wip

3 files changed, 175 insertions, 39 deletions

diff --git a/main/Kconfig b/main/Kconfig
index 6259a3f..11b89dc 100644
--- a/main/Kconfig
+++ b/main/Kconfig
@@ -27,4 +27,20 @@ config MAIN_FLASH_FD_BUF_SIZE
     int "FD buf size. Default: 32*4"
     default 128
 
+config MAIN_DEFAULT_WIFI_SSID
+    string "Default Wi-Fi SSID"
+    default "espressif"
+
+config MAIN_DEFAULT_WIFI_PASSWORD
+    string "Default Wi-Fi password"
+    default "12345678"
+
+config MAIN_DEFAULT_MQTT_HOST
+    string "Default MQTT host"
+    default "iot.eclipse.org"
+
+config MAIN_DEFAULT_MQTT_PORT
+    int "Default MQTT port"
+    default 1883
+
 endmenu
diff --git a/main/component.mk b/main/component.mk
index e69de29..500a63f 100644
--- a/main/component.mk
+++ b/main/component.mk
@@ -0,0 +1,9 @@
+
+ifdef WIFI_SSID
+CFLAGS += '-DWIFI_SSID="$(WIFI_SSID)"'
+CPPFLAGS += '-DWIFI_SSID="$(WIFI_SSID)"'
+endif
+ifdef WIFI_PASSWORD
+CFLAGS += '-DWIFI_PASSWORD="$(WIFI_PASSWORD)"'
+CPPFLAGS += '-DWIFI_PASSWORD="$(WIFI_PASSWORD)"'
+endif
diff --git a/main/main.cpp b/main/main.cpp
index 648c415..3acf080 100644
--- a/main/main.cpp
+++ b/main/main.cpp
@@ -17,8 +17,6 @@
 #include <errno.h>
 */
 
-#include "main-config.h"
-
 extern "C"
 uint32_t user_rf_cal_sector_set();
 
@@ -31,13 +29,22 @@ enum events {
 
 struct config {
     int loaded;
+    int ok_count;
+    char wifi_ssid[20];
+    char wifi_password[20];
+
     char mqtt_host[20];
     int mqtt_port;
 };
 
 struct _reent fs_reent;
+#define open(...) _spiffs_open_r(&fs_reent, __VA_ARGS__)
+#define close(...) _spiffs_close_r(&fs_reent, __VA_ARGS__)
+#define read(...) _spiffs_read_r(&fs_reent, __VA_ARGS__)
+#define write(...) _spiffs_write_r(&fs_reent, __VA_ARGS__)
+#define lseek(...) _spiffs_lseek_r(&fs_reent, __VA_ARGS__)
 
-xTaskHandle main_task_handle;
+xTaskHandle main_task_handle = 0;
 
 /******************************************************************************
  * FunctionName : user_rf_cal_sector_set
@@ -88,14 +95,49 @@ uint32_t user_rf_cal_sector_set()
     return rf_cal_sec;
 }
 
-static int config_load(spiffs *fs, struct config &config)
-{
-    int ret = 0;
+static int on_config_item(void *ctx, const char* key, const char* value) {
+    struct config *config = (struct config *)ctx;
+
+    if (strcmp(key, "wifi-ssid") == 0) {
+        strncpy(config->wifi_ssid, value, sizeof(config->wifi_ssid));
+    } else if (strcmp(key, "wifi-password") == 0) {
+        strncpy(config->wifi_password, value, sizeof(config->wifi_password));
+    } else if (strcmp(key, "mqtt-host") == 0) {
+        strncpy(config->mqtt_host, value, sizeof(config->mqtt_host));
+    } else if (strcmp(key, "mqtt-port") == 0) {
+        config->mqtt_port = atoi(value);
+    } else {
+        printf("Unknown key: %s\n", key);
+        return -1;
+    }
 
-    bzero(&config, sizeof(config));
+    config->ok_count++;
+
+    return 0;
+}
+
+typedef int (kv_event_handler)(void *ctx, const char *key, const char *value);
+
+static int config_read_on_line(kv_event_handler *event_handler, void *ctx, const char* line, size_t sz) {
+    printf("Handling line: #%d, %s\n", sz, line);
+    char *ptr = strchr(line, '=');
+    if (ptr == NULL) {
+        printf("Could not find '='\n");
+        return -1;
+    }
 
-    int fd = _spiffs_open_r(&fs_reent, "config", O_RDONLY, 0);
+    *ptr = '\0';
+    const char *key = line, *value = ++ptr;
 
+    return event_handler(ctx, key, value);
+}
+
+static int config_read(kv_event_handler *event_handler, void *ctx)
+{
+    int ret = 0;
+    int pos = 0;
+
+    int fd = open("config", O_RDONLY, 0);
     printf("fd=%d\n", fd);
     if (fd < 0) {
         ret = -1;
@@ -103,23 +145,43 @@ static int config_load(spiffs *fs, struct config &config)
     }
 
     while (1) {
-        char buf[6];
-        ret = _spiffs_read_r(&fs_reent, fd, buf, sizeof(buf));
-        printf("read: ret=%d\n", ret);
-        if (ret == 0) {
-            printf("EOF\n");
+        char buf[100];
+        ret = lseek(fd, pos, SEEK_SET);
+        if (ret == -1) {
+            printf("lseek failed: pos=%d, res=%d\n", pos, ret);
             break;
-        } else if (ret < 0) {
-            printf("FAIL\n");
+        }
+        if (pos > 200) {
+            break; // fail safe for now
+        }
+
+        ret = read(fd, buf, sizeof(buf));
+        printf("read: pos=%d, ret=%d\n", (int) pos, ret);
+        if (ret > 0) {
+            char *end = (char *)memchr(buf, 0, ret);
+            if (end == NULL) {
+                printf("too line line, could not find newline\n");
+                ret = -1;
+                goto fail;
+            } else {
+                *end = '\0';
+                size_t line_sz = end - buf;
+                printf("line_sz=%d\n", line_sz);
+                pos += line_sz + 1;
+
+                ret = config_read_on_line(event_handler, ctx, buf, line_sz);
+                if (ret) {
+                    goto fail;
+                }
+            }
+        } else {
             break;
         }
     }
 
-    config.loaded = 1;
-
 fail:
     if (fd != -1) {
-        _spiffs_close_r(&fs_reent, fd);
+        close(fd);
     }
 
     printf("%s: ret=%d\n", __FUNCTION__, ret);
@@ -127,34 +189,63 @@ fail:
     return ret;
 }
 
-static int config_write_string(int fd, const char *str) {
+static int config_write_kv(int fd, const char *key, const char *value) {
     int ret;
-    size_t sz = strlen(str);
-    printf("Writing %s, sz=%d\n", str, sz);
+    char c;
+    printf("Writing %s=%s\n", key, value);
+
+    size_t sz = strlen(key);
+    ret = write(fd, (char *)key, sz);
+    if (ret != sz) {
+        return ret;
+    }
+
+    c = '=';
+    ret = write(fd, &c, 1);
+    if (ret != 1) {
+        return ret;
+    }
 
-    ret = _spiffs_write_r(&fs_reent, fd, (char *)str, sz);
+    sz = strlen(value);
+    ret = write(fd, (char *)value, sz);
     if (ret != sz) {
         return ret;
     }
 
-    return _spiffs_write_r(&fs_reent, fd, (char *)"\0", 1) != 1;
+    c = 0;
+    return write(fd, &c, 1) != 1;
+}
+
+static int config_write_kv(int fd, const char *key, int value) {
+    char buf[100];
+    itoa(value, buf, 10);
+
+    return config_write_kv(fd, key, buf);
 }
 
 static int config_write(struct config &config) {
     int ret = 0;
 
-    int fd = _spiffs_open_r(&fs_reent, "config", O_WRONLY | O_CREAT, 0);
+    int fd = open("config", O_WRONLY | O_CREAT, 0);
 
     if (fd == -1) {
         ret = -1;
         goto fail;
     }
 
-    config_write_string(fd, config.mqtt_host);
+    ret = config_write_kv(fd, "mqtt-host", config.mqtt_host);
+    if (ret) {
+        goto fail;
+    }
+
+    ret = config_write_kv(fd, "mqtt-port", config.mqtt_port);
+    if (ret) {
+        goto fail;
+    }
 
 fail:
     if (fd != -1) {
-        _spiffs_close_r(&fs_reent, fd);
+        close(fd);
     }
 
     printf("%s: ret=%d\n", __FUNCTION__, ret);
@@ -170,8 +261,10 @@ void wifi_event_handler_cb(System_Event_t *event)
 
     if (event->event_id == EVENT_STAMODE_GOT_IP) {
         printf("STA GOT IP\n");
-        xTaskNotify(main_task_handle, EVENTS_GOT_IP, eSetBits);
-        printf("%s: xTaskGetCurrentTaskHandle=%p\n", __FUNCTION__, xTaskGetCurrentTaskHandle());
+        if (main_task_handle) {
+            xTaskNotify(main_task_handle, EVENTS_GOT_IP, eSetBits);
+            printf("%s: xTaskGetCurrentTaskHandle=%p\n", __FUNCTION__, xTaskGetCurrentTaskHandle());
+        }
     } else if (event->event_id == EVENT_STAMODE_CONNECTED) {
         os_printf("STA CONNECTED\n");
     } else if (event->event_id == EVENT_STAMODE_DISCONNECTED) {
@@ -199,7 +292,7 @@ static int fs_init()
     return esp_spiffs_init(&config);
 }
 
-static void set_station_mode() {
+static void set_station_mode(struct config &config) {
     printf("STATION_MODE\n");
     if (wifi_get_opmode_default() != NULL_MODE) {
         printf("Setting default station mode");
@@ -208,11 +301,11 @@ static void set_station_mode() {
 
     wifi_set_opmode_current(STATION_MODE);
 
-    struct station_config config;
-    bzero(&config, sizeof(struct station_config));
-    sprintf((char *)config.ssid, WIFI_SSID);
-    sprintf((char *)config.password, WIFI_PASSWORD);
-    wifi_station_set_config(&config);
+    struct station_config sc;
+    bzero(&sc, sizeof(struct station_config));
+    sprintf((char *)sc.ssid, config.wifi_ssid);
+    sprintf((char *)sc.password, config.wifi_password);
+    wifi_station_set_config(&sc);
 
     wifi_set_event_handler_cb(wifi_event_handler_cb); 
 
@@ -264,24 +357,42 @@ void user_init() {
     fs_init();
 
     struct config config;
-    ret = config_load(nullptr, config);
+    bzero(&config, sizeof(config));
+    ret = config_read(on_config_item, &config);
     if (ret) {
         printf("Loading file failed, errno=%d\n", fs_reent._errno);
     }
+    if (config.ok_count != 5) {
+        printf("Loading file failed, loaded bad number of items=%d\n", config.ok_count);
+    } else {
+        config.loaded = 1;
+    }
 
     if (!config.loaded) {
         printf("No config loaded, writing new\n");
-        strncpy(config.mqtt_host, "trygvis.io", sizeof(config.mqtt_host));
-        config.mqtt_port = 1883;
+#ifndef WIFI_SSID
+#define WIFI_SSID CONFIG_MAIN_DEFAULT_WIFI_SSID
+#endif
+#ifndef WIFI_PASSWORD
+#define WIFI_PASSWORD CONFIG_MAIN_DEFAULT_WIFI_PASSWORD
+#endif
+        strncpy(config.wifi_ssid, WIFI_SSID, sizeof(config.wifi_ssid));
+        strncpy(config.wifi_password, WIFI_PASSWORD, sizeof(config.wifi_password));
+        strncpy(config.mqtt_host, CONFIG_MAIN_DEFAULT_MQTT_HOST, sizeof(config.mqtt_host));
+        config.mqtt_port = CONFIG_MAIN_DEFAULT_MQTT_PORT;
         ret = config_write(config);
         if (ret) {
             printf("Writing file failed, errno=%d\n", fs_reent._errno);
         }
     }
 
-    return;
+    printf("Configuration:\n");
+    printf("  wifi-ssid=%s\n  wifi-password=%s\n", config.wifi_ssid, config.wifi_password);
+    printf("\n  mqtt-host=%s\n  mqtt-port=%d\n", config.mqtt_host, config.mqtt_port);
 
-    set_station_mode();
+    set_station_mode(config);
+
+    return;
 
     printf("%s: xTaskGetCurrentTaskHandle=%p\n", __FUNCTION__, xTaskGetCurrentTaskHandle());
     ret = xTaskCreate(main_task, THREAD_NAME, THREAD_STACK_WORDS, NULL, THREAD_PRIO, &main_task_handle);