aboutsummaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/data/pages/gotmyowndomain.html14
-rw-r--r--test/data/pages/gotmyowndomain_https.html14
-rwxr-xr-x[-rw-r--r--]test/init.sh4
-rw-r--r--test/misc_constants.py34
-rw-r--r--test/proxy_core.py4
-rwxr-xr-x[-rw-r--r--]test/server.py65
-rw-r--r--test/world_wide_library.py15
7 files changed, 110 insertions, 40 deletions
diff --git a/test/data/pages/gotmyowndomain.html b/test/data/pages/gotmyowndomain.html
new file mode 100644
index 0000000..daa9ab7
--- /dev/null
+++ b/test/data/pages/gotmyowndomain.html
@@ -0,0 +1,14 @@
+<html>
+ <head>
+ <meta name=charset value="latin1">
+ <title>
+ Schrodinger's Document
+ </title>
+ </head>
+ <body>
+ A nice, simple page for testing.
+ <script>
+ document.write('<p><b>Or so you thought...</b></p>');
+ </script>
+ </body>
+</html>
diff --git a/test/data/pages/gotmyowndomain_https.html b/test/data/pages/gotmyowndomain_https.html
new file mode 100644
index 0000000..ec2aa1f
--- /dev/null
+++ b/test/data/pages/gotmyowndomain_https.html
@@ -0,0 +1,14 @@
+<html>
+ <head>
+ <meta name=charset value="latin1">
+ <title>
+ Schrodinger's Document
+ </title>
+ </head>
+ <body>
+ A nice, simple page for testing (using HTTPS).
+ <script>
+ document.write('<p><b>Or so you thought...</b></p>');
+ </script>
+ </body>
+</html>
diff --git a/test/init.sh b/test/init.sh
index 915db76..c0e7647 100644..100755
--- a/test/init.sh
+++ b/test/init.sh
@@ -7,6 +7,10 @@
# Initialize the root certificate for the tests proxy server
# Make sure this is run in the directory where they will be put!
+
+if [ -n "$1" ]; then
+ cd "$1"
+fi
openssl genrsa -out ca.key 2048
openssl genrsa -out cert.key 2048
openssl req -new -x509 -days 3650 -key ca.key -out ca.crt -subj "/CN=Hachette Test"
diff --git a/test/misc_constants.py b/test/misc_constants.py
new file mode 100644
index 0000000..c6c134f
--- /dev/null
+++ b/test/misc_constants.py
@@ -0,0 +1,34 @@
+# Copyright (C) 2021 jahoti <jahoti@tilde.team>
+# Licensing information is collated in the `copyright` file
+
+"""
+Miscellaneous data that were found useful
+"""
+
+mime_types = {
+ "7z": "application/x-7z-compressed", "oga": "audio/ogg",
+ "abw": "application/x-abiword", "ogv": "video/ogg",
+ "arc": "application/x-freearc", "ogx": "application/ogg",
+ "bin": "application/octet-stream", "opus": "audio/opus",
+ "bz": "application/x-bzip", "otf": "font/otf",
+ "bz2": "application/x-bzip2", "pdf": "application/pdf",
+ "css": "text/css", "png": "image/png",
+ "csv": "text/csv", "sh": "application/x-sh",
+ "gif": "image/gif", "svg": "image/svg+xml",
+ "gz": "application/gzip", "tar": "application/x-tar",
+ "htm": "text/html", "ts": "video/mp2t",
+ "html": "text/html", "ttf": "font/ttf",
+ "ico": "image/vnd.microsoft.icon", "txt": "text/plain",
+ "js": "text/javascript", "wav": "audio/wav",
+ "jpeg": "image/jpeg", "weba": "audio/webm",
+ "jpg": "image/jpeg", "webm": "video/webm",
+ "json": "application/json", "woff": "font/woff",
+ "mjs": "text/javascript", "woff2": "font/woff2",
+ "mp3": "audio/mpeg", "xhtml": "application/xhtml+xml",
+ "mp4": "video/mp4", "zip": "application/zip",
+ "mpeg": "video/mpeg",
+ "odp": "application/vnd.oasis.opendocument.presentation",
+ "ods": "application/vnd.oasis.opendocument.spreadsheet",
+ "odt": "application/vnd.oasis.opendocument.text",
+ "xml": "application/xml" # text/xml if readable from casual users
+}
diff --git a/test/proxy_core.py b/test/proxy_core.py
index da90455..dd4225d 100644
--- a/test/proxy_core.py
+++ b/test/proxy_core.py
@@ -5,7 +5,7 @@
"""
The core for a "virtual network" proxy
-Be sure to run this inside your intended certificates directory.
+Be sure to set certdir to your intended certificates directory before running.
"""
import os, socket, ssl, subprocess, sys, threading, time
@@ -20,8 +20,6 @@ def popen(command, *args, **kwargs):
class ProxyRequestHandler(BaseHTTPRequestHandler):
"""Handles a network request made to the proxy"""
- certdir = ''
-
def log_error(self, format, *args):
# suppress "Request timed out: timeout('timed out',)"
if isinstance(args[0], socket.timeout):
diff --git a/test/server.py b/test/server.py
index 83a72fa..58a84bd 100644..100755
--- a/test/server.py
+++ b/test/server.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+#
# Copyright (C) 2021 jahoti <jahoti@tilde.team>
# Licensing information is collated in the `copyright` file
@@ -6,41 +8,12 @@ A modular "virtual network" proxy,
wrapping the classes in proxy_core.py
"""
-from proxy_core import *
+import proxy_core
from urllib.parse import parse_qs
+from misc_constants import *
+from world_wide_library import catalog as internet
-internet = {} # Add info here later
-mime_types = {
- "7z": "application/x-7z-compressed", "oga": "audio/ogg",
- "abw": "application/x-abiword", "ogv": "video/ogg",
- "arc": "application/x-freearc", "ogx": "application/ogg",
- "bin": "application/octet-stream", "opus": "audio/opus",
- "bz": "application/x-bzip", "otf": "font/otf",
- "bz2": "application/x-bzip2", "pdf": "application/pdf",
- "css": "text/css", "png": "image/png",
- "csv": "text/csv", "sh": "application/x-sh",
- "gif": "image/gif", "svg": "image/svg+xml",
- "gz": "application/gzip", "tar": "application/x-tar",
- "htm": "text/html", "ts": "video/mp2t",
- "html": "text/html", "ttf": "font/ttf",
- "ico": "image/vnd.microsoft.icon", "txt": "text/plain",
- "js": "text/javascript", "wav": "audio/wav",
- "jpeg": "image/jpeg", "weba": "audio/webm",
- "jpg": "image/jpeg", "webm": "video/webm",
- "json": "application/json", "woff": "font/woff",
- "mjs": "text/javascript", "woff2": "font/woff2",
- "mp3": "audio/mpeg", "xhtml": "application/xhtml+xml",
- "mp4": "video/mp4", "zip": "application/zip",
- "mpeg": "video/mpeg",
- "odp": "application/vnd.oasis.opendocument.presentation",
- "ods": "application/vnd.oasis.opendocument.spreadsheet",
- "odt": "application/vnd.oasis.opendocument.text",
- "xml": "application/xml" # text/xml if readable from casual users
-}
-
-class RequestHijacker(ProxyRequestHandler):
- certdir = global_certdir
-
+class RequestHijacker(proxy_core.ProxyRequestHandler):
def handle_request(self, req_body):
path_components = self.path.split('?', maxsplit=1)
path = path_components[0]
@@ -103,8 +76,26 @@ class RequestHijacker(ProxyRequestHandler):
def do_an_internet(certdir, port):
"""Start up the proxy/server"""
- global global_certdir
- global_certdir = certdir
-
- httpd = ThreadingHTTPServer(('', port), RequestHijacker)
+ proxy_core.certdir = certdir
+ httpd = proxy_core.ThreadingHTTPServer(('', port), RequestHijacker)
httpd.serve_forever()
+
+if __name__ == '__main__':
+ import sys
+ def fail(msg, error_code):
+ print('Error:', msg)
+ print('Usage:', sys.argv[0], '[certificates directory] (port)')
+ sys.exit(error_code)
+
+ if len(sys.argv) < 2:
+ fail('missing required argument "certificates directory".', 1)
+
+ certdir = sys.argv[1]
+ if not proxy_core.os.path.isdir(certdir):
+ fail('selected certificate directory does not exist.', 2)
+
+ port = sys.argv[2] if len(sys.argv) > 2 else '1337'
+ if not port.isnumeric():
+ fail('port must be an integer.', 3)
+
+ do_an_internet(certdir, int(port))
diff --git a/test/world_wide_library.py b/test/world_wide_library.py
new file mode 100644
index 0000000..fc9e095
--- /dev/null
+++ b/test/world_wide_library.py
@@ -0,0 +1,15 @@
+# Copyright (C) 2021 jahoti <jahoti@tilde.team>
+# Licensing information is collated in the `copyright` file
+
+"""
+Our helpful little stand-in for the Internet
+"""
+
+catalog = {
+ 'http://gotmyowndoma.in': (302, {'location': 'http://gotmyowndoma.in/index.html'}, b''),
+ 'http://gotmyowndoma.in/': (302, {'location': 'http://gotmyowndoma.in/index.html'}, b''),
+ 'http://gotmyowndoma.in/index.html': (200, {}, 'data/pages/gotmyowndomain.html'),
+ 'https://gotmyowndoma.in': (302, {'location': 'https://gotmyowndoma.in/index.html'}, b''),
+ 'https://gotmyowndoma.in/': (302, {'location': 'https://gotmyowndoma.in/index.html'}, b''),
+ 'https://gotmyowndoma.in/index.html': (200, {}, 'data/pages/gotmyowndomain_https.html')
+}