Re: [Qemu-devel] [PATCH V3 5/5] libqblock test example code

[Wenchao Xia <xiawenc@linux.vnet.ibm.com>]

于 2012-9-18 18:10, Paolo Bonzini 写道:
> Il 18/09/2012 11:01, Wenchao Xia ha scritto:
>>    In this example, user first create two qcow2 images, and then get the
>> backing file relationship information of them. Then does write and read
>> sync IO on them.
>
> Please use gtest so that this can be easily extensible.
>
> Paolo
>
   OK.

>> Signed-off-by: Wenchao Xia <xiawenc@linux.vnet.ibm.com>
>> ---
>>   tests/libqblock/libqblock-test.c |  233 ++++++++++++++++++++++++++++++++++++++
>>   1 files changed, 233 insertions(+), 0 deletions(-)
>>
>> diff --git a/tests/libqblock/libqblock-test.c b/tests/libqblock/libqblock-test.c
>> index c05c0c4..c0b7963 100644
>> --- a/tests/libqblock/libqblock-test.c
>> +++ b/tests/libqblock/libqblock-test.c
>> @@ -1,4 +1,237 @@
>> +/*
>> + * QEMU block layer library test
>> + *
>> + * Copyright IBM, Corp. 2012
>> + *
>> + * Authors:
>> + *  Wenchao Xia   <xiawenc@linux.vnet.ibm.com>
>> + *
>> + * This work is licensed under the terms of the GNU LGPL, version 2 or later.
>> + * See the COPYING.LIB file in the top-level directory.
>> + *
>> + */
>> +
>> +#include <stdarg.h>
>> +#include <stdio.h>
>> +#include <unistd.h>
>> +#include <inttypes.h>
>> +#include <string.h>
>> +#include <stdlib.h>
>> +#include "libqblock.h"
>> +
>> +#define TEST_BUF_SIZE 1024
>> +static unsigned char buf_r[TEST_BUF_SIZE];
>> +static unsigned char buf_w[TEST_BUF_SIZE] = {0, 0, 0, 0};
>> +
>> +typedef struct VerifyData {
>> +    unsigned char *buf_r;
>> +    unsigned char *buf_w;
>> +    int len;
>> +} VerifyData;
>> +
>> +static void print_loc(QBlockProtInfo *loc)
>> +{
>> +    if (loc == NULL) {
>> +        printf("backing file is NULL.");
>> +        return;
>> +    }
>> +    switch (loc->prot_type) {
>> +    case QB_PROT_NONE:
>> +        printf("protocol type [none].");
>> +        break;
>> +    case QB_PROT_FILE:
>> +        printf("protocol type [file], filename [%s].",
>> +               loc->prot_op.o_file.filename);
>> +        break;
>> +    default:
>> +        printf("protocol type not supported.");
>> +        break;
>> +    }
>> +}
>> +
>> +static void print_info_image_static(QBlockStaticInfo *info)
>> +{
>> +    printf("=======image location:\n");
>> +    print_loc(&info->loc);
>> +    printf("\nvirtual_size %" PRId64 ", format type %d [%s]",
>> +           *(info->member_addr->virt_size),
>> +           info->fmt.fmt_type, qb_fmttype2str(info->fmt.fmt_type));
>> +    printf("\nbacking image location:\n");
>> +    print_loc(info->member_addr->backing_loc);
>> +    printf("\n");
>> +}
>> +
>> +static void test_check(VerifyData *vdata)
>> +{
>> +    int cmp;
>> +    cmp = memcmp(vdata->buf_r, vdata->buf_w, vdata->len);
>> +    if (cmp == 0) {
>> +        printf("compare succeed, %d.\n", vdata->buf_r[24]);
>> +    } else {
>> +        printf("!!! compare fail, %d.\n", vdata->buf_r[24]);
>> +        exit(1);
>> +    }
>> +}
>> +
>>   int main(int argc, char **argv)
>>   {
>> +    const char *filename1, *filename2;
>> +    QBroker *broker = NULL;
>> +    QBlockState *qbs = NULL;
>> +    QBlockProtInfo *ol = NULL;
>> +    QBlockFmtInfo *of = NULL;
>> +    QBlockStaticInfo *info_st = NULL;
>> +    int ret, flag;
>> +    int test_offset = 510;
>> +    int test_len = 520;
>> +    VerifyData vdata;
>> +    char err_str[1024];
>> +
>> +    vdata.buf_r = buf_r;
>> +    vdata.buf_w = buf_w;
>> +    vdata.len = test_len;
>> +
>> +    filename1 = "./qemu_image1";
>> +    filename2 = "./qemu_image2";
>> +    printf("qemu test, filename1 is %s, filename2 is %s.\n",
>> +                                       filename1, filename2);
>> +
>> +    ret = qb_broker_new(&broker);
>> +    if (ret < 0) {
>> +        goto free;
>> +    }
>> +
>> +    ret = qb_state_new(broker, &qbs);
>> +    if (ret < 0) {
>> +        goto free;
>> +    }
>> +
>> +    ret = qb_prot_info_new(broker, &ol);
>> +    if (ret < 0) {
>> +        goto free;
>> +    }
>> +
>> +    ret = qb_fmt_info_new(broker, &of);
>> +    if (ret < 0) {
>> +        goto free;
>> +    }
>> +
>> +    /* create a new image */
>> +
>> +    ol->prot_type = QB_PROT_FILE;
>> +    ol->prot_op.o_file.filename = filename2;
>> +    of->fmt_type = QB_FMT_QCOW2;
>> +    of->fmt_op.o_qcow2.virt_size = 100 * 1024;
>> +    flag = 0;
>> +
>> +    ret = qb_create(broker, qbs, ol, of, flag);
>> +    if (ret < 0) {
>> +        qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +        printf("create fail 1. %s.\n", err_str);
>> +        goto unlink;
>> +    }
>> +
>> +    ol->prot_type = QB_PROT_FILE;
>> +    ol->prot_op.o_file.filename = filename1;
>> +    of->fmt_type = QB_FMT_QCOW2;
>> +    of->fmt_op.o_qcow2.backing_loc.prot_type = QB_PROT_FILE;
>> +    of->fmt_op.o_qcow2.backing_loc.prot_op.o_file.filename = filename2;
>> +    flag = 0;
>> +    ret = qb_create(broker, qbs, ol, of, flag);
>> +    if (ret < 0) {
>> +        qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +        printf("create fail 2. %s.\n", err_str);
>> +        goto unlink;
>> +    }
>> +
>> +    /* get informations */
>> +    ol->prot_type = QB_PROT_FILE;
>> +    ol->prot_op.o_file.filename = filename1;
>> +    of->fmt_type = QB_FMT_NONE;
>> +    flag = LIBQBLOCK_O_NO_BACKING;
>> +    ret = qb_open(broker, qbs, ol, of, flag);
>> +    if (ret < 0) {
>> +        qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +        printf("info getting, open failed. %s.\n", err_str);
>> +        goto free;
>> +    }
>> +
>> +    while (1) {
>> +        ret = qb_info_image_static_get(broker, qbs, &info_st);
>> +        if (ret < 0) {
>> +            qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +            printf("info get error. %s.\n", err_str);
>> +            goto close;
>> +        }
>> +        print_info_image_static(info_st);
>> +        qb_close(broker, qbs);
>> +        if ((info_st->member_addr->backing_loc == NULL) ||
>> +            (info_st->member_addr->backing_loc->prot_type == QB_FMT_NONE)) {
>> +            break;
>> +        }
>> +        *ol = *(info_st->member_addr->backing_loc);
>> +        ret = qb_open(broker, qbs, ol, of, flag);
>> +        if (ret < 0) {
>> +            qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +            printf("info getting, open failed in backing file. %s.\n",
>> +                                                       err_str);
>> +            goto free;
>> +        }
>> +        qb_info_image_static_delete(broker, &info_st);
>> +    }
>> +    /* read and write the image */
>> +    ol->prot_type = QB_PROT_FILE;
>> +    ol->prot_op.o_file.filename = filename1;
>> +    of->fmt_type = QB_FMT_NONE;
>> +    flag = LIBQBLOCK_O_RDWR;
>> +    ret = qb_open(broker, qbs, ol, of, flag);
>> +    if (ret < 0) {
>> +        qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +        printf("open failed. %s.\n", err_str);
>> +        goto free;
>> +    }
>> +
>> +    buf_w[1] = 1;
>> +    buf_w[2] = 2;
>> +    buf_w[514] = 4;
>> +    memset(buf_r, 0, sizeof(buf_r));
>> +
>> +    ret = qb_write(broker, qbs, buf_w, test_len, test_offset);
>> +    if (ret < 0) {
>> +        qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +        printf("%s.\n", err_str);
>> +        goto close;
>> +    }
>> +
>> +    ret = qb_read(broker, qbs, buf_r, test_len, test_offset);
>> +    if (ret < 0) {
>> +        qb_error_get_human_str(broker, err_str, sizeof(err_str));
>> +        printf("%s.\n", err_str);
>> +        goto close;
>> +    }
>> +
>> +    test_check(&vdata);
>> +
>> + close:
>> +    qb_close(broker, qbs);
>> + unlink:
>> +    unlink(filename1);
>> +    unlink(filename2);
>> + free:
>> +    if (info_st != NULL) {
>> +        qb_info_image_static_delete(broker, &info_st);
>> +    }
>> +    if (qbs != NULL) {
>> +        qb_state_delete(broker, &qbs);
>> +    }
>> +    if (ol != NULL) {
>> +        qb_prot_info_delete(broker, &ol);
>> +    }
>> +    if (of != NULL) {
>> +        qb_fmt_info_delete(broker, &of);
>> +    }
>> +    if (broker != NULL) {
>> +        qb_broker_delete(&broker);
>> +    }
>>       return 0;
>>   }
>>
>
>


-- 
Best Regards

Wenchao Xia