1 /*- 2 * Copyright (c) 2004 Marcel Moolenaar 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * 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 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #if HAVE_NBTOOL_CONFIG_H 28 #include "nbtool_config.h" 29 #endif 30 31 #include <sys/cdefs.h> 32 #ifdef __FBSDID 33 __FBSDID("$FreeBSD: src/sbin/gpt/remove.c,v 1.10 2006/10/04 18:20:25 marcel Exp $"); 34 #endif 35 #ifdef __RCSID 36 __RCSID("$NetBSD: uuid.c,v 1.2 2024/08/19 17:15:38 christos Exp $"); 37 #endif 38 39 #include <sys/types.h> 40 41 #include <err.h> 42 #include <stddef.h> 43 #include <stdio.h> 44 #include <stdlib.h> 45 #include <string.h> 46 #include <unistd.h> 47 48 #include "map.h" 49 #include "gpt.h" 50 #include "gpt_private.h" 51 52 static int cmd_uuid(gpt_t, int, char *[]); 53 54 static const char *uuidhelp[] = { 55 "-a", 56 "[-b blocknr] [-i index] [-L label] [-s sectors] [-t type] [-U newuuid]", 57 }; 58 59 struct gpt_cmd c_uuid = { 60 "uuid", 61 cmd_uuid, 62 uuidhelp, __arraycount(uuidhelp), 63 GPT_SYNC, 64 }; 65 66 #define usage() gpt_usage(NULL, &c_uuid) 67 68 static void 69 change_ent(struct gpt_ent *ent, void *v, int backup) 70 { 71 static gpt_uuid_t uuidstore; 72 73 if (v != NULL) { 74 memcpy(uuidstore, v, sizeof(uuidstore)); 75 } 76 else if (!backup) 77 gpt_uuid_generate(NULL, uuidstore); 78 memmove(ent->ent_guid, uuidstore, sizeof(ent->ent_guid)); 79 } 80 81 static void 82 change_hdr(struct gpt_hdr *hdr, void *v, int backup) 83 { 84 static gpt_uuid_t uuidstore; 85 86 if (!backup) 87 gpt_uuid_generate(NULL, uuidstore); 88 memmove(hdr->hdr_guid, uuidstore, sizeof(hdr->hdr_guid)); 89 } 90 91 static int 92 cmd_uuid(gpt_t gpt, int argc, char *argv[]) 93 { 94 int ch, rc; 95 struct gpt_find find; 96 gpt_uuid_t new_uuid; 97 void *v; 98 99 if (gpt == NULL) 100 return usage(); 101 102 memset(&find, 0, sizeof(find)); 103 find.msg = "UUID changed"; 104 105 /* Get the uuid options */ 106 v = NULL; 107 while ((ch = getopt(argc, argv, GPT_FIND "U:")) != -1) { 108 switch (ch) { 109 case 'U': 110 if (gpt_uuid_parse(optarg, new_uuid) == -1) 111 return usage(); 112 v = new_uuid; 113 break; 114 default: 115 if (gpt_add_find(gpt, &find, ch) == -1) 116 return usage(); 117 break; 118 } 119 } 120 121 if (argc != optind) 122 return usage(); 123 124 rc = gpt_change_ent(gpt, &find, change_ent, v); 125 if (rc != 0) 126 return rc; 127 128 if (find.all) 129 return gpt_change_hdr(gpt, &find, change_hdr, NULL); 130 131 return 0; 132 } 133