• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * tmon.c Thermal Monitor (TMON) main function and entry point
3  *
4  * Copyright (C) 2012 Intel Corporation. All rights reserved.
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License version
8  * 2 or later as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * Author: Jacob Pan <jacob.jun.pan@linux.intel.com>
16  *
17  */
18 
19 #include <getopt.h>
20 #include <unistd.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <sys/types.h>
25 #include <sys/stat.h>
26 #include <ncurses.h>
27 #include <ctype.h>
28 #include <time.h>
29 #include <signal.h>
30 #include <limits.h>
31 #include <sys/time.h>
32 #include <pthread.h>
33 #include <math.h>
34 #include <stdarg.h>
35 #include <syslog.h>
36 
37 #include "tmon.h"
38 
39 unsigned long ticktime = 1; /* seconds */
40 unsigned long no_control = 1; /* monitoring only or use cooling device for
41 			       * temperature control.
42 			       */
43 double time_elapsed = 0.0;
44 unsigned long target_temp_user = 65; /* can be select by tui later */
45 int dialogue_on;
46 int tmon_exit;
47 static short	daemon_mode;
48 static int logging; /* for recording thermal data to a file */
49 static int debug_on;
50 FILE *tmon_log;
51 /*cooling device used for the PID controller */
52 char ctrl_cdev[CDEV_NAME_SIZE] = "None";
53 int target_thermal_zone; /* user selected target zone instance */
54 static void	start_daemon_mode(void);
55 
56 pthread_t event_tid;
57 pthread_mutex_t input_lock;
usage()58 void usage()
59 {
60 	printf("Usage: tmon [OPTION...]\n");
61 	printf("  -c, --control         cooling device in control\n");
62 	printf("  -d, --daemon          run as daemon, no TUI\n");
63 	printf("  -g, --debug           debug message in syslog\n");
64 	printf("  -h, --help            show this help message\n");
65 	printf("  -l, --log             log data to /var/tmp/tmon.log\n");
66 	printf("  -t, --time-interval   sampling time interval, > 1 sec.\n");
67 	printf("  -T, --target-temp     initial target temperature\n");
68 	printf("  -v, --version         show version\n");
69 	printf("  -z, --zone            target thermal zone id\n");
70 
71 	exit(0);
72 }
73 
version()74 void version()
75 {
76 	printf("TMON version %s\n", VERSION);
77 	exit(EXIT_SUCCESS);
78 }
79 
tmon_cleanup(void)80 static void tmon_cleanup(void)
81 {
82 
83 	syslog(LOG_INFO, "TMON exit cleanup\n");
84 	fflush(stdout);
85 	refresh();
86 	if (tmon_log)
87 		fclose(tmon_log);
88 	if (event_tid) {
89 		pthread_mutex_lock(&input_lock);
90 		pthread_cancel(event_tid);
91 		pthread_mutex_unlock(&input_lock);
92 		pthread_mutex_destroy(&input_lock);
93 	}
94 	closelog();
95 	/* relax control knobs, undo throttling */
96 	set_ctrl_state(0);
97 
98 	keypad(stdscr, FALSE);
99 	echo();
100 	nocbreak();
101 	close_windows();
102 	endwin();
103 	free_thermal_data();
104 
105 	exit(1);
106 }
107 
108 
tmon_sig_handler(int sig)109 static void tmon_sig_handler(int sig)
110 {
111 	syslog(LOG_INFO, "TMON caught signal %d\n", sig);
112 	refresh();
113 	switch (sig) {
114 	case SIGTERM:
115 		printf("sigterm, exit and clean up\n");
116 		fflush(stdout);
117 		break;
118 	case SIGKILL:
119 		printf("sigkill, exit and clean up\n");
120 		fflush(stdout);
121 		break;
122 	case SIGINT:
123 		printf("ctrl-c, exit and clean up\n");
124 		fflush(stdout);
125 		break;
126 	default:
127 		break;
128 	}
129 	tmon_exit = true;
130 }
131 
132 
start_syslog(void)133 static void start_syslog(void)
134 {
135 	if (debug_on)
136 		setlogmask(LOG_UPTO(LOG_DEBUG));
137 	else
138 		setlogmask(LOG_UPTO(LOG_ERR));
139 	openlog("tmon.log", LOG_CONS | LOG_PID | LOG_NDELAY, LOG_LOCAL0);
140 	syslog(LOG_NOTICE, "TMON started by User %d", getuid());
141 }
142 
prepare_logging(void)143 static void prepare_logging(void)
144 {
145 	int i;
146 	struct stat logstat;
147 
148 	if (!logging)
149 		return;
150 	/* open local data log file */
151 	tmon_log = fopen(TMON_LOG_FILE, "w+");
152 	if (!tmon_log) {
153 		syslog(LOG_ERR, "failed to open log file %s\n", TMON_LOG_FILE);
154 		return;
155 	}
156 
157 	if (lstat(TMON_LOG_FILE, &logstat) < 0) {
158 		syslog(LOG_ERR, "Unable to stat log file %s\n", TMON_LOG_FILE);
159 		fclose(tmon_log);
160 		tmon_log = NULL;
161 		return;
162 	}
163 
164 	/* The log file must be a regular file owned by us */
165 	if (S_ISLNK(logstat.st_mode)) {
166 		syslog(LOG_ERR, "Log file is a symlink.  Will not log\n");
167 		fclose(tmon_log);
168 		tmon_log = NULL;
169 		return;
170 	}
171 
172 	if (logstat.st_uid != getuid()) {
173 		syslog(LOG_ERR, "We don't own the log file.  Not logging\n");
174 		fclose(tmon_log);
175 		tmon_log = NULL;
176 		return;
177 	}
178 
179 
180 	fprintf(tmon_log, "#----------- THERMAL SYSTEM CONFIG -------------\n");
181 	for (i = 0; i < ptdata.nr_tz_sensor; i++) {
182 		char binding_str[33]; /* size of long + 1 */
183 		int j;
184 
185 		memset(binding_str, 0, sizeof(binding_str));
186 		for (j = 0; j < 32; j++)
187 			binding_str[j] = (ptdata.tzi[i].cdev_binding & 1<<j) ?
188 				'1' : '0';
189 
190 		fprintf(tmon_log, "#thermal zone %s%02d cdevs binding: %32s\n",
191 			ptdata.tzi[i].type,
192 			ptdata.tzi[i].instance,
193 			binding_str);
194 		for (j = 0; j <	ptdata.tzi[i].nr_trip_pts; j++) {
195 			fprintf(tmon_log, "#\tTP%02d type:%s, temp:%lu\n", j,
196 				trip_type_name[ptdata.tzi[i].tp[j].type],
197 				ptdata.tzi[i].tp[j].temp);
198 		}
199 
200 	}
201 
202 	for (i = 0; i <	ptdata.nr_cooling_dev; i++)
203 		fprintf(tmon_log, "#cooling devices%02d: %s\n",
204 			i, ptdata.cdi[i].type);
205 
206 	fprintf(tmon_log, "#---------- THERMAL DATA LOG STARTED -----------\n");
207 	fprintf(tmon_log, "Samples TargetTemp ");
208 	for (i = 0; i < ptdata.nr_tz_sensor; i++) {
209 		fprintf(tmon_log, "%s%d    ", ptdata.tzi[i].type,
210 			ptdata.tzi[i].instance);
211 	}
212 	for (i = 0; i <	ptdata.nr_cooling_dev; i++)
213 		fprintf(tmon_log, "%s%d ", ptdata.cdi[i].type,
214 			ptdata.cdi[i].instance);
215 
216 	fprintf(tmon_log, "\n");
217 }
218 
219 static struct option opts[] = {
220 	{ "control", 1, NULL, 'c' },
221 	{ "daemon", 0, NULL, 'd' },
222 	{ "time-interval", 1, NULL, 't' },
223 	{ "target-temp", 1, NULL, 'T' },
224 	{ "log", 0, NULL, 'l' },
225 	{ "help", 0, NULL, 'h' },
226 	{ "version", 0, NULL, 'v' },
227 	{ "debug", 0, NULL, 'g' },
228 	{ 0, 0, NULL, 0 }
229 };
230 
231 
main(int argc,char ** argv)232 int main(int argc, char **argv)
233 {
234 	int err = 0;
235 	int id2 = 0, c;
236 	double yk = 0.0, temp; /* controller output */
237 	int target_tz_index;
238 
239 	if (geteuid() != 0) {
240 		printf("TMON needs to be run as root\n");
241 		exit(EXIT_FAILURE);
242 	}
243 
244 	while ((c = getopt_long(argc, argv, "c:dlht:T:vgz:", opts, &id2)) != -1) {
245 		switch (c) {
246 		case 'c':
247 			no_control = 0;
248 			strncpy(ctrl_cdev, optarg, CDEV_NAME_SIZE);
249 			break;
250 		case 'd':
251 			start_daemon_mode();
252 			printf("Run TMON in daemon mode\n");
253 			break;
254 		case 't':
255 			ticktime = strtod(optarg, NULL);
256 			if (ticktime < 1)
257 				ticktime = 1;
258 			break;
259 		case 'T':
260 			temp = strtod(optarg, NULL);
261 			if (temp < 0) {
262 				fprintf(stderr, "error: temperature must be positive\n");
263 				return 1;
264 			}
265 			target_temp_user = temp;
266 			break;
267 		case 'l':
268 			printf("Logging data to /var/tmp/tmon.log\n");
269 			logging = 1;
270 			break;
271 		case 'h':
272 			usage();
273 			break;
274 		case 'v':
275 			version();
276 			break;
277 		case 'g':
278 			debug_on = 1;
279 			break;
280 		case 'z':
281 			target_thermal_zone = strtod(optarg, NULL);
282 			break;
283 		default:
284 			break;
285 		}
286 	}
287 	if (pthread_mutex_init(&input_lock, NULL) != 0) {
288 		fprintf(stderr, "\n mutex init failed, exit\n");
289 		return 1;
290 	}
291 	start_syslog();
292 	if (signal(SIGINT, tmon_sig_handler) == SIG_ERR)
293 		syslog(LOG_DEBUG, "Cannot handle SIGINT\n");
294 	if (signal(SIGTERM, tmon_sig_handler) == SIG_ERR)
295 		syslog(LOG_DEBUG, "Cannot handle SIGINT\n");
296 
297 	if (probe_thermal_sysfs()) {
298 		pthread_mutex_destroy(&input_lock);
299 		closelog();
300 		return -1;
301 	}
302 	initialize_curses();
303 	setup_windows();
304 	signal(SIGWINCH, resize_handler);
305 	show_title_bar();
306 	show_sensors_w();
307 	show_cooling_device();
308 	update_thermal_data();
309 	show_data_w();
310 	prepare_logging();
311 	init_thermal_controller();
312 
313 	nodelay(stdscr, TRUE);
314 	err = pthread_create(&event_tid, NULL, &handle_tui_events, NULL);
315 	if (err != 0) {
316 		printf("\ncan't create thread :[%s]", strerror(err));
317 		tmon_cleanup();
318 		exit(EXIT_FAILURE);
319 	}
320 
321 	/* validate range of user selected target zone, default to the first
322 	 * instance if out of range
323 	 */
324 	target_tz_index = zone_instance_to_index(target_thermal_zone);
325 	if (target_tz_index < 0) {
326 		target_thermal_zone = ptdata.tzi[0].instance;
327 		syslog(LOG_ERR, "target zone is not found, default to %d\n",
328 			target_thermal_zone);
329 	}
330 	while (1) {
331 		sleep(ticktime);
332 		show_title_bar();
333 		show_sensors_w();
334 		update_thermal_data();
335 		if (!dialogue_on) {
336 			show_data_w();
337 			show_cooling_device();
338 		}
339 		time_elapsed += ticktime;
340 		controller_handler(trec[0].temp[target_tz_index] / 1000,
341 				&yk);
342 		trec[0].pid_out_pct = yk;
343 		if (!dialogue_on)
344 			show_control_w();
345 		if (tmon_exit)
346 			break;
347 	}
348 	tmon_cleanup();
349 	return 0;
350 }
351 
start_daemon_mode()352 static void start_daemon_mode()
353 {
354 	daemon_mode = 1;
355 	/* fork */
356 	pid_t	sid, pid = fork();
357 	if (pid < 0) {
358 		exit(EXIT_FAILURE);
359 	} else if (pid > 0)
360 		/* kill parent */
361 		exit(EXIT_SUCCESS);
362 
363 	/* disable TUI, it may not be necessary, but saves some resource */
364 	disable_tui();
365 
366 	/* change the file mode mask */
367 	umask(S_IWGRP | S_IWOTH);
368 
369 	/* new SID for the daemon process */
370 	sid = setsid();
371 	if (sid < 0)
372 		exit(EXIT_FAILURE);
373 
374 	/* change working directory */
375 	if ((chdir("/")) < 0)
376 		exit(EXIT_FAILURE);
377 
378 
379 	sleep(10);
380 
381 	close(STDIN_FILENO);
382 	close(STDOUT_FILENO);
383 	close(STDERR_FILENO);
384 
385 }
386