feedc0de/wolfssl
1
0
Fork 0
forked from wolfSSL/wolfssl
Code Pull Requests Activity

server: check that the first packet of udp connection is clientHello

Browse Source 
Used to allow for bi-directional shutdown tests with UDP and DTLS
This commit is contained in:
Marco Oliverio
2022-05-09 10:30:38 +02:00
parent 59fdf05155
commit d133fa6143
1 changed files with 36 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
examples/server/server.c
View File

@@ -2933,16 +2933,44 @@ THREAD_RETURN WOLFSSL_THREAD server_test(void* args)
if (doDTLS && dtlsUDP) { if (doDTLS && dtlsUDP) {
byte b[1500]; byte b[1500];
int n; int n;
int isClientHello = 0;
client_len = sizeof client_addr; while (!isClientHello) {
client_len = sizeof client_addr;
/* For DTLS, peek at the next datagram so we can get the client's /* For DTLS, peek at the next datagram so we can get the
* address and set it into the ssl object later to generate the * client's address and set it into the ssl object later to
* cookie. */ * generate the cookie. */
n = (int)recvfrom(clientfd, (char*)b, sizeof(b), MSG_PEEK, n = (int)recvfrom(clientfd, (char*)b, sizeof(b), MSG_PEEK,
(struct sockaddr*)&client_addr, &client_len); (struct sockaddr*)&client_addr, &client_len);
if (n <= 0)
err_sys_ex(runWithErrors, "recvfrom failed"); if (n <= 0)
err_sys_ex(runWithErrors, "recvfrom failed");
/* when doing resumption, it may happen that we receive the
alert used to shutdown the first connection as the first
packet of the second accept:
Client | Server
| WolfSSL_Shutdown()
| <- Alert
| recvfrom(peek)
WolfSSL_Shutdown() |
Alert-> |
| wolfSSL_set_dtls_peer()
but this will set the wrong src port, making the test fail.
Discard not-handshake message to avoid this.
*/
if (b[0] != 0x16) {
/* discard the packet */
n = (int)recvfrom(clientfd, (char *)b, sizeof(b), 0,
(struct sockaddr *)&client_addr, &client_len);
}
else {
isClientHello = 1;
}
}
if (doBlockSeq) { if (doBlockSeq) {
XMEMCPY(&dtlsCtx.peer.sa, &client_addr, client_len); XMEMCPY(&dtlsCtx.peer.sa, &client_addr, client_len);