1
0
mirror of https://github.com/DigitalDevices/octonet.git synced 2023-10-10 13:36:52 +02:00
octonet/octoserve/var/www/multicast.lua
2015-08-05 22:22:06 +02:00

225 lines
6.1 KiB
Lua

#!/usr/bin/lua
local host = os.getenv("HTTP_HOST")
local proto = os.getenv("SERVER_PROTOCOL")
local query = os.getenv("QUERY_STRING")
local method = os.getenv("REQUEST_METHOD")
local clength = os.getenv("CONTENT_LENGTH")
local ctype = os.getenv("CONTENT_TYPE")
function http_print(s)
if s then
io.stdout:write(tostring(s).."\r\n")
else
io.stdout:write("\r\n")
end
end
if #arg> 0 then
method="GET"
query="select=m3u"
proto = "HTTP/1.0"
end
function SendError(err,desc)
http_print(proto.." "..err)
http_print()
local file = io.open("e404.html")
if file then
local tmp = file:read("*a")
tmp = string.gsub(tmp,"404 Not Found",err .. " " .. desc)
http_print(tmp)
file:close()
end
end
function GetDevID()
local devid = nil
local tmp = io.open("/config/device.id")
if tmp then
devid = tonumber(tmp:read())
tmp:close()
end
return devid
end
function CreateM3U()
local m3u = nil
local devid = GetDevID()
if not devid then return nil end
local file = io.open("/config/mcsetup.csv","r")
if file then
local line = file:read()
if line then
line = string.gsub(line,'"','')
if string.match(line,"TITLE,REQUEST,PIDS,LANPORTS") then
local i = 1
m3u = "#EXTM3U".."\n"
while i < 13 do
line = file:read()
if not line then break end
if DebugFlag then print(line) end
local title,request,pids,lanports = string.match(line,'%"(.*)%",%"(.*)%",%"P(%a-[%d%:]-)%",%"L(%a-[%d%:]-)%"')
if not title then
title,request,pids = string.match(line,'%"(.*)%",%"(.*)%",%"P(%a-[%d%:]-)%"')
end
if not title or not request or not pids then
break
end
if pids == "" then break end
m3u = m3u .. "#EXTINF:0,"..title.."\n"
.. "rtp://239."..devid..".2."..i..":6670\n"
i = i + 1
end
elseif line:match("TITLE,REQUEST,PIDS,PROTO,IP,PORT,TTL,LANPORTS") then
local i = 1
m3u = "#EXTM3U".."\n"
while i < 13 do
line = file:read()
if not line then break end
if DebugFlag then print(line) end
local title,request,pids,proto,ip,port,ttl,lanports = line:match('%"(.*)%",%"(.*)%",%"P(%a-[%d%:]-)%",%"(%u%u%u)%",%"(%d*%.%d*%.%d*%.%d*)%",(%d*),(%d*),%"L(%a-[%d%:]-)%"')
if not title then
title,request,pids,proto,ip,port,ttl = line:match('%"(.*)%",%"(.*)%",%"P(%a-[%d%:]-)%",%"(%u%u%u)%",%"(%d*%.%d*%.%d*%.%d*)%",(%d*),(%d*)')
end
if not title or not request or not pids then
break
end
if pids == "" then break end
if proto ~= "UDP" and proto ~= "RTP" then break end
if proto == "UDP" then ip = "@"..ip end
m3u = m3u .. "#EXTINF:0,"..title.."\n"
.. proto:lower() .. "://"..ip..":"..port.."\n"
i = i + 1
end
end
end
file:close()
end
return m3u
end
if method == "GET" then
local path = nil
local disposition = nil
local subtype="csv"
if string.match(query,"select=sample") then
local name = string.match(query,"name=(%w+)")
if name then
path = "/var/mcsetup/"..name..".csv"
disposition = name..".csv"
else
SendError("404","Request Error")
return
end
elseif string.match(query,"select=current") then
path = "/config/mcsetup.csv"
disposition = "mcsetup.csv"
elseif string.match(query,"select=m3u") then
path = "m3u"
disposition = "mclist.m3u"
subtype = "m3u"
elseif string.match(query,"select=copy") then
local name = string.match(query,"name=(%w+)")
if name then
path = "/var/mcsetup/"..name..".csv"
disposition = "copy"
else
SendError("404","Request Error")
return
end
elseif string.match(query,"select=disable") then
disposition = "disable"
end
if disposition == "disable" then
os.remove("/config/mcsetup.csv")
os.execute('echo "1" >/tmp/mc.tmp;mv -f /tmp/mc.tmp /tmp/mc.signal');
http_print(proto.." 303")
http_print("Location: http://"..host.."/multicast.html")
http_print("")
return
end
if disposition == "copy" then
os.execute("cp "..path.." /config/mcsetup.csv")
os.execute('echo "1" >/tmp/mc.tmp;mv -f /tmp/mc.tmp /tmp/mc.signal');
http_print(proto.." 303")
http_print("Location: http://"..host.."/multicast.html")
http_print()
return
end
local data = nil
if path == "m3u" then
data = CreateM3U()
else
local tmp = io.open(path,"r")
if tmp then
data = tmp:read("*a")
tmp:close()
end
end
if data then
http_print(proto.." 200" )
http_print("Pragma: no-cache")
http_print("Content-Type: text/"..subtype)
http_print('Content-Disposition: filename="'..disposition..'"')
http_print(string.format("Content-Length: %d",#data))
http_print()
http_print(data)
else
SendError("404",disposition.." not found")
end
elseif method == "POST" and clength and ctype then
if not string.match(ctype,"multipart/form%-data") then
SendError("404","??")
return
end
local boundary = string.match(ctype,"boundary=(.*)")
if not boundary then
SendError("404","???")
return
end
while true do
local line = io.stdin:read()
line = string.gsub(line,"\r","")
if line == "" then break end
end
data = io.stdin:read(16384)
data = string.sub(data,1,#data - #boundary - 4)
if data:match("^\239\187\191") then data = data:sub(4) end
data = string.gsub(data,"\r\n","\n") -- Windows -> Unix
data = string.gsub(data,"\r","\n") -- MAC -> Unix
-- if data:match("^TITLE,REQUEST,PIDS,PROTO,IP,PORT,TTL,LANPORTS") then
if data:match("^TITLE,REQUEST,PIDS,LANPORTS") or data:match("^TITLE,REQUEST,PIDS,PROTO,IP,PORT,TTL,LANPORTS") then
file = io.open("/config/mcsetup.csv","w")
if file then
file:write(data)
file:close()
os.execute('echo "1" >/tmp/mc.tmp;mv -f /tmp/mc.tmp /tmp/mc.signal');
end
end
http_print(proto.." 303")
http_print("Location: http://"..host.."/multicast.html")
http_print()
else
SendError("500","What")
end