Chinaunix首页 | 论坛 | 博客
  • 博客访问: 125584
  • 博文数量: 31
  • 博客积分: 0
  • 博客等级: 民兵
  • 技术积分: 85
  • 用 户 组: 普通用户
  • 注册时间: 2014-12-18 15:18
文章分类
文章存档

2017年(1)

2016年(1)

2015年(26)

2014年(3)

我的朋友

分类: LINUX

2015-07-09 10:13:19

libaio是linux下原生的异步IO接口。网上对其使用方法讨论较少,这里做个简单说明。libaio的使用并不复杂,过程为:libaio的初始化,io请求的下发和回收,libaio销毁。
一、libaio接口

libaio提供下面五个主要API函数:

int io_setup(int maxevents, io_context_t *ctxp);
int io_destroy(io_context_t ctx);
int io_submit(io_context_t ctx, long nr, struct iocb *ios[]);
int io_cancel(io_context_t ctx, struct iocb *iocb, struct io_event *evt);
int io_getevents(io_context_t ctx_id, long min_nr, long nr, struct io_event *events, struct timespec *timeout);

五个宏定义:

void io_set_callback(struct iocb *iocb, io_callback_t cb);
void io_prep_pwrite(struct iocb *iocb, int fd, void *buf, size_t count, long long offset);
void io_prep_pread(struct iocb *iocb, int fd, void *buf, size_t count, long long offset);
void io_prep_pwritev(struct iocb *iocb, int fd, const struct iovec *iov, int iovcnt, long long offset);
void io_prep_preadv(struct iocb *iocb, int fd, const struct iovec *iov, int iovcnt, long long offset);

这五个宏定义都是操作struct iocb的结构体。struct iocb是libaio中很重要的一个结构体,用于表示IO,但是其结构略显复杂,为了保持封装性不建议直接操作其元素而用上面五个宏定义操作。
二、libaio的初始化和销毁

观察libaio五个主要API,都用到类型为io_context的变量,这个变量为libaio的工作空间。不用具体去了解这个变量的结构,只需要了解其相关操作。创建和销毁libaio分别用到io_setup(也可以用io_queue_init,区别只是名字不一样而已)和io_destroy。

int io_setup(int maxevents, io_context_t *ctxp);
int io_destroy(io_context_t ctx);
三、libaio读写请求的下发和回收

1. 请求下发

libaio的读写请求都用io_submit下发。下发前通过io_prep_pwrite和io_prep_pread生成iocb的结构体,做为io_submit的参数。这个结构体中指定了读写类型、起始扇区、长度和设备标志符。

libaio的初始化不是针对一个具体设备进行初始,而是创建一个libaio的工作环境。读写请求下发到哪个设备是通过open函数打开的设备标志符指定。

2. 请求返回

读写请求下发之后,使用io_getevents函数等待io结束信号:

int io_getevents(io_context_t ctx_id, long min_nr, long nr, struct io_event *events, struct timespec *timeout);

io_getevents返回events的数组,其参数events为数组首地址,nr为数组长度(即最大返回的event数),min_nr为最少返回的events数。timeout可填NULL表示无等待超时。io_event结构体的声明为:

struct io_event {
    PADDEDptr(void *data, __pad1);
    PADDEDptr(struct iocb *obj,  __pad2);
    PADDEDul(res,  __pad3);
    PADDEDul(res2, __pad4);
};

其中,res为实际完成的字节数;res2为读写成功状态,0表示成功;obj为之前下发的struct iocb结构体。这里有必要了解一下struct iocb这个结构体的主要内容:

iocbp->iocb.u.c.nbytes 字节数
iocbp->iocb.u.c.offset 偏移
iocbp->iocb.u.c.buf 缓冲空间
iocbp->iocb.u.c.flags 读写

3. 自定义字段

struct iocb除了自带的元素外,还留有供用户自定义的元素,包括回调函数和void *的data指针。如果在请求下发前用io_set_callback绑定用户自定义的回调函数,那么请求返回后就可以显示的调用该函数。回调函数的类型为:

void callback_function(io_context_t ctx, struct iocb *iocb, long res, long res2);

另外,还可以通过iocbp->data指针挂上用户自己的数据。

注意:实际使用中发现回调函数和data指针不能同时用,可能回调函数本身就是使用的data指针。
四、使用例子

通过上面的说明并不能完整的了解libaio的用法,下面通过简单的例子进一步说明。
     1    #include
     2    #include
     3    #include
     4    #include
     5    #include
     6    #include
     7    int srcfd = -1;
     8    int odsfd = -1;
     9    #define AIO_BLKSIZE  1024
    10    #define AIO_MAXIO 64
    11    static void wr_done(io_context_t ctx, struct iocb *iocb, long res, long res2)
    12    {
    13        if (res2 != 0)
    14        {
    15            printf("aio write error n");
    16        }
    17        if (res != iocb->u.c.nbytes)
    18        {
    19            printf("write missed bytes expect %d got %d n", iocb->u.c.nbytes, res);
    20            exit(1);
    21        }
    22        free(iocb->u.c.buf);
    23        free(iocb);
    24    }
    25    static void rd_done(io_context_t ctx, struct iocb *iocb, long res, long res2)
    26    {
    27        int iosize = iocb->u.c.nbytes;
    28        char *buf = (char *)iocb->u.c.buf;
    29        off_t offset = iocb->u.c.offset;
    30        int tmp;
    31        char *wrbuff = NULL;
    32        if (res2 != 0)
    33        {
    34            printf("aio read n");
    35        }
    36        if (res != iosize)
    37        {
    38            printf("read missing bytes expect % d got % d", iocb->u.c.nbytes, res);
    39            exit(1);
    40        }
    41        tmp = posix_memalign((void **)&wrbuff, getpagesize(), AIO_BLKSIZE);
    42        if (tmp < 0)
    43        {
    44            printf("posix_memalign222 n");
    45            exit(1);
    46        }
    47        snprintf(wrbuff, iosize + 1, "%s", buf);
    48        printf("wrbuff - len = %d:%s n", strlen(wrbuff), wrbuff);
    49        printf("wrbuff_len = %d n", strlen(wrbuff));
    50        free(buf);
    51        io_prep_pwrite(iocb, odsfd, wrbuff, iosize, offset);
    52        io_set_callback(iocb, wr_done);
    53        if (1 != (res = io_submit(ctx, 1, &iocb)))
    54            printf("io_submit write error n");
    55        printf("nsubmit % d write request n", res);
    56    }
    57    void main(int args, void *argv[])
    58    {
    59        int length = sizeof("abcdefg");
    60        char *content = (char *)malloc(length);
    61        io_context_t myctx;
    62        int rc;
    63        char *buff = NULL;
    64        int offset = 0;
    65        int num, i, tmp;
    66        if (args < 3)
    67        {
    68            printf("the number of param is wrong n");
    69            exit(1);
    70        }
    71        if ((srcfd = open(argv[1], O_RDWR)) < 0)
    72        {
    73            printf("open srcfile error n");
    74            exit(1);
    75        }
    76        printf("srcfd = %d n", srcfd);
    77        lseek(srcfd, 0, SEEK_SET);
    78        write(srcfd, "abcdefg", length);
    79        lseek(srcfd, 0, SEEK_SET);
    80        read(srcfd, content, length);
    81        printf("write in the srcfile successful, content is % s n", content);
    82        if ((odsfd = open(argv[2], O_RDWR)) < 0)
    83        {
    84            close(srcfd);
    85            printf("open odsfile error n");
    86            exit(1);
    87        }
    88        memset(&myctx, 0, sizeof(myctx));
    89        io_queue_init(AIO_MAXIO, &myctx);
    90        struct iocb *io = (struct iocb *)malloc(sizeof(struct iocb));
    91        int iosize = AIO_BLKSIZE;
    92        tmp = posix_memalign((void **)&buff, getpagesize(), AIO_BLKSIZE);
    93        if (tmp < 0)
    94        {
    95            printf("posix_memalign error n");
    96            exit(1);
    97        }
    98        if (NULL == io)
    99        {
   100            printf("io out of memeory n");
   101            exit(1);
   102        }
   103        io_prep_pread(io, srcfd, buff, iosize, offset);
   104        io_set_callback(io, rd_done);
   105        printf("START...n n");
   106        rc = io_submit(myctx, 1, &io);
   107        if (rc < 0)
   108            printf("io_submit read error n");
   109        printf("nsubmit % d read request n", rc);
   110        //m_io_queue_run(myctx);
   111        struct io_event events[AIO_MAXIO];
   112        io_callback_t cb;
   113        num = io_getevents(myctx, 1, AIO_MAXIO, events, NULL);
   114        printf("n % d io_request completed n n", num);
   115        for (i = 0; i < num; i++)
   116        {
   117            cb = (io_callback_t) events[i].data;
   118            struct iocb *io = events[i].obj;
   119            printf("events[%d].data = %x, res = %d, res2 = %d n", i, cb, events[i].res, events[i].res2);
   120            cb(myctx, io, events[i].res, events[i].res2);
   121        }
   122    }


阅读(5350) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~