1452f657cSIvan Voras /*-
24d846d26SWarner Losh * SPDX-License-Identifier: BSD-2-Clause
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/param.h>
30452f657cSIvan Voras #include <sys/systm.h>
31452f657cSIvan Voras #include <sys/kernel.h>
32452f657cSIvan Voras #include <sys/malloc.h>
33452f657cSIvan Voras #include <sys/kobj.h>
34452f657cSIvan Voras #include <sys/gpt.h>
355d807a0eSAndrey V. Elsukov #include <sys/sbuf.h>
36452f657cSIvan Voras
37452f657cSIvan Voras #include <geom/geom.h>
38452f657cSIvan Voras #include <geom/label/g_label.h>
39452f657cSIvan Voras #include <geom/part/g_part.h>
40452f657cSIvan Voras
41452f657cSIvan Voras #define PART_CLASS_NAME "PART"
42452f657cSIvan Voras #define SCHEME_NAME "GPT"
43452f657cSIvan Voras
4463f4d880SIvan Voras /* XXX: Also defined in geom/part/g_part_gpt.c */
45452f657cSIvan Voras struct g_part_gpt_entry {
46452f657cSIvan Voras struct g_part_entry base;
47452f657cSIvan Voras struct gpt_ent ent;
48452f657cSIvan Voras };
49452f657cSIvan Voras
5063f4d880SIvan Voras /* XXX: Shamelessly stolen from g_part_gpt.c */
51452f657cSIvan Voras static void
sbuf_nprintf_utf16(struct sbuf * sb,uint16_t * str,size_t len)52452f657cSIvan Voras sbuf_nprintf_utf16(struct sbuf *sb, uint16_t *str, size_t len)
53452f657cSIvan Voras {
54452f657cSIvan Voras u_int bo;
55452f657cSIvan Voras uint32_t ch;
56452f657cSIvan Voras uint16_t c;
57452f657cSIvan Voras
58452f657cSIvan Voras bo = LITTLE_ENDIAN; /* GPT is little-endian */
59452f657cSIvan Voras while (len > 0 && *str != 0) {
60452f657cSIvan Voras ch = (bo == BIG_ENDIAN) ? be16toh(*str) : le16toh(*str);
61452f657cSIvan Voras str++, len--;
62452f657cSIvan Voras if ((ch & 0xf800) == 0xd800) {
63452f657cSIvan Voras if (len > 0) {
64452f657cSIvan Voras c = (bo == BIG_ENDIAN) ? be16toh(*str)
65452f657cSIvan Voras : le16toh(*str);
66452f657cSIvan Voras str++, len--;
67452f657cSIvan Voras } else
68452f657cSIvan Voras c = 0xfffd;
69452f657cSIvan Voras if ((ch & 0x400) == 0 && (c & 0xfc00) == 0xdc00) {
70452f657cSIvan Voras ch = ((ch & 0x3ff) << 10) + (c & 0x3ff);
71452f657cSIvan Voras ch += 0x10000;
72452f657cSIvan Voras } else
73452f657cSIvan Voras ch = 0xfffd;
74452f657cSIvan Voras } else if (ch == 0xfffe) { /* BOM (U+FEFF) swapped. */
75452f657cSIvan Voras bo = (bo == BIG_ENDIAN) ? LITTLE_ENDIAN : BIG_ENDIAN;
76452f657cSIvan Voras continue;
77452f657cSIvan Voras } else if (ch == 0xfeff) /* BOM (U+FEFF) unswapped. */
78452f657cSIvan Voras continue;
79452f657cSIvan Voras
80452f657cSIvan Voras /* Write the Unicode character in UTF-8 */
81452f657cSIvan Voras if (ch < 0x80)
82452f657cSIvan Voras sbuf_printf(sb, "%c", ch);
83452f657cSIvan Voras else if (ch < 0x800)
84452f657cSIvan Voras sbuf_printf(sb, "%c%c", 0xc0 | (ch >> 6),
85452f657cSIvan Voras 0x80 | (ch & 0x3f));
86452f657cSIvan Voras else if (ch < 0x10000)
87452f657cSIvan Voras sbuf_printf(sb, "%c%c%c", 0xe0 | (ch >> 12),
88452f657cSIvan Voras 0x80 | ((ch >> 6) & 0x3f), 0x80 | (ch & 0x3f));
89452f657cSIvan Voras else if (ch < 0x200000)
90452f657cSIvan Voras sbuf_printf(sb, "%c%c%c%c", 0xf0 | (ch >> 18),
91452f657cSIvan Voras 0x80 | ((ch >> 12) & 0x3f),
92452f657cSIvan Voras 0x80 | ((ch >> 6) & 0x3f), 0x80 | (ch & 0x3f));
93452f657cSIvan Voras }
94452f657cSIvan Voras }
95452f657cSIvan Voras
96452f657cSIvan Voras static void
g_label_gpt_taste(struct g_consumer * cp,char * label,size_t size)97452f657cSIvan Voras g_label_gpt_taste(struct g_consumer *cp, char *label, size_t size)
98452f657cSIvan Voras {
99452f657cSIvan Voras struct g_provider *pp;
100452f657cSIvan Voras struct g_part_table *tp;
101452f657cSIvan Voras struct g_part_gpt_entry *part_gpt_entry;
102452f657cSIvan Voras struct sbuf *lbl;
103452f657cSIvan Voras
104452f657cSIvan Voras g_topology_assert_not();
105452f657cSIvan Voras pp = cp->provider;
106452f657cSIvan Voras tp = (struct g_part_table *)pp->geom->softc;
107452f657cSIvan Voras label[0] = '\0';
108452f657cSIvan Voras
10963f4d880SIvan Voras /* We taste only partitions handled by GPART */
110452f657cSIvan Voras if (strncmp(pp->geom->class->name, PART_CLASS_NAME, sizeof(PART_CLASS_NAME)))
111452f657cSIvan Voras return;
112452f657cSIvan Voras /* and only GPT */
113452f657cSIvan Voras if (strncmp(tp->gpt_scheme->name, SCHEME_NAME, sizeof(SCHEME_NAME)))
114452f657cSIvan Voras return;
115452f657cSIvan Voras
116452f657cSIvan Voras part_gpt_entry = (struct g_part_gpt_entry *)pp->private;
117452f657cSIvan Voras
118452f657cSIvan Voras /*
11963f4d880SIvan Voras * Create sbuf with biggest possible size.
12063f4d880SIvan Voras * We need max. 4 bytes for every 2-byte utf16 char.
121452f657cSIvan Voras */
122452f657cSIvan Voras lbl = sbuf_new(NULL, NULL, sizeof(part_gpt_entry->ent.ent_name) << 1, SBUF_FIXEDLEN);
12363f4d880SIvan Voras /* Size is the number of characters, not bytes */
124452f657cSIvan Voras sbuf_nprintf_utf16(lbl, part_gpt_entry->ent.ent_name, sizeof(part_gpt_entry->ent.ent_name) >> 1);
125452f657cSIvan Voras sbuf_finish(lbl);
126452f657cSIvan Voras strlcpy(label, sbuf_data(lbl), size);
127452f657cSIvan Voras sbuf_delete(lbl);
128452f657cSIvan Voras }
129452f657cSIvan Voras
130452f657cSIvan Voras static void
g_label_gpt_uuid_taste(struct g_consumer * cp,char * label,size_t size)131452f657cSIvan Voras g_label_gpt_uuid_taste(struct g_consumer *cp, char *label, size_t size)
132452f657cSIvan Voras {
133452f657cSIvan Voras struct g_provider *pp;
134452f657cSIvan Voras struct g_part_table *tp;
135452f657cSIvan Voras struct g_part_gpt_entry *part_gpt_entry;
136452f657cSIvan Voras
137452f657cSIvan Voras g_topology_assert_not();
138452f657cSIvan Voras pp = cp->provider;
139452f657cSIvan Voras tp = (struct g_part_table *)pp->geom->softc;
140452f657cSIvan Voras label[0] = '\0';
141452f657cSIvan Voras
14263f4d880SIvan Voras /* We taste only partitions handled by GPART */
143452f657cSIvan Voras if (strncmp(pp->geom->class->name, PART_CLASS_NAME, sizeof(PART_CLASS_NAME)))
144452f657cSIvan Voras return;
145452f657cSIvan Voras /* and only GPT */
146452f657cSIvan Voras if (strncmp(tp->gpt_scheme->name, SCHEME_NAME, sizeof(SCHEME_NAME)))
147452f657cSIvan Voras return;
148452f657cSIvan Voras
149452f657cSIvan Voras part_gpt_entry = (struct g_part_gpt_entry *)pp->private;
150452f657cSIvan Voras snprintf_uuid(label, size, &part_gpt_entry->ent.ent_uuid);
151452f657cSIvan Voras }
152452f657cSIvan Voras
1533ce9ca89SEdward Tomasz Napierala struct g_label_desc g_label_gpt = {
154452f657cSIvan Voras .ld_taste = g_label_gpt_taste,
155795c5f36SXin LI .ld_dirprefix = "gpt/",
1563ce9ca89SEdward Tomasz Napierala .ld_enabled = 1
157452f657cSIvan Voras };
158452f657cSIvan Voras
1593ce9ca89SEdward Tomasz Napierala struct g_label_desc g_label_gpt_uuid = {
160452f657cSIvan Voras .ld_taste = g_label_gpt_uuid_taste,
161795c5f36SXin LI .ld_dirprefix = "gptid/",
1623ce9ca89SEdward Tomasz Napierala .ld_enabled = 1
163452f657cSIvan Voras };
1643ce9ca89SEdward Tomasz Napierala
1653ce9ca89SEdward Tomasz Napierala G_LABEL_INIT(gpt, g_label_gpt, "Create device nodes for GPT labels");
1663ce9ca89SEdward Tomasz Napierala G_LABEL_INIT(gptid, g_label_gpt_uuid, "Create device nodes for GPT UUIDs");
167