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
> Signed-off-by: Wenchao Xia <xiaw...@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 <xiaw...@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;
> }
>
