gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-09-05 17:31:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix a cache overfitting issue: use re.IGNORECASE

Browse Source
This commit is contained in:
Namhyeon Go 2024-11-04 17:17:19 +09:00 committed by GitHub
parent 3af8879adb
commit 99f960307d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server.py
View File

@ -14,6 +14,7 @@ import argparse
import socket import socket
import sys import sys
import os import os
import re
from _thread import * from _thread import *
from subprocess import PIPE, Popen from subprocess import PIPE, Popen
import base64 import base64
@ -159,8 +160,11 @@ def conn_string(conn: socket.socket, data: bytes, addr: bytes):
local_domains = list(filter(None, map(str.strip, local_domain.split(',')))) local_domains = list(filter(None, map(str.strip, local_domain.split(','))))
for domain in local_domains: for domain in local_domains:
localserver = domain.encode(client_encoding) localserver = domain.encode(client_encoding)
if webserver == localserver or data.find(b"\nHost: " + localserver) > -1:
logger.info("[*] Detected the reverse proxy request: %s" % local_domain) # Fix a cache overfitting issue: use re.IGNORECASE
host_pattern = re.compile(rb"\n\s*host\s*:\s*" + re.escape(localserver), re.IGNORECASE)
if webserver == localserver or host_pattern.search(data):
logger.info("[*] Reverse proxy requested: %s" % local_domain)
scheme, _webserver, _port = proxy_pass.encode(client_encoding).split(b":") scheme, _webserver, _port = proxy_pass.encode(client_encoding).split(b":")
webserver = _webserver[2:] webserver = _webserver[2:]
port = int(_port.decode(client_encoding)) port = int(_port.decode(client_encoding))