コード例 #1
0
ファイル: client.py プロジェクト: dash1291/rest-security
    def create_request(self, url, **kwargs):
        headers = kwargs['headers']
        data = kwargs['data']

        params = {
            'certificate': self.local_certificate,
            'digest_algo': 'SHA1',
            'signature_algo': 'RSA-SHA1',
            'msg_encrypt_algo': 'AES',
            'key_encrypt_algo': 'RSA',
            'payload': data,
            'remote_public_key': self.server_certificate.public_key,
            'url': url,
            'is_request': True,
            'headers': headers 
        }

        rm = OutboundMessage(**params)
        rm.encrypt()
        (headers, content) = rm.to_message_data()

        return (url, headers, content)
コード例 #2
0
ファイル: utils.py プロジェクト: dash1291/rest-security
    def process_response(self, request, response):
        prt_list = PROTECT_LIST + ['/session_token/']

        if response.status_code == 403:
            return response

        if not (request.path in prt_list):
            return None

        server_cert = DjangoCertificateModel.get()

        params = {
            'certificate': server_cert,
            'digest_algo': 'SHA1',
            'signature_algo': 'RSA-SHA1',
            'msg_encrypt_algo': 'AES',
            'key_encrypt_algo': 'RSA',
            'payload': response.content,
            'remote_public_key': self.client_certificate.public_key,
            'is_request': False,
            'url': '',
            'headers': {},
        }

        response_msg = OutboundMessage(**params)

        response_msg.encrypt()
        (headers, encrypted) = response_msg.to_message_data()
        
        # insert headers fetched above into the response headers
        # and replace the content with above encrypted content.
        for key in headers.keys():
            response[key] = headers[key]

        response.content = encrypted.encode('hex')
        return response