Linux下实现web服务器

说明：暂时只是实现了静态网页的响应

 #include <stdio.h>
 #include <sys/types.h>          /* See NOTES */
 #include <sys/socket.h>
 #include <string.h>
 #include <arpa/inet.h>
 #include <unistd.h>
 #include <signal.h>
 #include <sys/wait.h>
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <stdlib.h>

 #define PORT 8000
 char* msg_err="HTTP/1.1 404 NO FOUND\r\nContent-Type:text/html\r\n\r\n";
 char* msg_text="HTTP/1.1 200 OK\r\nContent-Type:text/html\r\n\r\n";
 char* msg_jpg="HTTP/1.1 200 OK\r\nContent-Type:image/jpg\r\n\r\n";
 char* msg_png="HTTP/1.1 200 OK\r\nContent-Type:image/png\r\n\r\n";
 char* work_dir="/home/soft01/Hector/Unixc/webProject/hah/www";
 char* work_dir_err="/home/soft01/Hector/Unixc/webProject/hah/err_404.html";

 //定义http协议 结构体
 typedef struct http{
     char method[];
     char filename[];
     char protol[];
 }utr_t;

 //自定义信号处理函数
 //回收子进程
 void sig_do(){
     wait(NULL);
     return;
 }

 //响应
 void write_client(char* str,int fd,char* msg){
     char status[];
     int fd_open=open(str,O_RDONLY);
     if(fd_open==-){
         perror("open");
         int fd_open=open(work_dir_err,O_RDONLY);
         int nlen=read(fd_open,status,);
         write(fd,msg_err,strlen(msg_err));
         write(fd,status,nlen);
         return;
     }
     int nlen=read(fd_open,status,);
     write(fd,msg,strlen(msg));
     write(fd,status,nlen);
     return;
 }

 //判断图片文本
 void judge_text_or_png(char* str,int fd){
     char* msg="<html>\n<body>\n<h1>404\n</body>\n</html>\n";
     if(str==NULL){return;}
     char* tmp=NULL;
     if((tmp=strstr(str,".jpg"))!=NULL){
         write_client(str,fd,msg_jpg);
     }
     else if((tmp=strstr(str,".png"))!=NULL){
         write_client(str,fd,msg_png);
     }
     else if((tmp=strstr(str,".html"))!=NULL){//文本
         write_client(str,fd,msg_text);
     }
     else{
         write(fd,msg_err,strlen(msg_err));
         write(fd,msg,strlen(msg));
     }
     return;
 }

 //子进程响应客户端
 void request_do(int fd){
     utr_t http;
     char path[];
     char buf[];
     int len=read(fd,buf,);
     sscanf(buf,"%s %s %s\r\n",http.method,http.filename,http.protol);
     if(strcmp(http.filename,"/")==){
         strcpy(http.filename,"/index.html");
     }

     strcpy(path,work_dir);
     strcat(path,http.filename);
     strcpy(http.filename,path);
     printf("method:%s filename:%s protol:%s\n", http.method,http.filename,http.protol);

     //判断图片文本
     judge_text_or_png(http.filename,fd);

     return;
 }

 int main(){
     struct sockaddr_in serv;
     struct sockaddr_in clie;
     socklen_t clie_len;
     signal(SIGCHLD,sig_do);
     //创建socket通讯端口
     int sfd=socket(AF_INET,SOCK_STREAM,);
     if(sfd==-){
         perror("socket");
         return -;
     }

     //初始化服务器serv
     serv.sin_family=AF_INET;
     serv.sin_port=htons(PORT);
     serv.sin_addr.s_addr=htonl(INADDR_ANY);
     //将sfd绑定到服务器
     int b=bind(sfd,(struct sockaddr*)&serv,sizeof(serv));
     if(b==-){
         perror("bind");
         return -;
     }
     listen(sfd,);
     while(){
         clie_len=sizeof(clie);
         //建立和客户端的连接
         int cfd=accept(sfd,(struct sockaddr*)&clie,&clie_len);
         if(cfd==-){
             perror("accept");
             return -;
         }
         pid_t pid=fork();
         if(pid==-){
             perror("pid");
             return -;
         }
         if(pid==){//子进程
             close(sfd);
             //响应客户端请求
             request_do(cfd);
             close(cfd);
             exit();
         }
         else{//父进程
             close(cfd);
         }

     }
     return ;
 }