计算机网络课程设计——简单Web Server程序的设计与实现

上代码

运行完打开浏览器访问127.0.0.1:8000

#include <cstdio>
#include <string>
#include <fstream>
#include <WinSock2.h>
#include <iostream>
using namespace std;
#define BUFFER_SIZE 1024
#define HOST "127.0.0.1"
#define PORT 8000
#define HEADER "\
HTTP/1.1 200 OK\r\n\
Content-Type: text/html; charset=UTF-8\r\n\
Server: ZYhttp_v1.0.1\r\n\
Content-Length: %ld\r\n\r\n\
"
const string strPath = "1.html";
#pragma comment(lib, "WS2_32")
// 获取文件的大小
long GetFileLength(string strPath);
// 单线程函数
int main(int argc, char **argv)
{
    // 定义并且初始化一个服务器套接字
    sockaddr_in addrServer;
    addrServer.sin_family = AF_INET;
    addrServer.sin_addr.S_un.S_addr = INADDR_ANY;
    addrServer.sin_port = htons(PORT);
    // 初始化
    WSADATA wsaData;
    WORD socketVersion = MAKEWORD(2, 2);
    if (WSAStartup(socketVersion, &wsaData) != 0)
    {
        printf("初始化失败!");
        exit(1);
    }
    // 创建套接字
    SOCKET socketServer = socket(AF_INET, SOCK_STREAM, 0);
    if (socketServer == SOCKET_ERROR)
    {
        printf("创建失败!");
        exit(1);
    }
    // 绑定服务器套接字
    if (bind(socketServer, (LPSOCKADDR)&addrServer, sizeof(addrServer)) == SOCKET_ERROR)
    {
        printf("绑定失败!");
        exit(1);
    }
    // 监听
    if (listen(socketServer, 10) ==  SOCKET_ERROR)
    {
        printf("监听失败!");
        exit(1);
    }
    while (true)
    {
        printf("Listening ... \n");
        sockaddr_in addrClient;
        int nClientAddrLen = sizeof(addrClient);
        //服务器端建立连接
        SOCKET socketClient = accept(socketServer, (sockaddr*)&addrClient, &nClientAddrLen);
        if (SOCKET_ERROR == socketClient)
        {
            printf("接收失败!");
            break;
        }
        char buffer[BUFFER_SIZE];
        memset(buffer, 0, BUFFER_SIZE);
        //接收数据
        if (recv(socketClient, buffer, BUFFER_SIZE, 0) < 0)
        {
            printf("接收数据失败!");
            break;
        }
        printf("接收到的数据 : \n%s", buffer);
        // response
        // send header
        memset(buffer, 0, BUFFER_SIZE);
        sprintf(buffer, HEADER, GetFileLength(strPath));//把文件和头文件合并然后发送数据
        if (send(socketClient, buffer, strlen(buffer), 0) < 0)
        {
            printf("发送数据失败!");
            break;
        }
        //读取文本
        ifstream fin(strPath.c_str(), ios::in | ios::binary);
        if (fin.is_open())
        {
            memset(buffer, 0, BUFFER_SIZE);
            //每次读取一个字节
            while (fin.read(buffer, BUFFER_SIZE - 1))
            {
                if (send(socketClient, buffer, strlen(buffer), 0) < 0)
                {
                    printf("发送数据失败!");
                    break;
                }
                memset(buffer, 0, BUFFER_SIZE);
            }
            if (send(socketClient, buffer, strlen(buffer), 0) < 0)
            {
                printf("发送数据失败!");
                break;
            }
        }
        fin.close();
        closesocket(socketClient);
    }
    closesocket(socketServer);
    WSACleanup();
    return 0;
}
long GetFileLength(string strPath)
{
    ifstream fin(strPath.c_str(), ios::in | ios::binary);
    fin.seekg(0, ios_base::end);
    streampos pos = fin.tellg();//返回当前文件位置
    long lSize = static_cast<long>(pos);
    fin.close();
    cout<<"该文件的长度："<<lSize<<endl;
    return lSize;
}