码迷,mamicode.com
首页 > Web开发 > 详细

小型web服务器

时间:2016-06-03 23:10:26      阅读:475      评论:0      收藏:0      [点我收藏+]

标签:web

#pragma once                                                                    
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<sys/sendfile.h>
#include<fcntl.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<assert.h>
#include<pthread.h>
#include<errno.h>
#define _DET_PAGE_ index.html
#define _SIZE_ 1024
static void printLog(const char* const str,const char* const fun,int line);
void usage(const char*  const  proc);
int startup(char* ip,int port);
int get_line(int sock,char* buf,int size);
void clear_head(int sock);
void echo_html(int sock,char* path,ssize_t size);
void accept_request(int sock);
void* handle_client(void* arg);
 
#include"http.h"                                                                
static void printLog(const char* const str,const char* const fun,int line)
{
    printf("%s:%s:%d\n",str,fun,line);
}
void usage(const char*  const  proc)
{
    assert(proc);
    printLog(proc,__FUNCTION__,__LINE__);
}
int startup(char* ip,int port)
{
    int sock=socket(AF_INET,SOCK_STREAM,0);
    if(sock<0)
    {
        printLog(strerror(errno),__FUNCTION__,__LINE__);
        exit(1);
    }
    int opt=1;
    setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));
    struct sockaddr_in local;
    local.sin_family=AF_INET;
    local.sin_port=htons(port);
    local.sin_addr.s_addr=inet_addr(ip);
    if(bind(sock,(struct sockaddr*)&local,sizeof(local))<0)
    {
        printLog(strerror(errno),__FUNCTION__,__LINE__);
        exit(1);
    }
    if(listen(sock,5)<0)
    {
        printLog(strerror(errno),__FUNCTION__,__LINE__);
        exit(1);
    }
    return sock;
}
int get_line(int sock,char* buf,int size)
{
    assert(buf);
    int i=0;
    ssize_t _s=-1;
    char ch=‘\0‘;
    while(i<size-1)
    {
        _s=recv(sock,&ch,1,0);
        if(_s>0)                             
        {
            if(ch==‘\r‘)
            {
                if(_s=recv(sock,&ch,1,MSG_PEEK))
                {
                    if(_s>0&&ch==‘\n‘)
                        recv(sock,&ch,1,0);
                }
            }
            buf[i++]=ch;
        }
        else
        {
            break;
        }
    }
    buf[i]=‘\0‘;
    return i;
}
void clear_head(int sock)
{
    char buf[_SIZE_];
    buf[0]=‘\0‘;                
    ssize_t _s=-1;
    while(_s>0&&strcmp(buf,"\n")!=0)
    {
        get_line(sock,buf,sizeof(buf));
    }
}
void echo_html(int sock,char* path,ssize_t size)
{
    int fd=open(path,O_RDONLY);
    if(fd<0)
    {
        printLog(strerror(errno),__FUNCTION__,__LINE__);
        exit(1);
    }
    char* status_line="HTTP/1.1 200 ok/r/n/r/n";
    send(sock,status_line,strlen(status_line));
    if(sendfile(sock,fd,NULL,size)<0)
    {
        close(fd);
        return;
    }
    close(fd);
}                                    
void accept_request(int sock)
{
    char buf[_SIZE_];
#ifdef _DEBUG1_
    int ret=-1;
    int i=0,j=0;
    char method[_SIZE_/2];
    char url[_SIZE_];
    char path[_SIZE_];
    memset(method,‘\0‘,sizeof(method));
    memset(buf,‘\0‘,sizeof(buf));
    memset(path,‘\0‘,sizeof(path));
    memset(url,‘\0‘,sizeof(url));

    //while(ret=get_line(sock,buf,sizeof(buf)>0))
    //{
    //  if(ret==0)
    //  {
    //      printLog(strerror(errno),__FUNCTION__,__LINE__);
    //      break;
    //  }
    //  printf("%s",buf);
    //}                            
    if(get_line(sock,buf,sizeof(buf))==0)
    {
        printLog("errno",__FUNCTION__,__LINE__);
        return;
    }
    i=j=0;
    while(!isspace(buf[i])&&i<sizeof(buf)&&j<sizeof(method)-1)
    {
        method[j++]=buf[i++];//get method
    }
    method[j]=‘\0‘;
    
    j=0;
    while(isspace(buf[i]))
    {
        ++i;
    }
    while(!isspace(buf[i])&&i<sizeof(buf)&&j<sizeof(url)-1)
    {
        url[j++]=buf[i++];
    }
    int cgi=0;
    if(strncasecmp(method,"POST")!=0&&strncasecmp(method,"GET")!=0)  
    {
        printLog("errno",__FUNCTION__,__LINE__);
        return;
    }
    if(strncasecmp(method,"POST"))
    {
        cgi=1;
    }
    if(strncasecmp(method,"GET"))
    {
        char* query_string=url;
        while(*query_string!=‘\0‘&&*query_string!=‘?‘)
        {
            if(*query_string==‘?‘)
            {
                *query_string=‘\0‘;
                cgi=1;
            }
            ++query_string;
        }
        sprintf(path,"ntdos%s",url);//sprintf
        if(path[strlen(path)-1]==‘/‘)
        {                                     
            strcat(path,_DET_PAGE_);
        }

        struct stat st;
        if(stat(path,&st)<0)//not exist
        {
            printLog("errno",__FUNCTION__,__LINE__);
            return;
        }
        else if(S_ISDIR(st.st_mode))//dir
        {
            strcpy(path,"htdocs/");
            strcat(path,_DEF_PAGE_);
        }
        else if((st.st_mode&S_IXUSR)||(st.st_mode&S_IXGPR)||(st.st_mode&S_IXOTH))
        {
            cgi=1;
        }
        if(cgi)
        {
            execute_cgi(sock,path,method,query_string);
        }                                                                                          
        else
        {
            clear_head(sock);
            echo_html(sock,path,st.st_size);
        }
    }

#endif 
    close(sock);
}
void* handle_client(void* arg)
{
    int sock=(int)arg;
    accept_request(sock);
    return NULL;
}                          
//main.c 
#include"http.h"                                                                                   
int main(int argc,char* argv[])
{
    if(argc!=3)
    {
        usage(argv[0]);
        return 1;
    }
    char* ip=argv[1];
    int port=atoi(argv[2]);
    int listen_sock=startup(ip,port);
    struct sockaddr_in client;
    socklen_t len=sizeof(client);
    fflush(stdout);
    while(1)
    {
        int new_sock=accept(listen_sock,(struct sockaddr*)&client,&len);
        if(new_sock<0)
        {
            printf("no client");
            fflush(stdout);
            continue;
        }
        pthread_t id;
        pthread_create(&id,NULL,handle_client,(void*)new_sock);
        pthread_detach(id);//set detach return val:0/errno      
    }
    return 0;
}                                                                              					


本文出自 “小止” 博客,请务必保留此出处http://10541556.blog.51cto.com/10531556/1785983

小型web服务器

标签:web

原文地址:http://10541556.blog.51cto.com/10531556/1785983

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!