tools/thermal/tmon/tmon.c

   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;
  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
  74 void version()
  75 {
  76         printf("TMON version %s\n", VERSION);
  77         exit(EXIT_SUCCESS);
  78 }
  79
  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
 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
 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
 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
 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                 cur_thermal_record++;
 340                 time_elapsed += ticktime;
 341                 controller_handler(trec[0].temp[target_tz_index] / 1000,
 342                                 &yk);
 343                 trec[0].pid_out_pct = yk;
 344                 if (!dialogue_on)
 345                         show_control_w();
 346                 if (tmon_exit)
 347                         break;
 348         }
 349         tmon_cleanup();
 350         return 0;
 351 }
 352
 353 static void start_daemon_mode()
 354 {
 355         daemon_mode = 1;
 356         /* fork */
 357         pid_t   sid, pid = fork();
 358         if (pid < 0) {
 359                 exit(EXIT_FAILURE);
 360         } else if (pid > 0)
 361                 /* kill parent */
 362                 exit(EXIT_SUCCESS);
 363
 364         /* disable TUI, it may not be necessary, but saves some resource */
 365         disable_tui();
 366
 367         /* change the file mode mask */
 368         umask(S_IWGRP | S_IWOTH);
 369
 370         /* new SID for the daemon process */
 371         sid = setsid();
 372         if (sid < 0)
 373                 exit(EXIT_FAILURE);
 374
 375         /* change working directory */
 376         if ((chdir("/")) < 0)
 377                 exit(EXIT_FAILURE);
 378
 379
 380         sleep(10);
 381
 382         close(STDIN_FILENO);
 383         close(STDOUT_FILENO);
 384         close(STDERR_FILENO);
 385
 386 }