aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorWolfgang Draxinger <Wolfgang.Draxinger@draxit.de>2013-03-14 12:11:06 +0100
committerWolfgang Draxinger <Wolfgang.Draxinger@draxit.de>2013-03-14 12:11:06 +0100
commitdb106357956d84283dc85ea3e47d512da5113db3 (patch)
treed836486a57b7585aaee9b284d1529ffc3c767b62
parent62d44c3e46f63c891b18d05be5fbfa3c4a57feb5 (diff)
downloadlitheweb-db106357956d84283dc85ea3e47d512da5113db3.tar.gz
litheweb-db106357956d84283dc85ea3e47d512da5113db3.tar.bz2
Thu Mar 14 12:11:06 CET 2013
Diffstat
-rw-r--r--test/bsdsocket.c (renamed from test/bsd_socket.c)0
-rw-r--r--test/bufbsdsocket.c229
2 files changed, 229 insertions, 0 deletions
diff --git a/test/bsd_socket.c b/test/bsdsocket.c
index d4bd5a2..d4bd5a2 100644
--- a/test/bsd_socket.c
+++ b/test/bsdsocket.c
diff --git a/test/bufbsdsocket.c b/test/bufbsdsocket.c
new file mode 100644
index 0000000..27ce6ef
--- /dev/null
+++ b/test/bufbsdsocket.c
@@ -0,0 +1,229 @@
+#define _BSD_SOURCE
+
+#include <stddef.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <errno.h>
+#include <string.h>
+
+#include <unistd.h>
+
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/ip.h>
+
+#include "../picohttp.h"
+
+#define SENDBUF_LEN 256
+
+struct bufbsdsockData {
+ char * recvbuf;
+ size_t recvbuf_len;
+ size_t recvbuf_pos;
+ char sendbuf[SENDBUF_LEN];
+ size_t sendbuf_pos;
+ int fd;
+};
+
+int bufbsdsock_read(size_t count, char *buf, void *data_)
+{
+ struct bufbsdsockData *data = data_;
+
+ ssize_t rb = 0;
+ ssize_t r = 0;
+ do {
+ size_t len = 0;
+
+ if( !data->recvbuf ||
+ data->recvbuf_pos >= data->recvbuf_len ) {
+ if( data->recvbuf )
+ free( data->recvbuf );
+ data->recvbuf_len = 0;
+ data->recvbuf_pos = 0;
+
+ int avail = 0;
+ do {
+ struct pollfd pfd = {
+ .fd = data->fd,
+ .events = POLLIN | POLLPRI,
+ .revents = 0
+ };
+
+ int const pret = poll(&pfd, 1, -1);
+ if( 0 >= pret ) {
+ return -1;
+ }
+
+ assert(pfd.revents & (POLLIN | POLLPRI));
+
+ if( -1 == ioctl(fd, FIONREAD, &avail) ) {
+ perror("ioctl(FIONREAD)");
+ return -1;
+ }
+ } while( !avail );
+
+ data->recvbuf = malloc( avail);
+
+ int r;
+ while( 0 > (r = read(data->fd, data->recvbuf, avail)) )
+ if( EINTR == errno )
+ continue;
+
+ if( EAGAIN == errno ||
+ EWOULDBLOCK == errno ) {
+ usleep(200);
+ continue;
+ }
+
+ return -1;
+ }
+ data->recvbuf_len += r;
+ }
+
+ len = data->recvbuf_len - data->recvbuf_pos;
+ if( len > count )
+ len = count;
+
+ rb += len;
+ } while( rb < count );
+ return rb;
+}
+
+int bufbsdsock_write(size_t count, char const *buf, void *data)
+{
+ int fd = *((int*)data);
+
+ ssize_t wb = 0;
+ ssize_t w = 0;
+ do {
+ } while( wb < count );
+ return wb;
+}
+
+int16_t bufbsdsock_getch(void *data)
+{
+ char ch;
+ if( 1 != bufbsdsock_read(1, &ch, data) )
+ return -1;
+ return ch;
+}
+
+int bufbsdsock_putch(char ch, void *data)
+{
+ return bufbsdsock_write(1, &ch, data);
+}
+
+int bufbsdsock_flush(void *data)
+{
+ return 0;
+}
+
+int sockfd = -1;
+
+void bye(void)
+{
+ fputs("exiting\n", stderr);
+ int const one = 1;
+ /* allows for immediate reuse of address:port
+ * after program termination */
+ setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
+ shutdown(sockfd, SHUT_RDWR);
+ close(sockfd);
+}
+
+void rhRoot(struct picohttpRequest *req)
+{
+ fprintf(stderr, "handling request /%s\n", req->urltail);
+
+ char http_header[] = "HTTP/x.x 200 OK\r\nServer: picoweb\r\nContent-Type: text/html\r\n\r\n";
+ http_header[5] = '0'+req->httpversion.major;
+ http_header[7] = '0'+req->httpversion.minor;
+ picohttpIoWrite(req->ioops, sizeof(http_header)-1, http_header);
+ char http_test[] = "<html><head><title>handling request /</title></head>\n<body><a href=\"/test\">/test</a></body></html>\n";
+
+ picohttpIoWrite(req->ioops, sizeof(http_test)-1, http_test);
+}
+
+void rhTest(struct picohttpRequest *req)
+{
+ fprintf(stderr, "handling request /test%s\n", req->urltail);
+ char http_header[] = "HTTP/x.x 200 OK\r\nServer: picoweb\r\nContent-Type: text/text\r\n\r\n";
+ http_header[5] = '0'+req->httpversion.major;
+ http_header[7] = '0'+req->httpversion.minor;
+ picohttpIoWrite(req->ioops, sizeof(http_header)-1, http_header);
+ char http_test[] = "handling request /test";
+ picohttpIoWrite(req->ioops, sizeof(http_test)-1, http_test);
+ if(req->urltail) {
+ picohttpIoWrite(req->ioops, strlen(req->urltail), req->urltail);
+ }
+}
+
+int main(int argc, char *argv[])
+{
+ sockfd = socket(AF_INET, SOCK_STREAM, 0);
+ if( -1 == sockfd ) {
+ perror("socket");
+ return -1;
+ }
+#if 0
+ if( atexit(bye) ) {
+ return -1;
+ }
+#endif
+
+ struct sockaddr_in addr = {
+ .sin_family = AF_INET,
+ .sin_port = htons(8000),
+ .sin_addr = 0
+ };
+
+ int const one = 1;
+ setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
+ if( -1 == bind(sockfd, (struct sockaddr*)&addr, sizeof(addr)) ) {
+ perror("bind");
+ return -1;
+ }
+
+ if( -1 == listen(sockfd, 2) ) {
+ perror("listen");
+ return -1;
+ }
+
+ for(;;) {
+ socklen_t addrlen = 0;
+ int confd = accept(sockfd, (struct sockaddr*)&addr, &addrlen);
+ if( -1 == confd ) {
+ if( EAGAIN == errno ||
+ EWOULDBLOCK == errno ) {
+ usleep(1000);
+ continue;
+ } else {
+ perror("accept");
+ return -1;
+ }
+ }
+
+ struct picohttpIoOps ioops = {
+ .read = bsdsock_read,
+ .write = bsdsock_write,
+ .getch = bsdsock_getch,
+ .putch = bsdsock_putch,
+ .data = &confd
+ };
+
+ struct picohttpURLRoute routes[] = {
+ { "/test", 0, rhTest, 16, PICOHTTP_METHOD_GET },
+ { "/|", 0, rhRoot, 0, PICOHTTP_METHOD_GET },
+ { NULL, 0, 0, 0, 0 }
+ };
+
+ picohttpProcessRequest(&ioops, routes);
+
+ shutdown(confd, SHUT_RDWR);
+ close(confd);
+ }
+
+ return 0;
+}
+