chg: @piwi's code cleanup and some more.

[proxmark3-svn] / client / scripts / remagic.lua

diff --git a/client/scripts/remagic.lua b/client/scripts/remagic.lua
index 41a2cb83d160da30f6b246eec5dc462812e29956..806458999f2dd15dbc0a081b4e941d47e1676c81 100644 (file)
--- a/client/scripts/remagic.lua
+++ b/client/scripts/remagic.lua
@@ -59,6 +59,22 @@ local function cmdClassic()
        [4] = "hf 14a raw -c -a 5000",
        }
 end
+local function cmdRestoreST()
+       local arr = {}
+       for i = 0, 15 do
+               local blk = 3 + (4*i)
+               arr[i] = "hf mf csetbl "..blk.." FFFFFFFFFFFFFF078000FFFFFFFFFFFF"
+       end
+       return arr
+end 
+local function sendCmds( cmds )
+       for i = 0, #cmds do
+               if cmds[i]  then 
+                       print ( cmds[i]  )
+                       core.console( cmds[i] )
+               end
+       end
+end
 --- 
 -- The main entry point
 function main(args)
@@ -76,16 +92,10 @@ function main(args)
        core.clearCommandBuffer()
        
        if isUltralight then
-               cmds = cmdUltralight()
+               sendCmds ( cmdUltralight() )
        else
-               cmds = cmdClassic()
-       end
-       
-       for i = 0, #cmds do
-               if cmds[i]  then 
-                       print ( cmds[i] )
-                       core.console( cmds[i] )
-               end
+               sendCmds( cmdClassic() )
+               sendCmds( cmdRestoreST() )
        end
 end