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 36 RCSID("$Id: gen_copy.c 19539 2006-12-28 17:15:05Z lha $"); 37 38 static int used_fail; 39 40 static void 41 copy_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 48 static void 49 copy_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 221 void 222 generate_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