114 lines
3.6 KiB
C
114 lines
3.6 KiB
C
/*
|
|
* 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 "nodemcu_esp_event.h"
|
|
#include "wifi_common.h"
|
|
#include "esp_wifi.h"
|
|
#include <string.h>
|
|
|
|
|
|
static void do_connect (const system_event_t *evt)
|
|
{
|
|
(void)evt;
|
|
esp_wifi_connect ();
|
|
}
|
|
|
|
|
|
static int wifi_sta_config (lua_State *L)
|
|
{
|
|
luaL_checkanytable (L, 1);
|
|
bool save = luaL_optbool (L, 2, DEFAULT_SAVE);
|
|
lua_settop (L, 1);
|
|
|
|
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.sta.ssid))
|
|
len = sizeof (cfg.sta.ssid);
|
|
strncpy (cfg.sta.ssid, str, len);
|
|
|
|
lua_getfield (L, 1, "pwd");
|
|
str = luaL_optlstring (L, -1, "", &len);
|
|
if (len > sizeof (cfg.sta.password))
|
|
len = sizeof (cfg.sta.password);
|
|
strncpy (cfg.sta.password, str, len);
|
|
|
|
lua_getfield (L, 1, "bssid");
|
|
cfg.sta.bssid_set = false;
|
|
if (lua_isstring (L, -1))
|
|
{
|
|
const char *bssid = luaL_checklstring (L, -1, &len);
|
|
const char *fmts[] = {
|
|
"%hhx%hhx%hhx%hhx%hhx%hhx",
|
|
"%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
|
|
"%hhx-%hhx-%hhx-%hhx-%hhx-%hhx",
|
|
"%hhx %hhx %hhx %hhx %hhx %hhx",
|
|
NULL
|
|
};
|
|
for (unsigned i = 0; fmts[i]; ++i)
|
|
{
|
|
if (sscanf (bssid, fmts[i],
|
|
&cfg.sta.bssid[0], &cfg.sta.bssid[1], &cfg.sta.bssid[2],
|
|
&cfg.sta.bssid[3], &cfg.sta.bssid[4], &cfg.sta.bssid[5]) == 6)
|
|
{
|
|
cfg.sta.bssid_set = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!cfg.sta.bssid_set)
|
|
return luaL_error (L, "invalid BSSID: %s", bssid);
|
|
}
|
|
|
|
SET_SAVE_MODE(save);
|
|
esp_err_t err = esp_wifi_set_config (WIFI_IF_STA, &cfg);
|
|
return (err == ESP_OK) ?
|
|
0 : luaL_error (L, "failed to set wifi config, code %d", err);
|
|
}
|
|
|
|
|
|
const LUA_REG_TYPE wifi_sta_map[] = {
|
|
{ LSTRKEY( "config" ), LFUNCVAL( wifi_sta_config ) },
|
|
|
|
{ LNILKEY, LNILVAL }
|
|
};
|
|
|
|
|
|
// Currently no auto-connect, so do that in response to events
|
|
NODEMCU_ESP_EVENT(SYSTEM_EVENT_STA_START, do_connect);
|
|
NODEMCU_ESP_EVENT(SYSTEM_EVENT_STA_DISCONNECTED, do_connect);
|