nuclear@2: #include <stdio.h>
nuclear@2: #include <stdlib.h>
nuclear@2: #include <string.h>
nuclear@2: #include <errno.h>
nuclear@2: #include <assert.h>
nuclear@2: #include <unistd.h>
nuclear@2: #include <fcntl.h>
nuclear@2: #include <sys/socket.h>
nuclear@2: #include <arpa/inet.h>
nuclear@2: #include "srv.h"
nuclear@2: #include "dynarr.h"
nuclear@2: 
nuclear@2: static int lis_sock = -1, max_socket = -1;
nuclear@2: static int *csock;
nuclear@2: 
nuclear@2: int srv_init(int port)
nuclear@2: {
nuclear@2: 	struct sockaddr_in addr;
nuclear@2: 
nuclear@2: 	if(lis_sock != -1) {
nuclear@2: 		fprintf(stderr, "%s: already running\n", __func__);
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	if((lis_sock = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
nuclear@2: 		fprintf(stderr, "%s: failed to create listening socket\n", __func__);
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 	if(!(csock = dynarr_alloc(1, sizeof *csock))) {
nuclear@2: 		fprintf(stderr, "%s: failed to allocate client socket array\n", __func__);
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 	csock[0] = lis_sock;
nuclear@2: 	max_socket = lis_sock;
nuclear@2: 
nuclear@2: 	fcntl(lis_sock, F_SETFD, fcntl(lis_sock, F_GETFD) | O_NONBLOCK);
nuclear@2: 
nuclear@2: 	memset(&addr, 0, sizeof addr);
nuclear@2: 	addr.sin_family = AF_INET;
nuclear@2: 	addr.sin_port = htons(port);
nuclear@2: 	addr.sin_addr.s_addr = INADDR_ANY;
nuclear@2: 	if(bind(lis_sock, (struct sockaddr*)&addr, sizeof addr) == -1) {
nuclear@2: 		fprintf(stderr, "%s: failed to bind port %d\n", __func__, port);
nuclear@2: 		close(lis_sock);
nuclear@2: 		lis_sock = -1;
nuclear@2: 		return -1;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	return 0;
nuclear@2: }
nuclear@2: 
nuclear@2: void srv_shutdown(void)
nuclear@2: {
nuclear@2: 	int i, sz = dynarr_size(csock);
nuclear@2: 	for(i=0; i<sz; i++) {
nuclear@2: 		close(csock[i]);
nuclear@2: 	}
nuclear@2: 	dynarr_free(csock);
nuclear@2: 	csock = 0;
nuclear@2: 	lis_sock = -1;
nuclear@2: 	max_socket = -1;
nuclear@2: }
nuclear@2: 
nuclear@2: int srv_num_sockets(void)
nuclear@2: {
nuclear@2: 	return dynarr_size(csock);
nuclear@2: }
nuclear@2: 
nuclear@2: int *srv_sockets(void)
nuclear@2: {
nuclear@2: 	return csock;
nuclear@2: }
nuclear@2: 
nuclear@2: int srv_max_socket(void)
nuclear@2: {
nuclear@2: 	return max_socket;
nuclear@2: }
nuclear@2: 
nuclear@2: int srv_handle(int s)
nuclear@2: {
nuclear@2: 	static char buf[1024];
nuclear@2: 	int sz;
nuclear@2: 
nuclear@2: 	if(s == lis_sock) {
nuclear@2: 		/* incoming connection */
nuclear@2: 		struct sockaddr_in addr;
nuclear@2: 		int addr_size;
nuclear@2: 
nuclear@2: 		if((s = accept(lis_sock, (struct sockaddr*)&addr, (int*)&addr_size)) == -1) {
nuclear@2: 			fprintf(stderr, "%s: failed to accept incoming connection\n", __func__);
nuclear@2: 			return -1;
nuclear@2: 		}
nuclear@2: 		printf("%s: incoming connection from %s:%d\n", __func__,
nuclear@2: 				inet_ntoa(addr.sin_addr), ntohs(addr.sin_port));
nuclear@2: 		csock = dynarr_push(csock, &s);
nuclear@2: 		assert(csock);
nuclear@2: 		return 0;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	/* handle client */
nuclear@2: 	while((sz = read(s, buf, sizeof buf - 1)) > 0) {
nuclear@2: 		printf("%s: got data: %s\n", __func__, buf);
nuclear@2: 	}
nuclear@2: 	if(sz < 0 && errno != EAGAIN) {
nuclear@2: 		/* client closed connection probably */
nuclear@2: 		int i, num_clients = dynarr_size(csock);
nuclear@2: 		printf("%s: removing client\n", __func__);
nuclear@2: 		close(s);
nuclear@2: 		for(i=0; i<num_clients; i++) {
nuclear@2: 			if(csock[i] == s) {
nuclear@2: 				csock[i] = csock[num_clients - 1];
nuclear@2: 				csock = dynarr_pop(csock);
nuclear@2: 				break;
nuclear@2: 			}
nuclear@2: 		}
nuclear@2: 		assert(i < num_clients);
nuclear@2: 	}
nuclear@2: 	return 0;
nuclear@2: }
nuclear@2: 
nuclear@2: void srv_send_frame(unsigned char *frame, int xsz, int ysz)
nuclear@2: {
nuclear@2: }