src/lib/psensor.c

   1 /*
   2  * Copyright (C) 2010-2012 jeanfi@gmail.com
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License as
   6  * published by the Free Software Foundation; either version 2 of the
   7  * License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  17  * 02110-1301 USA
  18  */
  19 #include <stdlib.h>
  20 #include <string.h>
  21
  22 #include <locale.h>
  23 #include <libintl.h>
  24 #define _(str) gettext(str)
  25
  26 #include "hdd.h"
  27 #include "psensor.h"
  28 #include "lmsensor.h"
  29
  30 #ifdef HAVE_GTOP
  31 #include "cpu.h"
  32 #endif
  33
  34
  35 struct psensor *psensor_create(char *id, char *name,
  36                                unsigned int type, int values_max_length)
  37 {
  38         struct psensor *psensor
  39             = (struct psensor *)malloc(sizeof(struct psensor));
  40
  41         psensor->id = id;
  42         psensor->name = name;
  43         psensor->enabled = 1;
  44         psensor->min = UNKNOWN_DBL_VALUE;
  45         psensor->max = UNKNOWN_DBL_VALUE;
  46
  47         psensor->type = type;
  48
  49         psensor->values_max_length = values_max_length;
  50         psensor->measures = measures_dbl_create(values_max_length);
  51
  52         psensor->alarm_limit = 0;
  53
  54         psensor->cb_alarm_raised = NULL;
  55         psensor->cb_alarm_raised_data = NULL;
  56         psensor->alarm_raised = 0;
  57
  58         psensor->alarm_enabled = 0;
  59
  60         psensor->url = NULL;
  61
  62         psensor->color = NULL;
  63
  64         return psensor;
  65 }
  66
  67 void psensor_values_resize(struct psensor *s, int new_size)
  68 {
  69         struct measure *new_ms, *cur_ms;
  70         int cur_size;
  71
  72         cur_size = s->values_max_length;
  73         cur_ms = s->measures;
  74         new_ms = measures_dbl_create(new_size);
  75
  76         if (cur_ms) {
  77                 int i;
  78                 for (i = 0; i < new_size - 1 && i < cur_size - 1; i++)
  79                         measure_copy(&cur_ms[cur_size - i - 1],
  80                                      &new_ms[new_size - i - 1]);
  81
  82                 measures_free(s->measures);
  83         }
  84
  85         s->values_max_length = new_size;
  86         s->measures = new_ms;
  87 }
  88
  89 void psensor_free(struct psensor *sensor)
  90 {
  91         if (sensor) {
  92                 log_debug("Cleanup %s", sensor->id);
  93
  94                 free(sensor->name);
  95                 free(sensor->id);
  96
  97                 if (sensor->color)
  98                         free(sensor->color);
  99
 100                 measures_free(sensor->measures);
 101
 102                 free(sensor->url);
 103
 104                 free(sensor);
 105         }
 106 }
 107
 108 void psensor_list_free(struct psensor **sensors)
 109 {
 110         struct psensor **sensor_cur;
 111
 112         if (sensors) {
 113                 sensor_cur = sensors;
 114
 115                 while (*sensor_cur) {
 116                         psensor_free(*sensor_cur);
 117
 118                         sensor_cur++;
 119                 }
 120
 121                 free(sensors);
 122
 123                 sensors = NULL;
 124         }
 125 }
 126
 127 int psensor_list_size(struct psensor **sensors)
 128 {
 129         int size;
 130         struct psensor **sensor_cur;
 131
 132         if (!sensors)
 133                 return 0;
 134
 135         size = 0;
 136         sensor_cur = sensors;
 137
 138         while (*sensor_cur) {
 139                 size++;
 140                 sensor_cur++;
 141         }
 142         return size;
 143 }
 144
 145 int psensor_list_contains_type(struct psensor **sensors, unsigned int type)
 146 {
 147         struct psensor **s;
 148
 149         if (!sensors)
 150                 return 0;
 151
 152         s = sensors;
 153         while (*s) {
 154                 if ((*s)->type == type)
 155                         return 1;
 156                 s++;
 157         }
 158
 159         return 0;
 160 }
 161
 162 struct psensor **psensor_list_add(struct psensor **sensors,
 163                                   struct psensor *sensor)
 164 {
 165         int size = psensor_list_size(sensors);
 166
 167         struct psensor **result
 168             = malloc((size + 1 + 1) * sizeof(struct psensor *));
 169
 170         if (sensors)
 171                 memcpy(result, sensors, size * sizeof(struct psensor *));
 172
 173         result[size] = sensor;
 174         result[size + 1] = NULL;
 175
 176         return result;
 177 }
 178
 179 struct psensor *psensor_list_get_by_id(struct psensor **sensors, const char *id)
 180 {
 181         struct psensor **sensors_cur = sensors;
 182
 183         while (*sensors_cur) {
 184                 if (!strcmp((*sensors_cur)->id, id))
 185                         return *sensors_cur;
 186
 187                 sensors_cur++;
 188         }
 189
 190         return NULL;
 191 }
 192
 193 int is_temp_type(unsigned int type)
 194 {
 195         return type & SENSOR_TYPE_TEMP;
 196 }
 197
 198 int is_fan_type(unsigned int type)
 199 {
 200         return type & SENSOR_TYPE_FAN;
 201 }
 202
 203 double celcius_to_fahrenheit(double c)
 204 {
 205         return c * (9.0/5.0) + 32;
 206 }
 207
 208 double fahrenheit_to_celcius(double f)
 209 {
 210         return (f - 32) * (5.0/9.0);
 211 }
 212
 213 char *
 214 psensor_value_to_str(unsigned int type, double value, int use_celcius)
 215 {
 216         /* should not be possible to exceed 20 characters with temp or
 217            rpm values the .x part is never displayed */
 218         char *str = malloc(20);
 219
 220         char *unit;
 221
 222         if (is_temp_type(type))
 223                 if (use_celcius) {
 224                         unit = "°C";
 225                 } else {
 226                         unit = "°F";
 227                         value = celcius_to_fahrenheit(value);
 228                 }
 229         else if (type & SENSOR_TYPE_CPU_USAGE)
 230                 unit = "%";
 231         else
 232                 unit = "";
 233
 234         sprintf(str, "%.0f%s", value, unit);
 235
 236         return str;
 237 }
 238
 239 void psensor_set_current_value(struct psensor *sensor, double value)
 240 {
 241         struct timeval tv;
 242
 243         if (gettimeofday(&tv, NULL) != 0)
 244                 timerclear(&tv);
 245
 246         psensor_set_current_measure(sensor, value, tv);
 247 }
 248
 249 void
 250 psensor_set_current_measure(struct psensor *s,
 251                             double v, struct timeval tv)
 252 {
 253         memmove(s->measures,
 254                 &s->measures[1],
 255                 (s->values_max_length - 1) * sizeof(struct measure));
 256
 257         s->measures[s->values_max_length - 1].value = v;
 258         s->measures[s->values_max_length - 1].time = tv;
 259
 260         if (s->min == UNKNOWN_DBL_VALUE || v < s->min)
 261                 s->min = v;
 262
 263         if (s->max == UNKNOWN_DBL_VALUE || v > s->max)
 264                 s->max = v;
 265
 266         if (s->alarm_limit && s->alarm_enabled) {
 267                 if (v > s->alarm_limit) {
 268                         if (!s->alarm_raised && s->cb_alarm_raised)
 269                                 s->cb_alarm_raised(s,
 270                                                    s->cb_alarm_raised_data);
 271
 272                         s->alarm_raised = 1;
 273                 } else {
 274                         s->alarm_raised = 0;
 275                 }
 276         }
 277 }
 278
 279 double psensor_get_current_value(struct psensor *sensor)
 280 {
 281         return sensor->measures[sensor->values_max_length - 1].value;
 282 }
 283
 284 struct measure *psensor_get_current_measure(struct psensor *sensor)
 285 {
 286         return &sensor->measures[sensor->values_max_length - 1];
 287 }
 288
 289 /*
 290   Returns the minimal value of a given 'type' (SENSOR_TYPE_TEMP or
 291   SENSOR_TYPE_FAN)
 292  */
 293 double get_min_value(struct psensor **sensors, int type)
 294 {
 295         double m = UNKNOWN_DBL_VALUE;
 296         struct psensor **s = sensors;
 297
 298         while (*s) {
 299                 struct psensor *sensor = *s;
 300
 301                 if (sensor->enabled && (sensor->type & type)) {
 302                         int i;
 303                         double t;
 304
 305                         for (i = 0; i < sensor->values_max_length; i++) {
 306                                 t = sensor->measures[i].value;
 307
 308                                 if (t == UNKNOWN_DBL_VALUE)
 309                                         continue;
 310
 311                                 if (m == UNKNOWN_DBL_VALUE || t < m)
 312                                         m = t;
 313                         }
 314                 }
 315                 s++;
 316         }
 317
 318         return m;
 319 }
 320
 321 /*
 322   Returns the maximal value of a given 'type' (SENSOR_TYPE_TEMP or
 323   SENSOR_TYPE_FAN)
 324  */
 325 double get_max_value(struct psensor **sensors, int type)
 326 {
 327         double m = UNKNOWN_DBL_VALUE;
 328         struct psensor **s = sensors;
 329
 330         while (*s) {
 331                 struct psensor *sensor = *s;
 332
 333                 if (sensor->enabled && (sensor->type & type)) {
 334                         int i;
 335                         double t;
 336                         for (i = 0; i < sensor->values_max_length; i++) {
 337                                 t = sensor->measures[i].value;
 338
 339                                 if (t == UNKNOWN_DBL_VALUE)
 340                                         continue;
 341
 342                                 if (m == UNKNOWN_DBL_VALUE || t > m)
 343                                         m = t;
 344                         }
 345                 }
 346                 s++;
 347         }
 348
 349         return m;
 350 }
 351
 352 double
 353 psensor_get_max_current_value(struct psensor **sensors, unsigned int type)
 354 {
 355         double m = UNKNOWN_DBL_VALUE;
 356         struct psensor **s_cur = sensors;
 357
 358         while (*s_cur) {
 359                 struct psensor *s = *s_cur;
 360
 361                 if (s->enabled && (s->type & type)) {
 362                         double v = psensor_get_current_value(s);
 363
 364                         if (m == UNKNOWN_DBL_VALUE || v > m)
 365                                 m = v;
 366                 }
 367
 368                 s_cur++;
 369         }
 370
 371         return m;
 372 }
 373
 374 double get_min_temp(struct psensor **sensors)
 375 {
 376         return get_min_value(sensors, SENSOR_TYPE_TEMP);
 377 }
 378
 379 double get_min_rpm(struct psensor **sensors)
 380 {
 381         return get_min_value(sensors, SENSOR_TYPE_FAN);
 382 }
 383
 384 double get_max_rpm(struct psensor **sensors)
 385 {
 386         return get_max_value(sensors, SENSOR_TYPE_FAN);
 387 }
 388
 389 double get_max_temp(struct psensor **sensors)
 390 {
 391         return get_max_value(sensors, SENSOR_TYPE_TEMP);
 392 }
 393
 394 struct psensor **get_all_sensors(int use_libatasmart, int values_max_length)
 395 {
 396         struct psensor **psensors = NULL;
 397         struct psensor **tmp_psensors;
 398
 399         psensors = lmsensor_psensor_list_add(NULL, values_max_length);
 400
 401         if (!use_libatasmart) {
 402                 tmp_psensors = hddtemp_psensor_list_add(psensors,
 403                                                         values_max_length);
 404                 if (tmp_psensors != psensors) {
 405                         free(psensors);
 406                         psensors = tmp_psensors;
 407                 }
 408         }
 409 #ifdef HAVE_ATASMART
 410                 else {
 411                         tmp_psensors = hdd_psensor_list_add(psensors,
 412                                                             values_max_length);
 413                         if (tmp_psensors != psensors) {
 414                                 free(psensors);
 415                                 psensors = tmp_psensors;
 416                         }
 417                 }
 418 #endif
 419
 420         if (!psensors) {        /* there is no detected sensors */
 421                 psensors = malloc(sizeof(struct psensor *));
 422                 *psensors = NULL;
 423         }
 424
 425         return psensors;
 426 }
 427
 428 const char *psensor_type_to_str(unsigned int type)
 429 {
 430         if ((type & SENSOR_TYPE_NVIDIA_TEMP) == SENSOR_TYPE_NVIDIA_TEMP)
 431                 return "NVidia GPU Temperature";
 432
 433         if ((type & SENSOR_TYPE_AMD_TEMP) == SENSOR_TYPE_AMD_TEMP)
 434                 return "AMD GPU Temperature";
 435
 436         if ((type & SENSOR_TYPE_AMD_FAN) == SENSOR_TYPE_AMD_FAN)
 437                 return "AMD GPU Fan Speed";
 438
 439         if ((type & SENSOR_TYPE_HDD_TEMP) == SENSOR_TYPE_HDD_TEMP)
 440                 return "HDD Temperature";
 441
 442         if (type & SENSOR_TYPE_CPU_USAGE)
 443                 return "CPU Usage";
 444
 445         if (type & SENSOR_TYPE_TEMP)
 446                 return "Temperature";
 447
 448         if (type & SENSOR_TYPE_FAN)
 449                 return "Fan";
 450
 451         if (type & SENSOR_TYPE_REMOTE)
 452                 return "Remote";
 453
 454         return "N/A";           /* should not be possible */
 455 }
 456
 457
 458 const char *psensor_type_to_unit_str(unsigned int type, int use_celcius)
 459 {
 460         if (type & SENSOR_TYPE_TEMP) {
 461                 if (use_celcius)
 462                         return _("°C");
 463                 else
 464                         return _("°F");
 465         }
 466
 467         if (type & SENSOR_TYPE_FAN)
 468                 return _("RPM");
 469
 470         if (type & SENSOR_TYPE_CPU_USAGE)
 471                 return _("%");
 472
 473         return "N/A";
 474 }
 475
 476 void psensor_list_update_measures(struct psensor **sensors)
 477 {
 478         lmsensor_psensor_list_update(sensors);
 479
 480 #ifdef HAVE_GTOP
 481         cpu_psensor_list_update(sensors);
 482 #endif
 483
 484         if (psensor_list_contains_type(sensors, SENSOR_TYPE_HDD_TEMP_HDDTEMP))
 485                 hddtemp_psensor_list_update(sensors);
 486
 487 #ifdef HAVE_ATASMART
 488         if (psensor_list_contains_type(sensors,
 489                                        SENSOR_TYPE_HDD_TEMP_ATASMART))
 490                 hdd_psensor_list_update(sensors);
 491 #endif
 492 }
 493
 494 void psensor_log_measures(struct psensor **sensors)
 495 {
 496         if (log_level == LOG_DEBUG)
 497                 while (*sensors) {
 498                         log_debug("Measure: %s %.2f",
 499                                    (*sensors)->name,
 500                                    psensor_get_current_value(*sensors));
 501
 502                         sensors++;
 503                 }
 504 }
 505
 506 void psensor_init()
 507 {
 508         lmsensor_init();
 509 }
 510
 511 void psensor_cleanup()
 512 {
 513         lmsensor_cleanup();
 514 }