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$"); 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 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 227 void 228 generate_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