Branch data Line data Source code
1 : : /*
2 : : * Copyright (c) 2016, Citrix Systems, Inc.
3 : : *
4 : : * All rights reserved.
5 : : *
6 : : * Redistribution and use in source and binary forms, with or without
7 : : * modification, are permitted provided that the following conditions are met:
8 : : *
9 : : * 1. Redistributions of source code must retain the above copyright
10 : : * notice, this list of conditions and the following disclaimer.
11 : : * 2. Redistributions in binary form must reproduce the above copyright
12 : : * notice, this list of conditions and the following disclaimer in the
13 : : * documentation and/or other materials provided with the distribution.
14 : : * 3. Neither the name of the copyright holder nor the names of its
15 : : * contributors may be used to endorse or promote products derived from
16 : : * this software without specific prior written permission.
17 : : *
18 : : * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 : : * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 : : * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 : : * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
22 : : * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
23 : : * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
24 : : * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
25 : : * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
26 : : * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
27 : : * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28 : : * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 : : */
30 : :
31 : : #ifdef HAVE_CONFIG_H
32 : : #include "config.h"
33 : : #endif
34 : :
35 : : #include <errno.h>
36 : : #include <fcntl.h>
37 : : #include <stdio.h>
38 : : #include <stdlib.h>
39 : : #include <unistd.h>
40 : :
41 : : #include "libvhd.h"
42 : :
43 : : int
44 : 0 : vhd_util_set_field(int argc, char **argv)
45 : : {
46 : : long value;
47 : : int err, c;
48 : : vhd_context_t vhd;
49 : : char *name, *field;
50 : :
51 : 0 : err = -EINVAL;
52 : 0 : value = 0;
53 : 0 : name = NULL;
54 : 0 : field = NULL;
55 : :
56 [ # # ]: 0 : if (!argc || !argv)
57 : : goto usage;
58 : :
59 : 0 : optind = 0;
60 [ # # ]: 0 : while ((c = getopt(argc, argv, "n:f:v:h")) != -1) {
61 [ # # # # ]: 0 : switch (c) {
62 : : case 'n':
63 : 0 : name = optarg;
64 : 0 : break;
65 : : case 'f':
66 : 0 : field = optarg;
67 : 0 : break;
68 : : case 'v':
69 : 0 : err = 0;
70 : 0 : value = strtol(optarg, NULL, 10);
71 : 0 : break;
72 : : case 'h':
73 : : default:
74 : : goto usage;
75 : : }
76 : : }
77 : :
78 [ # # ][ # # ]: 0 : if (!name || !field || optind != argc || err)
[ # # ]
79 : : goto usage;
80 : :
81 [ # # ]: 0 : if (strnlen(field, 25) >= 25) {
82 : : printf("invalid field\n");
83 : : goto usage;
84 : : }
85 : :
86 [ # # ][ # # ]: 0 : if (strcmp(field, "hidden") && strcmp(field, "marker")) {
87 : : printf("invalid field %s\n", field);
88 : : goto usage;
89 : : }
90 : :
91 [ # # ]: 0 : if (value < 0 || value > 255) {
92 : : printf("invalid value %ld\n", value);
93 : : goto usage;
94 : : }
95 : :
96 : 0 : err = vhd_open(&vhd, name, VHD_OPEN_RDWR);
97 [ # # ]: 0 : if (err) {
98 : : printf("error opening %s: %d\n", name, err);
99 : 0 : return err;
100 : : }
101 : :
102 [ # # ]: 0 : if (!strcmp(field, "hidden")) {
103 : 0 : vhd.footer.hidden = (char)value;
104 : 0 : err = vhd_write_footer(&vhd, &vhd.footer);
105 [ # # ][ # # ]: 0 : if (err == -ENOSPC && vhd_type_dynamic(&vhd) && value)
[ # # ]
106 : : /* if no space to write the primary footer, at least write the
107 : : * backup footer so that it's possible to delete the VDI */
108 : 0 : err = vhd_write_footer_at(&vhd, &vhd.footer, 0);
109 : : } else {
110 : 0 : err = vhd_set_marker(&vhd, (char)value);
111 : : }
112 : :
113 : 0 : vhd_close(&vhd);
114 : : return err;
115 : :
116 : : usage:
117 : : printf("options: <-n name> <-f field> <-v value> [-h help]\n");
118 : : return -EINVAL;
119 : : }
|