diff options
Diffstat (limited to 'demos/guide/tls-client-non-block.c')
-rw-r--r-- | demos/guide/tls-client-non-block.c | 25 |
1 files changed, 18 insertions, 7 deletions
diff --git a/demos/guide/tls-client-non-block.c b/demos/guide/tls-client-non-block.c index 8748e4fffc..0b19d67762 100644 --- a/demos/guide/tls-client-non-block.c +++ b/demos/guide/tls-client-non-block.c @@ -27,7 +27,7 @@ #include <openssl/err.h> /* Helper function to create a BIO connected to the server */ -static BIO *create_socket_bio(const char *hostname, const char *port) +static BIO *create_socket_bio(const char *hostname, const char *port, int family) { int sock = -1; BIO_ADDRINFO *res; @@ -37,7 +37,7 @@ static BIO *create_socket_bio(const char *hostname, const char *port) /* * Lookup IP address info for the server. */ - if (!BIO_lookup_ex(hostname, port, BIO_LOOKUP_CLIENT, 0, SOCK_STREAM, 0, + if (!BIO_lookup_ex(hostname, port, BIO_LOOKUP_CLIENT, family, SOCK_STREAM, 0, &res)) return NULL; @@ -187,14 +187,25 @@ int main(int argc, char *argv[]) char buf[160]; int eof = 0; char *hostname, *port; + int argnext = 1; + int ipv6 = 0; - if (argc != 3) { - printf("Usage: tls-client-non-block hostname port\n"); + if (argc < 3) { + printf("Usage: tls-client-non-block [-6] hostname port\n"); goto end; } - hostname = argv[1]; - port = argv[2]; + if (!strcmp(argv[argnext], "-6")) { + if (argc < 4) { + printf("Usage: tls-client-non-block [-6] hostname port\n"); + goto end; + } + ipv6 = 1; + argnext++; + } + + hostname = argv[argnext++]; + port = argv[argnext]; /* * Create an SSL_CTX which we can use to create SSL objects from. We @@ -240,7 +251,7 @@ int main(int argc, char *argv[]) * Create the underlying transport socket/BIO and associate it with the * connection. */ - bio = create_socket_bio(hostname, port); + bio = create_socket_bio(hostname, port, ipv6 ? AF_INET6 : AF_INET); if (bio == NULL) { printf("Failed to crete the BIO\n"); goto end; |