User:Darien Caldwell/Advanced HTTP-DNS Code
< User:Darien Caldwell
Jump to navigation
Jump to search
Revision as of 20:14, 2 September 2012 by Darien Caldwell (talk | contribs)
dns.py File
<python>
import cgi, logging, urllib
from google.appengine.ext import webapp from google.appengine.ext.webapp.util import run_wsgi_app from google.appengine.ext import db
class Service(db.Model):
name = db.StringProperty(multiline=False) url = db.StringProperty(multiline=False,indexed=False) writepass = db.StringProperty(multiline=False,indexed=False) readpass = db.StringProperty(multiline = False,indexed=False) hidden = db.BooleanProperty(indexed=False)
class Main_DNS(webapp.RequestHandler):
def get(self):
obj_name=self.request.headers.get('X-SecondLife-Object-Name')
region=self.request.headers.get('X-SecondLife-Region')
local_pos=self.request.headers.get('X-SecondLife-Local-Position')
owner_name=self.request.headers.get('X-SecondLife-Owner-Name')
owner_key=self.request.headers.get('X-SecondLife-Owner-Key')
logging.info('DNS called from '+obj_name+' owned by '+owner_name+' ('+owner_key+') from Region:'+region+' Pos:'+local_pos)
logging.info('POST Body was: '+self.request.body)
admin_password = "adminpassword" # This password enables you to have unrestricted access to the DNS
# Putting "null" in password disables this feature.
if self.request.get('type')=='add': # Adding a new service to the DNS (You can also use Update but it won't tell you the service already exists)
param2=self.request.get('name') # the Name the service will be known by
param3=self.request.get('url') # the URL for the web service
param4=self.request.get('wpass') # the password for modifying the entry
param5=self.request.get('rpass') # the password for reading the entry
param6=self.request.get('hidden') # enables hiding the entry from listing
if admin_password != 'null' and admin_password == self.request.get('admin'):
newrec = Service.get_by_key_name(param2)
if newrec is None: # the service doesn't exist, so add it.
if param2=="" or param3=="" :
self.response.out.write('Error2')
else:
if param6 == "1":
param6 = True
else:
param6 = False
newrec=Service(key_name=param2,name=param2,url=param3,writepass=param4,readpass=param5,hidden=param6)
newrec.put()
logging.info('Added Service: '+param2)
self.response.out.write('Added')
else:
logging.info('Service: '+param2+' already found.')
self.response.out.write('Found') # service already exists so announce that and do nothing
else:
self.response.set_status(401)
logging.info('Rejected Service: '+param2+' Invalid admin password')
self.response.out.write('Rejected')
elif self.request.get('type')=='remove': #removing a service
param2=self.request.get('name') # the name the service is known by
record = Service.get_by_key_name(param2)
if record is None:
self.response.set_status(200)
self.response.out.write('None') # Service wasn't found
elif record.writepass == "" or record.writepass == self.request.get('pass') or (admin_password != 'null' and admin_password == self.request.get('admin')):
record.delete() # remove
logging.info('Removed Service: '+param2)
self.response.out.write('Removed')
else:
self.response.set_status(401)
logging.info('Rejected Service: '+param2+' remove. Not Found')
self.response.out.write('Rejected')
elif self.request.get('type')=='update': # update an existing service. Note this creates a new service, or updates an existing one
param2=self.request.get('name') #the Name the service will be known by
param3=self.request.get('url') # the URL for the web service
param4=self.request.get('pass') # the password for modifying the entry
if param2=="" :
logging.info('Could not Update Blank Service')
self.response.out.write('Error2')
return
record = Service.get_by_key_name(param2)
if record is None:
if param3 == "":
logging.info('Updating Service: '+param2+' failed. Blank URL')
self.response.out.write('Error3')
else:
newhidden = False
if self.request.get('hidden') == '1':
newhidden = True
newrec=Service(key_name=param2,name=param2,url=param3,writepass=param4,readpass=param4, hidden=newhidden)
if self.request.get('wpass') != "":
newrec.writepass=self.request.get('wpass')
if self.request.get('rpass') != "":
newrec.readpass=self.request.get('rpass')
newrec.put()
logging.info('Added Service: '+param2+' (update add)')
self.response.out.write('Added')
elif record.writepass == "" or record.writepass == param4 or (admin_password != 'null' and admin_password == self.request.get('admin')):
# record already exists, update it
newwpass = record.writepass
newrpass = record.readpass
newhidden = record.hidden
if param3 == "":
param3 = record.url
if self.request.get('wpass') != "":
if self.request.get('wpass') == "null":
newwpass = ""
else:
newwpass = self.request.get('wpass')
if self.request.get('rpass') != "":
if self.request.get('rpass') == "null":
newrpass = ""
else:
newrpass = self.request.get('rpass')
if self.request.get('hidden') != "":
if self.request.get('hidden') == "1":
newhidden = True
elif self.request.get('hidden') == "0":
newhidden = False
# add record, either replacing the deleted one, or adding a new one if it never existed
record.url=param3
record.writepass=newwpass
record.readpass=newrpass
record.hidden=newhidden
record.put()
logging.info('Updated Service: '+param2)
self.response.out.write('Updated')
else:
self.response.set_status(401)
logging.info('Update Service: '+param2+' failed. Invalid Password.')
self.response.out.write('Rejected')
elif self.request.get('type')=='retrieve': # get the current URL for a given service
param2=self.request.get('name') # the name the service is known by
record = Service.get_by_key_name(param2)
if record is None:
self.response.set_status(200)
logging.info('Retrieve Service: '+param2+' failed. Not Found.')
self.response.out.write('None') # Service wasn't found
elif record.readpass == "" or record.readpass == self.request.get('pass') or (admin_password != 'null' and admin_password == self.request.get('admin')):
logging.info('Retrieved Service: '+param2)
self.response.out.write(record.url) #print the URL
else:
self.response.set_status(401)
logging.info('Retrieve Service: '+param2+' failed. Invalid Password.')
self.response.out.write("Rejected")
elif self.request.get('type')=='list': # List the existing services
records = Service.all()
if records is None:
logging.info('Service List: Empty')
self.response.out.write('Empty') # Services weren't found
else:
got_admin = False
if admin_password != 'null' and admin_password == self.request.get('admin'):
got_admin = True;
for result in records:
if result.hidden == False or got_admin == True:
self.response.out.write(result.name+',')
logging.info('Service List: Success')
self.response.out.write('END') # Cap the list
else:
logging.info('Invalid Service command type: '+self.request.get('type'))
self.response.out.write('Error')
def post(self):
self.get()
class Redirector(webapp.RequestHandler):
def get(self):
import urllib
service_name=self.request.path
if service_name[-1]=='/' :
service_name=service_name[5:-1] #remove leading and trailing slash
else:
service_name=service_name[5:] # remove leading slash only
#un-escape just in case you're Kinki :p
service_name=urllib.unquote(service_name)
logging.debug('Looking up '+service_name+' (DNS)');
record = Service.get_by_key_name(str(service_name))
if record is None:
logging.info('Redirect Failed: Not Found')
self.response.out.write('None') # Service wasn't found
elif record.readpass != :
self.response.set_status(403)
logging.info('Redirect Failed: Read Protected. Use "retrieve" command type to get URL.')
self.response.out.write("<html><title>403 Forbidden</title><body>Cannot forward a read-protected entry.</body></html>")
else:
logging.info('Redirect Success')
if self.request.query_string != :
self.redirect(urllib.unquote(record.url)+'/?'+self.request.query_string) # redirect to the HTTP-IN URL with arugments
else:
self.redirect(urllib.unquote(record.url)) # redirect to the HTTP-IN URL
application = webapp.WSGIApplication(
[
('/', Main_DNS),
('/.*',Redirector)],
debug=True)
def main():
run_wsgi_app(application)
if __name__ == "__main__":
main()</python>
app.yaml File
<text> application: YOUR-APP-NAME-HERE version: 1 runtime: python api_version: 1
handlers: - url: /.*
script: dns.py</text>