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_repair(int argc,char ** argv)36 vhd_util_repair(int argc, char **argv)
37 {
38 	char *name;
39 	int err, c;
40 	off_t eof;
41 	vhd_context_t vhd;
42 
43 	name = NULL;
44 
45 	if (!argc || !argv)
46 		goto usage;
47 
48 	optind = 0;
49 	while ((c = getopt(argc, argv, "n:h")) != -1) {
50 		switch (c) {
51 		case 'n':
52 			name = optarg;
53 			break;
54 		case 'h':
55 		default:
56 			goto usage;
57 		}
58 	}
59 
60 	if (!name || optind != argc)
61 		goto usage;
62 
63 	err = vhd_open(&vhd, name, VHD_OPEN_RDWR);
64 	if (err) {
65 		printf("error opening %s: %d\n", name, err);
66 		return err;
67 	}
68 
69 	err = vhd_end_of_data(&vhd, &eof);
70 	if (err) {
71 		printf("error finding end of data: %d\n", err);
72 		goto done;
73 	}
74 
75 	err = vhd_write_footer_at(&vhd, &vhd.footer, eof);
76 
77  done:
78 	vhd_close(&vhd);
79 	return err;
80 
81 usage:
82 	printf("options: <-n name> [-h help]\n");
83 	return -EINVAL;
84 }
85