tools/thermal/tmon/tmon.c - kernel/lockdown - Git at Google

 /*
  * 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;

 	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;
 	}

 	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();
 		}
 		cur_thermal_record++;
 		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(0);

 	/* 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);

 }
	/*
	* 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;

	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;
	}

	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();
	}
	cur_thermal_record++;
	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(0);

	/* 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);

	}