我想知道是否有人知道如何用 C++ 计算 Berkeley 套接字的上传速度。我的 send 调用没有阻塞，发送 5 MB 数据需要 0.001 秒，但需要一段时间才能recv 响应(所以我知道它正在上传)。

这是到 HTTP 服务器的 TCP 套接字，我需要异步检查已上传/剩余的数据字节数。但是，我在 Winsock 中找不到任何 API 函数，所以我很困惑。

如有任何帮助，我们将不胜感激。

编辑:我找到了解决方案，并将尽快作为答案发布!

编辑 2: 添加正确的解决方案作为答案，将在 4 小时内添加为解决方案。

最佳答案

由于 bdolan 建议减少 SO_SNDBUF，我解决了我的问题。但是，要使用此代码，您必须注意您的代码使用 Winsock 2(用于重叠套接字和 WSASend)。除此之外，您的 SOCKET 句柄必须以类似于以下方式创建:

SOCKET sock = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);

请注意 WSA_FLAG_OVERLAPPED 标志作为最终参数。

在这个回答中，我将经历将数据上传到 TCP 服务器、跟踪每个上传 block 及其完成状态的阶段。这个概念需要将您的上传缓冲区分成多个 block (需要对现有代码进行最少的修改)并逐个上传，然后跟踪每个 block 。

我的代码流程

全局变量

您的代码文档必须具有以下全局变量:

#define UPLOAD_CHUNK_SIZE 4096

int g_nUploadChunks = 0;
int g_nChunksCompleted = 0;
WSAOVERLAPPED *g_pSendOverlapped = NULL;
int g_nBytesSent = 0;
float g_flLastUploadTimeReset = 0.0f;

注意:在我的测试中，减小 UPLOAD_CHUNK_SIZE 会提高上传速度的准确性，但会降低整体上传速度。增加 UPLOAD_CHUNK_SIZE 会降低上传速度的准确性，但会提高整体上传速度。 4 KB(4096 字节)对于大小约为 500kB 的文件来说是一个很好的妥协。

回调函数

此函数递增发送的字节数和完成的 block 变量(在 block 已完全上传到服务器后调用)

void CALLBACK SendCompletionCallback(DWORD dwError, DWORD cbTransferred, LPWSAOVERLAPPED lpOverlapped, DWORD dwFlags)
{
    g_nChunksCompleted++;
    g_nBytesSent += cbTransferred;
}

准备套接字

最初，必须通过将 SO_SNDBUF 减少到 0 来准备套接字。

注意:在我的测试中，任何大于 0 的值都会导致不良行为。

int nSndBuf = 0;
setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (char*)&nSndBuf, sizeof(nSndBuf));

创建WSAOVERLAPPED数组

必须创建一个 WSAOVERLAPPED 结构数组来保存我们所有上传 block 的重叠状态。为此，我只需:

// Calculate the amount of upload chunks we will have to create.
// nDataBytes is the size of data you wish to upload
g_nUploadChunks = ceil(nDataBytes / float(UPLOAD_CHUNK_SIZE));

// Overlapped array, should be delete'd after all uploads have completed
g_pSendOverlapped = new WSAOVERLAPPED[g_nUploadChunks];
memset(g_pSendOverlapped, 0, sizeof(WSAOVERLAPPED) * g_nUploadChunks);

上传数据

出于示例目的，所有需要发送的数据都保存在名为 pszData 的变量中。然后，使用 WSASend，数据以常量 UPLOAD_CHUNK_SIZE 定义的 block 形式发送。

WSABUF dataBuf;
DWORD dwBytesSent = 0;
int err;
int i, j;

for(i = 0, j = 0; i < nDataBytes; i += UPLOAD_CHUNK_SIZE, j++)
{
    int nTransferBytes = min(nDataBytes - i, UPLOAD_CHUNK_SIZE);

    dataBuf.buf = &pszData[i];
    dataBuf.len = nTransferBytes;

    // Now upload the data
    int rc = WSASend(sock, &dataBuf, 1, &dwBytesSent, 0, &g_pSendOverlapped[j], SendCompletionCallback);

    if ((rc == SOCKET_ERROR) && (WSA_IO_PENDING != (err = WSAGetLastError())))
    {
        fprintf(stderr, "WSASend failed: %d\n", err);
        exit(EXIT_FAILURE);
    }
}

等待游戏

现在我们可以在所有 block 上传的同时做任何我们想做的事。

注意:调用WSASend的线程必须定期放入alertable state中，以便我们的“传输完成”回调(SendCompletionCallback)从 APC(异步过程调用)列表中出队。

在我的代码中，我不断循环直到 g_nUploadChunks == g_nChunksCompleted。这是为了显示最终用户上传进度和速度(可以修改为显示预计完成时间、耗时等)

注意 2:此代码使用 Plat_FloatTime 作为第二个计数器，将其替换为您的代码使用的任何第二个计时器(或相应调整)

g_flLastUploadTimeReset = Plat_FloatTime();

// Clear the line on the screen with some default data
printf("(0 chunks of %d) Upload speed: ???? KiB/sec", g_nUploadChunks);

// Keep looping until ALL upload chunks have completed
while(g_nChunksCompleted < g_nUploadChunks)
{
    // Wait for 10ms so then we aren't repeatedly updating the screen
    SleepEx(10, TRUE);

    // Updata chunk count
    printf("\r(%d chunks of %d) ", g_nChunksCompleted, g_nUploadChunks);

    // Not enough time passed?
    if(g_flLastUploadTimeReset + 1 > Plat_FloatTime())
        continue;

    // Reset timer
    g_flLastUploadTimeReset = Plat_FloatTime();

    // Calculate how many kibibytes have been transmitted in the last second
    float flByteRate = g_nBytesSent/1024.0f;
    printf("Upload speed: %.2f KiB/sec", flByteRate);

    // Reset byte count
    g_nBytesSent = 0;
}

// Delete overlapped data (not used anymore)
delete [] g_pSendOverlapped;

// Note that the transfer has completed
Msg("\nTransfer completed successfully!\n");

结论

我真的希望这对将来希望在不进行任何服务器端修改的情况下计算 TCP 套接字上的上传速度的人有所帮助。我不知道 SO_SNDBUF = 0 对性能有多大影响，尽管我确信套接字大师会指出这一点。

关于c++ - 计算套接字上传速度，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/3145871/