mitmproxy mitmdump script that injects 404s for streaming chunks

404-ts-mitm.py

#!/usr/bin/env python
from mitmproxy import ctx
 
# an intermittent 404 for .ts files - HLS m3u8 streaming chunks
  
def start():
    return Foo()
 
class Foo:
  MUCHO = 0
  def response(self, flow):
    flow.response.headers["x-myproxy"] = "proxy"
    if not flow.request.path.endswith('.ts'):
        #and not flow.request.path.endswith('.m3u8'):
        return
    ctx.log.info("MUCHO {} - {}".format(self.MUCHO, flow.request.path))
    self.MUCHO += 1
    if not (self.MUCHO % 20 == 0): 
        # only every 20 or so return 404 - so it's intermittent.
        return
    if flow.response.status_code != 404:
        headers = flow.response.headers
        del headers["access-control-allow-origin"]
        flow.response.status_code = 404
        flow.response.reason = "We're in charge"
    else:
        flow.response.headers["x-myproxy"] = "404 not ours"