diff options
| author | James Taylor <user234683@users.noreply.github.com> | 2018-06-30 23:34:46 -0700 |
|---|---|---|
| committer | James Taylor <user234683@tutanota.com> | 2018-06-30 23:34:46 -0700 |
| commit | 98157bf1bf1223ffa7556d2d21cfac6f07675f9d (patch) | |
| tree | 9863e438deec87bb309a435c0f4cb0eb0e455dcc /server.py | |
| download | yt-local-98157bf1bf1223ffa7556d2d21cfac6f07675f9d.tar.lz yt-local-98157bf1bf1223ffa7556d2d21cfac6f07675f9d.tar.xz yt-local-98157bf1bf1223ffa7556d2d21cfac6f07675f9d.zip | |
initial commit
Diffstat (limited to 'server.py')
| -rw-r--r-- | server.py | 141 |
1 files changed, 141 insertions, 0 deletions
diff --git a/server.py b/server.py new file mode 100644 index 0000000..cc88d05 --- /dev/null +++ b/server.py @@ -0,0 +1,141 @@ +from gevent import monkey
+monkey.patch_all()
+import gevent.socket
+
+from gevent.pywsgi import WSGIServer
+from youtube.youtube import youtube
+import urllib
+import socket
+import socks
+import subprocess
+import re
+
+ROUTE_TOR = True
+TOR_PATH = ***REMOVED***
+PORT_NUMBER=80
+ALLOW_FOREIGN_ADDRESSES=True
+
+BAN_FILE = "banned_addresses.txt"
+with open(BAN_FILE, 'r') as f:
+ banned_addresses = f.read().splitlines()
+
+def ban_address(address):
+ banned_addresses.append(address)
+ with open(BAN_FILE, 'a') as f:
+ f.write(address + "\n")
+
+
+def youtu_be(env, start_response):
+ id = env['PATH_INFO'][1:]
+ env['PATH_INFO'] = '/watch'
+ env['QUERY_STRING'] = 'v=' + id
+ return youtube(env, start_response)
+
+def proxy_site(env, start_response):
+ headers = {
+ 'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64)',
+ 'Accept': '*/*',
+ }
+ url = "https://" + env['SERVER_NAME'] + env['PATH_INFO']
+ if env['QUERY_STRING']:
+ url += '?' + env['QUERY_STRING']
+ req = urllib.request.Request(url, headers=headers)
+ response = urllib.request.urlopen(req, timeout = 10)
+ start_response('200 OK', () )
+ return response.read()
+
+site_handlers = {
+ 'youtube.com':youtube,
+ 'youtu.be':youtu_be,
+ 'ytimg.com': proxy_site,
+ 'yt3.ggpht.com': proxy_site,
+ 'lh3.googleusercontent.com': proxy_site,
+
+}
+
+def split_url(url):
+ ''' Split https://sub.example.com/foo/bar.html into ('sub.example.com', '/foo/bar.html')'''
+ # XXX: Is this regex safe from REDOS?
+ # python STILL doesn't have a proper regular expression engine like grep uses built in...
+ match = re.match(r'(?:https?://)?([\w-]+(?:\.[\w-]+)+?)(/.*|$)', url)
+ if match is None:
+ raise ValueError('Invalid or unsupported url: ' + url)
+
+ return match.group(1), match.group(2)
+
+
+
+def error_code(code, start_response):
+ start_response(code, ())
+ return code.encode()
+
+def site_dispatch(env, start_response):
+ client_address = env['REMOTE_ADDR']
+ try:
+ method = env['REQUEST_METHOD']
+ path = env['PATH_INFO']
+ if client_address in banned_addresses:
+ yield error_code('403 Fuck Off', start_response)
+ return
+ if method=="POST" and client_address not in ('127.0.0.1', '::1'):
+ yield error_code('403 Forbidden', start_response)
+ return
+ if "phpmyadmin" in path or (path == "/" and method == "HEAD"):
+ ban_address(client_address)
+ start_response('403 Fuck Off', ())
+ yield b'403 Fuck Off'
+ return
+
+ '''if env['QUERY_STRING']:
+ path += '?' + env['QUERY_STRING']'''
+ #path_parts = urllib.parse.urlparse(path)
+ try:
+ env['SERVER_NAME'], env['PATH_INFO'] = split_url(path[1:])
+ except ValueError:
+ yield error_code('404 Not Found', start_response)
+ return
+
+ base_name = ''
+ for domain in reversed(env['SERVER_NAME'].split('.')):
+ if base_name == '':
+ base_name = domain
+ else:
+ base_name = domain + '.' + base_name
+
+ try:
+ handler = site_handlers[base_name]
+ except KeyError:
+ continue
+ else:
+ yield handler(env, start_response)
+ break
+ else: # did not break
+ yield error_code('404 Not Found', start_response)
+ return
+
+
+ except (socket.error, ConnectionAbortedError) as e:
+ start_response('500 Internal Server Error', ())
+ print(str(e))
+ yield b'500 Internal Server Error'
+
+ except Exception:
+ start_response('500 Internal Server Error', ())
+ raise
+ return
+
+
+
+
+if ROUTE_TOR:
+ #subprocess.Popen(TOR_PATH)
+ socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, '127.0.0.1', 9150)
+ socket.socket = socks.socksocket
+ gevent.socket.socket = socks.socksocket
+
+if ALLOW_FOREIGN_ADDRESSES:
+ server = WSGIServer(('0.0.0.0', PORT_NUMBER), site_dispatch)
+else:
+ server = WSGIServer(('127.0.0.1', PORT_NUMBER), site_dispatch)
+print('Started httpserver on port ' , PORT_NUMBER)
+server.serve_forever()
|