forked from Frichetten/aws_api_shapeshifter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
api_signer.py
180 lines (120 loc) · 7.18 KB
/
api_signer.py
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
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
import sys, os, base64, datetime, hashlib, hmac
import requests
from urllib3.exceptions import InsecureRequestWarning
# Suppress only the single warning from urllib3 needed.
requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning)
def _sign(key, msg):
return hmac.new(key, msg.encode("utf-8"), hashlib.sha256).digest()
def _getSignatureKey(key, date_stamp, regionName, serviceName):
kDate = _sign(('AWS4' + key).encode('utf-8'), date_stamp)
kRegion = _sign(kDate, regionName)
kService = _sign(kRegion, serviceName)
kSigning = _sign(kService, 'aws4_request')
return kSigning
def query_signer(credentials, method, endpoint_prefix,
host, region, endpoint,
request_uri, formatted_request):
request_parameters = formatted_request['body']
t = datetime.datetime.utcnow()
date_stamp = t.strftime('%Y%m%d')
canonical_uri = request_uri
## Step 3: Create the canonical query string. In this example, request
# parameters are passed in the body of the request and the query string
# is blank.
canonical_querystring = ''
canonical_headers = _build_canonical_headers(formatted_request['headers'])
signed_headers = _build_signed_headers(formatted_request['headers'])
payload_hash = hashlib.sha256(request_parameters.encode('utf-8')).hexdigest()
canonical_request = method + '\n' + canonical_uri + '\n' + canonical_querystring + '\n' + canonical_headers + '\n' + signed_headers + '\n' + payload_hash
algorithm = 'AWS4-HMAC-SHA256'
credential_scope = date_stamp + '/' + region + '/' + endpoint_prefix + '/' + 'aws4_request'
string_to_sign = algorithm + '\n' + formatted_request['headers']['X-Amz-Date'] + '\n' + credential_scope + '\n' + hashlib.sha256(canonical_request.encode('utf-8')).hexdigest()
signing_key = _getSignatureKey(credentials.secret_key, date_stamp, region, endpoint_prefix)
signature = hmac.new(signing_key, (string_to_sign).encode('utf-8'), hashlib.sha256).hexdigest()
authorization_header = algorithm + ' ' + 'Credential=' + credentials.access_key + '/' + credential_scope + ', ' + 'SignedHeaders=' + signed_headers + ', ' + 'Signature=' + signature
headers = formatted_request['headers']
headers['Authorization'] = authorization_header
r = requests.request(method, endpoint, data=request_parameters, headers=headers)
headers.pop("Authorization")
return r
def json_signer(credentials, method, endpoint_prefix,
host, region, endpoint, signing_name,
request_uri, formatted_request):
request_parameters = formatted_request['body']
t = datetime.datetime.utcnow()
date_stamp = t.strftime('%Y%m%d')
canonical_uri = request_uri
## Step 3: Create the canonical query string. In this example, request
# parameters are passed in the body of the request and the query string
# is blank. THIS BREAKS SIGNING. SIGN THIS STRING TO FIND THAT WEIRD THING
canonical_querystring = ''
canonical_headers = _build_canonical_headers(formatted_request['headers'])
signed_headers = _build_signed_headers(formatted_request['headers'])
payload_hash = hashlib.sha256(request_parameters.encode('utf-8')).hexdigest()
canonical_request = method + '\n' + canonical_uri + '\n' + canonical_querystring + '\n' + canonical_headers + '\n' + signed_headers + '\n' + payload_hash
algorithm = 'AWS4-HMAC-SHA256'
credential_scope = date_stamp + '/' + region + '/' + signing_name + '/' + 'aws4_request'
string_to_sign = algorithm + '\n' + formatted_request['headers']['X-Amz-Date'] + '\n' + credential_scope + '\n' + hashlib.sha256(canonical_request.encode('utf-8')).hexdigest()
signing_key = _getSignatureKey(credentials.secret_key, date_stamp, region, signing_name)
signature = hmac.new(signing_key, (string_to_sign).encode('utf-8'), hashlib.sha256).hexdigest()
authorization_header = algorithm + ' ' + 'Credential=' + credentials.access_key + '/' + credential_scope + ', ' + 'SignedHeaders=' + signed_headers + ', ' + 'Signature=' + signature
headers = formatted_request['headers']
headers['Authorization'] = authorization_header
r = requests.request(method, endpoint+canonical_uri, data=request_parameters, headers=headers)
headers.pop("Authorization")
return r
def rest_json_signer(credentials, method, endpoint_prefix,
host, region, endpoint, signing_name,
request_uri, formatted_request):
# For json we supply {}
# For rest-json we do NOT provide {}
request_parameters = formatted_request['body']
if request_parameters == "{}":
request_parameters = ""
t = datetime.datetime.utcnow()
date_stamp = t.strftime('%Y%m%d')
# TODO: UGLY HACK PLEASE FIX
if "prod" in host:
canonical_uri_in_request = "/prod" + request_uri
else:
canonical_uri_in_request = request_uri
canonical_uri = request_uri
## Step 3: Create the canonical query string. In this example, request
# parameters are passed in the body of the request and the query string
# is blank. THIS BREAKS SIGNING. SIGN THIS STRING TO FIND THAT WEIRD THING
canonical_querystring = ''
canonical_headers = _build_canonical_headers(formatted_request['headers'])
signed_headers = _build_signed_headers(formatted_request['headers'])
payload_hash = hashlib.sha256(request_parameters.encode('utf-8')).hexdigest()
canonical_request = method + '\n' + canonical_uri_in_request + '\n' + canonical_querystring + '\n' + canonical_headers + '\n' + signed_headers + '\n' + payload_hash
algorithm = 'AWS4-HMAC-SHA256'
credential_scope = date_stamp + '/' + region + '/' + signing_name + '/' + 'aws4_request'
string_to_sign = algorithm + '\n' + formatted_request['headers']['X-Amz-Date'] + '\n' + credential_scope + '\n' + hashlib.sha256(canonical_request.encode('utf-8')).hexdigest()
signing_key = _getSignatureKey(credentials.secret_key, date_stamp, region, signing_name)
signature = hmac.new(signing_key, (string_to_sign).encode('utf-8'), hashlib.sha256).hexdigest()
authorization_header = algorithm + ' ' + 'Credential=' + credentials.access_key + '/' + credential_scope + ', ' + 'SignedHeaders=' + signed_headers + ', ' + 'Signature=' + signature
headers = formatted_request['headers']
headers['Authorization'] = authorization_header
r = requests.request(method, endpoint+canonical_uri, data=request_parameters, headers=headers, verify=False)
headers.pop("Authorization")
return r
# FUZZ IDEA: Canonical headers must be sorted alphabetically
# What if we didn't do that
def _build_canonical_headers(headers):
headers_string = ""
header_list = list(headers.keys())
header_list.sort()
for header in header_list:
headers_string += header.lower() + ":" + headers[header] + "\n"
return headers_string
# FUZZ IDEA: Signed Headers must be sorted alphabetically
# What if we didn't do that
def _build_signed_headers(headers):
headers_string = ""
header_list = list(headers.keys())
header_list.sort()
for header in header_list:
headers_string += header.lower() + ";"
# Remove trailing ';'
headers_string = headers_string[:-1]
return headers_string