TCP IP网络编程2 简单的Web服务器

实现简单的web服务器有三种处理请求的方式：
多进程、多线程和I/O复用

下面利用多线程处理连接的请求，每个连接新增一个线程来处理
返回静态文件，不存在则返回400

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <pthread.h>
#include <sys/stat.h>

#define BUF_SIZE 1024
#define SMALL_BUF 100

void *request_handler(void *arg);
void send_data(FILE *fp, char *ct, char *file_name);
char *content_type(char *file);
void send_error(FILE *fp);
void error_handling(char *message);
long get_filesize(char *path);

int main(int argc, char *argv[])
{
        int serv_sock, clnt_sock;
        struct sockaddr_in serv_adr, clnt_adr;
        int clnt_adr_size;
        char buf[BUF_SIZE];
        pthread_t t_id;

        if(argc != 2)
        {
                printf("Usage : %s <port>\n", argv[0]);
                exit(1);
        }

        serv_sock = socket(PF_INET, SOCK_STREAM, 0);
        if(serv_sock == -1)
                error_handling("socket() error");

        memset(&serv_adr, 0, sizeof(serv_adr));
        serv_adr.sin_family = AF_INET;
        serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
        serv_adr.sin_port = htons(atoi(argv[1]));

        if(bind(serv_sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr)) == -1)
                error_handling("bind() error");
        if(listen(serv_sock, 20) == -1)
                error_handling("listen() error");

        while(1)
        {
                clnt_adr_size = sizeof(clnt_adr);
                clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_adr, &clnt_adr_size);
                printf("Connection Request: %s:%d\n",
                                inet_ntoa(clnt_adr.sin_addr), ntohs(clnt_adr.sin_port));
                pthread_create(&t_id, NULL, request_handler, &clnt_sock);
                pthread_detach(t_id);
        }

        close(serv_sock);

        return 0;
}

void *request_handler(void *arg)
{
        int clnt_sock = *(int *)arg;
        char req_line[SMALL_BUF];
        FILE *clnt_read;
        FILE *clnt_write;

        char method[10];
        char ct[15];
        char file_name[30];

        clnt_read = fdopen(clnt_sock, "r");
        clnt_write = fdopen(dup(clnt_sock), "w");
        fgets(req_line, SMALL_BUF, clnt_read);
        if(strstr(req_line, "HTTP/") == NULL)
        {
                send_error(clnt_write);
                fclose(clnt_read);
                fclose(clnt_write);
                return;
        }

        strcpy(method, strtok(req_line, " /"));
        strcpy(file_name, strtok(NULL, " /"));
        strcpy(ct, content_type(file_name));
        if(strcmp(method, "GET") != 0)
        {
                send_error(clnt_write);
                fclose(clnt_read);
                fclose(clnt_write);
                return;
        }

        fclose(clnt_read);
        send_data(clnt_write, ct, file_name);
}

void send_data(FILE *fp, char *ct, char *file_name)
{
        char protocol[] = "HTTP/1.1 200 OK\r\n";
        char server[] = "Server: Linux Web Server \r\n";
        char cnt_len[30] = {'\0'};
        char cnt_type[SMALL_BUF];
        char buf[BUF_SIZE];
        FILE *send_file;

        sprintf(cnt_type, "Content-type:%s\r\n\r\n", ct);
        send_file = fopen(file_name, "r");
        if(send_file == NULL)
        {
                send_error(fp);
                return;
        }

        sprintf(cnt_len, "Content-length:%ld\r\n", get_filesize(file_name));
        fputs(protocol, fp);
        fputs(server, fp);
        fputs(cnt_len, fp);
        fputs(cnt_type, fp);

        while(fgets(buf, BUF_SIZE, send_file) != NULL)
        {
                fputs(buf, fp);
                fflush(fp);
        }
        fflush(fp);
        fclose(fp);
}

char *content_type(char *file)
{
        char extension[SMALL_BUF];
        char file_name[SMALL_BUF];
        strcpy(file_name, file);
        strtok(file_name, ".");
        strcpy(extension, strtok(NULL, "."));

        if(!strcmp(extension, "html") || !strcmp(extension, "htm"))
                return "text/html";
        else
                return "text/plain";
}

void send_error(FILE *fp)
{
        char protocol[] = "HTTP/1.1 400 Bad Request\r\n";
        char server[] = "Server:Linux Web Server \r\n";
        char cnt_len[30] = {'\0'};
        char cnt_type[] = "Content-type:text/html\r\n\r\n";
        char content[] = "<html><head><title>Network</title></head>"
                "<body><font size=+5><br>Error occurred, please check the file name!"
                "</font></body></html>";

        sprintf(cnt_len, "Content-length:%d\r\n", sizeof(content));
        fputs(protocol, fp);
        fputs(server, fp);
        fputs(cnt_len, fp);
        fputs(cnt_type, fp);
        fputs(content, fp);
        fflush(fp);
        fclose(fp);
}

void error_handling(char *message)
{
        fputs(message, stderr);
        fputc('\n', stderr);
        exit(1);
}

long get_filesize(char *path)
{
        struct stat statbuff;
        if(stat(path, &statbuff) == -1)
                return -1;

        return statbuff.st_size;
}