gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-02-11 09:25:00 +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 sys
import os
import re
from _thread import *
from subprocess import PIPE, Popen
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(','))))
for domain in local_domains:
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":")
webserver = _webserver[2:]
port = int(_port.decode(client_encoding))