crypto/external/bsd/heimdal/dist/lib/asn1/gen_copy.c

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