Use "==0" instead of "!strcmp" etc
[openssl.git] / demos / bio / server-arg.c
1 /* NOCW */
2 /* demos/bio/server-arg.c */
3
4 /*
5  * A minimal program to serve an SSL connection. It uses blocking. It use the
6  * SSL_CONF API with the command line. cc -I../../include server-arg.c
7  * -L../.. -lssl -lcrypto -ldl
8  */
9
10 #include <stdio.h>
11 #include <signal.h>
12 #include <openssl/err.h>
13 #include <openssl/ssl.h>
14
15 int main(int argc, char *argv[])
16 {
17     char *port = "*:4433";
18     BIO *ssl_bio, *tmp;
19     SSL_CTX *ctx;
20     SSL_CONF_CTX *cctx;
21     char buf[512];
22     BIO *in = NULL;
23     int ret = 1, i;
24     char **args = argv + 1;
25     int nargs = argc - 1;
26
27     SSL_load_error_strings();
28
29     /* Add ciphers and message digests */
30     OpenSSL_add_ssl_algorithms();
31
32     ctx = SSL_CTX_new(SSLv23_server_method());
33
34     cctx = SSL_CONF_CTX_new();
35     SSL_CONF_CTX_set_flags(cctx, SSL_CONF_FLAG_SERVER);
36     SSL_CONF_CTX_set_flags(cctx, SSL_CONF_FLAG_CERTIFICATE);
37     SSL_CONF_CTX_set_ssl_ctx(cctx, ctx);
38     while (*args && **args == '-') {
39         int rv;
40         /* Parse standard arguments */
41         rv = SSL_CONF_cmd_argv(cctx, &nargs, &args);
42         if (rv == -3) {
43             fprintf(stderr, "Missing argument for %s\n", *args);
44             goto err;
45         }
46         if (rv < 0) {
47             fprintf(stderr, "Error in command %s\n", *args);
48             ERR_print_errors_fp(stderr);
49             goto err;
50         }
51         /* If rv > 0 we processed something so proceed to next arg */
52         if (rv > 0)
53             continue;
54         /* Otherwise application specific argument processing */
55         if (strcmp(*args, "-port") == 0) {
56             port = args[1];
57             if (port == NULL) {
58                 fprintf(stderr, "Missing -port argument\n");
59                 goto err;
60             }
61             args += 2;
62             nargs -= 2;
63             continue;
64         } else {
65             fprintf(stderr, "Unknown argument %s\n", *args);
66             goto err;
67         }
68     }
69
70     if (!SSL_CONF_CTX_finish(cctx)) {
71         fprintf(stderr, "Finish error\n");
72         ERR_print_errors_fp(stderr);
73         goto err;
74     }
75 #ifdef ITERATE_CERTS
76     /*
77      * Demo of how to iterate over all certificates in an SSL_CTX structure.
78      */
79     {
80         X509 *x;
81         int rv;
82         rv = SSL_CTX_set_current_cert(ctx, SSL_CERT_SET_FIRST);
83         while (rv) {
84             X509 *x = SSL_CTX_get0_certificate(ctx);
85             X509_NAME_print_ex_fp(stdout, X509_get_subject_name(x), 0,
86                                   XN_FLAG_ONELINE);
87             printf("\n");
88             rv = SSL_CTX_set_current_cert(ctx, SSL_CERT_SET_NEXT);
89         }
90         fflush(stdout);
91     }
92 #endif
93     /* Setup server side SSL bio */
94     ssl_bio = BIO_new_ssl(ctx, 0);
95
96     if ((in = BIO_new_accept(port)) == NULL)
97         goto err;
98
99     /*
100      * This means that when a new connection is accepted on 'in', The ssl_bio
101      * will be 'duplicated' and have the new socket BIO push into it.
102      * Basically it means the SSL BIO will be automatically setup
103      */
104     BIO_set_accept_bios(in, ssl_bio);
105
106  again:
107     /*
108      * The first call will setup the accept socket, and the second will get a
109      * socket.  In this loop, the first actual accept will occur in the
110      * BIO_read() function.
111      */
112
113     if (BIO_do_accept(in) <= 0)
114         goto err;
115
116     for (;;) {
117         i = BIO_read(in, buf, 512);
118         if (i == 0) {
119             /*
120              * If we have finished, remove the underlying BIO stack so the
121              * next time we call any function for this BIO, it will attempt
122              * to do an accept
123              */
124             printf("Done\n");
125             tmp = BIO_pop(in);
126             BIO_free_all(tmp);
127             goto again;
128         }
129         if (i < 0)
130             goto err;
131         fwrite(buf, 1, i, stdout);
132         fflush(stdout);
133     }
134
135     ret = 0;
136  err:
137     if (ret) {
138         ERR_print_errors_fp(stderr);
139     }
140     BIO_free(in);
141     exit(ret);
142     return (!ret);
143 }