1 | #!/usr/bin/env python |
---|
2 | |
---|
3 | """ |
---|
4 | to test with python |
---|
5 | |
---|
6 | cmd1:env keystore=`pwd` ./query.py |
---|
7 | cmd2: env ABAC_CN=1 keystore=`pwd` ./query.py |
---|
8 | |
---|
9 | """ |
---|
10 | |
---|
11 | import os |
---|
12 | import ABAC |
---|
13 | |
---|
14 | ctxt = ABAC.Context() |
---|
15 | |
---|
16 | # print "ABAC version %s" % ctxt.version() |
---|
17 | |
---|
18 | keystore=os.environ["keystore"] |
---|
19 | |
---|
20 | ctxt.load_directory(keystore) |
---|
21 | |
---|
22 | acmeID=ABAC.ID("Acme_ID.pem") |
---|
23 | acmeID.id_load_privkey_file("Acme_private.pem"); |
---|
24 | acme=acmeID.id_keyid() |
---|
25 | |
---|
26 | coyoteID=ABAC.ID("Coyote_ID.pem") |
---|
27 | coyoteID.id_load_privkey_file("Coyote_private.pem"); |
---|
28 | coyote=coyoteID.id_keyid() |
---|
29 | |
---|
30 | roadrunnerID=ABAC.ID("Roadrunner_ID.pem") |
---|
31 | roadrunnerID.id_load_privkey_file("Roadrunner_private.pem"); |
---|
32 | roadrunner=roadrunnerID.id_keyid() |
---|
33 | |
---|
34 | jackrabbitID=ABAC.ID("Jackrabbit_ID.pem") |
---|
35 | jackrabbitID.id_load_privkey_file("Jackrabbit_private.pem"); |
---|
36 | jackrabbit=jackrabbitID.id_keyid() |
---|
37 | |
---|
38 | ########################################################################## |
---|
39 | # role =[keyid:Acme].role:friendof([keyid:Roadrunner]) |
---|
40 | # p= [keyid:Coyote] |
---|
41 | param=ABAC.DataTerm(roadrunnerID) |
---|
42 | role = ABAC.Role(acme,"friendof") |
---|
43 | role.role_add_data_term(param) |
---|
44 | p = ABAC.Role(coyote) |
---|
45 | print "\n===bad============ Acme.friendOf(Roadrunner) <- Coyote" |
---|
46 | out = ctxt.query(role, p) |
---|
47 | |
---|
48 | for c in out[1]: |
---|
49 | print "%s <- %s" % (c.head_string(), c.tail_string()) |
---|
50 | |
---|
51 | ########################################################################## |
---|
52 | # role =[keyid:Acme].role:friendof([keyid:Roadrunner]) |
---|
53 | # p= [keyid:Jackrabbit] |
---|
54 | param=ABAC.DataTerm(roadrunnerID) |
---|
55 | role = ABAC.Role(acme,"friendof") |
---|
56 | role.role_add_data_term(param) |
---|
57 | p = ABAC.Role(jackrabbit) |
---|
58 | print "\n===good============ Acme.friendOf(Roadrunner) <- Jackrabbit" |
---|
59 | out = ctxt.query(role, p) |
---|
60 | |
---|
61 | for c in out[1]: |
---|
62 | print "%s <- %s" % (c.head_string(), c.tail_string()) |
---|
63 | |
---|
64 | |
---|
65 | ########################################################################## |
---|
66 | # role =[keyid:Acme].role:preferred_customer |
---|
67 | # p =[keyid:Jackrabbit] |
---|
68 | role = ABAC.Role(acme,"preferred_customer") |
---|
69 | p = ABAC.Role(jackrabbit) |
---|
70 | print "\n===good============ Acme.preferred_customer <- Jackrabbit" |
---|
71 | out = ctxt.query(role, p) |
---|
72 | |
---|
73 | for c in out[1]: |
---|
74 | print "%s <- %s" % (c.head_string(), c.tail_string()) |
---|
75 | |
---|
76 | ########################################################################## |
---|
77 | # role =[keyid:Acme].role:preferred_customer |
---|
78 | # p =[keyid:Coyote] |
---|
79 | role = ABAC.Role(acme,"preferred_customer") |
---|
80 | p = ABAC.Role(coyote) |
---|
81 | print "\n===good============ Acme.preferred_customer <- Coyote" |
---|
82 | out = ctxt.query(role, p) |
---|
83 | |
---|
84 | for c in out[1]: |
---|
85 | print "%s <- %s" % (c.head_string(), c.tail_string()) |
---|
86 | |
---|
87 | |
---|
88 | |
---|
89 | ########################################################################## |
---|
90 | # dump the loaded principals/policies |
---|
91 | # |
---|
92 | out = ctxt.context_principals() |
---|
93 | print "\n...final principal set..." |
---|
94 | for x in out[1]: |
---|
95 | print "%s " % x.string() |
---|
96 | print "\n" |
---|
97 | out = ctxt.context_credentials() |
---|
98 | print "\n...final policy attribute set..." |
---|
99 | for c in out[1]: |
---|
100 | print "%s <- %s" % (c.head_string(), c.tail_string()) |
---|
101 | print "\n" |
---|