vcs.maemo.org Git - monky/blob - src/hddtemp.c

   1 /* Conky, a system monitor, based on torsmo
   2  *
   3  * Any original torsmo code is licensed under the BSD license
   4  *
   5  * All code written since the fork of torsmo is licensed under the GPL
   6  *
   7  * Please see COPYING for details
   8  *
   9  * Copyright (c) 2004, Hannu Saransaari and Lauri Hakkarainen
  10  * Copyright (c) 2005-2008 Brenden Matthews, Philip Kovacs, et. al.
  11  *      (see AUTHORS)
  12  * All rights reserved.
  13  *
  14  * This program is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU General Public License as published by
  16  * the Free Software Foundation, either version 3 of the License, or
  17  * (at your option) any later version.
  18  *
  19  * This program is distributed in the hope that it will be useful,
  20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22  * GNU General Public License for more details.
  23  * You should have received a copy of the GNU General Public License
  24  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  25  *
  26  * $Id$ */
  27
  28 #include "conky.h"
  29 #include <errno.h>
  30 #include <unistd.h>
  31 #include <fcntl.h>
  32 #include <netdb.h>
  33 #include <sys/select.h>
  34 #include <sys/socket.h>
  35 #include <netinet/in.h>
  36
  37 #define BUFLEN 512
  38 #define PORT 7634
  39
  40 char buf[BUFLEN];
  41
  42 int scan_hddtemp(const char *arg, char **dev, char **addr, int *port, char** temp)
  43 {
  44         char buf1[32], buf2[64];
  45         int n, ret;
  46
  47         ret = sscanf(arg, "%31s %63s %d", buf1, buf2, &n);
  48
  49         if (ret < 1) {
  50                 return -1;
  51         }
  52
  53         if (strncmp(buf1, "/dev/", 5)) {
  54                 strncpy(buf1 + 5, buf1, 32 - 5);
  55                 strncpy(buf1, "/dev/", 5);
  56         }
  57         *dev = strndup(buf1, text_buffer_size);
  58
  59         if (ret >= 2) {
  60                 *addr = strndup(buf2, text_buffer_size);
  61         } else {
  62                 *addr = strndup("127.0.0.1", text_buffer_size);
  63         }
  64
  65         if (ret == 3) {
  66                 *port = n;
  67         } else {
  68                 *port = PORT;
  69         }
  70
  71         *temp = malloc(text_buffer_size);
  72         memset(*temp, 0, text_buffer_size);
  73
  74         return 0;
  75 }
  76
  77 char *get_hddtemp_info(char *dev, char *hostaddr, int port, char *unit)
  78 {
  79         int sockfd = 0;
  80         struct hostent he, *he_res = 0;
  81         int he_errno;
  82         char hostbuff[2048];
  83         struct sockaddr_in addr;
  84         struct timeval tv;
  85         fd_set rfds;
  86         int len, i, devlen = strlen(dev);
  87         char sep;
  88         char *p, *out, *r = NULL;
  89
  90         if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
  91                 perror("socket");
  92                 return NULL;
  93         }
  94
  95         do {
  96 #ifdef HAVE_GETHOSTBYNAME_R
  97                 if (gethostbyname_r(hostaddr, &he, hostbuff, sizeof(hostbuff), &he_res, &he_errno)) {   // get the host info
  98                         ERR("hddtemp gethostbyname_r: %s", hstrerror(h_errno));
  99                         break;
 100                 }
 101 #else /* HAVE_GETHOSTBYNAME_R */
 102                 he_res = gethostbyname(hostaddr);
 103                 if (!he_res) {
 104                         perror("gethostbyname");
 105                         break;
 106                 }
 107 #endif /* HAVE_GETHOSTBYNAME_R */
 108
 109                 addr.sin_family = AF_INET;
 110                 addr.sin_port = htons(port);
 111                 addr.sin_addr = *((struct in_addr *) he_res->h_addr);
 112                 memset(&(addr.sin_zero), 0, 8);
 113
 114                 if (connect(sockfd, (struct sockaddr *) &addr,
 115                                         sizeof(struct sockaddr)) == -1) {
 116                         perror("connect");
 117                         break;
 118                 }
 119
 120                 FD_ZERO(&rfds);
 121                 FD_SET(sockfd, &rfds);
 122
 123                 /* We're going to wait up to a quarter a second to see whether there's
 124                  * any data available. Polling with timeout set to 0 doesn't seem to work
 125                  * with hddtemp. */
 126                 tv.tv_sec = 0;
 127                 tv.tv_usec = 250000;
 128
 129                 i = select(sockfd + 1, &rfds, NULL, NULL, &tv);
 130                 if (i == -1) {
 131                         if (errno == EINTR) {   /* silently ignore interrupted system call */
 132                                 break;
 133                         } else {
 134                                 perror("select");
 135                         }
 136                 }
 137
 138                 /* No data available */
 139                 if (i <= 0) {
 140                         break;
 141                 }
 142
 143                 p = buf;
 144                 len = 0;
 145                 do {
 146                         i = recv(sockfd, p, BUFLEN - (p - buf), 0);
 147                         if (i < 0) {
 148                                 perror("recv");
 149                                 break;
 150                         }
 151                         len += i;
 152                         p += i;
 153                 } while (i > 0 && p < buf + BUFLEN - 1);
 154
 155                 if (len < 2) {
 156                         break;
 157                 }
 158
 159                 buf[len] = 0;
 160
 161                 /* The first character read is the separator. */
 162                 sep = buf[0];
 163                 p = buf + 1;
 164
 165                 while (*p) {
 166                         if (!strncmp(p, dev, devlen)) {
 167                                 p += devlen + 1;
 168                                 p = strchr(p, sep);
 169                                 if (!p) {
 170                                         break;
 171                                 }
 172                                 p++;
 173                                 out = p;
 174                                 p = strchr(p, sep);
 175                                 if (!p) {
 176                                         break;
 177                                 }
 178                                 *p = '\0';
 179                                 p++;
 180                                 *unit = *p;
 181                                 if (!strncmp(out, "NA", 2)) {
 182                                         strncpy(buf, "N/A", BUFLEN);
 183                                         r = buf;
 184                                 } else {
 185                                         r = out;
 186                                 }
 187                                 break;
 188                         } else {
 189                                 for (i = 0; i < 5; i++) {
 190                                         p = strchr(p, sep);
 191                                         if (!p) {
 192                                                 break;
 193                                         }
 194                                         p++;
 195                                 }
 196                                 if (!p && i < 5) {
 197                                         break;
 198                                 }
 199                         }
 200                 }
 201         } while (0);
 202         close(sockfd);
 203         return r;
 204 }