| /* | 
 |  * tmon.c Thermal Monitor (TMON) main function and entry point | 
 |  * | 
 |  * Copyright (C) 2012 Intel Corporation. All rights reserved. | 
 |  * | 
 |  * This program is free software; you can redistribute it and/or | 
 |  * modify it under the terms of the GNU General Public License version | 
 |  * 2 or later as published by the Free Software Foundation. | 
 |  * | 
 |  * This program is distributed in the hope that it will be useful, | 
 |  * but WITHOUT ANY WARRANTY; without even the implied warranty of | 
 |  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
 |  * GNU General Public License for more details. | 
 |  * | 
 |  * Author: Jacob Pan <jacob.jun.pan@linux.intel.com> | 
 |  * | 
 |  */ | 
 |  | 
 | #include <getopt.h> | 
 | #include <unistd.h> | 
 | #include <stdio.h> | 
 | #include <stdlib.h> | 
 | #include <string.h> | 
 | #include <sys/types.h> | 
 | #include <sys/stat.h> | 
 | #include <ncurses.h> | 
 | #include <ctype.h> | 
 | #include <time.h> | 
 | #include <signal.h> | 
 | #include <limits.h> | 
 | #include <sys/time.h> | 
 | #include <pthread.h> | 
 | #include <math.h> | 
 | #include <stdarg.h> | 
 | #include <syslog.h> | 
 |  | 
 | #include "tmon.h" | 
 |  | 
 | unsigned long ticktime = 1; /* seconds */ | 
 | unsigned long no_control = 1; /* monitoring only or use cooling device for | 
 | 			       * temperature control. | 
 | 			       */ | 
 | double time_elapsed = 0.0; | 
 | unsigned long target_temp_user = 65; /* can be select by tui later */ | 
 | int dialogue_on; | 
 | int tmon_exit; | 
 | static short	daemon_mode; | 
 | static int logging; /* for recording thermal data to a file */ | 
 | static int debug_on; | 
 | FILE *tmon_log; | 
 | /*cooling device used for the PID controller */ | 
 | char ctrl_cdev[CDEV_NAME_SIZE] = "None"; | 
 | int target_thermal_zone; /* user selected target zone instance */ | 
 | static void	start_daemon_mode(void); | 
 |  | 
 | pthread_t event_tid; | 
 | pthread_mutex_t input_lock; | 
 | void usage() | 
 | { | 
 | 	printf("Usage: tmon [OPTION...]\n"); | 
 | 	printf("  -c, --control         cooling device in control\n"); | 
 | 	printf("  -d, --daemon          run as daemon, no TUI\n"); | 
 | 	printf("  -g, --debug           debug message in syslog\n"); | 
 | 	printf("  -h, --help            show this help message\n"); | 
 | 	printf("  -l, --log             log data to /var/tmp/tmon.log\n"); | 
 | 	printf("  -t, --time-interval   sampling time interval, > 1 sec.\n"); | 
 | 	printf("  -T, --target-temp     initial target temperature\n"); | 
 | 	printf("  -v, --version         show version\n"); | 
 | 	printf("  -z, --zone            target thermal zone id\n"); | 
 |  | 
 | 	exit(0); | 
 | } | 
 |  | 
 | void version() | 
 | { | 
 | 	printf("TMON version %s\n", VERSION); | 
 | 	exit(EXIT_SUCCESS); | 
 | } | 
 |  | 
 | static void tmon_cleanup(void) | 
 | { | 
 |  | 
 | 	syslog(LOG_INFO, "TMON exit cleanup\n"); | 
 | 	fflush(stdout); | 
 | 	refresh(); | 
 | 	if (tmon_log) | 
 | 		fclose(tmon_log); | 
 | 	if (event_tid) { | 
 | 		pthread_mutex_lock(&input_lock); | 
 | 		pthread_cancel(event_tid); | 
 | 		pthread_mutex_unlock(&input_lock); | 
 | 		pthread_mutex_destroy(&input_lock); | 
 | 	} | 
 | 	closelog(); | 
 | 	/* relax control knobs, undo throttling */ | 
 | 	set_ctrl_state(0); | 
 |  | 
 | 	keypad(stdscr, FALSE); | 
 | 	echo(); | 
 | 	nocbreak(); | 
 | 	close_windows(); | 
 | 	endwin(); | 
 | 	free_thermal_data(); | 
 |  | 
 | 	exit(1); | 
 | } | 
 |  | 
 |  | 
 | static void tmon_sig_handler(int sig) | 
 | { | 
 | 	syslog(LOG_INFO, "TMON caught signal %d\n", sig); | 
 | 	refresh(); | 
 | 	switch (sig) { | 
 | 	case SIGTERM: | 
 | 		printf("sigterm, exit and clean up\n"); | 
 | 		fflush(stdout); | 
 | 		break; | 
 | 	case SIGKILL: | 
 | 		printf("sigkill, exit and clean up\n"); | 
 | 		fflush(stdout); | 
 | 		break; | 
 | 	case SIGINT: | 
 | 		printf("ctrl-c, exit and clean up\n"); | 
 | 		fflush(stdout); | 
 | 		break; | 
 | 	default: | 
 | 		break; | 
 | 	} | 
 | 	tmon_exit = true; | 
 | } | 
 |  | 
 |  | 
 | static void start_syslog(void) | 
 | { | 
 | 	if (debug_on) | 
 | 		setlogmask(LOG_UPTO(LOG_DEBUG)); | 
 | 	else | 
 | 		setlogmask(LOG_UPTO(LOG_ERR)); | 
 | 	openlog("tmon.log", LOG_CONS | LOG_PID | LOG_NDELAY, LOG_LOCAL0); | 
 | 	syslog(LOG_NOTICE, "TMON started by User %d", getuid()); | 
 | } | 
 |  | 
 | static void prepare_logging(void) | 
 | { | 
 | 	int i; | 
 | 	struct stat logstat; | 
 |  | 
 | 	if (!logging) | 
 | 		return; | 
 | 	/* open local data log file */ | 
 | 	tmon_log = fopen(TMON_LOG_FILE, "w+"); | 
 | 	if (!tmon_log) { | 
 | 		syslog(LOG_ERR, "failed to open log file %s\n", TMON_LOG_FILE); | 
 | 		return; | 
 | 	} | 
 |  | 
 | 	if (lstat(TMON_LOG_FILE, &logstat) < 0) { | 
 | 		syslog(LOG_ERR, "Unable to stat log file %s\n", TMON_LOG_FILE); | 
 | 		fclose(tmon_log); | 
 | 		tmon_log = NULL; | 
 | 		return; | 
 | 	} | 
 |  | 
 | 	/* The log file must be a regular file owned by us */ | 
 | 	if (S_ISLNK(logstat.st_mode)) { | 
 | 		syslog(LOG_ERR, "Log file is a symlink.  Will not log\n"); | 
 | 		fclose(tmon_log); | 
 | 		tmon_log = NULL; | 
 | 		return; | 
 | 	} | 
 |  | 
 | 	if (logstat.st_uid != getuid()) { | 
 | 		syslog(LOG_ERR, "We don't own the log file.  Not logging\n"); | 
 | 		fclose(tmon_log); | 
 | 		tmon_log = NULL; | 
 | 		return; | 
 | 	} | 
 |  | 
 |  | 
 | 	fprintf(tmon_log, "#----------- THERMAL SYSTEM CONFIG -------------\n"); | 
 | 	for (i = 0; i < ptdata.nr_tz_sensor; i++) { | 
 | 		char binding_str[33]; /* size of long + 1 */ | 
 | 		int j; | 
 |  | 
 | 		memset(binding_str, 0, sizeof(binding_str)); | 
 | 		for (j = 0; j < 32; j++) | 
 | 			binding_str[j] = (ptdata.tzi[i].cdev_binding & 1<<j) ? | 
 | 				'1' : '0'; | 
 |  | 
 | 		fprintf(tmon_log, "#thermal zone %s%02d cdevs binding: %32s\n", | 
 | 			ptdata.tzi[i].type, | 
 | 			ptdata.tzi[i].instance, | 
 | 			binding_str); | 
 | 		for (j = 0; j <	ptdata.tzi[i].nr_trip_pts; j++) { | 
 | 			fprintf(tmon_log, "#\tTP%02d type:%s, temp:%lu\n", j, | 
 | 				trip_type_name[ptdata.tzi[i].tp[j].type], | 
 | 				ptdata.tzi[i].tp[j].temp); | 
 | 		} | 
 |  | 
 | 	} | 
 |  | 
 | 	for (i = 0; i <	ptdata.nr_cooling_dev; i++) | 
 | 		fprintf(tmon_log, "#cooling devices%02d: %s\n", | 
 | 			i, ptdata.cdi[i].type); | 
 |  | 
 | 	fprintf(tmon_log, "#---------- THERMAL DATA LOG STARTED -----------\n"); | 
 | 	fprintf(tmon_log, "Samples TargetTemp "); | 
 | 	for (i = 0; i < ptdata.nr_tz_sensor; i++) { | 
 | 		fprintf(tmon_log, "%s%d    ", ptdata.tzi[i].type, | 
 | 			ptdata.tzi[i].instance); | 
 | 	} | 
 | 	for (i = 0; i <	ptdata.nr_cooling_dev; i++) | 
 | 		fprintf(tmon_log, "%s%d ", ptdata.cdi[i].type, | 
 | 			ptdata.cdi[i].instance); | 
 |  | 
 | 	fprintf(tmon_log, "\n"); | 
 | } | 
 |  | 
 | static struct option opts[] = { | 
 | 	{ "control", 1, NULL, 'c' }, | 
 | 	{ "daemon", 0, NULL, 'd' }, | 
 | 	{ "time-interval", 1, NULL, 't' }, | 
 | 	{ "target-temp", 1, NULL, 'T' }, | 
 | 	{ "log", 0, NULL, 'l' }, | 
 | 	{ "help", 0, NULL, 'h' }, | 
 | 	{ "version", 0, NULL, 'v' }, | 
 | 	{ "debug", 0, NULL, 'g' }, | 
 | 	{ 0, 0, NULL, 0 } | 
 | }; | 
 |  | 
 |  | 
 | int main(int argc, char **argv) | 
 | { | 
 | 	int err = 0; | 
 | 	int id2 = 0, c; | 
 | 	double yk = 0.0, temp; /* controller output */ | 
 | 	int target_tz_index; | 
 |  | 
 | 	if (geteuid() != 0) { | 
 | 		printf("TMON needs to be run as root\n"); | 
 | 		exit(EXIT_FAILURE); | 
 | 	} | 
 |  | 
 | 	while ((c = getopt_long(argc, argv, "c:dlht:T:vgz:", opts, &id2)) != -1) { | 
 | 		switch (c) { | 
 | 		case 'c': | 
 | 			no_control = 0; | 
 | 			strncpy(ctrl_cdev, optarg, CDEV_NAME_SIZE); | 
 | 			break; | 
 | 		case 'd': | 
 | 			start_daemon_mode(); | 
 | 			printf("Run TMON in daemon mode\n"); | 
 | 			break; | 
 | 		case 't': | 
 | 			ticktime = strtod(optarg, NULL); | 
 | 			if (ticktime < 1) | 
 | 				ticktime = 1; | 
 | 			break; | 
 | 		case 'T': | 
 | 			temp = strtod(optarg, NULL); | 
 | 			if (temp < 0) { | 
 | 				fprintf(stderr, "error: temperature must be positive\n"); | 
 | 				return 1; | 
 | 			} | 
 | 			target_temp_user = temp; | 
 | 			break; | 
 | 		case 'l': | 
 | 			printf("Logging data to /var/tmp/tmon.log\n"); | 
 | 			logging = 1; | 
 | 			break; | 
 | 		case 'h': | 
 | 			usage(); | 
 | 			break; | 
 | 		case 'v': | 
 | 			version(); | 
 | 			break; | 
 | 		case 'g': | 
 | 			debug_on = 1; | 
 | 			break; | 
 | 		case 'z': | 
 | 			target_thermal_zone = strtod(optarg, NULL); | 
 | 			break; | 
 | 		default: | 
 | 			break; | 
 | 		} | 
 | 	} | 
 | 	if (pthread_mutex_init(&input_lock, NULL) != 0) { | 
 | 		fprintf(stderr, "\n mutex init failed, exit\n"); | 
 | 		return 1; | 
 | 	} | 
 | 	start_syslog(); | 
 | 	if (signal(SIGINT, tmon_sig_handler) == SIG_ERR) | 
 | 		syslog(LOG_DEBUG, "Cannot handle SIGINT\n"); | 
 | 	if (signal(SIGTERM, tmon_sig_handler) == SIG_ERR) | 
 | 		syslog(LOG_DEBUG, "Cannot handle SIGINT\n"); | 
 |  | 
 | 	if (probe_thermal_sysfs()) { | 
 | 		pthread_mutex_destroy(&input_lock); | 
 | 		closelog(); | 
 | 		return -1; | 
 | 	} | 
 | 	initialize_curses(); | 
 | 	setup_windows(); | 
 | 	signal(SIGWINCH, resize_handler); | 
 | 	show_title_bar(); | 
 | 	show_sensors_w(); | 
 | 	show_cooling_device(); | 
 | 	update_thermal_data(); | 
 | 	show_data_w(); | 
 | 	prepare_logging(); | 
 | 	init_thermal_controller(); | 
 |  | 
 | 	nodelay(stdscr, TRUE); | 
 | 	err = pthread_create(&event_tid, NULL, &handle_tui_events, NULL); | 
 | 	if (err != 0) { | 
 | 		printf("\ncan't create thread :[%s]", strerror(err)); | 
 | 		tmon_cleanup(); | 
 | 		exit(EXIT_FAILURE); | 
 | 	} | 
 |  | 
 | 	/* validate range of user selected target zone, default to the first | 
 | 	 * instance if out of range | 
 | 	 */ | 
 | 	target_tz_index = zone_instance_to_index(target_thermal_zone); | 
 | 	if (target_tz_index < 0) { | 
 | 		target_thermal_zone = ptdata.tzi[0].instance; | 
 | 		syslog(LOG_ERR, "target zone is not found, default to %d\n", | 
 | 			target_thermal_zone); | 
 | 	} | 
 | 	while (1) { | 
 | 		sleep(ticktime); | 
 | 		show_title_bar(); | 
 | 		show_sensors_w(); | 
 | 		update_thermal_data(); | 
 | 		if (!dialogue_on) { | 
 | 			show_data_w(); | 
 | 			show_cooling_device(); | 
 | 		} | 
 | 		time_elapsed += ticktime; | 
 | 		controller_handler(trec[0].temp[target_tz_index] / 1000, | 
 | 				&yk); | 
 | 		trec[0].pid_out_pct = yk; | 
 | 		if (!dialogue_on) | 
 | 			show_control_w(); | 
 | 		if (tmon_exit) | 
 | 			break; | 
 | 	} | 
 | 	tmon_cleanup(); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static void start_daemon_mode() | 
 | { | 
 | 	daemon_mode = 1; | 
 | 	/* fork */ | 
 | 	pid_t	sid, pid = fork(); | 
 | 	if (pid < 0) { | 
 | 		exit(EXIT_FAILURE); | 
 | 	} else if (pid > 0) | 
 | 		/* kill parent */ | 
 | 		exit(EXIT_SUCCESS); | 
 |  | 
 | 	/* disable TUI, it may not be necessary, but saves some resource */ | 
 | 	disable_tui(); | 
 |  | 
 | 	/* change the file mode mask */ | 
 | 	umask(S_IWGRP | S_IWOTH); | 
 |  | 
 | 	/* new SID for the daemon process */ | 
 | 	sid = setsid(); | 
 | 	if (sid < 0) | 
 | 		exit(EXIT_FAILURE); | 
 |  | 
 | 	/* change working directory */ | 
 | 	if ((chdir("/")) < 0) | 
 | 		exit(EXIT_FAILURE); | 
 |  | 
 |  | 
 | 	sleep(10); | 
 |  | 
 | 	close(STDIN_FILENO); | 
 | 	close(STDOUT_FILENO); | 
 | 	close(STDERR_FILENO); | 
 |  | 
 | } |