Added Tahoe-LAFS gateway script

12 years ago · 444f294bea
parent 6264bc4e55
commit 444f294bea
1 changed files with 118 additions and 0 deletions
--- a/gateway.py
+++ b/gateway.py
@ -0,0 +1,118 @@
+from twisted.internet import protocol, reactor
+from twisted.web import static, server, proxy
+from twisted.web.resource import Resource
+import base64, urllib
+from os import fork, setsid, umask, dup2
+from sys import stdin, stdout, stderr
+from os import getpid
+from urllib import quote as urlquote
+
+pid_file = "/home/tahoe/gateway/gateway.pid"
+
+modHeaders = {
+	'Cache-Control': 'max-age=31536000',
+	'Server': 'Cryto-Tahoe-Gateway 1.1',
+	'Expires': 'Fri, 12 Dec 2012 05:00:00 GMT'
+}
+
+
+outfile = open(pid_file, 'w')
+outfile.write('%i' % getpid())
+outfile.close()
+if fork(): exit(0)
+umask(0) 
+setsid() 
+if fork(): exit(0)
+
+stdout.flush()
+stderr.flush()
+si = file('/dev/null', 'r')
+so = file('/dev/null', 'a+')
+se = file('/dev/null', 'a+', 0)
+dup2(si.fileno(), stdin.fileno())
+dup2(so.fileno(), stdout.fileno())
+dup2(se.fileno(), stderr.fileno())
+
+error_index = ("""\
+				<!DOCTYPE html>
+				<html>
+					<head>
+						<title>Cryto-Tahoe-Gateway: Index</title>
+					</head>
+					<body>
+						<h2>This gateway does not provide an index page.</h2>
+						Please use a direct URL to download a file hosted on this storage grid.<br><br>
+						Alternatively, <a href="http://cryto.net/">learn more about the Cryto Coding Collective</a>.
+						<br><br><br><hr>
+						<div style="text-align: right; font-style: italic;">Cryto-Tahoe-Gateway v1.1</div>
+					</body>
+				</html>""")
+
+error_404 = ("""\
+				<!DOCTYPE html>
+				<html>
+					<head>
+						<title>Cryto-Tahoe-Gateway: 404</title>
+					</head>
+					<body>
+						<h2>The specified resource was not found.</h2>
+						The file may have expired, or the hyperlink you followed may have been broken.<br><br>
+						<a href="http://cryto.net/">Learn more about the Cryto Coding Collective</a>.
+						<br><br><br><hr>
+						<div style="text-align: right; font-style: italic;">Cryto-Tahoe-Gateway v1.1</div>
+					</body>
+				</html>""")
+
+
+class ProxyClient(proxy.ProxyClient):
+	"""A proxy class that injects headers to the response."""
+	def handleEndHeaders(self):
+		for key, value in modHeaders.iteritems():
+			self.father.responseHeaders.setRawHeaders(key, [value])
+
+
+class ProxyClientFactory(proxy.ProxyClientFactory):
+	protocol = ProxyClient
+
+
+class ReverseProxyResource(proxy.ReverseProxyResource):
+	proxyClientFactoryClass = ProxyClientFactory
+
+	def getChild(self, path, request):
+		return ReverseProxyResource(
+			self.host, self.port, self.path + '/' + urlquote(path, safe=""))
+
+
+
+class GatewayResource(Resource):
+	isLeaf = False
+	allowedMethods = ("GET")
+
+	def getChild(self, name, request):
+		if name == "download":
+			try:
+				uri = request.path
+				uriParts = uri.split('/')
+				uriIdentifier = base64.urlsafe_b64decode(uriParts[2])
+				fileName = uriParts[3]
+				localUri = "/file/" + urllib.quote(uriIdentifier) + "/@@named=/" + urllib.quote(fileName)
+				return ReverseProxyResource('localhost', 3456, localUri)
+			except:
+				return self
+		else:
+			return self
+
+	def render_GET(self, request):
+		path = request.path
+		if path == "/":
+			return error_index
+		else:
+			return error_404
+
+resource = GatewayResource()
+
+site = server.Site(resource)
+reactor.listenTCP(3719, site)
+reactor.run()
+
+