simplified lf em command structure

[proxmark3-svn] / client / cmdscript.c

diff --git a/client/cmdscript.c b/client/cmdscript.c
index 8c1acf27104aa8443581c5b9b0245e4c2093841f..730f4e96ab26dd78e3392015af2ad5ef6d82a8af 100644 (file)
--- a/client/cmdscript.c
+++ b/client/cmdscript.c
@@ -25,7 +25,7 @@
 #include "cmdscript.h"
 #include "cmdhfmf.h"
 #include "pm3_binlib.h"
-
+#include "pm3_bitlib.h"
 #include <lua.h>
 #include <lualib.h>
 #include <lauxlib.h>
@@ -83,7 +83,7 @@ int CmdList(const char *Cmd)
     {
         while ((ep = readdir (dp)) != NULL)
         {
-            if(ep->d_name != NULL && str_ends_with(ep->d_name, ".lua"))
+            if(str_ends_with(ep->d_name, ".lua"))
                 PrintAndLog("%-16s %s", ep->d_name, "A script file");
         }
         (void) closedir (dp);
@@ -133,6 +133,8 @@ int CmdRun(const char *Cmd)
     //Add the 'bin' library
     set_bin_library(lua_state);
 
+       //Add the 'bit' library
+       set_bit_library(lua_state);
 
     char script_name[128] = {0};
     char arguments[256] = {0};