xref: /illumos-gate/usr/src/common/ctf/ctf_create.c (revision a61ed2ce7a86a4d6428f2a83eb4739fae945447e)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  */
22 
23 /*
24  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
25  * Use is subject to license terms.
26  */
27 /*
28  * Copyright 2020 Joyent, Inc.
29  */
30 
31 #include <sys/sysmacros.h>
32 #include <sys/param.h>
33 #include <sys/mman.h>
34 #include <ctf_impl.h>
35 #include <sys/debug.h>
36 
37 /*
38  * SSIZE_MAX is not available in the kernel, so we define it here rather than
39  * accidentally inject into headers where it's not wanted.
40  */
41 #ifndef SSIZE_MAX
42 #define	SSIZE_MAX (LONG_MAX)
43 #endif
44 
45 /*
46  * This static string is used as the template for initially populating a
47  * dynamic container's string table.  We always store \0 in the first byte,
48  * and we use the generic string "PARENT" to mark this container's parent
49  * if one is associated with the container using ctf_import().
50  */
51 static const char _CTF_STRTAB_TEMPLATE[] = "\0PARENT";
52 
53 /*
54  * To create an empty CTF container, we just declare a zeroed header and call
55  * ctf_bufopen() on it.  If ctf_bufopen succeeds, we mark the new container r/w
56  * and initialize the dynamic members.  We set dtstrlen to 1 to reserve the
57  * first byte of the string table for a \0 byte, and we start assigning type
58  * IDs at 1 because type ID 0 is used as a sentinel.
59  */
60 ctf_file_t *
61 ctf_create(int *errp)
62 {
63 	static const ctf_header_t hdr = { { CTF_MAGIC, CTF_VERSION, 0 } };
64 
65 	const ulong_t hashlen = 128;
66 	ctf_dtdef_t **hash = ctf_alloc(hashlen * sizeof (ctf_dtdef_t *));
67 	ctf_sect_t cts;
68 	ctf_file_t *fp;
69 
70 	if (hash == NULL)
71 		return (ctf_set_open_errno(errp, EAGAIN));
72 
73 	cts.cts_name = _CTF_SECTION;
74 	cts.cts_type = SHT_PROGBITS;
75 	cts.cts_flags = 0;
76 	cts.cts_data = &hdr;
77 	cts.cts_size = sizeof (hdr);
78 	cts.cts_entsize = 1;
79 	cts.cts_offset = 0;
80 
81 	if ((fp = ctf_bufopen(&cts, NULL, NULL, errp)) == NULL) {
82 		ctf_free(hash, hashlen * sizeof (ctf_dtdef_t *));
83 		return (NULL);
84 	}
85 
86 	fp->ctf_flags |= LCTF_RDWR;
87 	fp->ctf_dthashlen = hashlen;
88 	bzero(hash, hashlen * sizeof (ctf_dtdef_t *));
89 	fp->ctf_dthash = hash;
90 	fp->ctf_dtstrlen = sizeof (_CTF_STRTAB_TEMPLATE);
91 	fp->ctf_dtnextid = 1;
92 	fp->ctf_dtoldid = 0;
93 
94 	return (fp);
95 }
96 
97 ctf_file_t *
98 ctf_fdcreate(int fd, int *errp)
99 {
100 	ctf_file_t *fp;
101 	static const ctf_header_t hdr = { { CTF_MAGIC, CTF_VERSION, 0 } };
102 
103 	const ulong_t hashlen = 128;
104 	ctf_dtdef_t **hash;
105 	ctf_sect_t cts;
106 
107 	if (fd == -1)
108 		return (ctf_create(errp));
109 
110 	hash = ctf_alloc(hashlen * sizeof (ctf_dtdef_t *));
111 
112 	if (hash == NULL)
113 		return (ctf_set_open_errno(errp, EAGAIN));
114 
115 	cts.cts_name = _CTF_SECTION;
116 	cts.cts_type = SHT_PROGBITS;
117 	cts.cts_flags = 0;
118 	cts.cts_data = &hdr;
119 	cts.cts_size = sizeof (hdr);
120 	cts.cts_entsize = 1;
121 	cts.cts_offset = 0;
122 
123 	if ((fp = ctf_fdcreate_int(fd, errp, &cts)) == NULL) {
124 		ctf_free(hash, hashlen * sizeof (ctf_dtdef_t *));
125 		return (NULL);
126 	}
127 
128 	fp->ctf_flags |= LCTF_RDWR;
129 	fp->ctf_dthashlen = hashlen;
130 	bzero(hash, hashlen * sizeof (ctf_dtdef_t *));
131 	fp->ctf_dthash = hash;
132 	fp->ctf_dtstrlen = sizeof (_CTF_STRTAB_TEMPLATE);
133 	fp->ctf_dtnextid = 1;
134 	fp->ctf_dtoldid = 0;
135 
136 	return (fp);
137 }
138 
139 static uchar_t *
140 ctf_copy_smembers(ctf_dtdef_t *dtd, uint_t soff, uchar_t *t)
141 {
142 	ctf_dmdef_t *dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
143 	ctf_member_t ctm;
144 
145 	for (; dmd != NULL; dmd = ctf_list_next(dmd)) {
146 		if (dmd->dmd_name) {
147 			ctm.ctm_name = soff;
148 			soff += strlen(dmd->dmd_name) + 1;
149 		} else
150 			ctm.ctm_name = 0;
151 
152 		ctm.ctm_type = (ushort_t)dmd->dmd_type;
153 		ctm.ctm_offset = (ushort_t)dmd->dmd_offset;
154 
155 		bcopy(&ctm, t, sizeof (ctm));
156 		t += sizeof (ctm);
157 	}
158 
159 	return (t);
160 }
161 
162 static uchar_t *
163 ctf_copy_lmembers(ctf_dtdef_t *dtd, uint_t soff, uchar_t *t)
164 {
165 	ctf_dmdef_t *dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
166 	ctf_lmember_t ctlm;
167 
168 	for (; dmd != NULL; dmd = ctf_list_next(dmd)) {
169 		if (dmd->dmd_name) {
170 			ctlm.ctlm_name = soff;
171 			soff += strlen(dmd->dmd_name) + 1;
172 		} else
173 			ctlm.ctlm_name = 0;
174 
175 		ctlm.ctlm_type = (ushort_t)dmd->dmd_type;
176 		ctlm.ctlm_pad = 0;
177 		ctlm.ctlm_offsethi = CTF_OFFSET_TO_LMEMHI(dmd->dmd_offset);
178 		ctlm.ctlm_offsetlo = CTF_OFFSET_TO_LMEMLO(dmd->dmd_offset);
179 
180 		bcopy(&ctlm, t, sizeof (ctlm));
181 		t += sizeof (ctlm);
182 	}
183 
184 	return (t);
185 }
186 
187 static uchar_t *
188 ctf_copy_emembers(ctf_dtdef_t *dtd, uint_t soff, uchar_t *t)
189 {
190 	ctf_dmdef_t *dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
191 	ctf_enum_t cte;
192 
193 	for (; dmd != NULL; dmd = ctf_list_next(dmd)) {
194 		cte.cte_name = soff;
195 		cte.cte_value = dmd->dmd_value;
196 		soff += strlen(dmd->dmd_name) + 1;
197 		bcopy(&cte, t, sizeof (cte));
198 		t += sizeof (cte);
199 	}
200 
201 	return (t);
202 }
203 
204 static uchar_t *
205 ctf_copy_membnames(ctf_dtdef_t *dtd, uchar_t *s)
206 {
207 	ctf_dmdef_t *dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
208 	size_t len;
209 
210 	for (; dmd != NULL; dmd = ctf_list_next(dmd)) {
211 		if (dmd->dmd_name == NULL)
212 			continue; /* skip anonymous members */
213 		len = strlen(dmd->dmd_name) + 1;
214 		bcopy(dmd->dmd_name, s, len);
215 		s += len;
216 	}
217 
218 	return (s);
219 }
220 
221 /*
222  * Only types of dyanmic CTF containers contain reference counts. These
223  * containers are marked RD/WR. Because of that we basically make this a no-op
224  * for compatability with non-dynamic CTF sections. This is also a no-op for
225  * types which are not dynamic types. It is the responsibility of the caller to
226  * make sure it is a valid type. We help that caller out on debug builds.
227  *
228  * Note that the reference counts are not maintained for types that are not
229  * within this container. In other words if we have a type in a parent, that
230  * will not have its reference count increased. On the flip side, the parent
231  * will not be allowed to remove dynamic types if it has children.
232  */
233 static void
234 ctf_ref_inc(ctf_file_t *fp, ctf_id_t tid)
235 {
236 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, tid);
237 
238 	if (dtd == NULL)
239 		return;
240 
241 	if (!(fp->ctf_flags & LCTF_RDWR))
242 		return;
243 
244 	dtd->dtd_ref++;
245 }
246 
247 /*
248  * Just as with ctf_ref_inc, this is a no-op on non-writeable containers and the
249  * caller should ensure that this is already a valid type.
250  */
251 static void
252 ctf_ref_dec(ctf_file_t *fp, ctf_id_t tid)
253 {
254 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, tid);
255 
256 	if (dtd == NULL)
257 		return;
258 
259 	if (!(fp->ctf_flags & LCTF_RDWR))
260 		return;
261 
262 	ASSERT(dtd->dtd_ref >= 1);
263 	dtd->dtd_ref--;
264 }
265 
266 /*
267  * If the specified CTF container is writable and has been modified, reload
268  * this container with the updated type definitions.  In order to make this
269  * code and the rest of libctf as simple as possible, we perform updates by
270  * taking the dynamic type definitions and creating an in-memory CTF file
271  * containing the definitions, and then call ctf_bufopen() on it.  This not
272  * only leverages ctf_bufopen(), but also avoids having to bifurcate the rest
273  * of the library code with different lookup paths for static and dynamic
274  * type definitions.  We are therefore optimizing greatly for lookup over
275  * update, which we assume will be an uncommon operation.  We perform one
276  * extra trick here for the benefit of callers and to keep our code simple:
277  * ctf_bufopen() will return a new ctf_file_t, but we want to keep the fp
278  * constant for the caller, so after ctf_bufopen() returns, we use bcopy to
279  * swap the interior of the old and new ctf_file_t's, and then free the old.
280  *
281  * Note that the lists of dynamic types stays around and the resulting container
282  * is still writeable. Furthermore, the reference counts that are on the dtd's
283  * are still valid.
284  */
285 int
286 ctf_update(ctf_file_t *fp)
287 {
288 	ctf_file_t ofp, *nfp;
289 	ctf_header_t hdr, *bhdr;
290 	ctf_dtdef_t *dtd;
291 	ctf_dsdef_t *dsd;
292 	ctf_dldef_t *dld;
293 	ctf_sect_t cts, *symp, *strp;
294 
295 	uchar_t *s, *s0, *t;
296 	ctf_lblent_t *label;
297 	uint16_t *obj, *func;
298 	size_t size, objsize, funcsize, labelsize, plen;
299 	void *buf;
300 	int err;
301 	ulong_t i;
302 	const char *plabel;
303 	const char *sname;
304 
305 	uintptr_t symbase = (uintptr_t)fp->ctf_symtab.cts_data;
306 	uintptr_t strbase = (uintptr_t)fp->ctf_strtab.cts_data;
307 
308 	if (!(fp->ctf_flags & LCTF_RDWR))
309 		return (ctf_set_errno(fp, ECTF_RDONLY));
310 
311 	if (!(fp->ctf_flags & LCTF_DIRTY))
312 		return (0); /* no update required */
313 
314 	/*
315 	 * Fill in an initial CTF header.  We will leave the label, object,
316 	 * and function sections empty and only output a header, type section,
317 	 * and string table.  The type section begins at a 4-byte aligned
318 	 * boundary past the CTF header itself (at relative offset zero).
319 	 */
320 	bzero(&hdr, sizeof (hdr));
321 	hdr.cth_magic = CTF_MAGIC;
322 	hdr.cth_version = CTF_VERSION;
323 
324 	if (fp->ctf_flags & LCTF_CHILD) {
325 		if (fp->ctf_parname == NULL) {
326 			plen = 0;
327 			hdr.cth_parname = 1; /* i.e. _CTF_STRTAB_TEMPLATE[1] */
328 			plabel = NULL;
329 		} else {
330 			plen = strlen(fp->ctf_parname) + 1;
331 			plabel = ctf_label_topmost(fp->ctf_parent);
332 		}
333 	} else {
334 		plabel = NULL;
335 		plen = 0;
336 	}
337 
338 	/*
339 	 * Iterate over the labels that we have.
340 	 */
341 	for (labelsize = 0, dld = ctf_list_next(&fp->ctf_dldefs);
342 	    dld != NULL; dld = ctf_list_next(dld))
343 		labelsize += sizeof (ctf_lblent_t);
344 
345 	/*
346 	 * Iterate through the dynamic type definition list and compute the
347 	 * size of the CTF type section we will need to generate.
348 	 */
349 	for (size = 0, dtd = ctf_list_next(&fp->ctf_dtdefs);
350 	    dtd != NULL; dtd = ctf_list_next(dtd)) {
351 
352 		uint_t kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
353 		uint_t vlen = CTF_INFO_VLEN(dtd->dtd_data.ctt_info);
354 
355 		if (dtd->dtd_data.ctt_size != CTF_LSIZE_SENT)
356 			size += sizeof (ctf_stype_t);
357 		else
358 			size += sizeof (ctf_type_t);
359 
360 		switch (kind) {
361 		case CTF_K_INTEGER:
362 		case CTF_K_FLOAT:
363 			size += sizeof (uint_t);
364 			break;
365 		case CTF_K_ARRAY:
366 			size += sizeof (ctf_array_t);
367 			break;
368 		case CTF_K_FUNCTION:
369 			size += sizeof (ushort_t) * (vlen + (vlen & 1));
370 			break;
371 		case CTF_K_STRUCT:
372 		case CTF_K_UNION:
373 			if (dtd->dtd_data.ctt_size < CTF_LSTRUCT_THRESH)
374 				size += sizeof (ctf_member_t) * vlen;
375 			else
376 				size += sizeof (ctf_lmember_t) * vlen;
377 			break;
378 		case CTF_K_ENUM:
379 			size += sizeof (ctf_enum_t) * vlen;
380 			break;
381 		}
382 	}
383 
384 	/*
385 	 * An entry for each object must exist in the data section. However, if
386 	 * the symbol is SHN_UNDEF, then it is skipped. For objects, the storage
387 	 * is just the size of the 2-byte id. For functions it's always 2 bytes,
388 	 * plus 2 bytes per argument and the return type.
389 	 */
390 	dsd = ctf_list_next(&fp->ctf_dsdefs);
391 	for (objsize = 0, funcsize = 0, i = 0; i < fp->ctf_nsyms; i++) {
392 		int type;
393 
394 		if (fp->ctf_symtab.cts_entsize == sizeof (Elf32_Sym)) {
395 			const Elf32_Sym *symp = (Elf32_Sym *)symbase + i;
396 
397 			type = ELF32_ST_TYPE(symp->st_info);
398 			if (ctf_sym_valid(strbase, type, symp->st_shndx,
399 			    symp->st_value, symp->st_name) == B_FALSE)
400 				continue;
401 		} else {
402 			const Elf64_Sym *symp = (Elf64_Sym *)symbase + i;
403 
404 			type = ELF64_ST_TYPE(symp->st_info);
405 			if (ctf_sym_valid(strbase, type, symp->st_shndx,
406 			    symp->st_value, symp->st_name) == B_FALSE)
407 				continue;
408 		}
409 
410 		while (dsd != NULL && i > dsd->dsd_symidx)
411 			dsd = ctf_list_next(dsd);
412 		if (type == STT_OBJECT) {
413 			objsize += sizeof (uint16_t);
414 		} else {
415 			/* Every function has a uint16_t info no matter what */
416 			if (dsd == NULL || i < dsd->dsd_symidx) {
417 				funcsize += sizeof (uint16_t);
418 			} else {
419 				funcsize += sizeof (uint16_t) *
420 				    (dsd->dsd_nargs + 2);
421 			}
422 		}
423 	}
424 
425 	/*
426 	 * The objtoff and funcoffset must be 2-byte aligned. We're guaranteed
427 	 * that this is always true for the objtoff because labels are always 8
428 	 * bytes large. Similarly, because objects are always two bytes of data,
429 	 * this will always be true for funcoff.
430 	 */
431 	hdr.cth_objtoff = hdr.cth_lbloff + labelsize;
432 	hdr.cth_funcoff = hdr.cth_objtoff + objsize;
433 
434 	/*
435 	 * The type offset must be 4 byte aligned.
436 	 */
437 	hdr.cth_typeoff = hdr.cth_funcoff + funcsize;
438 	if (hdr.cth_typeoff & 3)
439 		hdr.cth_typeoff += 4 - (hdr.cth_typeoff & 3);
440 	ASSERT((hdr.cth_typeoff & 3) == 0);
441 
442 	/*
443 	 * Fill in the string table offset and size, compute the size of the
444 	 * entire CTF buffer we need, and then allocate a new buffer and
445 	 * bcopy the finished header to the start of the buffer.
446 	 */
447 	hdr.cth_stroff = hdr.cth_typeoff + size;
448 	hdr.cth_strlen = fp->ctf_dtstrlen + plen;
449 	size = sizeof (ctf_header_t) + hdr.cth_stroff + hdr.cth_strlen;
450 	ctf_dprintf("lbloff: %u\nobjtoff: %u\nfuncoff: %u\n"
451 	    "typeoff: %u\nstroff: %u\nstrlen: %u\n",
452 	    hdr.cth_lbloff, hdr.cth_objtoff, hdr.cth_funcoff,
453 	    hdr.cth_typeoff, hdr.cth_stroff, hdr.cth_strlen);
454 
455 	if ((buf = ctf_data_alloc(size)) == MAP_FAILED)
456 		return (ctf_set_errno(fp, EAGAIN));
457 
458 	bcopy(&hdr, buf, sizeof (ctf_header_t));
459 	bhdr = buf;
460 	label = (ctf_lblent_t *)((uintptr_t)buf + sizeof (ctf_header_t));
461 	t = (uchar_t *)buf + sizeof (ctf_header_t) + hdr.cth_typeoff;
462 	s = s0 = (uchar_t *)buf + sizeof (ctf_header_t) + hdr.cth_stroff;
463 	obj = (uint16_t *)((uintptr_t)buf + sizeof (ctf_header_t) +
464 	    hdr.cth_objtoff);
465 	func = (uint16_t *)((uintptr_t)buf + sizeof (ctf_header_t) +
466 	    hdr.cth_funcoff);
467 
468 	bcopy(_CTF_STRTAB_TEMPLATE, s, sizeof (_CTF_STRTAB_TEMPLATE));
469 	s += sizeof (_CTF_STRTAB_TEMPLATE);
470 
471 	/*
472 	 * We have an actual parent name and we're a child container, therefore
473 	 * we should make sure to note our parent's name here.
474 	 */
475 	if (plen != 0) {
476 		VERIFY(s + plen - s0 <= hdr.cth_strlen);
477 		bcopy(fp->ctf_parname, s, plen);
478 		bhdr->cth_parname = s - s0;
479 		s += plen;
480 	}
481 
482 	/*
483 	 * First pass over the labels and copy them out.
484 	 */
485 	for (dld = ctf_list_next(&fp->ctf_dldefs); dld != NULL;
486 	    dld = ctf_list_next(dld), label++) {
487 		size_t len = strlen(dld->dld_name) + 1;
488 
489 		VERIFY(s + len - s0 <= hdr.cth_strlen);
490 		bcopy(dld->dld_name, s, len);
491 		label->ctl_typeidx = dld->dld_type;
492 		label->ctl_label = s - s0;
493 		s += len;
494 
495 		if (plabel != NULL && strcmp(plabel, dld->dld_name) == 0)
496 			bhdr->cth_parlabel = label->ctl_label;
497 	}
498 
499 	/*
500 	 * We now take a final lap through the dynamic type definition list and
501 	 * copy the appropriate type records and strings to the output buffer.
502 	 */
503 	for (dtd = ctf_list_next(&fp->ctf_dtdefs);
504 	    dtd != NULL; dtd = ctf_list_next(dtd)) {
505 
506 		uint_t kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
507 		uint_t vlen = CTF_INFO_VLEN(dtd->dtd_data.ctt_info);
508 
509 		ctf_array_t cta;
510 		uint_t encoding;
511 		size_t len;
512 
513 		if (dtd->dtd_name != NULL) {
514 			dtd->dtd_data.ctt_name = (uint_t)(s - s0);
515 			len = strlen(dtd->dtd_name) + 1;
516 			VERIFY(s + len - s0 <= hdr.cth_strlen);
517 			bcopy(dtd->dtd_name, s, len);
518 			s += len;
519 		} else
520 			dtd->dtd_data.ctt_name = 0;
521 
522 		if (dtd->dtd_data.ctt_size != CTF_LSIZE_SENT)
523 			len = sizeof (ctf_stype_t);
524 		else
525 			len = sizeof (ctf_type_t);
526 
527 		bcopy(&dtd->dtd_data, t, len);
528 		t += len;
529 
530 		switch (kind) {
531 		case CTF_K_INTEGER:
532 		case CTF_K_FLOAT:
533 			if (kind == CTF_K_INTEGER) {
534 				encoding = CTF_INT_DATA(
535 				    dtd->dtd_u.dtu_enc.cte_format,
536 				    dtd->dtd_u.dtu_enc.cte_offset,
537 				    dtd->dtd_u.dtu_enc.cte_bits);
538 			} else {
539 				encoding = CTF_FP_DATA(
540 				    dtd->dtd_u.dtu_enc.cte_format,
541 				    dtd->dtd_u.dtu_enc.cte_offset,
542 				    dtd->dtd_u.dtu_enc.cte_bits);
543 			}
544 			bcopy(&encoding, t, sizeof (encoding));
545 			t += sizeof (encoding);
546 			break;
547 
548 		case CTF_K_ARRAY:
549 			cta.cta_contents = (ushort_t)
550 			    dtd->dtd_u.dtu_arr.ctr_contents;
551 			cta.cta_index = (ushort_t)
552 			    dtd->dtd_u.dtu_arr.ctr_index;
553 			cta.cta_nelems = dtd->dtd_u.dtu_arr.ctr_nelems;
554 			bcopy(&cta, t, sizeof (cta));
555 			t += sizeof (cta);
556 			break;
557 
558 		case CTF_K_FUNCTION: {
559 			ushort_t *argv = (ushort_t *)(uintptr_t)t;
560 			uint_t argc;
561 
562 			for (argc = 0; argc < vlen; argc++)
563 				*argv++ = (ushort_t)dtd->dtd_u.dtu_argv[argc];
564 
565 			if (vlen & 1)
566 				*argv++ = 0; /* pad to 4-byte boundary */
567 
568 			t = (uchar_t *)argv;
569 			break;
570 		}
571 
572 		case CTF_K_STRUCT:
573 		case CTF_K_UNION:
574 			if (dtd->dtd_data.ctt_size < CTF_LSTRUCT_THRESH)
575 				t = ctf_copy_smembers(dtd, (uint_t)(s - s0), t);
576 			else
577 				t = ctf_copy_lmembers(dtd, (uint_t)(s - s0), t);
578 			s = ctf_copy_membnames(dtd, s);
579 			break;
580 
581 		case CTF_K_ENUM:
582 			t = ctf_copy_emembers(dtd, (uint_t)(s - s0), t);
583 			s = ctf_copy_membnames(dtd, s);
584 			break;
585 		}
586 	}
587 
588 	/*
589 	 * Now we fill in our dynamic data and function sections. We use the
590 	 * same criteria as above, but also consult the dsd list.
591 	 */
592 	dsd = ctf_list_next(&fp->ctf_dsdefs);
593 	for (i = 0; i < fp->ctf_nsyms; i++) {
594 		int type;
595 		if (fp->ctf_symtab.cts_entsize == sizeof (Elf32_Sym)) {
596 			const Elf32_Sym *symp = (Elf32_Sym *)symbase + i;
597 			type = ELF32_ST_TYPE(symp->st_info);
598 
599 			if (ctf_sym_valid(strbase, type, symp->st_shndx,
600 			    symp->st_value, symp->st_name) == B_FALSE)
601 				continue;
602 		} else {
603 			const Elf64_Sym *symp = (Elf64_Sym *)symbase + i;
604 			type = ELF64_ST_TYPE(symp->st_info);
605 			if (ctf_sym_valid(strbase, type, symp->st_shndx,
606 			    symp->st_value, symp->st_name) == B_FALSE)
607 				continue;
608 		}
609 
610 		while (dsd != NULL && i > dsd->dsd_symidx) {
611 			dsd = ctf_list_next(dsd);
612 		}
613 		if (type == STT_OBJECT) {
614 			if (dsd == NULL || i < dsd->dsd_symidx) {
615 				*obj = 0;
616 			} else {
617 				*obj = dsd->dsd_tid;
618 			}
619 			obj++;
620 			VERIFY((uintptr_t)obj <= (uintptr_t)func);
621 		} else {
622 			if (dsd == NULL || i < dsd->dsd_symidx) {
623 				ushort_t data = CTF_TYPE_INFO(CTF_K_UNKNOWN,
624 				    0, 0);
625 				*func = data;
626 				func++;
627 			} else {
628 				int j;
629 				ushort_t data = CTF_TYPE_INFO(CTF_K_FUNCTION, 0,
630 				    dsd->dsd_nargs);
631 
632 				*func = data;
633 				func++;
634 				*func = dsd->dsd_tid;
635 				func++;
636 				for (j = 0; j < dsd->dsd_nargs; j++)
637 					func[j] = dsd->dsd_argc[j];
638 				func += dsd->dsd_nargs;
639 			}
640 		}
641 	}
642 
643 	/*
644 	 * Finally, we are ready to ctf_bufopen() the new container.  If this
645 	 * is successful, we then switch nfp and fp and free the old container.
646 	 */
647 	ctf_data_protect(buf, size);
648 	cts.cts_name = _CTF_SECTION;
649 	cts.cts_type = SHT_PROGBITS;
650 	cts.cts_flags = 0;
651 	cts.cts_data = buf;
652 	cts.cts_size = size;
653 	cts.cts_entsize = 1;
654 	cts.cts_offset = 0;
655 
656 	if (fp->ctf_nsyms == 0) {
657 		symp = NULL;
658 		strp = NULL;
659 	} else {
660 		symp = &fp->ctf_symtab;
661 		strp = &fp->ctf_strtab;
662 	}
663 
664 	if ((nfp = ctf_bufopen(&cts, symp, strp, &err)) == NULL) {
665 		ctf_data_free(buf, size);
666 		return (ctf_set_errno(fp, err));
667 	}
668 
669 	(void) ctf_setmodel(nfp, ctf_getmodel(fp));
670 	(void) ctf_import(nfp, fp->ctf_parent);
671 
672 	nfp->ctf_refcnt = fp->ctf_refcnt;
673 	nfp->ctf_flags |= fp->ctf_flags & ~LCTF_DIRTY;
674 	nfp->ctf_dthash = fp->ctf_dthash;
675 	nfp->ctf_dthashlen = fp->ctf_dthashlen;
676 	nfp->ctf_dtdefs = fp->ctf_dtdefs;
677 	nfp->ctf_dsdefs = fp->ctf_dsdefs;
678 	nfp->ctf_dldefs = fp->ctf_dldefs;
679 	nfp->ctf_dtstrlen = fp->ctf_dtstrlen;
680 	nfp->ctf_dtnextid = fp->ctf_dtnextid;
681 	nfp->ctf_dtoldid = fp->ctf_dtnextid - 1;
682 	nfp->ctf_specific = fp->ctf_specific;
683 
684 	fp->ctf_dthash = NULL;
685 	fp->ctf_dthashlen = 0;
686 	bzero(&fp->ctf_dtdefs, sizeof (ctf_list_t));
687 	bzero(&fp->ctf_dsdefs, sizeof (ctf_list_t));
688 	bzero(&fp->ctf_dldefs, sizeof (ctf_list_t));
689 
690 	/*
691 	 * Because the various containers share the data sections, we don't want
692 	 * to have ctf_close free it all. However, the name of the section is in
693 	 * fact unique to the ctf_sect_t. Thus we save the names of the symbol
694 	 * and string sections around the bzero() and restore them afterwards,
695 	 * ensuring that we don't result in a memory leak.
696 	 */
697 	sname = fp->ctf_symtab.cts_name;
698 	bzero(&fp->ctf_symtab, sizeof (ctf_sect_t));
699 	fp->ctf_symtab.cts_name = sname;
700 
701 	sname = fp->ctf_strtab.cts_name;
702 	bzero(&fp->ctf_strtab, sizeof (ctf_sect_t));
703 	fp->ctf_strtab.cts_name = sname;
704 
705 	bcopy(fp, &ofp, sizeof (ctf_file_t));
706 	bcopy(nfp, fp, sizeof (ctf_file_t));
707 	bcopy(&ofp, nfp, sizeof (ctf_file_t));
708 
709 	/*
710 	 * Initialize the ctf_lookup_by_name top-level dictionary.  We keep an
711 	 * array of type name prefixes and the corresponding ctf_hash to use.
712 	 * NOTE: This code must be kept in sync with the code in ctf_bufopen().
713 	 */
714 	fp->ctf_lookups[0].ctl_hash = &fp->ctf_structs;
715 	fp->ctf_lookups[1].ctl_hash = &fp->ctf_unions;
716 	fp->ctf_lookups[2].ctl_hash = &fp->ctf_enums;
717 	fp->ctf_lookups[3].ctl_hash = &fp->ctf_names;
718 
719 	nfp->ctf_refcnt = 1; /* force nfp to be freed */
720 	ctf_close(nfp);
721 
722 	return (0);
723 }
724 
725 void
726 ctf_dtd_insert(ctf_file_t *fp, ctf_dtdef_t *dtd)
727 {
728 	ulong_t h = dtd->dtd_type & (fp->ctf_dthashlen - 1);
729 
730 	dtd->dtd_hash = fp->ctf_dthash[h];
731 	fp->ctf_dthash[h] = dtd;
732 	ctf_list_append(&fp->ctf_dtdefs, dtd);
733 }
734 
735 void
736 ctf_dtd_delete(ctf_file_t *fp, ctf_dtdef_t *dtd)
737 {
738 	ulong_t h = dtd->dtd_type & (fp->ctf_dthashlen - 1);
739 	ctf_dtdef_t *p, **q = &fp->ctf_dthash[h];
740 	ctf_dmdef_t *dmd, *nmd;
741 	size_t len;
742 	int kind, i;
743 
744 	for (p = *q; p != NULL; p = p->dtd_hash) {
745 		if (p != dtd)
746 			q = &p->dtd_hash;
747 		else
748 			break;
749 	}
750 
751 	if (p != NULL)
752 		*q = p->dtd_hash;
753 
754 	kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
755 	switch (kind) {
756 	case CTF_K_STRUCT:
757 	case CTF_K_UNION:
758 	case CTF_K_ENUM:
759 		for (dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
760 		    dmd != NULL; dmd = nmd) {
761 			if (dmd->dmd_name != NULL) {
762 				len = strlen(dmd->dmd_name) + 1;
763 				ctf_free(dmd->dmd_name, len);
764 				fp->ctf_dtstrlen -= len;
765 			}
766 			if (kind != CTF_K_ENUM)
767 				ctf_ref_dec(fp, dmd->dmd_type);
768 			nmd = ctf_list_next(dmd);
769 			ctf_free(dmd, sizeof (ctf_dmdef_t));
770 		}
771 		break;
772 	case CTF_K_FUNCTION:
773 		ctf_ref_dec(fp, dtd->dtd_data.ctt_type);
774 		for (i = 0; i < CTF_INFO_VLEN(dtd->dtd_data.ctt_info); i++)
775 			if (dtd->dtd_u.dtu_argv[i] != 0)
776 				ctf_ref_dec(fp, dtd->dtd_u.dtu_argv[i]);
777 		ctf_free(dtd->dtd_u.dtu_argv, sizeof (ctf_id_t) *
778 		    CTF_INFO_VLEN(dtd->dtd_data.ctt_info));
779 		break;
780 	case CTF_K_ARRAY:
781 		ctf_ref_dec(fp, dtd->dtd_u.dtu_arr.ctr_contents);
782 		ctf_ref_dec(fp, dtd->dtd_u.dtu_arr.ctr_index);
783 		break;
784 	case CTF_K_TYPEDEF:
785 		ctf_ref_dec(fp, dtd->dtd_data.ctt_type);
786 		break;
787 	case CTF_K_POINTER:
788 	case CTF_K_VOLATILE:
789 	case CTF_K_CONST:
790 	case CTF_K_RESTRICT:
791 		ctf_ref_dec(fp, dtd->dtd_data.ctt_type);
792 		break;
793 	}
794 
795 	if (dtd->dtd_name) {
796 		len = strlen(dtd->dtd_name) + 1;
797 		ctf_free(dtd->dtd_name, len);
798 		fp->ctf_dtstrlen -= len;
799 	}
800 
801 	ctf_list_delete(&fp->ctf_dtdefs, dtd);
802 	ctf_free(dtd, sizeof (ctf_dtdef_t));
803 }
804 
805 ctf_dtdef_t *
806 ctf_dtd_lookup(ctf_file_t *fp, ctf_id_t type)
807 {
808 	ulong_t h = type & (fp->ctf_dthashlen - 1);
809 	ctf_dtdef_t *dtd;
810 
811 	if (fp->ctf_dthash == NULL)
812 		return (NULL);
813 
814 	for (dtd = fp->ctf_dthash[h]; dtd != NULL; dtd = dtd->dtd_hash) {
815 		if (dtd->dtd_type == type)
816 			break;
817 	}
818 
819 	return (dtd);
820 }
821 
822 ctf_dsdef_t *
823 ctf_dsd_lookup(ctf_file_t *fp, ulong_t idx)
824 {
825 	ctf_dsdef_t *dsd;
826 
827 	for (dsd = ctf_list_next(&fp->ctf_dsdefs); dsd != NULL;
828 	    dsd = ctf_list_next(dsd)) {
829 		if (dsd->dsd_symidx == idx)
830 			return (dsd);
831 	}
832 
833 	return (NULL);
834 }
835 
836 /*
837  * We order the ctf_dsdef_t by symbol index to make things better for updates.
838  */
839 void
840 ctf_dsd_insert(ctf_file_t *fp, ctf_dsdef_t *dsd)
841 {
842 	ctf_dsdef_t *i;
843 
844 	for (i = ctf_list_next(&fp->ctf_dsdefs); i != NULL;
845 	    i = ctf_list_next(i)) {
846 		if (i->dsd_symidx > dsd->dsd_symidx)
847 			break;
848 	}
849 
850 	if (i == NULL) {
851 		ctf_list_append(&fp->ctf_dsdefs, dsd);
852 		return;
853 	}
854 
855 	ctf_list_insert_before(&fp->ctf_dsdefs, i, dsd);
856 }
857 
858 /* ARGSUSED */
859 void
860 ctf_dsd_delete(ctf_file_t *fp, ctf_dsdef_t *dsd)
861 {
862 	if (dsd->dsd_nargs > 0)
863 		ctf_free(dsd->dsd_argc,
864 		    sizeof (ctf_id_t) * dsd->dsd_nargs);
865 	ctf_list_delete(&fp->ctf_dsdefs, dsd);
866 	ctf_free(dsd, sizeof (ctf_dsdef_t));
867 }
868 
869 ctf_dldef_t *
870 ctf_dld_lookup(ctf_file_t *fp, const char *name)
871 {
872 	ctf_dldef_t *dld;
873 
874 	for (dld = ctf_list_next(&fp->ctf_dldefs); dld != NULL;
875 	    dld = ctf_list_next(dld)) {
876 		if (strcmp(name, dld->dld_name) == 0)
877 			return (dld);
878 	}
879 
880 	return (NULL);
881 }
882 
883 void
884 ctf_dld_insert(ctf_file_t *fp, ctf_dldef_t *dld, uint_t pos)
885 {
886 	ctf_dldef_t *l;
887 
888 	if (pos == 0) {
889 		ctf_list_prepend(&fp->ctf_dldefs, dld);
890 		return;
891 	}
892 
893 	for (l = ctf_list_next(&fp->ctf_dldefs); pos != 0 && dld != NULL;
894 	    l = ctf_list_next(l), pos--)
895 		;
896 
897 	if (l == NULL)
898 		ctf_list_append(&fp->ctf_dldefs, dld);
899 	else
900 		ctf_list_insert_before(&fp->ctf_dsdefs, l, dld);
901 }
902 
903 void
904 ctf_dld_delete(ctf_file_t *fp, ctf_dldef_t *dld)
905 {
906 	ctf_list_delete(&fp->ctf_dldefs, dld);
907 
908 	if (dld->dld_name != NULL) {
909 		size_t len = strlen(dld->dld_name) + 1;
910 		ctf_free(dld->dld_name, len);
911 		fp->ctf_dtstrlen -= len;
912 	}
913 
914 	ctf_free(dld, sizeof (ctf_dldef_t));
915 }
916 
917 /*
918  * Discard all of the dynamic type definitions that have been added to the
919  * container since the last call to ctf_update().  We locate such types by
920  * scanning the list and deleting elements that have type IDs greater than
921  * ctf_dtoldid, which is set by ctf_update(), above. Note that to work properly
922  * with our reference counting schemes, we must delete the dynamic list in
923  * reverse.
924  */
925 int
926 ctf_discard(ctf_file_t *fp)
927 {
928 	ctf_dtdef_t *dtd, *ntd;
929 
930 	if (!(fp->ctf_flags & LCTF_RDWR))
931 		return (ctf_set_errno(fp, ECTF_RDONLY));
932 
933 	if (!(fp->ctf_flags & LCTF_DIRTY))
934 		return (0); /* no update required */
935 
936 	for (dtd = ctf_list_prev(&fp->ctf_dtdefs); dtd != NULL; dtd = ntd) {
937 		ntd = ctf_list_prev(dtd);
938 		if (dtd->dtd_type <= fp->ctf_dtoldid)
939 			continue; /* skip types that have been committed */
940 
941 		ctf_dtd_delete(fp, dtd);
942 	}
943 
944 	fp->ctf_dtnextid = fp->ctf_dtoldid + 1;
945 	fp->ctf_flags &= ~LCTF_DIRTY;
946 
947 	return (0);
948 }
949 
950 static ctf_id_t
951 ctf_add_generic(ctf_file_t *fp, uint_t flag, const char *name, ctf_dtdef_t **rp)
952 {
953 	ctf_dtdef_t *dtd;
954 	ctf_id_t type;
955 	char *s = NULL;
956 
957 	if (flag != CTF_ADD_NONROOT && flag != CTF_ADD_ROOT)
958 		return (ctf_set_errno(fp, EINVAL));
959 
960 	if (!(fp->ctf_flags & LCTF_RDWR))
961 		return (ctf_set_errno(fp, ECTF_RDONLY));
962 
963 	if (CTF_INDEX_TO_TYPE(fp->ctf_dtnextid, 1) > CTF_MAX_TYPE)
964 		return (ctf_set_errno(fp, ECTF_FULL));
965 
966 	if ((dtd = ctf_alloc(sizeof (ctf_dtdef_t))) == NULL)
967 		return (ctf_set_errno(fp, EAGAIN));
968 
969 	if (name != NULL && (s = ctf_strdup(name)) == NULL) {
970 		ctf_free(dtd, sizeof (ctf_dtdef_t));
971 		return (ctf_set_errno(fp, EAGAIN));
972 	}
973 
974 	type = fp->ctf_dtnextid++;
975 	type = CTF_INDEX_TO_TYPE(type, (fp->ctf_flags & LCTF_CHILD));
976 
977 	bzero(dtd, sizeof (ctf_dtdef_t));
978 	dtd->dtd_name = s;
979 	dtd->dtd_type = type;
980 
981 	if (s != NULL)
982 		fp->ctf_dtstrlen += strlen(s) + 1;
983 
984 	ctf_dtd_insert(fp, dtd);
985 	fp->ctf_flags |= LCTF_DIRTY;
986 
987 	*rp = dtd;
988 	return (type);
989 }
990 
991 ctf_id_t
992 ctf_add_encoded(ctf_file_t *fp, uint_t flag,
993     const char *name, const ctf_encoding_t *ep, uint_t kind)
994 {
995 	ctf_dtdef_t *dtd;
996 	ctf_id_t type;
997 
998 	if (ep == NULL)
999 		return (ctf_set_errno(fp, EINVAL));
1000 
1001 	if ((type = ctf_add_generic(fp, flag, name, &dtd)) == CTF_ERR)
1002 		return (CTF_ERR); /* errno is set for us */
1003 
1004 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(kind, flag, 0);
1005 
1006 	/*
1007 	 * If the type's size is not an even number of bytes, then we should
1008 	 * round up the type size to the nearest byte.
1009 	 */
1010 	dtd->dtd_data.ctt_size = ep->cte_bits / NBBY;
1011 	if ((ep->cte_bits % NBBY) != 0)
1012 		dtd->dtd_data.ctt_size++;
1013 	dtd->dtd_u.dtu_enc = *ep;
1014 
1015 	return (type);
1016 }
1017 
1018 ctf_id_t
1019 ctf_add_reftype(ctf_file_t *fp, uint_t flag,
1020     const char *name, ctf_id_t ref, uint_t kind)
1021 {
1022 	ctf_dtdef_t *dtd;
1023 	ctf_id_t type;
1024 
1025 	if (ref == CTF_ERR || ref < 0 || ref > CTF_MAX_TYPE)
1026 		return (ctf_set_errno(fp, EINVAL));
1027 
1028 	if ((type = ctf_add_generic(fp, flag, name, &dtd)) == CTF_ERR)
1029 		return (CTF_ERR); /* errno is set for us */
1030 
1031 	ctf_ref_inc(fp, ref);
1032 
1033 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(kind, flag, 0);
1034 	dtd->dtd_data.ctt_type = (ushort_t)ref;
1035 
1036 	return (type);
1037 }
1038 
1039 ctf_id_t
1040 ctf_add_integer(ctf_file_t *fp, uint_t flag,
1041     const char *name, const ctf_encoding_t *ep)
1042 {
1043 	return (ctf_add_encoded(fp, flag, name, ep, CTF_K_INTEGER));
1044 }
1045 
1046 ctf_id_t
1047 ctf_add_float(ctf_file_t *fp, uint_t flag,
1048     const char *name, const ctf_encoding_t *ep)
1049 {
1050 	return (ctf_add_encoded(fp, flag, name, ep, CTF_K_FLOAT));
1051 }
1052 
1053 ctf_id_t
1054 ctf_add_pointer(ctf_file_t *fp, uint_t flag, const char *name, ctf_id_t ref)
1055 {
1056 	return (ctf_add_reftype(fp, flag, name, ref, CTF_K_POINTER));
1057 }
1058 
1059 ctf_id_t
1060 ctf_add_array(ctf_file_t *fp, uint_t flag, const ctf_arinfo_t *arp)
1061 {
1062 	ctf_dtdef_t *dtd;
1063 	ctf_id_t type;
1064 	ctf_file_t *fpd;
1065 
1066 	if (arp == NULL)
1067 		return (ctf_set_errno(fp, EINVAL));
1068 
1069 	fpd = fp;
1070 	if (ctf_lookup_by_id(&fpd, arp->ctr_contents) == NULL &&
1071 	    ctf_dtd_lookup(fp, arp->ctr_contents) == NULL) {
1072 		ctf_dprintf("bad contents for array: %ld\n",
1073 		    arp->ctr_contents);
1074 		return (ctf_set_errno(fp, ECTF_BADID));
1075 	}
1076 
1077 	fpd = fp;
1078 	if (ctf_lookup_by_id(&fpd, arp->ctr_index) == NULL &&
1079 	    ctf_dtd_lookup(fp, arp->ctr_index) == NULL) {
1080 		ctf_dprintf("bad index for array: %ld\n", arp->ctr_index);
1081 		return (ctf_set_errno(fp, ECTF_BADID));
1082 	}
1083 
1084 	if ((type = ctf_add_generic(fp, flag, NULL, &dtd)) == CTF_ERR)
1085 		return (CTF_ERR); /* errno is set for us */
1086 
1087 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_ARRAY, flag, 0);
1088 	dtd->dtd_data.ctt_size = 0;
1089 	dtd->dtd_u.dtu_arr = *arp;
1090 	ctf_ref_inc(fp, arp->ctr_contents);
1091 	ctf_ref_inc(fp, arp->ctr_index);
1092 
1093 	return (type);
1094 }
1095 
1096 int
1097 ctf_set_array(ctf_file_t *fp, ctf_id_t type, const ctf_arinfo_t *arp)
1098 {
1099 	ctf_file_t *fpd;
1100 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, type);
1101 
1102 	if (!(fp->ctf_flags & LCTF_RDWR))
1103 		return (ctf_set_errno(fp, ECTF_RDONLY));
1104 
1105 	if (dtd == NULL || CTF_INFO_KIND(dtd->dtd_data.ctt_info) != CTF_K_ARRAY)
1106 		return (ctf_set_errno(fp, ECTF_BADID));
1107 
1108 	fpd = fp;
1109 	if (ctf_lookup_by_id(&fpd, arp->ctr_contents) == NULL &&
1110 	    ctf_dtd_lookup(fp, arp->ctr_contents) == NULL)
1111 		return (ctf_set_errno(fp, ECTF_BADID));
1112 
1113 	fpd = fp;
1114 	if (ctf_lookup_by_id(&fpd, arp->ctr_index) == NULL &&
1115 	    ctf_dtd_lookup(fp, arp->ctr_index) == NULL)
1116 		return (ctf_set_errno(fp, ECTF_BADID));
1117 
1118 	ctf_ref_dec(fp, dtd->dtd_u.dtu_arr.ctr_contents);
1119 	ctf_ref_dec(fp, dtd->dtd_u.dtu_arr.ctr_index);
1120 	fp->ctf_flags |= LCTF_DIRTY;
1121 	dtd->dtd_u.dtu_arr = *arp;
1122 	ctf_ref_inc(fp, arp->ctr_contents);
1123 	ctf_ref_inc(fp, arp->ctr_index);
1124 
1125 	return (0);
1126 }
1127 
1128 ctf_id_t
1129 ctf_add_funcptr(ctf_file_t *fp, uint_t flag,
1130     const ctf_funcinfo_t *ctc, const ctf_id_t *argv)
1131 {
1132 	ctf_dtdef_t *dtd;
1133 	ctf_id_t type;
1134 	uint_t vlen;
1135 	int i;
1136 	ctf_id_t *vdat = NULL;
1137 	ctf_file_t *fpd;
1138 
1139 	if (ctc == NULL || (ctc->ctc_flags & ~CTF_FUNC_VARARG) != 0 ||
1140 	    (ctc->ctc_argc != 0 && argv == NULL))
1141 		return (ctf_set_errno(fp, EINVAL));
1142 
1143 	vlen = ctc->ctc_argc;
1144 	if (ctc->ctc_flags & CTF_FUNC_VARARG)
1145 		vlen++; /* add trailing zero to indicate varargs (see below) */
1146 
1147 	if (vlen > CTF_MAX_VLEN)
1148 		return (ctf_set_errno(fp, EOVERFLOW));
1149 
1150 	fpd = fp;
1151 	if (ctf_lookup_by_id(&fpd, ctc->ctc_return) == NULL &&
1152 	    ctf_dtd_lookup(fp, ctc->ctc_return) == NULL)
1153 		return (ctf_set_errno(fp, ECTF_BADID));
1154 
1155 	for (i = 0; i < ctc->ctc_argc; i++) {
1156 		fpd = fp;
1157 		if (ctf_lookup_by_id(&fpd, argv[i]) == NULL &&
1158 		    ctf_dtd_lookup(fp, argv[i]) == NULL)
1159 			return (ctf_set_errno(fp, ECTF_BADID));
1160 	}
1161 
1162 	if (vlen != 0 && (vdat = ctf_alloc(sizeof (ctf_id_t) * vlen)) == NULL)
1163 		return (ctf_set_errno(fp, EAGAIN));
1164 
1165 	if ((type = ctf_add_generic(fp, flag, NULL, &dtd)) == CTF_ERR) {
1166 		ctf_free(vdat, sizeof (ctf_id_t) * vlen);
1167 		return (CTF_ERR); /* errno is set for us */
1168 	}
1169 
1170 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_FUNCTION, flag, vlen);
1171 	dtd->dtd_data.ctt_type = (ushort_t)ctc->ctc_return;
1172 
1173 	ctf_ref_inc(fp, ctc->ctc_return);
1174 	for (i = 0; i < ctc->ctc_argc; i++)
1175 		ctf_ref_inc(fp, argv[i]);
1176 
1177 	bcopy(argv, vdat, sizeof (ctf_id_t) * ctc->ctc_argc);
1178 	if (ctc->ctc_flags & CTF_FUNC_VARARG)
1179 		vdat[vlen - 1] = 0; /* add trailing zero to indicate varargs */
1180 	dtd->dtd_u.dtu_argv = vdat;
1181 
1182 	return (type);
1183 }
1184 
1185 ctf_id_t
1186 ctf_add_struct(ctf_file_t *fp, uint_t flag, const char *name)
1187 {
1188 	ctf_hash_t *hp = &fp->ctf_structs;
1189 	ctf_helem_t *hep = NULL;
1190 	ctf_dtdef_t *dtd = NULL;
1191 	ctf_id_t type = CTF_ERR;
1192 
1193 	if (name != NULL)
1194 		hep = ctf_hash_lookup(hp, fp, name, strlen(name));
1195 
1196 	if (hep != NULL && ctf_type_kind(fp, hep->h_type) == CTF_K_FORWARD) {
1197 		type = hep->h_type;
1198 		dtd = ctf_dtd_lookup(fp, type);
1199 		if (CTF_INFO_KIND(dtd->dtd_data.ctt_info) != CTF_K_FORWARD)
1200 			dtd = NULL;
1201 	}
1202 
1203 	if (dtd == NULL) {
1204 		type = ctf_add_generic(fp, flag, name, &dtd);
1205 		if (type == CTF_ERR)
1206 			return (CTF_ERR); /* errno is set for us */
1207 	}
1208 
1209 	VERIFY(type != CTF_ERR);
1210 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_STRUCT, flag, 0);
1211 	dtd->dtd_data.ctt_size = 0;
1212 
1213 	/*
1214 	 * Always dirty in case we modified a forward.
1215 	 */
1216 	fp->ctf_flags |= LCTF_DIRTY;
1217 
1218 	return (type);
1219 }
1220 
1221 ctf_id_t
1222 ctf_add_union(ctf_file_t *fp, uint_t flag, const char *name)
1223 {
1224 	ctf_hash_t *hp = &fp->ctf_unions;
1225 	ctf_helem_t *hep = NULL;
1226 	ctf_dtdef_t *dtd = NULL;
1227 	ctf_id_t type = CTF_ERR;
1228 
1229 	if (name != NULL)
1230 		hep = ctf_hash_lookup(hp, fp, name, strlen(name));
1231 
1232 	if (hep != NULL && ctf_type_kind(fp, hep->h_type) == CTF_K_FORWARD) {
1233 		type = hep->h_type;
1234 		dtd = ctf_dtd_lookup(fp, type);
1235 		if (CTF_INFO_KIND(dtd->dtd_data.ctt_info) != CTF_K_FORWARD)
1236 			dtd = NULL;
1237 	}
1238 
1239 	if (dtd == NULL) {
1240 		type = ctf_add_generic(fp, flag, name, &dtd);
1241 		if (type == CTF_ERR)
1242 			return (CTF_ERR); /* errno is set for us */
1243 	}
1244 
1245 	VERIFY(type != CTF_ERR);
1246 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_UNION, flag, 0);
1247 	dtd->dtd_data.ctt_size = 0;
1248 
1249 	/*
1250 	 * Always dirty in case we modified a forward.
1251 	 */
1252 	fp->ctf_flags |= LCTF_DIRTY;
1253 
1254 	return (type);
1255 }
1256 
1257 /*
1258  * If size is 0, we use the standard integer size. This is almost always the
1259  * case, except for packed enums.
1260  */
1261 ctf_id_t
1262 ctf_add_enum(ctf_file_t *fp, uint_t flag, const char *name, size_t size)
1263 {
1264 	ctf_hash_t *hp = &fp->ctf_enums;
1265 	ctf_helem_t *hep = NULL;
1266 	ctf_dtdef_t *dtd = NULL;
1267 	ctf_id_t type = CTF_ERR;
1268 
1269 	/* Check we could return something valid in ctf_type_size. */
1270 	if (size > SSIZE_MAX)
1271 		return (ctf_set_errno(fp, EINVAL));
1272 
1273 	if (name != NULL)
1274 		hep = ctf_hash_lookup(hp, fp, name, strlen(name));
1275 
1276 	if (hep != NULL && ctf_type_kind(fp, hep->h_type) == CTF_K_FORWARD) {
1277 		type = hep->h_type;
1278 		dtd = ctf_dtd_lookup(fp, type);
1279 		if (CTF_INFO_KIND(dtd->dtd_data.ctt_info) != CTF_K_FORWARD)
1280 			dtd = NULL;
1281 	}
1282 
1283 	if (dtd == NULL) {
1284 		type = ctf_add_generic(fp, flag, name, &dtd);
1285 		if (type == CTF_ERR)
1286 			return (CTF_ERR); /* errno is set for us */
1287 	}
1288 
1289 	VERIFY(type != CTF_ERR);
1290 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_ENUM, flag, 0);
1291 
1292 	ctf_set_ctt_size(&dtd->dtd_data, size == 0 ?
1293 	    fp->ctf_dmodel->ctd_int : size);
1294 
1295 	/*
1296 	 * Always dirty in case we modified a forward.
1297 	 */
1298 	fp->ctf_flags |= LCTF_DIRTY;
1299 
1300 	return (type);
1301 }
1302 
1303 ctf_id_t
1304 ctf_add_forward(ctf_file_t *fp, uint_t flag, const char *name, uint_t kind)
1305 {
1306 	ctf_hash_t *hp;
1307 	ctf_helem_t *hep;
1308 	ctf_dtdef_t *dtd;
1309 	ctf_id_t type;
1310 
1311 	switch (kind) {
1312 	case CTF_K_STRUCT:
1313 		hp = &fp->ctf_structs;
1314 		break;
1315 	case CTF_K_UNION:
1316 		hp = &fp->ctf_unions;
1317 		break;
1318 	case CTF_K_ENUM:
1319 		hp = &fp->ctf_enums;
1320 		break;
1321 	default:
1322 		return (ctf_set_errno(fp, ECTF_NOTSUE));
1323 	}
1324 
1325 	/*
1326 	 * If the type is already defined or exists as a forward tag, just
1327 	 * return the ctf_id_t of the existing definition.
1328 	 */
1329 	if (name != NULL && (hep = ctf_hash_lookup(hp,
1330 	    fp, name, strlen(name))) != NULL)
1331 		return (hep->h_type);
1332 
1333 	if ((type = ctf_add_generic(fp, flag, name, &dtd)) == CTF_ERR)
1334 		return (CTF_ERR); /* errno is set for us */
1335 
1336 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_FORWARD, flag, 0);
1337 	dtd->dtd_data.ctt_type = kind;
1338 
1339 	return (type);
1340 }
1341 
1342 ctf_id_t
1343 ctf_add_typedef(ctf_file_t *fp, uint_t flag, const char *name, ctf_id_t ref)
1344 {
1345 	ctf_dtdef_t *dtd;
1346 	ctf_id_t type;
1347 	ctf_file_t *fpd;
1348 
1349 	fpd = fp;
1350 	if (ref == CTF_ERR || (ctf_lookup_by_id(&fpd, ref) == NULL &&
1351 	    ctf_dtd_lookup(fp, ref) == NULL))
1352 		return (ctf_set_errno(fp, EINVAL));
1353 
1354 	if ((type = ctf_add_generic(fp, flag, name, &dtd)) == CTF_ERR)
1355 		return (CTF_ERR); /* errno is set for us */
1356 
1357 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(CTF_K_TYPEDEF, flag, 0);
1358 	dtd->dtd_data.ctt_type = (ushort_t)ref;
1359 	ctf_ref_inc(fp, ref);
1360 
1361 	return (type);
1362 }
1363 
1364 ctf_id_t
1365 ctf_add_volatile(ctf_file_t *fp, uint_t flag, const char *name, ctf_id_t ref)
1366 {
1367 	return (ctf_add_reftype(fp, flag, name, ref, CTF_K_VOLATILE));
1368 }
1369 
1370 ctf_id_t
1371 ctf_add_const(ctf_file_t *fp, uint_t flag, const char *name, ctf_id_t ref)
1372 {
1373 	return (ctf_add_reftype(fp, flag, name, ref, CTF_K_CONST));
1374 }
1375 
1376 ctf_id_t
1377 ctf_add_restrict(ctf_file_t *fp, uint_t flag, const char *name, ctf_id_t ref)
1378 {
1379 	return (ctf_add_reftype(fp, flag, name, ref, CTF_K_RESTRICT));
1380 }
1381 
1382 int
1383 ctf_add_enumerator(ctf_file_t *fp, ctf_id_t enid, const char *name, int value)
1384 {
1385 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, enid);
1386 	ctf_dmdef_t *dmd;
1387 
1388 	uint_t kind, vlen, root;
1389 	char *s;
1390 
1391 	if (name == NULL)
1392 		return (ctf_set_errno(fp, EINVAL));
1393 
1394 	if (!(fp->ctf_flags & LCTF_RDWR))
1395 		return (ctf_set_errno(fp, ECTF_RDONLY));
1396 
1397 	if (dtd == NULL)
1398 		return (ctf_set_errno(fp, ECTF_BADID));
1399 
1400 	kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
1401 	root = CTF_INFO_ISROOT(dtd->dtd_data.ctt_info);
1402 	vlen = CTF_INFO_VLEN(dtd->dtd_data.ctt_info);
1403 
1404 	if (kind != CTF_K_ENUM)
1405 		return (ctf_set_errno(fp, ECTF_NOTENUM));
1406 
1407 	if (vlen == CTF_MAX_VLEN)
1408 		return (ctf_set_errno(fp, ECTF_DTFULL));
1409 
1410 	for (dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
1411 	    dmd != NULL; dmd = ctf_list_next(dmd)) {
1412 		if (strcmp(dmd->dmd_name, name) == 0) {
1413 			ctf_dprintf("encountered duplicate member %s\n", name);
1414 			return (ctf_set_errno(fp, ECTF_DUPMEMBER));
1415 		}
1416 	}
1417 
1418 	if ((dmd = ctf_alloc(sizeof (ctf_dmdef_t))) == NULL)
1419 		return (ctf_set_errno(fp, EAGAIN));
1420 
1421 	if ((s = ctf_strdup(name)) == NULL) {
1422 		ctf_free(dmd, sizeof (ctf_dmdef_t));
1423 		return (ctf_set_errno(fp, EAGAIN));
1424 	}
1425 
1426 	dmd->dmd_name = s;
1427 	dmd->dmd_type = CTF_ERR;
1428 	dmd->dmd_offset = 0;
1429 	dmd->dmd_value = value;
1430 
1431 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(kind, root, vlen + 1);
1432 	ctf_list_append(&dtd->dtd_u.dtu_members, dmd);
1433 
1434 	fp->ctf_dtstrlen += strlen(s) + 1;
1435 	fp->ctf_flags |= LCTF_DIRTY;
1436 
1437 	return (0);
1438 }
1439 
1440 int
1441 ctf_add_member(ctf_file_t *fp, ctf_id_t souid, const char *name, ctf_id_t type,
1442     ulong_t offset)
1443 {
1444 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, souid);
1445 	ctf_dmdef_t *dmd;
1446 
1447 	ulong_t mbitsz;
1448 	ssize_t msize, malign, ssize;
1449 	uint_t kind, vlen, root;
1450 	int mkind;
1451 	char *s = NULL;
1452 
1453 	if (!(fp->ctf_flags & LCTF_RDWR))
1454 		return (ctf_set_errno(fp, ECTF_RDONLY));
1455 
1456 	if (dtd == NULL)
1457 		return (ctf_set_errno(fp, ECTF_BADID));
1458 
1459 	kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
1460 	root = CTF_INFO_ISROOT(dtd->dtd_data.ctt_info);
1461 	vlen = CTF_INFO_VLEN(dtd->dtd_data.ctt_info);
1462 
1463 	if (kind != CTF_K_STRUCT && kind != CTF_K_UNION)
1464 		return (ctf_set_errno(fp, ECTF_NOTSOU));
1465 
1466 	if (vlen == CTF_MAX_VLEN)
1467 		return (ctf_set_errno(fp, ECTF_DTFULL));
1468 
1469 	/*
1470 	 * Structures may have members which are anonymous. If they have two of
1471 	 * these, then the duplicate member detection would find it due to the
1472 	 * string of "", so we skip it.
1473 	 */
1474 	if (name != NULL && *name != '\0') {
1475 		for (dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
1476 		    dmd != NULL; dmd = ctf_list_next(dmd)) {
1477 			if (dmd->dmd_name != NULL &&
1478 			    strcmp(dmd->dmd_name, name) == 0) {
1479 				return (ctf_set_errno(fp, ECTF_DUPMEMBER));
1480 			}
1481 		}
1482 	}
1483 
1484 	if ((msize = ctf_type_size(fp, type)) == CTF_ERR ||
1485 	    (malign = ctf_type_align(fp, type)) == CTF_ERR ||
1486 	    (mkind = ctf_type_kind(fp, type)) == CTF_ERR)
1487 		return (CTF_ERR); /* errno is set for us */
1488 
1489 	/*
1490 	 * ctf_type_size returns sizes in bytes. However, for bitfields, that
1491 	 * means that it may misrepresent and actually rounds it up to a power
1492 	 * of two and store that in bytes. So instead we have to get the
1493 	 * Integers encoding and rely on that.
1494 	 */
1495 	if (mkind == CTF_K_INTEGER) {
1496 		ctf_encoding_t e;
1497 
1498 		if (ctf_type_encoding(fp, type, &e) == CTF_ERR)
1499 			return (CTF_ERR); /* errno is set for us */
1500 		mbitsz = e.cte_bits;
1501 	} else if (mkind == CTF_K_FORWARD) {
1502 		/*
1503 		 * This is a rather rare case. In general one cannot add a
1504 		 * forward to a structure. However, the CTF tools traditionally
1505 		 * tried to add a forward to the struct cpu as the last member.
1506 		 * Therefore, if we find one here, we're going to verify the
1507 		 * size and make sure it's zero. It's certainly odd, but that's
1508 		 * life.
1509 		 *
1510 		 * Further, if it's not an absolute position being specified,
1511 		 * then we refuse to add it.
1512 		 */
1513 		if (offset == ULONG_MAX)
1514 			return (ctf_set_errno(fp, EINVAL));
1515 		VERIFY(msize == 0);
1516 		mbitsz = msize;
1517 	} else {
1518 		mbitsz = msize * 8;
1519 	}
1520 
1521 	if ((dmd = ctf_alloc(sizeof (ctf_dmdef_t))) == NULL)
1522 		return (ctf_set_errno(fp, EAGAIN));
1523 
1524 	if (name != NULL && (s = ctf_strdup(name)) == NULL) {
1525 		ctf_free(dmd, sizeof (ctf_dmdef_t));
1526 		return (ctf_set_errno(fp, EAGAIN));
1527 	}
1528 
1529 	dmd->dmd_name = s;
1530 	dmd->dmd_type = type;
1531 	dmd->dmd_value = -1;
1532 
1533 	if (kind == CTF_K_STRUCT && vlen != 0) {
1534 		ctf_dmdef_t *lmd = ctf_list_prev(&dtd->dtd_u.dtu_members);
1535 		ctf_id_t ltype = ctf_type_resolve(fp, lmd->dmd_type);
1536 		size_t off;
1537 
1538 		if (offset == ULONG_MAX) {
1539 			ctf_encoding_t linfo;
1540 			ssize_t lsize;
1541 
1542 			off = lmd->dmd_offset;
1543 			if (ctf_type_encoding(fp, ltype, &linfo) != CTF_ERR)
1544 				off += linfo.cte_bits;
1545 			else if ((lsize = ctf_type_size(fp, ltype)) != CTF_ERR)
1546 				off += lsize * NBBY;
1547 
1548 			/*
1549 			 * Round up the offset of the end of the last member to
1550 			 * the next byte boundary, convert 'off' to bytes, and
1551 			 * then round it up again to the next multiple of the
1552 			 * alignment required by the new member.  Finally,
1553 			 * convert back to bits and store the result in
1554 			 * dmd_offset.  Technically we could do more efficient
1555 			 * packing if the new member is a bit-field, but we're
1556 			 * the "compiler" and ANSI says we can do as we choose.
1557 			 */
1558 			off = roundup(off, NBBY) / NBBY;
1559 			off = roundup(off, MAX(malign, 1));
1560 			dmd->dmd_offset = off * NBBY;
1561 			ssize = off + msize;
1562 		} else {
1563 			dmd->dmd_offset = offset;
1564 			ssize = (offset + mbitsz) / NBBY;
1565 		}
1566 	} else {
1567 		dmd->dmd_offset = 0;
1568 		ssize = ctf_get_ctt_size(fp, &dtd->dtd_data, NULL, NULL);
1569 		ssize = MAX(ssize, msize);
1570 	}
1571 
1572 	ctf_set_ctt_size(&dtd->dtd_data, ssize);
1573 
1574 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(kind, root, vlen + 1);
1575 	ctf_list_append(&dtd->dtd_u.dtu_members, dmd);
1576 
1577 	if (s != NULL)
1578 		fp->ctf_dtstrlen += strlen(s) + 1;
1579 
1580 	ctf_ref_inc(fp, type);
1581 	fp->ctf_flags |= LCTF_DIRTY;
1582 	return (0);
1583 }
1584 
1585 /*
1586  * This removes a type from the dynamic section. This will fail if the type is
1587  * referenced by another type. Note that the CTF ID is never reused currently by
1588  * CTF. Note that if this container is a parent container then we just outright
1589  * refuse to remove the type. There currently is no notion of searching for the
1590  * ctf_dtdef_t in parent containers. If there is, then this constraint could
1591  * become finer grained.
1592  */
1593 int
1594 ctf_delete_type(ctf_file_t *fp, ctf_id_t type)
1595 {
1596 	ctf_file_t *fpd;
1597 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, type);
1598 
1599 	if (!(fp->ctf_flags & LCTF_RDWR))
1600 		return (ctf_set_errno(fp, ECTF_RDONLY));
1601 
1602 	/*
1603 	 * We want to give as useful an errno as possible. That means that we
1604 	 * want to distinguish between a type which does not exist and one for
1605 	 * which the type is not dynamic.
1606 	 */
1607 	fpd = fp;
1608 	if (ctf_lookup_by_id(&fpd, type) == NULL &&
1609 	    ctf_dtd_lookup(fp, type) == NULL)
1610 		return (CTF_ERR); /* errno is set for us */
1611 
1612 	if (dtd == NULL)
1613 		return (ctf_set_errno(fp, ECTF_NOTDYN));
1614 
1615 	if (dtd->dtd_ref != 0 || fp->ctf_refcnt > 1)
1616 		return (ctf_set_errno(fp, ECTF_REFERENCED));
1617 
1618 	ctf_dtd_delete(fp, dtd);
1619 	fp->ctf_flags |= LCTF_DIRTY;
1620 	return (0);
1621 }
1622 
1623 static int
1624 enumcmp(const char *name, int value, void *arg)
1625 {
1626 	ctf_bundle_t *ctb = arg;
1627 	int bvalue;
1628 
1629 	return (ctf_enum_value(ctb->ctb_file, ctb->ctb_type,
1630 	    name, &bvalue) == CTF_ERR || value != bvalue);
1631 }
1632 
1633 static int
1634 enumadd(const char *name, int value, void *arg)
1635 {
1636 	ctf_bundle_t *ctb = arg;
1637 
1638 	return (ctf_add_enumerator(ctb->ctb_file, ctb->ctb_type,
1639 	    name, value) == CTF_ERR);
1640 }
1641 
1642 /*ARGSUSED*/
1643 static int
1644 membcmp(const char *name, ctf_id_t type, ulong_t offset, void *arg)
1645 {
1646 	ctf_bundle_t *ctb = arg;
1647 	ctf_membinfo_t ctm;
1648 
1649 	return (ctf_member_info(ctb->ctb_file, ctb->ctb_type,
1650 	    name, &ctm) == CTF_ERR || ctm.ctm_offset != offset);
1651 }
1652 
1653 static int
1654 membadd(const char *name, ctf_id_t type, ulong_t offset, void *arg)
1655 {
1656 	ctf_bundle_t *ctb = arg;
1657 	ctf_dmdef_t *dmd;
1658 	char *s = NULL;
1659 
1660 	if ((dmd = ctf_alloc(sizeof (ctf_dmdef_t))) == NULL)
1661 		return (ctf_set_errno(ctb->ctb_file, EAGAIN));
1662 
1663 	if (name != NULL && (s = ctf_strdup(name)) == NULL) {
1664 		ctf_free(dmd, sizeof (ctf_dmdef_t));
1665 		return (ctf_set_errno(ctb->ctb_file, EAGAIN));
1666 	}
1667 
1668 	/*
1669 	 * For now, dmd_type is copied as the src_fp's type; it is reset to an
1670 	 * equivalent dst_fp type by a final loop in ctf_add_type(), below.
1671 	 */
1672 	dmd->dmd_name = s;
1673 	dmd->dmd_type = type;
1674 	dmd->dmd_offset = offset;
1675 	dmd->dmd_value = -1;
1676 
1677 	ctf_list_append(&ctb->ctb_dtd->dtd_u.dtu_members, dmd);
1678 
1679 	if (s != NULL)
1680 		ctb->ctb_file->ctf_dtstrlen += strlen(s) + 1;
1681 
1682 	ctb->ctb_file->ctf_flags |= LCTF_DIRTY;
1683 	return (0);
1684 }
1685 
1686 /*
1687  * The ctf_add_type routine is used to copy a type from a source CTF container
1688  * to a dynamic destination container.  This routine operates recursively by
1689  * following the source type's links and embedded member types.  If the
1690  * destination container already contains a named type which has the same
1691  * attributes, then we succeed and return this type but no changes occur.
1692  */
1693 ctf_id_t
1694 ctf_add_type(ctf_file_t *dst_fp, ctf_file_t *src_fp, ctf_id_t src_type)
1695 {
1696 	ctf_id_t dst_type = CTF_ERR;
1697 	uint_t dst_kind = CTF_K_UNKNOWN;
1698 
1699 	const ctf_type_t *tp;
1700 	const char *name;
1701 	uint_t kind, flag, vlen;
1702 
1703 	ctf_bundle_t src, dst;
1704 	ctf_encoding_t src_en, dst_en;
1705 	ctf_arinfo_t src_ar, dst_ar;
1706 
1707 	ctf_dtdef_t *dtd;
1708 	ctf_funcinfo_t ctc;
1709 
1710 	ctf_hash_t *hp;
1711 	ctf_helem_t *hep;
1712 
1713 	if (dst_fp == src_fp)
1714 		return (src_type);
1715 
1716 	if (!(dst_fp->ctf_flags & LCTF_RDWR))
1717 		return (ctf_set_errno(dst_fp, ECTF_RDONLY));
1718 
1719 	if ((tp = ctf_lookup_by_id(&src_fp, src_type)) == NULL)
1720 		return (ctf_set_errno(dst_fp, ctf_errno(src_fp)));
1721 
1722 	name = ctf_strptr(src_fp, tp->ctt_name);
1723 	kind = LCTF_INFO_KIND(src_fp, tp->ctt_info);
1724 	flag = LCTF_INFO_ROOT(src_fp, tp->ctt_info);
1725 	vlen = LCTF_INFO_VLEN(src_fp, tp->ctt_info);
1726 
1727 	switch (kind) {
1728 	case CTF_K_STRUCT:
1729 		hp = &dst_fp->ctf_structs;
1730 		break;
1731 	case CTF_K_UNION:
1732 		hp = &dst_fp->ctf_unions;
1733 		break;
1734 	case CTF_K_ENUM:
1735 		hp = &dst_fp->ctf_enums;
1736 		break;
1737 	default:
1738 		hp = &dst_fp->ctf_names;
1739 		break;
1740 	}
1741 
1742 	/*
1743 	 * If the source type has a name and is a root type (visible at the
1744 	 * top-level scope), lookup the name in the destination container and
1745 	 * verify that it is of the same kind before we do anything else.
1746 	 */
1747 	if ((flag & CTF_ADD_ROOT) && name[0] != '\0' &&
1748 	    (hep = ctf_hash_lookup(hp, dst_fp, name, strlen(name))) != NULL) {
1749 		dst_type = (ctf_id_t)hep->h_type;
1750 		dst_kind = ctf_type_kind(dst_fp, dst_type);
1751 	}
1752 
1753 	/*
1754 	 * If an identically named dst_type exists, fail with ECTF_CONFLICT
1755 	 * unless dst_type is a forward declaration and src_type is a struct,
1756 	 * union, or enum (i.e. the definition of the previous forward decl).
1757 	 */
1758 	if (dst_type != CTF_ERR && dst_kind != kind && (
1759 	    dst_kind != CTF_K_FORWARD || (kind != CTF_K_ENUM &&
1760 	    kind != CTF_K_STRUCT && kind != CTF_K_UNION)))
1761 		return (ctf_set_errno(dst_fp, ECTF_CONFLICT));
1762 
1763 	/*
1764 	 * If the non-empty name was not found in the appropriate hash, search
1765 	 * the list of pending dynamic definitions that are not yet committed.
1766 	 * If a matching name and kind are found, assume this is the type that
1767 	 * we are looking for.  This is necessary to permit ctf_add_type() to
1768 	 * operate recursively on entities such as a struct that contains a
1769 	 * pointer member that refers to the same struct type.
1770 	 */
1771 	if (dst_type == CTF_ERR && name[0] != '\0') {
1772 		for (dtd = ctf_list_prev(&dst_fp->ctf_dtdefs); dtd != NULL &&
1773 		    dtd->dtd_type > dst_fp->ctf_dtoldid;
1774 		    dtd = ctf_list_prev(dtd)) {
1775 			if (CTF_INFO_KIND(dtd->dtd_data.ctt_info) == kind &&
1776 			    dtd->dtd_name != NULL &&
1777 			    strcmp(dtd->dtd_name, name) == 0)
1778 				return (dtd->dtd_type);
1779 		}
1780 	}
1781 
1782 	src.ctb_file = src_fp;
1783 	src.ctb_type = src_type;
1784 	src.ctb_dtd = NULL;
1785 
1786 	dst.ctb_file = dst_fp;
1787 	dst.ctb_type = dst_type;
1788 	dst.ctb_dtd = NULL;
1789 
1790 	/*
1791 	 * Now perform kind-specific processing.  If dst_type is CTF_ERR, then
1792 	 * we add a new type with the same properties as src_type to dst_fp.
1793 	 * If dst_type is not CTF_ERR, then we verify that dst_type has the
1794 	 * same attributes as src_type.  We recurse for embedded references.
1795 	 */
1796 	switch (kind) {
1797 	case CTF_K_INTEGER:
1798 	case CTF_K_FLOAT:
1799 		if (ctf_type_encoding(src_fp, src_type, &src_en) != 0)
1800 			return (ctf_set_errno(dst_fp, ctf_errno(src_fp)));
1801 
1802 		if (dst_type != CTF_ERR) {
1803 			if (ctf_type_encoding(dst_fp, dst_type, &dst_en) != 0)
1804 				return (CTF_ERR); /* errno is set for us */
1805 
1806 			if (bcmp(&src_en, &dst_en, sizeof (ctf_encoding_t)))
1807 				return (ctf_set_errno(dst_fp, ECTF_CONFLICT));
1808 
1809 		} else if (kind == CTF_K_INTEGER) {
1810 			dst_type = ctf_add_integer(dst_fp, flag, name, &src_en);
1811 		} else
1812 			dst_type = ctf_add_float(dst_fp, flag, name, &src_en);
1813 		break;
1814 
1815 	case CTF_K_POINTER:
1816 	case CTF_K_VOLATILE:
1817 	case CTF_K_CONST:
1818 	case CTF_K_RESTRICT:
1819 		src_type = ctf_type_reference(src_fp, src_type);
1820 		src_type = ctf_add_type(dst_fp, src_fp, src_type);
1821 
1822 		if (src_type == CTF_ERR)
1823 			return (CTF_ERR); /* errno is set for us */
1824 
1825 		dst_type = ctf_add_reftype(dst_fp, flag, NULL, src_type, kind);
1826 		break;
1827 
1828 	case CTF_K_ARRAY:
1829 		if (ctf_array_info(src_fp, src_type, &src_ar) == CTF_ERR)
1830 			return (ctf_set_errno(dst_fp, ctf_errno(src_fp)));
1831 
1832 		src_ar.ctr_contents =
1833 		    ctf_add_type(dst_fp, src_fp, src_ar.ctr_contents);
1834 		src_ar.ctr_index =
1835 		    ctf_add_type(dst_fp, src_fp, src_ar.ctr_index);
1836 		src_ar.ctr_nelems = src_ar.ctr_nelems;
1837 
1838 		if (src_ar.ctr_contents == CTF_ERR ||
1839 		    src_ar.ctr_index == CTF_ERR)
1840 			return (CTF_ERR); /* errno is set for us */
1841 
1842 		if (dst_type != CTF_ERR) {
1843 			if (ctf_array_info(dst_fp, dst_type, &dst_ar) != 0)
1844 				return (CTF_ERR); /* errno is set for us */
1845 
1846 			if (bcmp(&src_ar, &dst_ar, sizeof (ctf_arinfo_t)))
1847 				return (ctf_set_errno(dst_fp, ECTF_CONFLICT));
1848 		} else
1849 			dst_type = ctf_add_array(dst_fp, flag, &src_ar);
1850 		break;
1851 
1852 	case CTF_K_FUNCTION:
1853 		ctc.ctc_return = ctf_add_type(dst_fp, src_fp, tp->ctt_type);
1854 		ctc.ctc_argc = 0;
1855 		ctc.ctc_flags = 0;
1856 
1857 		if (ctc.ctc_return == CTF_ERR)
1858 			return (CTF_ERR); /* errno is set for us */
1859 
1860 		dst_type = ctf_add_funcptr(dst_fp, flag, &ctc, NULL);
1861 		break;
1862 
1863 	case CTF_K_STRUCT:
1864 	case CTF_K_UNION: {
1865 		ctf_dmdef_t *dmd;
1866 		int errs = 0;
1867 
1868 		/*
1869 		 * Technically to match a struct or union we need to check both
1870 		 * ways (src members vs. dst, dst members vs. src) but we make
1871 		 * this more optimal by only checking src vs. dst and comparing
1872 		 * the total size of the structure (which we must do anyway)
1873 		 * which covers the possibility of dst members not in src.
1874 		 * This optimization can be defeated for unions, but is so
1875 		 * pathological as to render it irrelevant for our purposes.
1876 		 */
1877 		if (dst_type != CTF_ERR && dst_kind != CTF_K_FORWARD) {
1878 			if (ctf_type_size(src_fp, src_type) !=
1879 			    ctf_type_size(dst_fp, dst_type))
1880 				return (ctf_set_errno(dst_fp, ECTF_CONFLICT));
1881 
1882 			if (ctf_member_iter(src_fp, src_type, membcmp, &dst))
1883 				return (ctf_set_errno(dst_fp, ECTF_CONFLICT));
1884 
1885 			break;
1886 		}
1887 
1888 		/*
1889 		 * Unlike the other cases, copying structs and unions is done
1890 		 * manually so as to avoid repeated lookups in ctf_add_member
1891 		 * and to ensure the exact same member offsets as in src_type.
1892 		 */
1893 		dst_type = ctf_add_generic(dst_fp, flag, name, &dtd);
1894 		if (dst_type == CTF_ERR)
1895 			return (CTF_ERR); /* errno is set for us */
1896 
1897 		dst.ctb_type = dst_type;
1898 		dst.ctb_dtd = dtd;
1899 
1900 		if (ctf_member_iter(src_fp, src_type, membadd, &dst) != 0)
1901 			errs++; /* increment errs and fail at bottom of case */
1902 
1903 		ctf_set_ctt_size(&dtd->dtd_data,
1904 		    ctf_type_size(src_fp, src_type));
1905 
1906 		dtd->dtd_data.ctt_info = CTF_TYPE_INFO(kind, flag, vlen);
1907 
1908 		/*
1909 		 * Make a final pass through the members changing each dmd_type
1910 		 * (a src_fp type) to an equivalent type in dst_fp.  We pass
1911 		 * through all members, leaving any that fail set to CTF_ERR.
1912 		 */
1913 		for (dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
1914 		    dmd != NULL; dmd = ctf_list_next(dmd)) {
1915 			if ((dmd->dmd_type = ctf_add_type(dst_fp, src_fp,
1916 			    dmd->dmd_type)) == CTF_ERR)
1917 				errs++;
1918 		}
1919 
1920 		if (errs)
1921 			return (CTF_ERR); /* errno is set for us */
1922 
1923 		/*
1924 		 * Now that we know that we can't fail, we go through and bump
1925 		 * all the reference counts on the member types.
1926 		 */
1927 		for (dmd = ctf_list_next(&dtd->dtd_u.dtu_members);
1928 		    dmd != NULL; dmd = ctf_list_next(dmd))
1929 			ctf_ref_inc(dst_fp, dmd->dmd_type);
1930 		break;
1931 	}
1932 
1933 	case CTF_K_ENUM:
1934 		if (dst_type != CTF_ERR && dst_kind != CTF_K_FORWARD) {
1935 			if (ctf_enum_iter(src_fp, src_type, enumcmp, &dst) ||
1936 			    ctf_enum_iter(dst_fp, dst_type, enumcmp, &src))
1937 				return (ctf_set_errno(dst_fp, ECTF_CONFLICT));
1938 		} else {
1939 			ssize_t size = ctf_type_size(src_fp, src_type);
1940 
1941 			if (size == CTF_ERR)
1942 				return (CTF_ERR); /* errno is set for us */
1943 
1944 			dst_type = ctf_add_enum(dst_fp, flag, name, size);
1945 			if ((dst.ctb_type = dst_type) == CTF_ERR ||
1946 			    ctf_enum_iter(src_fp, src_type, enumadd, &dst))
1947 				return (CTF_ERR); /* errno is set for us */
1948 		}
1949 		break;
1950 
1951 	case CTF_K_FORWARD:
1952 		if (dst_type == CTF_ERR) {
1953 			dst_type = ctf_add_forward(dst_fp,
1954 			    flag, name, CTF_K_STRUCT); /* assume STRUCT */
1955 		}
1956 		break;
1957 
1958 	case CTF_K_TYPEDEF:
1959 		src_type = ctf_type_reference(src_fp, src_type);
1960 		src_type = ctf_add_type(dst_fp, src_fp, src_type);
1961 
1962 		if (src_type == CTF_ERR)
1963 			return (CTF_ERR); /* errno is set for us */
1964 
1965 		/*
1966 		 * If dst_type is not CTF_ERR at this point, we should check if
1967 		 * ctf_type_reference(dst_fp, dst_type) != src_type and if so
1968 		 * fail with ECTF_CONFLICT.  However, this causes problems with
1969 		 * <sys/types.h> typedefs that vary based on things like if
1970 		 * _ILP32x then pid_t is int otherwise long.  We therefore omit
1971 		 * this check and assume that if the identically named typedef
1972 		 * already exists in dst_fp, it is correct or equivalent.
1973 		 */
1974 		if (dst_type == CTF_ERR) {
1975 			dst_type = ctf_add_typedef(dst_fp, flag,
1976 			    name, src_type);
1977 		}
1978 		break;
1979 
1980 	default:
1981 		return (ctf_set_errno(dst_fp, ECTF_CORRUPT));
1982 	}
1983 
1984 	return (dst_type);
1985 }
1986 
1987 int
1988 ctf_add_function(ctf_file_t *fp, ulong_t idx, const ctf_funcinfo_t *fip,
1989     const ctf_id_t *argc)
1990 {
1991 	int i;
1992 	ctf_dsdef_t *dsd;
1993 	ctf_file_t *afp;
1994 	uintptr_t symbase = (uintptr_t)fp->ctf_symtab.cts_data;
1995 
1996 	if (!(fp->ctf_flags & LCTF_RDWR))
1997 		return (ctf_set_errno(fp, ECTF_RDONLY));
1998 
1999 	if (ctf_dsd_lookup(fp, idx) != NULL)
2000 		return (ctf_set_errno(fp, ECTF_CONFLICT));
2001 
2002 	if (symbase == (uintptr_t)NULL)
2003 		return (ctf_set_errno(fp, ECTF_STRTAB));
2004 
2005 	if (idx > fp->ctf_nsyms)
2006 		return (ctf_set_errno(fp, ECTF_NOTDATA));
2007 
2008 	if (fp->ctf_symtab.cts_entsize == sizeof (Elf32_Sym)) {
2009 		const Elf32_Sym *symp = (Elf32_Sym *)symbase + idx;
2010 		if (ELF32_ST_TYPE(symp->st_info) != STT_FUNC)
2011 			return (ctf_set_errno(fp, ECTF_NOTFUNC));
2012 	} else {
2013 		const Elf64_Sym *symp = (Elf64_Sym *)symbase + idx;
2014 		if (ELF64_ST_TYPE(symp->st_info) != STT_FUNC)
2015 			return (ctf_set_errno(fp, ECTF_NOTFUNC));
2016 	}
2017 
2018 	afp = fp;
2019 	if (ctf_lookup_by_id(&afp, fip->ctc_return) == NULL)
2020 		return (CTF_ERR); /* errno is set for us */
2021 
2022 	for (i = 0; i < fip->ctc_argc; i++) {
2023 		afp = fp;
2024 		if (ctf_lookup_by_id(&afp, argc[i]) == NULL)
2025 			return (CTF_ERR); /* errno is set for us */
2026 	}
2027 
2028 	dsd = ctf_alloc(sizeof (ctf_dsdef_t));
2029 	if (dsd == NULL)
2030 		return (ctf_set_errno(fp, ENOMEM));
2031 	dsd->dsd_nargs = fip->ctc_argc;
2032 	if (fip->ctc_flags & CTF_FUNC_VARARG)
2033 		dsd->dsd_nargs++;
2034 	if (dsd->dsd_nargs != 0) {
2035 		dsd->dsd_argc = ctf_alloc(sizeof (ctf_id_t) * dsd->dsd_nargs);
2036 		if (dsd->dsd_argc == NULL) {
2037 			ctf_free(dsd, sizeof (ctf_dsdef_t));
2038 			return (ctf_set_errno(fp, ENOMEM));
2039 		}
2040 		bcopy(argc, dsd->dsd_argc, sizeof (ctf_id_t) * fip->ctc_argc);
2041 		if (fip->ctc_flags & CTF_FUNC_VARARG)
2042 			dsd->dsd_argc[fip->ctc_argc] = 0;
2043 	}
2044 	dsd->dsd_symidx = idx;
2045 	dsd->dsd_tid = fip->ctc_return;
2046 
2047 	ctf_dsd_insert(fp, dsd);
2048 	fp->ctf_flags |= LCTF_DIRTY;
2049 
2050 	return (0);
2051 }
2052 
2053 int
2054 ctf_add_object(ctf_file_t *fp, ulong_t idx, ctf_id_t type)
2055 {
2056 	ctf_dsdef_t *dsd;
2057 	ctf_file_t *afp;
2058 	uintptr_t symbase = (uintptr_t)fp->ctf_symtab.cts_data;
2059 
2060 	if (!(fp->ctf_flags & LCTF_RDWR))
2061 		return (ctf_set_errno(fp, ECTF_RDONLY));
2062 
2063 	if (!(fp->ctf_flags & LCTF_RDWR))
2064 		return (ctf_set_errno(fp, ECTF_RDONLY));
2065 
2066 	if (ctf_dsd_lookup(fp, idx) != NULL)
2067 		return (ctf_set_errno(fp, ECTF_CONFLICT));
2068 
2069 	if (symbase == (uintptr_t)NULL)
2070 		return (ctf_set_errno(fp, ECTF_STRTAB));
2071 
2072 	if (idx > fp->ctf_nsyms)
2073 		return (ctf_set_errno(fp, ECTF_NOTDATA));
2074 
2075 	if (fp->ctf_symtab.cts_entsize == sizeof (Elf32_Sym)) {
2076 		const Elf32_Sym *symp = (Elf32_Sym *)symbase + idx;
2077 		if (ELF32_ST_TYPE(symp->st_info) != STT_OBJECT)
2078 			return (ctf_set_errno(fp, ECTF_NOTDATA));
2079 	} else {
2080 		const Elf64_Sym *symp = (Elf64_Sym *)symbase + idx;
2081 		if (ELF64_ST_TYPE(symp->st_info) != STT_OBJECT)
2082 			return (ctf_set_errno(fp, ECTF_NOTDATA));
2083 	}
2084 
2085 	afp = fp;
2086 	if (ctf_lookup_by_id(&afp, type) == NULL)
2087 		return (CTF_ERR); /* errno is set for us */
2088 
2089 	dsd = ctf_alloc(sizeof (ctf_dsdef_t));
2090 	if (dsd == NULL)
2091 		return (ctf_set_errno(fp, ENOMEM));
2092 	dsd->dsd_symidx = idx;
2093 	dsd->dsd_tid = type;
2094 	dsd->dsd_argc = NULL;
2095 
2096 	ctf_dsd_insert(fp, dsd);
2097 	fp->ctf_flags |= LCTF_DIRTY;
2098 
2099 	return (0);
2100 }
2101 
2102 void
2103 ctf_dataptr(ctf_file_t *fp, const void **addrp, size_t *sizep)
2104 {
2105 	if (addrp != NULL)
2106 		*addrp = fp->ctf_base;
2107 	if (sizep != NULL)
2108 		*sizep = fp->ctf_size;
2109 }
2110 
2111 int
2112 ctf_add_label(ctf_file_t *fp, const char *name, ctf_id_t type, uint_t position)
2113 {
2114 	ctf_file_t *fpd;
2115 	ctf_dldef_t *dld;
2116 
2117 	if (name == NULL)
2118 		return (ctf_set_errno(fp, EINVAL));
2119 
2120 	if (!(fp->ctf_flags & LCTF_RDWR))
2121 		return (ctf_set_errno(fp, ECTF_RDONLY));
2122 
2123 	fpd = fp;
2124 	if (type != 0 && ctf_lookup_by_id(&fpd, type) == NULL)
2125 		return (CTF_ERR); /* errno is set for us */
2126 
2127 	if (type != 0 && (fp->ctf_flags & LCTF_CHILD) &&
2128 	    CTF_TYPE_ISPARENT(type))
2129 		return (ctf_set_errno(fp, ECTF_NOPARENT));
2130 
2131 	if (ctf_dld_lookup(fp, name) != NULL)
2132 		return (ctf_set_errno(fp, ECTF_LABELEXISTS));
2133 
2134 	if ((dld = ctf_alloc(sizeof (ctf_dldef_t))) == NULL)
2135 		return (ctf_set_errno(fp, EAGAIN));
2136 
2137 	if ((dld->dld_name = ctf_strdup(name)) == NULL) {
2138 		ctf_free(dld, sizeof (ctf_dldef_t));
2139 		return (ctf_set_errno(fp, EAGAIN));
2140 	}
2141 
2142 	ctf_dprintf("adding label %s, %ld\n", name, type);
2143 	dld->dld_type = type;
2144 	fp->ctf_dtstrlen += strlen(name) + 1;
2145 	ctf_dld_insert(fp, dld, position);
2146 	fp->ctf_flags |= LCTF_DIRTY;
2147 
2148 	return (0);
2149 }
2150 
2151 /*
2152  * Update the size of a structure or union. Note that we don't allow this to
2153  * shrink the size of a struct or union, only to increase it. This is useful for
2154  * cases when you have a structure whose actual size is larger than the sum of
2155  * its members due to padding for natural alignment.
2156  */
2157 int
2158 ctf_set_size(ctf_file_t *fp, ctf_id_t id, const ulong_t newsz)
2159 {
2160 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, id);
2161 	uint_t kind;
2162 	size_t oldsz;
2163 
2164 	if (!(fp->ctf_flags & LCTF_RDWR))
2165 		return (ctf_set_errno(fp, ECTF_RDONLY));
2166 
2167 	if (dtd == NULL)
2168 		return (ctf_set_errno(fp, ECTF_BADID));
2169 
2170 	kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
2171 
2172 	if (kind != CTF_K_STRUCT && kind != CTF_K_UNION)
2173 		return (ctf_set_errno(fp, ECTF_NOTSOU));
2174 
2175 	if ((oldsz = dtd->dtd_data.ctt_size) == CTF_LSIZE_SENT)
2176 		oldsz = CTF_TYPE_LSIZE(&dtd->dtd_data);
2177 
2178 	if (newsz < oldsz)
2179 		return (ctf_set_errno(fp, EINVAL));
2180 
2181 	ctf_set_ctt_size(&dtd->dtd_data, newsz);
2182 
2183 	fp->ctf_flags |= LCTF_DIRTY;
2184 	return (0);
2185 }
2186 
2187 int
2188 ctf_set_root(ctf_file_t *fp, ctf_id_t id, const boolean_t vis)
2189 {
2190 	ctf_dtdef_t *dtd = ctf_dtd_lookup(fp, id);
2191 	uint_t kind, vlen;
2192 
2193 	if (!(fp->ctf_flags & LCTF_RDWR))
2194 		return (ctf_set_errno(fp, ECTF_RDONLY));
2195 
2196 	if (dtd == NULL)
2197 		return (ctf_set_errno(fp, ECTF_BADID));
2198 
2199 	kind = CTF_INFO_KIND(dtd->dtd_data.ctt_info);
2200 	vlen = CTF_INFO_VLEN(dtd->dtd_data.ctt_info);
2201 
2202 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO(kind, vis, vlen);
2203 	return (0);
2204 }
2205