libs/httpd/luasrc/httpd/handler/luci.lua

   1 --[[
   2
   3 HTTP server implementation for LuCI - luci handler
   4 (c) 2008 Steven Barth <steven@midlink.org>
   5
   6 Licensed under the Apache License, Version 2.0 (the "License");
   7 you may not use this file except in compliance with the License.
   8 You may obtain a copy of the License at
   9
  10         http://www.apache.org/licenses/LICENSE-2.0
  11
  12 $Id$
  13
  14 ]]--
  15
  16 module("luci.httpd.handler.luci", package.seeall)
  17
  18 require("luci.dispatcher")
  19 require("luci.http")
  20 require("luci.http.protocol.date")
  21 local ltn12 = require("luci.ltn12")
  22
  23 Luci = luci.util.class(luci.httpd.module.Handler)
  24 Response = luci.httpd.module.Response
  25
  26 function Luci.__init__(self, limit)
  27         luci.httpd.module.Handler.__init__(self)
  28         self.limit = limit or 5
  29         self.running = {}
  30         setmetatable(self.running, {__mode = "v"})
  31 end
  32
  33 function Luci.handle_head(self, ...)
  34         local response, sourceout = self:handle_get(...)
  35         return response
  36 end
  37
  38 function Luci.handle_post(self, ...)
  39         return self:handle_get(...)
  40 end
  41
  42 function Luci.handle_get(self, request, sourcein, sinkerr)
  43         if self.limit and #self.running >= self.limit then
  44                 for k, v in ipairs(self.running) do
  45                         if coroutine.status(v) == "dead" then
  46                                 collectgarbage()
  47                                 break
  48                         end
  49                 end
  50                 if #self.running >= self.limit then
  51                         return self:failure(503, "Overload")
  52                 end
  53         end
  54         table.insert(self.running, coroutine.running())
  55
  56         local r = luci.http.Request(
  57                 request.env,
  58                 sourcein,
  59                 sinkerr
  60         )
  61
  62         local res, id, data1, data2 = true, 0, nil, nil
  63         local headers = {}
  64         local status = 200
  65         local active = true
  66
  67         local x = coroutine.create(luci.dispatcher.httpdispatch)
  68         while not id or id < 3 do
  69                 coroutine.yield()
  70
  71                 res, id, data1, data2 = coroutine.resume(x, r)
  72
  73                 if not res then
  74                         status = 500
  75                         headers["Content-Type"] = "text/plain"
  76                         local err = {id}
  77                         return Response( status, headers ), function() return table.remove(err) end
  78                 end
  79
  80                 if id == 1 then
  81                         status = data1
  82                 elseif id == 2 then
  83                         headers[data1] = data2
  84                 end
  85         end
  86
  87         local function iter()
  88                 local res, id, data = coroutine.resume(x)
  89                 if not res then
  90                         return nil, id
  91                 elseif not id or not active then
  92                         return true
  93                 elseif id == 5 then
  94                         active = false
  95
  96                         while (coroutine.resume(x)) do
  97                         end
  98
  99                         return nil
 100                 elseif id == 4 then
 101                         return data
 102                 end
 103                 if coroutine.status(x) == "dead" then
 104                         return nil
 105                 end
 106         end
 107
 108         return Response(status, headers), iter
 109 end