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
|
#!/usr/bin/env python2
###
# ============LICENSE_START=======================================================
# ONAP CLAMP
# ================================================================================
# Copyright (C) 2018 AT&T Intellectual Property. All rights
# reserved.
# ================================================================================
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============LICENSE_END============================================
# ===================================================================
# ECOMP is a trademark and service mark of AT&T Intellectual Property.
###
import json
import requests
import os
import sys
import SimpleHTTPServer
import SocketServer
import argparse
import tempfile
import signal
parser = argparse.ArgumentParser(description="SDC Cache & Replay")
parser.add_argument("--username", "-u", type=str, help="Set the username for contacting SDC")
parser.add_argument("--password", "-p", type=str, help="Set the password for contacting SDC")
parser.add_argument("--root", "-r", default=tempfile.mkdtemp, type=str, help="Root folder for the proxy cache")
parser.add_argument("--proxy" , type=str, help="Url of the Act as a proxy. If not set, this script only uses the cache and will return a 404 if files aren't found")
parser.add_argument("--port", "-P", type=int, default="8081", help="Port on which the proxy should listen to")
parser.add_argument("--verbose", "-v", type=bool, help="Print more information in case of error")
options = parser.parse_args()
PORT = options.port
SDC_HOST = options.proxy
SDC_AUTH = (options.username, options.password)
SDC_HEADERS = {'X-ECOMP-InstanceID':'CLAMP'}
CACHE_ROOT = options.root
def signal_handler(signal_sent, frame):
global httpd
if signal_sent == signal.SIGINT:
print('Got Ctrl-C (SIGINT)')
httpd.socket.close()
httpd.shutdown()
httpd.server_close()
class Proxy(SimpleHTTPServer.SimpleHTTPRequestHandler):
def print_headers(self):
for header,value in self.headers.items():
print("%s : %s" % (header, value))
def check_credentials(self):
pass
def _send_content(self, header_file, content_file):
self.send_response(200)
with open(header_file, 'rb') as f:
headers = json.load(f)
for key,value in headers.items():
if key in ('Transfer-Encoding',):
continue
self.send_header(key, value)
self.end_headers()
with open(content_file,'rb') as f:
fc = f.read()
self.wfile.write(fc)
def _write_cache(self,cached_file, header_file, content_file, response):
os.makedirs(cached_file, True)
with open(content_file, 'w') as f:
f.write(response.raw.read())
with open(header_file, 'w') as f:
json.dump(dict(response.raw.headers), f)
# Entry point of the code
def do_GET(self):
self.print_headers()
self.check_credentials()
cached_file = '%s/%s' % (CACHE_ROOT, self.path,)
print("Cached file name before escaping : %s" % cached_file)
cached_file = cached_file.replace('<','<').replace('>','>').replace('?','?').replace('*','*').replace('\\','*').replace(':',':').replace('|','|')
print("Cached file name after escaping (used for cache storage) : %s" % cached_file)
cached_file_content = "%s/.file" % (cached_file,)
cached_file_header = "%s/.header" % (cached_file,)
_file_available = os.path.exists(cached_file_content)
if not _file_available and not SDC_HOST:
self.send_response(404)
return "404 Not found"
if not _file_available:
print("SDC Request for data currently not present in cache: %s" % (cached_file,))
url = '%s%s' % (SDC_HOST, self.path)
response = requests.get(url, auth=SDC_AUTH, headers=SDC_HEADERS, stream=True)
if response.status_code == 200:
self._write_cache(cached_file, cached_file_header, cached_file_content, response)
else:
print('Error when requesting file :')
print('Requested url : %s' % (url,))
print('Status code : %s' % (response.status_code,))
print('Content : %s' % (response.content,))
self.send_response(response.status_code)
return response.content
self._send_content(cached_file_header, cached_file_content)
# Main code that start the HTTP server
httpd = SocketServer.ForkingTCPServer(('', PORT), Proxy)
httpd.allow_reuse_address = True
print "Listening on port "+ str(PORT) + " and caching in " + CACHE_ROOT + "(Press Ctrl+C to stop HTTPD Caching script)"
signal.signal(signal.SIGINT, signal_handler)
httpd.serve_forever()
|
