fix dtls example

Ichishino · Dec 19, 2023 · 3da499b · 3da499b
1 parent 7bd51ab
commit 3da499b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 28 deletions.
diff --git a/examples/dtls_client/main.c b/examples/dtls_client/main.c
@@ -111,28 +111,6 @@ app_on_tls_handshake(
 }
 
 #ifdef CO_USE_TLS
-
-#ifndef CO_USE_WOLFSSL
-// TODO
-int
-app_on_tls_verify_cookie(
-    SSL* ssl,
-    const unsigned char* cookie,
-    unsigned int cookie_len
-)
-{
-    (void)ssl;
-    (void)cookie;
-    (void)cookie_len;
-
-    printf("cookie: %*.*s\n",
-        cookie_len, cookie_len, (char*)cookie);
-
-    // ok
-    return 1;
-}
-#endif
-
 int
 app_on_tls_verify_peer(
     int preverify_ok,
@@ -145,7 +123,6 @@ app_on_tls_verify_peer(
     // ok
     return 1;
 }
-
 #endif
 
 
@@ -169,11 +146,6 @@ app_tls_setup(
     SSL_CTX_set_verify(
         ssl_ctx, SSL_VERIFY_PEER, app_on_tls_verify_peer);
 
-    #ifndef CO_USE_WOLFSSL
-    // TODO
-    SSL_CTX_set_cookie_verify_cb(ssl_ctx, app_on_tls_verify_cookie);
-    #endif
-
     tls_ctx->ssl_ctx = ssl_ctx;
 
 #endif

diff --git a/examples/dtls_server/main.c b/examples/dtls_server/main.c
@@ -165,6 +165,7 @@ app_on_tls_handshake(
 
 #ifdef CO_USE_TLS
 #ifndef CO_USE_WOLFSSL
+
 // TODO
 int
 app_on_tls_generate_cookie(
@@ -180,6 +181,22 @@ app_on_tls_generate_cookie(
 
     return 1;
 }
+
+// TODO
+int
+app_on_tls_verify_cookie(
+    SSL* ssl,
+    const unsigned char* cookie,
+    unsigned int cookie_len
+)
+{
+    (void)ssl;
+    (void)cookie;
+    (void)cookie_len;
+
+    return 1;
+}
+
 #endif
 #endif
 
@@ -226,6 +243,8 @@ app_tls_setup(
     // TODO
     SSL_CTX_set_cookie_generate_cb(
         ssl_ctx, app_on_tls_generate_cookie);
+    SSL_CTX_set_cookie_verify_cb(
+        ssl_ctx, app_on_tls_verify_cookie);
     #endif
 
     tls_ctx->ssl_ctx = ssl_ctx;