mirror of
https://github.com/gnh1201/caterpillar.git
synced 2025-02-06 15:05:00 +00:00
101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
#!/usr/bin/python3
|
|
#
|
|
# wayback.py
|
|
# Cached previous page (e.g. Wayback Machine) integration plugin for Caterpillar Proxy
|
|
#
|
|
# Caterpillar Proxy - The simple and parasitic web proxy with SPAM filter
|
|
# Namyheon Go (Catswords Research) <gnh1201@gmail.com>
|
|
# https://github.com/gnh1201/caterpillar
|
|
# Created at: 2024-03-13
|
|
# Updated at: 2024-03-13
|
|
#
|
|
|
|
import requests
|
|
|
|
from server import Extension
|
|
|
|
try:
|
|
client_encoding = config('CLIENT_ENCODING')
|
|
except Exception as e:
|
|
print ("[*] Invaild configration: %s" % (str(e)))
|
|
|
|
def get_cached_page_from_google(url):
|
|
status_code, text = (0, '')
|
|
|
|
# Google Cache URL
|
|
google_cache_url = "https://webcache.googleusercontent.com/search?q=cache:" + url
|
|
|
|
# Send a GET request to Google Cache URL
|
|
response = requests.get(google_cache_url)
|
|
|
|
# Check if the request was successful (status code 200)
|
|
if response.status_code == 200:
|
|
text = response.text # Extract content from response
|
|
else:
|
|
status_code = response.status_code
|
|
|
|
return status_code, text
|
|
|
|
# API documentation: https://archive.org/help/wayback_api.php
|
|
def get_cached_page_from_wayback(url):
|
|
status_code, text = (0, '')
|
|
|
|
# Wayback Machine API URL
|
|
wayback_api_url = "http://archive.org/wayback/available?url=" + url
|
|
|
|
# Send a GET request to Wayback Machine API
|
|
response = requests.get(wayback_api_url)
|
|
|
|
# Check if the request was successful (status code 200)
|
|
if response.status_code == 200:
|
|
try:
|
|
# Parse JSON response
|
|
data = response.json()
|
|
archived_snapshots = data.get("archived_snapshots", {})
|
|
closest_snapshot = archived_snapshots.get("closest", {})
|
|
|
|
# Check if the URL is available in the archive
|
|
if closest_snapshot:
|
|
archived_url = closest_snapshot.get("url", "")
|
|
|
|
# If URL is available, fetch the content of the archived page
|
|
if archived_url:
|
|
archived_page_response = requests.get(archived_url)
|
|
status_code = archived_page_response.status_code;
|
|
if status_code == 200:
|
|
text = archived_page_response.text
|
|
else:
|
|
status_code = 404
|
|
else:
|
|
status_code = 404
|
|
except:
|
|
status_code = 502
|
|
else:
|
|
status_code = response.status_code
|
|
|
|
return status_code, text
|
|
|
|
class Wayback(Extension):
|
|
def __init__(self):
|
|
self.type = "connector" # this is a connctor
|
|
self.connection_type = "wayback"
|
|
|
|
def connect(self, conn, data, webserver, port, scheme, method, url):
|
|
connected = False
|
|
|
|
target_url = url.decode(client_encoding)
|
|
|
|
if not connected:
|
|
status_code, text = get_cached_page_from_google(target_url)
|
|
if status_code == 200:
|
|
conn.send(text.encode(client_encoding))
|
|
connected = True
|
|
|
|
if not connected:
|
|
status_code, text = get_cached_page_from_wayback(target_url)
|
|
if status_code == 200:
|
|
conn.send(text.encode(client_encoding))
|
|
connected = True
|
|
|
|
return connected
|