summaryrefslogtreecommitdiff
path: root/M2Crypto/SSL/Session.py
blob: c364d4efc18431219bb6cf9d2787982473901d75 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
"""SSL Session

Copyright (c) 1999-2003 Ng Pheng Siong. All rights reserved."""

__all__ = ['Session', 'load_session']

from M2Crypto import BIO, Err, m2
from M2Crypto.SSL import SSLError
from typing import AnyStr  # noqa


class Session(object):

    m2_ssl_session_free = m2.ssl_session_free

    def __init__(self, session, _pyfree=0):
        # type: (bytes, int) -> None
        assert session is not None
        self.session = session
        self._pyfree = _pyfree

    def __del__(self):
        # type: () -> None
        if getattr(self, '_pyfree', 0):
            self.m2_ssl_session_free(self.session)

    def _ptr(self):
        # type: () -> bytes
        return self.session

    def as_text(self):
        # type: () -> bytes
        buf = BIO.MemoryBuffer()
        m2.ssl_session_print(buf.bio_ptr(), self.session)
        return buf.read_all()

    def as_der(self):
        # type: () -> bytes
        buf = BIO.MemoryBuffer()
        m2.i2d_ssl_session(buf.bio_ptr(), self.session)
        return buf.read_all()

    def write_bio(self, bio):
        # type: (BIO.BIO) -> int
        return m2.ssl_session_write_bio(bio.bio_ptr(), self.session)

    def get_time(self):
        # type: () -> int
        return m2.ssl_session_get_time(self.session)

    def set_time(self, t):
        # type: (int) -> int
        return m2.ssl_session_set_time(self.session, t)

    def get_timeout(self):
        # type: () -> int
        return m2.ssl_session_get_timeout(self.session)

    def set_timeout(self, t):
        # type: (int) -> int
        return m2.ssl_session_set_timeout(self.session, t)


def load_session(pemfile):
    # type: (AnyStr) -> Session
    with BIO.openfile(pemfile) as f:
        cptr = m2.ssl_session_read_pem(f.bio_ptr())

    return Session(cptr, 1)