From patchwork Fri Jun 24 08:18:19 2011 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Feiran Zheng X-Patchwork-Id: 101754 Return-Path: X-Original-To: incoming@patchwork.ozlabs.org Delivered-To: patchwork-incoming@bilbo.ozlabs.org Received: from lists.gnu.org (lists.gnu.org [140.186.70.17]) (using TLSv1 with cipher AES256-SHA (256/256 bits)) (Client did not present a certificate) by ozlabs.org (Postfix) with ESMTPS id EC659B6F76 for ; Fri, 24 Jun 2011 19:12:35 +1000 (EST) Received: from localhost ([::1]:38690 helo=lists.gnu.org) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1Qa1sY-0001IE-W1 for incoming@patchwork.ozlabs.org; Fri, 24 Jun 2011 04:36:43 -0400 Received: from eggs.gnu.org ([140.186.70.92]:54164) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1Qa1cg-0006OB-Og for qemu-devel@nongnu.org; Fri, 24 Jun 2011 04:20:20 -0400 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.71) (envelope-from ) id 1Qa1ce-0001pz-Jg for qemu-devel@nongnu.org; Fri, 24 Jun 2011 04:20:18 -0400 Received: from mail-iy0-f173.google.com ([209.85.210.173]:63570) by eggs.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1Qa1ce-0001ms-4F for qemu-devel@nongnu.org; Fri, 24 Jun 2011 04:20:16 -0400 Received: by mail-iy0-f173.google.com with SMTP id 3so2656746iyb.4 for ; Fri, 24 Jun 2011 01:20:15 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=gamma; h=domainkey-signature:from:to:cc:subject:date:message-id:x-mailer :in-reply-to:references; bh=8lyVzMvk9iXdH1l/W0+fTBkOrQOi64ezwXt2NVxNga0=; b=QGNy1TrD6ZU4fsZxi8QzjETdWwpmfD3IUxetz+h+VUuLEsq+3K2l8TJm+fJuuXnxqr Q/uuxT43TDcQ451ewD6F5Lk9e8ZbtodZ/A+LrD8zgrYjY37okmcBW/8jhD+2h75Y0XXt DAvyZez4KK+8DRWSxC7Pf9z2k67Vq1lTMJoh4= DomainKey-Signature: a=rsa-sha1; c=nofws; d=gmail.com; s=gamma; h=from:to:cc:subject:date:message-id:x-mailer:in-reply-to:references; b=EN0mheCCvGQXFmq3Ae70dxY8Z3Fc6lsiEa/Qd5pZuJyrOdpDprtTBE1oNos/cncrEe 9y+neitktuQ8gwRsFENKhxBkjudKX+BSvDtBbnKYl3UdRa/ap3xEyJYmeiCb56Pyu2a0 uPBoJd8xHO2fVLnEmQMtQrnTIJObMxOS2wHtY= Received: by 10.42.167.72 with SMTP id r8mr3193064icy.131.1308903615836; Fri, 24 Jun 2011 01:20:15 -0700 (PDT) Received: from localhost.localdomain (admis.tongji.edu.cn [202.120.189.202]) by mx.google.com with ESMTPS id f19sm1351912ibl.66.2011.06.24.01.20.11 (version=TLSv1/SSLv3 cipher=OTHER); Fri, 24 Jun 2011 01:20:15 -0700 (PDT) From: famcool@gmail.com To: qemu-devel@nongnu.org Date: Fri, 24 Jun 2011 16:18:19 +0800 Message-Id: <1308903507-10799-5-git-send-email-famcool@gmail.com> X-Mailer: git-send-email 1.7.5.2 In-Reply-To: <1308903507-10799-1-git-send-email-famcool@gmail.com> References: <1308903507-10799-1-git-send-email-famcool@gmail.com> X-detected-operating-system: by eggs.gnu.org: GNU/Linux 2.6 (newer, 2) X-Received-From: 209.85.210.173 Cc: kwolf@redhat.com, Fam Zheng , hch@lst.de, stefanha@gmail.com Subject: [Qemu-devel] [PATCH v2 04/12] VMDK: separate vmdk_open by format version X-BeenThere: qemu-devel@nongnu.org X-Mailman-Version: 2.1.14 Precedence: list List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Errors-To: qemu-devel-bounces+incoming=patchwork.ozlabs.org@nongnu.org Sender: qemu-devel-bounces+incoming=patchwork.ozlabs.org@nongnu.org From: Fam Zheng Separate vmdk_open by subformats to: * vmdk_open_vmdk3 * vmdk_open_vmdk4 Signed-off-by: Fam Zheng --- block/vmdk.c | 225 ++++++++++++++++++++++++++++++++++++++-------------------- 1 files changed, 147 insertions(+), 78 deletions(-) diff --git a/block/vmdk.c b/block/vmdk.c index 71d81bc..8d55e5f 100644 --- a/block/vmdk.c +++ b/block/vmdk.c @@ -398,84 +398,26 @@ static int vmdk_parent_open(BlockDriverState *bs) return 0; } -static int vmdk_open(BlockDriverState *bs, int flags) +static int vmdk_init_tables(BlockDriverState *bs, VmdkExtent *extent) { - BDRVVmdkState *s = bs->opaque; - uint32_t magic; + int ret; int l1_size, i; - VmdkExtent *extent = NULL; - - if (bdrv_pread(bs->file, 0, &magic, sizeof(magic)) != sizeof(magic)) - goto fail; - - magic = be32_to_cpu(magic); - if (magic == VMDK3_MAGIC) { - VMDK3Header header; - s->extents = qemu_mallocz(sizeof(VmdkExtent)); - s->num_extents = 1; - if (bdrv_pread(bs->file, sizeof(magic), &header, sizeof(header)) - != sizeof(header)) { - goto fail; - } - extent = s->extents; - extent->flat = false; - extent->file = bs->file; - extent->cluster_sectors = le32_to_cpu(header.granularity); - extent->l2_size = 1 << 9; - extent->l1_size = 1 << 6; - extent->sectors = le32_to_cpu(header.disk_sectors); - extent->end_sector = le32_to_cpu(header.disk_sectors); - extent->l1_table_offset = le32_to_cpu(header.l1dir_offset) << 9; - extent->l1_backup_table_offset = 0; - extent->l1_entry_sectors = extent->l2_size * extent->cluster_sectors; - } else if (magic == VMDK4_MAGIC) { - VMDK4Header header; - s->extents = qemu_mallocz(sizeof(VmdkExtent)); - s->num_extents = 1; - if (bdrv_pread(bs->file, sizeof(magic), &header, sizeof(header)) - != sizeof(header)) { - goto fail; - } - extent = s->extents; - extent->file = bs->file; - extent->sectors = le64_to_cpu(header.capacity); - extent->end_sector = le64_to_cpu(header.capacity); - extent->cluster_sectors = le64_to_cpu(header.granularity); - extent->l2_size = le32_to_cpu(header.num_gtes_per_gte); - extent->l1_entry_sectors = extent->l2_size * extent->cluster_sectors; - if (extent->l1_entry_sectors <= 0) { - goto fail; - } - extent->l1_size = (extent->sectors + extent->l1_entry_sectors - 1) - / extent->l1_entry_sectors; - extent->l1_table_offset = le64_to_cpu(header.rgd_offset) << 9; - extent->l1_backup_table_offset = le64_to_cpu(header.gd_offset) << 9; - - // try to open parent images, if exist - if (vmdk_parent_open(bs) != 0) - goto fail; - // write the CID once after the image creation - s->parent_cid = vmdk_read_cid(bs,1); - } else { - goto fail; - } - - /* sum up the total sectors */ - bs->total_sectors = 0; - for (i = 0; i < s->num_extents; i++) { - bs->total_sectors += s->extents[i].sectors; - } /* read the L1 table */ l1_size = extent->l1_size * sizeof(uint32_t); extent->l1_table = qemu_malloc(l1_size); - if (bdrv_pread(bs->file, - extent->l1_table_offset, - extent->l1_table, - l1_size) - != l1_size) { + if (!extent->l1_table) { + ret = -ENOMEM; goto fail; } + ret = bdrv_pread(bs->file, + extent->l1_table_offset, + extent->l1_table, + l1_size); + if (ret != l1_size) { + ret = ret < 0 ? ret : -EIO; + goto fail_l1; + } for (i = 0; i < extent->l1_size; i++) { le32_to_cpus(&extent->l1_table[i]); } @@ -490,21 +432,148 @@ static int vmdk_open(BlockDriverState *bs, int flags) goto fail; } for (i = 0; i < extent->l1_size; i++) { + if (!extent->l1_backup_table) { + ret = -ENOMEM; + goto fail_l1; + } + } + ret = bdrv_pread(bs->file, + extent->l1_backup_table_offset, + extent->l1_backup_table, + l1_size); + if (ret != l1_size) { + ret = ret < 0 ? ret : -EIO; + goto fail; + } + for (i = 0; i < extent->l1_size; i++) { le32_to_cpus(&extent->l1_backup_table[i]); } } extent->l2_cache = qemu_malloc(extent->l2_size * L2_CACHE_SIZE * sizeof(uint32_t)); + if (!extent->l2_cache) { + ret = -ENOMEM; + goto fail_l1b; + } return 0; + fail_l1b: + qemu_free(extent->l1_backup_table); + fail_l1: + qemu_free(extent->l1_table); fail: - for (i = 0; i < s->num_extents; i++) { - qemu_free(s->extents[i].l1_backup_table); - qemu_free(s->extents[i].l1_table); - qemu_free(s->extents[i].l2_cache); + return ret; +} + +static int vmdk_open_vmdk3(BlockDriverState *bs, int flags) +{ + int ret; + uint32_t magic; + BDRVVmdkState *s = bs->opaque; + VMDK3Header header; + VmdkExtent *extent; + + s->extents = qemu_mallocz(sizeof(VmdkExtent)); + if (!s->extents) { + ret = -ENOMEM; + goto fail; } + s->num_extents = 1; + ret = bdrv_pread(bs->file, sizeof(magic), &header, sizeof(header)); + if (ret != sizeof(header)) { + ret = ret < 0 ? ret : -EIO; + goto fail; + } + extent = s->extents; + extent->flat = false; + extent->file = bs->file; + extent->cluster_sectors = le32_to_cpu(header.granularity); + extent->l2_size = 1 << 9; + extent->l1_size = 1 << 6; + extent->sectors = le32_to_cpu(header.disk_sectors); + extent->end_sector = le32_to_cpu(header.disk_sectors); + extent->l1_table_offset = le32_to_cpu(header.l1dir_offset) << 9; + extent->l1_backup_table_offset = 0; + extent->l1_entry_sectors = extent->l2_size * extent->cluster_sectors; + bs->total_sectors = extent->sectors; + ret = vmdk_init_tables(bs, extent); + if (ret) { + goto fail; + } + return 0; + fail: + qemu_free(s->extents); + return ret; +} + +static int vmdk_open_vmdk4(BlockDriverState *bs, int flags) +{ + int ret; + uint32_t magic; + VMDK4Header header; + BDRVVmdkState *s = bs->opaque; + VmdkExtent *extent; + + s->extents = qemu_mallocz(sizeof(VmdkExtent)); + if (!s->extents) { + ret = -ENOMEM; + goto fail; + } + s->num_extents = 1; + ret = bdrv_pread(bs->file, sizeof(magic), &header, sizeof(header)); + if (ret != sizeof(header)) { + ret = ret < 0 ? ret : -EIO; + goto fail; + } + extent = s->extents; + extent->file = bs->file; + extent->sectors = le64_to_cpu(header.capacity); + extent->end_sector = le64_to_cpu(header.capacity); + extent->cluster_sectors = le64_to_cpu(header.granularity); + extent->l2_size = le32_to_cpu(header.num_gtes_per_gte); + extent->l1_entry_sectors = extent->l2_size * extent->cluster_sectors; + if (extent->l1_entry_sectors <= 0) { + ret = -EINVAL; + goto fail; + } + extent->l1_size = (extent->sectors + extent->l1_entry_sectors - 1) + / extent->l1_entry_sectors; + extent->l1_table_offset = le64_to_cpu(header.rgd_offset) << 9; + extent->l1_backup_table_offset = le64_to_cpu(header.gd_offset) << 9; + + /* try to open parent images, if exist */ + ret = vmdk_parent_open(bs); + if (ret) { + goto fail; + } + s->parent_cid = vmdk_read_cid(bs, 1); + bs->total_sectors = extent->sectors; + ret = vmdk_init_tables(bs, extent); + if (ret) { + goto fail; + } + return 0; + fail: qemu_free(s->extents); - return -1; + return ret; +} + +static int vmdk_open(BlockDriverState *bs, int flags) +{ + uint32_t magic; + + if (bdrv_pread(bs->file, 0, &magic, sizeof(magic)) != sizeof(magic)) { + return -EIO; + } + + magic = be32_to_cpu(magic); + if (magic == VMDK3_MAGIC) { + return vmdk_open_vmdk3(bs, flags); + } else if (magic == VMDK4_MAGIC) { + return vmdk_open_vmdk4(bs, flags); + } else { + return -EINVAL; + } } static int get_whole_cluster(BlockDriverState *bs, @@ -591,11 +660,11 @@ static uint64_t get_cluster_offset(BlockDriverState *bs, if (!l2_offset) { return 0; } - for(i = 0; i < L2_CACHE_SIZE; i++) { + for (i = 0; i < L2_CACHE_SIZE; i++) { if (l2_offset == extent->l2_cache_offsets[i]) { /* increment the hit count */ if (++extent->l2_cache_counts[i] == 0xffffffff) { - for(j = 0; j < L2_CACHE_SIZE; j++) { + for (j = 0; j < L2_CACHE_SIZE; j++) { extent->l2_cache_counts[j] >>= 1; } } @@ -606,7 +675,7 @@ static uint64_t get_cluster_offset(BlockDriverState *bs, /* not found: load a new entry in the least used one */ min_index = 0; min_count = 0xffffffff; - for(i = 0; i < L2_CACHE_SIZE; i++) { + for (i = 0; i < L2_CACHE_SIZE; i++) { if (extent->l2_cache_counts[i] < min_count) { min_count = extent->l2_cache_counts[i]; min_index = i;