git.sesse.net Git - vlc/blob - share/lua/intf/http.lua

   1 --[==========================================================================[
   2  http.lua: HTTP interface module for VLC
   3 --[==========================================================================[
   4  Copyright (C) 2007-2009 the VideoLAN team
   5  $Id$
   6
   7  Authors: Antoine Cellerier <dionoea at videolan dot org>
   8
   9  This program is free software; you can redistribute it and/or modify
  10  it under the terms of the GNU General Public License as published by
  11  the Free Software Foundation; either version 2 of the License, or
  12  (at your option) any later version.
  13
  14  This program is distributed in the hope that it will be useful,
  15  but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  GNU General Public License for more details.
  18
  19  You should have received a copy of the GNU General Public License
  20  along with this program; if not, write to the Free Software
  21  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22 --]==========================================================================]
  23
  24 --[==========================================================================[
  25 Configuration options:
  26  * host: A host to listen on.
  27  * dir: Directory to use as the http interface's root.
  28  * no_error_detail: If set, do not print the Lua error message when generating
  29                     a page fails.
  30  * no_index: If set, don't build directory indexes
  31 --]==========================================================================]
  32
  33
  34 require "common"
  35
  36 vlc.msg.info("Lua HTTP interface")
  37
  38 open_tag = "<?vlc"
  39 close_tag = "?>"
  40
  41 -- TODO: use internal VLC mime lookup function for mimes not included here
  42 local mimes = {
  43     txt = "text/plain",
  44     html = "text/html",
  45     xml = "text/xml",
  46     js = "text/javascript",
  47     css = "text/css",
  48     png = "image/png",
  49     jpg = "image/jpeg",
  50     jpeg = "image/jpeg",
  51     ico = "image/x-icon",
  52 }
  53
  54 function escape(s)
  55     return (string.gsub(s,"([%^%$%%%.%[%]%*%+%-%?])","%%%1"))
  56 end
  57
  58 function process_raw(filename)
  59     local input = io.open(filename):read("*a")
  60     -- find the longest [===[ or ]=====] type sequence and make sure that
  61     -- we use one that's longer.
  62     local str="X"
  63     for str2 in string.gmatch(input,"[%[%]]=*[%[%]]") do
  64         if #str < #str2 then str = str2 end
  65     end
  66     str=string.rep("=",#str-1)
  67
  68     --[[ FIXME:
  69     <?xml version="1.0" encoding="charset" standalone="yes" ?> is still a problem. The closing '?>' needs to be printed using '?<?vlc print ">" ?>' to prevent a parse error.
  70     --]]
  71     local code0 = string.gsub(input,escape(close_tag)," print(["..str.."[")
  72     local code1 = string.gsub(code0,escape(open_tag),"]"..str.."]) ")
  73     local code = "print(["..str.."["..code1.."]"..str.."])"
  74     --[[ Uncomment to debug
  75     if string.match(filename,"vlm_cmd.xml$") then
  76     io.write(code)
  77     io.write("\n")
  78     end
  79     --]]
  80     return assert(loadstring(code,filename))
  81 end
  82
  83 function process(filename)
  84     local mtime = 0    -- vlc.net.stat(filename).modification_time
  85     local func = false -- process_raw(filename)
  86     return function(...)
  87         local new_mtime = vlc.net.stat(filename).modification_time
  88         if new_mtime ~= mtime then
  89             -- Re-read the file if it changed
  90             if mtime == 0 then
  91                 vlc.msg.dbg("Loading `"..filename.."'")
  92             else
  93                 vlc.msg.dbg("Reloading `"..filename.."'")
  94             end
  95             func = process_raw(filename)
  96             mtime = new_mtime
  97         end
  98         return func(...)
  99     end
 100 end
 101
 102
 103 function callback_error(path,url,msg)
 104     local url = url or "&lt;page unknown&gt;"
 105     return  [[<html xmlns="http://www.w3.org/1999/xhtml">
 106 <head>
 107 <title>Error loading ]]..url..[[</title>
 108 </head>
 109 <body>
 110 <h1>Error loading ]]..url..[[</h1><pre>]]..(config.no_error_detail and "Remove configuration option `no_error_detail' on the server to get more information." or tostring(msg))..[[</pre>
 111 <p>
 112 <a href="http://www.videolan.org/">VideoLAN</a><br/>
 113 <a href="http://www.lua.org/manual/5.1/">Lua 5.1 Reference Manual</a>
 114 </p>
 115 </body>
 116 </html>]]
 117 end
 118
 119 function dirlisting(url,listing,acl_)
 120     local list = {}
 121     for _,f in ipairs(listing) do
 122         if not string.match(f,"^%.") then
 123             table.insert(list,"<li><a href='"..f.."'>"..f.."</a></li>")
 124         end
 125     end
 126     list = table.concat(list)
 127     local function callback()
 128         return [[<html xmlns="http://www.w3.org/1999/xhtml">
 129 <head>
 130 <title>Directory listing ]]..url..[[</title>
 131 </head>
 132 <body>
 133 <h1>Directory listing ]]..url..[[</h1><ul>]]..list..[[</ul>
 134 </body>
 135 </html>]]
 136     end
 137     return h:file(url,"text/html",nil,nil,acl_,callback,nil)
 138 end
 139
 140 -- FIXME: Experimental art support. Needs some cleaning up.
 141 function callback_art(data, request)
 142     local art = function(data, request)
 143         local metas = vlc.input.metas()
 144         local filename = vlc.strings.decode_uri(string.gsub(metas["artwork_url"],"file://",""))
 145         local size = vlc.net.stat(filename).size
 146         local ext = string.match(filename,"%.([^%.]-)$")
 147         local raw = io.open(filename):read("*a")
 148         local content = [[Content-Type: ]]..mimes[ext]..[[
 149
 150 Content-Length: ]]..size..[[
 151
 152
 153 ]]..raw..[[
 154
 155 ]]
 156         return content
 157     end
 158
 159     local ok, content = pcall(art, data, request)
 160     if not ok then
 161         return [[Status: 404
 162 Content-Type: text/plain
 163 Content-Length: 5
 164
 165 Error
 166 ]]
 167     end
 168     return content
 169 end
 170
 171 function file(h,path,url,acl_,mime)
 172     local generate_page = process(path)
 173     local callback = function(data,request)
 174         -- FIXME: I'm sure that we could define a real sandbox
 175         -- redefine print
 176         local page = {}
 177         local function pageprint(...)
 178             for i=1,select("#",...) do
 179                 if i== 1 then
 180                     table.insert(page,tostring(select(i,...)))
 181                 else
 182                     table.insert(page," "..tostring(select(i,...)))
 183                 end
 184             end
 185         end
 186         _G._GET = parse_url_request(request)
 187         local oldprint = print
 188         print = pageprint
 189         local ok, msg = pcall(generate_page)
 190         -- reset
 191         print = oldprint
 192         if not ok then
 193             return callback_error(path,url,msg)
 194         end
 195         return table.concat(page)
 196     end
 197     return h:file(url or path,mime,nil,nil,acl_,callback,nil)
 198 end
 199
 200 function rawfile(h,path,url,acl_)
 201     local filename = path
 202     local mtime = 0    -- vlc.net.stat(filename).modification_time
 203     local page = false -- io.open(filename):read("*a")
 204     local callback = function(data,request)
 205         local new_mtime = vlc.net.stat(filename).modification_time
 206         if mtime ~= new_mtime then
 207             -- Re-read the file if it changed
 208             if mtime == 0 then
 209                 vlc.msg.dbg("Loading `"..filename.."'")
 210             else
 211                 vlc.msg.dbg("Reloading `"..filename.."'")
 212             end
 213             page = io.open(filename):read("*a")
 214             mtime = new_mtime
 215         end
 216         return page
 217     end
 218     return h:file(url or path,nil,nil,nil,acl_,callback,nil)
 219 end
 220
 221 function parse_url_request(request)
 222     if not request then return {} end
 223     local t = {}
 224     for k,v in string.gmatch(request,"([^=&]+)=?([^=&]*)") do
 225         local k_ = vlc.strings.decode_uri(k)
 226         local v_ = vlc.strings.decode_uri(v)
 227         if t[k_] ~= nil then
 228             local t2
 229             if type(t[k_]) ~= "table" then
 230                 t2 = {}
 231                 table.insert(t2,t[k_])
 232                 t[k_] = t2
 233             else
 234                 t2 = t[k_]
 235             end
 236             table.insert(t2,v_)
 237         else
 238             t[k_] = v_
 239         end
 240     end
 241     return t
 242 end
 243
 244 local function find_datadir(name)
 245     local list = vlc.misc.datadir_list(name)
 246     for _, l in ipairs(list) do
 247         local s = vlc.net.stat(l)
 248         if s then
 249             return l
 250         end
 251     end
 252     error("Unable to find the `"..name.."' directory.")
 253 end
 254 http_dir = config.dir or find_datadir("http")
 255
 256 do
 257     local oldpath = package.path
 258     package.path = http_dir.."/?.lua"
 259     local ok, err = pcall(require,"custom")
 260     if not ok then
 261         vlc.msg.warn("Couldn't load "..http_dir.."/custom.lua",err)
 262     else
 263         vlc.msg.dbg("Loaded "..http_dir.."/custom.lua")
 264     end
 265     package.path = oldpath
 266 end
 267 local files = {}
 268 local function load_dir(dir,root,parent_acl)
 269     local root = root or "/"
 270     local has_index = false
 271     local my_acl = parent_acl
 272     do
 273         local af = dir.."/.hosts"
 274         local s = vlc.net.stat(af)
 275         if s and s.type == "file" then
 276             -- We found an acl
 277             my_acl = vlc.acl(false)
 278             my_acl:load_file(af)
 279         end
 280     end
 281     local d = vlc.net.opendir(dir)
 282     for _,f in ipairs(d) do
 283         if not string.match(f,"^%.") then
 284             local s = vlc.net.stat(dir.."/"..f)
 285             if s.type == "file" then
 286                 local url
 287                 if f == "index.html" then
 288                     url = root
 289                     has_index = true
 290                 else
 291                     url = root..f
 292                 end
 293                 local ext = string.match(f,"%.([^%.]-)$")
 294                 local mime = mimes[ext]
 295                 -- print(url,mime)
 296                 if mime and string.match(mime,"^text/") then
 297                     table.insert(files,file(h,dir.."/"..f,url,my_acl,mime))
 298                 else
 299                     table.insert(files,rawfile(h,dir.."/"..f,url,my_acl))
 300                 end
 301             elseif s.type == "dir" then
 302                 load_dir(dir.."/"..f,root..f.."/",my_acl)
 303             end
 304         end
 305     end
 306     if not has_index and not config.no_index then
 307         -- print("Adding index for", root)
 308         table.insert(files,dirlisting(root,d,my_acl))
 309     end
 310     return my_acl
 311 end
 312
 313 local u = vlc.net.url_parse( config.host or "0.0.0.0:8080" )
 314 h = vlc.httpd(u.host,u.port)
 315 local root_acl = load_dir( http_dir )
 316 local a = h:handler("/art",nil,nil,root_acl,callback_art,nil)
 317
 318 while not vlc.misc.lock_and_wait() do end -- everything happens in callbacks
 319