commands/rawspeed/rawspeed.c

   1 /*      rawspeed 1.15 - Measure speed of a device.      Author: Kees J. Bot
   2  *                                                              26 Apr 1992
   3  */
   4 #define nil 0
   5 #include <sys/types.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <unistd.h>
   9 #include <errno.h>
  10 #include <signal.h>
  11 #include <string.h>
  12 #include <time.h>
  13 #if !__minix
  14 #include <sys/time.h>
  15 #endif
  16 #include <fcntl.h>
  17 #include <limits.h>
  18 #include <sys/stat.h>
  19
  20 #define SECTOR_SIZE             512
  21 #define BLK_MAX_SECTORS         (sizeof(int) == 2 ? 32 : 64)
  22 #define CHR_MAX_SECTORS         (sizeof(int) == 2 ? 63 : 512)
  23 #define ABS_MAX_SECTORS         (INT_MAX / SECTOR_SIZE)
  24
  25 #define USEC    (!__minix || __minix_vmd)
  26
  27 /* Any good random number generator around? */
  28 #if __minix_vmd || __linux
  29 #define rand    random
  30 #define srand   srandom
  31 #endif
  32
  33 #if __sun && __svr4__
  34 #define rand    lrand48
  35 #define srand   srand48
  36 #endif
  37
  38 void report(const char *label)
  39 {
  40         fprintf(stderr, "rawspeed: %s: %s\n", label, strerror(errno));
  41 }
  42
  43 void fatal(const char *label)
  44 {
  45         report(label);
  46         exit(1);
  47 }
  48
  49 void usage(void)
  50 {
  51         fprintf(stderr,
  52 "Usage: rawspeed [-u unit] [-m max] [-t seconds] [-c] [-r limit] device\n");
  53         fprintf(stderr,
  54 "       -u unit = best sector multiple (default 2)\n");
  55         fprintf(stderr,
  56 "       -m max = read multiples of unit upto 'max' (default %d raw, %d file)\n",
  57                                         CHR_MAX_SECTORS, BLK_MAX_SECTORS);
  58         fprintf(stderr,
  59 "       -t seconds = time to run test (default 10)\n");
  60         fprintf(stderr,
  61 "       -c = cache test: rewind after each read or write of max size\n");
  62         fprintf(stderr,
  63 "       -r limit = random seeks upto sector 'limit' before reading or writing\n");
  64         exit(1);
  65 }
  66
  67 int done= 0;
  68
  69 void timeout(int sig)
  70 {
  71         done= 1;
  72 }
  73
  74 int main(int argc, char **argv)
  75 {
  76         int i, fd, n= 0, unit= -1, max= -1, cache= 0;
  77         int size, seconds= 10;
  78         long tenthsec;
  79 #if USEC
  80         struct timeval start_time, end_time;
  81         struct timezone dummy;
  82 #else
  83         time_t start_time;
  84 #endif
  85         off_t nbytes= 0, wbytes= 0, randlimit= 0;
  86         char *device, *chunk;
  87         struct stat st;
  88         off_t nseeks= 0;
  89
  90         for (i= 1; i < argc && argv[i][0] == '-' && argv[i][1] != 0; i++) {
  91                 char *opt;
  92
  93                 if (argv[i][1] == '-' && argv[i][2] == 0) { i++; break; }
  94
  95                 for (opt= argv[i]+1; *opt != 0; opt++) {
  96                         switch (*opt) {
  97                         case 'w':
  98                                 if (i == argc) usage();
  99                                 wbytes= atol(argv[++i]) * 1024;
 100                                 if (wbytes <= 0) usage();
 101                                 break;
 102                         case 'm':
 103                                 if (i == argc) usage();
 104                                 max= atoi(argv[++i]);
 105                                 if (max <= 0 || max > ABS_MAX_SECTORS)
 106                                         usage();
 107                                 break;
 108                         case 'u':
 109                                 if (i == argc) usage();
 110                                 unit= atoi(argv[++i]);
 111                                 if (unit <= 0 || unit > ABS_MAX_SECTORS)
 112                                         usage();
 113                                 break;
 114                         case 't':
 115                                 if (i == argc) usage();
 116                                 seconds= atoi(argv[++i]);
 117                                 if (seconds <= 0) usage();
 118                                 break;
 119                         case 'c':
 120                                 cache= 1;
 121                                 break;
 122                         case 'r':
 123                                 if (i == argc) usage();
 124                                 randlimit= atol(argv[++i]);
 125                                 if (randlimit <= 0) usage();
 126                                 break;
 127                         default:
 128                                 usage();
 129                         }
 130                 }
 131         }
 132
 133         if (i != argc - 1) usage();
 134
 135         if (strcmp(argv[i], "-") == 0) {
 136                 fd= wbytes == 0 ? 0 : 1;
 137                 device= "";
 138         } else {
 139                 device= argv[i];
 140                 if ((fd= open(device,
 141                         wbytes == 0 ? O_RDONLY : O_WRONLY | O_CREAT, 0666)) < 0)
 142                                 fatal(device);
 143         }
 144         if (max < 0) {
 145                 if (fstat(fd, &st) >= 0 && S_ISCHR(st.st_mode))
 146                         max= CHR_MAX_SECTORS;
 147                 else
 148                         max= BLK_MAX_SECTORS;
 149         }
 150
 151         if (unit < 0) unit= max > 1 ? 2 : 1;
 152         unit*= max / unit;
 153         if (unit == 0) usage();
 154         size= unit * SECTOR_SIZE;
 155         randlimit/= unit;
 156
 157         if ((chunk= malloc((size_t) size)) == nil) {
 158                 fprintf(stderr, "rawspeed: can't grab %d bytes: %s\n",
 159                         size, strerror(errno));
 160                 exit(1);
 161         }
 162
 163         /* Touch the pages to get real memory sending other processes to swap.
 164          */
 165         memset((void *) chunk, 0, (size_t) size);
 166
 167         /* Clean the cache. */
 168         sync();
 169
 170         signal(SIGALRM, timeout);
 171         signal(SIGINT, timeout);
 172 #if USEC
 173         gettimeofday(&start_time, &dummy);
 174         if (randlimit != 0) srand((int) (start_time.tv_sec & INT_MAX));
 175 #else
 176         start_time= time((time_t *) nil);
 177         if (randlimit != 0) srand((int) (start_time & INT_MAX));
 178 #endif
 179         alarm(seconds);
 180
 181         if (wbytes > 0) {
 182                 while (!done && (n= write(fd, chunk, size)) > 0
 183                                                 && (nbytes+= n) < wbytes) {
 184                         if (cache && lseek(fd, (off_t) 0, SEEK_SET) == -1)
 185                                 fatal(device);
 186                         if (randlimit != 0) {
 187                                 if (lseek(fd, (off_t)
 188                                         (rand() % randlimit * size),
 189                                                         SEEK_SET) == -1)
 190                                         fatal(device);
 191                                 nseeks++;
 192                         }
 193                 }
 194                 sync();
 195         } else {
 196                 while (!done && (n= read(fd, chunk, size)) > 0) {
 197                         nbytes+= n;
 198                         if (cache && lseek(fd, (off_t) 0, SEEK_SET) == -1)
 199                                 fatal(device);
 200                         if (randlimit != 0) {
 201                                 if (lseek(fd, (off_t)
 202                                         (rand() % randlimit * size),
 203                                                         SEEK_SET) == -1)
 204                                         fatal(device);
 205                                 nseeks++;
 206                         }
 207                 }
 208         }
 209
 210 #if USEC
 211         gettimeofday(&end_time, &dummy);
 212         tenthsec= (end_time.tv_sec - start_time.tv_sec) * 10
 213                 + (end_time.tv_usec - start_time.tv_usec) / 100000;
 214 #else
 215         tenthsec= (time((time_t *) 0) - start_time) * 10;
 216 #endif
 217         if (n < 0 && errno == EINTR) n= 0;
 218         if (n < 0) report(device);
 219
 220         if (nbytes > 0) {
 221                 off_t kBpts;
 222
 223                 fprintf(stderr, "%ld kB / %d.%d s = ",
 224                         (nbytes + 512) / 1024,
 225                         tenthsec / 10, tenthsec % 10);
 226                 if (tenthsec < 5)
 227                         fprintf(stderr, "infinite\n");
 228                 else {
 229                         if (nbytes > LONG_MAX / 100) {
 230                                 seconds = (tenthsec + 5) / 10;
 231                                 kBpts= (nbytes + 512L * seconds)
 232                                                         / (1024L * seconds);
 233                                 fprintf(stderr, "%ld kB/s\n", kBpts);
 234                         } else {
 235                                 kBpts= (100 * nbytes + 512L * tenthsec)
 236                                                         / (1024L * tenthsec);
 237                                 fprintf(stderr, "%ld.%ld kB/s\n",
 238                                         kBpts/10, kBpts%10);
 239                         }
 240                 }
 241         }
 242         if (randlimit != 0 && tenthsec >= 5) {
 243                 int rpm, disc= 0;
 244                 off_t tenthms;
 245
 246                 tenthms= (tenthsec * 1000 + nseeks/2) / nseeks;
 247
 248                 fprintf(stderr,
 249                         "%ld seeks / %d.%d s = %ld seeks/s = %ld.%ld ms/seek\n",
 250                         nseeks, tenthsec / 10, tenthsec % 10,
 251                         (nseeks * 10 + tenthsec/2) / tenthsec,
 252                         tenthms / 10, tenthms % 10);
 253
 254                 for (rpm= 3600; rpm <= 7200; rpm+= 1800) {
 255                         int rotms = (10000L / 2 * 60 + rpm/2) / rpm;
 256
 257                         if (tenthms <= rotms) continue;
 258
 259                         if (!disc) {
 260                                 fprintf(stderr,
 261                                         "discarding av. rotational delay:\n  ");
 262                                 disc= 1;
 263                         } else {
 264                                 fprintf(stderr, ", ");
 265                         }
 266                         fprintf(stderr, "%ld.%ld ms (%d rpm)",
 267                                 (tenthms - rotms) / 10, (tenthms - rotms) % 10,
 268                                 rpm);
 269                 }
 270                 if (disc) fputc('\n', stdout);
 271         }
 272         return n < 0 ? 1 : 0;
 273 }