1 /* Copyright (c) 2008, XenSource Inc.
2  * All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are met:
6  *     * Redistributions of source code must retain the above copyright
7  *       notice, this list of conditions and the following disclaimer.
8  *     * Redistributions in binary form must reproduce the above copyright
9  *       notice, this list of conditions and the following disclaimer in the
10  *       documentation and/or other materials provided with the distribution.
11  *     * Neither the name of XenSource Inc. nor the names of its contributors
12  *       may be used to endorse or promote products derived from this software
13  *       without specific prior written permission.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
19  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
23  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
24  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27 #include <errno.h>
28 #include <fcntl.h>
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <unistd.h>
32 
33 #include "libvhd.h"
34 
35 int
vhd_util_fill(int argc,char ** argv)36 vhd_util_fill(int argc, char **argv)
37 {
38 	int err, c;
39 	char *buf, *name;
40 	vhd_context_t vhd;
41 	uint64_t i, sec, secs;
42 
43 	buf  = NULL;
44 	name = NULL;
45 
46 	if (!argc || !argv)
47 		goto usage;
48 
49 	optind = 0;
50 	while ((c = getopt(argc, argv, "n:h")) != -1) {
51 		switch (c) {
52 		case 'n':
53 			name = optarg;
54 			break;
55 		case 'h':
56 		default:
57 			goto usage;
58 		}
59 	}
60 
61 	if (!name || optind != argc)
62 		goto usage;
63 
64 	err = vhd_open(&vhd, name, VHD_OPEN_RDWR);
65 	if (err) {
66 		printf("error opening %s: %d\n", name, err);
67 		return err;
68 	}
69 
70 	err = vhd_get_bat(&vhd);
71 	if (err)
72 		goto done;
73 
74 	err = posix_memalign((void **)&buf, 4096, vhd.header.block_size);
75 	if (err) {
76 		err = -err;
77 		goto done;
78 	}
79 
80 	sec  = 0;
81 	secs = vhd.header.block_size >> VHD_SECTOR_SHIFT;
82 
83 	for (i = 0; i < vhd.header.max_bat_size; i++) {
84 		err = vhd_io_read(&vhd, buf, sec, secs);
85 		if (err)
86 			goto done;
87 
88 		err = vhd_io_write(&vhd, buf, sec, secs);
89 		if (err)
90 			goto done;
91 
92 		sec += secs;
93 	}
94 
95 	err = 0;
96 
97  done:
98 	free(buf);
99 	vhd_close(&vhd);
100 	return err;
101 
102 usage:
103 	printf("options: <-n name> [-h help]\n");
104 	return -EINVAL;
105 }
106