Linux通过AIO进行异步读文件

nuligannima 2015-12-13

Linux通过AIO进行异步读文件,下面列出源代码:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <aio.h>
#include <unistd.h>
#include <signal.h>
#include <sys/stat.h>
#include <fcntl.h>


static char *memBuffer;
static int sFileDesc;
static struct sigaction sOldSigAction;

static void MySigQuitHandler(int sig)
{
    printf("Signal Quit! The number is: %d\n", sig);
}

static void MyFileReadCompleteProcedure(int sig, siginfo_t *si, void *ucontext)
{
    printf("The file length is: %zu, and the content is: %s\n", strlen(memBuffer), memBuffer);
    int status = close(sFileDesc);
    if(status == 0)
        puts("File closed successfully!");
    else
        printf("The error code is: %d\n", status);

    free(memBuffer);

    // 还原原来的SIGUSR1信号行为
    if(sigaction(SIGUSR1, &sOldSigAction, NULL) == -1)
        puts("SIGUSR1 signal restore failed!");
}

int main(void)
{
    struct sigaction sigAction = { .sa_flags = SA_RESTART, .sa_handler = &MySigQuitHandler };

    sigemptyset(&sigAction.sa_mask);

    if (sigaction(SIGQUIT, &sigAction, NULL) == -1)
    {
        puts("Signal failed!");
        return -1;
    }

    sigAction.sa_sigaction = &MyFileReadCompleteProcedure;
    if(sigaction(SIGUSR1, &sigAction, &sOldSigAction) == -1)
    {
        puts("Signal failed!");
        return -1;
    }

    const char *filePath = "myfile.txt";

    const mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
    sFileDesc = open(filePath, O_RDONLY, mode);
    if(sFileDesc == -1)
    {
        printf("The file: %s cannot be opened!\n", filePath);
        return -1;
    }

    const long fileLength = lseek(sFileDesc, 0, SEEK_END);
    lseek(sFileDesc, 0, SEEK_SET);

    memBuffer = malloc(fileLength + 1);
    memBuffer[fileLength] = '\0';

    struct aiocb aioBuffer;
    aioBuffer.aio_fildes = sFileDesc;
    aioBuffer.aio_offset = 0;
    aioBuffer.aio_buf = memBuffer;
    aioBuffer.aio_nbytes = fileLength;
    aioBuffer.aio_reqprio = 0;
    aioBuffer.aio_sigevent = (struct sigevent){.sigev_notify = SIGEV_SIGNAL, .sigev_signo = SIGUSR1, .sigev_value.sival_ptr = memBuffer };

    aio_read(&aioBuffer);

    getchar();

    return 0;
}

其中,上述代码实现中采用SIGUSR1信号进行捕获文件读完成事件,当然,这里也可以用SIGIO信号。

另外,在编译链接时必须添加-lrt命令选项,因为rt库才包含了系统底层的API。

相关推荐