大型文件上的aio_write

Question

我试图使用sendfile(2)和aio_write(3)以异步方式从本质上模拟aio_read(3)的功能。

除了测试大文件(> 150 k)外，一切看起来都很好。

我有一个简单的struct io_request用于跟踪传输：

struct io_request {
    int status;
    struct aiocb *aiocbp;
    int sfd;
};

首先，构建aio_read()调用：

struct io_request * ioreq = malloc(sizeof(struct io_request));
ioreq->status = EINPROGRESS;
ioreq->sfd = sfd;

struct aiocb * aiocbreq = malloc(sizeof(struct aiocb));
memset(aiocbreq, 0, sizeof(struct aiocb));

ioreq->aiocbp = aiocbreq;

ioreq->aiocbp->aio_fildes = ffd;

if (ioreq->aiocbp->aio_fildes == -1) {
    perror("aio_fildes");
}

ioreq->aiocbp->aio_buf = malloc(st.st_size);
if (ioreq->aiocbp->aio_buf == NULL) {
    perror("aio_buf malloc");
}

ioreq->aiocbp->aio_nbytes = st.st_size;
ioreq->aiocbp->aio_reqprio = 0;
ioreq->aiocbp->aio_offset = 0;
ioreq->aiocbp->aio_sigevent.sigev_signo = IO_READ_SIGNAL;
ioreq->aiocbp->aio_sigevent.sigev_value.sival_ptr = ioreq;
if (aio_read(ioreq->aiocbp) == -1) {
    perror("aio_read");
}

然后，将在IO_READ_SIGNAL处理程序中捕获以下内容：

static void
aio_read_handler(int sig, siginfo_t *si, void *ucontext)
{
    if (si->si_code == SI_ASYNCIO) {
        struct io_request *ioreq = si->si_value.sival_ptr;

        // Build the AIO write request
        struct aiocb aiocbreq;
        memset(&aiocbreq, 0, sizeof(struct aiocb));
        aiocbreq.aio_fildes = ioreq->sfd;
        aiocbreq.aio_buf = ioreq->aiocbp->aio_buf;
        aiocbreq.aio_nbytes = ioreq->aiocbp->aio_nbytes;
        aiocbreq.aio_sigevent.sigev_signo = IO_WRITE_SIGNAL;
        aiocbreq.aio_sigevent.sigev_value.sival_ptr = ioreq;

        if (aio_write((void *) &aiocbreq) == -1) {
            perror("aio_write");
        }
    }
}

我可以确认，在处理程序中，即使是大型文件，ioreq->aiocbp->aio_buf的内容也是完整的。

稍后，aio_write()将在IO_WRITE_SIGNAL处理程序中捕获：

static void
aio_write_handler(int sig, siginfo_t *si, void *ucontext)
{
    if (si->si_code == SI_ASYNCIO) {
        struct io_request *ioreq = si->si_value.sival_ptr;

        ssize_t bytes_written = aio_return(ioreq->aiocbp);
        printf("Wrote %zu of %zu bytes\n", bytes_written, ioreq->aiocbp->aio_nbytes);
        //free(ioreq->aiocbp);
        //free(ioreq);

        if (aio_error(ioreq->aiocbp) != 0) {
            perror("aio_write_handler");
        }
    }
}

此时，aio_write()应该已经完成。我检查返回值并采取相应的行动。这两个调用都报告了写入的适当字节数，并且在写入过程中没有出现错误。

更大的应用程序是HTTP服务器。我推测这个问题的出现是因为远程客户端阅读速度不够快，跟不上aio_write()。当我有一个sendfile()实现时，我不得不多次调用sendfile()来完成文件传输。

几个直接问题：

• 为什么aio_return()和aio_error()不报告任何问题？
• 我怎么才能纠正这种行为？
• 有办法缓冲aio_write()吗？我正在考虑在n_bytes内部设置struct aiocb传递给aio_write()的上限，然后从aio_write_handler()内部多次调用aio_write()。

谢谢你的帮忙!

Answer 1

如果我理解正确的话，您使用的是aio_return而不是aio_error。aio_return手册页面上说，

对于任何给定的请求，在aio_error(3)返回EINPROGRESS以外的其他内容后，应该只调用该函数一次。