sconnect.c 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. /*
  2. * Copyright 1998-2020 The OpenSSL Project Authors. All Rights Reserved.
  3. *
  4. * Licensed under the Apache License 2.0 (the "License"). You may not use
  5. * this file except in compliance with the License. You can obtain a copy
  6. * in the file LICENSE in the source distribution or at
  7. * https://www.openssl.org/source/license.html
  8. */
  9. /*-
  10. * A minimal program to do SSL to a passed host and port.
  11. * It is actually using non-blocking IO but in a very simple manner
  12. * sconnect host:port - it does a 'GET / HTTP/1.0'
  13. *
  14. * cc -I../../include sconnect.c -L../.. -lssl -lcrypto
  15. */
  16. #include <stdio.h>
  17. #include <stdlib.h>
  18. #include <unistd.h>
  19. #include <string.h>
  20. #include <errno.h>
  21. #include <openssl/err.h>
  22. #include <openssl/ssl.h>
  23. #define HOSTPORT "localhost:4433"
  24. #define CAFILE "root.pem"
  25. int main(int argc, char *argv[])
  26. {
  27. const char *hostport = HOSTPORT;
  28. const char *CAfile = CAFILE;
  29. const char *hostname;
  30. BIO *out = NULL;
  31. char buf[1024 * 10], *p;
  32. SSL_CTX *ssl_ctx = NULL;
  33. SSL *ssl;
  34. BIO *ssl_bio;
  35. int i, len, off, ret = EXIT_FAILURE;
  36. if (argc > 1)
  37. hostport = argv[1];
  38. if (argc > 2)
  39. CAfile = argv[2];
  40. #ifdef WATT32
  41. dbug_init();
  42. sock_init();
  43. #endif
  44. ssl_ctx = SSL_CTX_new(TLS_client_method());
  45. /* Enable trust chain verification */
  46. SSL_CTX_set_verify(ssl_ctx, SSL_VERIFY_PEER, NULL);
  47. SSL_CTX_load_verify_locations(ssl_ctx, CAfile, NULL);
  48. /* Lets make a SSL structure */
  49. ssl = SSL_new(ssl_ctx);
  50. SSL_set_connect_state(ssl);
  51. /* Use it inside an SSL BIO */
  52. ssl_bio = BIO_new(BIO_f_ssl());
  53. BIO_set_ssl(ssl_bio, ssl, BIO_CLOSE);
  54. /* Lets use a connect BIO under the SSL BIO */
  55. out = BIO_new(BIO_s_connect());
  56. BIO_set_conn_hostname(out, hostport);
  57. /* The BIO has parsed the host:port and even IPv6 literals in [] */
  58. hostname = BIO_get_conn_hostname(out);
  59. if (!hostname || SSL_set1_host(ssl, hostname) <= 0)
  60. goto err;
  61. BIO_set_nbio(out, 1);
  62. out = BIO_push(ssl_bio, out);
  63. p = "GET / HTTP/1.0\r\n\r\n";
  64. len = strlen(p);
  65. off = 0;
  66. for (;;) {
  67. i = BIO_write(out, &(p[off]), len);
  68. if (i <= 0) {
  69. if (BIO_should_retry(out)) {
  70. fprintf(stderr, "write DELAY\n");
  71. sleep(1);
  72. continue;
  73. } else {
  74. goto err;
  75. }
  76. }
  77. off += i;
  78. len -= i;
  79. if (len <= 0)
  80. break;
  81. }
  82. for (;;) {
  83. i = BIO_read(out, buf, sizeof(buf));
  84. if (i == 0)
  85. break;
  86. if (i < 0) {
  87. if (BIO_should_retry(out)) {
  88. fprintf(stderr, "read DELAY\n");
  89. sleep(1);
  90. continue;
  91. }
  92. goto err;
  93. }
  94. fwrite(buf, 1, i, stdout);
  95. }
  96. ret = EXIT_SUCCESS;
  97. goto done;
  98. err:
  99. if (ERR_peek_error() == 0) { /* system call error */
  100. fprintf(stderr, "errno=%d ", errno);
  101. perror("error");
  102. } else {
  103. ERR_print_errors_fp(stderr);
  104. }
  105. done:
  106. BIO_free_all(out);
  107. SSL_CTX_free(ssl_ctx);
  108. return ret;
  109. }