1/*
2 * Copyright (c) 1997 - 2005 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of the Institute nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#include "gen_locl.h"
35
36RCSID("$Id$");
37
38static int used_fail;
39
40static void
41copy_primitive (const char *typename, const char *from, const char *to)
42{
43    fprintf (codefile, "if(der_copy_%s(%s, %s)) goto fail;\n",
44	     typename, from, to);
45    used_fail++;
46}
47
48static void
49copy_type (const char *from, const char *to, const Type *t, int preserve)
50{
51    switch (t->type) {
52    case TType:
53#if 0
54	copy_type (from, to, t->symbol->type, preserve);
55#endif
56	fprintf (codefile, "if(copy_%s(%s, %s)) goto fail;\n",
57		 t->symbol->gen_name, from, to);
58	used_fail++;
59	break;
60    case TInteger:
61	if (t->range == NULL && t->members == NULL) {
62	    copy_primitive ("heim_integer", from, to);
63	    break;
64	}
65    case TBoolean:
66    case TEnumerated :
67	fprintf(codefile, "*(%s) = *(%s);\n", to, from);
68	break;
69    case TOctetString:
70	copy_primitive ("octet_string", from, to);
71	break;
72    case TBitString:
73	if (ASN1_TAILQ_EMPTY(t->members))
74	    copy_primitive ("bit_string", from, to);
75	else
76	    fprintf(codefile, "*(%s) = *(%s);\n", to, from);
77	break;
78    case TSet:
79    case TSequence:
80    case TChoice: {
81	Member *m, *have_ellipsis = NULL;
82
83	if(t->members == NULL)
84	    break;
85
86	if ((t->type == TSequence || t->type == TChoice) && preserve) {
87	    fprintf(codefile,
88		    "{ int ret;\n"
89		    "ret = der_copy_octet_string(&(%s)->_save, &(%s)->_save);\n"
90		    "if (ret) goto fail;\n"
91		    "}\n",
92		    from, to);
93	    used_fail++;
94	}
95
96	if(t->type == TChoice) {
97	    fprintf(codefile, "(%s)->element = (%s)->element;\n", to, from);
98	    fprintf(codefile, "switch((%s)->element) {\n", from);
99	}
100
101	ASN1_TAILQ_FOREACH(m, t->members, members) {
102	    char *fs;
103	    char *ts;
104
105	    if (m->ellipsis) {
106		have_ellipsis = m;
107		continue;
108	    }
109
110	    if(t->type == TChoice)
111		fprintf(codefile, "case %s:\n", m->label);
112
113	    if (asprintf (&fs, "%s(%s)->%s%s",
114			  m->optional ? "" : "&", from,
115			  t->type == TChoice ? "u." : "", m->gen_name) < 0)
116		errx(1, "malloc");
117	    if (fs == NULL)
118		errx(1, "malloc");
119	    if (asprintf (&ts, "%s(%s)->%s%s",
120			  m->optional ? "" : "&", to,
121			  t->type == TChoice ? "u." : "", m->gen_name) < 0)
122		errx(1, "malloc");
123	    if (ts == NULL)
124		errx(1, "malloc");
125	    if(m->optional){
126		fprintf(codefile, "if(%s) {\n", fs);
127		fprintf(codefile, "%s = malloc(sizeof(*%s));\n", ts, ts);
128		fprintf(codefile, "if(%s == NULL) goto fail;\n", ts);
129		used_fail++;
130	    }
131	    copy_type (fs, ts, m->type, FALSE);
132	    if(m->optional){
133		fprintf(codefile, "}else\n");
134		fprintf(codefile, "%s = NULL;\n", ts);
135	    }
136	    free (fs);
137	    free (ts);
138	    if(t->type == TChoice)
139		fprintf(codefile, "break;\n");
140	}
141	if(t->type == TChoice) {
142	    if (have_ellipsis) {
143		fprintf(codefile, "case %s: {\n"
144			"int ret;\n"
145			"ret=der_copy_octet_string(&(%s)->u.%s, &(%s)->u.%s);\n"
146			"if (ret) goto fail;\n"
147			"break;\n"
148			"}\n",
149			have_ellipsis->label,
150			from, have_ellipsis->gen_name,
151			to, have_ellipsis->gen_name);
152		used_fail++;
153	    }
154	    fprintf(codefile, "}\n");
155	}
156	break;
157    }
158    case TSetOf:
159    case TSequenceOf: {
160	char *f = NULL, *T = NULL;
161
162	fprintf (codefile, "if(((%s)->val = "
163		 "malloc((%s)->len * sizeof(*(%s)->val))) == NULL && (%s)->len != 0)\n",
164		 to, from, to, from);
165	fprintf (codefile, "goto fail;\n");
166	used_fail++;
167	fprintf(codefile,
168		"for((%s)->len = 0; (%s)->len < (%s)->len; (%s)->len++){\n",
169		to, to, from, to);
170	if (asprintf(&f, "&(%s)->val[(%s)->len]", from, to) < 0)
171	    errx(1, "malloc");
172	if (f == NULL)
173	    errx(1, "malloc");
174	if (asprintf(&T, "&(%s)->val[(%s)->len]", to, to) < 0)
175	    errx(1, "malloc");
176	if (T == NULL)
177	    errx(1, "malloc");
178	copy_type(f, T, t->subtype, FALSE);
179	fprintf(codefile, "}\n");
180	free(f);
181	free(T);
182	break;
183    }
184    case TGeneralizedTime:
185	fprintf(codefile, "*(%s) = *(%s);\n", to, from);
186	break;
187    case TGeneralString:
188	copy_primitive ("general_string", from, to);
189	break;
190    case TTeletexString:
191	copy_primitive ("general_string", from, to);
192	break;
193    case TUTCTime:
194	fprintf(codefile, "*(%s) = *(%s);\n", to, from);
195	break;
196    case TUTF8String:
197	copy_primitive ("utf8string", from, to);
198	break;
199    case TPrintableString:
200	copy_primitive ("printable_string", from, to);
201	break;
202    case TIA5String:
203	copy_primitive ("ia5_string", from, to);
204	break;
205    case TBMPString:
206	copy_primitive ("bmp_string", from, to);
207	break;
208    case TUniversalString:
209	copy_primitive ("universal_string", from, to);
210	break;
211    case TVisibleString:
212	copy_primitive ("visible_string", from, to);
213	break;
214    case TTag:
215	copy_type (from, to, t->subtype, preserve);
216	break;
217    case TOID:
218	copy_primitive ("oid", from, to);
219	break;
220    case TNull:
221	break;
222    default :
223	abort ();
224    }
225}
226
227void
228generate_type_copy (const Symbol *s)
229{
230  int preserve = preserve_type(s->name) ? TRUE : FALSE;
231
232  used_fail = 0;
233
234  fprintf (codefile, "int ASN1CALL\n"
235	   "copy_%s(const %s *from, %s *to)\n"
236	   "{\n"
237	   "memset(to, 0, sizeof(*to));\n",
238	   s->gen_name, s->gen_name, s->gen_name);
239  copy_type ("from", "to", s->type, preserve);
240  fprintf (codefile, "return 0;\n");
241
242  if (used_fail)
243      fprintf (codefile, "fail:\n"
244	       "free_%s(to);\n"
245	       "return ENOMEM;\n",
246	       s->gen_name);
247
248  fprintf(codefile,
249	  "}\n\n");
250}
251
252