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: gen_copy.c 19539 2006-12-28 17:15:05Z lha $");
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	    asprintf (&fs, "%s(%s)->%s%s",
114		      m->optional ? "" : "&", from,
115		      t->type == TChoice ? "u." : "", m->gen_name);
116	    if (fs == NULL)
117		errx(1, "malloc");
118	    asprintf (&ts, "%s(%s)->%s%s",
119		      m->optional ? "" : "&", to,
120		      t->type == TChoice ? "u." : "", m->gen_name);
121	    if (ts == NULL)
122		errx(1, "malloc");
123	    if(m->optional){
124		fprintf(codefile, "if(%s) {\n", fs);
125		fprintf(codefile, "%s = malloc(sizeof(*%s));\n", ts, ts);
126		fprintf(codefile, "if(%s == NULL) goto fail;\n", ts);
127		used_fail++;
128	    }
129	    copy_type (fs, ts, m->type, FALSE);
130	    if(m->optional){
131		fprintf(codefile, "}else\n");
132		fprintf(codefile, "%s = NULL;\n", ts);
133	    }
134	    free (fs);
135	    free (ts);
136	    if(t->type == TChoice)
137		fprintf(codefile, "break;\n");
138	}
139	if(t->type == TChoice) {
140	    if (have_ellipsis) {
141		fprintf(codefile, "case %s: {\n"
142			"int ret;\n"
143			"ret=der_copy_octet_string(&(%s)->u.%s, &(%s)->u.%s);\n"
144			"if (ret) goto fail;\n"
145			"break;\n"
146			"}\n",
147			have_ellipsis->label,
148			from, have_ellipsis->gen_name,
149			to, have_ellipsis->gen_name);
150		used_fail++;
151	    }
152	    fprintf(codefile, "}\n");
153	}
154	break;
155    }
156    case TSetOf:
157    case TSequenceOf: {
158	char *f;
159	char *T;
160
161	fprintf (codefile, "if(((%s)->val = "
162		 "malloc((%s)->len * sizeof(*(%s)->val))) == NULL && (%s)->len != 0)\n",
163		 to, from, to, from);
164	fprintf (codefile, "goto fail;\n");
165	used_fail++;
166	fprintf(codefile,
167		"for((%s)->len = 0; (%s)->len < (%s)->len; (%s)->len++){\n",
168		to, to, from, to);
169	asprintf(&f, "&(%s)->val[(%s)->len]", from, to);
170	if (f == NULL)
171	    errx(1, "malloc");
172	asprintf(&T, "&(%s)->val[(%s)->len]", to, to);
173	if (T == NULL)
174	    errx(1, "malloc");
175	copy_type(f, T, t->subtype, FALSE);
176	fprintf(codefile, "}\n");
177	free(f);
178	free(T);
179	break;
180    }
181    case TGeneralizedTime:
182	fprintf(codefile, "*(%s) = *(%s);\n", to, from);
183	break;
184    case TGeneralString:
185	copy_primitive ("general_string", from, to);
186	break;
187    case TUTCTime:
188	fprintf(codefile, "*(%s) = *(%s);\n", to, from);
189	break;
190    case TUTF8String:
191	copy_primitive ("utf8string", from, to);
192	break;
193    case TPrintableString:
194	copy_primitive ("printable_string", from, to);
195	break;
196    case TIA5String:
197	copy_primitive ("ia5_string", from, to);
198	break;
199    case TBMPString:
200	copy_primitive ("bmp_string", from, to);
201	break;
202    case TUniversalString:
203	copy_primitive ("universal_string", from, to);
204	break;
205    case TVisibleString:
206	copy_primitive ("visible_string", from, to);
207	break;
208    case TTag:
209	copy_type (from, to, t->subtype, preserve);
210	break;
211    case TOID:
212	copy_primitive ("oid", from, to);
213	break;
214    case TNull:
215	break;
216    default :
217	abort ();
218    }
219}
220
221void
222generate_type_copy (const Symbol *s)
223{
224  int preserve = preserve_type(s->name) ? TRUE : FALSE;
225
226  used_fail = 0;
227
228  fprintf (headerfile,
229	   "int    copy_%s  (const %s *, %s *);\n",
230	   s->gen_name, s->gen_name, s->gen_name);
231
232  fprintf (codefile, "int\n"
233	   "copy_%s(const %s *from, %s *to)\n"
234	   "{\n"
235	   "memset(to, 0, sizeof(*to));\n",
236	   s->gen_name, s->gen_name, s->gen_name);
237  copy_type ("from", "to", s->type, preserve);
238  fprintf (codefile, "return 0;\n");
239
240  if (used_fail)
241      fprintf (codefile, "fail:\n"
242	       "free_%s(to);\n"
243	       "return ENOMEM;\n",
244	       s->gen_name);
245
246  fprintf(codefile,
247	  "}\n\n");
248}
249
250