1 # ============LICENSE_START===============================================
2 # Copyright (C) 2020 Nordix Foundation. All rights reserved.
3 # ========================================================================
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15 # ============LICENSE_END=================================================
24 from pathlib import Path
25 from flask import Flask, escape, request, Response
26 from jsonschema import validate
27 from var_declaration import policy_instances, policy_types, policy_status, policy_fingerprint, forced_settings, hosts_set
28 from maincommon import *
29 from time import sleep
34 app = connexion.FlaskApp(__name__, specification_dir=apipath)
35 t=[] ##varialbe for test purpose
38 @app.route('/long', methods=['GET'])
43 return Response(str(t), 200, mimetype='text/plain')
46 @app.route('/short', methods=['GET'])
50 return Response(str(t), 200, mimetype='text/plain')
53 @app.route('/', methods=['GET'])
56 return Response("OK", 200, mimetype='text/plain')
58 #Return the current and all supported yamls for the this container
59 @app.route('/container_interfaces', methods=['GET'])
60 def container_interfaces():
62 return get_supported_interfaces_response()
64 #Delete all created instances and status
65 @app.route('/deleteinstances', methods=['POST'])
66 def deleteinstances():
68 for i in policy_instances.keys():
69 policy_instances[i]={}
72 forced_settings.clear()
73 forced_settings['code']=None
74 forced_settings['delay']=None
75 policy_fingerprint.clear()
76 return Response("All policy instances deleted", 200, mimetype='text/plain')
78 #Delete all - all reset
79 @app.route('/deleteall', methods=['POST'])
82 policy_instances.clear()
85 forced_settings['code']=None
86 forced_settings['delay']=None
87 policy_fingerprint.clear()
88 return Response("All policy instances and types deleted", 200, mimetype='text/plain')
91 @app.route('/policytype', methods=['PUT'])
94 policyTypeId=request.args.get('id')
95 if (policyTypeId is None):
96 return Response('Parameter <id> missing in request', status=400, mimetype='text/plain')
100 return Response("The policy type id is not an int", 400, mimetype='text/plain')
103 data = json.loads(data)
105 return Response("The policy type is corrupt or missing", 400, mimetype='text/plain')
107 if ('name' not in data.keys() or 'description' not in data.keys() or 'policy_type_id' not in data.keys() or'create_schema' not in data.keys()):
108 return Response("The policy type missing atributes", 400, mimetype='text/plain')
111 if (policyTypeId in policy_types.keys()):
113 if (len(policy_instances[policyTypeId]) > 0):
114 return Response("The policy type already exists and instances exists", 400, mimetype='text/plain')
116 policy_types[policyTypeId]=data
117 policy_instances[policyTypeId]={}
118 return Response("Policy type " + policyTypeId + " is OK.", retcode, mimetype='text/plain')
120 #Delete a policy type
121 @app.route('/policytype', methods=['DELETE'])
122 def del_policytype():
124 policyTypeId=request.args.get('id')
125 if (policyTypeId is None):
126 return Response('Parameter <id> missing in request', status=400, mimetype='text/plain')
128 val=int(policyTypeId)
130 return Response("The policy type id is not an int", 400, mimetype='text/plain')
132 if (policyTypeId in policy_types.keys()):
133 if (len(policy_instances[policyTypeId]) > 0):
134 return Response("The policy type already exists and instances exists", 400, mimetype='text/plain')
136 del policy_types[policyTypeId]
137 del policy_instances[policyTypeId]
138 return Response("Policy type " + policyTypeId + " is OK.", 204, mimetype='text/plain')
140 return Response("Policy type " + policyTypeId + " not found.", 204, mimetype='text/plain')
143 # Get all policy type ids
144 @app.route('/policytypes', methods=['GET'])
145 def get_policytype_ids():
147 return (json.dumps(list(policy_instances.keys())), 200)
149 #Set force response for one A1 response
150 #/forceresponse?code=<responsecode>
151 @app.route('/forceresponse', methods=['POST'])
155 forced_settings['code']=int(request.args.get('code'))
157 forced_settings['code']=None
158 return Response("Force response code: " + str(forced_settings['code']) + " set for one single A1 response", 200, mimetype='text/plain')
160 #Set force delay response, in seconds, for all A1 responses
161 #/froceesponse?delay=<seconds>
162 @app.route('/forcedelay', methods=['POST'])
166 forced_settings['delay']=int(request.args.get('delay'))
168 forced_settings['delay']=None
169 return Response("Force delay: " + str(forced_settings['delay']) + " sec set for all A1 responses", 200, mimetype='text/plain')
172 #Set status and reason
173 #/status?policyid=<policyid>&status=<status>[&deleted=<boolean>][&created_at=<timestamp>]
174 @app.route('/status', methods=['PUT'])
177 policyId=request.args.get('policyid')
178 if (policyId is None):
179 return Response('Parameter <policyid> missing in request', status=400, mimetype='text/plain')
181 if policyId not in policy_status.keys():
182 return Response('Policyid: '+policyId+' not found.', status=404, mimetype='text/plain')
183 status=request.args.get('status')
185 return Response('Parameter <status> missing in request', status=400, mimetype='text/plain')
186 policy_status[policyId]["instance_status"]=status
187 msg = "Status set to "+status
188 deleted_policy=request.args.get('deleted')
189 if (deleted_policy is not None):
190 policy_status[policyId]["has_been_deleted"]=deleted_policy
191 msg = msg + " and has_been_deleted set to "+deleted_policy
192 created_at = request.args.get('created_at')
193 if (created_at is not None):
194 policy_status[policyId]["created_at"]=created_at
195 msg = msg + " and created_at set to "+created_at
196 msg=msg + " for policy: " + policyId
197 return Response(msg, 200, mimetype='text/plain')
202 @app.route('/counter/<string:countername>', methods=['GET'])
203 def getCounter(countername):
205 if (countername == "num_instances"):
206 return Response(str(len(policy_fingerprint)), 200, mimetype='text/plain')
207 elif (countername == "num_types"):
208 return Response(str(len(policy_instances)),200, mimetype='text/plain')
209 elif (countername == "interface"):
212 return Response(str(pp[len(pp)-1]),200, mimetype='text/plain')
213 elif (countername == "remote_hosts"):
214 hosts=",".join(hosts_set)
215 return str(hosts),200
217 return Response("Counter name: "+countername+" not found.",404, mimetype='text/plain')
220 if len(sys.argv) >= 2:
221 if isinstance(sys.argv[1], int):
222 port_number = sys.argv[1]
224 app.add_api('openapi.yaml')
226 app.run(port=port_number, host="127.0.0.1", threaded=False)