webby/server.c at master · winterific/webby · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include "server.h"

HttpServer *ServerInit(
	int portno,
	ConnHandler connHandler,
	ErrorHandler errorHandler,
	ExitHandler exitHandler
)
{
	HttpServer *server = (HttpServer*)calloc(1, sizeof(HttpServer));
	server->portno = portno;
	server->loop = 1;
	server->connHandler = connHandler;
	server->errorHandler = errorHandler;
	server->exitHandler = exitHandler;
	return server;
}

void ServerStop(HttpServer *server)
{
	fprintf(stdout, "\nShutting down...\n");
	server->loop = 0;
	close(server->sockfd);
}

void ServerStart(HttpServer *server)
{
	socklen_t clilen;
	struct sockaddr_in serv_addr, cli_addr;

	server->sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (server->sockfd < 0)
	{
		server->errorHandler("Error opening socket.");
	}

	int value = 1;
	setsockopt(server->sockfd, SOL_SOCKET, SO_REUSEADDR, &value, sizeof(value));
	bzero((char*) &serv_addr, sizeof(serv_addr));
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = INADDR_ANY;
	serv_addr.sin_port = htons(server->portno);

	if (bind(server->sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
	{
		server->errorHandler("Error on binding.");
	}
	listen(server->sockfd, 5);

	clilen = sizeof(cli_addr);
	pid_t pid;

	fprintf(stdout, "Server listening on %s:%d...\n", "127.0.0.1", server->portno);

	while(server->loop)
	{
		server->newsockfd = accept(server->sockfd, (struct sockaddr *) &cli_addr, &clilen);
		if (server->newsockfd < 0)
		{
			server->errorHandler("ERROR on accept.");
		}
		pid = fork();
		if (pid < 0)
		{
			server->errorHandler("ERROR on fork.");
		}
		if (pid == 0)
		{
			close(server->sockfd);
			server->connHandler(server->newsockfd);
			close(server->newsockfd);
			exit(0);
		}
		else
		{
			close(server->newsockfd);
		}
	}
}