def build_sync_client(request):
# Get the BID assertion
is_authorization_defined = AUTHORIZATION_HEADER in request.headers
starts_with_browser_id = False
if is_authorization_defined:
authorization = request.headers[AUTHORIZATION_HEADER].lower()
starts_with_browser_id = authorization.startswith("browserid ")
if not is_authorization_defined or not starts_with_browser_id:
msg = "Provide a BID assertion %s header." % AUTHORIZATION_HEADER
response = http_error(httpexceptions.HTTPUnauthorized(), errno=ERRORS.MISSING_AUTH_TOKEN, message=msg)
response.headers.extend(forget(request))
raise response
bucket_id = request.matchdict["bucket_id"]
is_client_state_header_defined = CLIENT_STATE_HEADER in request.headers
if bucket_id == "syncto":
if not is_client_state_header_defined:
msg = "Provide the tokenserver %s header." % CLIENT_STATE_HEADER
response = http_error(httpexceptions.HTTPUnauthorized(), errno=ERRORS.MISSING_AUTH_TOKEN, message=msg)
response.headers.extend(forget(request))
raise response
client_state = request.headers[CLIENT_STATE_HEADER]
elif len(bucket_id) != CLIENT_STATE_LENGTH:
msg = "The provided bucket ID is incorrect."
response = http_error(httpexceptions.HTTPUnauthorized(), errno=ERRORS.MISSING_AUTH_TOKEN, message=msg)
response.headers.extend(forget(request))
raise response
else:
client_state = bucket_id
if is_client_state_header_defined:
send_alert(request, "%s header is deprecated and should not be " "provided anymore." % CLIENT_STATE_HEADER)
authorization_header = request.headers[AUTHORIZATION_HEADER]
bid_assertion = authorization_header.split(" ", 1)[1]
settings = request.registry.settings
cache = request.registry.cache
statsd = request.registry.statsd
token_server_url = settings["token_server_url"]
hmac_secret = settings["cache_hmac_secret"]
cache_key = "credentials_%s" % utils.hmac_digest(hmac_secret, bid_assertion)
ca_bundle = settings["certificate_ca_bundle"]
encrypted_credentials = cache.get(cache_key)
if not encrypted_credentials:
settings_ttl = int(settings["cache_credentials_ttl_seconds"])
bid_ttl = _extract_bid_assertion_ttl(bid_assertion)
ttl = min(settings_ttl, bid_ttl or settings_ttl)
tokenserver = TokenserverClient(bid_assertion, client_state, token_server_url, verify=ca_bundle)
if statsd:
statsd.watch_execution_time(tokenserver, prefix="tokenserver")
credentials = tokenserver.get_hawk_credentials(duration=ttl)
encrypted = encrypt(json.dumps(credentials), client_state, hmac_secret)
cache.set(cache_key, encrypted, ttl)
else:
credentials = json.loads(decrypt(encrypted_credentials, client_state, hmac_secret))
if statsd:
timer = statsd.timer("syncclient.start_time")
timer.start()
sync_client = SyncClient(verify=ca_bundle, **credentials)
if statsd:
timer.stop()
statsd.watch_execution_time(sync_client, prefix="syncclient")
return sync_client
def test_decrypt_return_unicode_string(self):
message = "Salut la companie"
encrypted = crypto.encrypt(message, "Client State", "Secret")
decrypted = crypto.decrypt(encrypted, "Client State", "Secret")
self.assertIsInstance(decrypted, text_type)
def test_encrypt_can_be_decrypted(self):
message = "Salut la companie"
encrypted = crypto.encrypt(message, "Client State", "Secret")
decrypted = crypto.decrypt(encrypted, "Client State", "Secret")
self.assertEquals(message, decrypted)
def test_encrypt_returns_hex_encoded_string(self):
message = "Salut la companie"
encrypted = crypto.encrypt(message, "Client State", "Secret")
hex_regexp = re.compile(r'[0-9a-f]+')
self.assertTrue(hex_regexp.match(encrypted))
def build_sync_client(request):
# Get the BID assertion
is_authorization_defined = AUTHORIZATION_HEADER in request.headers
starts_with_browser_id = False
if is_authorization_defined:
authorization = request.headers[AUTHORIZATION_HEADER].lower()
starts_with_browser_id = authorization.startswith("browserid ")
if not is_authorization_defined or not starts_with_browser_id:
msg = "Provide a BID assertion %s header." % AUTHORIZATION_HEADER
response = http_error(httpexceptions.HTTPUnauthorized(),
errno=ERRORS.MISSING_AUTH_TOKEN,
message=msg)
response.headers.extend(forget(request))
raise response
is_client_state_defined = CLIENT_STATE_HEADER in request.headers
if not is_client_state_defined:
msg = "Provide the tokenserver %s header." % CLIENT_STATE_HEADER
response = http_error(httpexceptions.HTTPUnauthorized(),
errno=ERRORS.MISSING_AUTH_TOKEN,
message=msg)
response.headers.extend(forget(request))
raise response
authorization_header = request.headers[AUTHORIZATION_HEADER]
bid_assertion = authorization_header.split(" ", 1)[1]
client_state = request.headers[CLIENT_STATE_HEADER]
settings = request.registry.settings
cache = request.registry.cache
statsd = request.registry.statsd
token_server_url = settings['token_server_url']
hmac_secret = settings['cache_hmac_secret']
cache_key = 'credentials_%s' % utils.hmac_digest(hmac_secret,
bid_assertion)
encrypted_credentials = cache.get(cache_key)
if not encrypted_credentials:
settings_ttl = int(settings['cache_credentials_ttl_seconds'])
bid_ttl = _extract_bid_assertion_ttl(bid_assertion)
ttl = min(settings_ttl, bid_ttl or settings_ttl)
tokenserver = TokenserverClient(bid_assertion, client_state,
token_server_url)
if statsd:
statsd.watch_execution_time(tokenserver, prefix="tokenserver")
credentials = tokenserver.get_hawk_credentials(duration=ttl)
encrypted = encrypt(json.dumps(credentials), client_state, hmac_secret)
cache.set(cache_key, encrypted, ttl)
else:
credentials = json.loads(
decrypt(encrypted_credentials, client_state, hmac_secret))
if statsd:
timer = statsd.timer("syncclient.start_time")
timer.start()
sync_client = SyncClient(**credentials)
if statsd:
timer.stop()
statsd.watch_execution_time(sync_client, prefix="syncclient")
return sync_client