c - 大文件上的 aio_write

Question

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

一切似乎都工作正常，除了测试大型(> 150k)文件。

我有一个简单的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()吗？我正在考虑将struct aiocb内的n_bytes传递给aio_write()，然后调用aio_write()从 aio_write_handler() 内部多次。

感谢您的帮助!

Answer 1

如果我理解正确的话，您在aio_error之前使用了aio_return。 aio_return 手册页显示，

This function should be called only once for any given request, after aio_error(3) returns something other than EINPROGRESS.