1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     6

# See COPYING file for copyrights details.

     7

     8

     9

"""

    10

The TLS-PSK adapter that handles SSL connections instead of regular sockets,

    11

but using Pre Shared Keys instead of Certificates

    22

"""

    23

    24

import socket

    25

import ssl

    26

    27

try:

    28

    import sslpsk

    29

except ImportError as e:

    30

    sslpsk = None

    31

    32

from erpc.transport import TCPTransport

    33

    34

class SSLPSKClientTransport(TCPTransport):

    35

    def __init__(self, host, port, psk):

    36

        """ overrides TCPTransport's __init__ to wrap socket in SSl wrapper """

    37

        super(TCPTransport, self).__init__()

    38

        self._host = host

    39

        self._port = port

    40

        self._sock = None

    42

    43

        if sslpsk is None:

    44

             raise ImportError("sslpsk module is not available")

    45

    50

        raw_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    51

        raw_sock.setsockopt(socket.SOL_TCP, socket.TCP_NODELAY, 1)

    52

        raw_sock.connect((self._host, self._port))

    55

    56