src/lib/hdd.c

   1 /*
   2     Copyright (C) 2010-2011 wpitchoune@gmail.com
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU 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
  20 #include <locale.h>
  21 #include <libintl.h>
  22 #define _(str) gettext(str)
  23
  24 /*
  25   Following code is based on GNOME sensors applet code hddtemp-plugin.c
  26   see http://sensors-applet.sourceforge.net/
  27 */
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <stdio.h>
  31 #include <sys/types.h>
  32 #include <sys/socket.h>
  33 #include <netinet/in.h>
  34 #include <arpa/inet.h>
  35 #include <unistd.h>
  36
  37 #include "psensor.h"
  38
  39 #define HDDTEMP_SERVER_IP_ADDRESS "127.0.0.1"
  40 #define HDDTEMP_PORT_NUMBER 7634
  41 #define HDDTEMP_OUTPUT_BUFFER_LENGTH 4048
  42
  43 struct hdd_info {
  44         char *name;
  45         int temp;
  46 };
  47
  48 char *hdd_fetch()
  49 {
  50         int sockfd;
  51         ssize_t n = 1;
  52         int output_length = 0;
  53         char *pc;
  54         char *buffer;
  55         struct sockaddr_in address;
  56
  57         sockfd = socket(AF_INET, SOCK_STREAM, 0);
  58         if (sockfd == -1) {
  59                 fprintf(stderr, _("ERROR: hdd_fetch, failed to open socket\n"));
  60                 return NULL;
  61         }
  62
  63         address.sin_family = AF_INET;
  64         address.sin_addr.s_addr = inet_addr(HDDTEMP_SERVER_IP_ADDRESS);
  65         address.sin_port = htons(HDDTEMP_PORT_NUMBER);
  66
  67         buffer = NULL;
  68
  69         if (connect(sockfd,
  70                     (struct sockaddr *)&address,
  71                     (socklen_t) sizeof(address)) == -1) {
  72                 fprintf(stderr,
  73                         _("ERROR: hdd_fetch, failed to open connection\n"));
  74         } else {
  75                 buffer = malloc(HDDTEMP_OUTPUT_BUFFER_LENGTH);
  76
  77                 pc = buffer;
  78                 while ((n = read(sockfd,
  79                                  pc,
  80                                  HDDTEMP_OUTPUT_BUFFER_LENGTH -
  81                                  output_length)) > 0) {
  82
  83                         output_length += n;
  84                         pc = &pc[n];
  85                 }
  86
  87                 buffer[output_length] = '\0';
  88         }
  89
  90         close(sockfd);
  91
  92         return buffer;
  93 }
  94
  95 int str_index(char *str, char d)
  96 {
  97         char *c;
  98         int i;
  99
 100         if (!str || *str == '\0')
 101                 return -1;
 102
 103         c = str;
 104
 105         i = 0;
 106         while (*c) {
 107                 if (*c == d)
 108                         return i;
 109                 i++;
 110                 c++;
 111         }
 112
 113         return -1;
 114 }
 115
 116 struct psensor *hdd_create_sensor(char *id, char *name, int values_max_length)
 117 {
 118         return psensor_create(id, name, SENSOR_TYPE_HDD_TEMP,
 119                               values_max_length);
 120 }
 121
 122 char *next_hdd_info(char *string, struct hdd_info *info)
 123 {
 124         char *c;
 125         int idx_name_n, i, temp;
 126
 127         if (!string || strlen(string) <= 5      /* at least 5 pipes */
 128             || string[0] != '|')
 129                 return NULL;
 130
 131         /* skip first pipe */
 132         c = string + 1;
 133
 134         /* name */
 135         idx_name_n = str_index(c, '|');
 136
 137         if (idx_name_n == -1)
 138                 return NULL;
 139         c = c + idx_name_n + 1;
 140
 141         /* skip label */
 142         i = str_index(c, '|');
 143         if (i == -1)
 144                 return NULL;
 145         c = c + i + 1;
 146
 147         /* temp */
 148         i = str_index(c, '|');
 149         if (i == -1)
 150                 return NULL;
 151         temp = atoi(c);
 152         c = c + i + 1;
 153
 154         /* skip unit  */
 155         i = str_index(c, '|');
 156         if (i == -1)
 157                 return NULL;
 158         c = c + i + 1;
 159
 160         info->name = malloc(idx_name_n + 1);
 161         strncpy(info->name, string + 1, idx_name_n);
 162         info->name[idx_name_n] = '\0';
 163
 164         info->temp = temp;
 165
 166         return c;
 167 }
 168
 169 struct psensor **hdd_psensor_list_add(struct psensor **sensors,
 170                                       int values_max_length)
 171 {
 172         char *hddtemp_output = hdd_fetch();
 173         char *c;
 174         struct hdd_info info;
 175         struct psensor **result;
 176
 177         if (!hddtemp_output)
 178                 return sensors;
 179
 180         if (hddtemp_output[0] != '|') {
 181                 fprintf(stderr,
 182                         _("ERROR: wrong hdd string: %s"), hddtemp_output);
 183
 184                 free(hddtemp_output);
 185
 186                 return sensors;
 187         }
 188
 189         c = hddtemp_output;
 190
 191         result = sensors;
 192
 193         while (c && (c = next_hdd_info(c, &info))) {
 194                 struct psensor *sensor;
 195                 struct psensor **tmp_sensors;
 196
 197                 char *id = malloc(strlen("hdd ") + strlen(info.name) + 1);
 198                 strcpy(id, "hdd ");
 199                 strcat(id, info.name);
 200
 201                 sensor = hdd_create_sensor(id, info.name, values_max_length);
 202
 203                 tmp_sensors = psensor_list_add(result, sensor);
 204
 205                 if (result != sensors)
 206                         free(result);
 207
 208                 result = tmp_sensors;
 209         }
 210
 211         free(hddtemp_output);
 212
 213         return result;
 214 }
 215
 216 void hdd_psensor_update(struct psensor **sensors, struct hdd_info *info)
 217 {
 218         struct psensor **sensor_cur = sensors;
 219
 220         while (*sensor_cur) {
 221                 if ((*sensor_cur)->type == SENSOR_TYPE_HDD_TEMP
 222                     && !strcmp((*sensor_cur)->id + 4, info->name))
 223                         psensor_set_current_value(*sensor_cur,
 224                                                   (float)info->temp);
 225
 226                 sensor_cur++;
 227         }
 228 }
 229
 230 void hdd_psensor_list_update(struct psensor **sensors)
 231 {
 232         char *hddtemp_output = hdd_fetch();
 233
 234         if (!hddtemp_output)
 235                 return;
 236
 237         if (hddtemp_output[0] == '|') {
 238
 239                 char *c = hddtemp_output;
 240                 struct hdd_info info;
 241                 info.name = NULL;
 242                 info.temp = 0;
 243
 244                 while (c && (c = next_hdd_info(c, &info))) {
 245
 246                         hdd_psensor_update(sensors, &info);
 247
 248                         free(info.name);
 249                 }
 250         } else {
 251                 fprintf(stderr,
 252                         _("ERROR: wrong hdd string: %s\n"), hddtemp_output);
 253         }
 254
 255         free(hddtemp_output);
 256 }