ab9f6f7537110237ecb4da981ad6cc22c8c1410e
2 * ustream-ssl - library for SSL over ustream
4 * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <sys/socket.h>
20 #include <netinet/in.h>
28 #include <libubox/ustream.h>
29 #include <libubox/uloop.h>
30 #include <libubox/usock.h>
31 #include "ustream-ssl.h"
35 static struct uloop_fd server
;
36 static const char *port
= "10000";
37 static struct client
*next_client
= NULL
;
40 struct sockaddr_in sin
;
43 struct ustream_ssl ssl
;
55 static void client_read_cb(struct ustream
*s
, int bytes
)
57 struct client
*cl
= container_of(s
, struct client
, ssl
.stream
);
58 struct ustream_buf
*buf
= s
->r
.head
;
62 str
= ustream_get_read_buf(s
, NULL
);
66 newline
= strchr(buf
->data
, '\n');
73 ustream_printf(s
, "HTTP/1.1 200 OK\nContent-Type:text/plain\n\n");
74 ustream_printf(s
, "Got request header: %s\n", str
);
82 ustream_state_change(s
);
86 ustream_printf(s
, "%s\n", str
);
93 ustream_consume(s
, newline
+ 1 - str
);
94 cl
->ctr
+= newline
+ 1 - str
;
97 if (s
->w
.data_bytes
> 256 && !ustream_read_blocked(s
)) {
98 fprintf(stderr
, "Block read, bytes: %d\n", s
->w
.data_bytes
);
99 ustream_set_read_blocked(s
, true);
103 static void client_close(struct ustream
*s
)
105 struct client
*cl
= container_of(s
, struct client
, ssl
.stream
);
107 fprintf(stderr
, "Connection closed\n");
109 ustream_free(&cl
->s
.stream
);
114 static void client_notify_write(struct ustream
*s
, int bytes
)
116 fprintf(stderr
, "Wrote %d bytes, pending: %d\n", bytes
, s
->w
.data_bytes
);
118 if (s
->w
.data_bytes
< 128 && ustream_read_blocked(s
)) {
119 fprintf(stderr
, "Unblock read\n");
120 ustream_set_read_blocked(s
, false);
124 static void client_notify_state(struct ustream
*s
)
126 struct client
*cl
= container_of(s
, struct client
, ssl
.stream
);
131 fprintf(stderr
, "eof!, pending: %d, total: %d\n", s
->w
.data_bytes
, cl
->ctr
);
132 if (!s
->w
.data_bytes
)
133 return client_close(s
);
136 static void client_notify_connected(struct ustream_ssl
*ssl
)
138 fprintf(stderr
, "SSL connection established\n");
141 static void client_notify_error(struct ustream_ssl
*ssl
, int error
, const char *str
)
143 fprintf(stderr
, "SSL connection error(%d): %s\n", error
, str
);
146 static void server_cb(struct uloop_fd
*fd
, unsigned int events
)
149 unsigned int sl
= sizeof(struct sockaddr_in
);
153 next_client
= calloc(1, sizeof(*next_client
));
156 sfd
= accept(server
.fd
, (struct sockaddr
*) &cl
->sin
, &sl
);
158 fprintf(stderr
, "Accept failed\n");
162 cl
->ssl
.stream
.string_data
= true;
163 cl
->ssl
.stream
.notify_read
= client_read_cb
;
164 cl
->ssl
.stream
.notify_state
= client_notify_state
;
165 cl
->ssl
.stream
.notify_write
= client_notify_write
;
166 cl
->ssl
.notify_connected
= client_notify_connected
;
167 cl
->ssl
.notify_error
= client_notify_error
;
169 ustream_fd_init(&cl
->s
, sfd
);
170 ustream_ssl_init(&cl
->ssl
, &cl
->s
.stream
, ctx
, true);
172 fprintf(stderr
, "New connection\n");
175 static int run_server(void)
178 server
.cb
= server_cb
;
179 server
.fd
= usock(USOCK_TCP
| USOCK_SERVER
| USOCK_IPV4ONLY
| USOCK_NUMERIC
, "127.0.0.1", port
);
186 uloop_fd_add(&server
, ULOOP_READ
);
192 static int usage(const char *name
)
194 fprintf(stderr
, "Usage: %s -p <port>\n", name
);
198 int main(int argc
, char **argv
)
202 ctx
= ustream_ssl_context_new(true);
203 ustream_ssl_context_set_crt_file(ctx
, "example.crt");
204 ustream_ssl_context_set_key_file(ctx
, "example.key");
206 while ((ch
= getopt(argc
, argv
, "p:")) != -1) {
212 return usage(argv
[0]);