]>
git.zerfleddert.de Git - proxmark3-svn/blob - client/scripting.c
1 //-----------------------------------------------------------------------------
2 // Copyright (C) 2013 m h swende <martin at swende.se>
4 // This code is licensed to you under the terms of the GNU GPL, version 2 or,
5 // at your option, any later version. See the LICENSE.txt file for the text of
7 //-----------------------------------------------------------------------------
8 // Some lua scripting glue to proxmark core.
9 //-----------------------------------------------------------------------------
14 #include "proxmark3.h"
17 #include "scripting.h"
19 #include "nonce2key/nonce2key.h"
20 #include "../common/iso15693tools.h"
21 #include "../common/crc16.h"
22 #include "../common/crc64.h"
25 * The following params expected:
31 static int l_SendCommand(lua_State
*L
){
35 The SendCommand (native) expects the following structure:
38 uint64_t cmd; //8 bytes
39 uint64_t arg[3]; // 8*3 bytes = 24 bytes
41 uint8_t asBytes[USB_CMD_DATA_SIZE]; // 1 byte * 512 = 512 bytes (OR)
42 uint32_t asDwords[USB_CMD_DATA_SIZE/4]; // 4 byte * 128 = 512 bytes
46 ==> A 544 byte buffer will do.
50 const char *data
= luaL_checklstring(L
, 1, &size
);
51 if(size
!= sizeof(UsbCommand
))
53 printf("Got data size %d, expected %d" , (int) size
,(int) sizeof(UsbCommand
));
54 lua_pushstring(L
,"Wrong data size");
58 // UsbCommand c = (*data);
59 SendCommand((UsbCommand
* )data
);
60 return 0; // no return values
63 * @brief The following params expected:
69 static int l_WaitForResponseTimeout(lua_State
*L
){
72 size_t ms_timeout
= -1;
74 //Check number of arguments
75 int n
= lua_gettop(L
);
78 //signal error by returning Nil, errorstring
80 lua_pushstring(L
,"You need to supply at least command to wait for");
81 return 2; // two return values
86 cmd
= luaL_checkunsigned(L
,1);
90 //Did the user send a timeout ?
91 //Check if the current top of stack is an integer
92 ms_timeout
= luaL_checkunsigned(L
,2);
93 //printf("Timeout set to %dms\n" , (int) ms_timeout);
98 if(WaitForResponseTimeout(cmd
, &response
, ms_timeout
))
100 //Push it as a string
101 lua_pushlstring(L
,(const char *)&response
,sizeof(UsbCommand
));
103 return 1;// return 1 to signal one return value
107 return 1;// one return value
111 static int returnToLuaWithError(lua_State
*L
, const char* fmt
, ...)
116 vsnprintf(buffer
, sizeof(buffer
), fmt
,args
);
120 lua_pushstring(L
,buffer
);
124 static int l_nonce2key(lua_State
*L
){
127 const char *p_uid
= luaL_checklstring(L
, 1, &size
);
128 if(size
!= 4) return returnToLuaWithError(L
,"Wrong size of uid, got %d bytes, expected 4", (int) size
);
130 const char *p_nt
= luaL_checklstring(L
, 2, &size
);
131 if(size
!= 4) return returnToLuaWithError(L
,"Wrong size of nt, got %d bytes, expected 4", (int) size
);
133 const char *p_nr
= luaL_checklstring(L
, 3, &size
);
134 if(size
!= 4) return returnToLuaWithError(L
,"Wrong size of nr, got %d bytes, expected 4", (int) size
);
136 const char *p_par_info
= luaL_checklstring(L
, 4, &size
);
137 if(size
!= 8) return returnToLuaWithError(L
,"Wrong size of par_info, got %d bytes, expected 8", (int) size
);
139 const char *p_pks_info
= luaL_checklstring(L
, 5, &size
);
140 if(size
!= 8) return returnToLuaWithError(L
,"Wrong size of ks_info, got %d bytes, expected 8", (int) size
);
143 uint32_t uid
= bytes_to_num(( uint8_t *)p_uid
,4);
144 uint32_t nt
= bytes_to_num(( uint8_t *)p_nt
,4);
146 uint32_t nr
= bytes_to_num(( uint8_t*)p_nr
,4);
147 uint64_t par_info
= bytes_to_num(( uint8_t *)p_par_info
,8);
148 uint64_t ks_info
= bytes_to_num(( uint8_t *)p_pks_info
,8);
152 int retval
= nonce2key(uid
,nt
, nr
, par_info
,ks_info
, &key
);
154 //Push the retval on the stack
155 lua_pushinteger(L
,retval
);
157 //Push the key onto the stack
159 num_to_bytes(key
,sizeof(dest_key
),dest_key
);
161 //printf("Pushing to lua stack: %012"llx"\n",key);
162 lua_pushlstring(L
,(const char *) dest_key
,sizeof(dest_key
));
164 return 2; //Two return values
166 //static int l_PrintAndLog(lua_State *L){ return CmdHF14AMfDump(luaL_checkstring(L, 1));}
167 static int l_clearCommandBuffer(lua_State
*L
){
168 clearCommandBuffer();
172 * @brief l_foobar is a dummy function to test lua-integration with
176 static int l_foobar(lua_State
*L
)
178 //Check number of arguments
179 int n
= lua_gettop(L
);
180 printf("foobar called with %d arguments" , n
);
182 printf("Arguments discarded, stack now contains %d elements", lua_gettop(L
));
184 // todo: this is not used, where was it intended for?
185 // UsbCommand response = {CMD_MIFARE_READBL, {1337, 1338, 1339}};
187 printf("Now returning a uint64_t as a string");
188 uint64_t x
= 0xDEADBEEF;
189 uint8_t destination
[8];
190 num_to_bytes(x
,sizeof(x
),destination
);
191 lua_pushlstring(L
,(const char *)&x
,sizeof(x
));
192 lua_pushlstring(L
,(const char *)destination
,sizeof(destination
));
199 * @brief Utility to check if a key has been pressed by the user. This method does not block.
201 * @return boolean, true if kbhit, false otherwise.
203 static int l_ukbhit(lua_State
*L
)
205 lua_pushboolean(L
,ukbhit() ? true : false);
209 * @brief Calls the command line parser to deal with the command. This enables
210 * lua-scripts to do stuff like "core.console('hf mf mifare')"
214 static int l_CmdConsole(lua_State
*L
)
216 CommandReceived((char *)luaL_checkstring(L
, 1));
220 static int l_iso15693_crc(lua_State
*L
)
222 // uint16_t Iso15693Crc(uint8_t *v, int n);
224 const char *v
= luaL_checklstring(L
, 1, &size
);
225 uint16_t retval
= Iso15693Crc((uint8_t *) v
, size
);
226 lua_pushinteger(L
, (int) retval
);
231 Simple AES 128 cbc hook up to OpenSSL.
234 static int l_aes128decrypt(lua_State
*L
)
236 //Check number of arguments
239 const char *p_key
= luaL_checklstring(L
, 1, &size
);
240 if(size
!= 32) return returnToLuaWithError(L
,"Wrong size of key, got %d bytes, expected 32", (int) size
);
242 const char *p_encTxt
= luaL_checklstring(L
, 2, &size
);
244 unsigned char indata
[16] = {0x00};
245 unsigned char outdata
[16] = {0x00};
246 unsigned char aes_key
[16] = {0x00};
247 unsigned char iv
[16] = {0x00};
249 // convert key to bytearray and convert input to bytearray
250 for (i
= 0; i
< 32; i
+= 2) {
251 sscanf(&p_encTxt
[i
], "%02x", (unsigned int *)&indata
[i
/ 2]);
252 sscanf(&p_key
[i
], "%02x", (unsigned int *)&aes_key
[i
/ 2]);
257 aes_setkey_dec(&ctx
, aes_key
, 128);
258 aes_crypt_cbc(&ctx
,AES_DECRYPT
,sizeof(indata
), iv
, indata
,outdata
);
259 //Push decrypted array as a string
260 lua_pushlstring(L
,(const char *)&outdata
, sizeof(outdata
));
261 return 1;// return 1 to signal one return value
263 static int l_aes128encrypt(lua_State
*L
)
265 //Check number of arguments
268 const char *p_key
= luaL_checklstring(L
, 1, &size
);
269 if(size
!= 32) return returnToLuaWithError(L
,"Wrong size of key, got %d bytes, expected 32", (int) size
);
271 const char *p_txt
= luaL_checklstring(L
, 2, &size
);
273 unsigned char indata
[16] = {0x00};
274 unsigned char outdata
[16] = {0x00};
275 unsigned char aes_key
[16] = {0x00};
276 unsigned char iv
[16] = {0x00};
278 for (i
= 0; i
< 32; i
+= 2) {
279 sscanf(&p_txt
[i
], "%02x", (unsigned int *)&indata
[i
/ 2]);
280 sscanf(&p_key
[i
], "%02x", (unsigned int *)&aes_key
[i
/ 2]);
285 aes_setkey_enc(&ctx
, aes_key
, 128);
286 aes_crypt_cbc(&ctx
, AES_ENCRYPT
, sizeof(indata
), iv
, indata
, outdata
);
287 //Push encrypted array as a string
288 lua_pushlstring(L
,(const char *)&outdata
, sizeof(outdata
));
289 return 1;// return 1 to signal one return value
292 static int l_crc16(lua_State
*L
)
295 const char *p_str
= luaL_checklstring(L
, 1, &size
);
297 uint16_t retval
= crc16_ccitt( (uint8_t*) p_str
, size
);
298 lua_pushinteger(L
, (int) retval
);
302 static int l_crc64(lua_State
*L
)
306 unsigned char outdata
[8] = {0x00};
308 const char *p_str
= luaL_checklstring(L
, 1, &size
);
310 crc64( (uint8_t*) p_str
, size
, &crc
);
312 outdata
[0] = (uint8_t)(crc
>> 56) & 0xff;
313 outdata
[1] = (uint8_t)(crc
>> 48) & 0xff;
314 outdata
[2] = (uint8_t)(crc
>> 40) & 0xff;
315 outdata
[3] = (uint8_t)(crc
>> 32) & 0xff;
316 outdata
[4] = (uint8_t)(crc
>> 24) & 0xff;
317 outdata
[5] = (uint8_t)(crc
>> 16) & 0xff;
318 outdata
[6] = (uint8_t)(crc
>> 8) & 0xff;
319 outdata
[7] = crc
& 0xff;
320 lua_pushlstring(L
,(const char *)&outdata
, sizeof(outdata
));
325 * @brief Sets the lua path to include "./lualibs/?.lua", in order for a script to be
326 * able to do "require('foobar')" if foobar.lua is within lualibs folder.
327 * Taken from http://stackoverflow.com/questions/4125971/setting-the-global-lua-path-variable-from-c-c
332 int setLuaPath( lua_State
* L
, const char* path
)
334 lua_getglobal( L
, "package" );
335 lua_getfield( L
, -1, "path" ); // get field "path" from table at top of stack (-1)
336 const char* cur_path
= lua_tostring( L
, -1 ); // grab path string from top of stack
337 int requiredLength
= strlen(cur_path
)+ strlen(path
)+10; //A few bytes too many, whatever we can afford it
338 char * buf
= malloc(requiredLength
);
339 snprintf(buf
, requiredLength
, "%s;%s", cur_path
, path
);
340 lua_pop( L
, 1 ); // get rid of the string on the stack we just pushed on line 5
341 lua_pushstring( L
, buf
); // push the new one
342 lua_setfield( L
, -2, "path" ); // set the field "path" in table at -2 with value at top of stack
343 lua_pop( L
, 1 ); // get rid of package table from top of stack
345 return 0; // all done!
349 int set_pm3_libraries(lua_State
*L
)
352 static const luaL_Reg libs
[] = {
353 {"SendCommand", l_SendCommand
},
354 {"WaitForResponseTimeout", l_WaitForResponseTimeout
},
355 {"nonce2key", l_nonce2key
},
356 //{"PrintAndLog", l_PrintAndLog},
357 {"foobar", l_foobar
},
358 {"ukbhit", l_ukbhit
},
359 {"clearCommandBuffer", l_clearCommandBuffer
},
360 {"console", l_CmdConsole
},
361 {"iso15693_crc", l_iso15693_crc
},
362 {"aes128_decrypt", l_aes128decrypt
},
363 {"aes128_encrypt", l_aes128encrypt
},
369 lua_pushglobaltable(L
);
370 // Core library is in this table. Contains '
371 //this is 'pm3' table
374 //Put the function into the hash table.
375 for (int i
= 0; libs
[i
].name
; i
++) {
376 lua_pushcfunction(L
, libs
[i
].func
);
377 lua_setfield(L
, -2, libs
[i
].name
);//set the name, pop stack
380 lua_setfield(L
, -2, "core");
382 //-- remove the global environment table from the stack
385 //-- Last but not least, add to the LUA_PATH (package.path in lua)
386 // so we can load libraries from the ./lualib/ - directory
387 setLuaPath(L
,"./lualibs/?.lua");