xref: /linux/tools/laptop/dslm/dslm.c (revision 26fbb4c8c7c3ee9a4c3b4de555a8587b5a19154e)
1 /*
2  * dslm.c
3  * Simple Disk Sleep Monitor
4  *  by Bartek Kania
5  * Licensed under the GPL
6  */
7 #include <unistd.h>
8 #include <stdlib.h>
9 #include <stdio.h>
10 #include <fcntl.h>
11 #include <errno.h>
12 #include <time.h>
13 #include <string.h>
14 #include <signal.h>
15 #include <sys/ioctl.h>
16 #include <linux/hdreg.h>
17 
18 #ifdef DEBUG
19 #define D(x) x
20 #else
21 #define D(x)
22 #endif
23 
24 int endit = 0;
25 
26 /* Check if the disk is in powersave-mode
27  * Most of the code is stolen from hdparm.
28  * 1 = active, 0 = standby/sleep, -1 = unknown */
29 static int check_powermode(int fd)
30 {
31     unsigned char args[4] = {WIN_CHECKPOWERMODE1,0,0,0};
32     int state;
33 
34     if (ioctl(fd, HDIO_DRIVE_CMD, &args)
35 	&& (args[0] = WIN_CHECKPOWERMODE2) /* try again with 0x98 */
36 	&& ioctl(fd, HDIO_DRIVE_CMD, &args)) {
37 	if (errno != EIO || args[0] != 0 || args[1] != 0) {
38 	    state = -1; /* "unknown"; */
39 	} else
40 	    state = 0; /* "sleeping"; */
41     } else {
42 	state = (args[2] == 255) ? 1 : 0;
43     }
44     D(printf(" drive state is:  %d\n", state));
45 
46     return state;
47 }
48 
49 static char *state_name(int i)
50 {
51     if (i == -1) return "unknown";
52     if (i == 0) return "sleeping";
53     if (i == 1) return "active";
54 
55     return "internal error";
56 }
57 
58 static char *myctime(time_t time)
59 {
60     char *ts = ctime(&time);
61     ts[strlen(ts) - 1] = 0;
62 
63     return ts;
64 }
65 
66 static void measure(int fd)
67 {
68     time_t start_time;
69     int last_state;
70     time_t last_time;
71     int curr_state;
72     time_t curr_time = 0;
73     time_t time_diff;
74     time_t active_time = 0;
75     time_t sleep_time = 0;
76     time_t unknown_time = 0;
77     time_t total_time = 0;
78     int changes = 0;
79     float tmp;
80 
81     printf("Starting measurements\n");
82 
83     last_state = check_powermode(fd);
84     start_time = last_time = time(0);
85     printf("  System is in state %s\n\n", state_name(last_state));
86 
87     while(!endit) {
88 	sleep(1);
89 	curr_state = check_powermode(fd);
90 
91 	if (curr_state != last_state || endit) {
92 	    changes++;
93 	    curr_time = time(0);
94 	    time_diff = curr_time - last_time;
95 
96 	    if (last_state == 1) active_time += time_diff;
97 	    else if (last_state == 0) sleep_time += time_diff;
98 	    else unknown_time += time_diff;
99 
100 	    last_state = curr_state;
101 	    last_time = curr_time;
102 
103 	    printf("%s: State-change to %s\n", myctime(curr_time),
104 		   state_name(curr_state));
105 	}
106     }
107     changes--; /* Compensate for SIGINT */
108 
109     total_time = time(0) - start_time;
110     printf("\nTotal running time:  %lus\n", curr_time - start_time);
111     printf(" State changed %d times\n", changes);
112 
113     tmp = (float)sleep_time / (float)total_time * 100;
114     printf(" Time in sleep state:   %lus (%.2f%%)\n", sleep_time, tmp);
115     tmp = (float)active_time / (float)total_time * 100;
116     printf(" Time in active state:  %lus (%.2f%%)\n", active_time, tmp);
117     tmp = (float)unknown_time / (float)total_time * 100;
118     printf(" Time in unknown state: %lus (%.2f%%)\n", unknown_time, tmp);
119 }
120 
121 static void ender(int s)
122 {
123     endit = 1;
124 }
125 
126 static void usage(void)
127 {
128     puts("usage: dslm [-w <time>] <disk>");
129     exit(0);
130 }
131 
132 int main(int argc, char **argv)
133 {
134     int fd;
135     char *disk = 0;
136     int settle_time = 60;
137 
138     /* Parse the simple command-line */
139     if (argc == 2)
140 	disk = argv[1];
141     else if (argc == 4) {
142 	settle_time = atoi(argv[2]);
143 	disk = argv[3];
144     } else
145 	usage();
146 
147     if (!(fd = open(disk, O_RDONLY|O_NONBLOCK))) {
148 	printf("Can't open %s, because: %s\n", disk, strerror(errno));
149 	exit(-1);
150     }
151 
152     if (settle_time) {
153 	printf("Waiting %d seconds for the system to settle down to "
154 	       "'normal'\n", settle_time);
155 	sleep(settle_time);
156     } else
157 	puts("Not waiting for system to settle down");
158 
159     signal(SIGINT, ender);
160 
161     measure(fd);
162 
163     close(fd);
164 
165     return 0;
166 }
167