Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 02:10:59 +00:00
Code Issues Projects Releases Wiki Activity

fix #3833

Browse Source
This commit is contained in:
naivekun 2020-02-28 19:15:42 +08:00
parent 6d3b8c9716
commit 3c1ced2b17
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mitmproxy/net/http/headers.py
View File

@ -162,8 +162,12 @@ class Headers(multidict.MultiDict):
pattern = re.compile(pattern, flags) pattern = re.compile(pattern, flags)
replacements = 0 replacements = 0
flag_count = count > 0 flag_count = count > 0
count_reached = False
fields = [] fields = []
for name, value in self.fields: for name, value in self.fields:
if count_reached:
fields.append((name, value))
continue
line, n = pattern.subn(repl, name + b": " + value, count=count) line, n = pattern.subn(repl, name + b": " + value, count=count)
try: try:
name, value = line.split(b": ", 1) name, value = line.split(b": ", 1)
@ -173,10 +177,12 @@ class Headers(multidict.MultiDict):
pass pass
else: else:
replacements += n replacements += n
fields.append((name, value))
if flag_count: if flag_count:
count -= n count -= n
if count == 0: if count == 0:
break count_reached = True
continue
fields.append((name, value)) fields.append((name, value))
self.fields = tuple(fields) self.fields = tuple(fields)
return replacements return replacements