cvs.zerfleddert.de Git - proxmark3-svn/blob - client/lualibs/html_dumplib.lua

   1 bin = require('bin')
   2
   3
   4 -------------------------------
   5 -- Some utilities
   6 -------------------------------
   7
   8 ---
   9 -- A debug printout-function
  10 local function dbg(args)
  11
  12     if type(args) == "table" then
  13                 local i = 1
  14                 while args[i] do
  15                         print("###", args[i])
  16                         i = i+1
  17                 end
  18         else
  19                 print("###", args)
  20         end
  21 end
  22 ---
  23 -- This is only meant to be used when errors occur
  24 local function oops(err)
  25         print("ERROR: ",err)
  26         return nil, err
  27 end
  28
  29 local function save_HTML(javascript, filename)
  30
  31         -- Read the HTML-skel file
  32         local skel = require("htmlskel")
  33         html = skel.getHTML(javascript);
  34
  35         -- Open the output file
  36
  37         local outfile = io.open(filename, "w")
  38         if outfile == nil then
  39                 return oops(string.format("Could not write to file %s",tostring(filename)))
  40         end
  41         -- Write the data into it
  42         outfile:write(html)
  43         io.close(outfile)
  44
  45         -- Done
  46         return filename
  47
  48 end
  49
  50 local function save_TEXT(data,filename)
  51         -- Open the output file
  52         local outfile = io.open(filename, "w")
  53         if outfile == nil then
  54                 return oops(string.format("Could not write to file %s",tostring(filename)))
  55         end
  56
  57         outfile:write(data)
  58         io.close(outfile)
  59         return filename
  60 end
  61
  62 local function save_BIN(data, filename)
  63         -- Open the output file
  64
  65         local outfile = io.open(filename, "wb")
  66         if outfile == nil then
  67                 return oops(string.format("Could not write to file %s",tostring(filename)))
  68         end
  69
  70         -- Write the data into it
  71         local i = 1
  72         while data[i] do
  73                 outfile:write(data[i])
  74                 i = i+1
  75         end
  76
  77         io.close(outfile)
  78         return filename
  79 end
  80
  81 local function convert_ascii_dump_to_JS(infile)
  82         local t = infile:read("*all")
  83         local cleaned
  84         local output = "[";
  85         for line in string.gmatch(t, "[^\n]+") do
  86             if string.byte(line,1) ~= string.byte("+",1) then
  87                         cleaned = (line or ''):gsub('%s+','')
  88                         output = output .. "'"..cleaned.."',\n"
  89         end
  90         end
  91         output = output .. "]"
  92         return output
  93 end
  94
  95 local function convert_binary_dump_to_JS(infile, blockLen)
  96          local bindata = infile:read("*all")
  97          len = string.len(bindata)
  98
  99         if len % blockLen ~= 0 then
 100                 return oops(("Bad data, length (%d) should be a multiple of blocklen (%d)"):format(len, blockLen))
 101         end
 102
 103         local _,hex = bin.unpack(("H%d"):format(len),bindata)
 104
 105         -- Now that we've converted binary data into hex, we doubled the size.
 106         -- One byte, like 0xDE is now
 107         -- the characters 'D' and 'E' : one byte each.
 108         -- Thus:
 109         blockLen = blockLen * 2
 110
 111         local js,i = "[";
 112         for i = 1, string.len(hex),blockLen do
 113                 js = js .."'" ..string.sub(hex,i,i+blockLen -1).."',\n"
 114         end
 115         js = js .. "]"
 116         return js
 117 end
 118
 119 local function convert_ascii_dump_to_BIN(infile)
 120         local t = infile:read("*all")
 121         local cleaned
 122         local output = {};
 123         for line in string.gmatch(t, "[^\n]+") do
 124                 if string.byte(line) ~= string.byte("+") then
 125                         cleaned = (line or ''):gsub('%s+','')
 126                         for c in cleaned:gmatch('..') do
 127                                 output[#output+1] = string.char( tonumber(c,16) )
 128                         end
 129                 end
 130         end
 131         return output
 132 end
 133
 134
 135 ---
 136 -- Converts a .eml-file into a HTML/Javascript file.
 137 -- @param input the file to convert
 138 -- @param output the file to write to
 139 -- @return the name of the new file.
 140 local function convert_eml_to_html(input, output)
 141         input = input or 'dumpdata.eml'
 142         output = output or input .. 'html'
 143
 144         local infile = io.open(input, "r")
 145         if infile == nil then
 146                 return oops(string.format("Could not read file %s",tostring(input)))
 147         end
 148
 149         -- Read file, get JS
 150         local javascript = convert_ascii_dump_to_JS(infile)
 151         io.close(infile)
 152         return save_HTML(javascript, output )
 153 end
 154
 155 --- Converts a binary dump into HTML/Javascript file
 156 -- @param input the file containing the dump  (defaults to dumpdata.bin)
 157 -- @param output the file to write to
 158 -- @param blockLen, the length of each block. Defaults to 16 bytes
 159 local function convert_bin_to_html(input, output, blockLen)
 160         input = input or 'dumpdata.bin'
 161         blockLen = blockLen or 16
 162         output = output or input .. 'html'
 163
 164         local infile = io.open(input, "rb")
 165         if infile == nil then
 166                 return oops(string.format("Could not read file %s",tostring(input)))
 167         end
 168         -- Read file, get JS
 169         local javascript = convert_binary_dump_to_JS(infile, blockLen)
 170         io.close(infile)
 171
 172         return save_HTML(javascript, output )
 173 end
 174
 175 --- Converts a eml dump into a binary file
 176 -- @param input the file containing the eml-dump  (defaults to dumpdata.eml)
 177 -- @param output the file to write to  ( defaults to dumpdata.bin)
 178 local function convert_eml_to_bin(input, output)
 179         input = input or 'dumpdata.eml'
 180         output = output or 'dumpdata.bin'
 181
 182         local infile = io.open(input, "rb")
 183         if infile == nil then
 184                 return oops(string.format("Could not read file %s",tostring(input)))
 185         end
 186         -- Read file, get BIN
 187         local data = convert_ascii_dump_to_BIN(infile)
 188         io.close(infile)
 189
 190         return save_BIN(data, output )
 191 end
 192
 193
 194 return {
 195         convert_bin_to_html = convert_bin_to_html,
 196         convert_eml_to_html = convert_eml_to_html,
 197         convert_eml_to_bin = convert_eml_to_bin,
 198     SaveAsBinary = save_BIN,
 199         SaveAsText = save_TEXT,
 200     SaveAsBinary = save_BIN,
 201         SaveAsText = save_TEXT,
 202 }