1 /*
2 * Copyright (C) 2002 Emmanuel VARAGNAT <hddtemp@guzu.net>
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 St, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 /*
20 * Added some SCSI support: Frederic LOCHON <lochon@roulaise.net>
21 *
22 */
23
24 // Include file generated by ./configure
25 #ifdef HAVE_CONFIG_H
26 #include <config.h>
27 #endif
28
29 // Gettext includes
30 #if ENABLE_NLS
31 #include <libintl.h>
32 #define _(String) gettext (String)
33 #else
34 #define _(String) (String)
35 #endif
36
37 // Standard includes
38 #include <unistd.h>
39 #include <stdlib.h>
40 #include <fcntl.h>
41 #include <stdio.h>
42 #include <string.h>
43 #include <sys/ioctl.h>
44 #include <sys/types.h>
45 #include <sys/socket.h>
46 #include <arpa/inet.h>
47 #include <netdb.h>
48 #include <getopt.h>
49 #include <signal.h>
50 #include <errno.h>
51 #include <locale.h>
52 #include <time.h>
53 #include <netinet/in.h>
54 #include <linux/hdreg.h>
55 #include <ctype.h>
56 #include <assert.h>
57
58 // Application specific includes
59 #include "ata.h"
60 #include "utf8.h"
61 #include "sata.h"
62 #include "scsi.h"
63 #include "db.h"
64 #include "hddtemp.h"
65 #include "backtrace.h"
66 #include "daemon.h"
67
68
69 #define PORT_NUMBER 7634
70 #define SEPARATOR '|'
71
72 char * database_path = DEFAULT_DATABASE_PATH;
73 long portnum, syslog_interval;
74 char * listen_addr;
75 char separator = SEPARATOR;
76
77 struct bustype * bus[BUS_TYPE_MAX];
78 int tcp_daemon, debug, quiet, numeric, wakeup, foreground, af_hint;
79
80 static enum { DEFAULT, CELSIUS, FAHRENHEIT } unit;
81
82 /*******************************************************
83 *******************************************************/
84
85 static void init_bus_types() {
86 bus[BUS_SATA] = &sata_bus;
87 bus[BUS_ATA] = &ata_bus;
88 bus[BUS_SCSI] = &scsi_bus;
89 }
90
91 /*******************************************************
92 *******************************************************/
93
94 int value_to_unit(struct disk *dsk) {
95 switch(unit) {
96 case CELSIUS:
97 if(dsk->db_entry->unit == 'F')
98 return F_to_C(dsk->value);
99 break;
100 case FAHRENHEIT:
101 if(dsk->db_entry->unit == 'C')
102 return C_to_F(dsk->value);
103 default:
104 break;
105 }
106
107 return dsk->value;
108 }
109
110 char get_unit(struct disk *dsk) {
111 switch(unit) {
112 case CELSIUS:
113 return 'C';
114 case FAHRENHEIT:
115 return 'F';
116 default:
117 return dsk->db_entry->unit;
118 }
119 }
120
121
122
123 static enum e_bustype probe_bus_type(struct disk *dsk) {
124 /* SATA disks answer to both ATA and SCSI commands so
125 they have to be probed first in order to be detected */
126 if(bus[BUS_SATA]->probe(dsk->fd))
127 return BUS_SATA;
128 else if(bus[BUS_ATA]->probe(dsk->fd))
129 return BUS_ATA;
130 else if(bus[BUS_SCSI]->probe(dsk->fd))
131 return BUS_SCSI;
132 else
133 return BUS_UNKNOWN;
134 }
135
136 /*
137 static int get_smart_threshold_values(int fd, unsigned char* buff) {
138 unsigned char cmd[516] = { WIN_SMART, 0, SMART_READ_THRESHOLDS, 1 };
139 int ret;
140
141 ret = ioctl(fd, HDIO_DRIVE_CMD, cmd);
142 if(ret)
143 return ret;
144
145 memcpy(buff, cmd+4, 512);
146 return 0;
147 }
148 */
149
150
151 static void display_temperature(struct disk *dsk) {
152 enum e_gettemp ret;
153 char *degree;
154
155 if(dsk->type != ERROR && debug ) {
156 printf(_("\n================= hddtemp %s ==================\n"
157 "Model: %s\n\n"), VERSION, dsk->model);
158 /* return;*/
159 }
160
161 if(dsk->type == ERROR
162 || bus[dsk->type]->get_temperature == NULL
163 || (ret = bus[dsk->type]->get_temperature(dsk)) == GETTEMP_ERROR )
164 {
165 fprintf(stderr, "%s: %s\n", dsk->drive, dsk->errormsg);
166 return;
167 }
168
169 if(debug)
170 return;
171
172 degree = degree_sign();
173 switch(ret) {
174 case GETTEMP_ERROR:
175 /* see above */
176 break;
177 case GETTEMP_NOT_APPLICABLE:
178
179 if (numeric && quiet)
180 printf("0\n");
181 else
182 printf("%s: %s: %s\n", dsk->drive, dsk->model, dsk->errormsg);
183
184 break;
185 case GETTEMP_UNKNOWN:
186
187 if(!quiet)
188 fprintf(stderr,
189 _("WARNING: Drive %s doesn't seem to have a temperature sensor.\n"
190 "WARNING: This doesn't mean it hasn't got one.\n"
191 "WARNING: If you are sure it has one, please contact me (hddtemp@guzu.net).\n"
192 "WARNING: See --help, --debug and --drivebase options.\n"), dsk->drive);
193
194 if (numeric && quiet)
195 printf("0\n");
196 else
197 fprintf(stderr, _("%s: %s: no sensor\n"), dsk->drive, dsk->model);
198
199 break;
200 case GETTEMP_KNOWN:
201
202 if (! numeric)
203 printf("%s: %s: %d%s%c\n",
204 dsk->drive,
205 dsk->model,
206 value_to_unit(dsk),
207 degree,
208 get_unit(dsk)
209 );
210 else
211 printf("%d\n", value_to_unit(dsk));
212
213 break;
214 case GETTEMP_DRIVE_SLEEP:
215
216 if (numeric && quiet)
217 printf("0\n");
218 else
219 fprintf(stderr, _("%s: %s: drive is sleeping\n"), dsk->drive, dsk->model);
220
221 break;
222 case GETTEMP_NOSENSOR:
223 if (numeric && quiet)
224 printf("0\n");
225 else
226 fprintf(stderr, _("%s: %s: drive supported, but it doesn't have a temperature sensor.\n"), dsk->drive, dsk->model);
227
228 break;
229 default:
230 fprintf(stderr, _("ERROR: %s: %s: unknown returned status\n"), dsk->drive, dsk->model);
231 break;
232 }
233 free(degree);
234 }
235
236
237 void do_direct_mode(struct disk *ldisks) {
238 struct disk *dsk;
239
240 for(dsk = ldisks; dsk; dsk = dsk->next) {
241 display_temperature(dsk);
242 }
243
244 if(debug) {
245 printf(_("\n"
246 "If one of the field value seems to match the temperature, be sure to read\n"
247 "the hddtemp man page before sending a report (section REPORT). Thanks.\n"
248 ));
249 }
250 }
251
252
253 int main(int argc, char* argv[]) {
254 int i, c, lindex = 0, db_loaded = 0;
255 int ret = 0;
256 int show_db;
257 struct disk * ldisks;
258
259 backtrace_sigsegv();
260 backtrace_sigill();
261 backtrace_sigbus();
262
263 setlocale (LC_ALL, "");
264 bindtextdomain (PACKAGE, LOCALEDIR);
265 textdomain (PACKAGE);
266
267 show_db = debug = numeric = quiet = wakeup = af_hint = syslog_interval = foreground = 0;
268 unit = DEFAULT;
269 portnum = PORT_NUMBER;
270 listen_addr = NULL;
271
272 /* Parse command line */
273 optind = 0;
274 while (1) {
275 static struct option long_options[] = {
276 {"help", 0, NULL, 'h'},
277 {"quiet", 0, NULL, 'q'},
278 {"daemon", 0, NULL, 'd'},
279 {"drivebase", 0, NULL, 'b'},
280 {"debug", 0, NULL, 'D'},
281 {"file", 1, NULL, 'f'},
282 {"foreground", 0, NULL, 'F'},
283 {"listen", 1, NULL, 'l'},
284 {"version", 0, NULL, 'v'},
285 {"port", 1, NULL, 'p'},
286 {"separator", 1, NULL, 's'},
287 {"numeric", 0, NULL, 'n'},
288 {"unit", 1, NULL, 'u'},
289 {"syslog", 1, NULL, 'S'},
290 {"wake-up", 0, NULL, 'w'},
291 {0, 0, 0, 0}
292 };
293
294 c = getopt_long (argc, argv, "bDdf:l:hp:qs:u:vnw46FS:", long_options, &lindex);
295 if (c == -1)
296 break;
297
298 switch (c) {
299 case 'q':
300 quiet = 1;
301 break;
302 case '4':
303 af_hint = AF_INET;
304 break;
305 case '6':
306 af_hint = AF_INET6;
307 break;
308 case 'b':
309 show_db = 1;
310 break;
311 case 'd':
312 tcp_daemon = 1;
313 break;
314 case 'D':
315 debug = 1;
316 break;
317 case 'f':
318 database_path = optarg;
319 break;
320 case 's':
321 separator = optarg[0];
322 if(separator == '\0') {
323 fprintf(stderr, _("ERROR: invalid separator.\n"));
324 exit(1);
325 }
326 break;
327 case 'p':
328 {
329 char *end = NULL;
330
331 portnum = strtol(optarg, &end, 10);
332
333 if(errno == ERANGE || end == optarg || *end != '\0' || portnum < 1) {
334 fprintf(stderr, _("ERROR: invalid port number.\n"));
335 exit(1);
336 }
337 }
338 break;
339 case 'u':
340 switch(*optarg) {
341 case 'c':
342 case 'C':
343 unit = CELSIUS;
344 break;
345 case 'f':
346 case 'F':
347 unit = FAHRENHEIT;
348 break;
349 }
350 break;
351 case 'l':
352 listen_addr = optarg;
353 break;
354 case '?':
355 case 'h':
356 printf(_(" Usage: hddtemp [OPTIONS] [TYPE:]DISK1 [[TYPE:]DISK2]...\n"
357 "\n"
358 " hddtemp displays the temperature of drives supplied in argument.\n"
359 " Drives must support S.M.A.R.T.\n"
360 "\n"
361 " TYPE could be SATA, PATA or SCSI. If omitted hddtemp will try to guess.\n"
362 "\n"
363 " -b --drivebase : display database file content that allow hddtemp to\n"
364 " recognize supported drives.\n"
365 " -D --debug : display various S.M.A.R.T. fields and their values.\n"
366 " Useful to find a value that seems to match the\n"
367 " temperature and/or to send me a report.\n"
368 " (done for every drive supplied).\n"
369 " -d --daemon : run hddtemp in TCP/IP daemon mode (port %d by default.)\n"
370 " -f --file=FILE : specify database file to use.\n"
371 " -F --foreground : don't daemonize, stay in foreground.\n"
372 " -l --listen=addr : listen on a specific interface (in TCP/IP daemon mode).\n"
373 " -n --numeric : print only the temperature.\n"
374 " -p --port=# : port to listen to (in TCP/IP daemon mode).\n"
375 " -s --separator=C : separator to use between fields (in TCP/IP daemon mode).\n"
376 " -S --syslog=s : log temperature to syslog every s seconds.\n"
377 " -u --unit=[C|F] : force output temperature either in Celsius or Fahrenheit.\n"
378 " -q --quiet : do not check if the drive is supported.\n"
379 " -v --version : display hddtemp version number.\n"
380 " -w --wake-up : wake-up the drive if need.\n"
381 " -4 : listen on IPv4 sockets only.\n"
382 " -6 : listen on IPv6 sockets only.\n"
383 "\n"
384 "Report bugs or new drives to <hddtemp@guzu.net>.\n"),
385 PORT_NUMBER);
386 break;
387 case 'v':
388 printf(_("hddtemp version %s\n"), VERSION);
389 exit(0);
390 break;
391 case 'n':
392 numeric = 1;
393 break;
394 case 'w':
395 wakeup = 1;
396 break;
397 case 'S':
398 {
399 char *end = NULL;
400
401 syslog_interval = strtol(optarg, &end, 10);
402
403 if(errno == ERANGE || end == optarg || *end != '\0' || syslog_interval < 1) {
404 fprintf(stderr, _("ERROR: invalid interval.\n"));
405 exit(1);
406 }
407 }
408 break;
409 case 'F':
410 foreground = 1;
411 break;
412 default:
413 exit(1);
414 }
415 }
416
417 if(show_db) {
418 load_database(database_path);
419 display_supported_drives();
420 exit(0);
421 }
422
423 if(argc - optind <= 0) {
424 fprintf(stderr, _("Too few arguments: you must specify one drive, at least.\n"));
425 exit(1);
426 }
427
428 if(debug) {
429 /* argc = optind + 1;*/
430 quiet = 1;
431 }
432
433 if(debug && (tcp_daemon || syslog_interval != 0)) {
434 fprintf(stderr, _("ERROR: can't use --debug and --daemon or --syslog options together.\n"));
435 exit(1);
436 }
437
438 init_bus_types();
439
440 /* collect disks informations */
441 ldisks = NULL;
442 for(i = argc - 1; i >= optind; i--) {
443 struct disk *dsk = (struct disk *) malloc(sizeof(struct disk));
444 char *p;
445
446 assert(dsk);
447
448 memset(dsk, 0, sizeof(*dsk));
449
450 p = strchr(argv[i], ':');
451 if(p == NULL)
452 dsk->drive = argv[i];
453 else {
454 char *q;
455 int j;
456
457 /* upper case type */
458 for(q = argv[i]; q != p; q++)
459 *q = (char) toupper(*q);
460
461 /* force bus type */
462 for(j = 0; j < BUS_TYPE_MAX; j++) {
463 if(bus[j] &&
464 bus[j]->name &&
465 strncmp(bus[j]->name, argv[i], p - argv[i] - 1) == 0)
466 {
467 dsk->type = j;
468 break;
469 }
470 }
471
472 dsk->drive = p + 1;
473 }
474
475 dsk->next = ldisks;
476 ldisks = dsk;
477
478 errno = 0;
479 dsk->errormsg[0] = '\0';
480 if( (dsk->fd = open(dsk->drive, O_RDONLY | O_NONBLOCK)) < 0) {
481 snprintf(dsk->errormsg, MAX_ERRORMSG_SIZE, "open: %s\n", strerror(errno));
482 dsk->type = ERROR;
483 ret = 1;
484 continue;
485 }
486
487 if( ! dsk->type ) {
488 dsk->type = probe_bus_type(dsk);
489 }
490
491 if(dsk->type == BUS_UNKNOWN) {
492 fprintf(stderr, _("ERROR: %s: can't determine bus type (or this bus type is unknown)\n"), dsk->drive);
493
494 ldisks = dsk->next;
495 free(dsk);
496 ret = 1;
497 continue;
498 }
499
500 dsk->model = bus[dsk->type]->model(dsk->fd);
501 dsk->value = -1;
502 if(dsk->type != BUS_SCSI) {
503 struct harddrive_entry *dbe;
504
505 if(!db_loaded) {
506 load_database(database_path);
507 db_loaded = 1;
508 }
509
510 dsk->db_entry = (struct harddrive_entry *)malloc(sizeof(struct harddrive_entry));
511 dbe = is_a_supported_drive(dsk->model);
512 if(dbe)
513 memcpy(dsk->db_entry, dbe, sizeof(struct harddrive_entry));
514 else {
515 dsk->db_entry->regexp = "";
516 dsk->db_entry->description = "";
517 dsk->db_entry->attribute_id = DEFAULT_ATTRIBUTE_ID;
518 dsk->db_entry->unit = 'C';
519 dsk->db_entry->next = NULL;
520 }
521 }
522 }
523
524 free_database();
525 if(tcp_daemon || syslog_interval != 0) {
526 do_daemon_mode(ldisks);
527 }
528 else {
529 do_direct_mode(ldisks);
530 }
531
532 return ret;
533 }
Debian
Debian-France
Debian-Facile
Debian-fr.org
Forum-Debian.fr
Debian ?
Communautés
