1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/* Written by Anthony Liguori <aliguori@us.ibm.com> */
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <string.h>
#include <unistd.h>
#include <termios.h>
#define MIN(a, b) (((a) < (b)) ? (a) : (b))
static void generate_random_buffer(char *buffer, size_t size)
{
int i;
for (i = 0; i < size; i++) {
buffer[i] = random() & 0xFF;
}
}
static void canonicalize(char *buffer)
{
char *reader, *writer;
reader = writer = buffer;
while (*reader) {
*writer = *reader;
if (*reader != '\r') writer++;
reader++;
}
*writer = *reader;
}
int main(int argc, char **argv)
{
char buffer[4096];
char *line;
unsigned int seed;
size_t size;
int runs;
unsigned long long total_bytes = 0;
struct termios term;
tcgetattr(STDIN_FILENO, &term);
cfmakeraw(&term);
tcsetattr(STDIN_FILENO, TCSAFLUSH, &term);
tcgetattr(STDOUT_FILENO, &term);
cfmakeraw(&term);
tcsetattr(STDOUT_FILENO, TCSAFLUSH, &term);
while ((line = fgets(buffer, sizeof(buffer), stdin))) {
canonicalize(line);
if (strcmp(line, "!!!XEN Test Begin!!!\n") == 0) {
break;
} else {
fprintf(stderr, "%s", line);
}
}
if (line == NULL) {
fprintf(stderr, "Client never sent start string.\n");
return 1;
}
seed = time(0);
printf("%u\n", seed); fflush(stdout);
fprintf(stderr, "Waiting for seed acknowledgement\n");
line = fgets(buffer, sizeof(buffer), stdin);
if (line == NULL) {
fprintf(stderr, "Client never acknowledge seed.\n");
return 1;
}
canonicalize(line);
if (strcmp(line, "Seed Okay.\n") != 0) {
fprintf(stderr, "Incorrect seed acknowledgement.\n");
fprintf(stderr, "[%s]", line);
return 1;
} else {
fprintf(stderr, "Processed seed.\n");
}
srandom(seed);
for (runs = (random() % 100000) + 4096; runs > 0; runs--) {
size = random() % 4096;
fprintf(stderr, "Writing %d bytes.\n", size);
generate_random_buffer(buffer, size);
fwrite(buffer, size, 1, stdout);
fflush(stdout);
do {
line = fgets(buffer, sizeof(buffer), stdin);
if (line == NULL) {
fprintf(stderr, "Premature EOF from client.\n");
return 1;
}
canonicalize(line);
fprintf(stderr, "%s", line);
} while (strcmp(line, "Okay.\n") != 0);
total_bytes += size;
}
fprintf(stderr, "PASS: processed %llu byte(s).\n", total_bytes);
return 0;
}
|
