48 lines
985 B
Python
48 lines
985 B
Python
|
"""
|
||
|
|
A mitmproxy script that introduces certain request failures in a deterministic
|
||
|
|
way.
|
||
|
|
|
||
|
|
Used mainly for Matrix style requests.
|
||
|
|
|
||
|
|
To run execute it with mitmproxy:
|
||
|
|
|
||
|
|
>>> mitmproxy -s failures.py`
|
||
|
|
|
||
|
|
"""
|
||
|
|
import time
|
||
|
|
import json
|
||
|
|
|
||
|
|
from mitmproxy import http
|
||
|
|
from mitmproxy.script import concurrent
|
||
|
|
|
||
|
|
REQUEST_COUNT = 0
|
||
|
|
|
||
|
|
|
||
|
|
@concurrent
|
||
|
|
def request(flow):
|
||
|
|
global REQUEST_COUNT
|
||
|
|
|
||
|
|
REQUEST_COUNT += 1
|
||
|
|
|
||
|
|
if REQUEST_COUNT % 2 == 0:
|
||
|
|
return
|
||
|
|
elif REQUEST_COUNT % 3 == 0:
|
||
|
|
flow.response = http.HTTPResponse.make(
|
||
|
|
500,
|
||
|
|
b"Gateway error",
|
||
|
|
)
|
||
|
|
elif REQUEST_COUNT % 7 == 0:
|
||
|
|
if "sync" in flow.request.pretty_url:
|
||
|
|
time.sleep(60)
|
||
|
|
else:
|
||
|
|
time.sleep(30)
|
||
|
|
else:
|
||
|
|
flow.response = http.HTTPResponse.make(
|
||
|
|
429,
|
||
|
|
json.dumps({
|
||
|
|
"errcode": "M_LIMIT_EXCEEDED",
|
||
|
|
"error": "Too many requests",
|
||
|
|
"retry_after_ms": 2000
|
||
|
|
})
|
||
|
|
)
|