Beginnings of network support.
TODO: rewrite to be addressfamily-independent. (getaddrinfo)
This commit is contained in:
parent
08dc193e53
commit
788724b4ff
1
src/Makefile
Normal file
1
src/Makefile
Normal file
@ -0,0 +1 @@
|
|||||||
|
CFLAGS=-g -std=c99
|
86
src/net.c
Normal file
86
src/net.c
Normal file
@ -0,0 +1,86 @@
|
|||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <stdarg.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <poll.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#define MAX_CLIENTS 2
|
||||||
|
#define SRV_FDS 1
|
||||||
|
|
||||||
|
int servsock;
|
||||||
|
struct pollfd pollfds[SRV_FDS+MAX_CLIENTS];
|
||||||
|
int nclients = 0;
|
||||||
|
|
||||||
|
int
|
||||||
|
errx(int eval, const char *fmt, ...) {
|
||||||
|
va_list ap;
|
||||||
|
vfprintf(stderr, fmt, ap);
|
||||||
|
exit(eval);
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
net_server(int port) {
|
||||||
|
struct sockaddr_in serveraddr;
|
||||||
|
serveraddr.sin_family = AF_INET;
|
||||||
|
serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||||||
|
serveraddr.sin_port = htons(port);
|
||||||
|
|
||||||
|
servsock=socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||||
|
if (servsock<0)
|
||||||
|
errx(1, "socket: %s", strerror(errno));
|
||||||
|
if (bind(servsock, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
|
||||||
|
errx(1, "bind port %d: %s", port, strerror(errno));
|
||||||
|
if (listen(servsock, 10) < 0)
|
||||||
|
errx(1, "listen: %s", strerror(errno));
|
||||||
|
pollfds[0].fd=servsock;
|
||||||
|
pollfds[0].events=POLLIN|POLLERR|POLLHUP;
|
||||||
|
for (int i=SRV_FDS; i<(SRV_FDS+MAX_CLIENTS); i++) {
|
||||||
|
pollfds[0].fd=-1;
|
||||||
|
pollfds[0].events=0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int
|
||||||
|
net_poll() {
|
||||||
|
int npoll;
|
||||||
|
npoll=poll(pollfds, SRV_FDS+MAX_CLIENTS, -1);
|
||||||
|
if (npoll<0)
|
||||||
|
errx(1, "poll: %s", strerror(errno));
|
||||||
|
if (pollfds[0].revents) {
|
||||||
|
struct sockaddr_in clientaddr;
|
||||||
|
int cfd;
|
||||||
|
socklen_t slen;
|
||||||
|
cfd=accept(servsock, (struct sockaddr *)&clientaddr, &slen);
|
||||||
|
if (cfd<0)
|
||||||
|
errx(1, "accept: %s", strerror(errno));
|
||||||
|
// FIXME log, shout, or something
|
||||||
|
if (nclients>MAX_CLIENTS) {
|
||||||
|
#define STR_SERVER_FULL "Sorry, the server is full"
|
||||||
|
write(cfd, STR_SERVER_FULL, sizeof(STR_SERVER_FULL)-1);
|
||||||
|
close(cfd);
|
||||||
|
} else {
|
||||||
|
int i;
|
||||||
|
for (i=SRV_FDS; i<(SRV_FDS+MAX_CLIENTS); i++)
|
||||||
|
if (pollfds[i].fd=-1) break;
|
||||||
|
pollfds[i].fd=cfd;
|
||||||
|
pollfds[i].events=POLLIN|POLLERR|POLLHUP;
|
||||||
|
nclients++;
|
||||||
|
}
|
||||||
|
npoll--;
|
||||||
|
}
|
||||||
|
for (int i=SRV_FDS; npoll && i<(SRV_FDS+MAX_CLIENTS); i++) {
|
||||||
|
if (pollfds[SRV_FDS+i].revents) {
|
||||||
|
int n;
|
||||||
|
char buf[100];
|
||||||
|
n=read(pollfds[SRV_FDS+i].fd, &buf, sizeof(buf));
|
||||||
|
// parse buf
|
||||||
|
// handle disconnections
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user