1 #define min(x,y) ((x)<(y)?(x):(y))
3 extern const char* side
;
5 #define HANDSHAKE_EXPECT(c, s, clierr, serverr) \
6 sret = cret = GNUTLS_E_AGAIN; \
9 if (cret == GNUTLS_E_AGAIN) \
12 cret = gnutls_handshake (c); \
14 if (sret == GNUTLS_E_AGAIN) \
17 sret = gnutls_handshake (s); \
20 while ((cret == GNUTLS_E_AGAIN || (cret == 0 && sret == GNUTLS_E_AGAIN)) && (sret == GNUTLS_E_AGAIN || (sret == 0 && cret == GNUTLS_E_AGAIN))); \
21 if (cret != clierr || sret != serverr) \
23 fprintf(stderr, "client: %s\n", gnutls_strerror(cret)); \
24 fprintf(stderr, "server: %s\n", gnutls_strerror(sret)); \
25 fail("Handshake failed\n"); \
29 #define HANDSHAKE(c, s) \
30 HANDSHAKE_EXPECT(c,s,0,0)
32 #define TRANSFER(c, s, msg, msglen, buf, buflen) \
36 ret = gnutls_record_send (c, msg, msglen); \
38 while(ret == GNUTLS_E_AGAIN); \
40 if (ret < 0) fail ("client send error: %s\n", gnutls_strerror (ret)); \
47 ret = gnutls_record_recv (s, buf, buflen); \
49 while(ret == GNUTLS_E_AGAIN); \
51 fail ("server: didn't receive any data\n"); \
54 fail ("server: error: %s\n", gnutls_strerror (ret)); \
63 ns = gnutls_record_send (server, msg, msglen); \
65 while (ns == GNUTLS_E_AGAIN); \
66 if (ns < 0) fail ("server send error: %s\n", gnutls_strerror (ret)); \
70 ret = gnutls_record_recv (client, buf, buflen); \
72 while(ret == GNUTLS_E_AGAIN); \
75 fail ("client: Peer has closed the TLS connection\n"); \
80 fputs ("!", stdout); \
81 fail ("client: Error: %s\n", gnutls_strerror (ret)); \
85 if (msglen != ret || memcmp (buf, msg, msglen) != 0) \
87 fail ("client: Transmitted data do not match\n"); \
93 ns = gnutls_record_send (client, buf, msglen); \
95 while (ns == GNUTLS_E_AGAIN); \
96 if (ns < 0) fail ("client send error: %s\n", gnutls_strerror (ret)); \
99 fputs (".", stdout); \
102 while (transferred < 70000)
104 static char to_server
[64*1024];
105 static size_t to_server_len
= 0;
107 static char to_client
[64*1024];
108 static size_t to_client_len
= 0;
111 #define RETURN_RND_EAGAIN(session) \
112 static unsigned char rnd = 0; \
113 if (rnd++ % 2 == 0) \
115 gnutls_transport_set_errno (session, EAGAIN); \
119 #define RETURN_RND_EAGAIN(session)
123 client_push (gnutls_transport_ptr_t tr
, const void *data
, size_t len
)
126 RETURN_RND_EAGAIN(tr
);
128 len
= min(len
, sizeof(to_server
)-to_server_len
);
130 newlen
= to_server_len
+ len
;
131 memcpy (to_server
+ to_server_len
, data
, len
);
132 to_server_len
= newlen
;
134 fprintf(stderr
, "eagain: pushed %d bytes to server (avail: %d)\n", (int)len
, (int)to_server_len
);
140 client_pull (gnutls_transport_ptr_t tr
, void *data
, size_t len
)
142 RETURN_RND_EAGAIN(tr
);
144 if (to_client_len
== 0)
147 fprintf(stderr
, "eagain: Not enough data by server (asked for: %d, have: %d)\n", (int)len
, (int)to_client_len
);
149 gnutls_transport_set_errno ((gnutls_session_t
)tr
, EAGAIN
);
153 len
= min(len
, to_client_len
);
155 memcpy (data
, to_client
, len
);
157 memmove (to_client
, to_client
+ len
, to_client_len
- len
);
158 to_client_len
-= len
;
160 fprintf(stderr
, "eagain: pulled %d bytes by client (avail: %d)\n", (int)len
, (int)to_client_len
);
166 server_pull (gnutls_transport_ptr_t tr
, void *data
, size_t len
)
168 //success ("server_pull len %d has %d\n", len, to_server_len);
169 RETURN_RND_EAGAIN(tr
);
171 if (to_server_len
== 0)
174 fprintf(stderr
, "eagain: Not enough data by client (asked for: %d, have: %d)\n", (int)len
, (int)to_server_len
);
176 gnutls_transport_set_errno ((gnutls_session_t
)tr
, EAGAIN
);
180 len
= min(len
, to_server_len
);
182 fprintf(stderr
, "eagain: pulled %d bytes by server (avail: %d)\n", (int)len
, (int)to_server_len
);
184 memcpy (data
, to_server
, len
);
186 memmove (to_server
, to_server
+ len
, to_server_len
- len
);
187 to_server_len
-= len
;
193 server_push (gnutls_transport_ptr_t tr
, const void *data
, size_t len
)
196 RETURN_RND_EAGAIN(tr
);
198 // hexprint (data, len);
200 len
= min(len
, sizeof(to_client
)-to_client_len
);
202 newlen
= to_client_len
+ len
;
203 memcpy (to_client
+ to_client_len
, data
, len
);
204 to_client_len
= newlen
;
206 fprintf(stderr
, "eagain: pushed %d bytes to client (avail: %d)\n", (int)len
, (int)to_client_len
);
212 /* inline is used to avoid a gcc warning if used in mini-eagain */
213 inline static int server_pull_timeout_func(gnutls_transport_ptr_t ptr
, unsigned int ms
)
217 if (to_server_len
> 0)
218 ret
= 1; /* available data */
220 ret
= 0; /* timeout */
223 fprintf(stderr
, "eagain: server_pull_timeout: %d (avail: cli %d, serv %d)\n", ret
, (int)to_client_len
, (int)to_server_len
);
229 inline static int client_pull_timeout_func(gnutls_transport_ptr_t ptr
, unsigned int ms
)
233 if (to_client_len
> 0)
239 fprintf(stderr
, "eagain: client_pull_timeout: %d (avail: cli %d, serv %d)\n", ret
, (int)to_client_len
, (int)to_server_len
);
245 inline static void reset_buffers(void)