31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from firegex.nfproxy.models.tcp import TCPInputStream, TCPOutputStream, TCPClientStream, TCPServerStream
|
|
from firegex.nfproxy.models.http import HttpRequest, HttpResponse, HttpRequestHeader, HttpResponseHeader
|
|
from firegex.nfproxy.internals.data import RawPacket
|
|
from enum import Enum
|
|
|
|
type_annotations_associations = {
|
|
"tcp": {
|
|
RawPacket: RawPacket._fetch_packet,
|
|
TCPInputStream: TCPInputStream._fetch_packet,
|
|
TCPOutputStream: TCPOutputStream._fetch_packet,
|
|
},
|
|
"http": {
|
|
RawPacket: RawPacket._fetch_packet,
|
|
TCPInputStream: TCPInputStream._fetch_packet,
|
|
TCPOutputStream: TCPOutputStream._fetch_packet,
|
|
HttpRequest: HttpRequest._fetch_packet,
|
|
HttpResponse: HttpResponse._fetch_packet,
|
|
HttpRequestHeader: HttpRequestHeader._fetch_packet,
|
|
HttpResponseHeader: HttpResponseHeader._fetch_packet,
|
|
}
|
|
}
|
|
|
|
class Protocols(Enum):
|
|
TCP = "tcp"
|
|
HTTP = "http"
|
|
|
|
__all__ = [
|
|
"RawPacket",
|
|
"TCPInputStream", "TCPOutputStream", "TCPClientStream", "TCPServerStream",
|
|
"HttpRequest", "HttpResponse", "HttpRequestHeader", "HttpResponseHeader", "Protocols"
|
|
] |