handling request /

#define _BSD_SOURCE #include #include #include #include #include #include #include #include #include #include #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[] = "handling request /\n/test\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; }