wps.c 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. // Module for WPS
  2. // by F.J. Exoo
  3. #include "module.h"
  4. #include "lauxlib.h"
  5. #include "platform.h"
  6. static int wps_callback_ref;
  7. // Lua: wps.disable()
  8. static int ICACHE_FLASH_ATTR wps_disable(lua_State* L)
  9. {
  10. wifi_wps_disable();
  11. return 0;
  12. }
  13. // Lua: wps.enable()
  14. static int ICACHE_FLASH_ATTR wps_enable(lua_State* L)
  15. {
  16. wifi_wps_enable(WPS_TYPE_PBC);
  17. return 0;
  18. }
  19. // WPS start callback function
  20. LOCAL void ICACHE_FLASH_ATTR user_wps_status_cb(int status)
  21. {
  22. lua_State *L = lua_getstate();
  23. if (wps_callback_ref != LUA_NOREF) {
  24. lua_rawgeti(L, LUA_REGISTRYINDEX, wps_callback_ref);
  25. lua_pushinteger(L, status);
  26. lua_call(L, 1, 0);
  27. }
  28. }
  29. // Lua: wps.start( function())
  30. static int ICACHE_FLASH_ATTR wps_start(lua_State* L)
  31. {
  32. // retrieve callback arg (optional)
  33. luaL_unref(L, LUA_REGISTRYINDEX, wps_callback_ref);
  34. wps_callback_ref = LUA_NOREF;
  35. if (lua_type(L, 1) == LUA_TFUNCTION || lua_type(L, 1) == LUA_TLIGHTFUNCTION)
  36. wps_callback_ref = luaL_ref(L, LUA_REGISTRYINDEX);
  37. else
  38. return luaL_error (L, "Argument not a function");
  39. wifi_set_wps_cb(user_wps_status_cb);
  40. wifi_wps_start();
  41. return 0;
  42. }
  43. // Module function map
  44. const LUA_REG_TYPE wps_map[] = {
  45. { LSTRKEY( "disable" ), LFUNCVAL( wps_disable ) },
  46. { LSTRKEY( "enable" ), LFUNCVAL( wps_enable ) },
  47. { LSTRKEY( "start" ), LFUNCVAL( wps_start ) },
  48. { LSTRKEY( "SUCCESS" ), LNUMVAL( WPS_CB_ST_SUCCESS ) },
  49. { LSTRKEY( "FAILED" ), LNUMVAL( WPS_CB_ST_FAILED ) },
  50. { LSTRKEY( "TIMEOUT" ), LNUMVAL( WPS_CB_ST_TIMEOUT ) },
  51. { LSTRKEY( "WEP" ), LNUMVAL( WPS_CB_ST_WEP ) },
  52. { LSTRKEY( "SCAN_ERR" ), LNUMVAL( 4 ) }, // WPS_CB_ST_SCAN_ERR
  53. { LNILKEY, LNILVAL }
  54. };
  55. int luaopen_wps( lua_State *L )
  56. {
  57. return 0;
  58. }
  59. NODEMCU_MODULE(WPS, "wps", wps_map, luaopen_wps);