Port hatena.py to Python 3 and modern Twisted
* Replace Python 2 syntax and string handling with Python 3-compatible code * Add explicit bytes/text conversion for Twisted requests and responses * Update request paths, headers, arguments, and child resource names for modern Twisted * Return byte responses from resource render methods * Replace deprecated `imp.load_source()` with `importlib.util` * Generate unique module names for dynamically loaded Hatena endpoints * Update HTTP response/header handling * Preserve Nintendo DSi connection-test handling * Preserve Flipnote Hatena proxy access restrictions * Preserve region routing for v2-xx, v2-eu, v2-us, and v2-jp * Preserve UGOXML loading and packing behavior * Update static file and folder resource handling * Improve 404 query-string reconstruction * Add compatibility fallback for modern Twisted client address APIs * Keep the original dynamic `hatenadir` resource-tree behavior
This commit is contained in:
Rekushi authored and GitHub committed
1 parent
76a9ac99e2
commit
fe44430a46
1 file changed
+271
-209
@@ -1,222 +1,284 @@
|
|||||||
from twisted.web import static, resource
|
from __future__ import annotations
|
||||||
import sys, os, imp
|
|
||||||
|
|
||||||
from Hatenatools import *#UGO, PPM, TMB and NTFT
|
import hashlib
|
||||||
#from DB import Database#flipnote handler
|
import importlib.util
|
||||||
#from DB import Database#flipnote handler
|
import os
|
||||||
ServerLog = None#Is set to the log class by server.py
|
import sys
|
||||||
|
from urllib.parse import urlencode
|
||||||
|
|
||||||
|
from twisted.web import resource, static
|
||||||
|
|
||||||
|
from Hatenatools import NTFT, PPM, TMB, UGO
|
||||||
|
|
||||||
|
ServerLog = None # Set by server.py
|
||||||
Silent = False
|
Silent = False
|
||||||
|
|
||||||
|
|
||||||
|
def as_text(value):
|
||||||
|
if isinstance(value, bytes):
|
||||||
|
return value.decode("utf-8", "replace")
|
||||||
|
return str(value)
|
||||||
|
|
||||||
|
|
||||||
|
def as_bytes(value):
|
||||||
|
if isinstance(value, bytes):
|
||||||
|
return value
|
||||||
|
return str(value).encode("utf-8")
|
||||||
|
|
||||||
|
|
||||||
|
def request_path(request):
|
||||||
|
return as_text(request.path)
|
||||||
|
|
||||||
|
|
||||||
|
def request_headers(request):
|
||||||
|
return {as_text(k).lower(): as_text(v) for k, v in request.getAllHeaders().items()}
|
||||||
|
|
||||||
|
|
||||||
|
def request_args(request):
|
||||||
|
out = {}
|
||||||
|
for key, values in request.args.items():
|
||||||
|
out[as_text(key)] = [as_text(v) for v in values]
|
||||||
|
return out
|
||||||
|
|
||||||
|
|
||||||
|
def client_ip(request):
|
||||||
|
# getClientIP exists on the Twisted versions this project historically used;
|
||||||
|
# getClientAddress is the modern fallback.
|
||||||
|
getter = getattr(request, "getClientIP", None)
|
||||||
|
if getter is not None:
|
||||||
|
value = getter()
|
||||||
|
if value:
|
||||||
|
return as_text(value)
|
||||||
|
address = request.getClientAddress()
|
||||||
|
return as_text(getattr(address, "host", address))
|
||||||
|
|
||||||
|
|
||||||
def Log(request, path=None, silent=Silent):
|
def Log(request, path=None, silent=Silent):
|
||||||
if not path:
|
if path is None:
|
||||||
path = "\"%s\"" % request.path
|
path = '"%s"' % request_path(request)
|
||||||
ServerLog.write("%s requested %s" % (request.getClientIP(), path), Silent)
|
if ServerLog is not None:
|
||||||
|
ServerLog.write("%s requested %s" % (client_ip(request), path), silent)
|
||||||
|
|
||||||
#Responses:
|
|
||||||
class AccessDenied(resource.Resource):
|
|
||||||
isLeaf = True
|
|
||||||
def render(self, request):
|
|
||||||
ServerLog.write("%s got 403 when requesting \"%s\"" % (request.getClientIP(), request.path), Silent)
|
|
||||||
print "debug:",request.getAllHeaders()
|
|
||||||
|
|
||||||
request.setResponseCode(403)
|
|
||||||
return "403 - Access denied\nThis proxy is only allowed to use for Flipnote Hatena for the DSi."
|
|
||||||
AccessDenied = AccessDenied()
|
|
||||||
class NotFound(resource.Resource):
|
|
||||||
isLeaf = True
|
|
||||||
def render(self, request):
|
|
||||||
args = "&".join(("%s=%s" % (i, request.args[i][0]) for i in request.args))
|
|
||||||
path = ("%s?%s" % (request.path, args)) if args else request.path
|
|
||||||
|
|
||||||
ServerLog.write("%s got 404 when requesting \"%s\"" % (request.getClientIP(), path), Silent)
|
|
||||||
request.setResponseCode(404)
|
|
||||||
return "404 - Not Found\nThis proxy is only allowed to use for Flipnote Hatena for the DSi."
|
|
||||||
NotFound = NotFound()
|
|
||||||
class ConnectionTest(resource.Resource):#used for people setting up the proxy in their DSi. It's nice to see it's actually working.
|
|
||||||
#http://conntest.nintendowifi.net/
|
|
||||||
isLeaf = True
|
|
||||||
def render(self, request):
|
|
||||||
ServerLog.write("%s performed a connection test" % request.getClientIP(), True)
|
|
||||||
request.responseHeaders.setRawHeaders('X-Organization', ['Nintendo'])#i hope this isn't illegal...
|
|
||||||
return '<html><head><title>HTML Page</title></head><body bgcolor="#FFFFFF">This is test.html page</body></html>'
|
|
||||||
ConnectionTest = ConnectionTest()
|
|
||||||
|
|
||||||
#Root structure:
|
class AccessDeniedResource(resource.Resource):
|
||||||
class Root(resource.Resource):#filters out non-hantena clients
|
isLeaf = True
|
||||||
isLeaf = False
|
|
||||||
def __init__(self):
|
def render(self, request):
|
||||||
resource.Resource.__init__(self)
|
if ServerLog is not None:
|
||||||
|
ServerLog.write(
|
||||||
self.dsResource = ds()
|
'%s got 403 when requesting "%s"' % (client_ip(request), request_path(request)),
|
||||||
self.cssResource = static.File("hatenadir/css/")
|
Silent,
|
||||||
self.imagesResource = static.File("hatenadir/images/")
|
)
|
||||||
def getChild(self, name, request):
|
request.setResponseCode(403)
|
||||||
if "x-dsi-sid" not in request.getAllHeaders():
|
request.setHeader(b"content-type", b"text/plain; charset=utf-8")
|
||||||
if "host" in request.getAllHeaders():
|
return b"403 - Access denied\nThis proxy is only allowed to use for Flipnote Hatena for the DSi."
|
||||||
if request.getAllHeaders()["host"] == "conntest.nintendowifi.net":
|
|
||||||
return ConnectionTest
|
|
||||||
return AccessDenied
|
AccessDenied = AccessDeniedResource()
|
||||||
|
|
||||||
if name == "ds":
|
|
||||||
return self.dsResource
|
class NotFoundResource(resource.Resource):
|
||||||
elif name == "css":
|
isLeaf = True
|
||||||
return self.cssResource
|
|
||||||
elif name == "images":
|
def render(self, request):
|
||||||
return self.imagesResource
|
args = request_args(request)
|
||||||
elif name == "":
|
query = urlencode([(k, v) for k, values in args.items() for v in values])
|
||||||
return self
|
path = request_path(request)
|
||||||
|
if query:
|
||||||
#return NotFound
|
path += "?" + query
|
||||||
return self
|
if ServerLog is not None:
|
||||||
def render(self, request):
|
ServerLog.write('%s got 404 when requesting "%s"' % (client_ip(request), path), Silent)
|
||||||
if request.getHost() == "conntest.nintendowifi.net":
|
request.setResponseCode(404)
|
||||||
return ConnectionTest
|
request.setHeader(b"content-type", b"text/plain; charset=utf-8")
|
||||||
Log(request, "root")
|
return b"404 - Not Found\nThis proxy is only allowed to use for Flipnote Hatena for the DSi."
|
||||||
return "Welcome to hatena.pbsds.net!\nThis is still in early stages, so please don't expect too much."
|
|
||||||
class ds(resource.Resource):#child of Root
|
|
||||||
isLeaf = False
|
NotFound = NotFoundResource()
|
||||||
def __init__(self):
|
|
||||||
resource.Resource.__init__(self)
|
|
||||||
|
class ConnectionTestResource(resource.Resource):
|
||||||
self.region = UgoRoot()
|
isLeaf = True
|
||||||
self.regions = ("v2-xx", "v2-eu", "v2-us", "v2-jp")
|
|
||||||
def getChild(self, name, request):
|
def render(self, request):
|
||||||
if name in self.regions:
|
if ServerLog is not None:
|
||||||
return self.region
|
ServerLog.write("%s performed a connection test" % client_ip(request), True)
|
||||||
elif name == "":
|
request.setHeader(b"X-Organization", b"Nintendo")
|
||||||
return self
|
request.setHeader(b"content-type", b"text/html; charset=utf-8")
|
||||||
return NotFound
|
return b'<html><head><title>HTML Page</title></head><body bgcolor="#FFFFFF">This is test.html page</body></html>'
|
||||||
def render(self, request):
|
|
||||||
Log(request)
|
|
||||||
return "ds desu"
|
ConnectionTest = ConnectionTestResource()
|
||||||
class UgoRoot(resource.Resource):#child of ds. (v2-xx)
|
|
||||||
isLeaf = False
|
|
||||||
def __init__(self):
|
class Root(resource.Resource):
|
||||||
resource.Resource.__init__(self)
|
isLeaf = False
|
||||||
|
|
||||||
LoadHatenadirStructure(self)
|
def __init__(self):
|
||||||
def getChild(self, name, request):
|
super().__init__()
|
||||||
if name == "":
|
self.dsResource = ds()
|
||||||
return self
|
self.cssResource = static.File("hatenadir/css/")
|
||||||
return NotFound
|
self.imagesResource = static.File("hatenadir/images/")
|
||||||
def render(self, request):
|
|
||||||
Log(request, "ugoroot")
|
def getChild(self, name, request):
|
||||||
return "UgoRoot desu"
|
name = as_text(name)
|
||||||
|
headers = request_headers(request)
|
||||||
|
if "x-dsi-sid" not in headers:
|
||||||
|
if headers.get("host", "").split(":", 1)[0].lower() == "conntest.nintendowifi.net":
|
||||||
|
return ConnectionTest
|
||||||
|
return AccessDenied
|
||||||
|
|
||||||
|
if name == "ds":
|
||||||
|
return self.dsResource
|
||||||
|
if name == "css":
|
||||||
|
return self.cssResource
|
||||||
|
if name == "images":
|
||||||
|
return self.imagesResource
|
||||||
|
if name == "":
|
||||||
|
return self
|
||||||
|
# The original server intentionally tolerated extra leading path pieces.
|
||||||
|
return self
|
||||||
|
|
||||||
|
def render(self, request):
|
||||||
|
headers = request_headers(request)
|
||||||
|
if headers.get("host", "").split(":", 1)[0].lower() == "conntest.nintendowifi.net":
|
||||||
|
return ConnectionTest.render(request)
|
||||||
|
Log(request, "root")
|
||||||
|
request.setHeader(b"content-type", b"text/plain; charset=utf-8")
|
||||||
|
return b"Welcome to hatena.pbsds.net!\nThis is still in early stages, so please don't expect too much."
|
||||||
|
|
||||||
|
|
||||||
|
class ds(resource.Resource):
|
||||||
|
isLeaf = False
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
super().__init__()
|
||||||
|
self.region = UgoRoot()
|
||||||
|
self.regions = {"v2-xx", "v2-eu", "v2-us", "v2-jp"}
|
||||||
|
|
||||||
|
def getChild(self, name, request):
|
||||||
|
name = as_text(name)
|
||||||
|
if name in self.regions:
|
||||||
|
return self.region
|
||||||
|
if name == "":
|
||||||
|
return self
|
||||||
|
return NotFound
|
||||||
|
|
||||||
|
def render(self, request):
|
||||||
|
Log(request)
|
||||||
|
return b"ds desu"
|
||||||
|
|
||||||
|
|
||||||
|
class UgoRoot(resource.Resource):
|
||||||
|
isLeaf = False
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
super().__init__()
|
||||||
|
LoadHatenadirStructure(self)
|
||||||
|
|
||||||
|
def getChild(self, name, request):
|
||||||
|
if as_text(name) == "":
|
||||||
|
return self
|
||||||
|
return NotFound
|
||||||
|
|
||||||
|
def render(self, request):
|
||||||
|
Log(request, "ugoroot")
|
||||||
|
return b"UgoRoot desu"
|
||||||
|
|
||||||
|
|
||||||
#UgoRoot filestructure:
|
|
||||||
class FileResource(resource.Resource):
|
class FileResource(resource.Resource):
|
||||||
isLeaf = True
|
isLeaf = True
|
||||||
def __init__(self, filepath, Store=False):
|
|
||||||
resource.Resource.__init__(self)
|
def __init__(self, filepath, Store=False):
|
||||||
|
super().__init__()
|
||||||
self.Store = Store
|
self.Store = Store
|
||||||
|
self.file = open(filepath, "rb").read() if Store else filepath
|
||||||
if Store:
|
self.html = filepath.split(".")[-1][:3].lower() == "htm"
|
||||||
f = open(filepath, "rb")
|
|
||||||
self.file = f.read()
|
def render(self, request):
|
||||||
f.close()
|
path = "/".join(request_path(request).split("/")[3:])
|
||||||
else:
|
Log(request, path)
|
||||||
self.file = filepath
|
if self.html:
|
||||||
|
request.setHeader(b"content-type", b"text/html; charset=utf-8")
|
||||||
self.html = filepath.split(".")[-1][:3] == "htm"
|
else:
|
||||||
def render(self, request):
|
request.setHeader(b"content-type", b"text/plain")
|
||||||
path = "/".join(request.path.split("/")[3:])
|
if self.Store:
|
||||||
Log(request, path)
|
return self.file
|
||||||
|
return static.File(self.file).render(request)
|
||||||
if self.html:
|
|
||||||
#request.responseHeaders.setRawHeaders('content-type', ['text/html'])
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
request.responseHeaders.setRawHeaders('content-type', ['text/plain'])
|
|
||||||
|
|
||||||
if self.Store:
|
|
||||||
return self.file
|
|
||||||
else:
|
|
||||||
#f = open(self.file, "rb")
|
|
||||||
#ret = f.read()
|
|
||||||
#f.close()
|
|
||||||
#return ret
|
|
||||||
|
|
||||||
return static.File(self.file).render(request)
|
|
||||||
class UGOXMLResource(resource.Resource):
|
class UGOXMLResource(resource.Resource):
|
||||||
isLeaf = True
|
isLeaf = True
|
||||||
def __init__(self, filepath):
|
|
||||||
resource.Resource.__init__(self)
|
def __init__(self, filepath):
|
||||||
|
super().__init__()
|
||||||
|
ugo = UGO().ReadXML(filepath, False)
|
||||||
self.ugofile = UGO().ReadXML(filepath, False)
|
if not ugo:
|
||||||
self.ugofile = self.ugofile.Pack()
|
raise ValueError("Could not parse UGOXML file: %s" % filepath)
|
||||||
def render(self, request):
|
self.ugofile = ugo.Pack()
|
||||||
path = "/".join(request.path.split("/")[3:])
|
|
||||||
Log(request, path)
|
def render(self, request):
|
||||||
|
path = "/".join(request_path(request).split("/")[3:])
|
||||||
#page = int(request.args["page"][0]) if "page" in request.args else 1
|
Log(request, path)
|
||||||
request.responseHeaders.setRawHeaders('content-type', ['text/plain'])
|
request.setHeader(b"content-type", b"text/plain")
|
||||||
return self.ugofile
|
return self.ugofile
|
||||||
|
|
||||||
|
|
||||||
class FolderResource(resource.Resource):
|
class FolderResource(resource.Resource):
|
||||||
isLeaf = False
|
isLeaf = False
|
||||||
def getChild(self, name, request):
|
|
||||||
if name == "":
|
def getChild(self, name, request):
|
||||||
return self
|
if as_text(name) == "":
|
||||||
return NotFound
|
return self
|
||||||
def render(self, request):
|
return NotFound
|
||||||
path = "/".join(request.path.split("/")[3:])
|
|
||||||
Log(request, path)
|
def render(self, request):
|
||||||
return "I am a folder, but I'm to lazy to list my contents..."
|
path = "/".join(request_path(request).split("/")[3:])
|
||||||
|
Log(request, path)
|
||||||
|
return b"I am a folder, but I'm too lazy to list my contents..."
|
||||||
|
|
||||||
|
|
||||||
|
def _load_python_resource(filepath):
|
||||||
|
module_name = "hatena_dynamic_" + hashlib.sha1(os.path.abspath(filepath).encode("utf-8")).hexdigest()
|
||||||
|
spec = importlib.util.spec_from_file_location(module_name, filepath)
|
||||||
|
if spec is None or spec.loader is None:
|
||||||
|
raise ImportError("Could not create import spec for %s" % filepath)
|
||||||
|
module = importlib.util.module_from_spec(spec)
|
||||||
|
sys.modules[module_name] = module
|
||||||
|
spec.loader.exec_module(module)
|
||||||
|
return module
|
||||||
|
|
||||||
|
|
||||||
def LoadHatenadirStructure(Resource, path=os.path.join("hatenadir", "ds", "v2-xx")):
|
def LoadHatenadirStructure(Resource, path=os.path.join("hatenadir", "ds", "v2-xx")):
|
||||||
for root, dirs, files in os.walk(path):
|
for root, dirs, files in os.walk(path):
|
||||||
if root <> path: continue#use recursion instead
|
if root != path:
|
||||||
for filename in files:
|
continue
|
||||||
filetype = filename.split(".")[-1].lower()
|
|
||||||
os.path.join(path, filename)
|
for filename in files:
|
||||||
|
filetype = filename.split(".")[-1].lower()
|
||||||
if filetype == "ugoxml":
|
filepath = os.path.join(path, filename)
|
||||||
Resource.putChild(filename[:-3], UGOXMLResource(os.path.join(path, filename)))
|
if filetype == "ugoxml":
|
||||||
elif filetype == "py":
|
Resource.putChild(filename[:-3].encode("utf-8"), UGOXMLResource(filepath))
|
||||||
try:
|
elif filetype == "py":
|
||||||
pyfile = imp.load_source("pyfile", os.path.abspath(os.path.join(path, filename)))
|
try:
|
||||||
except ImportError as err:
|
pyfile = _load_python_resource(os.path.abspath(filepath))
|
||||||
pyfile = None
|
except Exception as err:
|
||||||
print "Error!"
|
print("Error!")
|
||||||
print "Failed to import the python file \"%s\"" % os.path.join(path, filename)
|
print('Failed to import the python file "%s"' % filepath)
|
||||||
print err
|
print(err)
|
||||||
|
raise
|
||||||
if pyfile:
|
Resource.putChild(filename[:-3].encode("utf-8"), pyfile.PyResource())
|
||||||
Resource.putChild(filename[:-3], pyfile.PyResource())
|
elif filetype == "pyc":
|
||||||
elif filetype == "pyc":
|
continue
|
||||||
pass#ignore
|
else:
|
||||||
else:
|
Resource.putChild(filename.encode("utf-8"), FileResource(filepath))
|
||||||
Resource.putChild(filename, FileResource(os.path.join(path, filename)))
|
|
||||||
for foldername in dirs:# os.path.isdir(i):
|
for foldername in dirs:
|
||||||
if foldername[:2] <> "__":
|
if not foldername.startswith("__"):
|
||||||
folder = FolderResource()
|
folder = FolderResource()
|
||||||
LoadHatenadirStructure(folder, os.path.join(path, foldername))
|
LoadHatenadirStructure(folder, os.path.join(path, foldername))
|
||||||
Resource.putChild(foldername, folder)
|
Resource.putChild(foldername.encode("utf-8"), folder)
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
#when loading:
|
|
||||||
def Setup():
|
def Setup():
|
||||||
root = Root()
|
return Root()
|
||||||
return root
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Reference in new issue
Block a user