|
1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ |
|
2 /* This Source Code Form is subject to the terms of the Mozilla Public |
|
3 * License, v. 2.0. If a copy of the MPL was not distributed with this |
|
4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
|
5 |
|
6 #include "prio.h" |
|
7 #include "prmem.h" |
|
8 #include "prprf.h" |
|
9 #include "prinrval.h" |
|
10 |
|
11 #include "plerror.h" |
|
12 #include "plgetopt.h" |
|
13 |
|
14 static PRFileDesc *err = NULL; |
|
15 |
|
16 static void Help(void) |
|
17 { |
|
18 PR_fprintf(err, "Usage: [-S] [-K <n>] [-h] <filename>\n"); |
|
19 PR_fprintf(err, "\t-c Nuber of iterations (default: 10)\n"); |
|
20 PR_fprintf(err, "\t-S Sync the file (default: FALSE)\n"); |
|
21 PR_fprintf(err, "\t-K Size of file (K bytes) (default: 10)\n"); |
|
22 PR_fprintf(err, "\t Name of file to write (default: /usr/tmp/sync.dat)\n"); |
|
23 PR_fprintf(err, "\t-h This message and nothing else\n"); |
|
24 } /* Help */ |
|
25 |
|
26 int main(int argc, char **argv) |
|
27 { |
|
28 PRStatus rv; |
|
29 PLOptStatus os; |
|
30 PRUint8 *buffer; |
|
31 PRFileDesc *file = NULL; |
|
32 const char *filename = "sync.dat"; |
|
33 PRUint32 index, loops, iterations = 10, filesize = 10; |
|
34 PRIntn flags = PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE; |
|
35 PLOptState *opt = PL_CreateOptState(argc, argv, "hSK:c:"); |
|
36 PRIntervalTime time, total = 0, shortest = 0x7fffffff, longest = 0; |
|
37 |
|
38 err = PR_GetSpecialFD(PR_StandardError); |
|
39 |
|
40 while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) |
|
41 { |
|
42 if (PL_OPT_BAD == os) continue; |
|
43 switch (opt->option) |
|
44 { |
|
45 case 0: /* Name of file to create */ |
|
46 filename = opt->value; |
|
47 break; |
|
48 case 'S': /* Use sych option on file */ |
|
49 flags |= PR_SYNC; |
|
50 break; |
|
51 case 'K': /* Size of file to write */ |
|
52 filesize = atoi(opt->value); |
|
53 break; |
|
54 case 'c': /* Number of iterations */ |
|
55 iterations = atoi(opt->value); |
|
56 break; |
|
57 case 'h': /* user wants some guidance */ |
|
58 default: /* user needs some guidance */ |
|
59 Help(); /* so give him an earful */ |
|
60 return 2; /* but not a lot else */ |
|
61 } |
|
62 } |
|
63 PL_DestroyOptState(opt); |
|
64 |
|
65 file = PR_Open(filename, flags, 0666); |
|
66 if (NULL == file) |
|
67 { |
|
68 PL_FPrintError(err, "Failed to open file"); |
|
69 return 1; |
|
70 } |
|
71 |
|
72 buffer = (PRUint8*)PR_CALLOC(1024); |
|
73 if (NULL == buffer) |
|
74 { |
|
75 PL_FPrintError(err, "Cannot allocate buffer"); |
|
76 return 1; |
|
77 } |
|
78 |
|
79 for (index = 0; index < sizeof(buffer); ++index) |
|
80 buffer[index] = (PRUint8)index; |
|
81 |
|
82 for (loops = 0; loops < iterations; ++loops) |
|
83 { |
|
84 time = PR_IntervalNow(); |
|
85 for (index = 0; index < filesize; ++index) |
|
86 { |
|
87 PR_Write(file, buffer, 1024); |
|
88 } |
|
89 time = (PR_IntervalNow() - time); |
|
90 |
|
91 total += time; |
|
92 if (time < shortest) shortest = time; |
|
93 else if (time > longest) longest = time; |
|
94 if (0 != PR_Seek(file, 0, PR_SEEK_SET)) |
|
95 { |
|
96 PL_FPrintError(err, "Rewinding file"); |
|
97 return 1; |
|
98 } |
|
99 } |
|
100 |
|
101 total = total / iterations; |
|
102 PR_fprintf( |
|
103 err, "%u iterations over a %u kbyte %sfile: %u [%u] %u\n", |
|
104 iterations, filesize, ((flags & PR_SYNC) ? "SYNCH'd " : ""), |
|
105 PR_IntervalToMicroseconds(shortest), |
|
106 PR_IntervalToMicroseconds(total), |
|
107 PR_IntervalToMicroseconds(longest)); |
|
108 |
|
109 PR_DELETE(buffer); |
|
110 rv = PR_Close(file); |
|
111 if (PR_SUCCESS != rv) |
|
112 { |
|
113 PL_FPrintError(err, "Closing file failed"); |
|
114 return 1; |
|
115 } |
|
116 rv = PR_Delete(filename); |
|
117 if (PR_SUCCESS != rv) |
|
118 { |
|
119 PL_FPrintError(err, "Deleting file failed"); |
|
120 return 1; |
|
121 } |
|
122 return 0; |
|
123 } |