sctp-client-dtls.c 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. /* sctp-client-dtls.c
  2. *
  3. * Copyright (C) 2006-2020 wolfSSL Inc.
  4. *
  5. * This file is part of wolfSSL.
  6. *
  7. * wolfSSL is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * wolfSSL is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
  20. */
  21. /* wolfssl */
  22. #ifndef WOLFSSL_USER_SETTINGS
  23. #include <wolfssl/options.h>
  24. #endif
  25. #include <wolfssl/wolfcrypt/settings.h>
  26. #include <wolfssl/ssl.h>
  27. #if defined(WOLFSSL_SCTP) && defined(WOLFSSL_DTLS)
  28. /* sctp */
  29. #include <sys/socket.h>
  30. #include <sys/types.h>
  31. #include <arpa/inet.h>
  32. #include <netinet/in.h>
  33. /* std */
  34. #include <stdio.h>
  35. #include <stdlib.h>
  36. #include <string.h>
  37. #include <unistd.h>
  38. #define cacert "./certs/ca-cert.pem"
  39. static int err_sys(const char* msg)
  40. {
  41. perror(msg);
  42. exit(EXIT_FAILURE);
  43. }
  44. #endif /* WOLFSSL_SCTP && WOLFSSL_DTLS */
  45. int main()
  46. {
  47. #if defined(WOLFSSL_SCTP) && defined(WOLFSSL_DTLS)
  48. int sd = socket(PF_INET, SOCK_STREAM, IPPROTO_SCTP);
  49. if (sd < 0)
  50. err_sys("sctp socket error");
  51. struct sockaddr_in sa;
  52. memset(&sa, 0, sizeof(sa));
  53. sa.sin_family = AF_INET;
  54. sa.sin_addr.s_addr = inet_addr("127.0.0.1");
  55. sa.sin_port = htons(12345);
  56. int ret = connect(sd, (struct sockaddr*)&sa, sizeof(sa));
  57. if (ret < 0)
  58. err_sys("sctp connect error");
  59. const char* response = "hello there";
  60. char buffer[80];
  61. WOLFSSL_CTX* ctx = wolfSSL_CTX_new(wolfDTLSv1_2_client_method());
  62. if (ctx == NULL)
  63. err_sys("ctx new dtls client failed");
  64. ret = wolfSSL_CTX_dtls_set_sctp(ctx);
  65. if (ret != WOLFSSL_SUCCESS)
  66. err_sys("set sctp mode failed");
  67. ret = wolfSSL_CTX_load_verify_locations(ctx, cacert, NULL);
  68. if (ret != WOLFSSL_SUCCESS)
  69. err_sys("ca cert error");
  70. WOLFSSL* ssl = wolfSSL_new(ctx);
  71. if (ssl == NULL)
  72. err_sys("ssl new dtls client failed");
  73. wolfSSL_set_fd(ssl, sd);
  74. ret = wolfSSL_connect(ssl);
  75. if (ret != WOLFSSL_SUCCESS)
  76. err_sys("ssl connect failed");
  77. printf("TLS version is %s\n", wolfSSL_get_version(ssl));
  78. printf("Cipher Suite is %s\n",
  79. wolfSSL_CIPHER_get_name(wolfSSL_get_current_cipher(ssl)));
  80. wolfSSL_write(ssl, response, (int)strlen(response));
  81. int got = wolfSSL_read(ssl, buffer, sizeof(buffer));
  82. if (got > 0) {
  83. buffer[got] = 0;
  84. printf("server said: %s\n", buffer);
  85. }
  86. unsigned char bigBuf[4096];
  87. unsigned int i;
  88. for (i = 0; i < (int)sizeof(bigBuf); i++)
  89. bigBuf[i] = (unsigned char)(i & 0xFF);
  90. wolfSSL_write(ssl, bigBuf, sizeof(bigBuf));
  91. memset(bigBuf, 0, sizeof(bigBuf));
  92. wolfSSL_read(ssl, bigBuf, sizeof(bigBuf));
  93. for (i = 0; i < sizeof(bigBuf); i++) {
  94. if (bigBuf[i] != (unsigned char)(i & 0xFF)) {
  95. printf("big message check fail\n");
  96. break;
  97. }
  98. }
  99. wolfSSL_shutdown(ssl);
  100. wolfSSL_free(ssl);
  101. wolfSSL_CTX_free(ctx);
  102. close(sd);
  103. #endif /* WOLFSSL_SCTP && WOLFSSL_DTLS */
  104. return 0;
  105. }