asn1.c

   1 /*
   2    Unix SMB/Netbios implementation.
   3    Version 3.0
   4    simple SPNEGO routines
   5    Copyright (C) Andrew Tridgell 2001
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include <sys/types.h>
  23 #include <stdio.h>
  24 #include <unistd.h>
  25 #include <string.h>
  26 #include <ctype.h>
  27
  28 #define BOOL int
  29 #define True 1
  30 #define False 0
  31 #define Realloc realloc
  32
  33 struct nesting {
  34         off_t start;
  35         char *name;
  36         struct nesting *next;
  37 };
  38
  39 typedef struct {
  40         uint8 *data;
  41         size_t length;
  42         off_t ofs;
  43         struct nesting *nesting;
  44 } ASN1_DATA;
  45
  46 static BOOL asn1_write(ASN1_DATA data, void *p, int len)
  47 {
  48         if (data.length < data.ofs+len) {
  49                 data.data = Realloc(data.data, data.ofs+len);
  50                 if (!data.data) return False;
  51                 data.length = data.ofs+len;
  52         }
  53         memcpy(data.data + data.ofs, p, len);
  54         data.ofs += len;
  55 }
  56
  57 static BOOL asn1_push_tag(ASN1_DATA data, const char *tagstr, uint8 tag)
  58 {
  59         struct nesting *nesting;
  60         uint8 fill;
  61
  62         asn1_write(data, tag);
  63         nesting = (struct nesting *)malloc(sizeof(struct nesting));
  64         if (!nesting) return False;
  65
  66         nesting->start = data.ofs;
  67         nesting->name = strdup(tagstr);
  68         nesting->next = data.nesting;
  69         data.nesting = nesting;
  70         fill = 0xff;
  71         asn1_write(data, &fill, 1);
  72         return True;
  73 }
  74
  75 static BOOL asn1_pop_tag(ASN1_DATA data, const char *tagstr)
  76 {
  77         struct nesting *nesting;
  78
  79         nesting = data.nesting;
  80
  81         if (!nesting || strcmp(tagstr, nesting->name) != 0) {
  82                 return False;
  83         }
  84         data.data[nesting->start] = data.ofs - nesting->start;
  85         data.nesting = nesting->next;
  86         free(nesting->name);
  87         free(nesting);
  88         return True;
  89 }
  90
  91
  92 static BOOL asn1_write_uint8(ASN1_DATA data, uint8 v)
  93 {
  94         return asn1_write(data, &v, 1);
  95 }
  96
  97
  98 static BOOL asn1_write_OID(ASN1_DATA data, const char *OID)
  99 {
 100         long v, v2;
 101         char *p = (char *)OID;
 102
 103         asn1_push_tag(data, "OID", 0x6);
 104         v = strtol(p, &p, 10);
 105         v2 = strtol(p, &p, 10);
 106         asn1_write_uint8(data, 40*v + v2);
 107         while (*p) {
 108                 v = strtol(p, &p, 10);
 109                 if (v >= (1<<29)) asn1_write_uint8(data, 0x80 | ((v>>29)&0xff));
 110                 if (v >= (1<<22)) asn1_write_uint8(data, 0x80 | ((v>>22)&0xff));
 111                 if (v >= (1<<15)) asn1_write_uint8(data, 0x80 | ((v>>15)&0xff));
 112                 if (v >= (1<<8)) asn1_write_uint8(data, 0x80 | ((v>>8)&0xff));
 113                 asn1_write_uint8(data, 0x80 | ((v>>15)&0xff));
 114         }
 115         asn1_pop_tag(data, "OID");
 116         return True;
 117 }
 118
 119 static BOOL asn1_write_GeneralString(ASN1_DATA data, const char *s)
 120 {
 121         asn1_push_tag(data, "GS", 0x1b);
 122         asn1_write(data, s, strlen(s));
 123         asn1_pop_tag(data, "GS");
 124 }
 125
 126 /*
 127   generate a negTokenInit packet given a GUID, a list of supported
 128   OIDs (the mechanisms) and a principle name string
 129 */
 130 PACKET_DATA spnego_gen_negTokenInit(uint8 guid[16],
 131                                     const char **OIDs,
 132                                     const char *principle)
 133 {
 134         int i;
 135
 136         ASN1_DATA data;
 137
 138         ZERO_STRUCT(data);
 139
 140         asn1_write(data, guid, 16);
 141         asn1_push_tag(data,"SPNEGO", 0x60);
 142         asn1_write_OID(data,"1 3 6 1 5 5 2");
 143         asn1_push_tag(data,"MECHS", 0xa0);
 144         asn1_push_tag(data,"SEQ1", 0x30);
 145
 146         asn1_push_tag(data,"[0]", 0xa0);
 147         asn1_push_tag(data,"SEQ1.1", 0x30);
 148         for (i=0; OIDs[i]; i++) {
 149                 asn1_write_OID(data,OIDs[i]);
 150         }
 151         asn1_pop_tag(data,"SEQ1.1");
 152         asn1_pop_tag(data,"[0]");
 153
 154         asn1_push_tag(data,"[3]", 0xa3);
 155         asn1_push_tag(data,"SEQ1.2", 0x30);
 156         asn1_push_tag(data,"[0]", 0xa0);
 157         asn1_write_GeneralString(data,principle);
 158         asn1_pop_tag(data,"[0]");
 159         asn1_pop_tag(data,"SEQ1.2");
 160         asn1_pop_tag(data,"[3]");
 161
 162         asn1_pop_tag(data,"SEQ1");
 163         asn1_pop_tag(data,"MECHS");
 164
 165         asn1_check_empty(data);
 166 }
 167
 168
 169 #if 1
 170 main()
 171 {
 172         PACKET_DATA data;
 173         uint8 guid[16] = "012345678901234";
 174
 175         data = spnego_gen_negTokenInit(guid,
 176                 {"1 2 840 48018 1 2 2", "1 2 840 113554 1 2 2"},
 177                                        "blu$@VNET2.HOME.SAMBA.ORG");
 178 }
 179
 180 #endif