server.c

#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <signal.h>
#include <fcntl.h>
#include "print_file.c"
#include "get_mime.c"
#include "directory_listing.c"
#include "error.c"
#include "validate_request.c"
#include "get_headers.c"
#include "handle_cgi.c"
#include "request_type.c"

#define WEBROOT "./www"
#define MIMEFILE "/etc/mime.types"
#define LOGFILE "/var/log/httpd.log"
#define GID 65534
#define UID 65534

#define LOKAL_PORT 80
#define BAK_LOGG 10 // Størrelse på for kø ventende forespørsler

void daemonize()
{
	// Forker og lukker foreldreprosessen
	if (fork()!=0){
		exit(0);
	}

	// Setter til sesjonsleder
	setsid();

	// Ignorerer SIGHUP
	signal(SIGHUP, SIG_IGN);

	// Forker og lukker foreldreprosessen
	if (fork()!=0){
		exit(0);
	}

	// Binder stderr til loggfila
	close(2);
	open(LOGFILE, O_WRONLY | O_APPEND | O_CREAT);

	// Stenger stdout
	close(1);
	close(0);
}

int main ()
{

	struct sockaddr_in	lok_adr;
	int sd, ny_sd;

	// Åpner mimefil
	FILE* mimefile = fopen(MIMEFILE, "r");

	// Demoniserer prosessen
	daemonize();

	// Setter opp socket-strukturen
	sd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

	// For at operativsystemet ikke skal holde porten reservert etter tjenerens død
	setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &(int){ 1 }, sizeof(int));

	// Initierer lokal adresse
	lok_adr.sin_family			= AF_INET;
	lok_adr.sin_port				= htons((u_short)LOKAL_PORT);
	lok_adr.sin_addr.s_addr = htonl(				 INADDR_ANY);

	// Kobler sammen socket og lokal adresse
	if ( 0==bind(sd, (struct sockaddr *)&lok_adr, sizeof(lok_adr)) )
		dprintf(2, "Prosess %d er knyttet til port %d.\n", getpid(), LOKAL_PORT);
	else {
		dprintf(2, "Kunne ikke binde port\n");
		exit(1);
	}


	// Venter på forespørsel om forbindelse
	listen(sd, BAK_LOGG);
	while(1){

		// Aksepterer mottatt forespørsel og tar vare på ip-adressen til klienten
		struct sockaddr_in client_addr;
		socklen_t slen = sizeof(client_addr);
		ny_sd = accept(sd, (struct sockaddr *)&client_addr, &slen);


		if(0==fork()) {

			dup2(ny_sd, 1); // redirigerer socket til standard utgang

			// åpner socket og leser første linje inn i variabel
			char* line = NULL;
			FILE* request = fdopen(ny_sd, "r");
			size_t len;
			getline(&line, &len, request);

			read_headers(request);

			// Remove escape characters \r and \n
			while(strchr(line, '\r') != NULL)
			{
				strchr(line, '\r')[0] = '\0';
			}

			while(strchr(line, '\n') != NULL)
			{
				strchr(line, '\n')[0] = '\0';
			}

			// henter url fra linjen
			char* saveptr = NULL;
			char* line_copy = malloc(strlen(line) + 1);
			strcpy(line_copy, line);
			strtok_r(line_copy, " ", &saveptr);
			char* url = malloc(strlen(line));
			strcpy(url, strtok_r(NULL, " ", &saveptr));

			// Logger requesten til konsollen
			dprintf(2, "%s - %s - ", inet_ntoa(client_addr.sin_addr), line);

			// Finner path og query fra url ved å terminere på ?
			char* path = malloc(strlen(url) + 1);
			strcpy(path, url);
			char* query = strchr(path, '?');
			if(query == NULL)
			{
				query = "";
			}
			else
			{
				query[0] = '\0';
				query++;
			}

			// Stripper vekk fragment fra path/query og legger det i egen variabel
			char* fragment;
			if (strcmp(query,"") == 0) fragment = strchr(path, '#');
			else fragment = strchr(query, '#');
			if(fragment == NULL)
			{
				fragment = "";
			}
			else
			{
				fragment[0] = '\0';
				fragment++;
			}

			validate_request(line);

			char* req_type = request_type(line);

			if(strncmp(url, "/cgi-bin/", 9) == 0)
			{
				char* prefixed_path = malloc(1000);
				sprintf(prefixed_path, "/www%s", path);

				// Sjekker om fila finnes
				if (access( prefixed_path, F_OK ) == -1){
					error(404);
				}

				// Endrer bruker og gruppe til nobody
				setgid(GID);
				setuid(UID);

				// Kjører CGI programmet
				handle_cgi(prefixed_path, query, req_type, request);
			}


			// Chroot til webroten
			chdir(WEBROOT);
			chroot(".");

			// Endrer bruker og gruppe til nobody
			setgid(GID);
			setuid(UID);

			// Sjekker om fila finnes
			if (access( path, F_OK ) == -1){
				error(404);
			}

			if (strcmp(req_type, "GET") == 0){
				// Sjekker om url er /
				if (strcmp(url,"/")==0){
					printf("HTTP/1.1 200 OK\n");
					printf("Content-Type: text/plain\n");
					printf("\n");
					directory_listing(url);

					dprintf(2, "200 OK\n");
					fflush(stdout);
					shutdown(ny_sd, SHUT_RDWR);
					exit(0);
				}

				char* mime = get_mime(url, mimefile);

				// Hvis filtypen ikke gjenkjennes, gi feilmelding
				if (mime==NULL){
					error(415);
				}

				printf("HTTP/1.1 200 OK\n");
				printf("Content-Type: %s\n", mime);
				printf("\n");
				fflush(stdout); // nødvendig for å få riktig rekkefølge
				print_file(url);
			}

			else if (strcmp(req_type, "POST") == 0){
				if(strncmp(url, "/cgi-bin/", 9) != 0)
				{
					error(405);
				}
			}

			dprintf(2, "200 OK\n");
			fflush(stdout);

			// Sørger for å stenge socket for skriving og lesing
			// NB! Frigjør ingen plass i fildeskriptortabellen
			shutdown(ny_sd, SHUT_RDWR);
			exit(0);
		}

		else {
			close(ny_sd);
		}
	}
	return 0;
}