source: libabac/abac_attribute.c @ 7764378

abac0-leak
Last change on this file since 7764378 was 7764378, checked in by Mei <mei@…>, 11 years ago

1) tweak according valgrind's leak report

  • Property mode set to 100644
File size: 13.9 KB
Line 
1
2/* abac_attribute.c */
3
4#define _GNU_SOURCE
5#include <stdio.h>
6#include <assert.h>
7#include <ctype.h>
8#include <string.h>
9#include <time.h>
10
11#include "libabac_common.h"
12#include "abac_list.h"
13#include "abac_util.h"
14#include "abac_xml.h"
15
16#define ROLE_SEPARATOR " <- "
17#define INTERSECTION_SEP " & "
18#define SHA1_LENGTH 40
19
20#define DEFAULT_OUTPUT_FORMAT "GENIv1.1"
21
22static int debug=0;
23
24// a GENI XML attribute chunk might contain multiple
25// attribute rules. It will be translate into multiple
26// abac_attribute structures but with cert ptr pointing
27// to the same xml chunk
28// issuer can be missing but then it won't be bakable
29// unless it is baked just for creddy's roles call
30struct _abac_attribute_t {
31    abac_id_t *issuer_id; 
32    char *role;
33    long validity;
34    int ntails;
35
36    char *head_string;
37    char **tail_strings;
38    char *output_format;
39    abac_keyid_map_t *keymap;
40
41    abac_chunk_t cert; // the XML chunk
42};
43
44char *abac_attribute_role_string(abac_attribute_t *attr);
45extern abac_id_t *abac_verifier_lookup_id(abac_list_t*, char *keyid);
46static char *_validate_principal(char *keyid);
47
48/************************************************************/
49abac_chunk_t abac_attribute_cert(abac_attribute_t *ptr)
50{
51    assert(ptr);
52    return ptr->cert;
53}
54
55abac_id_t *abac_attribute_issuer_id(abac_attribute_t *ptr)
56{
57    assert(ptr);
58    return ptr->issuer_id;
59}
60
61/* Get the format for this attribute to be output in.  This is NULL if the
62 * attribute has been read from a file. */
63char *abac_attribute_get_output_format(abac_attribute_t *a) {
64    return a->output_format;
65}
66
67/* Set the format for this attribute to be output in.  Valid formats are:
68 * GENIv1.0
69 * GENIv1.1
70 */
71void abac_attribute_set_output_format(abac_attribute_t *a, char *fmt) {
72    if (a->output_format) 
73        free(a->output_format);
74    a->output_format = abac_xstrdup(fmt);
75}
76
77
78// validity is measured in seconds (as of 0.2.0)
79// Acme.customer
80int abac_attribute_create(abac_attribute_t **ret, abac_id_t *issuer_id, char *role, long validity) {
81    libabac_init();
82    if (!abac_id_has_privkey(issuer_id))
83        return ABAC_ATTRIBUTE_ISSUER_NOKEY;
84    if (!abac_clean_name(role))
85        return ABAC_ATTRIBUTE_INVALID_ROLE;
86    if (validity < 0)
87        return ABAC_ATTRIBUTE_INVALID_VALIDITY;
88    if (!abac_id_still_valid(issuer_id))
89        return ABAC_ATTRIBUTE_INVALID_ISSUER;
90
91    if(validity == 0) validity = (long)(60*60*24*(365));
92
93    abac_attribute_t *attr = abac_xmalloc(sizeof(abac_attribute_t));
94    if(issuer_id) attr->issuer_id = abac_id_dup(issuer_id);
95        else attr->issuer_id = NULL;
96    attr->role = abac_xstrdup(role);
97    attr->validity = validity;
98
99    attr->head_string = NULL;
100    int rc=asprintf(&attr->head_string,"%s.%s",abac_id_keyid(issuer_id),role);
101    attr->ntails = 0;
102    attr->tail_strings = NULL;
103    attr->keymap = NULL;
104
105    attr->output_format = abac_xstrdup(DEFAULT_OUTPUT_FORMAT);
106
107    // NULL until baked
108    attr->cert.ptr=NULL;
109    attr->cert.len=0;
110
111    *ret = attr;
112    return ABAC_SUCCESS;
113}
114
115/**
116 * Get the validity period.(xml module returns the diff from expire time - now()
117 */
118int abac_attribute_validity(abac_attribute_t *attr,struct tm *not_before,struct tm *not_after) {
119    assert(attr);
120    memset(not_before, 0, sizeof(struct tm));
121    memset(not_after, 0, sizeof(struct tm));
122
123    time_t now;
124    time(&now);
125    gmtime_r(&now, not_before);
126    char *xml=(char *)attr->cert.ptr;
127    long validity=get_validity_from_xml(xml);
128
129    time_t etime = now + validity;
130    gmtime_r(&etime, not_after);
131
132    if(debug) fprintf(stderr,"validity from the xml blob is %ld\n",validity);
133    if(validity == 0)
134        return ABAC_FAILURE;
135    return ABAC_SUCCESS;
136}
137
138int abac_attribute_still_valid(abac_attribute_t *attr)
139{
140    assert(attr);
141    assert(attr->cert.ptr);
142    long v=get_validity_from_xml((char *)attr->cert.ptr);
143    if (v > 0.0)
144        return 1;
145    else return 0;
146}
147
148/* string is a malloc copy */
149int abac_attribute_add_tail(abac_attribute_t *attr, char *string) {
150    assert(attr);
151
152    char **old_tail = attr->tail_strings;
153    int newsize = (attr->ntails+1)*sizeof(char *);
154
155    if ( !(attr->tail_strings = realloc(attr->tail_strings, newsize))) {
156        attr->tail_strings = old_tail;
157        return 0;
158    }
159    attr->tail_strings[attr->ntails++] = string;
160    return 1;
161}
162
163
164void abac_attribute_set_head(abac_attribute_t *attr, char *string)
165{
166    assert(attr);
167    attr->head_string=string;
168}
169
170char *abac_attribute_get_head(abac_attribute_t *attr)
171{
172    assert(attr);
173    return attr->head_string;
174}
175
176/*
177 * Return the number of tail strings
178 */
179int abac_attribute_get_ntails(abac_attribute_t *attr) {
180    assert(attr);
181    return attr->ntails;
182}
183
184/*
185 * Return the nth tail string or NULL if it is undefined
186 */
187
188char *abac_attribute_get_tail_n(abac_attribute_t *attr, int n) {
189    assert(attr);
190    if ( n < 0 || n > attr->ntails) return NULL;
191    return attr->tail_strings[n];
192}
193
194
195/* A.b->C, return copy of a A */
196char *abac_attribute_get_principal(abac_attribute_t *attr)
197{
198    /* already a copy */
199    char *tmp=abac_attribute_role_string(attr);
200    char *head_tail[2];
201    int ret = 2;
202    abac_split(tmp, "<-", head_tail, &ret);
203    if (ret != 2) goto err;
204    abac_split(head_tail[0], ".", head_tail, &ret);
205    if (ret != 2) goto err;
206    char *prin=strdup(head_tail[0]);
207    free(tmp);
208    return prin;
209
210err:     
211     free(tmp);
212     return NULL;
213}
214
215int abac_attribute_principal(abac_attribute_t *attr, char *keyid) {
216    char *copy = _validate_principal(keyid);
217    if (copy == NULL)
218        return 0;
219
220    return abac_attribute_add_tail(attr,copy);
221}
222
223int abac_attribute_role(abac_attribute_t *attr, char *keyid, char *role) {
224    if (!abac_clean_name(role))
225        return 0;
226
227    char *copy = _validate_principal(keyid);
228    char *newcopy=NULL;
229    if (copy == NULL)
230        return 0;
231
232    int rc=asprintf(&newcopy,"%s.%s", copy,role);
233    free(copy);
234    return abac_attribute_add_tail(attr, newcopy);
235}
236
237int abac_attribute_linking_role(abac_attribute_t *attr, char *keyid, char *role, char *linked) {
238    if (!abac_clean_name(role) || !abac_clean_name(linked))
239        return 0;
240
241    char *copy = _validate_principal(keyid);
242    if (copy == NULL)
243        return 0;
244
245    char *newcopy=NULL;
246    int rc=asprintf(&newcopy,"%s.%s.%s", copy,role,linked);
247    free(copy);
248    return abac_attribute_add_tail(attr, newcopy);
249}
250
251
252
253// 0 for fail to bake, 1 is baked okay
254int abac_attribute_bake_context(abac_attribute_t *attr, abac_context_t *ctxt) {
255    assert(attr);
256    assert(attr->head_string);
257    assert(attr->tail_strings);
258    abac_keyid_map_t *km = NULL;
259
260    abac_chunk_t id_chunk = { NULL, 0 };
261    int ret=abac_id_PEM(attr->issuer_id, &id_chunk);
262    if(ret != ABAC_CERT_SUCCESS)
263        return 0; 
264
265    if ( ctxt && (km = abac_context_get_keyid_map(ctxt))) {
266        if (attr->keymap) abac_keyid_map_free(attr->keymap);
267        attr->keymap = abac_keyid_map_dup(km);
268    }
269
270    /* Make an new GENI abac credential with the rt0 rule that expires secs
271     * from now.  cert is the PEM encoded X.509 of the issuer's certificate as
272     * a string.  certlen is the length of cert.  Returns the XML. Caller is
273     * responsible for freeing it. */
274    char *attr_cert=make_credential(attr, attr->validity, 
275            (char *)id_chunk.ptr, id_chunk.len);
276
277    /*MEI, free id_chunk */
278    abac_chunk_free(id_chunk);
279
280    if (attr_cert == NULL)
281        return 0;
282
283    attr->cert.ptr = (unsigned char *)attr_cert;
284    attr->cert.len = strlen(attr_cert);
285
286    return 1;
287}
288// 0 for fail to bake, 1 is baked okay
289int abac_attribute_bake(abac_attribute_t *attr) {
290    return abac_attribute_bake_context(attr, NULL);
291}
292
293/*
294 * caller is responsible to free up the chunk after use
295*/
296abac_chunk_t abac_attribute_cert_chunk(abac_attribute_t *attr) {
297    abac_chunk_t chunk= {NULL,0};
298
299    if (attr->cert.ptr == NULL)
300        return chunk;
301
302    /* return the xml chunk */
303    chunk.ptr= (unsigned char *)abac_xstrdup((char *)attr->cert.ptr);
304    chunk.len=attr->cert.len;
305    return chunk;
306}
307
308int abac_attribute_baked(abac_attribute_t *attr) {
309    return (attr->cert.ptr != NULL);
310}
311
312
313static abac_attribute_t *_load_attr(abac_list_t *id_certs,char *rstring, char *xml, abac_keyid_map_t *km)
314{
315    /* make a copy of rle_string */
316    char *role_string=abac_xstrdup(rstring);
317
318    if(debug) fprintf(stderr,"loading -> %s \n", role_string);
319
320    char *head_tail[2];
321    char *role_rest[2];
322    int ret = 2;
323    abac_split(role_string, "<-", head_tail, &ret);
324    if (ret != 2) return NULL; 
325
326    char *keyid=get_keyid_from_xml(xml);
327    abac_id_t *issuer_id=abac_verifier_lookup_id(id_certs,keyid);
328
329    long validity=get_validity_from_xml(xml);
330   
331    abac_attribute_t *attr = abac_xmalloc(sizeof(abac_attribute_t));
332    if(issuer_id)
333        attr->issuer_id = abac_id_dup(issuer_id);
334    else attr->issuer_id=NULL;
335    attr->validity = validity;
336    attr->ntails = 0;
337    attr->tail_strings = NULL;
338
339    /* If there is a keymap, make a reference to it. */
340    if ( km ) attr->keymap = abac_keyid_map_dup(km);
341    else attr->keymap = NULL;
342
343
344    attr->head_string = abac_xstrdup(head_tail[0]);
345    do {
346        ret = 2;
347        abac_split(head_tail[1], " & ", role_rest, &ret);
348        abac_attribute_add_tail(attr, abac_xstrdup(role_rest[0]));
349        head_tail[1] =role_rest[1];
350    } while (ret == 2);
351
352    char *tmp=strstr(attr->head_string,".");
353    attr->role =abac_xstrdup(tmp+1);
354
355    attr->cert.ptr=(unsigned char *)abac_xstrdup(xml);
356    attr->cert.len=strlen(xml);
357    attr->output_format = NULL;
358
359    free(keyid);
360    free(role_string);
361    return attr;
362}
363
364abac_list_t *abac_attribute_certs_from_file(abac_list_t *id_certs,char *filename)
365{
366    libabac_init();
367    abac_list_t *alist=abac_list_new();
368    char *xml=NULL;
369    char *rt0=NULL;
370    abac_keyid_map_t *km = abac_keyid_map_new();
371
372    char **rt0s=read_credential((void *)id_certs,filename, &xml, km);
373    if(rt0s == NULL) return alist;
374    if(xml == NULL || strlen(xml)==0) return alist;
375
376    abac_attribute_t *attr;
377
378    int i=0;
379    do {
380        rt0 = rt0s[i]; 
381        if(rt0 == NULL) break;
382        attr=_load_attr(id_certs,rt0, xml, km);
383        if(attr)
384            abac_list_add(alist, attr);
385        free(rt0);
386        i++;
387    } while (rt0s[i] !=NULL);
388    abac_keyid_map_free(km);
389
390    free(rt0s);
391    free(xml);
392
393    return alist;
394}
395
396abac_list_t *abac_attribute_certs_from_chunk(abac_list_t *id_certs,abac_chunk_t chunk) {
397    libabac_init();
398
399    abac_list_t *alist=abac_list_new();
400    char *xml=(char *)chunk.ptr;
401    abac_keyid_map_t *km = abac_keyid_map_new();
402
403    if(chunk.len==0) return alist;
404
405    char **rt0s=get_rt0_from_xml((void *) id_certs, xml, km);
406    char *rt0=NULL;
407    if(rt0s==NULL) return alist;
408
409    abac_attribute_t *attr;
410    int i=0;
411    do {
412        rt0 = rt0s[i]; 
413        if(rt0 == NULL) break;
414        attr=_load_attr(id_certs,rt0, xml, km);
415        if(attr)
416            abac_list_add(alist, attr);
417        free(rt0);
418        i++;
419    } while (rt0s[i] !=NULL);
420    abac_keyid_map_free(km);
421
422    free(rt0s);
423    return alist;
424}
425
426// returns ABAC_FAILURE if the cert hasn't been baked
427int abac_attribute_write(abac_attribute_t *attr, FILE *out) {
428    assert(attr != NULL);
429
430    if (attr->cert.ptr == NULL)
431        return ABAC_FAILURE;
432
433    // write to file
434    fwrite(attr->cert.ptr, attr->cert.len, 1, out);
435
436    return ABAC_SUCCESS;
437}
438
439// returns ABAC_FAILURE if the cert hasn't been baked
440int abac_attribute_write_file(abac_attribute_t *attr, const char *fname) {
441    if (attr->cert.ptr == NULL)
442        return ABAC_FAILURE;
443
444    FILE *fp=fopen(fname,"w+");
445    if(fp) {
446         // write to file
447         fwrite(attr->cert.ptr, attr->cert.len, 1, fp);
448    } else return ABAC_FAILURE;
449    fclose(fp);
450
451    return ABAC_SUCCESS;
452}
453
454/* return a copy of the local name mappings, if any.  The returned value is not
455 * reference counted, so callers will need to call abac_keyid_map_dup on it if
456 * they need to keep a copy of the pointer.
457 */
458abac_keyid_map_t *abac_attribute_get_keyid_map(abac_attribute_t *attr) {
459    return attr->keymap;
460}
461
462
463void abac_attribute_free(abac_attribute_t *attr) {
464
465if(debug) fprintf(stderr,"calling abac_attribute_free on %s\n", attr->head_string);
466    int i = 0;
467
468    if (attr == NULL)
469        return;
470
471    if(attr->issuer_id) abac_id_free(attr->issuer_id);
472
473    free(attr->role);
474    free(attr->head_string);
475    for (i=0; i < attr->ntails; i++) 
476        free(attr->tail_strings[i]);
477    free(attr->tail_strings);
478    if ( attr->output_format ) 
479        free(attr->output_format);
480
481    if ( attr->keymap ) abac_keyid_map_free(attr->keymap);
482
483    free(attr->cert.ptr);
484
485    free(attr);
486}
487
488//
489// Helper functions below
490//
491
492// validate a princpal's name
493// makes sure it's a valid SHA1 identifier
494// return values:
495//  success: malloc'd copy with all hex digits lowercase
496//  fail: NULL
497static char *_validate_principal(char *keyid) {
498    int i;
499    char *copy = NULL;
500
501    if (strlen(keyid) != SHA1_LENGTH)
502        return NULL;
503
504    copy = abac_xstrdup(keyid);
505    for (i = 0; i < SHA1_LENGTH; ++i) {
506        copy[i] = tolower(copy[i]);
507        if (!isxdigit(copy[i]))
508            goto error;
509    }
510
511    return copy;
512
513error:
514    free(copy);
515    return NULL;
516}
517
518static int abac_attribute_role_string_size(abac_attribute_t *attr) {
519    int sz = 3; /* Start with the end of string character and <-*/
520    int i;      /* Scratch */
521
522    if ( !attr) return sz;
523    if ( attr->head_string) 
524        sz += strlen(attr->head_string);
525    for (i = 0; i < attr->ntails; i++) 
526        if ( attr->tail_strings[i]) 
527            sz += strlen(attr->tail_strings[i]);
528    /* " & " between each pair of tails */
529    sz += 3 * (attr->ntails-1);
530    return sz;
531}
532
533// combine up the attribute's rule string, explicit copy
534char *abac_attribute_role_string(abac_attribute_t *attr) {
535    assert(attr);
536
537    int sz = abac_attribute_role_string_size(attr);
538    char *role_string=abac_xmalloc(sz);
539    int i;
540
541    if ( !role_string) return NULL;
542
543    sz -= snprintf(role_string, sz, "%s<-", attr->head_string);
544    for ( i = 0 ; i < attr->ntails; i++ ) {
545        if ( i > 0 ) {
546            strncat(role_string, " & ", 3);
547            sz -= 3;
548        }
549        strncat(role_string,attr->tail_strings[i], sz); 
550
551        sz -= strlen(attr->tail_strings[i]);
552        if (sz < 0 ) return NULL;
553    }
554    return role_string;
555}
Note: See TracBrowser for help on using the repository browser.