cookbooks/tile/templates/default/export.erb

   1 #!/usr/bin/python -u
   2 # -*- coding: utf-8 -*-
   3
   4 import cairo
   5 import cgi
   6 import Cookie
   7 import mapnik
   8 import os
   9 import pyotp
  10 import resource
  11 import shutil
  12 import signal
  13 import sys
  14 import tempfile
  15
  16 # Limit maximum CPU time
  17 # The Postscript output format can sometimes take hours
  18 resource.setrlimit(resource.RLIMIT_CPU,(180,180))
  19
  20 # Limit memory usage
  21 # Some odd requests can cause extreme memory usage
  22 resource.setrlimit(resource.RLIMIT_AS,(4000000000, 4000000000))
  23
  24 # Routine to output HTTP headers
  25 def output_headers(content_type, filename = "", length = 0):
  26   print "Content-Type: %s" % content_type
  27   if filename:
  28     print "Content-Disposition: attachment; filename=\"%s\"" % filename
  29   if length:
  30     print "Content-Length: %d" % length
  31   print ""
  32
  33 # Routine to output the contents of a file
  34 def output_file(file):
  35   file.seek(0)
  36   shutil.copyfileobj(file, sys.stdout)
  37
  38 # Routine to get the size of a file
  39 def file_size(file):
  40   return os.fstat(file.fileno()).st_size
  41
  42 # Routine to report an error
  43 def output_error(message, status = "400 Bad Request"):
  44   print "Status: %s" % status
  45   output_headers("text/html")
  46   print "<html>"
  47   print "<head>"
  48   print "<title>Error</title>"
  49   print "</head>"
  50   print "<body>"
  51   print "<h1>Error</h1>"
  52   print "<p>%s</p>" % message
  53   print "</body>"
  54   print "</html>"
  55
  56 # Create TOTP token validator
  57 totp = pyotp.TOTP('<%= @totp_key %>', interval = 3600)
  58
  59 # Parse CGI parameters
  60 form = cgi.FieldStorage()
  61
  62 # Import cookies
  63 cookies = Cookie.SimpleCookie(os.environ.get('HTTP_COOKIE'))
  64
  65 # Make sure we have a user agent
  66 if not os.environ.has_key('HTTP_USER_AGENT'):
  67   os.environ['HTTP_USER_AGENT'] = 'NONE'
  68
  69 # Make sure we have a referer
  70 if not os.environ.has_key('HTTP_REFERER'):
  71   os.environ['HTTP_REFERER'] = 'NONE'
  72
  73 # Look for TOTP token
  74 if cookies.has_key('_osm_totp_token'):
  75   token = cookies['_osm_totp_token'].value
  76 else:
  77   token = None
  78
  79 # Get the load average
  80 cputimes = [float(n) for n in open("/proc/stat").readline().rstrip().split()[1:-1]]
  81 idletime = cputimes[3] / sum(cputimes)
  82
  83 # Process the request
  84 if not totp.verify(token, valid_window = 1):
  85   # Abort if the request didn't have a valid TOTP token
  86   output_error("Missing or invalid token")
  87 elif idletime < 0.2:
  88   # Abort if the CPU idle time on the machine is too low
  89   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
  90 <% @blocks["user_agents"].each do |user_agent| -%>
  91 elif os.environ['HTTP_USER_AGENT'] == '<%= user_agent %>':
  92   # Block scraper
  93   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
  94 <% end -%>
  95 <% @blocks["referers"].each do |referer| -%>
  96 elif os.environ['HTTP_REFERER'] == '<%= referer %>':
  97   # Block scraper
  98   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
  99 <% end -%>
 100 elif not form.has_key("bbox"):
 101   # No bounding box specified
 102   output_error("No bounding box specified")
 103 elif not form.has_key("scale"):
 104   # No scale specified
 105   output_error("No scale specified")
 106 elif not form.has_key("format"):
 107   # No format specified
 108   output_error("No format specified")
 109 else:
 110   # Create projection object
 111   prj = mapnik.Projection("+proj=merc +a=6378137 +b=6378137 +lat_ts=0.0 +lon_0=0.0 +x_0=0.0 +y_0=0 +k=1.0 +units=m +nadgrids=@null +no_defs +over");
 112
 113   # Get the bounds of the area to render
 114   bbox = [float(x) for x in form.getvalue("bbox").split(",")]
 115
 116   if bbox[0] >= bbox[2] or bbox[1] >= bbox[3]:
 117     # Bogus bounding box
 118     output_error("Invalid bounding box")
 119   else:
 120     # Project the bounds to the map projection
 121     bbox = mapnik.forward_(mapnik.Box2d(*bbox), prj)
 122
 123     # Get the style to use
 124     style = form.getvalue("style", "default")
 125
 126     # Calculate the size of the final rendered image
 127     scale = float(form.getvalue("scale"))
 128     width = int(bbox.width() / scale / 0.00028)
 129     height = int(bbox.height() / scale / 0.00028)
 130
 131     # Limit the size of map we are prepared to produce
 132     if width * height > 4000000:
 133       # Map is too large (limit is approximately A2 size)
 134       output_error("Map too large")
 135     else:
 136       # Create map
 137       map = mapnik.Map(width, height)
 138
 139       # Load map configuration
 140       mapnik.load_map(map, "/srv/tile.openstreetmap.org/styles/%s/project.xml" % style)
 141
 142       # Zoom the map to the bounding box
 143       map.zoom_to_box(bbox)
 144
 145       # Fork so that we can handle crashes rendering the map
 146       pid = os.fork()
 147
 148       # Render the map
 149       if pid == 0:
 150         if form.getvalue("format") == "png":
 151           image = mapnik.Image(map.width, map.height)
 152           mapnik.render(map, image)
 153           png = image.tostring("png")
 154           output_headers("image/png", "map.png", len(png))
 155           sys.stdout.write(png)
 156         elif form.getvalue("format") == "jpeg":
 157           image = mapnik.Image(map.width, map.height)
 158           mapnik.render(map, image)
 159           jpeg = image.tostring("jpeg")
 160           output_headers("image/jpeg", "map.jpg", len(jpeg))
 161           sys.stdout.write(jpeg)
 162         elif form.getvalue("format") == "svg":
 163           file = tempfile.NamedTemporaryFile(prefix = "export")
 164           surface = cairo.SVGSurface(file.name, map.width, map.height)
 165           mapnik.render(map, surface)
 166           surface.finish()
 167           output_headers("image/svg+xml", "map.svg", file_size(file))
 168           output_file(file)
 169         elif form.getvalue("format") == "pdf":
 170           file = tempfile.NamedTemporaryFile(prefix = "export")
 171           surface = cairo.PDFSurface(file.name, map.width, map.height)
 172           mapnik.render(map, surface)
 173           surface.finish()
 174           output_headers("application/pdf", "map.pdf", file_size(file))
 175           output_file(file)
 176         elif form.getvalue("format") == "ps":
 177           file = tempfile.NamedTemporaryFile(prefix = "export")
 178           surface = cairo.PSSurface(file.name, map.width, map.height)
 179           mapnik.render(map, surface)
 180           surface.finish()
 181           output_headers("application/postscript", "map.ps", file_size(file))
 182           output_file(file)
 183         else:
 184           output_error("Unknown format '%s'" % form.getvalue("format"))
 185       else:
 186         pid, status = os.waitpid(pid, 0)
 187         if status & 0xff == signal.SIGXCPU:
 188           output_error("CPU time limit exceeded", "509 Resource Limit Exceeded")
 189         elif status & 0xff == signal.SIGSEGV:
 190           output_error("Memory limit exceeded", "509 Resource Limit Exceeded")
 191         elif status != 0:
 192           output_error("Internal server error", "500 Internal Server Error")