tinyhttp

Posted 2022-12-19 letlifestop

/*************************************************************************
    > File Name: http.c
    > Author: 
    > Mail: 
    > Created Time: 2019年09月30日 星期一 16时32分14秒
 ************************************************************************/

#include<stdio.h>
#include<stdlib.h>
#include<sys/socket.h>
#include<sys/types.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<ctype.h>
#include<string.h>
#include<sys/stat.h>
#include<pthread.h>
#include<sys/wait.h>

# define SERVER_STRING "Server: jdhttpd/0.1.0\r\n"
# define ISspace(x) isspace((int)(x))
void accept_request(void *);
void bad_request(int);
void cat(int, FILE *);
void cannot_excute(int);
void error_dis(const char *);
void execute_cgi(int, const char *, const char *, const char *);
void headers(int, const char *);
void not_found(int);
void serve_file(int, const char *);
void not_found(int);
void serve_file(int, const char *);
int startup(int *);
void unimplemented(int);
int get_line(int, char *, int);
int pthread_create(pthread_t *thread,const pthread_attr_t *attr, void *(*start_routine)(void *), void *arg);

void accept_request(void *arg)
    
    char buf[1024];
    char method[255];
    char url[255];
    char path[512];
    int client = (intptr_t)arg;
    int cgi = 0;
    struct stat st;
    char *query_string = NULL;
    int numchars = 0;
    int i, j;

    numchars = get_line(client, buf, sizeof(buf));
    i = 0, j = 0;
    while(!ISspace(buf[i]) && (i < sizeof(method) - 1))
        method[i] = buf[i];
        i++;
    
    j = i;
    method[i] = ‘\0‘;

    if(strcasecmp(method, "GET") && strcasecmp(method, "POST"))
        unimplemented(client);
        return ;
    
    
    if((strcasecmp, "POST") == 0)
        cgi = 1;
    
    
    i = 0;
    while(ISspace(buf[j]) && (j < numchars))
        j++;
    
    while(!ISspace(buf[j]) && (i < sizeof(url) - 1) && (j < numchars))
        url[i] = buf[j];
        i++;
        j++;
    
    url[i] = ‘\0‘;

    if(strcasecmp(method, "GET") == 0)
        query_string = url;
        while((*query_string != ‘?‘) && (*query_string != ‘\0‘))
            query_string++;
        
        if(*query_string == ‘?‘)
            cgi = 1;
            *query_string = ‘\0‘;
            query_string++;
        
    

    sprintf(path, "htdocs%s", url);
    if(path[strlen(path) - 1] == ‘/‘)
        strcat(path, "index.html");
    
    if(stat(path, &st) == -1)
        while((numchars > 0) && strcmp("\n", buf))
            numchars = get_line(client, buf, sizeof(buf));
        
        not_found(client);
    
    else 
        if((st.st_mode & S_IFMT) == S_IFDIR)
            strcat(path, "/index.html");
        
        // 如果不是cgi, 就直接把服务器文件返回
        if((st.st_mode & S_IXUSR) || (st.st_mode & S_IXGRP) || 
           (st.st_mode & S_IXOTH))
        cgi = 1;
        if(!cgi)
        serve_file(client, path);
        else 
        // client 客户端对应的socket,　path 是对应的路径，method是请求方法
        // query_string 是请求的内容
        execute_cgi(client, path, method, query_string);
    
    close(client);
    return ;


void execute_cgi(int client, const char *path, const char* method, const char * query_string )
    
    char buf[1024];
    char c;
    int numchars = 1, content_length = -1;
    buf[0] = ‘A‘;
    buf[1] = ‘\0‘;
    int cgi_output[2];
    int cgi_input[2];
    int status;
    pid_t pid;

    if(strcasecmp(method, "GET") == 0)
        while((numchars > 0 ) && (strcmp("\n", buf))) 
            numchars = get_line(client, buf, sizeof(buf));
        
    
    else if(strcasecmp(method, "POST") == 0)
        numchars = get_line(client, buf, sizeof(buf));
        
        while((numchars > 0) && strcmp("\n", buf))
            buf[15] = ‘\0‘;
        if(strcasecmp(buf, "Content-Length:") == 0)
            content_length = atoi(&buf[16]);
        
        
         numchars = get_line(client, buf, sizeof(buf));
        

        if(content_length == -1)
            bad_request(client);
            return ;
        
    
    
    if(pipe(cgi_output) < 0)
        cannot_excute(client);
        return ;
    

    if(pipe(cgi_input) < 0)
        cannot_excute(client);
    

    
    if((pid = fork()) < 0 )
        cannot_excute(client);
    

    sprintf(buf, "HTTP/1.0 200 OK\r\n");
    send(client, buf, sizeof(buf), 0);

    if(pid == 0)
        
        char meth_env[255];
        char query_env[255];
        char length_env[255];

        dup2(cgi_input[0], 0);
        dup2(cgi_output[1], 1);
        close(cgi_input[1]);
        close(cgi_output[0]);
        
        sprintf(meth_env, "REQUEST_METHOD=%s", method);
        putenv(meth_env);
        
        if(strcasecmp(method, "GET") == 0)
            sprintf(query_env, "QUERY_STRING=%s", query_string);
            putenv(query_env);
        
        else 
            sprintf(query_env, "QUERY_STRING=%s", query_string);
            putenv(length_env);
        

        execl(path, path, NULL);
        exit(0);
    
    else
        close(cgi_output[1]);
        close(cgi_input[0]);
        
        if(strcasecmp(method, "POST") == 0)
        int i;
            for(i = 0; i < content_length; i++)
            recv(client, &c, 1, 0);
            write(cgi_input[1], &c, 1);
            
        

        while(read(cgi_output[0], &c, 1) > 0)
            send(client, &c, 1, 0);
        

        close(cgi_output[0]);
        close(cgi_input[1]);

        waitpid(pid, &status, 0);
    

    return ;


void cannot_excute(int client)
    
    char buf[1024];

    sprintf(buf, "HTTPS/1.0 500 Internal Server ERror\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "Cpntent-type: text/html\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "<P>Error prohibited CGI execution.\r\n");
    send(client, buf, strlen(buf), 0);
    return ;


void bad_request(int client) 
    char buf[1024];

    sprintf(buf, "HTTP/1.0 400 BAD REQUEST\r\n");
    send(client, buf, sizeof(buf), 0);
    sprintf(buf, "Content-type: text/html\r\n");
    send(client, buf, sizeof(buf), 0);
    sprintf(buf, "\r\n");
    send(client, buf, sizeof(buf), 0);
    sprintf(buf, "<P>Your browser sent a bad request. ");
    send(client, buf, sizeof(buf), 0);;
    sprintf(buf, "such as a POST without a content_length.\r\n");
    send(client, buf, sizeof(buf), 0);
    return ;

void serve_file(int client, const char *filename)
    
    FILE *resource = NULL;
    int numchars = 1;
    char buf[1024];

    buf[0] = ‘A‘;
    buf[1] = ‘\0‘;
    while((numchars > 0) && strcmp("\n", buf))
        numchars = get_line(client, buf, sizeof(buf));
    

    resource = fopen(filename, "r");
    if(resource == NULL)
        not_found(client);
    
    else 
        // 先把http头部信息发送过去，再把文件中的内容通过socket发送出去
        headers(client, filename);
        cat(client, resource);
    
    return ;


void headers(int client, const char *filename)
    char buf[1024];
    (void)filename;

    strcpy(buf, "HTTP/1.0 200 OK\r\n");
    send(client, buf, strlen(buf), 0);
    strcpy(buf, SERVER_STRING);
    send(client, buf, strlen(buf), 0);
    strcpy(buf, "\r\n");
    send(client, buf, strlen(buf), 0);
    return ;


void cat(int client, FILE *resource)
    char buf[1024];

    fgets(buf, sizeof(buf), resource);
    while(!feof(resource))
       send(client, buf, strlen(buf), 0);
       fgets(buf, sizeof(buf), resource);
    
    return ;
 

void not_found(int client)
    char buf[1024];

    sprintf(buf, "HTTP/1.0 404 NOT FOUND\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, SERVER_STRING);
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "Content-Type: text/html\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "<HTML><TITLe>Not Found</TITLE>\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "<BODY><P>The server could notfullfill\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "your request because the resource specified\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "is unavauilable or nonexistent/\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "</BODY></HTML>\r\n");
    send(client, buf, strlen(buf), 0);
    return ;


void unimplemented(int client)
    
    char buf[1024];

    sprintf(buf, "HTTP/1.0 501 Method Not Implemented\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf,SERVER_STRING);
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "Content-Type: text/html\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "<HTML><HEAD><TITLE>Method not Implemented\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "</TITLE></HEAD>\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "<BODY><P>HTTP request method not supported.\r\n");
    send(client, buf, strlen(buf), 0);
    sprintf(buf, "</BODY></HTML>\r\n");
    send(client, buf, strlen(buf), 0);
    return ;


int get_line(int sock, char *buf, int size)
    int i = 0;
    char c = ‘\0‘;
    int n;
    
    while((i < size - 1) && (c != ‘\n‘))
        n = recv(sock, &c, 1, 0);
        if(n > 0)
            if(c == ‘\r‘)
                n = recv(sock, &c, 1, MSG_PEEK);
                if((n > 0) && (c == ‘\n‘))
                    recv(sock, &c, 1, 0);
                
                else 
                    c = ‘\n‘;
                
            
            buf[i] = c;
            i++;
        
        else c = ‘\n‘;
    

    buf[i] = ‘\0‘;

    return i;


int  startup(int *port)
    int httpd = -1, on = 1;
    struct sockaddr_in name;

    httpd = socket(AF_INET,SOCK_STREAM, 0);
    if(httpd == -1)
        perror("create socket error");
        exit(1);
    
    // SO_REUSEADDR允许地址重用，每次bind的时候，
    // 如果这个端口正在被使用，bind就会把端口抢过来  
    if((setsockopt(httpd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0))
        perror("setsockopt error");
        exit(1);
    

    memset(&name, 0, sizeof(name));
    name.sin_family =  AF_INET;
    name.sin_port = htons(*port);
    name.sin_addr.s_addr = htonl(INADDR_ANY);
    if(bind(httpd, (struct sockaddr *)&name,sizeof(name)) < 0)
        perror("bind error");
        exit(1);
    

    if(*port == 0)
        socklen_t namelen = sizeof(name);
        if(getsockname(httpd, (struct sockaddr *)&name, &namelen) < 0)
            perror("get sock error");
            exit(1);
        
        *port = ntohs(name.sin_port);
    

     if(listen(httpd, 5) < 0)
        perror("set listen error");
        exit(1);
        
    return httpd;


int main()
    int serv_sock = -1;
    int port = 0;
    int clie_sock = -1;
    struct sockaddr_in client_name;
    pthread_t newthread;
    socklen_t len_client_name;

    serv_sock = startup(&port);
    printf("httpd running on port %d\n", port);
    
    len_client_name = sizeof(client_name);
    while(1)
        clie_sock = accept(serv_sock, (struct sockaddr *)&client_name, &len_client_name);
        if(clie_sock == -1)
            perror("create accept error");
            exit(1);
          

        if(pthread_create(&newthread, NULL, (void *)accept_request, (void *)(intptr_t)(clie_sock)) != 0)
            perror("create pthread error");
            exit(1);
        
        
    
    close(serv_sock);
    return 0;