mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 08:11:00 +00:00
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
|
|
||
|
|
import netlib.websockets
|
||
|
|
import contrib.pyparsing as pp
|
||
|
|
from . import base, generators
|
||
|
|
|
||
|
|
|
||
|
|
class WebsocketFrame(base._Message):
|
||
|
|
comps = (
|
||
|
|
base.Body,
|
||
|
|
base.PauseAt,
|
||
|
|
base.DisconnectAt,
|
||
|
|
base.InjectAt
|
||
|
|
)
|
||
|
|
logattrs = ["body"]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def expr(klass):
|
||
|
|
parts = [i.expr() for i in klass.comps]
|
||
|
|
atom = pp.MatchFirst(parts)
|
||
|
|
resp = pp.And(
|
||
|
|
[
|
||
|
|
base.WF.expr(),
|
||
|
|
base.Sep,
|
||
|
|
pp.ZeroOrMore(base.Sep + atom)
|
||
|
|
]
|
||
|
|
)
|
||
|
|
resp = resp.setParseAction(klass)
|
||
|
|
return resp
|
||
|
|
|
||
|
|
def values(self, settings):
|
||
|
|
vals = []
|
||
|
|
if self.body:
|
||
|
|
bodygen = self.body.value.get_generator(settings)
|
||
|
|
length = len(self.body.value.get_generator(settings))
|
||
|
|
else:
|
||
|
|
bodygen = None
|
||
|
|
length = 0
|
||
|
|
frame = netlib.websockets.FrameHeader(
|
||
|
|
mask = True,
|
||
|
|
payload_length = length
|
||
|
|
)
|
||
|
|
vals = [frame.to_bytes()]
|
||
|
|
if self.body:
|
||
|
|
masker = netlib.websockets.Masker(frame.masking_key)
|
||
|
|
vals.append(
|
||
|
|
generators.TransformGenerator(
|
||
|
|
bodygen,
|
||
|
|
masker.mask
|
||
|
|
)
|
||
|
|
)
|
||
|
|
return vals
|
||
|
|
|
||
|
|
def resolve(self, settings, msg=None):
|
||
|
|
return self.__class__(
|
||
|
|
[i.resolve(settings, msg) for i in self.tokens]
|
||
|
|
)
|
||
|
|
|
||
|
|
def spec(self):
|
||
|
|
return ":".join([i.spec() for i in self.tokens])
|