19a747e4fSDavid du Colombier #include <u.h>
29a747e4fSDavid du Colombier #include <libc.h>
39a747e4fSDavid du Colombier #include <mp.h>
49a747e4fSDavid du Colombier #include <libsec.h>
59a747e4fSDavid du Colombier
69a747e4fSDavid du Colombier #define STRLEN(s) (sizeof(s)-1)
79a747e4fSDavid du Colombier
89a747e4fSDavid du Colombier uchar*
decodePEM(char * s,char * type,int * len,char ** new_s)9*51711cb6SDavid du Colombier decodePEM(char *s, char *type, int *len, char **new_s)
109a747e4fSDavid du Colombier {
119a747e4fSDavid du Colombier uchar *d;
129a747e4fSDavid du Colombier char *t, *e, *tt;
139a747e4fSDavid du Colombier int n;
149a747e4fSDavid du Colombier
150ee2e829SDavid du Colombier *len = 0;
160ee2e829SDavid du Colombier
179a747e4fSDavid du Colombier /*
189a747e4fSDavid du Colombier * find the correct section of the file, stripping garbage at the beginning and end.
199a747e4fSDavid du Colombier * the data is delimited by -----BEGIN <type>-----\n and -----END <type>-----\n
209a747e4fSDavid du Colombier */
219a747e4fSDavid du Colombier n = strlen(type);
229a747e4fSDavid du Colombier e = strchr(s, '\0');
239a747e4fSDavid du Colombier for(t = s; t != nil && t < e; ){
249a747e4fSDavid du Colombier tt = t;
259a747e4fSDavid du Colombier t = strchr(tt, '\n');
269a747e4fSDavid du Colombier if(t != nil)
279a747e4fSDavid du Colombier t++;
289a747e4fSDavid du Colombier if(strncmp(tt, "-----BEGIN ", STRLEN("-----BEGIN ")) == 0
299a747e4fSDavid du Colombier && strncmp(&tt[STRLEN("-----BEGIN ")], type, n) == 0
309a747e4fSDavid du Colombier && strncmp(&tt[STRLEN("-----BEGIN ")+n], "-----\n", STRLEN("-----\n")) == 0)
319a747e4fSDavid du Colombier break;
329a747e4fSDavid du Colombier }
339a747e4fSDavid du Colombier for(tt = t; tt != nil && tt < e; tt++){
349a747e4fSDavid du Colombier if(strncmp(tt, "-----END ", STRLEN("-----END ")) == 0
359a747e4fSDavid du Colombier && strncmp(&tt[STRLEN("-----END ")], type, n) == 0
369a747e4fSDavid du Colombier && strncmp(&tt[STRLEN("-----END ")+n], "-----\n", STRLEN("-----\n")) == 0)
379a747e4fSDavid du Colombier break;
389a747e4fSDavid du Colombier tt = strchr(tt, '\n');
399a747e4fSDavid du Colombier if(tt == nil)
409a747e4fSDavid du Colombier break;
419a747e4fSDavid du Colombier }
429a747e4fSDavid du Colombier if(tt == nil || tt == e){
439a747e4fSDavid du Colombier werrstr("incorrect .pem file format: bad header or trailer");
449a747e4fSDavid du Colombier return nil;
459a747e4fSDavid du Colombier }
469a747e4fSDavid du Colombier
47*51711cb6SDavid du Colombier if(new_s)
48*51711cb6SDavid du Colombier *new_s = tt+1;
499a747e4fSDavid du Colombier n = ((tt - t) * 6 + 7) / 8;
509a747e4fSDavid du Colombier d = malloc(n);
519a747e4fSDavid du Colombier if(d == nil){
529a747e4fSDavid du Colombier werrstr("out of memory");
539a747e4fSDavid du Colombier return nil;
549a747e4fSDavid du Colombier }
559a747e4fSDavid du Colombier n = dec64(d, n, t, tt - t);
569a747e4fSDavid du Colombier if(n < 0){
579a747e4fSDavid du Colombier free(d);
589a747e4fSDavid du Colombier werrstr("incorrect .pem file format: bad base64 encoded data");
599a747e4fSDavid du Colombier return nil;
609a747e4fSDavid du Colombier }
619a747e4fSDavid du Colombier *len = n;
629a747e4fSDavid du Colombier return d;
639a747e4fSDavid du Colombier }
64*51711cb6SDavid du Colombier
65*51711cb6SDavid du Colombier PEMChain*
decodepemchain(char * s,char * type)66*51711cb6SDavid du Colombier decodepemchain(char *s, char *type)
67*51711cb6SDavid du Colombier {
68*51711cb6SDavid du Colombier PEMChain *first = nil, *last = nil, *chp;
69*51711cb6SDavid du Colombier uchar *d;
70*51711cb6SDavid du Colombier char *e;
71*51711cb6SDavid du Colombier int n;
72*51711cb6SDavid du Colombier
73*51711cb6SDavid du Colombier e = strchr(s, '\0');
74*51711cb6SDavid du Colombier while (s < e) {
75*51711cb6SDavid du Colombier d = decodePEM(s, type, &n, &s);
76*51711cb6SDavid du Colombier if(d == nil)
77*51711cb6SDavid du Colombier break;
78*51711cb6SDavid du Colombier chp = malloc(sizeof(PEMChain));
79*51711cb6SDavid du Colombier chp->next = nil;
80*51711cb6SDavid du Colombier chp->pem = d;
81*51711cb6SDavid du Colombier chp->pemlen = n;
82*51711cb6SDavid du Colombier if (first == nil)
83*51711cb6SDavid du Colombier first = chp;
84*51711cb6SDavid du Colombier else
85*51711cb6SDavid du Colombier last->next = chp;
86*51711cb6SDavid du Colombier last = chp;
87*51711cb6SDavid du Colombier }
88*51711cb6SDavid du Colombier return first;
89*51711cb6SDavid du Colombier }
90