def test_extract_payload_with_multipart(self): signed_mime_message = MIMEMultipart('signed', protocol="application/pkcs7-signature") payload = SMIMEHelper.extract_payload(signed_mime_message) assert_not_equal(None, payload) assert_not_equal("", payload) assert_equal(True, payload.startswith("--"))
def test_extract_payload(self): text_plain = "test message" mime_message = email.Message.Message() mime_message.set_payload(text_plain) payload = SMIMEHelper.extract_payload(mime_message) assert_not_equal(None, payload) assert_not_equal("", payload) assert_equal(text_plain, payload)
def _signature(self): if not self.is_signed: self.context.trace("signature ignored") return cert_thumbprint = self.context.agreement.outbound_agreement.message_signature_certificate.thumbprint cert_local_file_path = self.context.agreement.outbound_agreement.message_signature_certificate.local_file_path cert_pass_phrase = self.context.agreement.outbound_agreement.message_signature_certificate.pass_phrase cert_signature_algorithm = self.context.agreement.outbound_agreement.message_signature_algorithm try: f_mime_string = SMIMEHelper.format_with_cr_lf( SMIMEHelper.mime_to_string(self.mime_message)) self.mic_content = f_mime_string self.mic_algorithm = self.context.agreement.outbound_agreement.message_signature_algorithm detached_signed_message = SMIMEHelper.sign_to_mime_detached( f_mime_string, cert_local_file_path, cert_pass_phrase, cert_signature_algorithm) signature = SMIMEHelper.get_signature_from_mime(detached_signed_message) if signature is None: raise Exception("signature is none from detached signed mime message") signed_mime_message = MIMEMultipart('signed', boundary=SMIMEHelper.get_random_boundary(), protocol="application/pkcs7-signature") del signed_mime_message['MIME-Version'] signed_mime_message.set_param('micalg', self.mic_algorithm) signed_mime_message.attach(self.mime_message) signed_mime_message.attach(signature) self.body = SMIMEHelper.format_with_cr_lf(SMIMEHelper.extract_payload(signed_mime_message)) self.mime_message = signed_mime_message self.context.trace("signature finished; thumbprint: {thumbprint}, algorithm: {algorithm}", thumbprint=cert_thumbprint, algorithm=cert_signature_algorithm) except: logger.exception("signature failed; message-id: {id}".format(id=self.message_id)) raise AS2SignatureException( "signature failed; thumbprint: {thumbprint}, algorithm: {algorithm}, due to: {message}", thumbprint=cert_thumbprint, algorithm=cert_signature_algorithm, message=sys.exc_info()[1])
def encode(self): try: self._init_mode(self.context.message.message_mdn.mdn_mode) self._init_message_id(self.context.message.message_mdn) self._init_disposition(self.context.message.message_mdn) self._init_mic(self.context.message.message_is_mic, self.context.message.message_mdn) mdn_report = MIMEMultipart( 'report', boundary=SMIMEHelper.get_random_boundary(), report_type="disposition-notification") mdn_confirmation_text = self._build_confirmation_mime() mdn_feedback_content = self._build_feedback_content_mime() mdn_report.attach(mdn_confirmation_text) mdn_report.attach(mdn_feedback_content) del mdn_report['MIME-Version'] if self.context.agreement.inbound_agreement.is_mdn_signed: mdn_message = self._signature_mdn_mime(mdn_report) else: mdn_message = mdn_report self.context.trace("mdn signature ignore") mdn_headers = self._build_mdn_headers(mdn_message) mdn_body = SMIMEHelper.format_with_cr_lf( SMIMEHelper.extract_payload(mdn_message)) return mdn_headers, mdn_body except AS2MdnException as ex: self.context.trace_error(ex.msg) raise except: self.context.trace_error(sys.exc_info()[1]) raise