CrepeSR/proxy.py
2022-07-31 12:46:48 -07:00

36 lines
1.3 KiB
Python

"""MITM script for Star Rail."""
from mitmproxy import http
from mitmproxy import ctx
from mitmproxy.proxy import layer, layers
def load(loader):
# ctx.options.web_open_browser = False
# We change the connection strategy to lazy so that next_layer happens before we actually connect upstream.
ctx.options.connection_strategy = "lazy"
ctx.options.upstream_cert = False
ctx.options.ssl_insecure = True
def next_layer(nextlayer: layer.NextLayer):
ctx.log(
f"{nextlayer.context=}\n"
f"{nextlayer.data_client()[:70]=}\n"
)
sni = nextlayer.context.client.sni
if nextlayer.context.client.tls and sni and (sni.endswith("yuanshen.com") or sni.endswith("mihoyo.com") or sni.endswith("hoyoverse.com") or sni.endswith("starrails.com")):
ctx.log('sni:' + sni)
nextlayer.context.server.address = ("127.0.0.1", 443)
def request(flow: http.HTTPFlow) -> None:
# flow.request.scheme = "http"
# pretty_host takes the "Host" header of the request into account
if flow.request.pretty_url.startswith('http://log-upload-os.mihoyo.com'):
flow.response = http.Response.make(
404, # (optional) status code
b"404 not found", # (optional) content
{"Content-Type": "text/html"} # (optional) headers
)
return