popeize/popeproxy.py

44 lines
1 KiB
Python
Raw Permalink Normal View History

2015-02-21 17:32:43 +00:00
"""
usage: mitmdump -T --anticache --host -s popeproxy.py
"""
from libmproxy.protocol.http import decoded
import cv2
import sys
import os
import numpy as np
sys.path.insert(0, os.getcwd())
import popeize
papaj = cv2.imread("zoltamorda.png", -1)
faceCascade = cv2.CascadeClassifier('features/haarcascade_frontalface_default.xml')
types = {
'image/jpeg': '.jpg',
'image/png': '.png',
}
def response(context, flow):
if 'content-type' not in flow.response.headers:
return
ctype = flow.response.headers['content-type'][0]
if ctype not in types:
return
with decoded(flow.response):
image = cv2.imdecode(np.fromstring(flow.response.content,
dtype=np.uint8), -1)
popeize.popeize(faceCascade, image, papaj, 1.1)
success, encimg = cv2.imencode(types[ctype], image)
if success:
flow.response.content = encimg.tostring()
print '*'*20, 'length', len(flow.reply.obj.response.content)
print '*'*20, 'type', ctype