xref: /dflybsd-src/contrib/binutils-2.34/libctf/ctf-create.c (revision b52ef7118d1621abed722c5bbbd542210290ecef)
1*fae548d3Szrj /* CTF file creation.
2*fae548d3Szrj    Copyright (C) 2019-2020 Free Software Foundation, Inc.
3*fae548d3Szrj 
4*fae548d3Szrj    This file is part of libctf.
5*fae548d3Szrj 
6*fae548d3Szrj    libctf is free software; you can redistribute it and/or modify it under
7*fae548d3Szrj    the terms of the GNU General Public License as published by the Free
8*fae548d3Szrj    Software Foundation; either version 3, or (at your option) any later
9*fae548d3Szrj    version.
10*fae548d3Szrj 
11*fae548d3Szrj    This program is distributed in the hope that it will be useful, but
12*fae548d3Szrj    WITHOUT ANY WARRANTY; without even the implied warranty of
13*fae548d3Szrj    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
14*fae548d3Szrj    See the GNU General Public License for more details.
15*fae548d3Szrj 
16*fae548d3Szrj    You should have received a copy of the GNU General Public License
17*fae548d3Szrj    along with this program; see the file COPYING.  If not see
18*fae548d3Szrj    <http://www.gnu.org/licenses/>.  */
19*fae548d3Szrj 
20*fae548d3Szrj #include <ctf-impl.h>
21*fae548d3Szrj #include <sys/param.h>
22*fae548d3Szrj #include <assert.h>
23*fae548d3Szrj #include <string.h>
24*fae548d3Szrj #include <zlib.h>
25*fae548d3Szrj 
26*fae548d3Szrj #ifndef roundup
27*fae548d3Szrj #define roundup(x, y)  ((((x) + ((y) - 1)) / (y)) * (y))
28*fae548d3Szrj #endif
29*fae548d3Szrj 
30*fae548d3Szrj /* Make sure the ptrtab has enough space for at least one more type.
31*fae548d3Szrj 
32*fae548d3Szrj    We start with 4KiB of ptrtab, enough for a thousand types, then grow it 25%
33*fae548d3Szrj    at a time.  */
34*fae548d3Szrj 
35*fae548d3Szrj static int
ctf_grow_ptrtab(ctf_file_t * fp)36*fae548d3Szrj ctf_grow_ptrtab (ctf_file_t *fp)
37*fae548d3Szrj {
38*fae548d3Szrj   size_t new_ptrtab_len = fp->ctf_ptrtab_len;
39*fae548d3Szrj 
40*fae548d3Szrj   /* We allocate one more ptrtab entry than we need, for the initial zero,
41*fae548d3Szrj      plus one because the caller will probably allocate a new type.  */
42*fae548d3Szrj 
43*fae548d3Szrj   if (fp->ctf_ptrtab == NULL)
44*fae548d3Szrj     new_ptrtab_len = 1024;
45*fae548d3Szrj   else if ((fp->ctf_typemax + 2) > fp->ctf_ptrtab_len)
46*fae548d3Szrj     new_ptrtab_len = fp->ctf_ptrtab_len * 1.25;
47*fae548d3Szrj 
48*fae548d3Szrj   if (new_ptrtab_len != fp->ctf_ptrtab_len)
49*fae548d3Szrj     {
50*fae548d3Szrj       uint32_t *new_ptrtab;
51*fae548d3Szrj 
52*fae548d3Szrj       if ((new_ptrtab = realloc (fp->ctf_ptrtab,
53*fae548d3Szrj 				 new_ptrtab_len * sizeof (uint32_t))) == NULL)
54*fae548d3Szrj 	return (ctf_set_errno (fp, ENOMEM));
55*fae548d3Szrj 
56*fae548d3Szrj       fp->ctf_ptrtab = new_ptrtab;
57*fae548d3Szrj       memset (fp->ctf_ptrtab + fp->ctf_ptrtab_len, 0,
58*fae548d3Szrj 	      (new_ptrtab_len - fp->ctf_ptrtab_len) * sizeof (uint32_t));
59*fae548d3Szrj       fp->ctf_ptrtab_len = new_ptrtab_len;
60*fae548d3Szrj     }
61*fae548d3Szrj   return 0;
62*fae548d3Szrj }
63*fae548d3Szrj 
64*fae548d3Szrj /* To create an empty CTF container, we just declare a zeroed header and call
65*fae548d3Szrj    ctf_bufopen() on it.  If ctf_bufopen succeeds, we mark the new container r/w
66*fae548d3Szrj    and initialize the dynamic members.  We start assigning type IDs at 1 because
67*fae548d3Szrj    type ID 0 is used as a sentinel and a not-found indicator.  */
68*fae548d3Szrj 
69*fae548d3Szrj ctf_file_t *
ctf_create(int * errp)70*fae548d3Szrj ctf_create (int *errp)
71*fae548d3Szrj {
72*fae548d3Szrj   static const ctf_header_t hdr = { .cth_preamble = { CTF_MAGIC, CTF_VERSION, 0 } };
73*fae548d3Szrj 
74*fae548d3Szrj   ctf_dynhash_t *dthash;
75*fae548d3Szrj   ctf_dynhash_t *dvhash;
76*fae548d3Szrj   ctf_dynhash_t *structs = NULL, *unions = NULL, *enums = NULL, *names = NULL;
77*fae548d3Szrj   ctf_sect_t cts;
78*fae548d3Szrj   ctf_file_t *fp;
79*fae548d3Szrj 
80*fae548d3Szrj   libctf_init_debug();
81*fae548d3Szrj   dthash = ctf_dynhash_create (ctf_hash_integer, ctf_hash_eq_integer,
82*fae548d3Szrj 			       NULL, NULL);
83*fae548d3Szrj   if (dthash == NULL)
84*fae548d3Szrj     {
85*fae548d3Szrj       ctf_set_open_errno (errp, EAGAIN);
86*fae548d3Szrj       goto err;
87*fae548d3Szrj     }
88*fae548d3Szrj 
89*fae548d3Szrj   dvhash = ctf_dynhash_create (ctf_hash_string, ctf_hash_eq_string,
90*fae548d3Szrj 			       NULL, NULL);
91*fae548d3Szrj   if (dvhash == NULL)
92*fae548d3Szrj     {
93*fae548d3Szrj       ctf_set_open_errno (errp, EAGAIN);
94*fae548d3Szrj       goto err_dt;
95*fae548d3Szrj     }
96*fae548d3Szrj 
97*fae548d3Szrj   structs = ctf_dynhash_create (ctf_hash_string, ctf_hash_eq_string,
98*fae548d3Szrj 				NULL, NULL);
99*fae548d3Szrj   unions = ctf_dynhash_create (ctf_hash_string, ctf_hash_eq_string,
100*fae548d3Szrj 			       NULL, NULL);
101*fae548d3Szrj   enums = ctf_dynhash_create (ctf_hash_string, ctf_hash_eq_string,
102*fae548d3Szrj 			      NULL, NULL);
103*fae548d3Szrj   names = ctf_dynhash_create (ctf_hash_string, ctf_hash_eq_string,
104*fae548d3Szrj 			      NULL, NULL);
105*fae548d3Szrj   if (!structs || !unions || !enums || !names)
106*fae548d3Szrj     {
107*fae548d3Szrj       ctf_set_open_errno (errp, EAGAIN);
108*fae548d3Szrj       goto err_dv;
109*fae548d3Szrj     }
110*fae548d3Szrj 
111*fae548d3Szrj   cts.cts_name = _CTF_SECTION;
112*fae548d3Szrj   cts.cts_data = &hdr;
113*fae548d3Szrj   cts.cts_size = sizeof (hdr);
114*fae548d3Szrj   cts.cts_entsize = 1;
115*fae548d3Szrj 
116*fae548d3Szrj   if ((fp = ctf_bufopen_internal (&cts, NULL, NULL, NULL, 1, errp)) == NULL)
117*fae548d3Szrj     goto err_dv;
118*fae548d3Szrj 
119*fae548d3Szrj   fp->ctf_structs.ctn_writable = structs;
120*fae548d3Szrj   fp->ctf_unions.ctn_writable = unions;
121*fae548d3Szrj   fp->ctf_enums.ctn_writable = enums;
122*fae548d3Szrj   fp->ctf_names.ctn_writable = names;
123*fae548d3Szrj   fp->ctf_dthash = dthash;
124*fae548d3Szrj   fp->ctf_dvhash = dvhash;
125*fae548d3Szrj   fp->ctf_dtoldid = 0;
126*fae548d3Szrj   fp->ctf_snapshots = 1;
127*fae548d3Szrj   fp->ctf_snapshot_lu = 0;
128*fae548d3Szrj 
129*fae548d3Szrj   ctf_set_ctl_hashes (fp);
130*fae548d3Szrj   ctf_setmodel (fp, CTF_MODEL_NATIVE);
131*fae548d3Szrj   if (ctf_grow_ptrtab (fp) < 0)
132*fae548d3Szrj     {
133*fae548d3Szrj       ctf_set_open_errno (errp, ctf_errno (fp));
134*fae548d3Szrj       ctf_file_close (fp);
135*fae548d3Szrj       return NULL;
136*fae548d3Szrj     }
137*fae548d3Szrj 
138*fae548d3Szrj   return fp;
139*fae548d3Szrj 
140*fae548d3Szrj  err_dv:
141*fae548d3Szrj   ctf_dynhash_destroy (structs);
142*fae548d3Szrj   ctf_dynhash_destroy (unions);
143*fae548d3Szrj   ctf_dynhash_destroy (enums);
144*fae548d3Szrj   ctf_dynhash_destroy (names);
145*fae548d3Szrj   ctf_dynhash_destroy (dvhash);
146*fae548d3Szrj  err_dt:
147*fae548d3Szrj   ctf_dynhash_destroy (dthash);
148*fae548d3Szrj  err:
149*fae548d3Szrj   return NULL;
150*fae548d3Szrj }
151*fae548d3Szrj 
152*fae548d3Szrj static unsigned char *
ctf_copy_smembers(ctf_file_t * fp,ctf_dtdef_t * dtd,unsigned char * t)153*fae548d3Szrj ctf_copy_smembers (ctf_file_t *fp, ctf_dtdef_t *dtd, unsigned char *t)
154*fae548d3Szrj {
155*fae548d3Szrj   ctf_dmdef_t *dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
156*fae548d3Szrj   ctf_member_t ctm;
157*fae548d3Szrj 
158*fae548d3Szrj   for (; dmd != NULL; dmd = ctf_list_next (dmd))
159*fae548d3Szrj     {
160*fae548d3Szrj       ctf_member_t *copied;
161*fae548d3Szrj 
162*fae548d3Szrj       ctm.ctm_name = 0;
163*fae548d3Szrj       ctm.ctm_type = (uint32_t) dmd->dmd_type;
164*fae548d3Szrj       ctm.ctm_offset = (uint32_t) dmd->dmd_offset;
165*fae548d3Szrj 
166*fae548d3Szrj       memcpy (t, &ctm, sizeof (ctm));
167*fae548d3Szrj       copied = (ctf_member_t *) t;
168*fae548d3Szrj       if (dmd->dmd_name)
169*fae548d3Szrj 	ctf_str_add_ref (fp, dmd->dmd_name, &copied->ctm_name);
170*fae548d3Szrj 
171*fae548d3Szrj       t += sizeof (ctm);
172*fae548d3Szrj     }
173*fae548d3Szrj 
174*fae548d3Szrj   return t;
175*fae548d3Szrj }
176*fae548d3Szrj 
177*fae548d3Szrj static unsigned char *
ctf_copy_lmembers(ctf_file_t * fp,ctf_dtdef_t * dtd,unsigned char * t)178*fae548d3Szrj ctf_copy_lmembers (ctf_file_t *fp, ctf_dtdef_t *dtd, unsigned char *t)
179*fae548d3Szrj {
180*fae548d3Szrj   ctf_dmdef_t *dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
181*fae548d3Szrj   ctf_lmember_t ctlm;
182*fae548d3Szrj 
183*fae548d3Szrj   for (; dmd != NULL; dmd = ctf_list_next (dmd))
184*fae548d3Szrj     {
185*fae548d3Szrj       ctf_lmember_t *copied;
186*fae548d3Szrj 
187*fae548d3Szrj       ctlm.ctlm_name = 0;
188*fae548d3Szrj       ctlm.ctlm_type = (uint32_t) dmd->dmd_type;
189*fae548d3Szrj       ctlm.ctlm_offsethi = CTF_OFFSET_TO_LMEMHI (dmd->dmd_offset);
190*fae548d3Szrj       ctlm.ctlm_offsetlo = CTF_OFFSET_TO_LMEMLO (dmd->dmd_offset);
191*fae548d3Szrj 
192*fae548d3Szrj       memcpy (t, &ctlm, sizeof (ctlm));
193*fae548d3Szrj       copied = (ctf_lmember_t *) t;
194*fae548d3Szrj       if (dmd->dmd_name)
195*fae548d3Szrj 	ctf_str_add_ref (fp, dmd->dmd_name, &copied->ctlm_name);
196*fae548d3Szrj 
197*fae548d3Szrj       t += sizeof (ctlm);
198*fae548d3Szrj     }
199*fae548d3Szrj 
200*fae548d3Szrj   return t;
201*fae548d3Szrj }
202*fae548d3Szrj 
203*fae548d3Szrj static unsigned char *
ctf_copy_emembers(ctf_file_t * fp,ctf_dtdef_t * dtd,unsigned char * t)204*fae548d3Szrj ctf_copy_emembers (ctf_file_t *fp, ctf_dtdef_t *dtd, unsigned char *t)
205*fae548d3Szrj {
206*fae548d3Szrj   ctf_dmdef_t *dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
207*fae548d3Szrj   ctf_enum_t cte;
208*fae548d3Szrj 
209*fae548d3Szrj   for (; dmd != NULL; dmd = ctf_list_next (dmd))
210*fae548d3Szrj     {
211*fae548d3Szrj       ctf_enum_t *copied;
212*fae548d3Szrj 
213*fae548d3Szrj       cte.cte_value = dmd->dmd_value;
214*fae548d3Szrj       memcpy (t, &cte, sizeof (cte));
215*fae548d3Szrj       copied = (ctf_enum_t *) t;
216*fae548d3Szrj       ctf_str_add_ref (fp, dmd->dmd_name, &copied->cte_name);
217*fae548d3Szrj       t += sizeof (cte);
218*fae548d3Szrj     }
219*fae548d3Szrj 
220*fae548d3Szrj   return t;
221*fae548d3Szrj }
222*fae548d3Szrj 
223*fae548d3Szrj /* Sort a newly-constructed static variable array.  */
224*fae548d3Szrj 
225*fae548d3Szrj typedef struct ctf_sort_var_arg_cb
226*fae548d3Szrj {
227*fae548d3Szrj   ctf_file_t *fp;
228*fae548d3Szrj   ctf_strs_t *strtab;
229*fae548d3Szrj } ctf_sort_var_arg_cb_t;
230*fae548d3Szrj 
231*fae548d3Szrj static int
ctf_sort_var(const void * one_,const void * two_,void * arg_)232*fae548d3Szrj ctf_sort_var (const void *one_, const void *two_, void *arg_)
233*fae548d3Szrj {
234*fae548d3Szrj   const ctf_varent_t *one = one_;
235*fae548d3Szrj   const ctf_varent_t *two = two_;
236*fae548d3Szrj   ctf_sort_var_arg_cb_t *arg = arg_;
237*fae548d3Szrj 
238*fae548d3Szrj   return (strcmp (ctf_strraw_explicit (arg->fp, one->ctv_name, arg->strtab),
239*fae548d3Szrj 		  ctf_strraw_explicit (arg->fp, two->ctv_name, arg->strtab)));
240*fae548d3Szrj }
241*fae548d3Szrj 
242*fae548d3Szrj /* Compatibility: just update the threshold for ctf_discard.  */
243*fae548d3Szrj int
ctf_update(ctf_file_t * fp)244*fae548d3Szrj ctf_update (ctf_file_t *fp)
245*fae548d3Szrj {
246*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
247*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
248*fae548d3Szrj 
249*fae548d3Szrj   fp->ctf_dtoldid = fp->ctf_typemax;
250*fae548d3Szrj   return 0;
251*fae548d3Szrj }
252*fae548d3Szrj 
253*fae548d3Szrj /* If the specified CTF container is writable and has been modified, reload this
254*fae548d3Szrj    container with the updated type definitions, ready for serialization.  In
255*fae548d3Szrj    order to make this code and the rest of libctf as simple as possible, we
256*fae548d3Szrj    perform updates by taking the dynamic type definitions and creating an
257*fae548d3Szrj    in-memory CTF file containing the definitions, and then call
258*fae548d3Szrj    ctf_simple_open_internal() on it.  We perform one extra trick here for the
259*fae548d3Szrj    benefit of callers and to keep our code simple: ctf_simple_open_internal()
260*fae548d3Szrj    will return a new ctf_file_t, but we want to keep the fp constant for the
261*fae548d3Szrj    caller, so after ctf_simple_open_internal() returns, we use memcpy to swap
262*fae548d3Szrj    the interior of the old and new ctf_file_t's, and then free the old.  */
263*fae548d3Szrj int
ctf_serialize(ctf_file_t * fp)264*fae548d3Szrj ctf_serialize (ctf_file_t *fp)
265*fae548d3Szrj {
266*fae548d3Szrj   ctf_file_t ofp, *nfp;
267*fae548d3Szrj   ctf_header_t hdr, *hdrp;
268*fae548d3Szrj   ctf_dtdef_t *dtd;
269*fae548d3Szrj   ctf_dvdef_t *dvd;
270*fae548d3Szrj   ctf_varent_t *dvarents;
271*fae548d3Szrj   ctf_strs_writable_t strtab;
272*fae548d3Szrj 
273*fae548d3Szrj   unsigned char *t;
274*fae548d3Szrj   unsigned long i;
275*fae548d3Szrj   size_t buf_size, type_size, nvars;
276*fae548d3Szrj   unsigned char *buf, *newbuf;
277*fae548d3Szrj   int err;
278*fae548d3Szrj 
279*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
280*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
281*fae548d3Szrj 
282*fae548d3Szrj   /* Update required?  */
283*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_DIRTY))
284*fae548d3Szrj     return 0;
285*fae548d3Szrj 
286*fae548d3Szrj   /* Fill in an initial CTF header.  We will leave the label, object,
287*fae548d3Szrj      and function sections empty and only output a header, type section,
288*fae548d3Szrj      and string table.  The type section begins at a 4-byte aligned
289*fae548d3Szrj      boundary past the CTF header itself (at relative offset zero).  */
290*fae548d3Szrj 
291*fae548d3Szrj   memset (&hdr, 0, sizeof (hdr));
292*fae548d3Szrj   hdr.cth_magic = CTF_MAGIC;
293*fae548d3Szrj   hdr.cth_version = CTF_VERSION;
294*fae548d3Szrj 
295*fae548d3Szrj   /* Iterate through the dynamic type definition list and compute the
296*fae548d3Szrj      size of the CTF type section we will need to generate.  */
297*fae548d3Szrj 
298*fae548d3Szrj   for (type_size = 0, dtd = ctf_list_next (&fp->ctf_dtdefs);
299*fae548d3Szrj        dtd != NULL; dtd = ctf_list_next (dtd))
300*fae548d3Szrj     {
301*fae548d3Szrj       uint32_t kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
302*fae548d3Szrj       uint32_t vlen = LCTF_INFO_VLEN (fp, dtd->dtd_data.ctt_info);
303*fae548d3Szrj 
304*fae548d3Szrj       if (dtd->dtd_data.ctt_size != CTF_LSIZE_SENT)
305*fae548d3Szrj 	type_size += sizeof (ctf_stype_t);
306*fae548d3Szrj       else
307*fae548d3Szrj 	type_size += sizeof (ctf_type_t);
308*fae548d3Szrj 
309*fae548d3Szrj       switch (kind)
310*fae548d3Szrj 	{
311*fae548d3Szrj 	case CTF_K_INTEGER:
312*fae548d3Szrj 	case CTF_K_FLOAT:
313*fae548d3Szrj 	  type_size += sizeof (uint32_t);
314*fae548d3Szrj 	  break;
315*fae548d3Szrj 	case CTF_K_ARRAY:
316*fae548d3Szrj 	  type_size += sizeof (ctf_array_t);
317*fae548d3Szrj 	  break;
318*fae548d3Szrj 	case CTF_K_SLICE:
319*fae548d3Szrj 	  type_size += sizeof (ctf_slice_t);
320*fae548d3Szrj 	  break;
321*fae548d3Szrj 	case CTF_K_FUNCTION:
322*fae548d3Szrj 	  type_size += sizeof (uint32_t) * (vlen + (vlen & 1));
323*fae548d3Szrj 	  break;
324*fae548d3Szrj 	case CTF_K_STRUCT:
325*fae548d3Szrj 	case CTF_K_UNION:
326*fae548d3Szrj 	  if (dtd->dtd_data.ctt_size < CTF_LSTRUCT_THRESH)
327*fae548d3Szrj 	    type_size += sizeof (ctf_member_t) * vlen;
328*fae548d3Szrj 	  else
329*fae548d3Szrj 	    type_size += sizeof (ctf_lmember_t) * vlen;
330*fae548d3Szrj 	  break;
331*fae548d3Szrj 	case CTF_K_ENUM:
332*fae548d3Szrj 	  type_size += sizeof (ctf_enum_t) * vlen;
333*fae548d3Szrj 	  break;
334*fae548d3Szrj 	}
335*fae548d3Szrj     }
336*fae548d3Szrj 
337*fae548d3Szrj   /* Computing the number of entries in the CTF variable section is much
338*fae548d3Szrj      simpler.  */
339*fae548d3Szrj 
340*fae548d3Szrj   for (nvars = 0, dvd = ctf_list_next (&fp->ctf_dvdefs);
341*fae548d3Szrj        dvd != NULL; dvd = ctf_list_next (dvd), nvars++);
342*fae548d3Szrj 
343*fae548d3Szrj   /* Compute the size of the CTF buffer we need, sans only the string table,
344*fae548d3Szrj      then allocate a new buffer and memcpy the finished header to the start of
345*fae548d3Szrj      the buffer.  (We will adjust this later with strtab length info.)  */
346*fae548d3Szrj 
347*fae548d3Szrj   hdr.cth_typeoff = hdr.cth_varoff + (nvars * sizeof (ctf_varent_t));
348*fae548d3Szrj   hdr.cth_stroff = hdr.cth_typeoff + type_size;
349*fae548d3Szrj   hdr.cth_strlen = 0;
350*fae548d3Szrj 
351*fae548d3Szrj   buf_size = sizeof (ctf_header_t) + hdr.cth_stroff + hdr.cth_strlen;
352*fae548d3Szrj 
353*fae548d3Szrj   if ((buf = malloc (buf_size)) == NULL)
354*fae548d3Szrj     return (ctf_set_errno (fp, EAGAIN));
355*fae548d3Szrj 
356*fae548d3Szrj   memcpy (buf, &hdr, sizeof (ctf_header_t));
357*fae548d3Szrj   t = (unsigned char *) buf + sizeof (ctf_header_t) + hdr.cth_varoff;
358*fae548d3Szrj 
359*fae548d3Szrj   hdrp = (ctf_header_t *) buf;
360*fae548d3Szrj   if ((fp->ctf_flags & LCTF_CHILD) && (fp->ctf_parname != NULL))
361*fae548d3Szrj     ctf_str_add_ref (fp, fp->ctf_parname, &hdrp->cth_parname);
362*fae548d3Szrj   if (fp->ctf_cuname != NULL)
363*fae548d3Szrj     ctf_str_add_ref (fp, fp->ctf_cuname, &hdrp->cth_cuname);
364*fae548d3Szrj 
365*fae548d3Szrj   /* Work over the variable list, translating everything into ctf_varent_t's and
366*fae548d3Szrj      prepping the string table.  */
367*fae548d3Szrj 
368*fae548d3Szrj   dvarents = (ctf_varent_t *) t;
369*fae548d3Szrj   for (i = 0, dvd = ctf_list_next (&fp->ctf_dvdefs); dvd != NULL;
370*fae548d3Szrj        dvd = ctf_list_next (dvd), i++)
371*fae548d3Szrj     {
372*fae548d3Szrj       ctf_varent_t *var = &dvarents[i];
373*fae548d3Szrj 
374*fae548d3Szrj       ctf_str_add_ref (fp, dvd->dvd_name, &var->ctv_name);
375*fae548d3Szrj       var->ctv_type = dvd->dvd_type;
376*fae548d3Szrj     }
377*fae548d3Szrj   assert (i == nvars);
378*fae548d3Szrj 
379*fae548d3Szrj   t += sizeof (ctf_varent_t) * nvars;
380*fae548d3Szrj 
381*fae548d3Szrj   assert (t == (unsigned char *) buf + sizeof (ctf_header_t) + hdr.cth_typeoff);
382*fae548d3Szrj 
383*fae548d3Szrj   /* We now take a final lap through the dynamic type definition list and copy
384*fae548d3Szrj      the appropriate type records to the output buffer, noting down the
385*fae548d3Szrj      strings as we go.  */
386*fae548d3Szrj 
387*fae548d3Szrj   for (dtd = ctf_list_next (&fp->ctf_dtdefs);
388*fae548d3Szrj        dtd != NULL; dtd = ctf_list_next (dtd))
389*fae548d3Szrj     {
390*fae548d3Szrj       uint32_t kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
391*fae548d3Szrj       uint32_t vlen = LCTF_INFO_VLEN (fp, dtd->dtd_data.ctt_info);
392*fae548d3Szrj 
393*fae548d3Szrj       ctf_array_t cta;
394*fae548d3Szrj       uint32_t encoding;
395*fae548d3Szrj       size_t len;
396*fae548d3Szrj       ctf_stype_t *copied;
397*fae548d3Szrj       const char *name;
398*fae548d3Szrj 
399*fae548d3Szrj       if (dtd->dtd_data.ctt_size != CTF_LSIZE_SENT)
400*fae548d3Szrj 	len = sizeof (ctf_stype_t);
401*fae548d3Szrj       else
402*fae548d3Szrj 	len = sizeof (ctf_type_t);
403*fae548d3Szrj 
404*fae548d3Szrj       memcpy (t, &dtd->dtd_data, len);
405*fae548d3Szrj       copied = (ctf_stype_t *) t;  /* name is at the start: constant offset.  */
406*fae548d3Szrj       if (copied->ctt_name
407*fae548d3Szrj 	  && (name = ctf_strraw (fp, copied->ctt_name)) != NULL)
408*fae548d3Szrj 	ctf_str_add_ref (fp, name, &copied->ctt_name);
409*fae548d3Szrj       t += len;
410*fae548d3Szrj 
411*fae548d3Szrj       switch (kind)
412*fae548d3Szrj 	{
413*fae548d3Szrj 	case CTF_K_INTEGER:
414*fae548d3Szrj 	case CTF_K_FLOAT:
415*fae548d3Szrj 	  if (kind == CTF_K_INTEGER)
416*fae548d3Szrj 	    {
417*fae548d3Szrj 	      encoding = CTF_INT_DATA (dtd->dtd_u.dtu_enc.cte_format,
418*fae548d3Szrj 				       dtd->dtd_u.dtu_enc.cte_offset,
419*fae548d3Szrj 				       dtd->dtd_u.dtu_enc.cte_bits);
420*fae548d3Szrj 	    }
421*fae548d3Szrj 	  else
422*fae548d3Szrj 	    {
423*fae548d3Szrj 	      encoding = CTF_FP_DATA (dtd->dtd_u.dtu_enc.cte_format,
424*fae548d3Szrj 				      dtd->dtd_u.dtu_enc.cte_offset,
425*fae548d3Szrj 				      dtd->dtd_u.dtu_enc.cte_bits);
426*fae548d3Szrj 	    }
427*fae548d3Szrj 	  memcpy (t, &encoding, sizeof (encoding));
428*fae548d3Szrj 	  t += sizeof (encoding);
429*fae548d3Szrj 	  break;
430*fae548d3Szrj 
431*fae548d3Szrj 	case CTF_K_SLICE:
432*fae548d3Szrj 	  memcpy (t, &dtd->dtd_u.dtu_slice, sizeof (struct ctf_slice));
433*fae548d3Szrj 	  t += sizeof (struct ctf_slice);
434*fae548d3Szrj 	  break;
435*fae548d3Szrj 
436*fae548d3Szrj 	case CTF_K_ARRAY:
437*fae548d3Szrj 	  cta.cta_contents = (uint32_t) dtd->dtd_u.dtu_arr.ctr_contents;
438*fae548d3Szrj 	  cta.cta_index = (uint32_t) dtd->dtd_u.dtu_arr.ctr_index;
439*fae548d3Szrj 	  cta.cta_nelems = dtd->dtd_u.dtu_arr.ctr_nelems;
440*fae548d3Szrj 	  memcpy (t, &cta, sizeof (cta));
441*fae548d3Szrj 	  t += sizeof (cta);
442*fae548d3Szrj 	  break;
443*fae548d3Szrj 
444*fae548d3Szrj 	case CTF_K_FUNCTION:
445*fae548d3Szrj 	  {
446*fae548d3Szrj 	    uint32_t *argv = (uint32_t *) (uintptr_t) t;
447*fae548d3Szrj 	    uint32_t argc;
448*fae548d3Szrj 
449*fae548d3Szrj 	    for (argc = 0; argc < vlen; argc++)
450*fae548d3Szrj 	      *argv++ = (uint32_t) dtd->dtd_u.dtu_argv[argc];
451*fae548d3Szrj 
452*fae548d3Szrj 	    if (vlen & 1)
453*fae548d3Szrj 	      *argv++ = 0;	/* Pad to 4-byte boundary.  */
454*fae548d3Szrj 
455*fae548d3Szrj 	    t = (unsigned char *) argv;
456*fae548d3Szrj 	    break;
457*fae548d3Szrj 	  }
458*fae548d3Szrj 
459*fae548d3Szrj 	case CTF_K_STRUCT:
460*fae548d3Szrj 	case CTF_K_UNION:
461*fae548d3Szrj 	  if (dtd->dtd_data.ctt_size < CTF_LSTRUCT_THRESH)
462*fae548d3Szrj 	    t = ctf_copy_smembers (fp, dtd, t);
463*fae548d3Szrj 	  else
464*fae548d3Szrj 	    t = ctf_copy_lmembers (fp, dtd, t);
465*fae548d3Szrj 	  break;
466*fae548d3Szrj 
467*fae548d3Szrj 	case CTF_K_ENUM:
468*fae548d3Szrj 	  t = ctf_copy_emembers (fp, dtd, t);
469*fae548d3Szrj 	  break;
470*fae548d3Szrj 	}
471*fae548d3Szrj     }
472*fae548d3Szrj   assert (t == (unsigned char *) buf + sizeof (ctf_header_t) + hdr.cth_stroff);
473*fae548d3Szrj 
474*fae548d3Szrj   /* Construct the final string table and fill out all the string refs with the
475*fae548d3Szrj      final offsets.  Then purge the refs list, because we're about to move this
476*fae548d3Szrj      strtab onto the end of the buf, invalidating all the offsets.  */
477*fae548d3Szrj   strtab = ctf_str_write_strtab (fp);
478*fae548d3Szrj   ctf_str_purge_refs (fp);
479*fae548d3Szrj 
480*fae548d3Szrj   if (strtab.cts_strs == NULL)
481*fae548d3Szrj     {
482*fae548d3Szrj       free (buf);
483*fae548d3Szrj       return (ctf_set_errno (fp, EAGAIN));
484*fae548d3Szrj     }
485*fae548d3Szrj 
486*fae548d3Szrj   /* Now the string table is constructed, we can sort the buffer of
487*fae548d3Szrj      ctf_varent_t's.  */
488*fae548d3Szrj   ctf_sort_var_arg_cb_t sort_var_arg = { fp, (ctf_strs_t *) &strtab };
489*fae548d3Szrj   ctf_qsort_r (dvarents, nvars, sizeof (ctf_varent_t), ctf_sort_var,
490*fae548d3Szrj 	       &sort_var_arg);
491*fae548d3Szrj 
492*fae548d3Szrj   if ((newbuf = ctf_realloc (fp, buf, buf_size + strtab.cts_len)) == NULL)
493*fae548d3Szrj     {
494*fae548d3Szrj       free (buf);
495*fae548d3Szrj       free (strtab.cts_strs);
496*fae548d3Szrj       return (ctf_set_errno (fp, EAGAIN));
497*fae548d3Szrj     }
498*fae548d3Szrj   buf = newbuf;
499*fae548d3Szrj   memcpy (buf + buf_size, strtab.cts_strs, strtab.cts_len);
500*fae548d3Szrj   hdrp = (ctf_header_t *) buf;
501*fae548d3Szrj   hdrp->cth_strlen = strtab.cts_len;
502*fae548d3Szrj   buf_size += hdrp->cth_strlen;
503*fae548d3Szrj   free (strtab.cts_strs);
504*fae548d3Szrj 
505*fae548d3Szrj   /* Finally, we are ready to ctf_simple_open() the new container.  If this
506*fae548d3Szrj      is successful, we then switch nfp and fp and free the old container.  */
507*fae548d3Szrj 
508*fae548d3Szrj   if ((nfp = ctf_simple_open_internal ((char *) buf, buf_size, NULL, 0,
509*fae548d3Szrj 				       0, NULL, 0, fp->ctf_syn_ext_strtab,
510*fae548d3Szrj 				       1, &err)) == NULL)
511*fae548d3Szrj     {
512*fae548d3Szrj       free (buf);
513*fae548d3Szrj       return (ctf_set_errno (fp, err));
514*fae548d3Szrj     }
515*fae548d3Szrj 
516*fae548d3Szrj   (void) ctf_setmodel (nfp, ctf_getmodel (fp));
517*fae548d3Szrj   (void) ctf_import (nfp, fp->ctf_parent);
518*fae548d3Szrj 
519*fae548d3Szrj   nfp->ctf_refcnt = fp->ctf_refcnt;
520*fae548d3Szrj   nfp->ctf_flags |= fp->ctf_flags & ~LCTF_DIRTY;
521*fae548d3Szrj   if (nfp->ctf_dynbase == NULL)
522*fae548d3Szrj     nfp->ctf_dynbase = buf;		/* Make sure buf is freed on close.  */
523*fae548d3Szrj   nfp->ctf_dthash = fp->ctf_dthash;
524*fae548d3Szrj   nfp->ctf_dtdefs = fp->ctf_dtdefs;
525*fae548d3Szrj   nfp->ctf_dvhash = fp->ctf_dvhash;
526*fae548d3Szrj   nfp->ctf_dvdefs = fp->ctf_dvdefs;
527*fae548d3Szrj   nfp->ctf_dtoldid = fp->ctf_dtoldid;
528*fae548d3Szrj   nfp->ctf_add_processing = fp->ctf_add_processing;
529*fae548d3Szrj   nfp->ctf_snapshots = fp->ctf_snapshots + 1;
530*fae548d3Szrj   nfp->ctf_specific = fp->ctf_specific;
531*fae548d3Szrj   nfp->ctf_ptrtab = fp->ctf_ptrtab;
532*fae548d3Szrj   nfp->ctf_ptrtab_len = fp->ctf_ptrtab_len;
533*fae548d3Szrj   nfp->ctf_link_inputs = fp->ctf_link_inputs;
534*fae548d3Szrj   nfp->ctf_link_outputs = fp->ctf_link_outputs;
535*fae548d3Szrj   nfp->ctf_str_prov_offset = fp->ctf_str_prov_offset;
536*fae548d3Szrj   nfp->ctf_syn_ext_strtab = fp->ctf_syn_ext_strtab;
537*fae548d3Szrj   nfp->ctf_link_cu_mapping = fp->ctf_link_cu_mapping;
538*fae548d3Szrj   nfp->ctf_link_type_mapping = fp->ctf_link_type_mapping;
539*fae548d3Szrj   nfp->ctf_link_memb_name_changer = fp->ctf_link_memb_name_changer;
540*fae548d3Szrj   nfp->ctf_link_memb_name_changer_arg = fp->ctf_link_memb_name_changer_arg;
541*fae548d3Szrj 
542*fae548d3Szrj   nfp->ctf_snapshot_lu = fp->ctf_snapshots;
543*fae548d3Szrj 
544*fae548d3Szrj   memcpy (&nfp->ctf_lookups, fp->ctf_lookups, sizeof (fp->ctf_lookups));
545*fae548d3Szrj   nfp->ctf_structs = fp->ctf_structs;
546*fae548d3Szrj   nfp->ctf_unions = fp->ctf_unions;
547*fae548d3Szrj   nfp->ctf_enums = fp->ctf_enums;
548*fae548d3Szrj   nfp->ctf_names = fp->ctf_names;
549*fae548d3Szrj 
550*fae548d3Szrj   fp->ctf_dthash = NULL;
551*fae548d3Szrj   ctf_str_free_atoms (nfp);
552*fae548d3Szrj   nfp->ctf_str_atoms = fp->ctf_str_atoms;
553*fae548d3Szrj   nfp->ctf_prov_strtab = fp->ctf_prov_strtab;
554*fae548d3Szrj   fp->ctf_str_atoms = NULL;
555*fae548d3Szrj   fp->ctf_prov_strtab = NULL;
556*fae548d3Szrj   memset (&fp->ctf_dtdefs, 0, sizeof (ctf_list_t));
557*fae548d3Szrj   fp->ctf_add_processing = NULL;
558*fae548d3Szrj   fp->ctf_ptrtab = NULL;
559*fae548d3Szrj   fp->ctf_link_inputs = NULL;
560*fae548d3Szrj   fp->ctf_link_outputs = NULL;
561*fae548d3Szrj   fp->ctf_syn_ext_strtab = NULL;
562*fae548d3Szrj   fp->ctf_link_cu_mapping = NULL;
563*fae548d3Szrj   fp->ctf_link_type_mapping = NULL;
564*fae548d3Szrj 
565*fae548d3Szrj   fp->ctf_dvhash = NULL;
566*fae548d3Szrj   memset (&fp->ctf_dvdefs, 0, sizeof (ctf_list_t));
567*fae548d3Szrj   memset (fp->ctf_lookups, 0, sizeof (fp->ctf_lookups));
568*fae548d3Szrj   fp->ctf_structs.ctn_writable = NULL;
569*fae548d3Szrj   fp->ctf_unions.ctn_writable = NULL;
570*fae548d3Szrj   fp->ctf_enums.ctn_writable = NULL;
571*fae548d3Szrj   fp->ctf_names.ctn_writable = NULL;
572*fae548d3Szrj 
573*fae548d3Szrj   memcpy (&ofp, fp, sizeof (ctf_file_t));
574*fae548d3Szrj   memcpy (fp, nfp, sizeof (ctf_file_t));
575*fae548d3Szrj   memcpy (nfp, &ofp, sizeof (ctf_file_t));
576*fae548d3Szrj 
577*fae548d3Szrj   nfp->ctf_refcnt = 1;		/* Force nfp to be freed.  */
578*fae548d3Szrj   ctf_file_close (nfp);
579*fae548d3Szrj 
580*fae548d3Szrj   return 0;
581*fae548d3Szrj }
582*fae548d3Szrj 
583*fae548d3Szrj ctf_names_t *
ctf_name_table(ctf_file_t * fp,int kind)584*fae548d3Szrj ctf_name_table (ctf_file_t *fp, int kind)
585*fae548d3Szrj {
586*fae548d3Szrj   switch (kind)
587*fae548d3Szrj     {
588*fae548d3Szrj     case CTF_K_STRUCT:
589*fae548d3Szrj       return &fp->ctf_structs;
590*fae548d3Szrj     case CTF_K_UNION:
591*fae548d3Szrj       return &fp->ctf_unions;
592*fae548d3Szrj     case CTF_K_ENUM:
593*fae548d3Szrj       return &fp->ctf_enums;
594*fae548d3Szrj     default:
595*fae548d3Szrj       return &fp->ctf_names;
596*fae548d3Szrj     }
597*fae548d3Szrj }
598*fae548d3Szrj 
599*fae548d3Szrj int
ctf_dtd_insert(ctf_file_t * fp,ctf_dtdef_t * dtd,int kind)600*fae548d3Szrj ctf_dtd_insert (ctf_file_t *fp, ctf_dtdef_t *dtd, int kind)
601*fae548d3Szrj {
602*fae548d3Szrj   const char *name;
603*fae548d3Szrj   if (ctf_dynhash_insert (fp->ctf_dthash, (void *) dtd->dtd_type, dtd) < 0)
604*fae548d3Szrj     return -1;
605*fae548d3Szrj 
606*fae548d3Szrj   if (dtd->dtd_data.ctt_name
607*fae548d3Szrj       && (name = ctf_strraw (fp, dtd->dtd_data.ctt_name)) != NULL)
608*fae548d3Szrj     {
609*fae548d3Szrj       if (ctf_dynhash_insert (ctf_name_table (fp, kind)->ctn_writable,
610*fae548d3Szrj 			      (char *) name, (void *) dtd->dtd_type) < 0)
611*fae548d3Szrj 	{
612*fae548d3Szrj 	  ctf_dynhash_remove (fp->ctf_dthash, (void *) dtd->dtd_type);
613*fae548d3Szrj 	  return -1;
614*fae548d3Szrj 	}
615*fae548d3Szrj     }
616*fae548d3Szrj   ctf_list_append (&fp->ctf_dtdefs, dtd);
617*fae548d3Szrj   return 0;
618*fae548d3Szrj }
619*fae548d3Szrj 
620*fae548d3Szrj void
ctf_dtd_delete(ctf_file_t * fp,ctf_dtdef_t * dtd)621*fae548d3Szrj ctf_dtd_delete (ctf_file_t *fp, ctf_dtdef_t *dtd)
622*fae548d3Szrj {
623*fae548d3Szrj   ctf_dmdef_t *dmd, *nmd;
624*fae548d3Szrj   int kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
625*fae548d3Szrj   const char *name;
626*fae548d3Szrj 
627*fae548d3Szrj   ctf_dynhash_remove (fp->ctf_dthash, (void *) dtd->dtd_type);
628*fae548d3Szrj 
629*fae548d3Szrj   switch (kind)
630*fae548d3Szrj     {
631*fae548d3Szrj     case CTF_K_STRUCT:
632*fae548d3Szrj     case CTF_K_UNION:
633*fae548d3Szrj     case CTF_K_ENUM:
634*fae548d3Szrj       for (dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
635*fae548d3Szrj 	   dmd != NULL; dmd = nmd)
636*fae548d3Szrj 	{
637*fae548d3Szrj 	  if (dmd->dmd_name != NULL)
638*fae548d3Szrj 	      free (dmd->dmd_name);
639*fae548d3Szrj 	  nmd = ctf_list_next (dmd);
640*fae548d3Szrj 	  free (dmd);
641*fae548d3Szrj 	}
642*fae548d3Szrj       break;
643*fae548d3Szrj     case CTF_K_FUNCTION:
644*fae548d3Szrj       free (dtd->dtd_u.dtu_argv);
645*fae548d3Szrj       break;
646*fae548d3Szrj     }
647*fae548d3Szrj 
648*fae548d3Szrj   if (dtd->dtd_data.ctt_name
649*fae548d3Szrj       && (name = ctf_strraw (fp, dtd->dtd_data.ctt_name)) != NULL)
650*fae548d3Szrj     {
651*fae548d3Szrj       ctf_dynhash_remove (ctf_name_table (fp, kind)->ctn_writable,
652*fae548d3Szrj 			  name);
653*fae548d3Szrj       ctf_str_remove_ref (fp, name, &dtd->dtd_data.ctt_name);
654*fae548d3Szrj     }
655*fae548d3Szrj 
656*fae548d3Szrj   ctf_list_delete (&fp->ctf_dtdefs, dtd);
657*fae548d3Szrj   free (dtd);
658*fae548d3Szrj }
659*fae548d3Szrj 
660*fae548d3Szrj ctf_dtdef_t *
ctf_dtd_lookup(const ctf_file_t * fp,ctf_id_t type)661*fae548d3Szrj ctf_dtd_lookup (const ctf_file_t *fp, ctf_id_t type)
662*fae548d3Szrj {
663*fae548d3Szrj   return (ctf_dtdef_t *) ctf_dynhash_lookup (fp->ctf_dthash, (void *) type);
664*fae548d3Szrj }
665*fae548d3Szrj 
666*fae548d3Szrj ctf_dtdef_t *
ctf_dynamic_type(const ctf_file_t * fp,ctf_id_t id)667*fae548d3Szrj ctf_dynamic_type (const ctf_file_t *fp, ctf_id_t id)
668*fae548d3Szrj {
669*fae548d3Szrj   ctf_id_t idx;
670*fae548d3Szrj 
671*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
672*fae548d3Szrj     return NULL;
673*fae548d3Szrj 
674*fae548d3Szrj   if ((fp->ctf_flags & LCTF_CHILD) && LCTF_TYPE_ISPARENT (fp, id))
675*fae548d3Szrj     fp = fp->ctf_parent;
676*fae548d3Szrj 
677*fae548d3Szrj   idx = LCTF_TYPE_TO_INDEX(fp, id);
678*fae548d3Szrj 
679*fae548d3Szrj   if ((unsigned long) idx <= fp->ctf_typemax)
680*fae548d3Szrj     return ctf_dtd_lookup (fp, id);
681*fae548d3Szrj   return NULL;
682*fae548d3Szrj }
683*fae548d3Szrj 
684*fae548d3Szrj int
ctf_dvd_insert(ctf_file_t * fp,ctf_dvdef_t * dvd)685*fae548d3Szrj ctf_dvd_insert (ctf_file_t *fp, ctf_dvdef_t *dvd)
686*fae548d3Szrj {
687*fae548d3Szrj   if (ctf_dynhash_insert (fp->ctf_dvhash, dvd->dvd_name, dvd) < 0)
688*fae548d3Szrj     return -1;
689*fae548d3Szrj   ctf_list_append (&fp->ctf_dvdefs, dvd);
690*fae548d3Szrj   return 0;
691*fae548d3Szrj }
692*fae548d3Szrj 
693*fae548d3Szrj void
ctf_dvd_delete(ctf_file_t * fp,ctf_dvdef_t * dvd)694*fae548d3Szrj ctf_dvd_delete (ctf_file_t *fp, ctf_dvdef_t *dvd)
695*fae548d3Szrj {
696*fae548d3Szrj   ctf_dynhash_remove (fp->ctf_dvhash, dvd->dvd_name);
697*fae548d3Szrj   free (dvd->dvd_name);
698*fae548d3Szrj 
699*fae548d3Szrj   ctf_list_delete (&fp->ctf_dvdefs, dvd);
700*fae548d3Szrj   free (dvd);
701*fae548d3Szrj }
702*fae548d3Szrj 
703*fae548d3Szrj ctf_dvdef_t *
ctf_dvd_lookup(const ctf_file_t * fp,const char * name)704*fae548d3Szrj ctf_dvd_lookup (const ctf_file_t *fp, const char *name)
705*fae548d3Szrj {
706*fae548d3Szrj   return (ctf_dvdef_t *) ctf_dynhash_lookup (fp->ctf_dvhash, name);
707*fae548d3Szrj }
708*fae548d3Szrj 
709*fae548d3Szrj /* Discard all of the dynamic type definitions and variable definitions that
710*fae548d3Szrj    have been added to the container since the last call to ctf_update().  We
711*fae548d3Szrj    locate such types by scanning the dtd list and deleting elements that have
712*fae548d3Szrj    type IDs greater than ctf_dtoldid, which is set by ctf_update(), above, and
713*fae548d3Szrj    by scanning the variable list and deleting elements that have update IDs
714*fae548d3Szrj    equal to the current value of the last-update snapshot count (indicating that
715*fae548d3Szrj    they were added after the most recent call to ctf_update()).  */
716*fae548d3Szrj int
ctf_discard(ctf_file_t * fp)717*fae548d3Szrj ctf_discard (ctf_file_t *fp)
718*fae548d3Szrj {
719*fae548d3Szrj   ctf_snapshot_id_t last_update =
720*fae548d3Szrj     { fp->ctf_dtoldid,
721*fae548d3Szrj       fp->ctf_snapshot_lu + 1 };
722*fae548d3Szrj 
723*fae548d3Szrj   /* Update required?  */
724*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_DIRTY))
725*fae548d3Szrj     return 0;
726*fae548d3Szrj 
727*fae548d3Szrj   return (ctf_rollback (fp, last_update));
728*fae548d3Szrj }
729*fae548d3Szrj 
730*fae548d3Szrj ctf_snapshot_id_t
ctf_snapshot(ctf_file_t * fp)731*fae548d3Szrj ctf_snapshot (ctf_file_t *fp)
732*fae548d3Szrj {
733*fae548d3Szrj   ctf_snapshot_id_t snapid;
734*fae548d3Szrj   snapid.dtd_id = fp->ctf_typemax;
735*fae548d3Szrj   snapid.snapshot_id = fp->ctf_snapshots++;
736*fae548d3Szrj   return snapid;
737*fae548d3Szrj }
738*fae548d3Szrj 
739*fae548d3Szrj /* Like ctf_discard(), only discards everything after a particular ID.  */
740*fae548d3Szrj int
ctf_rollback(ctf_file_t * fp,ctf_snapshot_id_t id)741*fae548d3Szrj ctf_rollback (ctf_file_t *fp, ctf_snapshot_id_t id)
742*fae548d3Szrj {
743*fae548d3Szrj   ctf_dtdef_t *dtd, *ntd;
744*fae548d3Szrj   ctf_dvdef_t *dvd, *nvd;
745*fae548d3Szrj 
746*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
747*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
748*fae548d3Szrj 
749*fae548d3Szrj   if (fp->ctf_snapshot_lu >= id.snapshot_id)
750*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_OVERROLLBACK));
751*fae548d3Szrj 
752*fae548d3Szrj   for (dtd = ctf_list_next (&fp->ctf_dtdefs); dtd != NULL; dtd = ntd)
753*fae548d3Szrj     {
754*fae548d3Szrj       int kind;
755*fae548d3Szrj       const char *name;
756*fae548d3Szrj 
757*fae548d3Szrj       ntd = ctf_list_next (dtd);
758*fae548d3Szrj 
759*fae548d3Szrj       if (LCTF_TYPE_TO_INDEX (fp, dtd->dtd_type) <= id.dtd_id)
760*fae548d3Szrj 	continue;
761*fae548d3Szrj 
762*fae548d3Szrj       kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
763*fae548d3Szrj 
764*fae548d3Szrj       if (dtd->dtd_data.ctt_name
765*fae548d3Szrj 	  && (name = ctf_strraw (fp, dtd->dtd_data.ctt_name)) != NULL)
766*fae548d3Szrj 	{
767*fae548d3Szrj 	  ctf_dynhash_remove (ctf_name_table (fp, kind)->ctn_writable,
768*fae548d3Szrj 			      name);
769*fae548d3Szrj 	  ctf_str_remove_ref (fp, name, &dtd->dtd_data.ctt_name);
770*fae548d3Szrj 	}
771*fae548d3Szrj 
772*fae548d3Szrj       ctf_dynhash_remove (fp->ctf_dthash, (void *) dtd->dtd_type);
773*fae548d3Szrj       ctf_dtd_delete (fp, dtd);
774*fae548d3Szrj     }
775*fae548d3Szrj 
776*fae548d3Szrj   for (dvd = ctf_list_next (&fp->ctf_dvdefs); dvd != NULL; dvd = nvd)
777*fae548d3Szrj     {
778*fae548d3Szrj       nvd = ctf_list_next (dvd);
779*fae548d3Szrj 
780*fae548d3Szrj       if (dvd->dvd_snapshots <= id.snapshot_id)
781*fae548d3Szrj 	continue;
782*fae548d3Szrj 
783*fae548d3Szrj       ctf_dvd_delete (fp, dvd);
784*fae548d3Szrj     }
785*fae548d3Szrj 
786*fae548d3Szrj   fp->ctf_typemax = id.dtd_id;
787*fae548d3Szrj   fp->ctf_snapshots = id.snapshot_id;
788*fae548d3Szrj 
789*fae548d3Szrj   if (fp->ctf_snapshots == fp->ctf_snapshot_lu)
790*fae548d3Szrj     fp->ctf_flags &= ~LCTF_DIRTY;
791*fae548d3Szrj 
792*fae548d3Szrj   return 0;
793*fae548d3Szrj }
794*fae548d3Szrj 
795*fae548d3Szrj static ctf_id_t
ctf_add_generic(ctf_file_t * fp,uint32_t flag,const char * name,int kind,ctf_dtdef_t ** rp)796*fae548d3Szrj ctf_add_generic (ctf_file_t *fp, uint32_t flag, const char *name, int kind,
797*fae548d3Szrj 		 ctf_dtdef_t **rp)
798*fae548d3Szrj {
799*fae548d3Szrj   ctf_dtdef_t *dtd;
800*fae548d3Szrj   ctf_id_t type;
801*fae548d3Szrj 
802*fae548d3Szrj   if (flag != CTF_ADD_NONROOT && flag != CTF_ADD_ROOT)
803*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
804*fae548d3Szrj 
805*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
806*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
807*fae548d3Szrj 
808*fae548d3Szrj   if (LCTF_INDEX_TO_TYPE (fp, fp->ctf_typemax, 1) >= CTF_MAX_TYPE)
809*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_FULL));
810*fae548d3Szrj 
811*fae548d3Szrj   if (LCTF_INDEX_TO_TYPE (fp, fp->ctf_typemax, 1) == (CTF_MAX_PTYPE - 1))
812*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_FULL));
813*fae548d3Szrj 
814*fae548d3Szrj   /* Make sure ptrtab always grows to be big enough for all types.  */
815*fae548d3Szrj   if (ctf_grow_ptrtab (fp) < 0)
816*fae548d3Szrj       return CTF_ERR;		/* errno is set for us. */
817*fae548d3Szrj 
818*fae548d3Szrj   if ((dtd = malloc (sizeof (ctf_dtdef_t))) == NULL)
819*fae548d3Szrj     return (ctf_set_errno (fp, EAGAIN));
820*fae548d3Szrj 
821*fae548d3Szrj   type = ++fp->ctf_typemax;
822*fae548d3Szrj   type = LCTF_INDEX_TO_TYPE (fp, type, (fp->ctf_flags & LCTF_CHILD));
823*fae548d3Szrj 
824*fae548d3Szrj   memset (dtd, 0, sizeof (ctf_dtdef_t));
825*fae548d3Szrj   dtd->dtd_data.ctt_name = ctf_str_add_ref (fp, name, &dtd->dtd_data.ctt_name);
826*fae548d3Szrj   dtd->dtd_type = type;
827*fae548d3Szrj 
828*fae548d3Szrj   if (dtd->dtd_data.ctt_name == 0 && name != NULL && name[0] != '\0')
829*fae548d3Szrj     {
830*fae548d3Szrj       free (dtd);
831*fae548d3Szrj       return (ctf_set_errno (fp, EAGAIN));
832*fae548d3Szrj     }
833*fae548d3Szrj 
834*fae548d3Szrj   if (ctf_dtd_insert (fp, dtd, kind) < 0)
835*fae548d3Szrj     {
836*fae548d3Szrj       free (dtd);
837*fae548d3Szrj       return CTF_ERR;			/* errno is set for us.  */
838*fae548d3Szrj     }
839*fae548d3Szrj   fp->ctf_flags |= LCTF_DIRTY;
840*fae548d3Szrj 
841*fae548d3Szrj   *rp = dtd;
842*fae548d3Szrj   return type;
843*fae548d3Szrj }
844*fae548d3Szrj 
845*fae548d3Szrj /* When encoding integer sizes, we want to convert a byte count in the range
846*fae548d3Szrj    1-8 to the closest power of 2 (e.g. 3->4, 5->8, etc).  The clp2() function
847*fae548d3Szrj    is a clever implementation from "Hacker's Delight" by Henry Warren, Jr.  */
848*fae548d3Szrj static size_t
clp2(size_t x)849*fae548d3Szrj clp2 (size_t x)
850*fae548d3Szrj {
851*fae548d3Szrj   x--;
852*fae548d3Szrj 
853*fae548d3Szrj   x |= (x >> 1);
854*fae548d3Szrj   x |= (x >> 2);
855*fae548d3Szrj   x |= (x >> 4);
856*fae548d3Szrj   x |= (x >> 8);
857*fae548d3Szrj   x |= (x >> 16);
858*fae548d3Szrj 
859*fae548d3Szrj   return (x + 1);
860*fae548d3Szrj }
861*fae548d3Szrj 
862*fae548d3Szrj static ctf_id_t
ctf_add_encoded(ctf_file_t * fp,uint32_t flag,const char * name,const ctf_encoding_t * ep,uint32_t kind)863*fae548d3Szrj ctf_add_encoded (ctf_file_t *fp, uint32_t flag,
864*fae548d3Szrj 		 const char *name, const ctf_encoding_t *ep, uint32_t kind)
865*fae548d3Szrj {
866*fae548d3Szrj   ctf_dtdef_t *dtd;
867*fae548d3Szrj   ctf_id_t type;
868*fae548d3Szrj 
869*fae548d3Szrj   if (ep == NULL)
870*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
871*fae548d3Szrj 
872*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, name, kind, &dtd)) == CTF_ERR)
873*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
874*fae548d3Szrj 
875*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (kind, flag, 0);
876*fae548d3Szrj   dtd->dtd_data.ctt_size = clp2 (P2ROUNDUP (ep->cte_bits, CHAR_BIT)
877*fae548d3Szrj 				 / CHAR_BIT);
878*fae548d3Szrj   dtd->dtd_u.dtu_enc = *ep;
879*fae548d3Szrj 
880*fae548d3Szrj   return type;
881*fae548d3Szrj }
882*fae548d3Szrj 
883*fae548d3Szrj static ctf_id_t
ctf_add_reftype(ctf_file_t * fp,uint32_t flag,ctf_id_t ref,uint32_t kind)884*fae548d3Szrj ctf_add_reftype (ctf_file_t *fp, uint32_t flag, ctf_id_t ref, uint32_t kind)
885*fae548d3Szrj {
886*fae548d3Szrj   ctf_dtdef_t *dtd;
887*fae548d3Szrj   ctf_id_t type;
888*fae548d3Szrj   ctf_file_t *tmp = fp;
889*fae548d3Szrj   int child = fp->ctf_flags & LCTF_CHILD;
890*fae548d3Szrj 
891*fae548d3Szrj   if (ref == CTF_ERR || ref > CTF_MAX_TYPE)
892*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
893*fae548d3Szrj 
894*fae548d3Szrj   if (ctf_lookup_by_id (&tmp, ref) == NULL)
895*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
896*fae548d3Szrj 
897*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, NULL, kind, &dtd)) == CTF_ERR)
898*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
899*fae548d3Szrj 
900*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (kind, flag, 0);
901*fae548d3Szrj   dtd->dtd_data.ctt_type = (uint32_t) ref;
902*fae548d3Szrj 
903*fae548d3Szrj   if (kind != CTF_K_POINTER)
904*fae548d3Szrj     return type;
905*fae548d3Szrj 
906*fae548d3Szrj   /* If we are adding a pointer, update the ptrtab, both the directly pointed-to
907*fae548d3Szrj      type and (if an anonymous typedef node is being pointed at) the type that
908*fae548d3Szrj      points at too.  Note that ctf_typemax is at this point one higher than we
909*fae548d3Szrj      want to check against, because it's just been incremented for the addition
910*fae548d3Szrj      of this type.  */
911*fae548d3Szrj 
912*fae548d3Szrj   uint32_t type_idx = LCTF_TYPE_TO_INDEX (fp, type);
913*fae548d3Szrj   uint32_t ref_idx = LCTF_TYPE_TO_INDEX (fp, ref);
914*fae548d3Szrj 
915*fae548d3Szrj   if (LCTF_TYPE_ISCHILD (fp, ref) == child
916*fae548d3Szrj       && ref_idx < fp->ctf_typemax)
917*fae548d3Szrj     {
918*fae548d3Szrj       fp->ctf_ptrtab[ref_idx] = type_idx;
919*fae548d3Szrj 
920*fae548d3Szrj       ctf_id_t refref_idx = LCTF_TYPE_TO_INDEX (fp, dtd->dtd_data.ctt_type);
921*fae548d3Szrj 
922*fae548d3Szrj       if (tmp == fp
923*fae548d3Szrj 	  && (LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info) == CTF_K_TYPEDEF)
924*fae548d3Szrj 	  && strcmp (ctf_strptr (fp, dtd->dtd_data.ctt_name), "") == 0
925*fae548d3Szrj 	  && refref_idx < fp->ctf_typemax)
926*fae548d3Szrj 	fp->ctf_ptrtab[refref_idx] = type_idx;
927*fae548d3Szrj     }
928*fae548d3Szrj 
929*fae548d3Szrj   return type;
930*fae548d3Szrj }
931*fae548d3Szrj 
932*fae548d3Szrj ctf_id_t
ctf_add_slice(ctf_file_t * fp,uint32_t flag,ctf_id_t ref,const ctf_encoding_t * ep)933*fae548d3Szrj ctf_add_slice (ctf_file_t *fp, uint32_t flag, ctf_id_t ref,
934*fae548d3Szrj 	       const ctf_encoding_t *ep)
935*fae548d3Szrj {
936*fae548d3Szrj   ctf_dtdef_t *dtd;
937*fae548d3Szrj   ctf_id_t type;
938*fae548d3Szrj   int kind;
939*fae548d3Szrj   const ctf_type_t *tp;
940*fae548d3Szrj   ctf_file_t *tmp = fp;
941*fae548d3Szrj 
942*fae548d3Szrj   if (ep == NULL)
943*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
944*fae548d3Szrj 
945*fae548d3Szrj   if ((ep->cte_bits > 255) || (ep->cte_offset > 255))
946*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_SLICEOVERFLOW));
947*fae548d3Szrj 
948*fae548d3Szrj   if (ref == CTF_ERR || ref > CTF_MAX_TYPE)
949*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
950*fae548d3Szrj 
951*fae548d3Szrj   if ((tp = ctf_lookup_by_id (&tmp, ref)) == NULL)
952*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
953*fae548d3Szrj 
954*fae548d3Szrj   kind = ctf_type_kind_unsliced (tmp, ref);
955*fae548d3Szrj   if ((kind != CTF_K_INTEGER) && (kind != CTF_K_FLOAT) &&
956*fae548d3Szrj       (kind != CTF_K_ENUM))
957*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_NOTINTFP));
958*fae548d3Szrj 
959*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, NULL, CTF_K_SLICE, &dtd)) == CTF_ERR)
960*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
961*fae548d3Szrj 
962*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_SLICE, flag, 0);
963*fae548d3Szrj   dtd->dtd_data.ctt_size = clp2 (P2ROUNDUP (ep->cte_bits, CHAR_BIT)
964*fae548d3Szrj 				 / CHAR_BIT);
965*fae548d3Szrj   dtd->dtd_u.dtu_slice.cts_type = ref;
966*fae548d3Szrj   dtd->dtd_u.dtu_slice.cts_bits = ep->cte_bits;
967*fae548d3Szrj   dtd->dtd_u.dtu_slice.cts_offset = ep->cte_offset;
968*fae548d3Szrj 
969*fae548d3Szrj   return type;
970*fae548d3Szrj }
971*fae548d3Szrj 
972*fae548d3Szrj ctf_id_t
ctf_add_integer(ctf_file_t * fp,uint32_t flag,const char * name,const ctf_encoding_t * ep)973*fae548d3Szrj ctf_add_integer (ctf_file_t *fp, uint32_t flag,
974*fae548d3Szrj 		 const char *name, const ctf_encoding_t *ep)
975*fae548d3Szrj {
976*fae548d3Szrj   return (ctf_add_encoded (fp, flag, name, ep, CTF_K_INTEGER));
977*fae548d3Szrj }
978*fae548d3Szrj 
979*fae548d3Szrj ctf_id_t
ctf_add_float(ctf_file_t * fp,uint32_t flag,const char * name,const ctf_encoding_t * ep)980*fae548d3Szrj ctf_add_float (ctf_file_t *fp, uint32_t flag,
981*fae548d3Szrj 	       const char *name, const ctf_encoding_t *ep)
982*fae548d3Szrj {
983*fae548d3Szrj   return (ctf_add_encoded (fp, flag, name, ep, CTF_K_FLOAT));
984*fae548d3Szrj }
985*fae548d3Szrj 
986*fae548d3Szrj ctf_id_t
ctf_add_pointer(ctf_file_t * fp,uint32_t flag,ctf_id_t ref)987*fae548d3Szrj ctf_add_pointer (ctf_file_t *fp, uint32_t flag, ctf_id_t ref)
988*fae548d3Szrj {
989*fae548d3Szrj   return (ctf_add_reftype (fp, flag, ref, CTF_K_POINTER));
990*fae548d3Szrj }
991*fae548d3Szrj 
992*fae548d3Szrj ctf_id_t
ctf_add_array(ctf_file_t * fp,uint32_t flag,const ctf_arinfo_t * arp)993*fae548d3Szrj ctf_add_array (ctf_file_t *fp, uint32_t flag, const ctf_arinfo_t *arp)
994*fae548d3Szrj {
995*fae548d3Szrj   ctf_dtdef_t *dtd;
996*fae548d3Szrj   ctf_id_t type;
997*fae548d3Szrj   ctf_file_t *tmp = fp;
998*fae548d3Szrj 
999*fae548d3Szrj   if (arp == NULL)
1000*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
1001*fae548d3Szrj 
1002*fae548d3Szrj   if (ctf_lookup_by_id (&tmp, arp->ctr_contents) == NULL)
1003*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1004*fae548d3Szrj 
1005*fae548d3Szrj   tmp = fp;
1006*fae548d3Szrj   if (ctf_lookup_by_id (&tmp, arp->ctr_index) == NULL)
1007*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1008*fae548d3Szrj 
1009*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, NULL, CTF_K_ARRAY, &dtd)) == CTF_ERR)
1010*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1011*fae548d3Szrj 
1012*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_ARRAY, flag, 0);
1013*fae548d3Szrj   dtd->dtd_data.ctt_size = 0;
1014*fae548d3Szrj   dtd->dtd_u.dtu_arr = *arp;
1015*fae548d3Szrj 
1016*fae548d3Szrj   return type;
1017*fae548d3Szrj }
1018*fae548d3Szrj 
1019*fae548d3Szrj int
ctf_set_array(ctf_file_t * fp,ctf_id_t type,const ctf_arinfo_t * arp)1020*fae548d3Szrj ctf_set_array (ctf_file_t *fp, ctf_id_t type, const ctf_arinfo_t *arp)
1021*fae548d3Szrj {
1022*fae548d3Szrj   ctf_dtdef_t *dtd = ctf_dtd_lookup (fp, type);
1023*fae548d3Szrj 
1024*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
1025*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
1026*fae548d3Szrj 
1027*fae548d3Szrj   if (dtd == NULL
1028*fae548d3Szrj       || LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info) != CTF_K_ARRAY)
1029*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_BADID));
1030*fae548d3Szrj 
1031*fae548d3Szrj   fp->ctf_flags |= LCTF_DIRTY;
1032*fae548d3Szrj   dtd->dtd_u.dtu_arr = *arp;
1033*fae548d3Szrj 
1034*fae548d3Szrj   return 0;
1035*fae548d3Szrj }
1036*fae548d3Szrj 
1037*fae548d3Szrj ctf_id_t
ctf_add_function(ctf_file_t * fp,uint32_t flag,const ctf_funcinfo_t * ctc,const ctf_id_t * argv)1038*fae548d3Szrj ctf_add_function (ctf_file_t *fp, uint32_t flag,
1039*fae548d3Szrj 		  const ctf_funcinfo_t *ctc, const ctf_id_t *argv)
1040*fae548d3Szrj {
1041*fae548d3Szrj   ctf_dtdef_t *dtd;
1042*fae548d3Szrj   ctf_id_t type;
1043*fae548d3Szrj   uint32_t vlen;
1044*fae548d3Szrj   ctf_id_t *vdat = NULL;
1045*fae548d3Szrj   ctf_file_t *tmp = fp;
1046*fae548d3Szrj   size_t i;
1047*fae548d3Szrj 
1048*fae548d3Szrj   if (ctc == NULL || (ctc->ctc_flags & ~CTF_FUNC_VARARG) != 0
1049*fae548d3Szrj       || (ctc->ctc_argc != 0 && argv == NULL))
1050*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
1051*fae548d3Szrj 
1052*fae548d3Szrj   vlen = ctc->ctc_argc;
1053*fae548d3Szrj   if (ctc->ctc_flags & CTF_FUNC_VARARG)
1054*fae548d3Szrj     vlen++;	       /* Add trailing zero to indicate varargs (see below).  */
1055*fae548d3Szrj 
1056*fae548d3Szrj   if (ctf_lookup_by_id (&tmp, ctc->ctc_return) == NULL)
1057*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1058*fae548d3Szrj 
1059*fae548d3Szrj   for (i = 0; i < ctc->ctc_argc; i++)
1060*fae548d3Szrj     {
1061*fae548d3Szrj       tmp = fp;
1062*fae548d3Szrj       if (ctf_lookup_by_id (&tmp, argv[i]) == NULL)
1063*fae548d3Szrj 	return CTF_ERR;		/* errno is set for us.  */
1064*fae548d3Szrj     }
1065*fae548d3Szrj 
1066*fae548d3Szrj   if (vlen > CTF_MAX_VLEN)
1067*fae548d3Szrj     return (ctf_set_errno (fp, EOVERFLOW));
1068*fae548d3Szrj 
1069*fae548d3Szrj   if (vlen != 0 && (vdat = malloc (sizeof (ctf_id_t) * vlen)) == NULL)
1070*fae548d3Szrj     return (ctf_set_errno (fp, EAGAIN));
1071*fae548d3Szrj 
1072*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, NULL, CTF_K_FUNCTION,
1073*fae548d3Szrj 			       &dtd)) == CTF_ERR)
1074*fae548d3Szrj     {
1075*fae548d3Szrj       free (vdat);
1076*fae548d3Szrj       return CTF_ERR;		   /* errno is set for us.  */
1077*fae548d3Szrj     }
1078*fae548d3Szrj 
1079*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_FUNCTION, flag, vlen);
1080*fae548d3Szrj   dtd->dtd_data.ctt_type = (uint32_t) ctc->ctc_return;
1081*fae548d3Szrj 
1082*fae548d3Szrj   memcpy (vdat, argv, sizeof (ctf_id_t) * ctc->ctc_argc);
1083*fae548d3Szrj   if (ctc->ctc_flags & CTF_FUNC_VARARG)
1084*fae548d3Szrj     vdat[vlen - 1] = 0;		   /* Add trailing zero to indicate varargs.  */
1085*fae548d3Szrj   dtd->dtd_u.dtu_argv = vdat;
1086*fae548d3Szrj 
1087*fae548d3Szrj   return type;
1088*fae548d3Szrj }
1089*fae548d3Szrj 
1090*fae548d3Szrj ctf_id_t
ctf_add_struct_sized(ctf_file_t * fp,uint32_t flag,const char * name,size_t size)1091*fae548d3Szrj ctf_add_struct_sized (ctf_file_t *fp, uint32_t flag, const char *name,
1092*fae548d3Szrj 		      size_t size)
1093*fae548d3Szrj {
1094*fae548d3Szrj   ctf_dtdef_t *dtd;
1095*fae548d3Szrj   ctf_id_t type = 0;
1096*fae548d3Szrj 
1097*fae548d3Szrj   /* Promote forwards to structs.  */
1098*fae548d3Szrj 
1099*fae548d3Szrj   if (name != NULL)
1100*fae548d3Szrj     type = ctf_lookup_by_rawname (fp, CTF_K_STRUCT, name);
1101*fae548d3Szrj 
1102*fae548d3Szrj   if (type != 0 && ctf_type_kind (fp, type) == CTF_K_FORWARD)
1103*fae548d3Szrj     dtd = ctf_dtd_lookup (fp, type);
1104*fae548d3Szrj   else if ((type = ctf_add_generic (fp, flag, name, CTF_K_STRUCT,
1105*fae548d3Szrj 				    &dtd)) == CTF_ERR)
1106*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1107*fae548d3Szrj 
1108*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_STRUCT, flag, 0);
1109*fae548d3Szrj 
1110*fae548d3Szrj   if (size > CTF_MAX_SIZE)
1111*fae548d3Szrj     {
1112*fae548d3Szrj       dtd->dtd_data.ctt_size = CTF_LSIZE_SENT;
1113*fae548d3Szrj       dtd->dtd_data.ctt_lsizehi = CTF_SIZE_TO_LSIZE_HI (size);
1114*fae548d3Szrj       dtd->dtd_data.ctt_lsizelo = CTF_SIZE_TO_LSIZE_LO (size);
1115*fae548d3Szrj     }
1116*fae548d3Szrj   else
1117*fae548d3Szrj     dtd->dtd_data.ctt_size = (uint32_t) size;
1118*fae548d3Szrj 
1119*fae548d3Szrj   return type;
1120*fae548d3Szrj }
1121*fae548d3Szrj 
1122*fae548d3Szrj ctf_id_t
ctf_add_struct(ctf_file_t * fp,uint32_t flag,const char * name)1123*fae548d3Szrj ctf_add_struct (ctf_file_t *fp, uint32_t flag, const char *name)
1124*fae548d3Szrj {
1125*fae548d3Szrj   return (ctf_add_struct_sized (fp, flag, name, 0));
1126*fae548d3Szrj }
1127*fae548d3Szrj 
1128*fae548d3Szrj ctf_id_t
ctf_add_union_sized(ctf_file_t * fp,uint32_t flag,const char * name,size_t size)1129*fae548d3Szrj ctf_add_union_sized (ctf_file_t *fp, uint32_t flag, const char *name,
1130*fae548d3Szrj 		     size_t size)
1131*fae548d3Szrj {
1132*fae548d3Szrj   ctf_dtdef_t *dtd;
1133*fae548d3Szrj   ctf_id_t type = 0;
1134*fae548d3Szrj 
1135*fae548d3Szrj   /* Promote forwards to unions.  */
1136*fae548d3Szrj   if (name != NULL)
1137*fae548d3Szrj     type = ctf_lookup_by_rawname (fp, CTF_K_UNION, name);
1138*fae548d3Szrj 
1139*fae548d3Szrj   if (type != 0 && ctf_type_kind (fp, type) == CTF_K_FORWARD)
1140*fae548d3Szrj     dtd = ctf_dtd_lookup (fp, type);
1141*fae548d3Szrj   else if ((type = ctf_add_generic (fp, flag, name, CTF_K_UNION,
1142*fae548d3Szrj 				    &dtd)) == CTF_ERR)
1143*fae548d3Szrj     return CTF_ERR;		/* errno is set for us */
1144*fae548d3Szrj 
1145*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_UNION, flag, 0);
1146*fae548d3Szrj 
1147*fae548d3Szrj   if (size > CTF_MAX_SIZE)
1148*fae548d3Szrj     {
1149*fae548d3Szrj       dtd->dtd_data.ctt_size = CTF_LSIZE_SENT;
1150*fae548d3Szrj       dtd->dtd_data.ctt_lsizehi = CTF_SIZE_TO_LSIZE_HI (size);
1151*fae548d3Szrj       dtd->dtd_data.ctt_lsizelo = CTF_SIZE_TO_LSIZE_LO (size);
1152*fae548d3Szrj     }
1153*fae548d3Szrj   else
1154*fae548d3Szrj     dtd->dtd_data.ctt_size = (uint32_t) size;
1155*fae548d3Szrj 
1156*fae548d3Szrj   return type;
1157*fae548d3Szrj }
1158*fae548d3Szrj 
1159*fae548d3Szrj ctf_id_t
ctf_add_union(ctf_file_t * fp,uint32_t flag,const char * name)1160*fae548d3Szrj ctf_add_union (ctf_file_t *fp, uint32_t flag, const char *name)
1161*fae548d3Szrj {
1162*fae548d3Szrj   return (ctf_add_union_sized (fp, flag, name, 0));
1163*fae548d3Szrj }
1164*fae548d3Szrj 
1165*fae548d3Szrj ctf_id_t
ctf_add_enum(ctf_file_t * fp,uint32_t flag,const char * name)1166*fae548d3Szrj ctf_add_enum (ctf_file_t *fp, uint32_t flag, const char *name)
1167*fae548d3Szrj {
1168*fae548d3Szrj   ctf_dtdef_t *dtd;
1169*fae548d3Szrj   ctf_id_t type = 0;
1170*fae548d3Szrj 
1171*fae548d3Szrj   /* Promote forwards to enums.  */
1172*fae548d3Szrj   if (name != NULL)
1173*fae548d3Szrj     type = ctf_lookup_by_rawname (fp, CTF_K_ENUM, name);
1174*fae548d3Szrj 
1175*fae548d3Szrj   if (type != 0 && ctf_type_kind (fp, type) == CTF_K_FORWARD)
1176*fae548d3Szrj     dtd = ctf_dtd_lookup (fp, type);
1177*fae548d3Szrj   else if ((type = ctf_add_generic (fp, flag, name, CTF_K_ENUM,
1178*fae548d3Szrj 				    &dtd)) == CTF_ERR)
1179*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1180*fae548d3Szrj 
1181*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_ENUM, flag, 0);
1182*fae548d3Szrj   dtd->dtd_data.ctt_size = fp->ctf_dmodel->ctd_int;
1183*fae548d3Szrj 
1184*fae548d3Szrj   return type;
1185*fae548d3Szrj }
1186*fae548d3Szrj 
1187*fae548d3Szrj ctf_id_t
ctf_add_enum_encoded(ctf_file_t * fp,uint32_t flag,const char * name,const ctf_encoding_t * ep)1188*fae548d3Szrj ctf_add_enum_encoded (ctf_file_t *fp, uint32_t flag, const char *name,
1189*fae548d3Szrj 		      const ctf_encoding_t *ep)
1190*fae548d3Szrj {
1191*fae548d3Szrj   ctf_id_t type = 0;
1192*fae548d3Szrj 
1193*fae548d3Szrj   /* First, create the enum if need be, using most of the same machinery as
1194*fae548d3Szrj      ctf_add_enum(), to ensure that we do not allow things past that are not
1195*fae548d3Szrj      enums or forwards to them.  (This includes other slices: you cannot slice a
1196*fae548d3Szrj      slice, which would be a useless thing to do anyway.)  */
1197*fae548d3Szrj 
1198*fae548d3Szrj   if (name != NULL)
1199*fae548d3Szrj     type = ctf_lookup_by_rawname (fp, CTF_K_ENUM, name);
1200*fae548d3Szrj 
1201*fae548d3Szrj   if (type != 0)
1202*fae548d3Szrj     {
1203*fae548d3Szrj       if ((ctf_type_kind (fp, type) != CTF_K_FORWARD) &&
1204*fae548d3Szrj 	  (ctf_type_kind_unsliced (fp, type) != CTF_K_ENUM))
1205*fae548d3Szrj 	return (ctf_set_errno (fp, ECTF_NOTINTFP));
1206*fae548d3Szrj     }
1207*fae548d3Szrj   else if ((type = ctf_add_enum (fp, flag, name)) == CTF_ERR)
1208*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1209*fae548d3Szrj 
1210*fae548d3Szrj   /* Now attach a suitable slice to it.  */
1211*fae548d3Szrj 
1212*fae548d3Szrj   return ctf_add_slice (fp, flag, type, ep);
1213*fae548d3Szrj }
1214*fae548d3Szrj 
1215*fae548d3Szrj ctf_id_t
ctf_add_forward(ctf_file_t * fp,uint32_t flag,const char * name,uint32_t kind)1216*fae548d3Szrj ctf_add_forward (ctf_file_t *fp, uint32_t flag, const char *name,
1217*fae548d3Szrj 		 uint32_t kind)
1218*fae548d3Szrj {
1219*fae548d3Szrj   ctf_dtdef_t *dtd;
1220*fae548d3Szrj   ctf_id_t type = 0;
1221*fae548d3Szrj 
1222*fae548d3Szrj   if (kind != CTF_K_STRUCT && kind != CTF_K_UNION && kind != CTF_K_ENUM)
1223*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_NOTSUE));
1224*fae548d3Szrj 
1225*fae548d3Szrj   /* If the type is already defined or exists as a forward tag, just
1226*fae548d3Szrj      return the ctf_id_t of the existing definition.  */
1227*fae548d3Szrj 
1228*fae548d3Szrj   if (name != NULL)
1229*fae548d3Szrj     type = ctf_lookup_by_rawname (fp, kind, name);
1230*fae548d3Szrj 
1231*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, name, CTF_K_FORWARD,&dtd)) == CTF_ERR)
1232*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1233*fae548d3Szrj 
1234*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_FORWARD, flag, 0);
1235*fae548d3Szrj   dtd->dtd_data.ctt_type = kind;
1236*fae548d3Szrj 
1237*fae548d3Szrj   return type;
1238*fae548d3Szrj }
1239*fae548d3Szrj 
1240*fae548d3Szrj ctf_id_t
ctf_add_typedef(ctf_file_t * fp,uint32_t flag,const char * name,ctf_id_t ref)1241*fae548d3Szrj ctf_add_typedef (ctf_file_t *fp, uint32_t flag, const char *name,
1242*fae548d3Szrj 		 ctf_id_t ref)
1243*fae548d3Szrj {
1244*fae548d3Szrj   ctf_dtdef_t *dtd;
1245*fae548d3Szrj   ctf_id_t type;
1246*fae548d3Szrj   ctf_file_t *tmp = fp;
1247*fae548d3Szrj 
1248*fae548d3Szrj   if (ref == CTF_ERR || ref > CTF_MAX_TYPE)
1249*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
1250*fae548d3Szrj 
1251*fae548d3Szrj   if (ctf_lookup_by_id (&tmp, ref) == NULL)
1252*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1253*fae548d3Szrj 
1254*fae548d3Szrj   if ((type = ctf_add_generic (fp, flag, name, CTF_K_TYPEDEF,
1255*fae548d3Szrj 			       &dtd)) == CTF_ERR)
1256*fae548d3Szrj     return CTF_ERR;		/* errno is set for us.  */
1257*fae548d3Szrj 
1258*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (CTF_K_TYPEDEF, flag, 0);
1259*fae548d3Szrj   dtd->dtd_data.ctt_type = (uint32_t) ref;
1260*fae548d3Szrj 
1261*fae548d3Szrj   return type;
1262*fae548d3Szrj }
1263*fae548d3Szrj 
1264*fae548d3Szrj ctf_id_t
ctf_add_volatile(ctf_file_t * fp,uint32_t flag,ctf_id_t ref)1265*fae548d3Szrj ctf_add_volatile (ctf_file_t *fp, uint32_t flag, ctf_id_t ref)
1266*fae548d3Szrj {
1267*fae548d3Szrj   return (ctf_add_reftype (fp, flag, ref, CTF_K_VOLATILE));
1268*fae548d3Szrj }
1269*fae548d3Szrj 
1270*fae548d3Szrj ctf_id_t
ctf_add_const(ctf_file_t * fp,uint32_t flag,ctf_id_t ref)1271*fae548d3Szrj ctf_add_const (ctf_file_t *fp, uint32_t flag, ctf_id_t ref)
1272*fae548d3Szrj {
1273*fae548d3Szrj   return (ctf_add_reftype (fp, flag, ref, CTF_K_CONST));
1274*fae548d3Szrj }
1275*fae548d3Szrj 
1276*fae548d3Szrj ctf_id_t
ctf_add_restrict(ctf_file_t * fp,uint32_t flag,ctf_id_t ref)1277*fae548d3Szrj ctf_add_restrict (ctf_file_t *fp, uint32_t flag, ctf_id_t ref)
1278*fae548d3Szrj {
1279*fae548d3Szrj   return (ctf_add_reftype (fp, flag, ref, CTF_K_RESTRICT));
1280*fae548d3Szrj }
1281*fae548d3Szrj 
1282*fae548d3Szrj int
ctf_add_enumerator(ctf_file_t * fp,ctf_id_t enid,const char * name,int value)1283*fae548d3Szrj ctf_add_enumerator (ctf_file_t *fp, ctf_id_t enid, const char *name,
1284*fae548d3Szrj 		    int value)
1285*fae548d3Szrj {
1286*fae548d3Szrj   ctf_dtdef_t *dtd = ctf_dtd_lookup (fp, enid);
1287*fae548d3Szrj   ctf_dmdef_t *dmd;
1288*fae548d3Szrj 
1289*fae548d3Szrj   uint32_t kind, vlen, root;
1290*fae548d3Szrj   char *s;
1291*fae548d3Szrj 
1292*fae548d3Szrj   if (name == NULL)
1293*fae548d3Szrj     return (ctf_set_errno (fp, EINVAL));
1294*fae548d3Szrj 
1295*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
1296*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
1297*fae548d3Szrj 
1298*fae548d3Szrj   if (dtd == NULL)
1299*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_BADID));
1300*fae548d3Szrj 
1301*fae548d3Szrj   kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
1302*fae548d3Szrj   root = LCTF_INFO_ISROOT (fp, dtd->dtd_data.ctt_info);
1303*fae548d3Szrj   vlen = LCTF_INFO_VLEN (fp, dtd->dtd_data.ctt_info);
1304*fae548d3Szrj 
1305*fae548d3Szrj   if (kind != CTF_K_ENUM)
1306*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_NOTENUM));
1307*fae548d3Szrj 
1308*fae548d3Szrj   if (vlen == CTF_MAX_VLEN)
1309*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_DTFULL));
1310*fae548d3Szrj 
1311*fae548d3Szrj   for (dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
1312*fae548d3Szrj        dmd != NULL; dmd = ctf_list_next (dmd))
1313*fae548d3Szrj     {
1314*fae548d3Szrj       if (strcmp (dmd->dmd_name, name) == 0)
1315*fae548d3Szrj 	return (ctf_set_errno (fp, ECTF_DUPLICATE));
1316*fae548d3Szrj     }
1317*fae548d3Szrj 
1318*fae548d3Szrj   if ((dmd = malloc (sizeof (ctf_dmdef_t))) == NULL)
1319*fae548d3Szrj     return (ctf_set_errno (fp, EAGAIN));
1320*fae548d3Szrj 
1321*fae548d3Szrj   if ((s = strdup (name)) == NULL)
1322*fae548d3Szrj     {
1323*fae548d3Szrj       free (dmd);
1324*fae548d3Szrj       return (ctf_set_errno (fp, EAGAIN));
1325*fae548d3Szrj     }
1326*fae548d3Szrj 
1327*fae548d3Szrj   dmd->dmd_name = s;
1328*fae548d3Szrj   dmd->dmd_type = CTF_ERR;
1329*fae548d3Szrj   dmd->dmd_offset = 0;
1330*fae548d3Szrj   dmd->dmd_value = value;
1331*fae548d3Szrj 
1332*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (kind, root, vlen + 1);
1333*fae548d3Szrj   ctf_list_append (&dtd->dtd_u.dtu_members, dmd);
1334*fae548d3Szrj 
1335*fae548d3Szrj   fp->ctf_flags |= LCTF_DIRTY;
1336*fae548d3Szrj 
1337*fae548d3Szrj   return 0;
1338*fae548d3Szrj }
1339*fae548d3Szrj 
1340*fae548d3Szrj int
ctf_add_member_offset(ctf_file_t * fp,ctf_id_t souid,const char * name,ctf_id_t type,unsigned long bit_offset)1341*fae548d3Szrj ctf_add_member_offset (ctf_file_t *fp, ctf_id_t souid, const char *name,
1342*fae548d3Szrj 		       ctf_id_t type, unsigned long bit_offset)
1343*fae548d3Szrj {
1344*fae548d3Szrj   ctf_dtdef_t *dtd = ctf_dtd_lookup (fp, souid);
1345*fae548d3Szrj   ctf_dmdef_t *dmd;
1346*fae548d3Szrj 
1347*fae548d3Szrj   ssize_t msize, malign, ssize;
1348*fae548d3Szrj   uint32_t kind, vlen, root;
1349*fae548d3Szrj   char *s = NULL;
1350*fae548d3Szrj 
1351*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
1352*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
1353*fae548d3Szrj 
1354*fae548d3Szrj   if (dtd == NULL)
1355*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_BADID));
1356*fae548d3Szrj 
1357*fae548d3Szrj   kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
1358*fae548d3Szrj   root = LCTF_INFO_ISROOT (fp, dtd->dtd_data.ctt_info);
1359*fae548d3Szrj   vlen = LCTF_INFO_VLEN (fp, dtd->dtd_data.ctt_info);
1360*fae548d3Szrj 
1361*fae548d3Szrj   if (kind != CTF_K_STRUCT && kind != CTF_K_UNION)
1362*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_NOTSOU));
1363*fae548d3Szrj 
1364*fae548d3Szrj   if (vlen == CTF_MAX_VLEN)
1365*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_DTFULL));
1366*fae548d3Szrj 
1367*fae548d3Szrj   if (name != NULL)
1368*fae548d3Szrj     {
1369*fae548d3Szrj       for (dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
1370*fae548d3Szrj 	   dmd != NULL; dmd = ctf_list_next (dmd))
1371*fae548d3Szrj 	{
1372*fae548d3Szrj 	  if (dmd->dmd_name != NULL && strcmp (dmd->dmd_name, name) == 0)
1373*fae548d3Szrj 	    return (ctf_set_errno (fp, ECTF_DUPLICATE));
1374*fae548d3Szrj 	}
1375*fae548d3Szrj     }
1376*fae548d3Szrj 
1377*fae548d3Szrj   if ((msize = ctf_type_size (fp, type)) < 0 ||
1378*fae548d3Szrj       (malign = ctf_type_align (fp, type)) < 0)
1379*fae548d3Szrj     return -1;			/* errno is set for us.  */
1380*fae548d3Szrj 
1381*fae548d3Szrj   if ((dmd = malloc (sizeof (ctf_dmdef_t))) == NULL)
1382*fae548d3Szrj     return (ctf_set_errno (fp, EAGAIN));
1383*fae548d3Szrj 
1384*fae548d3Szrj   if (name != NULL && (s = strdup (name)) == NULL)
1385*fae548d3Szrj     {
1386*fae548d3Szrj       free (dmd);
1387*fae548d3Szrj       return (ctf_set_errno (fp, EAGAIN));
1388*fae548d3Szrj     }
1389*fae548d3Szrj 
1390*fae548d3Szrj   dmd->dmd_name = s;
1391*fae548d3Szrj   dmd->dmd_type = type;
1392*fae548d3Szrj   dmd->dmd_value = -1;
1393*fae548d3Szrj 
1394*fae548d3Szrj   if (kind == CTF_K_STRUCT && vlen != 0)
1395*fae548d3Szrj     {
1396*fae548d3Szrj       if (bit_offset == (unsigned long) - 1)
1397*fae548d3Szrj 	{
1398*fae548d3Szrj 	  /* Natural alignment.  */
1399*fae548d3Szrj 
1400*fae548d3Szrj 	  ctf_dmdef_t *lmd = ctf_list_prev (&dtd->dtd_u.dtu_members);
1401*fae548d3Szrj 	  ctf_id_t ltype = ctf_type_resolve (fp, lmd->dmd_type);
1402*fae548d3Szrj 	  size_t off = lmd->dmd_offset;
1403*fae548d3Szrj 
1404*fae548d3Szrj 	  ctf_encoding_t linfo;
1405*fae548d3Szrj 	  ssize_t lsize;
1406*fae548d3Szrj 
1407*fae548d3Szrj 	  if (ctf_type_encoding (fp, ltype, &linfo) == 0)
1408*fae548d3Szrj 	    off += linfo.cte_bits;
1409*fae548d3Szrj 	  else if ((lsize = ctf_type_size (fp, ltype)) > 0)
1410*fae548d3Szrj 	    off += lsize * CHAR_BIT;
1411*fae548d3Szrj 
1412*fae548d3Szrj 	  /* Round up the offset of the end of the last member to
1413*fae548d3Szrj 	     the next byte boundary, convert 'off' to bytes, and
1414*fae548d3Szrj 	     then round it up again to the next multiple of the
1415*fae548d3Szrj 	     alignment required by the new member.  Finally,
1416*fae548d3Szrj 	     convert back to bits and store the result in
1417*fae548d3Szrj 	     dmd_offset.  Technically we could do more efficient
1418*fae548d3Szrj 	     packing if the new member is a bit-field, but we're
1419*fae548d3Szrj 	     the "compiler" and ANSI says we can do as we choose.  */
1420*fae548d3Szrj 
1421*fae548d3Szrj 	  off = roundup (off, CHAR_BIT) / CHAR_BIT;
1422*fae548d3Szrj 	  off = roundup (off, MAX (malign, 1));
1423*fae548d3Szrj 	  dmd->dmd_offset = off * CHAR_BIT;
1424*fae548d3Szrj 	  ssize = off + msize;
1425*fae548d3Szrj 	}
1426*fae548d3Szrj       else
1427*fae548d3Szrj 	{
1428*fae548d3Szrj 	  /* Specified offset in bits.  */
1429*fae548d3Szrj 
1430*fae548d3Szrj 	  dmd->dmd_offset = bit_offset;
1431*fae548d3Szrj 	  ssize = ctf_get_ctt_size (fp, &dtd->dtd_data, NULL, NULL);
1432*fae548d3Szrj 	  ssize = MAX (ssize, ((signed) bit_offset / CHAR_BIT) + msize);
1433*fae548d3Szrj 	}
1434*fae548d3Szrj     }
1435*fae548d3Szrj   else
1436*fae548d3Szrj     {
1437*fae548d3Szrj       dmd->dmd_offset = 0;
1438*fae548d3Szrj       ssize = ctf_get_ctt_size (fp, &dtd->dtd_data, NULL, NULL);
1439*fae548d3Szrj       ssize = MAX (ssize, msize);
1440*fae548d3Szrj     }
1441*fae548d3Szrj 
1442*fae548d3Szrj   if ((size_t) ssize > CTF_MAX_SIZE)
1443*fae548d3Szrj     {
1444*fae548d3Szrj       dtd->dtd_data.ctt_size = CTF_LSIZE_SENT;
1445*fae548d3Szrj       dtd->dtd_data.ctt_lsizehi = CTF_SIZE_TO_LSIZE_HI (ssize);
1446*fae548d3Szrj       dtd->dtd_data.ctt_lsizelo = CTF_SIZE_TO_LSIZE_LO (ssize);
1447*fae548d3Szrj     }
1448*fae548d3Szrj   else
1449*fae548d3Szrj     dtd->dtd_data.ctt_size = (uint32_t) ssize;
1450*fae548d3Szrj 
1451*fae548d3Szrj   dtd->dtd_data.ctt_info = CTF_TYPE_INFO (kind, root, vlen + 1);
1452*fae548d3Szrj   ctf_list_append (&dtd->dtd_u.dtu_members, dmd);
1453*fae548d3Szrj 
1454*fae548d3Szrj   fp->ctf_flags |= LCTF_DIRTY;
1455*fae548d3Szrj   return 0;
1456*fae548d3Szrj }
1457*fae548d3Szrj 
1458*fae548d3Szrj int
ctf_add_member_encoded(ctf_file_t * fp,ctf_id_t souid,const char * name,ctf_id_t type,unsigned long bit_offset,const ctf_encoding_t encoding)1459*fae548d3Szrj ctf_add_member_encoded (ctf_file_t *fp, ctf_id_t souid, const char *name,
1460*fae548d3Szrj 			ctf_id_t type, unsigned long bit_offset,
1461*fae548d3Szrj 			const ctf_encoding_t encoding)
1462*fae548d3Szrj {
1463*fae548d3Szrj   ctf_dtdef_t *dtd = ctf_dtd_lookup (fp, type);
1464*fae548d3Szrj   int kind = LCTF_INFO_KIND (fp, dtd->dtd_data.ctt_info);
1465*fae548d3Szrj   int otype = type;
1466*fae548d3Szrj 
1467*fae548d3Szrj   if ((kind != CTF_K_INTEGER) && (kind != CTF_K_FLOAT) && (kind != CTF_K_ENUM))
1468*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_NOTINTFP));
1469*fae548d3Szrj 
1470*fae548d3Szrj   if ((type = ctf_add_slice (fp, CTF_ADD_NONROOT, otype, &encoding)) == CTF_ERR)
1471*fae548d3Szrj     return -1;			/* errno is set for us.  */
1472*fae548d3Szrj 
1473*fae548d3Szrj   return ctf_add_member_offset (fp, souid, name, type, bit_offset);
1474*fae548d3Szrj }
1475*fae548d3Szrj 
1476*fae548d3Szrj int
ctf_add_member(ctf_file_t * fp,ctf_id_t souid,const char * name,ctf_id_t type)1477*fae548d3Szrj ctf_add_member (ctf_file_t *fp, ctf_id_t souid, const char *name,
1478*fae548d3Szrj 		ctf_id_t type)
1479*fae548d3Szrj {
1480*fae548d3Szrj   return ctf_add_member_offset (fp, souid, name, type, (unsigned long) - 1);
1481*fae548d3Szrj }
1482*fae548d3Szrj 
1483*fae548d3Szrj int
ctf_add_variable(ctf_file_t * fp,const char * name,ctf_id_t ref)1484*fae548d3Szrj ctf_add_variable (ctf_file_t *fp, const char *name, ctf_id_t ref)
1485*fae548d3Szrj {
1486*fae548d3Szrj   ctf_dvdef_t *dvd;
1487*fae548d3Szrj   ctf_file_t *tmp = fp;
1488*fae548d3Szrj 
1489*fae548d3Szrj   if (!(fp->ctf_flags & LCTF_RDWR))
1490*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_RDONLY));
1491*fae548d3Szrj 
1492*fae548d3Szrj   if (ctf_dvd_lookup (fp, name) != NULL)
1493*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_DUPLICATE));
1494*fae548d3Szrj 
1495*fae548d3Szrj   if (ctf_lookup_by_id (&tmp, ref) == NULL)
1496*fae548d3Szrj     return -1;			/* errno is set for us.  */
1497*fae548d3Szrj 
1498*fae548d3Szrj   /* Make sure this type is representable.  */
1499*fae548d3Szrj   if ((ctf_type_resolve (fp, ref) == CTF_ERR)
1500*fae548d3Szrj       && (ctf_errno (fp) == ECTF_NONREPRESENTABLE))
1501*fae548d3Szrj     return -1;
1502*fae548d3Szrj 
1503*fae548d3Szrj   if ((dvd = malloc (sizeof (ctf_dvdef_t))) == NULL)
1504*fae548d3Szrj     return (ctf_set_errno (fp, EAGAIN));
1505*fae548d3Szrj 
1506*fae548d3Szrj   if (name != NULL && (dvd->dvd_name = strdup (name)) == NULL)
1507*fae548d3Szrj     {
1508*fae548d3Szrj       free (dvd);
1509*fae548d3Szrj       return (ctf_set_errno (fp, EAGAIN));
1510*fae548d3Szrj     }
1511*fae548d3Szrj   dvd->dvd_type = ref;
1512*fae548d3Szrj   dvd->dvd_snapshots = fp->ctf_snapshots;
1513*fae548d3Szrj 
1514*fae548d3Szrj   if (ctf_dvd_insert (fp, dvd) < 0)
1515*fae548d3Szrj     {
1516*fae548d3Szrj       free (dvd->dvd_name);
1517*fae548d3Szrj       free (dvd);
1518*fae548d3Szrj       return -1;			/* errno is set for us.  */
1519*fae548d3Szrj     }
1520*fae548d3Szrj 
1521*fae548d3Szrj   fp->ctf_flags |= LCTF_DIRTY;
1522*fae548d3Szrj   return 0;
1523*fae548d3Szrj }
1524*fae548d3Szrj 
1525*fae548d3Szrj static int
enumcmp(const char * name,int value,void * arg)1526*fae548d3Szrj enumcmp (const char *name, int value, void *arg)
1527*fae548d3Szrj {
1528*fae548d3Szrj   ctf_bundle_t *ctb = arg;
1529*fae548d3Szrj   int bvalue;
1530*fae548d3Szrj 
1531*fae548d3Szrj   if (ctf_enum_value (ctb->ctb_file, ctb->ctb_type, name, &bvalue) < 0)
1532*fae548d3Szrj     {
1533*fae548d3Szrj       ctf_dprintf ("Conflict due to member %s iteration error: %s.\n", name,
1534*fae548d3Szrj 		   ctf_errmsg (ctf_errno (ctb->ctb_file)));
1535*fae548d3Szrj       return 1;
1536*fae548d3Szrj     }
1537*fae548d3Szrj   if (value != bvalue)
1538*fae548d3Szrj     {
1539*fae548d3Szrj       ctf_dprintf ("Conflict due to value change: %i versus %i\n",
1540*fae548d3Szrj 		   value, bvalue);
1541*fae548d3Szrj       return 1;
1542*fae548d3Szrj     }
1543*fae548d3Szrj   return 0;
1544*fae548d3Szrj }
1545*fae548d3Szrj 
1546*fae548d3Szrj static int
enumadd(const char * name,int value,void * arg)1547*fae548d3Szrj enumadd (const char *name, int value, void *arg)
1548*fae548d3Szrj {
1549*fae548d3Szrj   ctf_bundle_t *ctb = arg;
1550*fae548d3Szrj 
1551*fae548d3Szrj   return (ctf_add_enumerator (ctb->ctb_file, ctb->ctb_type,
1552*fae548d3Szrj 			      name, value) < 0);
1553*fae548d3Szrj }
1554*fae548d3Szrj 
1555*fae548d3Szrj static int
membcmp(const char * name,ctf_id_t type _libctf_unused_,unsigned long offset,void * arg)1556*fae548d3Szrj membcmp (const char *name, ctf_id_t type _libctf_unused_, unsigned long offset,
1557*fae548d3Szrj 	 void *arg)
1558*fae548d3Szrj {
1559*fae548d3Szrj   ctf_bundle_t *ctb = arg;
1560*fae548d3Szrj   ctf_membinfo_t ctm;
1561*fae548d3Szrj 
1562*fae548d3Szrj   if (ctf_member_info (ctb->ctb_file, ctb->ctb_type, name, &ctm) < 0)
1563*fae548d3Szrj     {
1564*fae548d3Szrj       ctf_dprintf ("Conflict due to member %s iteration error: %s.\n", name,
1565*fae548d3Szrj 		   ctf_errmsg (ctf_errno (ctb->ctb_file)));
1566*fae548d3Szrj       return 1;
1567*fae548d3Szrj     }
1568*fae548d3Szrj   if (ctm.ctm_offset != offset)
1569*fae548d3Szrj     {
1570*fae548d3Szrj       ctf_dprintf ("Conflict due to member %s offset change: "
1571*fae548d3Szrj 		   "%lx versus %lx\n", name, ctm.ctm_offset, offset);
1572*fae548d3Szrj       return 1;
1573*fae548d3Szrj     }
1574*fae548d3Szrj   return 0;
1575*fae548d3Szrj }
1576*fae548d3Szrj 
1577*fae548d3Szrj static int
membadd(const char * name,ctf_id_t type,unsigned long offset,void * arg)1578*fae548d3Szrj membadd (const char *name, ctf_id_t type, unsigned long offset, void *arg)
1579*fae548d3Szrj {
1580*fae548d3Szrj   ctf_bundle_t *ctb = arg;
1581*fae548d3Szrj   ctf_dmdef_t *dmd;
1582*fae548d3Szrj   char *s = NULL;
1583*fae548d3Szrj 
1584*fae548d3Szrj   if ((dmd = malloc (sizeof (ctf_dmdef_t))) == NULL)
1585*fae548d3Szrj     return (ctf_set_errno (ctb->ctb_file, EAGAIN));
1586*fae548d3Szrj 
1587*fae548d3Szrj   if (name != NULL && (s = strdup (name)) == NULL)
1588*fae548d3Szrj     {
1589*fae548d3Szrj       free (dmd);
1590*fae548d3Szrj       return (ctf_set_errno (ctb->ctb_file, EAGAIN));
1591*fae548d3Szrj     }
1592*fae548d3Szrj 
1593*fae548d3Szrj   /* For now, dmd_type is copied as the src_fp's type; it is reset to an
1594*fae548d3Szrj     equivalent dst_fp type by a final loop in ctf_add_type(), below.  */
1595*fae548d3Szrj   dmd->dmd_name = s;
1596*fae548d3Szrj   dmd->dmd_type = type;
1597*fae548d3Szrj   dmd->dmd_offset = offset;
1598*fae548d3Szrj   dmd->dmd_value = -1;
1599*fae548d3Szrj 
1600*fae548d3Szrj   ctf_list_append (&ctb->ctb_dtd->dtd_u.dtu_members, dmd);
1601*fae548d3Szrj 
1602*fae548d3Szrj   ctb->ctb_file->ctf_flags |= LCTF_DIRTY;
1603*fae548d3Szrj   return 0;
1604*fae548d3Szrj }
1605*fae548d3Szrj 
1606*fae548d3Szrj /* The ctf_add_type routine is used to copy a type from a source CTF container
1607*fae548d3Szrj    to a dynamic destination container.  This routine operates recursively by
1608*fae548d3Szrj    following the source type's links and embedded member types.  If the
1609*fae548d3Szrj    destination container already contains a named type which has the same
1610*fae548d3Szrj    attributes, then we succeed and return this type but no changes occur.  */
1611*fae548d3Szrj static ctf_id_t
ctf_add_type_internal(ctf_file_t * dst_fp,ctf_file_t * src_fp,ctf_id_t src_type,ctf_file_t * proc_tracking_fp)1612*fae548d3Szrj ctf_add_type_internal (ctf_file_t *dst_fp, ctf_file_t *src_fp, ctf_id_t src_type,
1613*fae548d3Szrj 		       ctf_file_t *proc_tracking_fp)
1614*fae548d3Szrj {
1615*fae548d3Szrj   ctf_id_t dst_type = CTF_ERR;
1616*fae548d3Szrj   uint32_t dst_kind = CTF_K_UNKNOWN;
1617*fae548d3Szrj   ctf_file_t *tmp_fp = dst_fp;
1618*fae548d3Szrj   ctf_id_t tmp;
1619*fae548d3Szrj 
1620*fae548d3Szrj   const char *name;
1621*fae548d3Szrj   uint32_t kind, forward_kind, flag, vlen;
1622*fae548d3Szrj 
1623*fae548d3Szrj   const ctf_type_t *src_tp, *dst_tp;
1624*fae548d3Szrj   ctf_bundle_t src, dst;
1625*fae548d3Szrj   ctf_encoding_t src_en, dst_en;
1626*fae548d3Szrj   ctf_arinfo_t src_ar, dst_ar;
1627*fae548d3Szrj 
1628*fae548d3Szrj   ctf_funcinfo_t ctc;
1629*fae548d3Szrj 
1630*fae548d3Szrj   ctf_id_t orig_src_type = src_type;
1631*fae548d3Szrj 
1632*fae548d3Szrj   if (!(dst_fp->ctf_flags & LCTF_RDWR))
1633*fae548d3Szrj     return (ctf_set_errno (dst_fp, ECTF_RDONLY));
1634*fae548d3Szrj 
1635*fae548d3Szrj   if ((src_tp = ctf_lookup_by_id (&src_fp, src_type)) == NULL)
1636*fae548d3Szrj     return (ctf_set_errno (dst_fp, ctf_errno (src_fp)));
1637*fae548d3Szrj 
1638*fae548d3Szrj   if ((ctf_type_resolve (src_fp, src_type) == CTF_ERR)
1639*fae548d3Szrj       && (ctf_errno (src_fp) == ECTF_NONREPRESENTABLE))
1640*fae548d3Szrj     return (ctf_set_errno (dst_fp, ECTF_NONREPRESENTABLE));
1641*fae548d3Szrj 
1642*fae548d3Szrj   name = ctf_strptr (src_fp, src_tp->ctt_name);
1643*fae548d3Szrj   kind = LCTF_INFO_KIND (src_fp, src_tp->ctt_info);
1644*fae548d3Szrj   flag = LCTF_INFO_ISROOT (src_fp, src_tp->ctt_info);
1645*fae548d3Szrj   vlen = LCTF_INFO_VLEN (src_fp, src_tp->ctt_info);
1646*fae548d3Szrj 
1647*fae548d3Szrj   /* If this is a type we are currently in the middle of adding, hand it
1648*fae548d3Szrj      straight back.  (This lets us handle self-referential structures without
1649*fae548d3Szrj      considering forwards and empty structures the same as their completed
1650*fae548d3Szrj      forms.)  */
1651*fae548d3Szrj 
1652*fae548d3Szrj   tmp = ctf_type_mapping (src_fp, src_type, &tmp_fp);
1653*fae548d3Szrj 
1654*fae548d3Szrj   if (tmp != 0)
1655*fae548d3Szrj     {
1656*fae548d3Szrj       if (ctf_dynhash_lookup (proc_tracking_fp->ctf_add_processing,
1657*fae548d3Szrj 			      (void *) (uintptr_t) src_type))
1658*fae548d3Szrj 	return tmp;
1659*fae548d3Szrj 
1660*fae548d3Szrj       /* If this type has already been added from this container, and is the same
1661*fae548d3Szrj 	 kind and (if a struct or union) has the same number of members, hand it
1662*fae548d3Szrj 	 straight back.  */
1663*fae548d3Szrj 
1664*fae548d3Szrj       if ((ctf_type_kind_unsliced (tmp_fp, tmp) == (int) kind)
1665*fae548d3Szrj 	  && (kind == CTF_K_STRUCT || kind == CTF_K_UNION
1666*fae548d3Szrj 	      || kind == CTF_K_ENUM))
1667*fae548d3Szrj 	{
1668*fae548d3Szrj 	  if ((dst_tp = ctf_lookup_by_id (&tmp_fp, dst_type)) != NULL)
1669*fae548d3Szrj 	    if (vlen == LCTF_INFO_VLEN (tmp_fp, dst_tp->ctt_info))
1670*fae548d3Szrj 	      return tmp;
1671*fae548d3Szrj 	}
1672*fae548d3Szrj     }
1673*fae548d3Szrj 
1674*fae548d3Szrj   forward_kind = kind;
1675*fae548d3Szrj   if (kind == CTF_K_FORWARD)
1676*fae548d3Szrj     forward_kind = src_tp->ctt_type;
1677*fae548d3Szrj 
1678*fae548d3Szrj   /* If the source type has a name and is a root type (visible at the
1679*fae548d3Szrj      top-level scope), lookup the name in the destination container and
1680*fae548d3Szrj      verify that it is of the same kind before we do anything else.  */
1681*fae548d3Szrj 
1682*fae548d3Szrj   if ((flag & CTF_ADD_ROOT) && name[0] != '\0'
1683*fae548d3Szrj       && (tmp = ctf_lookup_by_rawname (dst_fp, forward_kind, name)) != 0)
1684*fae548d3Szrj     {
1685*fae548d3Szrj       dst_type = tmp;
1686*fae548d3Szrj       dst_kind = ctf_type_kind_unsliced (dst_fp, dst_type);
1687*fae548d3Szrj     }
1688*fae548d3Szrj 
1689*fae548d3Szrj   /* If an identically named dst_type exists, fail with ECTF_CONFLICT
1690*fae548d3Szrj      unless dst_type is a forward declaration and src_type is a struct,
1691*fae548d3Szrj      union, or enum (i.e. the definition of the previous forward decl).
1692*fae548d3Szrj 
1693*fae548d3Szrj      We also allow addition in the opposite order (addition of a forward when a
1694*fae548d3Szrj      struct, union, or enum already exists), which is a NOP and returns the
1695*fae548d3Szrj      already-present struct, union, or enum.  */
1696*fae548d3Szrj 
1697*fae548d3Szrj   if (dst_type != CTF_ERR && dst_kind != kind)
1698*fae548d3Szrj     {
1699*fae548d3Szrj       if (kind == CTF_K_FORWARD
1700*fae548d3Szrj 	  && (dst_kind == CTF_K_ENUM || dst_kind == CTF_K_STRUCT
1701*fae548d3Szrj 	      || dst_kind == CTF_K_UNION))
1702*fae548d3Szrj 	{
1703*fae548d3Szrj 	  ctf_add_type_mapping (src_fp, src_type, dst_fp, dst_type);
1704*fae548d3Szrj 	  return dst_type;
1705*fae548d3Szrj 	}
1706*fae548d3Szrj 
1707*fae548d3Szrj       if (dst_kind != CTF_K_FORWARD
1708*fae548d3Szrj 	  || (kind != CTF_K_ENUM && kind != CTF_K_STRUCT
1709*fae548d3Szrj 	      && kind != CTF_K_UNION))
1710*fae548d3Szrj 	{
1711*fae548d3Szrj 	  ctf_dprintf ("Conflict for type %s: kinds differ, new: %i; "
1712*fae548d3Szrj 		       "old (ID %lx): %i\n", name, kind, dst_type, dst_kind);
1713*fae548d3Szrj 	  return (ctf_set_errno (dst_fp, ECTF_CONFLICT));
1714*fae548d3Szrj 	}
1715*fae548d3Szrj     }
1716*fae548d3Szrj 
1717*fae548d3Szrj   /* We take special action for an integer, float, or slice since it is
1718*fae548d3Szrj      described not only by its name but also its encoding.  For integers,
1719*fae548d3Szrj      bit-fields exploit this degeneracy.  */
1720*fae548d3Szrj 
1721*fae548d3Szrj   if (kind == CTF_K_INTEGER || kind == CTF_K_FLOAT || kind == CTF_K_SLICE)
1722*fae548d3Szrj     {
1723*fae548d3Szrj       if (ctf_type_encoding (src_fp, src_type, &src_en) != 0)
1724*fae548d3Szrj 	return (ctf_set_errno (dst_fp, ctf_errno (src_fp)));
1725*fae548d3Szrj 
1726*fae548d3Szrj       if (dst_type != CTF_ERR)
1727*fae548d3Szrj 	{
1728*fae548d3Szrj 	  ctf_file_t *fp = dst_fp;
1729*fae548d3Szrj 
1730*fae548d3Szrj 	  if ((dst_tp = ctf_lookup_by_id (&fp, dst_type)) == NULL)
1731*fae548d3Szrj 	    return CTF_ERR;
1732*fae548d3Szrj 
1733*fae548d3Szrj 	  if (ctf_type_encoding (dst_fp, dst_type, &dst_en) != 0)
1734*fae548d3Szrj 	    return CTF_ERR;			/* errno set for us.  */
1735*fae548d3Szrj 
1736*fae548d3Szrj 	  if (LCTF_INFO_ISROOT (fp, dst_tp->ctt_info) & CTF_ADD_ROOT)
1737*fae548d3Szrj 	    {
1738*fae548d3Szrj 	      /* The type that we found in the hash is also root-visible.  If
1739*fae548d3Szrj 		 the two types match then use the existing one; otherwise,
1740*fae548d3Szrj 		 declare a conflict.  Note: slices are not certain to match
1741*fae548d3Szrj 		 even if there is no conflict: we must check the contained type
1742*fae548d3Szrj 		 too.  */
1743*fae548d3Szrj 
1744*fae548d3Szrj 	      if (memcmp (&src_en, &dst_en, sizeof (ctf_encoding_t)) == 0)
1745*fae548d3Szrj 		{
1746*fae548d3Szrj 		  if (kind != CTF_K_SLICE)
1747*fae548d3Szrj 		    {
1748*fae548d3Szrj 		      ctf_add_type_mapping (src_fp, src_type, dst_fp, dst_type);
1749*fae548d3Szrj 		      return dst_type;
1750*fae548d3Szrj 		    }
1751*fae548d3Szrj 		}
1752*fae548d3Szrj 	      else
1753*fae548d3Szrj 		  {
1754*fae548d3Szrj 		    return (ctf_set_errno (dst_fp, ECTF_CONFLICT));
1755*fae548d3Szrj 		  }
1756*fae548d3Szrj 	    }
1757*fae548d3Szrj 	  else
1758*fae548d3Szrj 	    {
1759*fae548d3Szrj 	      /* We found a non-root-visible type in the hash.  If its encoding
1760*fae548d3Szrj 		 is the same, we can reuse it, unless it is a slice.  */
1761*fae548d3Szrj 
1762*fae548d3Szrj 	      if (memcmp (&src_en, &dst_en, sizeof (ctf_encoding_t)) == 0)
1763*fae548d3Szrj 		{
1764*fae548d3Szrj 		  if (kind != CTF_K_SLICE)
1765*fae548d3Szrj 		    {
1766*fae548d3Szrj 		      ctf_add_type_mapping (src_fp, src_type, dst_fp, dst_type);
1767*fae548d3Szrj 		      return dst_type;
1768*fae548d3Szrj 		    }
1769*fae548d3Szrj 		}
1770*fae548d3Szrj 	    }
1771*fae548d3Szrj 	}
1772*fae548d3Szrj     }
1773*fae548d3Szrj 
1774*fae548d3Szrj   src.ctb_file = src_fp;
1775*fae548d3Szrj   src.ctb_type = src_type;
1776*fae548d3Szrj   src.ctb_dtd = NULL;
1777*fae548d3Szrj 
1778*fae548d3Szrj   dst.ctb_file = dst_fp;
1779*fae548d3Szrj   dst.ctb_type = dst_type;
1780*fae548d3Szrj   dst.ctb_dtd = NULL;
1781*fae548d3Szrj 
1782*fae548d3Szrj   /* Now perform kind-specific processing.  If dst_type is CTF_ERR, then we add
1783*fae548d3Szrj      a new type with the same properties as src_type to dst_fp.  If dst_type is
1784*fae548d3Szrj      not CTF_ERR, then we verify that dst_type has the same attributes as
1785*fae548d3Szrj      src_type.  We recurse for embedded references.  Before we start, we note
1786*fae548d3Szrj      that we are processing this type, to prevent infinite recursion: we do not
1787*fae548d3Szrj      re-process any type that appears in this list.  The list is emptied
1788*fae548d3Szrj      wholesale at the end of processing everything in this recursive stack.  */
1789*fae548d3Szrj 
1790*fae548d3Szrj   if (ctf_dynhash_insert (proc_tracking_fp->ctf_add_processing,
1791*fae548d3Szrj 			  (void *) (uintptr_t) src_type, (void *) 1) < 0)
1792*fae548d3Szrj     return ctf_set_errno (dst_fp, ENOMEM);
1793*fae548d3Szrj 
1794*fae548d3Szrj   switch (kind)
1795*fae548d3Szrj     {
1796*fae548d3Szrj     case CTF_K_INTEGER:
1797*fae548d3Szrj       /*  If we found a match we will have either returned it or declared a
1798*fae548d3Szrj 	  conflict.  */
1799*fae548d3Szrj       dst_type = ctf_add_integer (dst_fp, flag, name, &src_en);
1800*fae548d3Szrj       break;
1801*fae548d3Szrj 
1802*fae548d3Szrj     case CTF_K_FLOAT:
1803*fae548d3Szrj       /* If we found a match we will have either returned it or declared a
1804*fae548d3Szrj        conflict.  */
1805*fae548d3Szrj       dst_type = ctf_add_float (dst_fp, flag, name, &src_en);
1806*fae548d3Szrj       break;
1807*fae548d3Szrj 
1808*fae548d3Szrj     case CTF_K_SLICE:
1809*fae548d3Szrj       /* We have checked for conflicting encodings: now try to add the
1810*fae548d3Szrj 	 contained type.  */
1811*fae548d3Szrj       src_type = ctf_type_reference (src_fp, src_type);
1812*fae548d3Szrj       src_type = ctf_add_type_internal (dst_fp, src_fp, src_type,
1813*fae548d3Szrj 					proc_tracking_fp);
1814*fae548d3Szrj 
1815*fae548d3Szrj       if (src_type == CTF_ERR)
1816*fae548d3Szrj 	return CTF_ERR;				/* errno is set for us.  */
1817*fae548d3Szrj 
1818*fae548d3Szrj       dst_type = ctf_add_slice (dst_fp, flag, src_type, &src_en);
1819*fae548d3Szrj       break;
1820*fae548d3Szrj 
1821*fae548d3Szrj     case CTF_K_POINTER:
1822*fae548d3Szrj     case CTF_K_VOLATILE:
1823*fae548d3Szrj     case CTF_K_CONST:
1824*fae548d3Szrj     case CTF_K_RESTRICT:
1825*fae548d3Szrj       src_type = ctf_type_reference (src_fp, src_type);
1826*fae548d3Szrj       src_type = ctf_add_type_internal (dst_fp, src_fp, src_type,
1827*fae548d3Szrj 					proc_tracking_fp);
1828*fae548d3Szrj 
1829*fae548d3Szrj       if (src_type == CTF_ERR)
1830*fae548d3Szrj 	return CTF_ERR;				/* errno is set for us.  */
1831*fae548d3Szrj 
1832*fae548d3Szrj       dst_type = ctf_add_reftype (dst_fp, flag, src_type, kind);
1833*fae548d3Szrj       break;
1834*fae548d3Szrj 
1835*fae548d3Szrj     case CTF_K_ARRAY:
1836*fae548d3Szrj       if (ctf_array_info (src_fp, src_type, &src_ar) != 0)
1837*fae548d3Szrj 	return (ctf_set_errno (dst_fp, ctf_errno (src_fp)));
1838*fae548d3Szrj 
1839*fae548d3Szrj       src_ar.ctr_contents =
1840*fae548d3Szrj 	ctf_add_type_internal (dst_fp, src_fp, src_ar.ctr_contents,
1841*fae548d3Szrj 			       proc_tracking_fp);
1842*fae548d3Szrj       src_ar.ctr_index = ctf_add_type_internal (dst_fp, src_fp,
1843*fae548d3Szrj 						src_ar.ctr_index,
1844*fae548d3Szrj 						proc_tracking_fp);
1845*fae548d3Szrj       src_ar.ctr_nelems = src_ar.ctr_nelems;
1846*fae548d3Szrj 
1847*fae548d3Szrj       if (src_ar.ctr_contents == CTF_ERR || src_ar.ctr_index == CTF_ERR)
1848*fae548d3Szrj 	return CTF_ERR;				/* errno is set for us.  */
1849*fae548d3Szrj 
1850*fae548d3Szrj       if (dst_type != CTF_ERR)
1851*fae548d3Szrj 	{
1852*fae548d3Szrj 	  if (ctf_array_info (dst_fp, dst_type, &dst_ar) != 0)
1853*fae548d3Szrj 	    return CTF_ERR;			/* errno is set for us.  */
1854*fae548d3Szrj 
1855*fae548d3Szrj 	  if (memcmp (&src_ar, &dst_ar, sizeof (ctf_arinfo_t)))
1856*fae548d3Szrj 	    {
1857*fae548d3Szrj 	      ctf_dprintf ("Conflict for type %s against ID %lx: "
1858*fae548d3Szrj 			   "array info differs, old %lx/%lx/%x; "
1859*fae548d3Szrj 			   "new: %lx/%lx/%x\n", name, dst_type,
1860*fae548d3Szrj 			   src_ar.ctr_contents, src_ar.ctr_index,
1861*fae548d3Szrj 			   src_ar.ctr_nelems, dst_ar.ctr_contents,
1862*fae548d3Szrj 			   dst_ar.ctr_index, dst_ar.ctr_nelems);
1863*fae548d3Szrj 	      return (ctf_set_errno (dst_fp, ECTF_CONFLICT));
1864*fae548d3Szrj 	    }
1865*fae548d3Szrj 	}
1866*fae548d3Szrj       else
1867*fae548d3Szrj 	dst_type = ctf_add_array (dst_fp, flag, &src_ar);
1868*fae548d3Szrj       break;
1869*fae548d3Szrj 
1870*fae548d3Szrj     case CTF_K_FUNCTION:
1871*fae548d3Szrj       ctc.ctc_return = ctf_add_type_internal (dst_fp, src_fp,
1872*fae548d3Szrj 					      src_tp->ctt_type,
1873*fae548d3Szrj 					      proc_tracking_fp);
1874*fae548d3Szrj       ctc.ctc_argc = 0;
1875*fae548d3Szrj       ctc.ctc_flags = 0;
1876*fae548d3Szrj 
1877*fae548d3Szrj       if (ctc.ctc_return == CTF_ERR)
1878*fae548d3Szrj 	return CTF_ERR;				/* errno is set for us.  */
1879*fae548d3Szrj 
1880*fae548d3Szrj       dst_type = ctf_add_function (dst_fp, flag, &ctc, NULL);
1881*fae548d3Szrj       break;
1882*fae548d3Szrj 
1883*fae548d3Szrj     case CTF_K_STRUCT:
1884*fae548d3Szrj     case CTF_K_UNION:
1885*fae548d3Szrj       {
1886*fae548d3Szrj 	ctf_dmdef_t *dmd;
1887*fae548d3Szrj 	int errs = 0;
1888*fae548d3Szrj 	size_t size;
1889*fae548d3Szrj 	ssize_t ssize;
1890*fae548d3Szrj 	ctf_dtdef_t *dtd;
1891*fae548d3Szrj 
1892*fae548d3Szrj 	/* Technically to match a struct or union we need to check both
1893*fae548d3Szrj 	   ways (src members vs. dst, dst members vs. src) but we make
1894*fae548d3Szrj 	   this more optimal by only checking src vs. dst and comparing
1895*fae548d3Szrj 	   the total size of the structure (which we must do anyway)
1896*fae548d3Szrj 	   which covers the possibility of dst members not in src.
1897*fae548d3Szrj 	   This optimization can be defeated for unions, but is so
1898*fae548d3Szrj 	   pathological as to render it irrelevant for our purposes.  */
1899*fae548d3Szrj 
1900*fae548d3Szrj 	if (dst_type != CTF_ERR && kind != CTF_K_FORWARD
1901*fae548d3Szrj 	    && dst_kind != CTF_K_FORWARD)
1902*fae548d3Szrj 	  {
1903*fae548d3Szrj 	    if (ctf_type_size (src_fp, src_type) !=
1904*fae548d3Szrj 		ctf_type_size (dst_fp, dst_type))
1905*fae548d3Szrj 	      {
1906*fae548d3Szrj 		ctf_dprintf ("Conflict for type %s against ID %lx: "
1907*fae548d3Szrj 			     "union size differs, old %li, new %li\n",
1908*fae548d3Szrj 			     name, dst_type,
1909*fae548d3Szrj 			     (long) ctf_type_size (src_fp, src_type),
1910*fae548d3Szrj 			     (long) ctf_type_size (dst_fp, dst_type));
1911*fae548d3Szrj 		return (ctf_set_errno (dst_fp, ECTF_CONFLICT));
1912*fae548d3Szrj 	      }
1913*fae548d3Szrj 
1914*fae548d3Szrj 	    if (ctf_member_iter (src_fp, src_type, membcmp, &dst))
1915*fae548d3Szrj 	      {
1916*fae548d3Szrj 		ctf_dprintf ("Conflict for type %s against ID %lx: "
1917*fae548d3Szrj 			     "members differ, see above\n", name, dst_type);
1918*fae548d3Szrj 		return (ctf_set_errno (dst_fp, ECTF_CONFLICT));
1919*fae548d3Szrj 	      }
1920*fae548d3Szrj 
1921*fae548d3Szrj 	    break;
1922*fae548d3Szrj 	  }
1923*fae548d3Szrj 
1924*fae548d3Szrj 	/* Unlike the other cases, copying structs and unions is done
1925*fae548d3Szrj 	   manually so as to avoid repeated lookups in ctf_add_member
1926*fae548d3Szrj 	   and to ensure the exact same member offsets as in src_type.  */
1927*fae548d3Szrj 
1928*fae548d3Szrj 	dst_type = ctf_add_generic (dst_fp, flag, name, kind, &dtd);
1929*fae548d3Szrj 	if (dst_type == CTF_ERR)
1930*fae548d3Szrj 	  return CTF_ERR;			/* errno is set for us.  */
1931*fae548d3Szrj 
1932*fae548d3Szrj 	dst.ctb_type = dst_type;
1933*fae548d3Szrj 	dst.ctb_dtd = dtd;
1934*fae548d3Szrj 
1935*fae548d3Szrj 	/* Pre-emptively add this struct to the type mapping so that
1936*fae548d3Szrj 	   structures that refer to themselves work.  */
1937*fae548d3Szrj 	ctf_add_type_mapping (src_fp, src_type, dst_fp, dst_type);
1938*fae548d3Szrj 
1939*fae548d3Szrj 	if (ctf_member_iter (src_fp, src_type, membadd, &dst) != 0)
1940*fae548d3Szrj 	  errs++;	       /* Increment errs and fail at bottom of case.  */
1941*fae548d3Szrj 
1942*fae548d3Szrj 	if ((ssize = ctf_type_size (src_fp, src_type)) < 0)
1943*fae548d3Szrj 	  return CTF_ERR;			/* errno is set for us.  */
1944*fae548d3Szrj 
1945*fae548d3Szrj 	size = (size_t) ssize;
1946*fae548d3Szrj 	if (size > CTF_MAX_SIZE)
1947*fae548d3Szrj 	  {
1948*fae548d3Szrj 	    dtd->dtd_data.ctt_size = CTF_LSIZE_SENT;
1949*fae548d3Szrj 	    dtd->dtd_data.ctt_lsizehi = CTF_SIZE_TO_LSIZE_HI (size);
1950*fae548d3Szrj 	    dtd->dtd_data.ctt_lsizelo = CTF_SIZE_TO_LSIZE_LO (size);
1951*fae548d3Szrj 	  }
1952*fae548d3Szrj 	else
1953*fae548d3Szrj 	  dtd->dtd_data.ctt_size = (uint32_t) size;
1954*fae548d3Szrj 
1955*fae548d3Szrj 	dtd->dtd_data.ctt_info = CTF_TYPE_INFO (kind, flag, vlen);
1956*fae548d3Szrj 
1957*fae548d3Szrj 	/* Make a final pass through the members changing each dmd_type (a
1958*fae548d3Szrj 	   src_fp type) to an equivalent type in dst_fp.  We pass through all
1959*fae548d3Szrj 	   members, leaving any that fail set to CTF_ERR, unless they fail
1960*fae548d3Szrj 	   because they are marking a member of type not representable in this
1961*fae548d3Szrj 	   version of CTF, in which case we just want to silently omit them:
1962*fae548d3Szrj 	   no consumer can do anything with them anyway.  */
1963*fae548d3Szrj 	for (dmd = ctf_list_next (&dtd->dtd_u.dtu_members);
1964*fae548d3Szrj 	     dmd != NULL; dmd = ctf_list_next (dmd))
1965*fae548d3Szrj 	  {
1966*fae548d3Szrj 	    ctf_file_t *dst = dst_fp;
1967*fae548d3Szrj 	    ctf_id_t memb_type;
1968*fae548d3Szrj 
1969*fae548d3Szrj 	    memb_type = ctf_type_mapping (src_fp, dmd->dmd_type, &dst);
1970*fae548d3Szrj 	    if (memb_type == 0)
1971*fae548d3Szrj 	      {
1972*fae548d3Szrj 		if ((dmd->dmd_type =
1973*fae548d3Szrj 		     ctf_add_type_internal (dst_fp, src_fp, dmd->dmd_type,
1974*fae548d3Szrj 					    proc_tracking_fp)) == CTF_ERR)
1975*fae548d3Szrj 		  {
1976*fae548d3Szrj 		    if (ctf_errno (dst_fp) != ECTF_NONREPRESENTABLE)
1977*fae548d3Szrj 		      errs++;
1978*fae548d3Szrj 		  }
1979*fae548d3Szrj 	      }
1980*fae548d3Szrj 	    else
1981*fae548d3Szrj 	      dmd->dmd_type = memb_type;
1982*fae548d3Szrj 	  }
1983*fae548d3Szrj 
1984*fae548d3Szrj 	if (errs)
1985*fae548d3Szrj 	  return CTF_ERR;			/* errno is set for us.  */
1986*fae548d3Szrj 	break;
1987*fae548d3Szrj       }
1988*fae548d3Szrj 
1989*fae548d3Szrj     case CTF_K_ENUM:
1990*fae548d3Szrj       if (dst_type != CTF_ERR && kind != CTF_K_FORWARD
1991*fae548d3Szrj 	  && dst_kind != CTF_K_FORWARD)
1992*fae548d3Szrj 	{
1993*fae548d3Szrj 	  if (ctf_enum_iter (src_fp, src_type, enumcmp, &dst)
1994*fae548d3Szrj 	      || ctf_enum_iter (dst_fp, dst_type, enumcmp, &src))
1995*fae548d3Szrj 	    {
1996*fae548d3Szrj 	      ctf_dprintf ("Conflict for enum %s against ID %lx: "
1997*fae548d3Szrj 			   "members differ, see above\n", name, dst_type);
1998*fae548d3Szrj 	      return (ctf_set_errno (dst_fp, ECTF_CONFLICT));
1999*fae548d3Szrj 	    }
2000*fae548d3Szrj 	}
2001*fae548d3Szrj       else
2002*fae548d3Szrj 	{
2003*fae548d3Szrj 	  dst_type = ctf_add_enum (dst_fp, flag, name);
2004*fae548d3Szrj 	  if ((dst.ctb_type = dst_type) == CTF_ERR
2005*fae548d3Szrj 	      || ctf_enum_iter (src_fp, src_type, enumadd, &dst))
2006*fae548d3Szrj 	    return CTF_ERR;			/* errno is set for us */
2007*fae548d3Szrj 	}
2008*fae548d3Szrj       break;
2009*fae548d3Szrj 
2010*fae548d3Szrj     case CTF_K_FORWARD:
2011*fae548d3Szrj       if (dst_type == CTF_ERR)
2012*fae548d3Szrj 	  dst_type = ctf_add_forward (dst_fp, flag, name, forward_kind);
2013*fae548d3Szrj       break;
2014*fae548d3Szrj 
2015*fae548d3Szrj     case CTF_K_TYPEDEF:
2016*fae548d3Szrj       src_type = ctf_type_reference (src_fp, src_type);
2017*fae548d3Szrj       src_type = ctf_add_type_internal (dst_fp, src_fp, src_type,
2018*fae548d3Szrj 					proc_tracking_fp);
2019*fae548d3Szrj 
2020*fae548d3Szrj       if (src_type == CTF_ERR)
2021*fae548d3Szrj 	return CTF_ERR;				/* errno is set for us.  */
2022*fae548d3Szrj 
2023*fae548d3Szrj       /* If dst_type is not CTF_ERR at this point, we should check if
2024*fae548d3Szrj 	 ctf_type_reference(dst_fp, dst_type) != src_type and if so fail with
2025*fae548d3Szrj 	 ECTF_CONFLICT.  However, this causes problems with bitness typedefs
2026*fae548d3Szrj 	 that vary based on things like if 32-bit then pid_t is int otherwise
2027*fae548d3Szrj 	 long.  We therefore omit this check and assume that if the identically
2028*fae548d3Szrj 	 named typedef already exists in dst_fp, it is correct or
2029*fae548d3Szrj 	 equivalent.  */
2030*fae548d3Szrj 
2031*fae548d3Szrj       if (dst_type == CTF_ERR)
2032*fae548d3Szrj 	  dst_type = ctf_add_typedef (dst_fp, flag, name, src_type);
2033*fae548d3Szrj 
2034*fae548d3Szrj       break;
2035*fae548d3Szrj 
2036*fae548d3Szrj     default:
2037*fae548d3Szrj       return (ctf_set_errno (dst_fp, ECTF_CORRUPT));
2038*fae548d3Szrj     }
2039*fae548d3Szrj 
2040*fae548d3Szrj   if (dst_type != CTF_ERR)
2041*fae548d3Szrj     ctf_add_type_mapping (src_fp, orig_src_type, dst_fp, dst_type);
2042*fae548d3Szrj   return dst_type;
2043*fae548d3Szrj }
2044*fae548d3Szrj 
2045*fae548d3Szrj ctf_id_t
ctf_add_type(ctf_file_t * dst_fp,ctf_file_t * src_fp,ctf_id_t src_type)2046*fae548d3Szrj ctf_add_type (ctf_file_t *dst_fp, ctf_file_t *src_fp, ctf_id_t src_type)
2047*fae548d3Szrj {
2048*fae548d3Szrj   ctf_id_t id;
2049*fae548d3Szrj 
2050*fae548d3Szrj   if (!src_fp->ctf_add_processing)
2051*fae548d3Szrj     src_fp->ctf_add_processing = ctf_dynhash_create (ctf_hash_integer,
2052*fae548d3Szrj 						     ctf_hash_eq_integer,
2053*fae548d3Szrj 						     NULL, NULL);
2054*fae548d3Szrj 
2055*fae548d3Szrj   /* We store the hash on the source, because it contains only source type IDs:
2056*fae548d3Szrj      but callers will invariably expect errors to appear on the dest.  */
2057*fae548d3Szrj   if (!src_fp->ctf_add_processing)
2058*fae548d3Szrj     return (ctf_set_errno (dst_fp, ENOMEM));
2059*fae548d3Szrj 
2060*fae548d3Szrj   id = ctf_add_type_internal (dst_fp, src_fp, src_type, src_fp);
2061*fae548d3Szrj   ctf_dynhash_empty (src_fp->ctf_add_processing);
2062*fae548d3Szrj 
2063*fae548d3Szrj   return id;
2064*fae548d3Szrj }
2065*fae548d3Szrj 
2066*fae548d3Szrj /* Write the compressed CTF data stream to the specified gzFile descriptor.  */
2067*fae548d3Szrj int
ctf_gzwrite(ctf_file_t * fp,gzFile fd)2068*fae548d3Szrj ctf_gzwrite (ctf_file_t *fp, gzFile fd)
2069*fae548d3Szrj {
2070*fae548d3Szrj   const unsigned char *buf;
2071*fae548d3Szrj   ssize_t resid;
2072*fae548d3Szrj   ssize_t len;
2073*fae548d3Szrj 
2074*fae548d3Szrj   resid = sizeof (ctf_header_t);
2075*fae548d3Szrj   buf = (unsigned char *) fp->ctf_header;
2076*fae548d3Szrj   while (resid != 0)
2077*fae548d3Szrj     {
2078*fae548d3Szrj       if ((len = gzwrite (fd, buf, resid)) <= 0)
2079*fae548d3Szrj 	return (ctf_set_errno (fp, errno));
2080*fae548d3Szrj       resid -= len;
2081*fae548d3Szrj       buf += len;
2082*fae548d3Szrj     }
2083*fae548d3Szrj 
2084*fae548d3Szrj   resid = fp->ctf_size;
2085*fae548d3Szrj   buf = fp->ctf_buf;
2086*fae548d3Szrj   while (resid != 0)
2087*fae548d3Szrj     {
2088*fae548d3Szrj       if ((len = gzwrite (fd, buf, resid)) <= 0)
2089*fae548d3Szrj 	return (ctf_set_errno (fp, errno));
2090*fae548d3Szrj       resid -= len;
2091*fae548d3Szrj       buf += len;
2092*fae548d3Szrj     }
2093*fae548d3Szrj 
2094*fae548d3Szrj   return 0;
2095*fae548d3Szrj }
2096*fae548d3Szrj 
2097*fae548d3Szrj /* Compress the specified CTF data stream and write it to the specified file
2098*fae548d3Szrj    descriptor.  */
2099*fae548d3Szrj int
ctf_compress_write(ctf_file_t * fp,int fd)2100*fae548d3Szrj ctf_compress_write (ctf_file_t *fp, int fd)
2101*fae548d3Szrj {
2102*fae548d3Szrj   unsigned char *buf;
2103*fae548d3Szrj   unsigned char *bp;
2104*fae548d3Szrj   ctf_header_t h;
2105*fae548d3Szrj   ctf_header_t *hp = &h;
2106*fae548d3Szrj   ssize_t header_len = sizeof (ctf_header_t);
2107*fae548d3Szrj   ssize_t compress_len;
2108*fae548d3Szrj   ssize_t len;
2109*fae548d3Szrj   int rc;
2110*fae548d3Szrj   int err = 0;
2111*fae548d3Szrj 
2112*fae548d3Szrj   if (ctf_serialize (fp) < 0)
2113*fae548d3Szrj     return -1;					/* errno is set for us.  */
2114*fae548d3Szrj 
2115*fae548d3Szrj   memcpy (hp, fp->ctf_header, header_len);
2116*fae548d3Szrj   hp->cth_flags |= CTF_F_COMPRESS;
2117*fae548d3Szrj   compress_len = compressBound (fp->ctf_size);
2118*fae548d3Szrj 
2119*fae548d3Szrj   if ((buf = malloc (compress_len)) == NULL)
2120*fae548d3Szrj     return (ctf_set_errno (fp, ECTF_ZALLOC));
2121*fae548d3Szrj 
2122*fae548d3Szrj   if ((rc = compress (buf, (uLongf *) &compress_len,
2123*fae548d3Szrj 		      fp->ctf_buf, fp->ctf_size)) != Z_OK)
2124*fae548d3Szrj     {
2125*fae548d3Szrj       ctf_dprintf ("zlib deflate err: %s\n", zError (rc));
2126*fae548d3Szrj       err = ctf_set_errno (fp, ECTF_COMPRESS);
2127*fae548d3Szrj       goto ret;
2128*fae548d3Szrj     }
2129*fae548d3Szrj 
2130*fae548d3Szrj   while (header_len > 0)
2131*fae548d3Szrj     {
2132*fae548d3Szrj       if ((len = write (fd, hp, header_len)) < 0)
2133*fae548d3Szrj 	{
2134*fae548d3Szrj 	  err = ctf_set_errno (fp, errno);
2135*fae548d3Szrj 	  goto ret;
2136*fae548d3Szrj 	}
2137*fae548d3Szrj       header_len -= len;
2138*fae548d3Szrj       hp += len;
2139*fae548d3Szrj     }
2140*fae548d3Szrj 
2141*fae548d3Szrj   bp = buf;
2142*fae548d3Szrj   while (compress_len > 0)
2143*fae548d3Szrj     {
2144*fae548d3Szrj       if ((len = write (fd, bp, compress_len)) < 0)
2145*fae548d3Szrj 	{
2146*fae548d3Szrj 	  err = ctf_set_errno (fp, errno);
2147*fae548d3Szrj 	  goto ret;
2148*fae548d3Szrj 	}
2149*fae548d3Szrj       compress_len -= len;
2150*fae548d3Szrj       bp += len;
2151*fae548d3Szrj     }
2152*fae548d3Szrj 
2153*fae548d3Szrj ret:
2154*fae548d3Szrj   free (buf);
2155*fae548d3Szrj   return err;
2156*fae548d3Szrj }
2157*fae548d3Szrj 
2158*fae548d3Szrj /* Optionally compress the specified CTF data stream and return it as a new
2159*fae548d3Szrj    dynamically-allocated string.  */
2160*fae548d3Szrj unsigned char *
ctf_write_mem(ctf_file_t * fp,size_t * size,size_t threshold)2161*fae548d3Szrj ctf_write_mem (ctf_file_t *fp, size_t *size, size_t threshold)
2162*fae548d3Szrj {
2163*fae548d3Szrj   unsigned char *buf;
2164*fae548d3Szrj   unsigned char *bp;
2165*fae548d3Szrj   ctf_header_t *hp;
2166*fae548d3Szrj   ssize_t header_len = sizeof (ctf_header_t);
2167*fae548d3Szrj   ssize_t compress_len;
2168*fae548d3Szrj   int rc;
2169*fae548d3Szrj 
2170*fae548d3Szrj   if (ctf_serialize (fp) < 0)
2171*fae548d3Szrj     return NULL;				/* errno is set for us.  */
2172*fae548d3Szrj 
2173*fae548d3Szrj   compress_len = compressBound (fp->ctf_size);
2174*fae548d3Szrj   if (fp->ctf_size < threshold)
2175*fae548d3Szrj     compress_len = fp->ctf_size;
2176*fae548d3Szrj   if ((buf = malloc (compress_len
2177*fae548d3Szrj 		     + sizeof (struct ctf_header))) == NULL)
2178*fae548d3Szrj     {
2179*fae548d3Szrj       ctf_set_errno (fp, ENOMEM);
2180*fae548d3Szrj       return NULL;
2181*fae548d3Szrj     }
2182*fae548d3Szrj 
2183*fae548d3Szrj   hp = (ctf_header_t *) buf;
2184*fae548d3Szrj   memcpy (hp, fp->ctf_header, header_len);
2185*fae548d3Szrj   bp = buf + sizeof (struct ctf_header);
2186*fae548d3Szrj   *size = sizeof (struct ctf_header);
2187*fae548d3Szrj 
2188*fae548d3Szrj   if (fp->ctf_size < threshold)
2189*fae548d3Szrj     {
2190*fae548d3Szrj       hp->cth_flags &= ~CTF_F_COMPRESS;
2191*fae548d3Szrj       memcpy (bp, fp->ctf_buf, fp->ctf_size);
2192*fae548d3Szrj       *size += fp->ctf_size;
2193*fae548d3Szrj     }
2194*fae548d3Szrj   else
2195*fae548d3Szrj     {
2196*fae548d3Szrj       hp->cth_flags |= CTF_F_COMPRESS;
2197*fae548d3Szrj       if ((rc = compress (bp, (uLongf *) &compress_len,
2198*fae548d3Szrj 			  fp->ctf_buf, fp->ctf_size)) != Z_OK)
2199*fae548d3Szrj 	{
2200*fae548d3Szrj 	  ctf_dprintf ("zlib deflate err: %s\n", zError (rc));
2201*fae548d3Szrj 	  ctf_set_errno (fp, ECTF_COMPRESS);
2202*fae548d3Szrj 	  free (buf);
2203*fae548d3Szrj 	  return NULL;
2204*fae548d3Szrj 	}
2205*fae548d3Szrj       *size += compress_len;
2206*fae548d3Szrj     }
2207*fae548d3Szrj   return buf;
2208*fae548d3Szrj }
2209*fae548d3Szrj 
2210*fae548d3Szrj /* Write the uncompressed CTF data stream to the specified file descriptor.  */
2211*fae548d3Szrj int
ctf_write(ctf_file_t * fp,int fd)2212*fae548d3Szrj ctf_write (ctf_file_t *fp, int fd)
2213*fae548d3Szrj {
2214*fae548d3Szrj   const unsigned char *buf;
2215*fae548d3Szrj   ssize_t resid;
2216*fae548d3Szrj   ssize_t len;
2217*fae548d3Szrj 
2218*fae548d3Szrj   if (ctf_serialize (fp) < 0)
2219*fae548d3Szrj     return -1;					/* errno is set for us.  */
2220*fae548d3Szrj 
2221*fae548d3Szrj   resid = sizeof (ctf_header_t);
2222*fae548d3Szrj   buf = (unsigned char *) fp->ctf_header;
2223*fae548d3Szrj   while (resid != 0)
2224*fae548d3Szrj     {
2225*fae548d3Szrj       if ((len = write (fd, buf, resid)) <= 0)
2226*fae548d3Szrj 	return (ctf_set_errno (fp, errno));
2227*fae548d3Szrj       resid -= len;
2228*fae548d3Szrj       buf += len;
2229*fae548d3Szrj     }
2230*fae548d3Szrj 
2231*fae548d3Szrj   resid = fp->ctf_size;
2232*fae548d3Szrj   buf = fp->ctf_buf;
2233*fae548d3Szrj   while (resid != 0)
2234*fae548d3Szrj     {
2235*fae548d3Szrj       if ((len = write (fd, buf, resid)) <= 0)
2236*fae548d3Szrj 	return (ctf_set_errno (fp, errno));
2237*fae548d3Szrj       resid -= len;
2238*fae548d3Szrj       buf += len;
2239*fae548d3Szrj     }
2240*fae548d3Szrj 
2241*fae548d3Szrj   return 0;
2242*fae548d3Szrj }
2243