Linux高性能服务器编程第6章之简单的cgi服务器和HTTP应答

CGI服务器

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <assert.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <libgen.h>
int main(int argc, char* argv[])
{
    if(argc <= 2)
    {
                // 注意，basename函数在libgen.h头文件中，作用是只保留文件名，去除前面的路径
        printf("usage: %s ip_address port_num\n", basename(argv[0]));
        return 1;
    }
    const char* ip = argv[1];
    int port = atoi(argv[2]);
    struct sockaddr_in address;
    bzero(&address, sizeof(address));
    address.sin_family = AF_INET;  // AF_INET 跟 PF_INET都是表示ipv4协议族
    // 把ip转换为网络字节序填入sin.addr
    inet_pton(AF_INET, ip, &address.sin_addr);
    address.sin_port = htons(port); // 端口转成网络字节序

    int sock = socket(PF_INET, SOCK_STREAM, 0);
    assert(sock >= 0);

    int ret = bind(sock, (struct sockaddr *)&address, sizeof(address));
    assert(ret != -1);

    ret = listen(sock, 5);
    assert(ret != -1);

    struct sockaddr_in client;
    socklen_t client_addrlength = sizeof(client);
    // sock 是监听套接字（经过了listen）
    // connfd 是连接套接字（accept返回的）
    // 两者都是socket文件描述符
    // 因此connfd才是真正代表创建出的这条网络连接
    int connfd = accept(sock, (struct sockaddr *)&client, &client_addrlength);
    if(connfd < 0)
    {
        printf("errno is: %d\n", errno);
    }
    else
    {
        close(STDOUT_FILENO); // 关闭标准输出
        dup(connfd);          // 重定向标准输出到网络连接
        printf("abcd\n");
        close(connfd);
    }
    close(sock);
    return 0;
}

执行效果

image.png

使用writev函数进行http应答

下面这个例子由于使用了new，所以是c++程序，要用cpp后缀
要使用g++编译，或者 gcc -lstdc++ 进行编译

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <libgen.h>

#define BUFFER_SIZE 1024
#define bool char
/* 定义两种状态码和状态信息 */
static const char* status_line[2] = {"200 OK", "500 Internal server error"};

int main(int argc, char* argv[]){
    if(argc <= 3){
        // basename(path) 去掉路径留下最后那个文件名  // basename函数用不了
        printf("usage: %s ip port filenme\n", basename(argv[0]));
        return 1;
    }
    const char* ip = argv[1];
    int port = atoi(argv[2]);
    const char* filename = argv[3];

    struct sockaddr_in address;
    __bzero(&address, sizeof(address));
    address.sin_family = AF_INET;
    address.sin_port = htons(port);
    inet_pton(AF_INET, ip, &address.sin_addr);
    // 创建socket
    int sock = socket(PF_INET, SOCK_STREAM, 0);
    assert(sock >= 0);
    // 绑定sock和address
    int ret = bind(sock, (struct sockaddr *)&address, sizeof(address));
    assert(ret != -1);
    // 监听
    ret = listen(sock, 5);
    assert(ret != -1);

    struct sockaddr_in client;
    socklen_t client_addrlen = sizeof(client);
    // 接收客户端请求
    int connfd = accept(sock, (struct sockaddr *)&client, &client_addrlen);
    if (connfd < 0)
    {
        printf("errno is: %d\n", errno);
    }
    else
    {
        /* 用于保存HTTP应答的状态行、头部字段和一个空行 */
        char header_buf[BUFFER_SIZE];
        memset(header_buf, '\0', BUFFER_SIZE);
        /* 存放目标文件内容的缓存区 */
        char* file_buf;
        /* 用于获取目标文件属性，是否目录，文件大小 */
        struct stat file_stat;
        /* 记录文件是否有效文件 */
        bool valid = 1;
        /* 缓存区header_buf目前已经使用了多少字节的空间 */
        int len = 0;
        if(stat(filename, &file_stat) < 0) /* 文件不存在 */
        {
            valid = 0;
        }
        else
        {
            if(S_ISDIR(file_stat.st_mode)) /* 目标文件是目录 */
            {
                valid = 0;
            }
            else if(file_stat.st_mode & S_IROTH) /* 当前用户有读取文件的权限 */
            {
                /* 分配缓存区file_buf大小 */
                int fd = open(filename, O_RDONLY);
                file_buf = new char[file_stat.st_size + 1];
                memset(file_buf, 0, file_stat.st_size + 1);
                if(read(fd, file_buf, file_stat.st_size) < 0)
                {
                    valid = 0;
                }
            }
            else
            {
                valid = 0;
            }
        }
        /* 目标文件有效，发送正常的HTTP应答 */
        if(valid)
        {
            /* 将HTTP应答的状态行，"Conten-Length"头部字段和一个空白行写入header_buf  */
            ret = snprintf(header_buf, BUFFER_SIZE - 1, "%s %s\r\n",
                            "HTTP/1.1", status_line[0]);
            len += ret;
            ret = snprintf(header_buf + len, BUFFER_SIZE - len - 1,
                            "Content-Length: %d\r\n", file_stat.st_size);
            len += ret;
            ret = snprintf(header_buf + len, BUFFER_SIZE - 1 - len,
                            "%s", "\r\n");
            /* Io结构体， 里面存放了一个指针，和指针指向的数据的长度 */
            struct iovec iv[2];
            iv[0].iov_base = header_buf;
            iv[0].iov_len = strlen(header_buf);
            iv[1].iov_base = file_buf;
            iv[1].iov_len = strlen(file_buf);
            // 将iv数组写入文件描述符，长度2
            ret = writev(connfd, iv, 2);
        }
        else  /* 文件无效，则通知客户端服务器发生了“内部错误” */
        {
            ret = snprintf(header_buf, BUFFER_SIZE - 1, "%s %s\r\n",
                        "HTTP/1.1", status_line[1]);
            len += ret;
            ret = snprintf(header_buf + len, BUFFER_SIZE - 1 - len,
                            "%s", "\r\n");
            send(connfd, header_buf, strlen(header_buf), 0);
        }
        close(connfd);
        delete [] file_buf;

    }
    close(sock);
    return 0;
}

执行效果

image.png