分块传输编码 - 页面显示不正确

Question

我的任务是实现简单的 HTTP 服务器。当我发送响应时，我应该支持分块传输编码。这是我向客户端发送响应的函数。

static int serve_request(int sock, struct conf_arg *arg, char version[])
{
    FILE *html = NULL;
    char buf[MAX_MSG];

    strcat(arg->root, arg->defdoc);
    html = fopen(arg->root, "r");
    if (!html) {
        not_found(sock, version);
        return 0;
    }
    good_responce(sock, version);
    do {
        fgets(buf, sizeof(buf), html);
        const unsigned chunk = CHUNK_SIZE;
        char *pbuf = buf;
        char tempbuf[chunk + 10];

        while (strlen(pbuf) >= chunk) {
            sprintf(tempbuf, "%x\r\n", chunk);
            write(sock, tempbuf, strlen(tempbuf));
            write(sock, pbuf, chunk);
            pbuf += chunk;
            strcpy(tempbuf, "\r\n");
            write(sock, tempbuf, strlen(tempbuf));
        }
        if (strlen(pbuf) == 0) {
            sprintf(tempbuf, "%x\r\n", 0);
            write(sock, tempbuf, strlen(tempbuf));
        }
        if (strlen(pbuf) > 0) {
            sprintf(tempbuf, "%x\r\n", (unsigned)strlen(pbuf));
            write(sock, tempbuf, strlen(tempbuf));
            write(sock, pbuf, strlen(pbuf));
            sprintf(tempbuf, "%x\r\n", 0);
            write(sock, tempbuf, strlen(tempbuf));
        }
        strcpy(tempbuf, "\r\n");
        write(sock, tempbuf, strlen(tempbuf));
    } while (!feof(html));
    fclose(html);
    return 0;
}

CHUNK_SIZE 定义为 1024，因为我想发送 1KB 大小的块。打开页面 enter image description here 时出现问题页面显示不正常。我还设置了 Transfer-Encoding: chunked

strcpy(buf, ENCODING);
send(sock, buf, strlen(buf), 0);

ENCODING 定义为 "Transfer-Encoding: chunked\r\n"

Answer 1

我想我知道问题出在哪里，但不完全确定。

在您的 do 循环中，您会收到一个 buf 充满数据的数据，然后将其发送。然后你得到另一个充满数据的缓冲区并发送它。但是在发送了每个数据缓冲区之后，您可以通过发送 0\r\n 来终止传输。例如：

1024    // send first chunk
1024    // send second chunk
256     // last part of first bufer
0       // terminate transfer
1024    // send first chunk of second buffer
1024    //...
256
0

尽管在发送最后一个块之前再次填充缓冲区可能更好（使用 memmove 将最后一部分向下移动，然后调用 fgets 填充剩余部分），您可能"saved" 通过仅在 do ... while 循环之后发送 0\r\n，例如：

        if (strlen(pbuf) > 0) {
            sprintf(tempbuf, "%x\r\n", (unsigned)strlen(pbuf));
            write(sock, tempbuf, strlen(tempbuf));
            write(sock, pbuf, strlen(pbuf));
            //sprintf(tempbuf, "%x\r\n", 0);
            //write(sock, tempbuf, strlen(tempbuf));
        }
        //strcpy(tempbuf, "\r\n");
        //write(sock, tempbuf, strlen(tempbuf));
    } while (!feof(html));
    sprintf(tempbuf, "%x\r\n", 0);
    write(sock, tempbuf, strlen(tempbuf));
    strcpy(tempbuf, "\r\n");
    write(sock, tempbuf, strlen(tempbuf));

另请注意，您必须检查fgets的结果，因为它可以return在eof上归零；缓冲区将不会被刷新，您将再次发送最后一部分：

    if (fgets(buf, sizeof(buf), html)==NULL) break;

另请参阅关于您不必要地使用 tempbuf 的评论。

分块传输编码 - 页面显示不正确

Chunked transfer encoding - Page is not displayed properly

c

http

chunked-encoding

http-chunked

server