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.1 2019/06/25 04:53:40 jnemeth 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]", 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 (!backup) 74 gpt_uuid_generate(NULL, uuidstore); 75 memmove(ent->ent_guid, uuidstore, sizeof(ent->ent_guid)); 76 } 77 78 static void 79 change_hdr(struct gpt_hdr *hdr, void *v, int backup) 80 { 81 static gpt_uuid_t uuidstore; 82 83 if (!backup) 84 gpt_uuid_generate(NULL, uuidstore); 85 memmove(hdr->hdr_guid, uuidstore, sizeof(hdr->hdr_guid)); 86 } 87 88 static int 89 cmd_uuid(gpt_t gpt, int argc, char *argv[]) 90 { 91 int ch, rc; 92 struct gpt_find find; 93 94 memset(&find, 0, sizeof(find)); 95 find.msg = "UUID changed"; 96 97 /* Get the uuid options */ 98 while ((ch = getopt(argc, argv, GPT_FIND)) != -1) { 99 if (gpt == NULL || gpt_add_find(gpt, &find, ch) == -1) 100 return usage(); 101 } 102 103 if (gpt == NULL || argc != optind) 104 return usage(); 105 106 rc = gpt_change_ent(gpt, &find, change_ent, NULL); 107 if (rc != 0) 108 return rc; 109 110 if (find.all) 111 return gpt_change_hdr(gpt, &find, change_hdr, NULL); 112 113 return 0; 114 } 115