1"""SSL Session 2 3Copyright (c) 1999-2003 Ng Pheng Siong. All rights reserved.""" 4 5__all__ = ['Session', 'load_session'] 6 7from M2Crypto import BIO, Err, m2 8from M2Crypto.SSL import SSLError 9from typing import AnyStr # noqa 10 11 12class Session(object): 13 14 m2_ssl_session_free = m2.ssl_session_free 15 16 def __init__(self, session, _pyfree=0): 17 # type: (bytes, int) -> None 18 assert session is not None 19 self.session = session 20 self._pyfree = _pyfree 21 22 def __del__(self): 23 # type: () -> None 24 if getattr(self, '_pyfree', 0): 25 self.m2_ssl_session_free(self.session) 26 27 def _ptr(self): 28 # type: () -> bytes 29 return self.session 30 31 def as_text(self): 32 # type: () -> bytes 33 buf = BIO.MemoryBuffer() 34 m2.ssl_session_print(buf.bio_ptr(), self.session) 35 return buf.read_all() 36 37 def as_der(self): 38 # type: () -> bytes 39 buf = BIO.MemoryBuffer() 40 m2.i2d_ssl_session(buf.bio_ptr(), self.session) 41 return buf.read_all() 42 43 def write_bio(self, bio): 44 # type: (BIO.BIO) -> int 45 return m2.ssl_session_write_bio(bio.bio_ptr(), self.session) 46 47 def get_time(self): 48 # type: () -> int 49 return m2.ssl_session_get_time(self.session) 50 51 def set_time(self, t): 52 # type: (int) -> int 53 return m2.ssl_session_set_time(self.session, t) 54 55 def get_timeout(self): 56 # type: () -> int 57 return m2.ssl_session_get_timeout(self.session) 58 59 def set_timeout(self, t): 60 # type: (int) -> int 61 return m2.ssl_session_set_timeout(self.session, t) 62 63 64def load_session(pemfile): 65 # type: (AnyStr) -> Session 66 with BIO.openfile(pemfile) as f: 67 cptr = m2.ssl_session_read_pem(f.bio_ptr()) 68 69 return Session(cptr, 1) 70