Add -s to the usage text...
[doldaconnect.git] / config / speedrec.c
1 #include <string.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4 #include <sys/file.h>
5 #include <unistd.h>
6 #include <fcntl.h>
7 #include <errno.h>
8 #include <time.h>
9 #include <signal.h>
10
11 char buf[65536];
12 volatile int eof;
13
14 void sighandler(int sig)
15 {
16     eof = 1;
17 }
18
19 int main(int argc, char **argv)
20 {
21     int i;
22     int ret, fd;
23     time_t starttime, endtime;
24     long long numbytes;
25     size_t datalen;
26     FILE *recfile;
27     int thisrec, numrecs, numuses, maxrec;
28     int recs[5];
29     
30     if(argc < 2)
31     {
32         fprintf(stderr, "usage: speedrec recfile\n");
33         exit(1);
34     }
35     numbytes = 0;
36     starttime = endtime = 0;
37     datalen = 0;
38     eof = 0;
39     signal(SIGHUP, sighandler);
40     signal(SIGINT, sighandler);
41     signal(SIGTERM, sighandler);
42     while(1)
43     {
44         ret = read(0, buf, sizeof(buf));
45         if((ret < 0) && (errno != EINTR))
46         {
47             perror("cannot read");
48             exit(1);
49         }
50         if((ret == 0) || eof)
51             break;
52         if(starttime == 0)
53             starttime = time(NULL);
54         endtime = time(NULL);
55         datalen = ret;
56         numbytes += ret;
57         while(datalen > 0)
58         {
59             ret = write(1, buf, datalen);
60             if((ret < 0) && (errno != EINTR))
61             {
62                 perror("cannot write");
63                 exit(1);
64             }
65             memmove(buf, buf + ret, datalen -= ret);
66         }
67     }
68     if((starttime == 0) || (endtime == 0) || (endtime == starttime))
69         exit(0);
70     if(numbytes == 0)
71         exit(0);
72     thisrec = (int)(numbytes / ((long long)(endtime - starttime)));
73     if((fd = open(argv[1], O_RDWR | O_CREAT, 0666)) < 0)
74     {
75         perror(argv[1]);
76         exit(1);
77     }
78     recfile = fdopen(fd, "r+");
79     close(0);
80     close(1);
81     flock(fd, LOCK_EX);
82     if(fscanf(recfile, "%i\n", &numuses) < 1)
83         numuses = 0;
84     if(fscanf(recfile, "%i\n", &maxrec) < 1)
85         maxrec = 0;
86     if(fscanf(recfile, "%i\n", &numrecs) < 1)
87         numrecs = 0;
88     for(i = 0; i < numrecs; i++)
89         fscanf(recfile, "%i\n", &recs[i]);
90     if(numrecs == 5)
91     {
92         for(i = 0; i < 4; i++)
93             recs[i] = recs[i + 1];
94         numrecs = 4;
95     }
96     recs[numrecs++] = thisrec;
97     rewind(recfile);
98     ftruncate(fd, 0);
99     fprintf(recfile, "%i\n", numuses + 1);
100     fprintf(recfile, "%i\n", (thisrec > maxrec)?thisrec:maxrec);
101     fprintf(recfile, "%i\n", numrecs);
102     for(i = 0; i < numrecs; i++)
103         fprintf(recfile, "%i\n", recs[i]);
104     flock(fd, LOCK_UN);
105     fclose(recfile);
106     return(0);
107 }