基于Linux的多线程Web Server

来源：互联网发布：南风知我意叶小意txt 编辑：程序博客网时间：2024/05/29 14:06
#include <stdio.h>#include <stdlib.h>#include <unistd.h>#include <string.h>#include <arpa/inet.h>#include <sys/socket.h>#include <pthread.h>#define BUF_SIZE 1024#define SMALL_BUF 100void* 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);int main(int argc,char *argv[]){    int serv_sock,clnt_sock;//create sev and clnt sock;    struct sockaddr_in serv_adr,clnt_adr;    int clnt_adr_size;    char buf[BUF_SIZE];    pthread_t t_id;//create thread id    if(argc!=2)    {        printf("Usage :%s <port>\n",argv[0]);        exit(1);    }    serv_sock = socket(PF_INET,SOCK_STREAM,0);//create      memset(&serv_adr,0,sizeof(serv_adr));//init the srv addr     serv_adr.sin_family = AF_INET;//协议地址族，v4    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!");//监听请求    //监听队列大小为20，这时候相当于创建了服务器套接字（充当守门员）    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);        //线程id为t_id，NULL表示创建默认属性，传递线程的thread_main函数，最后一个        //参数是线程main函数的参数        pthread_detach(t_id);//销毁子线程            }       close(serv_sock);//关闭套接字    return 0;}//请求套接字的处理void* request_handler(void* arg){    int clnt_sock = *((int *)arg);    char read_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");    //为了实现半关闭状态，这里复制套接字，实现较好的IO分流    fgets(read_line,SMALL_BUF,clnt_read);    //读取可读套接字缓冲区中的数据    if(strstr(read_line,"HTTP/")==NULL)    {        send_error(clnt_write);        fclose(clnt_write);        fclose(clnt_read);        //return;        //exit(-1);    }    strcpy(method,strtok(read_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;        //exit(-1);    }           fclose(clnt_read);    send_data(clnt_write,ct,file_name);    //return ;}void send_data(FILE* fp,char* ct,char *file_name){    char protocol[] = "HTTP/1.0 200 ok\r\n";    char server[] = "Server:Linux web Server\r\n";    char clnt_len[] = "Content-length:2048";    char clnt_type[SMALL_BUF];    char buf[BUF_SIZE];    FILE* send_file;    sprintf(clnt_type,"Content type :%s\r\n\r\n",ct);    send_file = fopen(file_name,"r");    if(send_file==NULL)    {        send_error(fp);        return ;    }    //传输头信息    fputs(protocol,fp);    fputs(server,fp);    fputs(clnt_len,fp);    fputs(clnt_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.0 400 Bad Request\r\n";    char server[] = "Server:Linux Web Server \r\n";    char cnt_len[] = "Content-length : 2048\r\n";    char cnt_type[] = "Content-type:text/html\r\n\r\n";    char content[] = "<html><head><title></title>NETWORK</head>"     "<body><font size=+5><br>发生错误！查看请求文件名和请求方式！"     "</font></body></html>";    fputs(protocol,fp);    fputs(server,fp);    fputs(cnt_len,fp);    fputs(cnt_type,fp);    fflush(fp);}void error_handling(char *message){    fputs(message,stderr);    fputc('\n',stderr);    exit(1);}
阅读全文
0 0