C实现简单web服务器-1(windows)

yzqzs

// SimpleHTTPServer.cpp　　
// 功能：实现简单的web服务器功能，能同时响应多个浏览器的请求：
　　
//       1、如果该文件存在，则在浏览器上显示该文件；
　　
//       2、如果文件不存在，则返回404-file not found页面
　　
//       3、只支持GET、HEAD方法
　　
// HTTP1.1 与 1.0不同，默认是持续连接的(keep-alive)
　　

　　
#include <Winsock2.h>
　　
#include <time.h>
　　
#include <stdio.h>
　　
#include <string.h>
　　
#include <direct.h>     // 目录头文件
　　
#pragma comment(lib,"Ws2_32.lib")
　　

　　
// http 默认端口是80，如果80端口被占用那么改个端口即可
　　
#define DEFAULT_PORT 80
　　
#define BUF_LENGTH 1024
　　
#define MIN_BUF 128
　　
#define USER_ERROR -1
　　
#define SERVER "Server: csr_http1.1\r\n"
　　

　　
int file_not_found(SOCKET sAccept);
　　
int file_ok(SOCKET sAccept, long flen);
　　
int send_file(SOCKET sAccept, FILE *resource);
　　
int send_not_found(SOCKET sAccept);
　　

　　
DWORD WINAPI SimpleHTTPServer(LPVOID lparam)
　　
{
　　
    SOCKET sAccept = (SOCKET)(LPVOID)lparam;
　　
    char recv_buf[BUF_LENGTH];
　　
    char method[MIN_BUF];
　　
    char url[MIN_BUF];
　　
    char path[_MAX_PATH];
　　
    int i, j;
　　

　　
    // 缓存清0，每次操作前都要记得清缓存，养成习惯；
　　
    // 不清空可能出现的现象：输出乱码、换台机器乱码还各不相同
　　
    // 原因：不清空会输出遇到 '\0'字符为止，所以前面的不是'\0' 也会一起输出
　　
    memset(recv_buf,0,sizeof(recv_buf));
　　
    if (recv(sAccept,recv_buf,sizeof(recv_buf),0) == SOCKET_ERROR)   //接收错误
　　
    {
　　
        printf("recv() Failed:%d\n",WSAGetLastError());
　　
        return USER_ERROR;
　　
    }
　　
    else
　　
        printf("recv data from client:%s\n",recv_buf); //接收成功，打印请求报文
　　

　　
    //处理接收数据
　　
    i = 0; j = 0;
　　
    // 取出第一个单词，一般为HEAD、GET、POST
　　
    while (!(' ' == recv_buf[j]) && (i < sizeof(method) - 1))
　　
    {
　　
        method = recv_buf[j];
　　
        i++; j++;
　　
    }
　　
    method = '\0';   // 结束符，这里也是初学者很容易忽视的地方
　　

　　
    // 如果不是GET或HEAD方法，则直接断开本次连接
　　
    // 如果想做的规范些可以返回浏览器一个501未实现的报头和页面
　　
    if (stricmp(method, "GET") && stricmp(method, "HEAD"))
　　
    {
　　
        closesocket(sAccept); //释放连接套接字，结束与该客户的通信
　　
        printf("not get or head method.\nclose ok.\n");
　　
        printf("***********************\n\n\n\n");
　　
        return USER_ERROR;
　　
    }
　　
    printf("method: %s\n", method);
　　

　　
    // 提取出第二个单词(url文件路径，空格结束)，并把'/'改为windows下的路径分隔符'\'
　　
    // 这里只考虑静态请求(比如url中出现'?'表示非静态，需要调用CGI脚本，'?'后面的字符串表示参数，多个参数用'+'隔开
　　
    // 例如：www.csr.com/cgi_bin/cgi?arg1+arg2 该方法有时也叫查询，早期常用于搜索)
　　
    i = 0;
　　
    while ((' ' == recv_buf[j]) && (j < sizeof(recv_buf)))
　　
        j++;
　　
    while (!(' ' == recv_buf[j]) && (i < sizeof(recv_buf) - 1) && (j < sizeof(recv_buf)))
　　
    {
　　
        if (recv_buf[j] == '/')
　　
            url = '\\';
　　
        else if(recv_buf[j] == ' ')
　　
            break;
　　
        else
　　
            url = recv_buf[j];
　　
        i++; j++;
　　
    }
　　
    url = '\0';
　　
    printf("url: %s\n",url);
　　

　　
    // 将请求的url路径转换为本地路径
　　
    _getcwd(path,_MAX_PATH);
　　
    strcat(path,url);
　　
    printf("path: %s\n",path);
　　

　　
    // 打开本地路径下的文件，网络传输中用r文本方式打开会出错
　　
    FILE *resource = fopen(path,"rb");
　　

　　
    // 没有该文件则发送一个简单的404-file not found的html页面，并断开本次连接
　　
    if(resource==NULL)
　　
    {
　　
        file_not_found(sAccept);
　　
        // 如果method是GET，则发送自定义的file not found页面
　　
        if(0 == stricmp(method, "GET"))
　　
            send_not_found(sAccept);
　　

　　
        closesocket(sAccept); //释放连接套接字，结束与该客户的通信
　　
        printf("file not found.\nclose ok.\n");
　　
        printf("***********************\n\n\n\n");
　　
        return USER_ERROR;
　　
    }
　　

　　
    // 求出文件长度，记得重置文件指针到文件头
　　
    fseek(resource,0,SEEK_SET);
　　
    fseek(resource,0,SEEK_END);
　　
    long flen=ftell(resource);
　　
    printf("file length: %ld\n", flen);
　　
    fseek(resource,0,SEEK_SET);
　　

　　
    // 发送200 OK HEAD
　　
    file_ok(sAccept, flen);
　　

　　
    // 如果是GET方法则发送请求的资源
　　
    if(0 == stricmp(method, "GET"))
　　
    {
　　
        if(0 == send_file(sAccept, resource))
　　
            printf("file send ok.\n");
　　
        else
　　
            printf("file send fail.\n");
　　
    }
　　
    fclose(resource);
　　

　　
    closesocket(sAccept); //释放连接套接字，结束与该客户的通信
　　
    printf("close ok.\n");
　　
    printf("***********************\n\n\n\n");
　　

　　
    return 0;
　　

　　
}
　　

　　
// 发送404 file_not_found报头
　　
int file_not_found(SOCKET sAccept)
　　
{
　　
    char send_buf[MIN_BUF];
　　
//  time_t timep;
　　
//  time(&timep);
　　
    sprintf(send_buf, "HTTP/1.1 404 NOT FOUND\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
//  sprintf(send_buf, "Date: %s\r\n", ctime(&timep));
　　
//  send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "Connection: keep-alive\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, SERVER);
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "Content-Type: text/html\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    return 0;
　　
}
　　

　　
// 发送200 ok报头
　　
int file_ok(SOCKET sAccept, long flen)
　　
{
　　
    char send_buf[MIN_BUF];
　　
//  time_t timep;
　　
//  time(&timep);
　　
    sprintf(send_buf, "HTTP/1.1 200 OK\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "Connection: keep-alive\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
//  sprintf(send_buf, "Date: %s\r\n", ctime(&timep));
　　
//  send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, SERVER);
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "Content-Length: %ld\r\n", flen);
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "Content-Type: text/html\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    return 0;
　　
}
　　

　　
// 发送自定义的file_not_found页面
　　
int send_not_found(SOCKET sAccept)
　　
{
　　
    char send_buf[MIN_BUF];
　　
    sprintf(send_buf, "<HTML><TITLE>Not Found</TITLE>\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "<BODY><h1 align='center'>404</h1><br/><h1 align='center'>file not found.</h1>\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    sprintf(send_buf, "</BODY></HTML>\r\n");
　　
    send(sAccept, send_buf, strlen(send_buf), 0);
　　
    return 0;
　　
}
　　

　　
// 发送请求的资源
　　
int send_file(SOCKET sAccept, FILE *resource)
　　
{
　　
    char send_buf[BUF_LENGTH];
　　
    while (1)
　　
    {
　　
        memset(send_buf,0,sizeof(send_buf));       //缓存清0
　　
        fgets(send_buf, sizeof(send_buf), resource);
　　
    //  printf("send_buf: %s\n",send_buf);
　　
        if (SOCKET_ERROR == send(sAccept, send_buf, strlen(send_buf), 0))
　　
        {
　　
            printf("send() Failed:%d\n",WSAGetLastError());
　　
            return USER_ERROR;
　　
        }
　　
        if(feof(resource))
　　
            return 0;
　　
    }
　　
}
　　

　　
int main()
　　
{
　　
    WSADATA wsaData;
　　
    SOCKET sListen,sAccept;        //服务器监听套接字，连接套接字
　　
    int serverport=DEFAULT_PORT;   //服务器端口号
　　
    struct sockaddr_in ser,cli;   //服务器地址，客户端地址
　　
    int iLen;
　　

　　
    printf("-----------------------\n");
　　
    printf("Server waiting\n");
　　
    printf("-----------------------\n");
　　

　　
    //第一步：加载协议栈
　　
    if (WSAStartup(MAKEWORD(2,2),&wsaData) !=0)
　　
    {
　　
        printf("Failed to load Winsock.\n");
　　
        return USER_ERROR;
　　
    }
　　

　　
    //第二步：创建监听套接字，用于监听客户请求
　　
    sListen =socket(AF_INET,SOCK_STREAM,0);
　　
    if (sListen == INVALID_SOCKET)
　　
    {
　　
        printf("socket() Failed:%d\n",WSAGetLastError());
　　
        return USER_ERROR;
　　
    }
　　

　　
    //创建服务器地址：IP+端口号
　　
    ser.sin_family=AF_INET;
　　
    ser.sin_port=htons(serverport);               //服务器端口号
　　
    ser.sin_addr.s_addr=htonl(INADDR_ANY);   //服务器IP地址，默认使用本机IP
　　

　　
    //第三步：绑定监听套接字和服务器地址
　　
    if (bind(sListen,(LPSOCKADDR)&ser,sizeof(ser))==SOCKET_ERROR)
　　
    {
　　
        printf("blind() Failed:%d\n",WSAGetLastError());
　　
        return USER_ERROR;
　　
    }
　　

　　
    //第五步：通过监听套接字进行监听
　　
    if (listen(sListen,5)==SOCKET_ERROR)
　　
    {
　　
        printf("listen() Failed:%d\n",WSAGetLastError());
　　
        return USER_ERROR;
　　
    }
　　
    while (1)  //循环等待客户的请求
　　
    {
　　
        //第六步：接受客户端的连接请求，返回与该客户建立的连接套接字
　　
        iLen=sizeof(cli);
　　
        sAccept=accept(sListen,(struct sockaddr*)&cli,&iLen);
　　
        if (sAccept==INVALID_SOCKET)
　　
        {
　　
            printf("accept() Failed:%d\n",WSAGetLastError());
　　
            break;
　　
        }
　　
        //第七步，创建线程接受浏览器请求
　　
        DWORD ThreadID;
　　
        CreateThread(NULL,0,SimpleHTTPServer,(LPVOID)sAccept,0,&ThreadID);
　　
    }
　　
    closesocket(sListen);
　　
    WSACleanup();
　　
    return 0;
　　
}