kore

An easy to use, scalable and secure web application framework for writing web APIs in C.
Commits | Files | Refs | README | LICENSE | git clone https://git.kore.io/kore.git

commit 857c3f91214120e7a54062f12c303e41972250b2
Author: Joris Vink <joris@coders.se>
Date:   Wed, 17 Apr 2013 22:34:27 +0200

first commit

Diffstat:
Makefile | 21+++++++++++++++++++++
includes/.kore.h.swp | 0
includes/kore.h | 59+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
src/kore.c | 191+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
src/utils.c | 102+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
5 files changed, 373 insertions(+), 0 deletions(-)

diff --git a/Makefile b/Makefile @@ -0,0 +1,21 @@ +# Kore Makefile + +CC=clang +BIN=kore + +S_SRC= src/kore.c src/utils.c +S_OBJS= $(S_SRC:.c=.o) + +CFLAGS+=-Wall -Wstrict-prototypes -Wmissing-prototypes +CFLAGS+=-Wmissing-declarations -Wshadow -Wpointer-arith -Wcast-qual +CFLAGS+=-Wsign-compare -Iincludes -g +LDFLAGS= + +light: $(S_OBJS) + $(CC) $(CFLAGS) $(LDFLAGS) $(S_OBJS) -o $(BIN) + +.c.o: $< + $(CC) $(CFLAGS) -c $< -o $@ + +clean: + rm -f src/*.o $(BIN) diff --git a/includes/.kore.h.swp b/includes/.kore.h.swp Binary files differ. diff --git a/includes/kore.h b/includes/kore.h @@ -0,0 +1,59 @@ +/* + * Copyright (c) 2013 Joris Vink <joris@coders.se> + * + * Permission to use, copy, modify, and distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + */ + +#ifndef __H_KORE_H +#define __H_KORE_H + +#define KORE_RESULT_ERROR 0 +#define KORE_RESULT_OK 1 + +#define errno_s strerror(errno) +#define kore_log(fmt, ...) \ + kore_log_internal(__FILE__, __LINE__, fmt, ##__VA_ARGS__) + +struct netbuf { + u_int8_t *data; + u_int32_t offset; + u_int32_t length; + void *owner; + int (*cb)(struct netbuf *); + + TAILQ_ENTRY(netbuf) list; +}; + +struct listener { + int fd; + struct sockaddr_in sin; +}; + +struct connection { + int fd; + struct sockaddr_in sin; + void *owner; + + TAILQ_HEAD(, netbuf) send_queue; + TAILQ_HEAD(, netbuf) recv_queue; +}; + +void *kore_malloc(size_t); +void *kore_calloc(size_t, size_t); +void *kore_realloc(void *, size_t); +char *kore_strdup(const char *); + +void fatal(const char *, ...); +void kore_log_internal(char *, int, const char *, ...); + +#endif /* !__H_KORE_H */ diff --git a/src/kore.c b/src/kore.c @@ -0,0 +1,191 @@ +/* + * Copyright (c) 2013 Joris Vink <joris@coders.se> + * + * Permission to use, copy, modify, and distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + */ + +#include <sys/param.h> +#include <sys/types.h> +#include <sys/socket.h> +#include <sys/queue.h> +#include <sys/epoll.h> + +#include <netinet/in.h> +#include <arpa/inet.h> + +#include <errno.h> +#include <fcntl.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + +#include "kore.h" + +#define EPOLL_EVENTS 500 + +static int efd = -1; + +static int kore_server_bind(struct listener *, const char *, int); +static int kore_server_accept(struct listener *); +static int kore_connection_handle(struct connection *, int); +static int kore_socket_nonblock(int); +static void kore_event(int, int, void *); + +int +main(int argc, char *argv[]) +{ + struct connection *c; + struct listener server; + struct epoll_event *events; + int n, i, *fd; + + if (argc != 3) + fatal("Usage: kore [ip] [port]"); + + if (!kore_server_bind(&server, argv[1], atoi(argv[2]))) + fatal("cannot bind to %s:%s", argv[1], argv[2]); + + if ((efd = epoll_create(1000)) == -1) + fatal("epoll_create(): %s", errno_s); + + kore_event(server.fd, EPOLLIN, &server); + events = kore_calloc(EPOLL_EVENTS, sizeof(struct epoll_event)); + for (;;) { + n = epoll_wait(efd, events, EPOLL_EVENTS, -1); + if (n == -1) + fatal("epoll_wait(): %s", errno_s); + + for (i = 0; i < n; i++) { + fd = (int *)events[i].data.ptr; + + if (events[i].events & EPOLLERR || + events[i].events & EPOLLHUP) { + if (*fd == server.fd) + fatal("error on server socket"); + + c = (struct connection *)events[i].data.ptr; + continue; + } + + if (*fd == server.fd) { + kore_server_accept(&server); + } else { + c = (struct connection *)events[i].data.ptr; + if (!kore_connection_handle(c, events[i].events)) + /* Disconnect. */; + } + } + } + + close(server.fd); + return (0); +} + +static int +kore_server_bind(struct listener *l, const char *ip, int port) +{ + if ((l->fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { + kore_log("socket(): %s", errno_s); + return (KORE_RESULT_ERROR); + } + + if (!kore_socket_nonblock(l->fd)) { + close(l->fd); + return (KORE_RESULT_ERROR); + } + + memset(&(l->sin), 0, sizeof(l->sin)); + l->sin.sin_family = AF_INET; + l->sin.sin_port = htons(port); + l->sin.sin_addr.s_addr = inet_addr(ip); + + if (bind(l->fd, (struct sockaddr *)&(l->sin), sizeof(l->sin)) == -1) { + close(l->fd); + kore_log("bind(): %s", errno_s); + return (KORE_RESULT_ERROR); + } + + if (listen(l->fd, 50) == -1) { + close(l->fd); + kore_log("listen(): %s", errno_s); + return (KORE_RESULT_ERROR); + } + + return (KORE_RESULT_OK); +} + +static int +kore_server_accept(struct listener *l) +{ + socklen_t len; + struct connection *c; + + len = sizeof(struct sockaddr_in); + c = (struct connection *)kore_malloc(sizeof(*c)); + if ((c->fd = accept(l->fd, (struct sockaddr *)&(c->sin), &len)) == -1) { + free(c); + kore_log("accept(): %s", errno_s); + return (KORE_RESULT_ERROR); + } + + if (!kore_socket_nonblock(c->fd)) { + close(c->fd); + free(c); + return (KORE_RESULT_ERROR); + } + + c->owner = l; + TAILQ_INIT(&(c->send_queue)); + TAILQ_INIT(&(c->recv_queue)); + kore_event(c->fd, EPOLLIN | EPOLLET, c); + kore_log("new connection from %s", inet_ntoa(c->sin.sin_addr)); + + return (KORE_RESULT_OK); +} + +static int +kore_connection_handle(struct connection *c, int flags) +{ + return (KORE_RESULT_OK); +} + +static int +kore_socket_nonblock(int fd) +{ + int flags; + + if ((flags = fcntl(fd, F_GETFL, 0)) == -1) { + kore_log("fcntl(): F_GETFL %s", errno_s); + return (KORE_RESULT_ERROR); + } + + flags |= O_NONBLOCK; + if (fcntl(fd, F_SETFL, flags) == -1) { + kore_log("fcntl(): F_SETFL %s", errno_s); + return (KORE_RESULT_ERROR); + } + + return (KORE_RESULT_OK); +} + +static void +kore_event(int fd, int flags, void *udata) +{ + struct epoll_event evt; + + evt.events = flags; + evt.data.ptr = udata; + if (epoll_ctl(efd, EPOLL_CTL_ADD, fd, &evt) == -1) + fatal("epoll_ctl(): %s", errno_s); +} diff --git a/src/utils.c b/src/utils.c @@ -0,0 +1,102 @@ +/* + * Copyright (c) 2013 Joris Vink <joris@coders.se> + * + * Permission to use, copy, modify, and distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + */ + +#include <sys/param.h> +#include <sys/types.h> +#include <sys/socket.h> +#include <sys/queue.h> + +#include <netinet/in.h> +#include <arpa/inet.h> + +#include <errno.h> +#include <fcntl.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "kore.h" + +void * +kore_malloc(size_t len) +{ + void *ptr; + + if ((ptr = malloc(len)) == NULL) + fatal("kore_malloc(%d): %d", len, errno); + + return (ptr); +} + +void * +kore_realloc(void *ptr, size_t len) +{ + void *nptr; + + if ((nptr = realloc(ptr, len)) == NULL) + fatal("kore_realloc(%p, %d): %d", ptr, len, errno); + + return (ptr); +} + +void * +kore_calloc(size_t memb, size_t len) +{ + void *ptr; + + if ((ptr = calloc(memb, len)) == NULL) + fatal("kore_calloc(%d, %d): %d", memb, len, errno); + + return (ptr); +} + +char * +kore_strdup(const char *str) +{ + char *nstr; + + if ((nstr = strdup(str)) == NULL) + fatal("kore_strdup(): %d", errno); + + return (nstr); +} + +void +kore_log_internal(char *file, int line, const char *fmt, ...) +{ + va_list args; + char buf[2048]; + + va_start(args, fmt); + vsnprintf(buf, sizeof(buf), fmt, args); + va_end(args); + + printf("%s:%d - %s\n", file, line, buf); +} + +void +fatal(const char *fmt, ...) +{ + va_list args; + char buf[2048]; + + va_start(args, fmt); + vsnprintf(buf, sizeof(buf), fmt, args); + va_end(args); + + printf("error: %s\n", buf); + exit(1); +}