1452f657cSIvan Voras /*- 23728855aSPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 33728855aSPedro F. Giffuni * 4452f657cSIvan Voras * Copyright (c) 2008 Marius Nuennerich 5452f657cSIvan Voras * All rights reserved. 6452f657cSIvan Voras * 7452f657cSIvan Voras * Redistribution and use in source and binary forms, with or without 8452f657cSIvan Voras * modification, are permitted provided that the following conditions 9452f657cSIvan Voras * are met: 10452f657cSIvan Voras * 1. Redistributions of source code must retain the above copyright 11452f657cSIvan Voras * notice, this list of conditions and the following disclaimer. 12452f657cSIvan Voras * 2. Redistributions in binary form must reproduce the above copyright 13452f657cSIvan Voras * notice, this list of conditions and the following disclaimer in the 14452f657cSIvan Voras * documentation and/or other materials provided with the distribution. 15452f657cSIvan Voras * 16452f657cSIvan Voras * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17452f657cSIvan Voras * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18452f657cSIvan Voras * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19452f657cSIvan Voras * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20452f657cSIvan Voras * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21452f657cSIvan Voras * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22452f657cSIvan Voras * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23452f657cSIvan Voras * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24452f657cSIvan Voras * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25452f657cSIvan Voras * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26452f657cSIvan Voras * SUCH DAMAGE. 27452f657cSIvan Voras */ 28452f657cSIvan Voras 29452f657cSIvan Voras #include <sys/cdefs.h> 30452f657cSIvan Voras __FBSDID("$FreeBSD$"); 31452f657cSIvan Voras 32452f657cSIvan Voras #include <sys/param.h> 33452f657cSIvan Voras #include <sys/systm.h> 34452f657cSIvan Voras #include <sys/kernel.h> 35452f657cSIvan Voras #include <sys/malloc.h> 36452f657cSIvan Voras #include <sys/kobj.h> 37452f657cSIvan Voras #include <sys/gpt.h> 385d807a0eSAndrey V. Elsukov #include <sys/sbuf.h> 39452f657cSIvan Voras 40452f657cSIvan Voras #include <geom/geom.h> 41452f657cSIvan Voras #include <geom/label/g_label.h> 42452f657cSIvan Voras #include <geom/part/g_part.h> 43452f657cSIvan Voras 44452f657cSIvan Voras #define PART_CLASS_NAME "PART" 45452f657cSIvan Voras #define SCHEME_NAME "GPT" 46452f657cSIvan Voras 4763f4d880SIvan Voras /* XXX: Also defined in geom/part/g_part_gpt.c */ 48452f657cSIvan Voras struct g_part_gpt_entry { 49452f657cSIvan Voras struct g_part_entry base; 50452f657cSIvan Voras struct gpt_ent ent; 51452f657cSIvan Voras }; 52452f657cSIvan Voras 5363f4d880SIvan Voras /* XXX: Shamelessly stolen from g_part_gpt.c */ 54452f657cSIvan Voras static void 55452f657cSIvan Voras sbuf_nprintf_utf16(struct sbuf *sb, uint16_t *str, size_t len) 56452f657cSIvan Voras { 57452f657cSIvan Voras u_int bo; 58452f657cSIvan Voras uint32_t ch; 59452f657cSIvan Voras uint16_t c; 60452f657cSIvan Voras 61452f657cSIvan Voras bo = LITTLE_ENDIAN; /* GPT is little-endian */ 62452f657cSIvan Voras while (len > 0 && *str != 0) { 63452f657cSIvan Voras ch = (bo == BIG_ENDIAN) ? be16toh(*str) : le16toh(*str); 64452f657cSIvan Voras str++, len--; 65452f657cSIvan Voras if ((ch & 0xf800) == 0xd800) { 66452f657cSIvan Voras if (len > 0) { 67452f657cSIvan Voras c = (bo == BIG_ENDIAN) ? be16toh(*str) 68452f657cSIvan Voras : le16toh(*str); 69452f657cSIvan Voras str++, len--; 70452f657cSIvan Voras } else 71452f657cSIvan Voras c = 0xfffd; 72452f657cSIvan Voras if ((ch & 0x400) == 0 && (c & 0xfc00) == 0xdc00) { 73452f657cSIvan Voras ch = ((ch & 0x3ff) << 10) + (c & 0x3ff); 74452f657cSIvan Voras ch += 0x10000; 75452f657cSIvan Voras } else 76452f657cSIvan Voras ch = 0xfffd; 77452f657cSIvan Voras } else if (ch == 0xfffe) { /* BOM (U+FEFF) swapped. */ 78452f657cSIvan Voras bo = (bo == BIG_ENDIAN) ? LITTLE_ENDIAN : BIG_ENDIAN; 79452f657cSIvan Voras continue; 80452f657cSIvan Voras } else if (ch == 0xfeff) /* BOM (U+FEFF) unswapped. */ 81452f657cSIvan Voras continue; 82452f657cSIvan Voras 83452f657cSIvan Voras /* Write the Unicode character in UTF-8 */ 84452f657cSIvan Voras if (ch < 0x80) 85452f657cSIvan Voras sbuf_printf(sb, "%c", ch); 86452f657cSIvan Voras else if (ch < 0x800) 87452f657cSIvan Voras sbuf_printf(sb, "%c%c", 0xc0 | (ch >> 6), 88452f657cSIvan Voras 0x80 | (ch & 0x3f)); 89452f657cSIvan Voras else if (ch < 0x10000) 90452f657cSIvan Voras sbuf_printf(sb, "%c%c%c", 0xe0 | (ch >> 12), 91452f657cSIvan Voras 0x80 | ((ch >> 6) & 0x3f), 0x80 | (ch & 0x3f)); 92452f657cSIvan Voras else if (ch < 0x200000) 93452f657cSIvan Voras sbuf_printf(sb, "%c%c%c%c", 0xf0 | (ch >> 18), 94452f657cSIvan Voras 0x80 | ((ch >> 12) & 0x3f), 95452f657cSIvan Voras 0x80 | ((ch >> 6) & 0x3f), 0x80 | (ch & 0x3f)); 96452f657cSIvan Voras } 97452f657cSIvan Voras } 98452f657cSIvan Voras 99452f657cSIvan Voras static void 100452f657cSIvan Voras g_label_gpt_taste(struct g_consumer *cp, char *label, size_t size) 101452f657cSIvan Voras { 102452f657cSIvan Voras struct g_provider *pp; 103452f657cSIvan Voras struct g_part_table *tp; 104452f657cSIvan Voras struct g_part_gpt_entry *part_gpt_entry; 105452f657cSIvan Voras struct sbuf *lbl; 106452f657cSIvan Voras 107452f657cSIvan Voras g_topology_assert_not(); 108452f657cSIvan Voras pp = cp->provider; 109452f657cSIvan Voras tp = (struct g_part_table *)pp->geom->softc; 110452f657cSIvan Voras label[0] = '\0'; 111452f657cSIvan Voras 11263f4d880SIvan Voras /* We taste only partitions handled by GPART */ 113452f657cSIvan Voras if (strncmp(pp->geom->class->name, PART_CLASS_NAME, sizeof(PART_CLASS_NAME))) 114452f657cSIvan Voras return; 115452f657cSIvan Voras /* and only GPT */ 116452f657cSIvan Voras if (strncmp(tp->gpt_scheme->name, SCHEME_NAME, sizeof(SCHEME_NAME))) 117452f657cSIvan Voras return; 118452f657cSIvan Voras 119452f657cSIvan Voras part_gpt_entry = (struct g_part_gpt_entry *)pp->private; 120452f657cSIvan Voras 121452f657cSIvan Voras /* 12263f4d880SIvan Voras * Create sbuf with biggest possible size. 12363f4d880SIvan Voras * We need max. 4 bytes for every 2-byte utf16 char. 124452f657cSIvan Voras */ 125452f657cSIvan Voras lbl = sbuf_new(NULL, NULL, sizeof(part_gpt_entry->ent.ent_name) << 1, SBUF_FIXEDLEN); 12663f4d880SIvan Voras /* Size is the number of characters, not bytes */ 127452f657cSIvan Voras sbuf_nprintf_utf16(lbl, part_gpt_entry->ent.ent_name, sizeof(part_gpt_entry->ent.ent_name) >> 1); 128452f657cSIvan Voras sbuf_finish(lbl); 129452f657cSIvan Voras strlcpy(label, sbuf_data(lbl), size); 130452f657cSIvan Voras sbuf_delete(lbl); 131452f657cSIvan Voras } 132452f657cSIvan Voras 133452f657cSIvan Voras static void 134452f657cSIvan Voras g_label_gpt_uuid_taste(struct g_consumer *cp, char *label, size_t size) 135452f657cSIvan Voras { 136452f657cSIvan Voras struct g_provider *pp; 137452f657cSIvan Voras struct g_part_table *tp; 138452f657cSIvan Voras struct g_part_gpt_entry *part_gpt_entry; 139452f657cSIvan Voras 140452f657cSIvan Voras g_topology_assert_not(); 141452f657cSIvan Voras pp = cp->provider; 142452f657cSIvan Voras tp = (struct g_part_table *)pp->geom->softc; 143452f657cSIvan Voras label[0] = '\0'; 144452f657cSIvan Voras 14563f4d880SIvan Voras /* We taste only partitions handled by GPART */ 146452f657cSIvan Voras if (strncmp(pp->geom->class->name, PART_CLASS_NAME, sizeof(PART_CLASS_NAME))) 147452f657cSIvan Voras return; 148452f657cSIvan Voras /* and only GPT */ 149452f657cSIvan Voras if (strncmp(tp->gpt_scheme->name, SCHEME_NAME, sizeof(SCHEME_NAME))) 150452f657cSIvan Voras return; 151452f657cSIvan Voras 152452f657cSIvan Voras part_gpt_entry = (struct g_part_gpt_entry *)pp->private; 153452f657cSIvan Voras snprintf_uuid(label, size, &part_gpt_entry->ent.ent_uuid); 154452f657cSIvan Voras } 155452f657cSIvan Voras 1563ce9ca89SEdward Tomasz Napierala struct g_label_desc g_label_gpt = { 157452f657cSIvan Voras .ld_taste = g_label_gpt_taste, 158795c5f36SXin LI .ld_dirprefix = "gpt/", 1593ce9ca89SEdward Tomasz Napierala .ld_enabled = 1 160452f657cSIvan Voras }; 161452f657cSIvan Voras 1623ce9ca89SEdward Tomasz Napierala struct g_label_desc g_label_gpt_uuid = { 163452f657cSIvan Voras .ld_taste = g_label_gpt_uuid_taste, 164795c5f36SXin LI .ld_dirprefix = "gptid/", 1653ce9ca89SEdward Tomasz Napierala .ld_enabled = 1 166452f657cSIvan Voras }; 1673ce9ca89SEdward Tomasz Napierala 1683ce9ca89SEdward Tomasz Napierala G_LABEL_INIT(gpt, g_label_gpt, "Create device nodes for GPT labels"); 1693ce9ca89SEdward Tomasz Napierala G_LABEL_INIT(gptid, g_label_gpt_uuid, "Create device nodes for GPT UUIDs"); 170