From 2809f92eeb8b727e20167fe82e4cb9c3627d4870 Mon Sep 17 00:00:00 2001 From: HampusM Date: Sun, 8 May 2022 18:55:42 +0200 Subject: chore: move most files to minion folder --- minion/src/wifi_module.cpp | 312 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 312 insertions(+) create mode 100644 minion/src/wifi_module.cpp (limited to 'minion/src/wifi_module.cpp') diff --git a/minion/src/wifi_module.cpp b/minion/src/wifi_module.cpp new file mode 100644 index 0000000..2e25dc9 --- /dev/null +++ b/minion/src/wifi_module.cpp @@ -0,0 +1,312 @@ +#include "wifi_module.hpp" + +#include "util.hpp" + +#include +#include + +WiFiModule::WiFiModule(uint8_t receive_pin, uint8_t transmit_pin) noexcept + : _serial(receive_pin, transmit_pin) +{ +} + +void WiFiModule::begin(size_t baudrate) noexcept +{ + _serial.begin(baudrate); +} + +int WiFiModule::get_available() noexcept +{ + return _serial.available(); +} + +void WiFiModule::reset() noexcept +{ + _send_serial("AT+RST"); + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + const auto response_status = _read(2000U, response); + + Serial.println( + response_status == ResponseStatus::OK ? "Reset wifi module" + : "Failed to reset wifi module" + ); +} + +bool WiFiModule::connect(const char *ssid, const char *password) noexcept +{ + const char cmd[] = "AT+CWJAP"; + + auto command_length = strlen(cmd) + strlen(ssid) + strlen(password) + 6U + 1U; + + auto *command = util::malloc(command_length); + + if (command == nullptr) + { + Serial.println("Memory allocation failed for creating full connection command"); + return false; + } + + snprintf(command, command_length, "%s=\"%s\",\"%s\"", cmd, ssid, password); + + const auto send_success = _send_serial(command); + + if (!send_success) + { + Serial.println("Failed to send connection command"); + return false; + } + + free(command); + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + const auto response_status = _read(20000U, response); + + if (response_status != ResponseStatus::OK) + { + Serial.print("Connection command responded with status "); + Serial.println(response_status); + return false; + } + + return true; +} + +void WiFiModule::set_wifi_mode(WifiMode wifi_mode) noexcept +{ + const char cmd[] = "AT+CWMODE_CUR"; + + auto command_length = strlen(cmd) + 2U + 1U; + + auto *command = util::malloc(command_length); + + snprintf(command, command_length, "%s=%u", cmd, static_cast(wifi_mode)); + + _send_serial(command); + + free(command); + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + _read(1500U, response); +} + +void WiFiModule::set_multiple_connections_enabled(bool is_enabled) noexcept +{ + const char cmd[] = "AT+CIPMUX"; + + auto command_length = strlen(cmd) + 2U + 1U; + + auto *command = util::malloc(command_length); + + snprintf(command, command_length, "%s=%u", cmd, is_enabled ? 1U : 0U); + + _send_serial(command); + + free(command); + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + _read(1500U, response); + + // Serial.println(response); +} + +void WiFiModule::set_echo_enabled(bool is_enabled) noexcept +{ + const char cmd[] = "ATE"; + + auto command_length = strlen(cmd) + 1U + 1U; + + auto *command = util::malloc(command_length); + + snprintf(command, command_length, "%s%u", cmd, is_enabled ? 1U : 0U); + + _send_serial(command); + + free(command); + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + const auto response_status = _read(1500U, response); + + if (response_status == ResponseStatus::OK) + { + Serial.print("Turned "); + Serial.print(is_enabled ? "on" : "off"); + Serial.println(" AT commands echo"); + } + else + { + Serial.print("Failed to turn "); + Serial.print(is_enabled ? "on" : "off"); + Serial.println(" AT commands echo"); + } +} + +void WiFiModule::create_tcp_server(size_t port) noexcept +{ + const auto cmd = "AT+CIPSERVER"; + + auto command_length = strlen(cmd) + 7U + 1U; + + auto *command = util::malloc(command_length); + + snprintf(command, command_length, "%s=1,%u", cmd, port); + + _send_serial(command); + + free(command); + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + _read(1500U, response); + + // Serial.println(response); +} + +bool WiFiModule::test() noexcept +{ + const auto send_success = _send_serial("AT"); + + if (!send_success) + { + return false; + } + + char response[MAX_NETWORK_MODULE_RESPONSE_LENGTH] = ""; + + const auto response_status = _read(5000U, response); + + if (response_status == ResponseStatus::TIMEOUT) + { + return false; + } + + return strcmp(response, "") != 0; +} + +const char *WiFiModule::get_local_ip(char *local_ip_out) noexcept +{ + const auto command = "AT+CIFSR"; + + const auto send_success = _send_serial(command); + + if (!send_success) + { + return local_ip_out; + } + + auto *buf = util::malloc(strlen(local_ip_out)); + + if (buf == nullptr) + { + strcpy(local_ip_out, "Memory allocation failure"); + return local_ip_out; + } + + const auto response_status = _read(10000U, buf); + + if (response_status != ResponseStatus::OK) + { + free(buf); + + sprintf( + local_ip_out, + "Response status was not OK. Was %d", + static_cast(response_status) + ); + + return local_ip_out; + } + + const auto *local_ip_end = strstr(buf, "\r\n+CIFSR:STAMAC"); + + if (local_ip_end == nullptr) + { + free(buf); + strcpy(local_ip_out, "Response parsing error"); + return local_ip_out; + } + + const auto staip_title_length = 16U; // The length of '+CIFSR:STAIP,"' + + util::substr(buf + staip_title_length, local_ip_end, local_ip_out); + + free(buf); + + return local_ip_out; +} + +bool WiFiModule::_send_serial(const char *command) noexcept +{ + auto full_command_length = strlen(command) + 2U + 1U; + + auto *full_command = util::malloc(full_command_length); + + if (full_command == nullptr) + { + return false; + } + + snprintf(full_command, full_command_length, "%s\r\n", command); + + _serial.print(full_command); + + // Serial.print("Sent command: "); + // Serial.println(full_command); + + free(full_command); + + return true; +} + +ResponseStatus WiFiModule::_read(uint64_t timeout, char *response_out) noexcept +{ + const auto start_time = millis(); + + bool has_end = false; + auto status = ResponseStatus::TIMEOUT; + + // auto *response_out_head = response_out; + + while (!has_end && (start_time + timeout) > millis()) + { + while (_serial.available() != 0) + { + char character = _read_byte(); + + strncat(response_out, &character, 1U); + + // *response_out_head = character; + // response_out_head++; + + if (util::str_ends_with(response_out, "OK")) + { + status = ResponseStatus::OK; + has_end = true; + } + + if (util::str_ends_with(response_out, "ERROR")) + { + status = ResponseStatus::ERROR; + has_end = true; + } + + if (util::str_ends_with(response_out, "FAIL")) + { + status = ResponseStatus::FAIL; + has_end = true; + } + } + } + + return status; +} + +char WiFiModule::_read_byte() noexcept +{ + return static_cast(_serial.read()); +} -- cgit v1.2.3-18-g5258