Changed the name of writeraw into 14araw

[proxmark3-svn] / client / scripting.c

diff --git a/client/scripting.c b/client/scripting.c
index 331a8877569034a2937ecbb45ec6f9d35afe18cb..82d6ffc9c9171b624689bafd45b2a2b9fbdf6745 100644 (file)
--- a/client/scripting.c
+++ b/client/scripting.c
@@ -15,6 +15,7 @@
 #include "usb_cmd.h"
 #include "cmdmain.h"
 #include "scripting.h"
 #include "usb_cmd.h"
 #include "cmdmain.h"
 #include "scripting.h"

+#include "util.h"

 /**
  * The following params expected:
  *  UsbCommand c
 /**
  * The following params expected:
  *  UsbCommand c


@@ -50,13 +51,8 @@ static int l_SendCommand(lua_State *L){
     }
 
 //    UsbCommand c = (*data);
     }
 
 //    UsbCommand c = (*data);

-    SendCommand(data);
-    return 0;
-    //UsbCommand *c = (UsbCommand *)lua_touserdata(L, 1);
-    //luaL_argcheck(L, c != NULL, 1, "'UsbCommand' expected");
-
-    //SendCommand(c);
-    //return 0;
+    SendCommand((UsbCommand* )data);
+    return 0; // no return values

 }
 /**
  * @brief The following params expected:
 }
 /**
  * @brief The following params expected:


@@ -67,52 +63,90 @@ static int l_SendCommand(lua_State *L){
  */
 static int l_WaitForResponseTimeout(lua_State *L){
 
  */
 static int l_WaitForResponseTimeout(lua_State *L){
 

-    //pop cmd
-    uint32_t cmd = luaL_checkunsigned(L,1);
-    printf("in l_WaitForResponseTimeout, got cmd 0x%0x\n",(int) cmd);
-    //UsbCommand response;
-
-     //We allocate the usbcommand as userdata on the Lua-stack
-    size_t nbytes = sizeof(UsbCommand);
-
-    UsbCommand *response = (UsbCommand *)lua_newuserdata(L, nbytes);
+    uint32_t cmd = 0;
+    size_t ms_timeout = -1;

 
 

-    size_t ms_timeout = 2000;
-    //Did the user send a timeout ?
-    //Check if the current top of stack is an integer
-
-    if(lua_isnumber( L, 2))
+    //Check number of arguments
+    int n = lua_gettop(L);
+    if(n == 0)
+    {
+        //signal error by returning Nil, errorstring
+        lua_pushnil(L);
+        lua_pushstring(L,"You need to supply at least command to wait for");
+        return 2; // two return values
+    }
+    if(n >= 1)

     {
     {

-        printf("You sent a timout-value\n");
+        //pop cmd
+        cmd = luaL_checkunsigned(L,1);
+    }
+    if(n >= 2)
+    {
+        //Did the user send a timeout ?
+        //Check if the current top of stack is an integer

         ms_timeout = luaL_checkunsigned(L,2);
         ms_timeout = luaL_checkunsigned(L,2);

+        //printf("Timeout set to %dms\n" , (int) ms_timeout);

     }
     }

-    printf("Timeout set to %dms\n" , (int) ms_timeout);

 
 

-    if(WaitForResponseTimeout(cmd, response, ms_timeout))
-    {
-        //Return the UsbCommand as userdata
-        //the usbcommand is already on the stack
-        // return 1 to signal one return value
-        return 1;
-    }else
+    UsbCommand response;
+
+    if(WaitForResponseTimeout(cmd, &response, ms_timeout))

     {
     {

-        //Don't return the UsbCommand. Pop it.
-        lua_pop(L,-1);
+        //Push it as a string
+         lua_pushlstring(L,(const char *)&response,sizeof(UsbCommand));
+
+        return 1;// return 1 to signal one return value
+    }else{

         //Push a Nil instead
         lua_pushnil(L);
         //Push a Nil instead
         lua_pushnil(L);

-        return 1;
+        return 1;// one return value

     }
 }
     }
 }

-static int l_nonce2key(lua_State *L){ return CmdHF14AMfRdSc(luaL_checkstring(L, 1));}
-static int l_PrintAndLog(lua_State *L){ return CmdHF14AMfDump(luaL_checkstring(L, 1));}
+//static int l_nonce2key(lua_State *L){ return CmdHF14AMfRdSc(luaL_checkstring(L, 1));}
+//static int l_PrintAndLog(lua_State *L){ return CmdHF14AMfDump(luaL_checkstring(L, 1));}
+static int l_clearCommandBuffer(lua_State *L){
+    clearCommandBuffer();
+    return 0;
+}
+/**
+ * @brief l_foobar is a dummy function to test lua-integration with
+ * @param L
+ * @return
+ */
+static int l_foobar(lua_State *L)
+{
+    //Check number of arguments
+    int n = lua_gettop(L);
+    printf("foobar called with %d arguments" , n);
+    lua_settop(L, 0);
+    printf("Arguments discarded, stack now contains %d elements", lua_gettop(L));
+    UsbCommand response =  {CMD_MIFARE_READBL, {1337, 1338, 1339}};
+    printf("Now returning a UsbCommand as a string");
+    lua_pushlstring(L,(const char *)&response,sizeof(UsbCommand));
+    return 1;
+}
+
+/**
+ * @brief Utility to check if a key has been pressed by the user. This method does not block.
+ * @param L
+ * @return boolean, true if kbhit, false otherwise.
+ */
+static int l_ukbhit(lua_State *L)
+{
+    lua_pushboolean(L,ukbhit() ? true : false);
+    return 1;
+}

 
 int set_pm3_libraries(lua_State *L)
 {
     static const luaL_Reg libs[] = {
         {"SendCommand",                 l_SendCommand},
         {"WaitForResponseTimeout",      l_WaitForResponseTimeout},
 
 int set_pm3_libraries(lua_State *L)
 {
     static const luaL_Reg libs[] = {
         {"SendCommand",                 l_SendCommand},
         {"WaitForResponseTimeout",      l_WaitForResponseTimeout},

-        {"nonce2key",                   l_nonce2key},
-        {"PrintAndLog",                 l_PrintAndLog},
+        //{"nonce2key",                   l_nonce2key},
+        //{"PrintAndLog",                 l_PrintAndLog},
+        {"foobar",                      l_foobar},
+        {"ukbhit",                      l_ukbhit},
+        {"clearCommandBuffer",          l_clearCommandBuffer},

         {NULL, NULL}
     };
 
         {NULL, NULL}
     };