148 lines
3.3 KiB
Lua
148 lines
3.3 KiB
Lua
local arg = arg or {...}
|
|
local string, table = string, table
|
|
|
|
local POS_BITS = 12
|
|
local LEN_BITS = 16 - POS_BITS
|
|
local POS_SIZE = 1 << POS_BITS
|
|
local LEN_SIZE = 1 << LEN_BITS
|
|
local LEN_MIN = 3
|
|
|
|
function lzss_decompress(input)
|
|
local offset, output = 1, {}
|
|
local window = ''
|
|
|
|
while offset <= #input do
|
|
local flags = string.byte(input, offset)
|
|
offset = offset + 1
|
|
|
|
for i = 1, 8 do
|
|
local str = nil
|
|
if (flags & 1) ~= 0 then
|
|
if offset <= #input then
|
|
str = string.sub(input, offset, offset)
|
|
offset = offset + 1
|
|
end
|
|
else
|
|
if offset + 1 <= #input then
|
|
local tmp = string.unpack('>I2', input, offset)
|
|
offset = offset + 2
|
|
local pos = (tmp >> LEN_BITS) + 1
|
|
local len = (tmp & (LEN_SIZE - 1)) + LEN_MIN
|
|
str = string.sub(window, pos, pos + len - 1)
|
|
end
|
|
end
|
|
flags = flags >> 1
|
|
if str then
|
|
output[#output + 1] = str
|
|
window = string.sub(window .. str, -POS_SIZE)
|
|
end
|
|
end
|
|
end
|
|
|
|
return table.concat(output)
|
|
end
|
|
--print("Decompressing CPIO...")
|
|
--local code = lzss_decompress(@--[{io.stdin:read("*a")}])
|
|
local code = @[{io.stdin:read("*a")}]
|
|
local dat = code
|
|
local tbl = {}
|
|
|
|
local pos = 1
|
|
local function read(n)
|
|
local d = dat:sub(pos, pos+n-1)
|
|
pos = pos + n
|
|
return d
|
|
end
|
|
|
|
local function seek(n)
|
|
pos = pos + n
|
|
return pos
|
|
end
|
|
|
|
--[[local function readint(amt, rev)
|
|
local tmp = 0
|
|
for i=(rev and amt) or 1, (rev and 1) or amt, (rev and -1) or 1 do
|
|
tmp = tmp | (read(1):byte() << ((i-1)*8))
|
|
end
|
|
return tmp
|
|
end
|
|
|
|
while true do
|
|
local dent = {}
|
|
dent.magic = readint(2)
|
|
local rev = false
|
|
if (dent.magic ~= tonumber("070707", 8)) then rev = true end
|
|
dent.dev = readint(2)
|
|
dent.ino = readint(2)
|
|
dent.mode = readint(2)
|
|
dent.uid = readint(2)
|
|
dent.gid = readint(2)
|
|
dent.nlink = readint(2)
|
|
dent.rdev = readint(2)
|
|
dent.mtime = (readint(2) << 16) | readint(2)
|
|
dent.namesize = readint(2)
|
|
dent.filesize = (readint(2) << 16) | readint(2)
|
|
local name = read(dent.namesize):sub(1, dent.namesize-1)
|
|
if (name == "TRAILER!!!") then break end
|
|
--for k, v in pairs(dent) do
|
|
-- print(k, v)
|
|
--end
|
|
dent.name = name
|
|
if (dent.namesize % 2 ~= 0) then
|
|
pos = pos + 1
|
|
end
|
|
if (dent.mode & 32768 ~= 0) then
|
|
--fwrite()
|
|
end
|
|
dent.pos = pos
|
|
pos = pos + dent.filesize
|
|
if (dent.filesize % 2 ~= 0) then
|
|
pos = pos + 1
|
|
end
|
|
tbl[#tbl+1] = dent
|
|
end
|
|
]]
|
|
|
|
local magic = 0x5f7d
|
|
local magic_rev = 0x7d5f
|
|
local header_fmt = "I2I2I2I2I2I6I6"
|
|
local en = string.unpack("=I2", string.char(0x7d, 0x5f)) == magic -- true = LE, false = BE
|
|
local function get_end(e)
|
|
return (e and "<") or ">"
|
|
end
|
|
local function read_header(dat)
|
|
local e = get_end(en)
|
|
local m = string.unpack(e.."I2", dat)
|
|
if m ~= magic and m ~= magic_rev then return nil, "bad magic" end
|
|
if m ~= magic then
|
|
e = get_end(not en)
|
|
end
|
|
local ent = {}
|
|
ent.magic, ent.namesize, ent.mode, ent.uid, ent.gid, ent.filesize, ent.mtime = string.unpack(e..header_fmt, dat)
|
|
return ent
|
|
end
|
|
|
|
local lname = ""
|
|
while lname ~= "TRAILER!!!" do
|
|
local dat = read(22)
|
|
local e = assert(read_header(dat))
|
|
e.name = read(e.namesize)
|
|
e.pos = seek(e.namesize & 1)
|
|
seek(e.filesize + (e.filesize & 1))
|
|
lname = e.name
|
|
if lname ~= "TRAILER!!!" then
|
|
tbl[#tbl+1] = e
|
|
end
|
|
end
|
|
|
|
local unpack = unpack or table.unpack
|
|
|
|
for i=1, #tbl do
|
|
if (tbl[i].name == "init.lua") then
|
|
load(dat:sub(tbl[i].pos, tbl[i].pos+tbl[i].filesize-1))(tbl, dat, unpack(arg))
|
|
if os.exit then os.exit(0) else return end
|
|
end
|
|
end
|
|
|
|
error("Init not found.")
|