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
|
# $Id$
# Copyright (C) 2014 Dragon Research Labs ("DRL")
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND DRL DISCLAIMS ALL WARRANTIES WITH
# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS. IN NO EVENT SHALL DRL BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
# OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
# PERFORMANCE OF THIS SOFTWARE.
"""
HTTP using Python standard libraries, for RPKI programs that don't
need the full-blown rpki.http asynchronous code.
"""
import logging
import httplib
import urlparse
import BaseHTTPServer
logger = logging.getLogger(__name__)
rpki_content_type = "application/x-rpki"
class HTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
"""
HTTP request handler simple RPKI servers.
"""
def do_POST(self):
try:
content_type = self.headers.get("Content-Type")
content_length = self.headers.get("Content-Length")
if content_type != rpki_content_type:
self.send_error(415, "No handler for Content-Type %s" % content_type)
else:
for prefix, handler in self.rpki_handlers:
if self.path.startswith(prefix):
handler(self, (self.rfile.read()
if content_length is None else
self.rfile.read(int(content_length))))
break
else:
self.send_error(404, "No handler for path %s" % self.path)
except Exception, e:
logger.exception("Unhandled exception")
self.send_error(501, "Unhandled exception")
def send_cms_response(self, der):
self.send_response(200)
self.send_header("Content-Type", rpki_content_type)
self.send_header("Content-Length", str(len(der)))
self.end_headers()
self.wfile.write(der)
def log_message(self, *args):
logger.info(*args, extra = dict(context = "%s:%s" % self.client_address))
def send_error(self, code, message = None):
# BaseHTTPRequestHandler.send_error() generates HTML error messages,
# which we don't want, so we override the method to suppress this.
self.send_response(code, message)
self.send_header("Content-Type", rpki_content_type)
self.send_header("Connection", "close")
self.end_headers()
def server(handlers, port, host = ""):
"""
Run an HTTP server and wait (forever) for connections.
"""
if not isinstance(handlers, (tuple, list)):
handlers = (("/", handlers),)
class RequestHandler(HTTPRequestHandler):
rpki_handlers = handlers
BaseHTTPServer.HTTPServer((host, port), RequestHandler).serve_forever()
class BadURL(Exception):
"Bad contact URL"
class RequestFailed(Exception):
"HTTP returned failure"
class BadContentType(Exception):
"Wrong HTTP Content-Type"
def client(url, query):
"""
Issue single a query and return the response.
Might want to add CMS processing here, not sure yet.
"""
u = urlparse.urlparse(url)
if u.scheme not in ("", "http") or u.username or u.password or u.params or u.query or u.fragment:
raise BadURL("Unusable URL %s", url)
http = httplib.HTTPConnection(u.hostname, u.port or httplib.HTTP_PORT)
http.request("POST", u.path, query, {"Content-Type" : rpki_content_type})
r = http.getresponse()
if r.status != 200:
raise RequestFailed("HTTP request failed with status %r reason %r" % (r.status, r.reason))
if r.getheader("Content-Type") != rpki_content_type:
raise BadContentType("HTTP Content-Type %r, expected %r" % (r.getheader("Content-Type"), rpki_content_type))
return r.read()
|
