mirror of
https://github.com/LuaJIT/LuaJIT.git
synced 2025-02-07 23:24:09 +00:00
126 lines
3.2 KiB
Lua
126 lines
3.2 KiB
Lua
----------------------------------------------------------------------------
|
|
-- LuaJIT module to save/list bytecode.
|
|
--
|
|
-- Copyright (C) 2005-2011 Mike Pall. All rights reserved.
|
|
-- Released under the MIT/X license. See Copyright Notice in luajit.h
|
|
----------------------------------------------------------------------------
|
|
--
|
|
-- This module saves or lists the bytecode for an input file.
|
|
-- It's run by the -b command line option.
|
|
--
|
|
------------------------------------------------------------------------------
|
|
|
|
-- Cache some library functions and objects.
|
|
local jit = require("jit")
|
|
assert(jit.version_num == 20000, "LuaJIT core/library version mismatch")
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
local function usage()
|
|
io.stderr:write[[
|
|
Save LuaJIT bytecode: luajit -b[options] input output
|
|
-l Only list bytecode.
|
|
-s Strip debug info (default).
|
|
-g Keep debug info.
|
|
-e chunk Use chunk string as input.
|
|
-- Stop handling options.
|
|
- Use stdin as input and/or stdout as output.
|
|
]]
|
|
os.exit(1)
|
|
end
|
|
|
|
local function readfile(input)
|
|
if type(input) == "function" then return input end
|
|
if input == "-" then input = nil end
|
|
local f, err = loadfile(input)
|
|
if not f then
|
|
io.stderr:write("luajit: ", err, "\n")
|
|
os.exit(1)
|
|
end
|
|
return f
|
|
end
|
|
|
|
local function readstring(input)
|
|
local f, err = loadstring(input)
|
|
if not f then
|
|
io.stderr:write("luajit: ", err, "\n")
|
|
os.exit(1)
|
|
end
|
|
return f
|
|
end
|
|
|
|
local function savefile(name, mode)
|
|
if name == "-" then return io.stdout end
|
|
local fp, err = io.open(name, mode)
|
|
if not fp then
|
|
io.stderr:write("luajit: cannot write ", err, "\n")
|
|
os.exit(1)
|
|
end
|
|
return fp
|
|
end
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
local function bclist(input, output)
|
|
local f = readfile(input)
|
|
require("jit.bc").dump(f, savefile(output, "w"), true)
|
|
end
|
|
|
|
local function bcsave(input, output, strip)
|
|
local f = readfile(input)
|
|
local s = string.dump(f, strip)
|
|
local fp = savefile(output, "wb")
|
|
local ok, err = fp:write(s)
|
|
if ok and output ~= "-" then ok, err = fp:close() end
|
|
if not ok then
|
|
io.stderr:write("luajit: cannot write ", arg[2], ": ", err, "\n")
|
|
os.exit(1)
|
|
end
|
|
end
|
|
|
|
local function docmd(...)
|
|
local arg = {...}
|
|
local n = 1
|
|
local list = false
|
|
local strip = true
|
|
while n <= #arg do
|
|
local a = arg[n]
|
|
if type(a) == "string" and string.sub(a, 1, 1) == "-" and a ~= "-" then
|
|
if a == "--" then table.remove(arg, n); break end
|
|
for m=2,#a do
|
|
local opt = string.sub(a, m, m)
|
|
if opt == "l" then
|
|
list = true
|
|
elseif opt == "s" then
|
|
strip = true
|
|
elseif opt == "g" then
|
|
strip = false
|
|
elseif opt == "e" then
|
|
if n ~= 1 or #arg < 2 or m ~= #a then usage() end
|
|
arg[2] = readstring(arg[2])
|
|
else
|
|
usage()
|
|
end
|
|
end
|
|
table.remove(arg, n)
|
|
else
|
|
n = n + 1
|
|
end
|
|
end
|
|
if list then
|
|
if #arg == 0 or #arg > 2 then usage() end
|
|
bclist(arg[1], arg[2] or "-")
|
|
else
|
|
if #arg ~= 2 then usage() end
|
|
bcsave(arg[1], arg[2], strip)
|
|
end
|
|
end
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
-- Public module functions.
|
|
module(...)
|
|
|
|
start = docmd -- Process -b command line option.
|
|
|