mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 10:26:23 +00:00
9aae3213b9
SSL is an outdated protocol superseeded by TLS. Although the commonly used library is called OpenSSL, it is no reason to still use outdated language for attributes.
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import time
|
|
|
|
from mitmproxy.net import websockets
|
|
from pathod import language
|
|
from mitmproxy import exceptions
|
|
|
|
|
|
class WebsocketsProtocol:
|
|
|
|
def __init__(self, pathod_handler):
|
|
self.pathod_handler = pathod_handler
|
|
|
|
def handle_websocket(self, logger):
|
|
while True:
|
|
with logger.ctx() as lg:
|
|
started = time.time()
|
|
try:
|
|
frm = websockets.Frame.from_file(self.pathod_handler.rfile)
|
|
except exceptions.NetlibException as e:
|
|
lg("Error reading websocket frame: %s" % e)
|
|
return None, None
|
|
ended = time.time()
|
|
lg(repr(frm))
|
|
retlog = dict(
|
|
type="inbound",
|
|
protocol="websockets",
|
|
started=started,
|
|
duration=ended - started,
|
|
frame=dict(
|
|
),
|
|
cipher=None,
|
|
)
|
|
if self.pathod_handler.tls_established:
|
|
retlog["cipher"] = self.pathod_handler.get_current_cipher()
|
|
self.pathod_handler.addlog(retlog)
|
|
ld = language.websockets.NESTED_LEADER
|
|
if frm.payload.startswith(ld):
|
|
nest = frm.payload[len(ld):]
|
|
try:
|
|
wf_gen = language.parse_websocket_frame(nest.decode())
|
|
except language.exceptions.ParseException as v:
|
|
logger.write(
|
|
"Parse error in reflected frame specifcation:"
|
|
" %s" % v.msg
|
|
)
|
|
return None, None
|
|
for frm in wf_gen:
|
|
with logger.ctx() as lg:
|
|
frame_log = language.serve(
|
|
frm,
|
|
self.pathod_handler.wfile,
|
|
self.pathod_handler.settings
|
|
)
|
|
lg("crafting websocket spec: %s" % frame_log["spec"])
|
|
self.pathod_handler.addlog(frame_log)
|