aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/protocol2/auto.py
blob: 1c4293ac5e5288c87be2e0aea614e9212e1ec4ff (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from __future__ import (absolute_import, print_function, division, unicode_literals)
from .layer import Layer


class AutoLayer(Layer):
    def __call__(self):
        d = self.client_conn.rfile.peek(1)
        if d[0] == "\x16":
            layer = SslLayer(self, True, True)
        else:
            layer = TcpLayer(self)
        for m in layer():
            yield m

from .rawtcp import TcpLayer
from .ssl import SslLayer