ithewei 4 роки тому
батько
коміт
10b180087d
2 змінених файлів з 9 додано та 3 видалено
  1. 6 3
      examples/websocket_client_test.cpp
  2. 3 0
      ssl/mbedtls.c

+ 6 - 3
examples/websocket_client_test.cpp

@@ -20,7 +20,7 @@ int main(int argc, char** argv) {
     const char* url = argv[1];
     const char* url = argv[1];
 
 
     WebSocketClient ws;
     WebSocketClient ws;
-    ws.onopen = [&ws]() {
+    ws.onopen = []() {
         printf("onopen\n");
         printf("onopen\n");
     };
     };
     ws.onclose = []() {
     ws.onclose = []() {
@@ -43,9 +43,12 @@ int main(int argc, char** argv) {
 
 
     std::string str;
     std::string str;
     while (std::getline(std::cin, str)) {
     while (std::getline(std::cin, str)) {
-        if (ws.isConnected()) {
-            ws.send(str);
+        if (!ws.isConnected()) break;
+        if (str == "quit") {
+            ws.close();
+            break;
         }
         }
+        ws.send(str);
     }
     }
 
 
     return 0;
     return 0;

+ 3 - 0
ssl/mbedtls.c

@@ -183,6 +183,9 @@ int hssl_close(hssl_t ssl) {
 }
 }
 
 
 int hssl_set_sni_hostname(hssl_t ssl, const char* hostname) {
 int hssl_set_sni_hostname(hssl_t ssl, const char* hostname) {
+#ifdef MBEDTLS_X509_CRT_PARSE_C
+    mbedtls_ssl_set_hostname(ssl, hostname);
+#endif
     return 0;
     return 0;
 }
 }