def test_crypto(self):
key = binascii.a2b_base64("yyUKYpE1IsMCBdmSAs7FpxWwRszfSbPBqD+Obi3P7Cg28SpLuqB4eA==")
#s = "vtpGn1C4TbvR/Z7P4qVRij/UGYwcB16D5gTR18cOF+AZhgJqSM/LkVXWAMxFWs5NK83EyZ42/6Y0kkpMJgMGw9IrRBawtsrwI8+5Ty1r+5Wi8KsMXY5hbUmujbwghuVOGpWR4WrTQ/kUARo5oE8U196t97jXVp6r7zSgPe94MyGSHw0+EDZzYXS/q1jePoHaZSSD+meyts9KGzMHAAvvN2E1"
se = cryptotool.encrypt("12345678", key)
print se
so = cryptotool.decrypt(se, key)
self.assertEqual(so, "12345678")
print so
def test_crypto(self):
key = binascii.a2b_base64("yyUKYpE1IsMCBdmSAs7FpxWwRszfSbPBqD+Obi3P7Cg28SpLuqB4eA==")
#s = "vtpGn1C4TbvR/Z7P4qVRij/UGYwcB16D5gTR18cOF+AZhgJqSM/LkVXWAMxFWs5NK83EyZ42/6Y0kkpMJgMGw9IrRBawtsrwI8+5Ty1r+5Wi8KsMXY5hbUmujbwghuVOGpWR4WrTQ/kUARo5oE8U196t97jXVp6r7zSgPe94MyGSHw0+EDZzYXS/q1jePoHaZSSD+meyts9KGzMHAAvvN2E1"
se = cryptotool.encrypt("12345678", key)
print se
so = cryptotool.decrypt(se, key)
self.assertEqual(so, "12345678")
print so
def out_protocol_filter(self, producer, queue, message, headers):
try:
# Encrypt message
cnf = bus.cnf
self._logger.debug('Encrypting message')
crypto_key = binascii.a2b_base64(cnf.read_key(self.crypto_key_path))
data = cryptotool.encrypt(message, crypto_key)
# Generate signature
signature, timestamp = cryptotool.sign_http_request(data, crypto_key)
# Add extra headers
headers['Date'] = timestamp
headers['X-Signature'] = signature
headers['X-Server-Id'] = self.server_id
return data
except (BaseException, Exception), e:
raise MessagingError('Cannot encrypt message. error: %s' % (e))
def out_protocol_filter(self, producer, queue, message, headers):
try:
# Encrypt message
cnf = bus.cnf
self._logger.debug('Encrypting message')
crypto_key = binascii.a2b_base64(cnf.read_key(self.crypto_key_path))
data = cryptotool.encrypt(message, crypto_key)
# Generate signature
signature, timestamp = cryptotool.sign_http_request(data, crypto_key)
# Add extra headers
headers['Date'] = timestamp
headers['X-Signature'] = signature
headers['X-Server-Id'] = self.server_id
return data
except (BaseException, Exception), e:
raise MessagingError('Cannot encrypt message. error: %s' % (e))
def encrypt_data(self, data):
try:
return cryptotool.encrypt(data, self._read_crypto_key())
except:
raise rpc.InvalidRequestError('Failed to encrypt data. Error: %s' % (sys.exc_info()[1], ))
def test_decrypt_data(self):
s = 'ABC'
encrypted = cryptotool.encrypt(s, self._read_crypto_key())
assert self.app.decrypt_data(encrypted) == s
def test_decrypt_data(self):
s = 'ABC'
encrypted = cryptotool.encrypt(s, self._read_crypto_key())
assert self.app.decrypt_data(encrypted) == s
def encrypt_data(self, data):
try:
return cryptotool.encrypt(data, self._read_crypto_key())
except:
raise rpc.InvalidRequestError('Failed to encrypt data. Error: %s' %
(sys.exc_info()[1], ))
