mirror of
https://github.com/NeoFlock/neonucleus.git
synced 2025-09-25 01:23:31 +02:00
106 lines
2.7 KiB
Lua
106 lines
2.7 KiB
Lua
local component = require("component")
|
|
local shell = require("shell")
|
|
local fs = require("filesystem")
|
|
|
|
local args, options = shell.parse(...)
|
|
|
|
if #args < 1 and not options.l then
|
|
io.write("Usage: flash [-qlr] [<bios.lua>] [label]\n")
|
|
io.write(" q: quiet mode, don't ask questions.\n")
|
|
io.write(" l: print current contents of installed EEPROM.\n")
|
|
io.write(" r: save the current contents of installed EEPROM to file.\n")
|
|
return
|
|
end
|
|
|
|
local function printRom()
|
|
local eeprom = component.eeprom
|
|
io.write(eeprom.get())
|
|
|
|
return nil
|
|
end
|
|
|
|
local function readRom()
|
|
local eeprom = component.eeprom
|
|
local fileName = shell.resolve(args[1])
|
|
if not options.q then
|
|
if fs.exists(fileName) then
|
|
io.write("Are you sure you want to overwrite " .. fileName .. "?\n")
|
|
io.write("Type `y` to confirm.\n")
|
|
repeat
|
|
local response = io.read()
|
|
until response and response:lower():sub(1, 1) == "y"
|
|
end
|
|
io.write("Reading EEPROM " .. eeprom.address .. ".\n" )
|
|
end
|
|
local bios = eeprom.get()
|
|
local file = assert(io.open(fileName, "wb"))
|
|
file:write(bios)
|
|
file:close()
|
|
if not options.q then
|
|
io.write("All done!\nThe label is '" .. eeprom.getLabel() .. "'.\n")
|
|
end
|
|
|
|
return nil
|
|
end
|
|
|
|
local function writeRom()
|
|
local file = assert(io.open(args[1], "rb"))
|
|
local bios = file:read("*a")
|
|
file:close()
|
|
|
|
if not options.q then
|
|
io.write("Insert the EEPROM you would like to flash.\n")
|
|
io.write("When ready to write, type `y` to confirm.\n")
|
|
repeat
|
|
local response = io.read()
|
|
until response and response:lower():sub(1, 1) == "y"
|
|
io.write("Beginning to flash EEPROM.\n")
|
|
end
|
|
|
|
local eeprom = component.eeprom
|
|
|
|
if not options.q then
|
|
io.write("Flashing EEPROM " .. eeprom.address .. ".\n")
|
|
io.write("Please do NOT power down or restart your computer during this operation!\n")
|
|
end
|
|
|
|
local result, reason = eeprom.set(bios)
|
|
if reason then
|
|
return nil, reason
|
|
end
|
|
|
|
local label = args[2]
|
|
if not options.q and not label then
|
|
io.write("Enter new label for this EEPROM. Leave input blank to leave the label unchanged.\n")
|
|
label = io.read()
|
|
end
|
|
if label and #label > 0 then
|
|
local result, reason = eeprom.setLabel(label)
|
|
if reason then
|
|
return nil, reason
|
|
end
|
|
|
|
if not options.q then
|
|
io.write("Set label to '" .. eeprom.getLabel() .. "'.\n")
|
|
end
|
|
end
|
|
|
|
if not options.q then
|
|
io.write("All done! You can remove the EEPROM and re-insert the previous one now.\n")
|
|
end
|
|
|
|
return nil
|
|
end
|
|
|
|
local result, reason
|
|
if options.l then
|
|
result, reason = printRom()
|
|
elseif options.r then
|
|
result, reason = readRom()
|
|
else
|
|
result, reason = writeRom()
|
|
end
|
|
if reason then
|
|
io.stderr:write(reason..'\n')
|
|
return 1
|
|
end |