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