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                 :           0 :         long value;
      47                 :           0 :         int err, c;
      48                 :           0 :         vhd_context_t vhd;
      49                 :           0 :         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                 :           0 :                 goto usage;
      58                 :             : 
      59                 :           0 :         optind = 0;
      60         [ #  # ]:           0 :         while ((c = getopt(argc, argv, "n:f:v:h")) != -1) {
      61   [ #  #  #  # ]:           0 :                 switch (c) {
      62                 :           0 :                 case 'n':
      63                 :           0 :                         name = optarg;
      64                 :           0 :                         break;
      65                 :           0 :                 case 'f':
      66                 :           0 :                         field = optarg;
      67                 :           0 :                         break;
      68                 :           0 :                 case 'v':
      69                 :           0 :                         err   = 0;
      70                 :           0 :                         value = strtol(optarg, NULL, 10);
      71                 :           0 :                         break;
      72                 :           0 :                 case 'h':
      73                 :             :                 default:
      74                 :           0 :                         goto usage;
      75                 :             :                 }
      76                 :             :         }
      77                 :             : 
      78   [ #  #  #  #  :           0 :         if (!name || !field || optind != argc || err)
                   #  # ]
      79                 :           0 :                 goto usage;
      80                 :             : 
      81         [ #  # ]:           0 :         if (strnlen(field, 25) >= 25) {
      82                 :           0 :                 printf("invalid field\n");
      83                 :           0 :                 goto usage;
      84                 :             :         }
      85                 :             : 
      86   [ #  #  #  # ]:           0 :         if (strcmp(field, "hidden") && strcmp(field, "marker")) {
      87                 :           0 :                 printf("invalid field %s\n", field);
      88                 :           0 :                 goto usage;
      89                 :             :         }
      90                 :             : 
      91         [ #  # ]:           0 :         if (value < 0 || value > 255) {
      92                 :           0 :                 printf("invalid value %ld\n", value);
      93                 :           0 :                 goto usage;
      94                 :             :         }
      95                 :             : 
      96                 :           0 :         err = vhd_open(&vhd, name, VHD_OPEN_RDWR);
      97         [ #  # ]:           0 :         if (err) {
      98                 :           0 :                 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                 :           0 : usage:
     117                 :           0 :         printf("options: <-n name> <-f field> <-v value> [-h help]\n");
     118                 :             :         return -EINVAL;
     119                 :             : }