1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
import argparse
import logging
from importlib.machinery import SourceFileLoader
from utils.pdp import *
from utils.models import *
from utils.policies import *
parser = argparse.ArgumentParser()
parser.add_argument('filename', help='scenario filename', nargs=1)
parser.add_argument("--verbose", "-v", action='store_true', help="verbose mode")
args = parser.parse_args()
FORMAT = '%(asctime)-15s %(levelname)s %(message)s'
logging.basicConfig(
format=FORMAT,
level=logging.WARNING)
logger = logging.getLogger(__name__)
if args.filename:
print("Loading: {}".format(args.filename[0]))
m = SourceFileLoader("scenario", args.filename[0])
scenario = m.load_module()
def create_model():
if args.verbose:
logger.warning("Creating model {}".format(scenario.model_name))
_model_id = add_model(name=scenario.model_name)
for cat in scenario.subject_categories:
scenario.subject_categories[cat] = add_subject_category(name=cat)
for cat in scenario.object_categories:
scenario.object_categories[cat] = add_object_category(name=cat)
for cat in scenario.action_categories:
scenario.action_categories[cat] = add_action_category(name=cat)
sub_cat = []
ob_cat = []
act_cat = []
meta_rule_list = []
for item_name, item_value in scenario.meta_rule.items():
for item in item_value["value"]:
if item in scenario.subject_categories:
sub_cat.append(scenario.subject_categories[item])
elif item in scenario.object_categories:
ob_cat.append(scenario.object_categories[item])
elif item in scenario.action_categories:
act_cat.append(scenario.action_categories[item])
meta_rule_id = add_meta_rule(item_name, sub_cat, ob_cat, act_cat)
item_value["id"] = meta_rule_id
meta_rule_list.append(meta_rule_id)
return _model_id, meta_rule_list
def create_policy(model_id, meta_rule_list):
if args.verbose:
logger.warning("Creating policy {}".format(scenario.policy_name))
policy_id = add_policy(name=scenario.policy_name)
update_policy(policy_id, model_id)
for meta_rule_id in meta_rule_list:
add_meta_rule_to_model(model_id, meta_rule_id)
for subject_cat_name in scenario.subject_data:
for subject_data_name in scenario.subject_data[subject_cat_name]:
data_id = scenario.subject_data[subject_cat_name][subject_data_name] = add_subject_data(
policy_id=policy_id,
category_id=scenario.subject_categories[subject_cat_name], name=subject_data_name)
scenario.subject_data[subject_cat_name][subject_data_name] = data_id
for object_cat_name in scenario.object_data:
for object_data_name in scenario.object_data[object_cat_name]:
data_id = scenario.object_data[object_cat_name][object_data_name] = add_object_data(
policy_id=policy_id,
category_id=scenario.object_categories[object_cat_name], name=object_data_name)
scenario.object_data[object_cat_name][object_data_name] = data_id
for action_cat_name in scenario.action_data:
for action_data_name in scenario.action_data[action_cat_name]:
data_id = scenario.action_data[action_cat_name][action_data_name] = add_action_data(
policy_id=policy_id,
category_id=scenario.action_categories[action_cat_name], name=action_data_name)
scenario.action_data[action_cat_name][action_data_name] = data_id
for name in scenario.subjects:
scenario.subjects[name] = add_subject(policy_id, name=name)
for name in scenario.objects:
scenario.objects[name] = add_object(policy_id, name=name)
for name in scenario.actions:
scenario.actions[name] = add_action(policy_id, name=name)
for subject_name in scenario.subject_assignments:
for subject_category_name in scenario.subject_assignments[subject_name]:
subject_id = scenario.subjects[subject_name]
subject_cat_id = scenario.subject_categories[subject_category_name]
subject_data_id = scenario.subject_data[subject_category_name][scenario.subject_assignments[subject_name][subject_category_name]]
add_subject_assignments(policy_id, subject_id, subject_cat_id, subject_data_id)
for object_name in scenario.object_assignments:
for object_category_name in scenario.object_assignments[object_name]:
object_id = scenario.objects[object_name]
object_cat_id = scenario.object_categories[object_category_name]
object_data_id = scenario.object_data[object_category_name][scenario.object_assignments[object_name][object_category_name]]
add_object_assignments(policy_id, object_id, object_cat_id, object_data_id)
for action_name in scenario.action_assignments:
for action_category_name in scenario.action_assignments[action_name]:
action_id = scenario.actions[action_name]
action_cat_id = scenario.action_categories[action_category_name]
action_data_id = scenario.action_data[action_category_name][scenario.action_assignments[action_name][action_category_name]]
add_action_assignments(policy_id, action_id, action_cat_id, action_data_id)
for meta_rule_name in scenario.rules:
meta_rule_value = scenario.meta_rule[meta_rule_name]
for rule in scenario.rules[meta_rule_name]:
data_list = []
_meta_rule = list(meta_rule_value["value"])
for data_name in rule["rule"]:
category_name = _meta_rule.pop(0)
if category_name in scenario.subject_categories:
data_list.append(scenario.subject_data[category_name][data_name])
elif category_name in scenario.object_categories:
data_list.append(scenario.object_data[category_name][data_name])
elif category_name in scenario.action_categories:
data_list.append(scenario.action_data[category_name][data_name])
instructions = rule["instructions"]
add_rule(policy_id, meta_rule_value["id"], data_list, instructions)
return policy_id
def create_pdp(policy_id=None):
if args.verbose:
logger.warning("Creating PDP {}".format(scenario.pdp_name))
projects = get_keystone_projects()
admin_project_id = None
for _project in projects['projects']:
if _project['name'] == "admin":
admin_project_id = _project['id']
assert admin_project_id
pdps = check_pdp()["pdps"]
for pdp_id, pdp_value in pdps.items():
if scenario.pdp_name == pdp_value["name"]:
update_pdp(pdp_id, policy_id=policy_id)
return pdp_id
_pdp_id = add_pdp(name=scenario.pdp_name, policy_id=policy_id)
map_to_keystone(pdp_id=_pdp_id, keystone_project_id=admin_project_id)
return _pdp_id
if __name__ == "__main__":
model_id, meta_rule_list = create_model()
policy_id = create_policy(model_id, meta_rule_list)
pdp_id = create_pdp(policy_id)
|
