基于Linux的多线程Web Server

#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 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);

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
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值