local getopt = require('getopt')
local bin = require('bin')
local ansicolors = require('ansicolors')

copyright = 'Copyright (c) 2019 Bogito. All rights reserved.'
author = 'Bogito'
version = 'v1.1.2'
desc = [[
This script will read the flash memory of RDV4 using SPIFFS and print the stored passwords.
It was meant to be used as a help tool after using the BogRun standalone mode.
]]
example = [[
    -- This will read the hf_bog.log file in SPIFFS and print the stored passwords
    script run mem_spiffs_readpwd

    -- This will read the other.log file in SPIFFS and print the stored passwords
    script run mem_spiffs_readpwd -f other.log

    -- This will delete the hf_bog.log file from SPIFFS
    script run mem_spiffs_readpwd -r
]]
usage = [[
    script run mem_spiffs_readpwd [-h] [-f <filename>] [-r]
]]
arguments = [[
    -h              :  this help
    -f <filename>   :  filename in SPIFFS
    -r              :  delete filename from SPIFFS
]]
---
-- This is only meant to be used when errors occur
local function oops(err)
    print('ERROR:', err)
    core.clearCommandBuffer()
    return nil, err
end
---
-- Usage help
local function help()
    print(copyright)
    print(author)
    print(version)
    print(desc)
    print(ansicolors.cyan..'Usage'..ansicolors.reset)
    print(usage)
    print(ansicolors.cyan..'Arguments'..ansicolors.reset)
    print(arguments)
    print(ansicolors.cyan..'Example usage'..ansicolors.reset)
    print(example)
end
---
-- The main entry point
local function main(args)

    print( string.rep('--',20) )
    print('Read passwords stored in memory (SPIFFS)')
    print( string.rep('--',20) )
    print()

    local data, length, err, removeflag
    local filename = 'hf_bog.log'
    local keylength = 4

    for o, a in getopt.getopt(args, 'rf:h') do

        -- help
        if o == 'h' then return help() end

        -- offset
        if o == 'f' then filename = a end

        -- remove
        if o == 'r' then removeflag = true end

    end

    if removeflag then
        print('Deleting file '..filename.. ' from SPIFFS if exists')
        core.console("mem spiffs remove -f " ..filename)
        return
    end

    data, length, err = core.GetFromFlashMemSpiffs(filename)
    if data == nil then return oops('Problem while reading file from SPIFFS') end

    --print('Filename', filename)
    --print('Filesize (B)', length)

    _, s = bin.unpack('H'..length, data)

    local cnt = 0, i
    for i = 1, length/keylength do
        key  = string.sub(s, (i-1)*8+1, i*8)
        print(string.format('[%02d] %s',i, key))
        cnt = cnt + 1
    end
    print( string.rep('--',20) )
    print( ('[+] found %d passwords'):format(cnt))

end

main(args)