Fork of the espurna firmware for `mhsw` switches
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

401 lines
11 KiB

8 years ago
7 years ago
8 years ago
8 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
7 years ago
8 years ago
7 years ago
8 years ago
7 years ago
8 years ago
7 years ago
8 years ago
  1. /*
  2. ESPurna
  3. Copyright (C) 2016-2017 by Xose Pérez <xose dot perez at gmail dot com>
  4. This program is free software: you can redistribute it and/or modify
  5. it under the terms of the GNU General Public License as published by
  6. the Free Software Foundation, either version 3 of the License, or
  7. (at your option) any later version.
  8. This program is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. GNU General Public License for more details.
  12. You should have received a copy of the GNU General Public License
  13. along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. */
  15. #include "config/all.h"
  16. #include <EEPROM.h>
  17. // -----------------------------------------------------------------------------
  18. // METHODS
  19. // -----------------------------------------------------------------------------
  20. unsigned long _loopDelay = 0;
  21. void hardwareSetup() {
  22. EEPROM.begin(EEPROM_SIZE);
  23. #if DEBUG_SERIAL_SUPPORT
  24. DEBUG_PORT.begin(SERIAL_BAUDRATE);
  25. #if DEBUG_ESP_WIFI
  26. DEBUG_PORT.setDebugOutput(true);
  27. #endif
  28. #elif defined(SERIAL_BAUDRATE)
  29. Serial.begin(SERIAL_BAUDRATE);
  30. #endif
  31. #if SPIFFS_SUPPORT
  32. SPIFFS.begin();
  33. #endif
  34. #if defined(ESPLIVE)
  35. //The ESPLive has an ADC MUX which needs to be configured.
  36. pinMode(16, OUTPUT);
  37. digitalWrite(16, HIGH); //Defualt CT input (pin B, solder jumper B)
  38. #endif
  39. }
  40. void hardwareLoop() {
  41. // Heartbeat
  42. static unsigned long last_uptime = 0;
  43. static bool on_connect = true;
  44. bool send = (last_uptime == 0);
  45. send = send || (millis() - last_uptime > HEARTBEAT_INTERVAL);
  46. if (mqttConnected()) {
  47. send = send || on_connect;
  48. } else {
  49. on_connect = true;
  50. }
  51. if (send) {
  52. last_uptime = millis();
  53. if (mqttConnected()) on_connect = false;
  54. heartbeat();
  55. }
  56. }
  57. // -----------------------------------------------------------------------------
  58. // BOOTING
  59. // -----------------------------------------------------------------------------
  60. unsigned int sectors(size_t size) {
  61. return (int) (size + SPI_FLASH_SEC_SIZE - 1) / SPI_FLASH_SEC_SIZE;
  62. }
  63. void welcome() {
  64. DEBUG_MSG_P(PSTR("\n\n"));
  65. DEBUG_MSG_P(PSTR("[INIT] %s %s\n"), (char *) APP_NAME, (char *) APP_VERSION);
  66. DEBUG_MSG_P(PSTR("[INIT] %s\n"), (char *) APP_AUTHOR);
  67. DEBUG_MSG_P(PSTR("[INIT] %s\n\n"), (char *) APP_WEBSITE);
  68. DEBUG_MSG_P(PSTR("[INIT] CPU chip ID: 0x%06X\n"), ESP.getChipId());
  69. DEBUG_MSG_P(PSTR("[INIT] CPU frequency: %d MHz\n"), ESP.getCpuFreqMHz());
  70. DEBUG_MSG_P(PSTR("[INIT] SDK version: %s\n"), ESP.getSdkVersion());
  71. DEBUG_MSG_P(PSTR("[INIT] Core version: %s\n"), getCoreVersion().c_str());
  72. DEBUG_MSG_P(PSTR("[INIT] Core revision: %s\n"), getCoreRevision().c_str());
  73. DEBUG_MSG_P(PSTR("\n"));
  74. // -------------------------------------------------------------------------
  75. FlashMode_t mode = ESP.getFlashChipMode();
  76. DEBUG_MSG_P(PSTR("[INIT] Flash chip ID: 0x%06X\n"), ESP.getFlashChipId());
  77. DEBUG_MSG_P(PSTR("[INIT] Flash speed: %u Hz\n"), ESP.getFlashChipSpeed());
  78. DEBUG_MSG_P(PSTR("[INIT] Flash mode: %s\n"), mode == FM_QIO ? "QIO" : mode == FM_QOUT ? "QOUT" : mode == FM_DIO ? "DIO" : mode == FM_DOUT ? "DOUT" : "UNKNOWN");
  79. DEBUG_MSG_P(PSTR("\n"));
  80. DEBUG_MSG_P(PSTR("[INIT] Flash sector size: %8u bytes\n"), SPI_FLASH_SEC_SIZE);
  81. DEBUG_MSG_P(PSTR("[INIT] Flash size (CHIP): %8u bytes\n"), ESP.getFlashChipRealSize());
  82. DEBUG_MSG_P(PSTR("[INIT] Flash size (SDK): %8u bytes / %4d sectors\n"), ESP.getFlashChipSize(), sectors(ESP.getFlashChipSize()));
  83. DEBUG_MSG_P(PSTR("[INIT] Firmware size: %8u bytes / %4d sectors\n"), ESP.getSketchSize(), sectors(ESP.getSketchSize()));
  84. DEBUG_MSG_P(PSTR("[INIT] OTA size: %8u bytes / %4d sectors\n"), ESP.getFreeSketchSpace(), sectors(ESP.getFreeSketchSpace()));
  85. #if SPIFFS_SUPPORT
  86. FSInfo fs_info;
  87. bool fs = SPIFFS.info(fs_info);
  88. if (fs) {
  89. DEBUG_MSG_P(PSTR("[INIT] SPIFFS size: %8u bytes / %4d sectors\n"), fs_info.totalBytes, sectors(fs_info.totalBytes));
  90. }
  91. #else
  92. DEBUG_MSG_P(PSTR("[INIT] SPIFFS size: %8u bytes / %4d sectors\n"), 0, 0);
  93. #endif
  94. DEBUG_MSG_P(PSTR("[INIT] EEPROM size: %8u bytes / %4d sectors\n"), settingsMaxSize(), sectors(settingsMaxSize()));
  95. DEBUG_MSG_P(PSTR("[INIT] Empty space: %8u bytes / 4 sectors\n"), 4 * SPI_FLASH_SEC_SIZE);
  96. DEBUG_MSG_P(PSTR("\n"));
  97. // -------------------------------------------------------------------------
  98. #if SPIFFS_SUPPORT
  99. if (fs) {
  100. DEBUG_MSG_P(PSTR("[INIT] SPIFFS total size: %8u bytes\n"), fs_info.totalBytes);
  101. DEBUG_MSG_P(PSTR("[INIT] used size: %8u bytes\n"), fs_info.usedBytes);
  102. DEBUG_MSG_P(PSTR("[INIT] block size: %8u bytes\n"), fs_info.blockSize);
  103. DEBUG_MSG_P(PSTR("[INIT] page size: %8u bytes\n"), fs_info.pageSize);
  104. DEBUG_MSG_P(PSTR("[INIT] max files: %8u\n"), fs_info.maxOpenFiles);
  105. DEBUG_MSG_P(PSTR("[INIT] max length: %8u\n"), fs_info.maxPathLength);
  106. } else {
  107. DEBUG_MSG_P(PSTR("[INIT] No SPIFFS partition\n"));
  108. }
  109. DEBUG_MSG_P(PSTR("\n"));
  110. #endif
  111. // -------------------------------------------------------------------------
  112. DEBUG_MSG_P(PSTR("[INIT] MANUFACTURER: %s\n"), MANUFACTURER);
  113. DEBUG_MSG_P(PSTR("[INIT] DEVICE: %s\n"), DEVICE);
  114. DEBUG_MSG_P(PSTR("[INIT] SUPPORT:"));
  115. #if ALEXA_SUPPORT
  116. DEBUG_MSG_P(PSTR(" ALEXA"));
  117. #endif
  118. #if ANALOG_SUPPORT
  119. DEBUG_MSG_P(PSTR(" ANALOG"));
  120. #endif
  121. #if COUNTER_SUPPORT
  122. DEBUG_MSG_P(PSTR(" COUNTER"));
  123. #endif
  124. #if DEBUG_SERIAL_SUPPORT
  125. DEBUG_MSG_P(PSTR(" DEBUG_SERIAL"));
  126. #endif
  127. #if DEBUG_TELNET_SUPPORT
  128. DEBUG_MSG_P(PSTR(" DEBUG_TELNET"));
  129. #endif
  130. #if DEBUG_UDP_SUPPORT
  131. DEBUG_MSG_P(PSTR(" DEBUG_UDP"));
  132. #endif
  133. #if DHT_SUPPORT
  134. DEBUG_MSG_P(PSTR(" DHT"));
  135. #endif
  136. #if DOMOTICZ_SUPPORT
  137. DEBUG_MSG_P(PSTR(" DOMOTICZ"));
  138. #endif
  139. #if DS18B20_SUPPORT
  140. DEBUG_MSG_P(PSTR(" DS18B20"));
  141. #endif
  142. #if HOMEASSISTANT_SUPPORT
  143. DEBUG_MSG_P(PSTR(" HOMEASSISTANT"));
  144. #endif
  145. #if I2C_SUPPORT
  146. DEBUG_MSG_P(PSTR(" I2C"));
  147. #endif
  148. #if INFLUXDB_SUPPORT
  149. DEBUG_MSG_P(PSTR(" INFLUXDB"));
  150. #endif
  151. #if LLMNR_SUPPORT
  152. DEBUG_MSG_P(PSTR(" LLMNR"));
  153. #endif
  154. #if MDNS_SUPPORT
  155. DEBUG_MSG_P(PSTR(" MDNS"));
  156. #endif
  157. #if NETBIOS_SUPPORT
  158. DEBUG_MSG_P(PSTR(" NETBIOS"));
  159. #endif
  160. #if NOFUSS_SUPPORT
  161. DEBUG_MSG_P(PSTR(" NOFUSS"));
  162. #endif
  163. #if NTP_SUPPORT
  164. DEBUG_MSG_P(PSTR(" NTP"));
  165. #endif
  166. #if RF_SUPPORT
  167. DEBUG_MSG_P(PSTR(" RF"));
  168. #endif
  169. #if SPIFFS_SUPPORT
  170. DEBUG_MSG_P(PSTR(" SPIFFS"));
  171. #endif
  172. #if TELNET_SUPPORT
  173. DEBUG_MSG_P(PSTR(" TELNET"));
  174. #endif
  175. #if TERMINAL_SUPPORT
  176. DEBUG_MSG_P(PSTR(" TERMINAL"));
  177. #endif
  178. #if WEB_SUPPORT
  179. DEBUG_MSG_P(PSTR(" WEB"));
  180. #endif
  181. DEBUG_MSG_P(PSTR("\n\n"));
  182. // -------------------------------------------------------------------------
  183. unsigned char reason = resetReason();
  184. if (reason > 0) {
  185. char buffer[32];
  186. strcpy_P(buffer, custom_reset_string[reason-1]);
  187. DEBUG_MSG_P(PSTR("[INIT] Last reset reason: %s\n"), buffer);
  188. } else {
  189. DEBUG_MSG_P(PSTR("[INIT] Last reset reason: %s\n"), (char *) ESP.getResetReason().c_str());
  190. }
  191. DEBUG_MSG_P(PSTR("[INIT] Free heap: %u bytes\n"), getFreeHeap());
  192. #if ADC_VCC_ENABLED
  193. DEBUG_MSG_P(PSTR("[INIT] Power: %d mV\n"), ESP.getVcc());
  194. #endif
  195. DEBUG_MSG_P(PSTR("[INIT] Power saving delay value: %lu ms\n"), _loopDelay);
  196. DEBUG_MSG_P(PSTR("\n"));
  197. }
  198. void setup() {
  199. // Init EEPROM, Serial and SPIFFS
  200. hardwareSetup();
  201. // Question system stability
  202. #if SYSTEM_CHECK_ENABLED
  203. systemCheck(false);
  204. #endif
  205. // Init persistance and terminal features
  206. settingsSetup();
  207. if (getSetting("hostname").length() == 0) {
  208. setSetting("hostname", getIdentifier());
  209. }
  210. // Cache loop delay value to speed things (recommended max 250ms)
  211. _loopDelay = atol(getSetting("loopDelay", LOOP_DELAY_TIME).c_str());
  212. // Show welcome message and system configuration
  213. welcome();
  214. // Basic modules, will always run
  215. wifiSetup();
  216. otaSetup();
  217. #if TELNET_SUPPORT
  218. telnetSetup();
  219. #endif
  220. // Do not run the next services if system is flagged stable
  221. #if SYSTEM_CHECK_ENABLED
  222. if (!systemCheck()) return;
  223. #endif
  224. // Init webserver required before any module that uses API
  225. #if WEB_SUPPORT
  226. webSetup();
  227. wsSetup();
  228. apiSetup();
  229. #endif
  230. #if LIGHT_PROVIDER != LIGHT_PROVIDER_NONE
  231. lightSetup();
  232. #endif
  233. relaySetup();
  234. buttonSetup();
  235. ledSetup();
  236. #if MQTT_SUPPORT
  237. mqttSetup();
  238. #endif
  239. #if MDNS_SUPPORT
  240. mdnsSetup();
  241. #endif
  242. #if LLMNR_SUPPORT
  243. llmnrSetup();
  244. #endif
  245. #if NETBIOS_SUPPORT
  246. netbiosSetup();
  247. #endif
  248. #if NTP_SUPPORT
  249. ntpSetup();
  250. #endif
  251. #if I2C_SUPPORT
  252. i2cSetup();
  253. #endif
  254. #ifdef ITEAD_SONOFF_RFBRIDGE
  255. rfbSetup();
  256. #endif
  257. #if POWER_PROVIDER != POWER_PROVIDER_NONE
  258. powerSetup();
  259. #endif
  260. #if ALEXA_SUPPORT
  261. alexaSetup();
  262. #endif
  263. #if NOFUSS_SUPPORT
  264. nofussSetup();
  265. #endif
  266. #if INFLUXDB_SUPPORT
  267. idbSetup();
  268. #endif
  269. #if DS18B20_SUPPORT
  270. dsSetup();
  271. #endif
  272. #if COUNTER_SUPPORT
  273. counterSetup();
  274. #endif
  275. #if RF_SUPPORT
  276. rfSetup();
  277. #endif
  278. #if IR_SUPPORT
  279. irSetup();
  280. #endif
  281. #if DOMOTICZ_SUPPORT
  282. domoticzSetup();
  283. #endif
  284. #if HOMEASSISTANT_SUPPORT
  285. haSetup();
  286. #endif
  287. sensorSetup();
  288. // Prepare configuration for version 2.0
  289. hwUpwardsCompatibility();
  290. saveSettings();
  291. }
  292. void loop() {
  293. hardwareLoop();
  294. settingsLoop();
  295. wifiLoop();
  296. otaLoop();
  297. #if SYSTEM_CHECK_ENABLED
  298. systemCheckLoop();
  299. // Do not run the next services if system is flagged stable
  300. if (!systemCheck()) return;
  301. #endif
  302. #if LIGHT_PROVIDER != LIGHT_PROVIDER_NONE
  303. lightLoop();
  304. #endif
  305. relayLoop();
  306. buttonLoop();
  307. ledLoop();
  308. #if MQTT_SUPPORT
  309. mqttLoop();
  310. #endif
  311. #ifdef ITEAD_SONOFF_RFBRIDGE
  312. rfbLoop();
  313. #endif
  314. #if POWER_PROVIDER != POWER_PROVIDER_NONE
  315. powerLoop();
  316. #endif
  317. #if NTP_SUPPORT
  318. ntpLoop();
  319. #endif
  320. #if ALEXA_SUPPORT
  321. alexaLoop();
  322. #endif
  323. #if NOFUSS_SUPPORT
  324. nofussLoop();
  325. #endif
  326. #if DS18B20_SUPPORT
  327. dsLoop();
  328. #endif
  329. #if COUNTER_SUPPORT
  330. counterLoop();
  331. #endif
  332. #if RF_SUPPORT
  333. rfLoop();
  334. #endif
  335. #if IR_SUPPORT
  336. irLoop();
  337. #endif
  338. sensorLoop();
  339. // Power saving delay
  340. delay(_loopDelay);
  341. }