123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189 |
- /*
- * Copyright 2016 Dius Computing Pty Ltd. All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- *
- * - Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * - Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the
- * distribution.
- * - Neither the name of the copyright holders nor the names of
- * its contributors may be used to endorse or promote products derived
- * from this software without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
- * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
- * THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
- * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
- * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
- * OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- * @author Johny Mattsson <jmattsson@dius.com.au>
- */
- #include "module.h"
- #include "lauxlib.h"
- #include "lextra.h"
- #include "wifi_common.h"
- #include "ip_fmt.h"
- #include "nodemcu_esp_event.h"
- #include <string.h>
- // Note: these are documented in wifi.md, update there too if changed here!
- #define DEFAULT_AP_CHANNEL 11
- #define DEFAULT_AP_MAXCONNS 4
- #define DEFAULT_AP_BEACON 100
- // --- Event handling ----------------------------------------------------
- static void ap_staconn (lua_State *L, const system_event_t *evt);
- static void ap_stadisconn (lua_State *L, const system_event_t *evt);
- static void ap_probe_req (lua_State *L, const system_event_t *evt);
- static void empty_arg (lua_State *L, const system_event_t *evt) {}
- static const event_desc_t events[] =
- {
- { "start", SYSTEM_EVENT_AP_START, empty_arg },
- { "stop", SYSTEM_EVENT_AP_STOP, empty_arg },
- { "sta_connected", SYSTEM_EVENT_AP_STACONNECTED, ap_staconn },
- { "sta_disconnected", SYSTEM_EVENT_AP_STADISCONNECTED, ap_stadisconn },
- { "probe_req", SYSTEM_EVENT_AP_PROBEREQRECVED, ap_probe_req }
- };
- static int event_cb[ARRAY_LEN(events)];
- static void ap_staconn (lua_State *L, const system_event_t *evt)
- {
- char mac[MAC_STR_SZ];
- macstr (mac, evt->event_info.sta_connected.mac);
- lua_pushstring (L, mac);
- lua_setfield (L, -2, "mac");
- lua_pushinteger (L, evt->event_info.sta_connected.aid);
- lua_setfield (L, -2, "id");
- }
- static void ap_stadisconn (lua_State *L, const system_event_t *evt)
- {
- char mac[MAC_STR_SZ];
- macstr (mac, evt->event_info.sta_disconnected.mac);
- lua_pushstring (L, mac);
- lua_setfield (L, -2, "mac");
- lua_pushinteger (L, evt->event_info.sta_disconnected.aid);
- lua_setfield (L, -2, "id");
- }
- static void ap_probe_req (lua_State *L, const system_event_t *evt)
- {
- char str[MAC_STR_SZ];
- macstr (str, evt->event_info.ap_probereqrecved.mac);
- lua_pushstring (L, str);
- lua_setfield (L, -2, "from");
- lua_pushinteger (L, evt->event_info.ap_probereqrecved.rssi);
- lua_setfield (L, -2, "rssi");
- }
- static void on_event (const system_event_t *evt)
- {
- int idx = wifi_event_idx_by_id (events, ARRAY_LEN(events), evt->event_id);
- if (idx < 0 || event_cb[idx] == LUA_NOREF)
- return;
- lua_State *L = lua_getstate ();
- lua_rawgeti (L, LUA_REGISTRYINDEX, event_cb[idx]);
- lua_pushstring (L, events[idx].name);
- lua_createtable (L, 0, 5);
- events[idx].fill_cb_arg (L, evt);
- lua_call (L, 2, 0);
- }
- NODEMCU_ESP_EVENT(SYSTEM_EVENT_AP_START, on_event);
- NODEMCU_ESP_EVENT(SYSTEM_EVENT_AP_STOP, on_event);
- NODEMCU_ESP_EVENT(SYSTEM_EVENT_AP_STACONNECTED, on_event);
- NODEMCU_ESP_EVENT(SYSTEM_EVENT_AP_STADISCONNECTED, on_event);
- NODEMCU_ESP_EVENT(SYSTEM_EVENT_AP_PROBEREQRECVED, on_event);
- void wifi_ap_init (void)
- {
- for (unsigned i = 0; i < ARRAY_LEN(event_cb); ++i)
- event_cb[i] = LUA_NOREF;
- }
- // --- Lua API funcs -----------------------------------------------------
- static int wifi_ap_config (lua_State *L)
- {
- luaL_checkanytable (L, 1);
- bool save = luaL_optbool (L, 2, DEFAULT_SAVE);
- wifi_config_t cfg;
- memset (&cfg, 0, sizeof (cfg));
- lua_getfield (L, 1, "ssid");
- size_t len;
- const char *str = luaL_checklstring (L, -1, &len);
- if (len > sizeof (cfg.ap.ssid))
- len = sizeof (cfg.ap.ssid);
- strncpy ((char *)cfg.ap.ssid, str, len);
- cfg.ap.ssid_len = len;
- lua_getfield (L, 1, "pwd");
- str = luaL_optlstring (L, -1, "", &len);
- if (len > sizeof (cfg.ap.password))
- len = sizeof (cfg.ap.password);
- strncpy ((char *)cfg.ap.password, str, len);
- lua_getfield (L, 1, "auth");
- int authmode = luaL_optint (L, -1, WIFI_AUTH_WPA2_PSK);
- if (authmode < 0 || authmode >= WIFI_AUTH_MAX)
- return luaL_error (L, "unknown auth mode %d", authmode);
- cfg.ap.authmode = authmode;
- lua_getfield (L, 1, "channel");
- cfg.ap.channel = luaL_optint (L, -1, DEFAULT_AP_CHANNEL);
- lua_getfield (L, 1, "hidden");
- cfg.ap.ssid_hidden = luaL_optbool (L, -1, false);
- lua_getfield (L, 1, "max");
- cfg.ap.max_connection = luaL_optint (L, -1, DEFAULT_AP_MAXCONNS);
- lua_getfield (L, 1, "beacon");
- cfg.ap.beacon_interval = luaL_optint (L, -1, DEFAULT_AP_BEACON);
-
- SET_SAVE_MODE(save);
- esp_err_t err = esp_wifi_set_config (WIFI_IF_AP, &cfg);
- return (err == ESP_OK) ?
- 0 : luaL_error (L, "failed to set wifi config, code %d", err);
- }
- static int wifi_ap_getmac (lua_State *L)
- {
- return wifi_getmac(WIFI_IF_AP, L);
- }
- static int wifi_ap_on (lua_State *L)
- {
- return wifi_on (L, events, ARRAY_LEN(events), event_cb);
- }
- const LUA_REG_TYPE wifi_ap_map[] =
- {
- { LSTRKEY( "config" ), LFUNCVAL( wifi_ap_config ) },
- { LSTRKEY( "on" ), LFUNCVAL( wifi_ap_on ) },
- { LSTRKEY( "getmac" ), LFUNCVAL( wifi_ap_getmac ) },
- { LNILKEY, LNILVAL }
- };
|