-
Notifications
You must be signed in to change notification settings - Fork 301
Expand file tree
/
Copy pathtcp_load_test.c
More file actions
135 lines (97 loc) · 3.58 KB
/
tcp_load_test.c
File metadata and controls
135 lines (97 loc) · 3.58 KB
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
#include <libusockets.h>
const int SSL = 0;
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
char request[] = "Hello there!";
char *host;
int port;
int connections;
int responses;
/* We don't need any of these */
void on_wakeup(struct us_loop_t *loop) {
}
void on_pre(struct us_loop_t *loop) {
}
/* This is not HTTP POST, it is merely an event emitted post loop iteration */
void on_post(struct us_loop_t *loop) {
}
struct us_socket_t *on_http_socket_writable(struct us_socket_t *s) {
return s;
}
struct us_socket_t *on_http_socket_close(struct us_socket_t *s, int code, void *reason) {
return s;
}
struct us_socket_t *on_http_socket_end(struct us_socket_t *s) {
return us_socket_close(SSL, s, 0, NULL);
}
struct us_socket_t *on_http_socket_data(struct us_socket_t *s, char *data, int length) {
us_socket_write(SSL, s, request, sizeof(request) - 1, 0);
responses++;
return s;
}
struct us_socket_t *on_http_socket_open(struct us_socket_t *s, int is_client, char *ip, int ip_length) {
/* Send a request */
us_socket_write(SSL, s, request, sizeof(request) - 1, 0);
if (--connections) {
us_socket_context_connect(SSL, us_socket_context(SSL, s), host, port, NULL, 0, 0);
} else {
printf("Running benchmark now...\n");
us_socket_timeout(SSL, s, LIBUS_TIMEOUT_GRANULARITY);
us_socket_long_timeout(SSL, s, 1);
}
return s;
}
struct us_socket_t *on_http_socket_long_timeout(struct us_socket_t *s) {
/* Print current statistics */
printf("--- Minute mark ---\n");
us_socket_long_timeout(SSL, s, 1);
return s;
}
struct us_socket_t *on_http_socket_timeout(struct us_socket_t *s) {
/* Print current statistics */
printf("Req/sec: %f\n", ((float)responses) / LIBUS_TIMEOUT_GRANULARITY);
responses = 0;
us_socket_timeout(SSL, s, LIBUS_TIMEOUT_GRANULARITY);
return s;
}
struct us_socket_t *on_http_socket_connect_error(struct us_socket_t *s, int code) {
printf("Cannot connect to server\n");
return s;
}
int main(int argc, char **argv) {
/* Parse host and port */
if (argc != 4) {
printf("Usage: connections host port\n");
return 0;
}
port = atoi(argv[3]);
host = malloc(strlen(argv[2]) + 1);
memcpy(host, argv[2], strlen(argv[2]) + 1);
connections = atoi(argv[1]);
/* Create the event loop */
struct us_loop_t *loop = us_create_loop(0, on_wakeup, on_pre, on_post, 0);
/* Create a socket context for HTTP */
struct us_socket_context_options_t options = {};
struct us_socket_context_t *http_context = us_create_socket_context(SSL, loop, 0, options);
if (!http_context) {
printf("Could not load SSL cert/key\n");
exit(0);
}
/* Set up event handlers */
us_socket_context_on_open(SSL, http_context, on_http_socket_open);
us_socket_context_on_data(SSL, http_context, on_http_socket_data);
us_socket_context_on_writable(SSL, http_context, on_http_socket_writable);
us_socket_context_on_close(SSL, http_context, on_http_socket_close);
us_socket_context_on_timeout(SSL, http_context, on_http_socket_timeout);
us_socket_context_on_long_timeout(SSL, http_context, on_http_socket_long_timeout);
us_socket_context_on_end(SSL, http_context, on_http_socket_end);
us_socket_context_on_connect_error(SSL, http_context, on_http_socket_connect_error);
/* Start making HTTP connections */
if (!us_socket_context_connect(SSL, http_context, host, port, NULL, 0, 0)) {
printf("Cannot connect to server\n");
}
us_loop_run(loop);
us_socket_context_free(SSL, http_context);
us_loop_free(loop);
}