diff --git a/.configs/TelnetTeraTerm.INI b/.configs/TelnetTeraTerm.INI new file mode 100644 index 0000000..e70423c Binary files /dev/null and b/.configs/TelnetTeraTerm.INI differ diff --git a/.configs/UartTeraTerm.INI b/.configs/UartTeraTerm.INI new file mode 100644 index 0000000..985dfde Binary files /dev/null and b/.configs/UartTeraTerm.INI differ diff --git a/.vscode/settings.json b/.vscode/settings.json index 59c830f..2f4a520 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -2,8 +2,6 @@ "C_Cpp.clang_format_fallbackStyle": "{ BasedOnStyle: LLVM, UseTab: Never, IndentWidth: 4, TabWidth: 4, AllowShortIfStatementsOnASingleLine: false, IndentCaseLabels: false, ColumnLimit: 0, AccessModifierOffset: -2, NamespaceIndentation: All, FixNamespaceComments: false }", "C_Cpp.intelliSenseEngine": "default", - "idf.adapterTargetName": "esp32", - "idf.customExtraPaths": "C:\\Tools\\ESP\\tools\\xtensa-esp-elf-gdb\\14.2_20240403\\xtensa-esp-elf-gdb\\bin;C:\\Tools\\ESP\\tools\\riscv32-esp-elf-gdb\\14.2_20240403\\riscv32-esp-elf-gdb\\bin;C:\\Tools\\ESP\\tools\\xtensa-esp-elf\\esp-13.2.0_20240530\\xtensa-esp-elf\\bin;C:\\Tools\\ESP\\tools\\riscv32-esp-elf\\esp-13.2.0_20240530\\riscv32-esp-elf\\bin;C:\\Tools\\ESP\\tools\\esp32ulp-elf\\2.38_20240113\\esp32ulp-elf\\bin;C:\\Tools\\ESP\\tools\\cmake\\3.24.0\\bin;C:\\Tools\\ESP\\tools\\openocd-esp32\\v0.12.0-esp32-20240318\\openocd-esp32\\bin;C:\\Tools\\ESP\\tools\\ninja\\1.11.1;C:\\Tools\\ESP\\tools\\idf-exe\\1.0.3;C:\\Tools\\ESP\\tools\\ccache\\4.8\\ccache-4.8-windows-x86_64;C:\\Tools\\ESP\\tools\\dfu-util\\0.11\\dfu-util-0.11-win64;C:\\Tools\\ESP\\tools\\esp-rom-elfs\\20240305", "idf.customExtraVars": { "OPENOCD_SCRIPTS": "C:\\Tools\\ESP\\tools\\openocd-esp32\\v0.12.0-esp32-20240318/openocd-esp32/share/openocd/scripts", "IDF_CCACHE_ENABLE": "1", @@ -14,8 +12,8 @@ "board/esp32c6-builtin.cfg" ], "idf.portWin": "COM9", - "idf.pythonBinPathWin": "C:\\Tools\\ESP\\python_env\\idf5.3_py3.11_env\\Scripts\\python.exe", "idf.toolsPathWin": "C:\\Tools\\ESP", "idf.flashType": "UART", - "C_Cpp.autoAddFileAssociations": false + "C_Cpp.autoAddFileAssociations": false, + "project.deviceIp": "172.16.17.56" } diff --git a/.vscode/tasks.json b/.vscode/tasks.json index 1dc7915..1807338 100644 --- a/.vscode/tasks.json +++ b/.vscode/tasks.json @@ -254,6 +254,18 @@ } } } + }, + { + "label": "Start TeraTerm Telnet", + "type": "shell", + "command": "C:/Program Files (x86)/teraterm5/ttermpro.exe", + "args": [ + "telnet://${config:project.deviceIp}:23", + "/W=\"Device-Platform\"", + "/T=1", + "/F=\"${workspaceFolder}/.configs/TelnetTeraTerm.INI\"" + ], + "problemMatcher": [] } ] } \ No newline at end of file diff --git a/components/project_bases/CMakeLists.txt b/components/project_bases/CMakeLists.txt index 397aead..e02288d 100644 --- a/components/project_bases/CMakeLists.txt +++ b/components/project_bases/CMakeLists.txt @@ -1,4 +1,4 @@ idf_component_register( - SRC_DIRS "." - INCLUDE_DIRS "." + SRC_DIRS "src" + INCLUDE_DIRS "include" ) diff --git a/components/project_bases/include/countDown.hpp b/components/project_bases/include/countDown.hpp new file mode 100644 index 0000000..f3f35fd --- /dev/null +++ b/components/project_bases/include/countDown.hpp @@ -0,0 +1,36 @@ +/** + * @file countDown.hpp + * @author Gustice + * @brief Classes for typical index manipulation + * + * @copyright Copyright (c) 2020 + */ +#pragma once + +class CountDown { + public: + CountDown(unsigned max) : Max(max) {} + + void Tick() { + _cnt++; + if (_cnt >= Max) { + _cnt = 0; + _pFlag = true; + } + } + + bool GetVolatileFlag() { + if (!_pFlag) + return false; + + // Reset _pFlag in any case, so it only be consumed once + // hence volatile + _pFlag = false; + return true; + } + + private: + const unsigned Max; + unsigned _cnt; + bool _pFlag; +}; diff --git a/components/project_bases/portBase.hpp b/components/project_bases/include/portBase.hpp similarity index 100% rename from components/project_bases/portBase.hpp rename to components/project_bases/include/portBase.hpp diff --git a/components/project_bases/include/rotatingIndex.hpp b/components/project_bases/include/rotatingIndex.hpp new file mode 100644 index 0000000..7fc0916 --- /dev/null +++ b/components/project_bases/include/rotatingIndex.hpp @@ -0,0 +1,58 @@ +/** + * @file rotatingIndex.hpp + * @author Gustice + * @brief Classes for typical index manipulation + * + * @copyright Copyright (c) 2020 + */ +#pragma once + +#include + +/** + * @brief Index with automatic revolution back to start position + * @details Useful for Buffer-Pointer, or similar tasks + */ +class RotatingIndex { + public: + /// @brief Value at witch the index rotates back to beginning + const uint16_t MaxIndex; + + /// @brief Constructor for rotating index + /// @brief output index will be set to 0 if incremented value reaches Maximum + /// @param max Maximum index for overflow-condition + RotatingIndex(uint16_t max) : MaxIndex(max) { _index = 0; } + + /// @brief Get the current index and inkrement index afterwards + /// @return uint16_t current index + uint16_t getIndexAndIncrement(void) { + auto retVal = _index; + _index++; + if (_index >= MaxIndex) { + _index = 0; + } + return retVal; + }; + + /// @brief Get the current index without incrementing the index + /// @return uint16_t current index + uint16_t getIndex(void) { return _index; } + + /// @brief Check if next get will lead to overflow condition + bool peekRevolution(void) { + if (MaxIndex == 0) + return true; + + if (_index >= MaxIndex - 1) { + return true; + } + return false; + } + + /// @brief Reset Index to 0 + void resetIndex(void) { _index = 0; } + + private: + /// Current Index + uint16_t _index; +}; diff --git a/components/project_bases/include/utils.hpp b/components/project_bases/include/utils.hpp new file mode 100644 index 0000000..8ed23d4 --- /dev/null +++ b/components/project_bases/include/utils.hpp @@ -0,0 +1,95 @@ +/** + * @file utils.hpp + * @author Gustice + * @brief Template + * + * @copyright Copyright (c) 2024 + */ +#pragma once + +#include +#include +#include +#include +#include + +/** + * @brief Simple exception + */ +class CoreException : public std::exception { + char const *Text; + + public: + /// @brief Constructor with simple pointer to text + CoreException(char const *text) : Text(text) { + } + + /// @brief error message getter + char const *what() const throw() override { return Text; } +}; + +/** + * @brief Common interface for character streams + */ +class CharStream { + public: + /// @brief Read method + /// @return string + virtual std::string read() = 0; + + /// @brief Write character + /// @param c character to write + virtual void write(char c) = 0; + + /// @brief Write string + /// @param str string to write + virtual void write(std::string str) = 0; +}; + +/** + * @brief Common interface + */ +class ByteStream { + /// @brief Read method + /// @return series of bytes + virtual std::vector read() = 0; + + /// @brief write single byte + /// @param b byte to write + virtual void write(uint8_t b) = 0; + + /// @brief write series of bytes + /// @param data series to write + virtual void write(std::vector data) = 0; +}; + +/** + * @brief Base for constructs that allow only one concurrently active instance + * + * @tparam T + * @note needs a `T * T::SingleActive::active{};` definition in source + */ +template +class SingleActive { + static bool active; + + public: + /// @brief Constructor, memorizes if already on instance was successfully build + SingleActive() { + if (active) { + throw "Instance already active"; + } + assert(active == false && "Constructor my only be run if no instance active"); + active = true; + } + + /// @brief Destructor, clears memory of existing instance + ~SingleActive() { + assert(active == true && "Destructor my only be run on active instance"); + active = false; + } + + // Disable move and assignment + SingleActive(SingleActive &other) = delete; + void operator=(const SingleActive &) = delete; +}; diff --git a/components/project_bases/src/utils.cpp b/components/project_bases/src/utils.cpp new file mode 100644 index 0000000..c31d1f5 --- /dev/null +++ b/components/project_bases/src/utils.cpp @@ -0,0 +1,9 @@ +/** + * @file utils.cpp + * @author Gustice + * @brief Implementation of utilities + * + * @copyright Copyright (c) 2024 + */ + +#include "utils.hpp" diff --git a/components/sockets/CMakeLists.txt b/components/sockets/CMakeLists.txt new file mode 100644 index 0000000..473ab9d --- /dev/null +++ b/components/sockets/CMakeLists.txt @@ -0,0 +1,5 @@ +idf_component_register( + SRC_DIRS "src" + INCLUDE_DIRS "include" + REQUIRES esp_netif lwip project_bases freeRtosApi +) diff --git a/components/sockets/include/etherCommon.hpp b/components/sockets/include/etherCommon.hpp new file mode 100644 index 0000000..db13aa7 --- /dev/null +++ b/components/sockets/include/etherCommon.hpp @@ -0,0 +1,196 @@ +/** + * @file etherCommon.hpp + * @author Gustice + * @brief Helper Functions for Ethernet usage + * + * @copyright Copyright (c) 2024 + */ +#pragma once + +#include "esp_log.h" +#include "esp_netif.h" +#include "lwip/err.h" +#include "lwip/sockets.h" +#include "lwip/sys.h" +#include +#include +#include +#include + +#if defined(CONFIG_LWIP_NETBUF_RECVINFO) +constexpr bool NETBUF_RECVINFO = true; +#else +constexpr bool NETBUF_RECVINFO = false; +#endif +#if defined(CONFIG_EXAMPLE_IPV4) +constexpr bool USE_IPV4 = true; +#else +constexpr bool USE_IPV4 = false; +#endif +#if defined(CONFIG_EXAMPLE_IPV6) +constexpr bool USE_IPV6 = true; +#else +constexpr bool USE_IPV6 = false; +#endif + +class Ip { + public: + Ip(uint32_t ipV4) { + _address.word = ipV4; + } + Ip(esp_ip4_addr_t ipV4) { + _address.word = ipV4.addr; + } + + Ip(uint8_t b0, uint8_t b1, uint8_t b2, uint8_t b3) { + _address.bytes[0] = b0; + _address.bytes[1] = b1; + _address.bytes[2] = b2; + _address.bytes[3] = b3; + } + + Ip(std::string ipStr) { + _address.word = ipaddr_addr(ipStr.c_str()); + } + + uint8_t operator[](int index) const { return _address.bytes[index]; }; + uint32_t asWord() const { return _address.word; } + std::string asString() const { return inet_ntoa(_address.word); } + + private: + union { + uint8_t bytes[4]; // IPv4 address + uint32_t word; + } _address; +}; + +class Mac { + public: + Mac(uint8_t val[NETIF_MAX_HWADDR_LEN]) { + std::memcpy(bytes.begin(), val, NETIF_MAX_HWADDR_LEN); + } + + uint8_t operator[](int index) const { + return bytes[index]; + }; + + std::string asString() const { + char buffer[64]; + sprintf(buffer, "%02x:%02x:%02x:%02x:%02x:%02x", bytes[0], bytes[1], + bytes[2], bytes[3], bytes[4], bytes[5]); + return {buffer}; + } + + const std::array &asArray() const { + return bytes; + } + + private: + std::array bytes; // IPv4 address +}; + +/** + * @brief IP-set class + */ +struct IpInfo { + /// @brief IP address + Ip address{0u}; + /// @brief IP subnet + Ip subnet{0u}; + /// @brief Gateway IP + Ip gateway{0u}; + /// @brief MAC-Address + Mac macAddress; + + /// @brief Get broadcast address from IP and subnet + /// @return + Ip getBradcastAddress() const { + auto sn = address.asWord() | ~subnet.asWord(); + return Ip(sn); + } +}; + +struct IpAdr { + public: + IpAdr(std::string ipStr, uint16_t port); + IpAdr(uint32_t ip, uint16_t port); + IpAdr(uint16_t port); + IpAdr(sockaddr_in ipV4data); + + static constexpr int IpProtocol = IPPROTO_IP; // IPPROTO_IPV6 for version 6 + static constexpr int ProtocolFamily = AF_INET; // AF_INET6 for version 6 + /// @brief StackFamily e.g. AF_INET or AF_INET6 + sa_family_t getProtocolFamily() const { return addrStore.ss_family; } + + std::string getIp() { + struct sockaddr_in *ipV4 = (struct sockaddr_in *)&addrStore; + return inet_ntoa(ipV4->sin_addr.s_addr); + } + uint16_t getPort() const { + struct sockaddr_in *ipV4 = (struct sockaddr_in *)&addrStore; + return ntohs(ipV4->sin_port); + }; + struct sockaddr *asSockAddrPtr() const { return (struct sockaddr *)&addrStore; } + + static constexpr int Size = sizeof(sockaddr_storage); + private: + struct sockaddr_storage addrStore; +}; + +struct IpAdrV6 { + public: + IpAdrV6(std::string ipStr, uint16_t port); + IpAdrV6(in6_addr ip, uint16_t port); + IpAdrV6(uint16_t port); + IpAdrV6(sockaddr_in6 ipV6data); + + static constexpr int IpProtocol = IPPROTO_IPV6; + static constexpr int ProtocolFamily = AF_INET6; + /// @brief StackFamily e.g. AF_INET or AF_INET6 + sa_family_t getProtocolFamily() const { return addrStore.ss_family; } + + std::string getIp() { + struct sockaddr_in6 *ipV6 = (struct sockaddr_in6 *)&addrStore; + return inet6_ntoa(ipV6->sin6_addr); + + } + uint16_t getPort() const { + struct sockaddr_in6 *ipV6 = (struct sockaddr_in6 *)&addrStore; + return ntohs(ipV6->sin6_port); + }; + struct sockaddr *asSockAddrPtr() const { return (struct sockaddr *)&addrStore; } + + static constexpr int Size = sizeof(sockaddr_storage); + private: + struct sockaddr_storage addrStore; +}; + +/** + * @brief Simplified abstraction for sockets + */ +class SocketBase { + public: + /// @brief Get if stream is active + /// @return true if active + virtual bool isActive() { + return active; + } + + IpInfo getIpInfo() { + auto *netIf = esp_netif_get_default_netif(); + esp_netif_ip_info_t ip; + uint8_t mac[NETIF_MAX_HWADDR_LEN]; + ESP_ERROR_CHECK(esp_netif_get_ip_info(netIf, &ip)); + ESP_ERROR_CHECK(esp_netif_get_mac(netIf, mac)); + + IpInfo ips{ + .address = Ip{ip.ip}, + .subnet = Ip(ip.netmask), + .gateway = Ip(ip.netmask), + .macAddress = Mac(mac)}; + return ips; + } + + protected: + bool active; +}; diff --git a/components/sockets/include/tcpSockets.hpp b/components/sockets/include/tcpSockets.hpp new file mode 100644 index 0000000..93444f9 --- /dev/null +++ b/components/sockets/include/tcpSockets.hpp @@ -0,0 +1,307 @@ +/** + * @file tcpSockets.hpp + * @author Gustice + * @brief Function wrapper for ESP-IDF socket implementation + * + * @copyright Copyright (c) 2024 + */ + +#pragma once + +#include "esp_log.h" +#include "etherCommon.hpp" +#include "utils.hpp" +#include +#include + +/** + * @brief TCP-Socket wrapper for ESP-IDF socket implementation + */ +class TcpSocket { + public: + /// @brief Constructor + /// @note Socket not active until call of attach method + /// @param addr Server/Client address + TcpSocket(const IpAdr &addr) + : Local(addr) { + _listenSockId = socket(addr.ProtocolFamily, SOCK_STREAM, addr.IpProtocol); + if (_listenSockId < 0) { + ESP_LOGE("TCP", "Unable to create socket: errno %d", errno); + throw CoreException("Unable to create socket"); + } + + if (addr.ProtocolFamily == IPPROTO_IP) { + int opt = 1; + setsockopt(_listenSockId, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)); + } + + if constexpr (USE_IPV4 && USE_IPV6) { + int opt = 1; + // Note that by default IPV6 binds to both protocols, it is must be disabled + // if both protocols used at the same time (used in CI) + setsockopt(_listenSockId, IPPROTO_IPV6, IPV6_V6ONLY, &opt, sizeof(opt)); + } + + ESP_LOGI("TCP", "Socket created"); + } + + /// @brief Destructor + ~TcpSocket() { + close(_listenSockId); + } + + /// @brief Attach to configured socket + /// @note Blocks until successful or failed + /// @return true if successful + bool bindTo() { + int err = bind(_listenSockId, Local.asSockAddrPtr(), sizeof(sockaddr_storage)); + if (err != 0) { + ESP_LOGE("TCP", "Socket unable to bind: errno %d", errno); + ESP_LOGE("TCP", "IPPROTO: %d", Local.ProtocolFamily); + throw CoreException("Socket unable to bind"); + } + ESP_LOGI("TCP", "Socket bound, port %d", Local.getPort()); + + err = listen(_listenSockId, 1); + if (err != 0) { + ESP_LOGE("TCP", "Error occurred during listen: errno %d", errno); + throw CoreException("Error occurred during listen"); + } + return true; + } + + /// @brief Get socket id for stream-operations + /// @return Id of socket + int getSocketId() { + return _listenSockId; + } + + /// @brief Address of this station (server perspective) + const IpAdr &Local; + + private: + int _listenSockId; +}; + +/** + * @brief TCP session + */ +class TcpSession : public SocketBase, public CharStream { + public: + /** + * @brief Configuration structure for Session + */ + struct Config { + int keepAlive; + int keepIdle; + int keepInterval; + int keepCount; + }; + + /// @brief Constructor + /// @note Constructor blocks + /// @param config Configuration for session + /// @param listener reference to socket + TcpSession(Config &config, TcpSocket &listener) : _other(IPADDR_ANY, 0) { + socklen_t addr_len = _other.Size; + _socket = accept(listener.getSocketId(), _other.asSockAddrPtr(), &addr_len); + if (_socket < 0) { + ESP_LOGE("TCP", "Unable to accept connection: errno %d", errno); + active = false; + throw CoreException("Unable to accept connection"); + } + + // Set tcp keepalive option + setsockopt(_socket, SOL_SOCKET, SO_KEEPALIVE, &config.keepAlive, sizeof(int)); + setsockopt(_socket, IPPROTO_TCP, TCP_KEEPIDLE, &config.keepIdle, sizeof(int)); + setsockopt(_socket, IPPROTO_TCP, TCP_KEEPINTVL, &config.keepInterval, sizeof(int)); + setsockopt(_socket, IPPROTO_TCP, TCP_KEEPCNT, &config.keepCount, sizeof(int)); + + std::string addrStr; + if (_other.ProtocolFamily == listener.Local.ProtocolFamily) { + if (_other.getProtocolFamily() == PF_INET) { + addrStr = _other.getIp(); + // } else if (_other.addrStore.ss_family == PF_INET6) { + // inet6_ntoa_r(((struct sockaddr_in6 *)&_other.addrStore)->sin6_addr, addrStr, sizeof(addrStr) - 1); + } else { + throw CoreException("Unknown case"); + } + } else { + ESP_LOGW("TCP", "Wrong IP-version request"); + active = false; + } + + ESP_LOGI("TCP", "Socket accepted ip address: %s", addrStr.c_str()); + + active = true; + } + + /// @brief Destructor + ~TcpSession() { + if (_socket < 0) + return; + + ESP_LOGD("TCP", "Tearing down socket"); + shutdown(_socket, 0); + } + + /// @brief Get IP of remote + /// @return + IpAdr getSource() { + return _other; + } + + /// @brief Read string + /// @note This call blocks until data is received + /// @return read string + std::string read() override { + auto data = readRaw(); + if (data.size() == 0) { + return ""; + } + + std::string ret((const char *)&data.data()[0]); + return ret; + } + + /// @brief Write character to socket + /// @param b character to write + void write(char c) override { + if (!isActive()) { + return; + } + + int written = send(_socket, &c, 1, 0); + if (written < 0) { + ESP_LOGE("TCP", "Error occurred during sending: errno %d", errno); + } + } + + /// @brief Write string to specific socket + /// @param message payload + /// @param dest Destination for message + void write(std::string message) override { + if (!isActive()) { + return; + } + + int written = send(_socket, message.c_str(), message.size(), 0); + if (written < 0) { + ESP_LOGE("TCP", "Error occurred during sending: errno %d", errno); + } + } + + IpAdr getOther() { + return _other; + } + + protected: + int _socket = 0; + IpAdr _other; + + std::vector readRaw() { + uint8_t data[128]; + int len = recv(_socket, data, sizeof(data) - 1, 0); + + if (len < 0) { + ESP_LOGE("TCP", "Error occurred during receiving: errno %d", errno); + active = false; + return {}; + } + + data[len] = 0; + if (len == 0) { + ESP_LOGW("TCP", "Connection closed"); + active = false; + return {}; + } + + ESP_LOGD("TCP", "Received %d bytes: %s", len, data); + std::vector vec(data, data + len + 1); + return vec; + } +}; + +constexpr TcpSession::Config DefaultTcpConfig{ + .keepAlive = 1, + .keepIdle = 5, + .keepInterval = 5, + .keepCount = 3, +}; + +constexpr uint16_t TelnetPort = 23; + +/** + * @brief TCP-Socket for Telnet-Applications + */ +class TelnetServer { + + /** + * @brief Specialization of TCP-Session for Telnet + */ + class TelnetSession : public TcpSession { + + public: + // Constructor + TelnetSession(Config &config, TcpSocket &listener) : TcpSession(config, listener) { + } + + /// @brief Read string + /// @note This call blocks until data is received + /// @return read string + std::string read() override { + auto data = readRaw(); + if (data.size() == 0) { + return ""; + } + + data.pop_back(); // 0-termination is not needed here + + int i = 0; + while (data[i] == 0xFF) { // while telnet command + i += 3; // skip this command + } + + if (i == data.size()) { // only configuration + send(_socket, data.data(), data.size(), 0); + return ""; + } + + std::string ret((const char *)&data[i]); + return ret; + } + }; + + public: + /// @brief Constructor + /// @param addrFamily IPV4 or IPV6 + TelnetServer(int addrFamily) + : Local(addrFamily == AF_INET ? IpAdr(INADDR_ANY, TelnetPort) : IpAdr(TelnetPort)), + Config(DefaultTcpConfig) { + } + + /** + * @brief Start server + * + * @param initCall Callback for shell init + * @param cyclicCall Callback for cyclic polling + */ + void run(std::function initCall, std::function cyclicCall) { + TcpSocket socket(Local); + socket.bindTo(); + while (true) { + // Run telnet-session as long socket is active else restart socket + ESP_LOGI("Telnet", "Socket waiting for connect"); + TelnetSession session(Config, socket); + + initCall(session); + while (session.isActive()) { + cyclicCall(session); + } + } + } + + private: + IpAdr Local; + TcpSession::Config Config; +}; diff --git a/components/sockets/include/udpSockets.hpp b/components/sockets/include/udpSockets.hpp new file mode 100644 index 0000000..3f6682c --- /dev/null +++ b/components/sockets/include/udpSockets.hpp @@ -0,0 +1,303 @@ +/** + * @file udpSockets.hpp + * @author Gustice + * @brief UDP-Socket wrapper for ESP-IDF socket implementation + * + * @copyright Copyright (c) 2024 + */ +#pragma once + +#include "Task.hpp" +#include "esp_log.h" +#include "etherCommon.hpp" +#include "utils.hpp" +#include +#include +#include + +constexpr bool UDP_WITH_TIMEOUT = false; + +/** + * @brief UDP-Socket wrapper for ESP-IDF socket implementation + */ +class UdpSocket : public ByteStream, public SocketBase { + public: + /// @brief Constructor for UDP-Socket-Server + /// @note Socket not active until call of attach method + /// @param addr Server/Client address + UdpSocket(const IpAdr &addr) + : Local(addr), _other(INADDR_ANY, 0) { + _listenSockId = socket(addr.ProtocolFamily, SOCK_DGRAM, addr.IpProtocol); + if (_listenSockId < 0) { + ESP_LOGE("UDP", "Unable to create socket: errno %d", errno); + throw CoreException("Unable to create socket"); + } + ESP_LOGD("UDP", "Socket created"); + + if constexpr (NETBUF_RECVINFO && !USE_IPV6) { + int enable = 1; + lwip_setsockopt(_listenSockId, IPPROTO_IP, IP_PKTINFO, &enable, sizeof(enable)); + } + + if constexpr (USE_IPV4 && USE_IPV6) { + if (addr.ProtocolFamily == AF_INET6) { + // Note that by default IPV6 binds to both protocols, it is must be disabled + // if both protocols used at the same time (used in CI) + int opt = 1; + setsockopt(_listenSockId, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)); + setsockopt(_listenSockId, IPPROTO_IPV6, IPV6_V6ONLY, &opt, sizeof(opt)); + } + } + + if constexpr (UDP_WITH_TIMEOUT) { // Set timeout if desired + struct timeval timeout { + .tv_sec = 10, + .tv_usec = 0, + }; + setsockopt(_listenSockId, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(timeout)); + } + } + + /// @brief Destructor + ~UdpSocket() { + active = false; + ESP_LOGD("UDP", "Shutting down socket and restarting..."); + shutdown(_listenSockId, 0); + close(_listenSockId); + } + + /// @brief Attach to configured socket + /// @note Blocks until successful or failed + /// @return true if successful + bool bindTo() { + active = false; + int err = bind(_listenSockId, Local.asSockAddrPtr(), sizeof(sockaddr_storage)); + if (err < 0) { + ESP_LOGE("UDP", "Socket unable to bind: errno %d", errno); + throw CoreException("Socket unable to bind"); + } + ESP_LOGI("UDP", "Socket bound, port %d", Local.getPort()); + + if constexpr (NETBUF_RECVINFO && !USE_IPV6) { + _iov.iov_base = _rxBuffer; + _iov.iov_len = sizeof(_rxBuffer); + _msg.msg_control = _cmsg_buf; + _msg.msg_controllen = sizeof(_cmsg_buf); + _msg.msg_flags = 0; + _msg.msg_iov = &_iov; + _msg.msg_iovlen = 1; + _msg.msg_name = _other.asSockAddrPtr(); + _msg.msg_namelen = _other.Size; + } + + active = true; + return active; + } + + /// @brief Read from socket + /// @note This call blocks until data is received + /// @return read bytes + std::vector read() override { + char buffer[256]; + int len = 0; + if constexpr (NETBUF_RECVINFO && !USE_IPV6) { + len = recvmsg(_listenSockId, &_msg, 0); + } else { + socklen_t socklen = _other.Size; + len = recvfrom(_listenSockId, buffer, sizeof(buffer) - 1, 0, _other.asSockAddrPtr(), &socklen); + } + if (len < 0) { + ESP_LOGE("UDP", "recvfrom failed: errno %d", errno); + return {}; + } + + std::string addrStr; + // Get the sender's ip address as string + if (Local.ProtocolFamily == PF_INET) { + addrStr = _other.getIp(); + + if constexpr (NETBUF_RECVINFO && !USE_IPV6) { + struct cmsghdr *cmsgtmp; + for (cmsgtmp = CMSG_FIRSTHDR(&_msg); cmsgtmp != NULL; cmsgtmp = CMSG_NXTHDR(&_msg, cmsgtmp)) { + if (cmsgtmp->cmsg_level == IPPROTO_IP && cmsgtmp->cmsg_type == IP_PKTINFO) { + struct in_pktinfo *pktinfo; + pktinfo = (struct in_pktinfo *)CMSG_DATA(cmsgtmp); + ESP_LOGI("UDP", "dest ip: %s", inet_ntoa(pktinfo->ipi_addr)); + } + } + } + } // else if (Local.ProtocolFamily == PF_INET6) { + // inet6_ntoa_r(((struct sockaddr_in6 *)&_other.addrStore)->sin6_addr, addrStr, sizeof(addrStr) - 1); + // } + + ESP_LOGD("UDP", "Received %d bytes from %s:", len, addrStr.c_str()); + std::vector ret(buffer, buffer + len); + return ret; + } + + /// @brief Write character to socket + /// @param b character to write + void write(uint8_t b) override {} + + /// @brief Write bytes to socket + /// @param data payload as series of bytes + void write(std::vector data) override { + write(&data[0], data.size()); + } + + /// @brief Write string to specific socket + /// @param message payload + /// @param dest Destination for message + void write(std::string &message, IpAdr *dest = nullptr) { + auto data = message.c_str(); + write((uint8_t *)data, message.size() + 1, dest); + } + + /// @brief Write string to specific socket or to last source + /// @param data payload + /// @param size size of data + /// @param addr Destination for message (if nullptr it will be send to last source) + void write(uint8_t *data, uint size, IpAdr *dest = nullptr) { + int err; + if (dest == nullptr) { + err = sendto(_listenSockId, data, size, 0, _other.asSockAddrPtr(), sizeof(sockaddr_storage)); + } else { + struct sockaddr_in *ipV4 = (struct sockaddr_in *)dest->asSockAddrPtr(); + // ESP_LOGI("UDP", "EthIP: %s", inet_ntoa(ipV4->sin_addr.s_addr)); + // ESP_LOGI("UDP", "PORT: %i", ntohs(ipV4->sin_port)); + err = sendto(_listenSockId, data, size, 0, dest->asSockAddrPtr(), sizeof(sockaddr_storage)); + } + + if (err < 0) { + ESP_LOGE("UDP", "Error occurred during sending: errno %d", errno); + active = false; + } + } + + /// @brief Standard destination address + const IpAdr &Local; + + /// @brief query if stream is still active + /// @return true if active + bool isValid() { + return _listenSockId >= 0; + } + + IpAdr getOther() { + return _other; + } + + private: + int _listenSockId; // os handle for socket + IpAdr _other; // Remote Station + char _rxBuffer[128]; + + // Only used if NETBUF_RECVINFO is active + iovec _iov; + msghdr _msg; + u8_t _cmsg_buf[CMSG_SPACE(sizeof(struct in_pktinfo))]; +}; + +/** + * @brief + * @details according to RFC-5424 + * https://datatracker.ietf.org/doc/html/rfc5424 + */ +class Syslog { + public: + static constexpr uint16_t SyslogPort = 514; + static constexpr std::string_view NilValue = "-"; + + enum class Severity { + Emergency = 0, + Alert = 1, + Critical = 2, + Error = 3, + Warning = 4, + Notice = 5, + Informational = 6, + Debug = 7, + }; + static constexpr uint8_t SeverityMask = 0x000F; + + enum class Facility { + kern = 0 << 3, + User = 1 << 3, + Mail = 2 << 3, + Daemon = 3 << 3, + Auth = 4 << 3, + Syslog = 5 << 3, + Lpr = 6 << 3, + News = 7 << 3, + Uucp = 8 << 3, + Cron = 9 << 3, + Authpriv = 10 << 3, + Ftp = 11 << 3, + Ntp = 12 << 3, + LogAudit = 13 << 3, + LogAlert = 14 << 3, + Clock = 15 << 3, + Local0 = 16 << 3, + Local1 = 17 << 3, + Local2 = 18 << 3, + Local3 = 19 << 3, + Local4 = 20 << 3, + Local5 = 21 << 3, + Local6 = 22 << 3, + Local7 = 23 << 3, + }; + + static void log(Severity severity, std::string message) { + if (_Socket == nullptr) { + return; + } + + char buffer[512]; + // V T H A P M D Msg + // alternatively with BOM flag UTF8 \xEF\xBB\xBF + auto pos = sprintf(buffer, "<%i>1 - %s %s - - - %s", + static_cast(_Facility) | static_cast(severity), + _HostName.c_str(), + _AppName.c_str(), + message.c_str()); + _Socket->write((uint8_t *)buffer, pos, &_Target); + } + + template + static void logv(Severity severity, const char *fmt, Params&&... args) { + if (_Socket == nullptr) { + return; + } + + char buffer[512]; + // V T H A P M D Msg + // alternatively with BOM flag UTF8 \xEF\xBB\xBF + auto pos = sprintf(buffer, "<%i>1 - %s %s - - - ", + static_cast(_Facility) | static_cast(severity), + _HostName.c_str(), + _AppName.c_str()); + pos += sprintf(&buffer[pos], fmt, std::forward (args)...); + + _Socket->write((uint8_t *)buffer, pos, &_Target); + } + + static void setupTask(IpAdr target) { + if (_Socket) { + return; + } + _Facility = Facility::Syslog; + _HostName = "ESP"; + _AppName = "App"; + _Target = IpAdr{target.getIp(), SyslogPort}; + + static fos::Task syslogTask(SyslogTask, "SysLog", 4096); + } + + private: + static void SyslogTask(); + static Facility _Facility; + static std::string _HostName; + static std::string _AppName; + static UdpSocket *_Socket; + static IpAdr _Target; +}; \ No newline at end of file diff --git a/components/sockets/src/etherCommon.cpp b/components/sockets/src/etherCommon.cpp new file mode 100644 index 0000000..70ebc70 --- /dev/null +++ b/components/sockets/src/etherCommon.cpp @@ -0,0 +1,68 @@ +/** + * @file etherCommon.cpp + * @author Gustice + * @brief Helper Functions for Ethernet usage + * + * @copyright Copyright (c) 2024 + */ + +#include "etherCommon.hpp" +#include + +IpAdr::IpAdr(std::string ipStr, uint16_t port) + : IpAdr(ipaddr_addr(ipStr.c_str()), port) { +} + +IpAdr::IpAdr(uint32_t ip, uint16_t port) { + struct sockaddr_in *ipV4 = (struct sockaddr_in *)&addrStore; + ipV4->sin_len = sizeof(sockaddr_in); + ipV4->sin_addr.s_addr = ip; + ipV4->sin_family = AF_INET; + ipV4->sin_port = htons(port); +} + +IpAdr::IpAdr(uint16_t port) { + struct sockaddr_in *ipV4 = (struct sockaddr_in *)&addrStore; + ipV4->sin_len = sizeof(sockaddr_in); + ipV4->sin_addr.s_addr = INADDR_ANY; + ipV4->sin_family = AF_INET; + ipV4->sin_port = htons(port); +} + +IpAdr::IpAdr(sockaddr_in ipV4data) { + struct sockaddr_storage *store = (struct sockaddr_storage *)&ipV4data; + assert(ipV4data.sin_family == AF_INET); + addrStore = *store; +} + + +IpAdrV6::IpAdrV6(std::string ipStr, uint16_t port) { + struct sockaddr_in6 *ipV6 = (struct sockaddr_in6 *)&addrStore; + ipV6->sin6_len = sizeof(sockaddr_in6); + inet6_aton(ipStr.c_str(), &ipV6->sin6_addr); + ipV6->sin6_family = AF_INET6; + ipV6->sin6_port = htons(port); +} + +IpAdrV6::IpAdrV6(in6_addr ip, uint16_t port) { + struct sockaddr_in6 *ipV6 = (struct sockaddr_in6 *)&addrStore; + ipV6->sin6_len = sizeof(sockaddr_in6); + memcpy(&ip, &ipV6->sin6_addr, sizeof(in6_addr)); + ipV6->sin6_family = AF_INET6; + ipV6->sin6_port = htons(port); +} + +IpAdrV6::IpAdrV6(uint16_t port) { + struct sockaddr_in6 *ipV6 = (struct sockaddr_in6 *)&addrStore; + ipV6->sin6_len = sizeof(sockaddr_in6); + // ip6_addr_set_any() + bzero(&ipV6->sin6_addr.un, sizeof(ipV6->sin6_addr.un)); + ipV6->sin6_family = AF_INET6; + ipV6->sin6_port = htons(port); +} + +IpAdrV6::IpAdrV6(sockaddr_in6 ipV6data) { + struct sockaddr_storage *store = (struct sockaddr_storage *)&ipV6data; + assert(ipV6data.sin6_family == AF_INET); + addrStore = *store; +} diff --git a/components/sockets/src/tcpSockets.cpp b/components/sockets/src/tcpSockets.cpp new file mode 100644 index 0000000..09bbb01 --- /dev/null +++ b/components/sockets/src/tcpSockets.cpp @@ -0,0 +1,9 @@ +/** + * @file tcpSockets.cpp + * @author Gustice + * @brief Function frames for handling sockets + * + * @copyright Copyright (c) 2024 + */ + +#include "tcpSockets.hpp" \ No newline at end of file diff --git a/components/sockets/src/udpSocekts.cpp b/components/sockets/src/udpSocekts.cpp new file mode 100644 index 0000000..e50eee9 --- /dev/null +++ b/components/sockets/src/udpSocekts.cpp @@ -0,0 +1,35 @@ +/** + * @file udpSockets.cpp + * @author Gustice + * @brief Function frames for handling sockets + * + * @copyright Copyright (c) 2024 + */ + +#include "udpSockets.hpp" + +UdpSocket * Syslog::_Socket; +Syslog::Facility Syslog::_Facility; +std::string Syslog::_HostName; +std::string Syslog::_AppName; +IpAdr Syslog::_Target(INADDR_ANY, 0); + +void Syslog::SyslogTask() { + ESP_LOGI("SysLog", "Starting Syslog Task"); + try { + while (true) { + IpAdr ip(INADDR_ANY, SyslogPort); + UdpSocket socket(ip); + _Socket = &socket; + socket.bindTo(); + + while (socket.isActive()) { + (void)socket.read(); + // ignore all incomming data + } + } + _Socket = nullptr; + } catch (const std::exception &e) { + ESP_LOGE("SysLog", "%s", e.what()); + } +} diff --git a/components/tftpShell/CMakeLists.txt b/components/tftpShell/CMakeLists.txt new file mode 100644 index 0000000..77b2c2a --- /dev/null +++ b/components/tftpShell/CMakeLists.txt @@ -0,0 +1,4 @@ +idf_component_register( + SRC_DIRS "extern" + INCLUDE_DIRS "include" "extern" +) diff --git a/components/tftpShell/extern/embedded_cli.c b/components/tftpShell/extern/embedded_cli.c new file mode 100644 index 0000000..cffbf53 --- /dev/null +++ b/components/tftpShell/extern/embedded_cli.c @@ -0,0 +1,1156 @@ +#include +#include + +#include "embedded_cli.h" + +#define CLI_TOKEN_NPOS 0xffff + +#define UNUSED(x) (void)x + +#define PREPARE_IMPL(t) EmbeddedCliImpl *impl = (EmbeddedCliImpl *)t->_impl + +#define IS_FLAG_SET(flags, flag) (((flags) & (flag)) != 0) + +#define SET_FLAG(flags, flag) ((flags) |= (flag)) + +#define UNSET_U8FLAG(flags, flag) ((flags) &= (uint8_t) ~(flag)) + +/** + * Marks binding as candidate for autocompletion + * This flag is updated each time getAutocompletedCommand is called + */ +#define BINDING_FLAG_AUTOCOMPLETE 1u + +/** + * Indicates that rx buffer overflow happened. In such case last command + * that wasn't finished (no \r or \n were received) will be discarded + */ +#define CLI_FLAG_OVERFLOW 0x01u + +/** + * Indicates that initialization is completed. Initialization is completed in + * first call to process and needed, for example, to print invitation message. + */ +#define CLI_FLAG_INIT_COMPLETE 0x02u + +/** + * Indicates that CLI structure and internal structures were allocated with + * malloc and should bre freed + */ +#define CLI_FLAG_ALLOCATED 0x04u + +/** + * Indicates that CLI structure and internal structures were allocated with + * malloc and should bre freed + */ +#define CLI_FLAG_ESCAPE_MODE 0x08u + +/** + * Indicates that CLI in mode when it will print directly to output without + * clear of current command and printing it back + */ +#define CLI_FLAG_DIRECT_PRINT 0x10u + +/** + * Indicates that live autocompletion is enabled + */ +#define CLI_FLAG_AUTOCOMPLETE_ENABLED 0x20u + +typedef struct EmbeddedCliImpl EmbeddedCliImpl; +typedef struct AutocompletedCommand AutocompletedCommand; +typedef struct FifoBuf FifoBuf; +typedef struct CliHistory CliHistory; + +struct FifoBuf { + char *buf; + /** + * Position of first element in buffer. From this position elements are taken + */ + uint16_t front; + /** + * Position after last element. At this position new elements are inserted + */ + uint16_t back; + /** + * Size of buffer + */ + uint16_t size; +}; + +struct CliHistory { + /** + * Items in buffer are separated by null-chars + */ + char *buf; + + /** + * Total size of buffer + */ + uint16_t bufferSize; + + /** + * Index of currently selected element. This allows to navigate history + * After command is sent, current element is reset to 0 (no element) + */ + uint16_t current; + + /** + * Number of items in buffer + * Items are counted from top to bottom (and are 1 based). + * So the most recent item is 1 and the oldest is itemCount. + */ + uint16_t itemsCount; +}; + +struct EmbeddedCliImpl { + /** + * Invitation string. Is printed at the beginning of each line with user + * input + */ + const char *invitation; + + CliHistory history; + + /** + * Buffer for storing received chars. + * Chars are stored in FIFO mode. + */ + FifoBuf rxBuffer; + + /** + * Buffer for current command + */ + char *cmdBuffer; + + /** + * Size of current command + */ + uint16_t cmdSize; + + /** + * Total size of command buffer + */ + uint16_t cmdMaxSize; + + CliCommandBinding *bindings; + + /** + * Flags for each binding. Sizes are the same as for bindings array + */ + uint8_t *bindingsFlags; + + uint16_t bindingsCount; + + uint16_t maxBindingsCount; + + /** + * Total length of input line. This doesn't include invitation but + * includes current command and its live autocompletion + */ + uint16_t inputLineLength; + + /** + * Stores last character that was processed. + */ + char lastChar; + + /** + * Flags are defined as CLI_FLAG_* + */ + uint8_t flags; +}; + +struct AutocompletedCommand { + /** + * Name of autocompleted command (or first candidate for autocompletion if + * there are multiple candidates). + * NULL if autocomplete not possible. + */ + const char *firstCandidate; + + /** + * Number of characters that can be completed safely. For example, if there + * are two possible commands "get-led" and "get-adc", then for prefix "g" + * autocompletedLen will be 4. If there are only one candidate, this number + * is always equal to length of the command. + */ + uint16_t autocompletedLen; + + /** + * Total number of candidates for autocompletion + */ + uint16_t candidateCount; +}; + +static EmbeddedCliConfig defaultConfig; + +/** + * Number of commands that cli adds. Commands: + * - help + */ +static const uint16_t cliInternalBindingCount = 1; + +static const char *lineBreak = "\r\n"; + +/** + * Navigate through command history back and forth. If navigateUp is true, + * navigate to older commands, otherwise navigate to newer. + * When history end is reached, nothing happens. + * @param cli + * @param navigateUp + */ +static void navigateHistory(EmbeddedCli *cli, bool navigateUp); + +/** + * Process escaped character. After receiving ESC+[ sequence, all chars up to + * ending character are sent to this function + * @param cli + * @param c + */ +static void onEscapedInput(EmbeddedCli *cli, char c); + +/** + * Process input character. Character is valid displayable char and should be + * added to current command string and displayed to client. + * @param cli + * @param c + */ +static void onCharInput(EmbeddedCli *cli, char c); + +/** + * Process control character (like \r or \n) possibly altering state of current + * command or executing onCommand callback. + * @param cli + * @param c + */ +static void onControlInput(EmbeddedCli *cli, char c); + +/** + * Parse command in buffer and execute callback + * @param cli + */ +static void parseCommand(EmbeddedCli *cli); + +/** + * Setup bindings for internal commands, like help + * @param cli + */ +static void initInternalBindings(EmbeddedCli *cli); + +/** + * Show help for given tokens (or default help if no tokens) + * @param cli + * @param tokens + * @param context - not used + */ +static void onHelp(EmbeddedCli *cli, char *tokens, void *context); + +/** + * Show error about unknown command + * @param cli + * @param name + */ +static void onUnknownCommand(EmbeddedCli *cli, const char *name); + +/** + * Return autocompleted command for given prefix. + * Prefix is compared to all known command bindings and autocompleted result + * is returned + * @param cli + * @param prefix + * @return + */ +static AutocompletedCommand getAutocompletedCommand(EmbeddedCli *cli, const char *prefix); + +/** + * Prints autocompletion result while keeping current command unchanged + * Prints only if autocompletion is present and only one candidate exists. + * @param cli + */ +static void printLiveAutocompletion(EmbeddedCli *cli); + +/** + * Handles autocomplete request. If autocomplete possible - fills current + * command with autocompleted command. When multiple commands satisfy entered + * prefix, they are printed to output. + * @param cli + */ +static void onAutocompleteRequest(EmbeddedCli *cli); + +/** + * Removes all input from current line (replaces it with whitespaces) + * And places cursor at the beginning of the line + * @param cli + */ +static void clearCurrentLine(EmbeddedCli *cli); + +/** + * Write given string to cli output + * @param cli + * @param str + */ +static void writeToOutput(EmbeddedCli *cli, const char *str); + +/** + * Returns true if provided char is a supported control char: + * \r, \n, \b or 0x7F (treated as \b) + * @param c + * @return + */ +static bool isControlChar(char c); + +/** + * Returns true if provided char is a valid displayable character: + * a-z, A-Z, 0-9, whitespace, punctuation, etc. + * Currently only ASCII is supported + * @param c + * @return + */ +static bool isDisplayableChar(char c); + +/** + * How many elements are currently available in buffer + * @param buffer + * @return number of elements + */ +static uint16_t fifoBufAvailable(FifoBuf *buffer); + +/** + * Return first character from buffer and remove it from buffer + * Buffer must be non-empty, otherwise 0 is returned + * @param buffer + * @return + */ +static char fifoBufPop(FifoBuf *buffer); + +/** + * Push character into fifo buffer. If there is no space left, character is + * discarded and false is returned + * @param buffer + * @param a - character to add + * @return true if char was added to buffer, false otherwise + */ +static bool fifoBufPush(FifoBuf *buffer, char a); + +/** + * Copy provided string to the history buffer. + * If it is already inside history, it will be removed from it and added again. + * So after addition, it will always be on top + * If available size is not enough (and total size is enough) old elements will + * be removed from history so this item can be put to it + * @param history + * @param str + * @return true if string was put in history + */ +static bool historyPut(CliHistory *history, const char *str); + +/** + * Get item from history. Items are counted from 1 so if item is 0 or greater + * than itemCount, NULL is returned + * @param history + * @param item + * @return true if string was put in history + */ +static const char *historyGet(CliHistory *history, uint16_t item); + +/** + * Remove specific item from history + * @param history + * @param str - string to remove + * @return + */ +static void historyRemove(CliHistory *history, const char *str); + +/** + * Return position (index of first char) of specified token + * @param tokenizedStr - tokenized string (separated by \0 with + * \0\0 at the end) + * @param pos - token position (counted from 1) + * @return index of first char of specified token + */ +static uint16_t getTokenPosition(const char *tokenizedStr, uint16_t pos); + +EmbeddedCliConfig *embeddedCliDefaultConfig(void) { + defaultConfig.rxBufferSize = 64; + defaultConfig.cmdBufferSize = 64; + defaultConfig.historyBufferSize = 128; + defaultConfig.cliBuffer = NULL; + defaultConfig.cliBufferSize = 0; + defaultConfig.maxBindingCount = 8; + defaultConfig.enableAutoComplete = true; + defaultConfig.invitation = "> "; + return &defaultConfig; +} + +uint16_t embeddedCliRequiredSize(EmbeddedCliConfig *config) { + uint16_t bindingCount = (uint16_t)(config->maxBindingCount + cliInternalBindingCount); + return (uint16_t)(CLI_UINT_SIZE * + (BYTES_TO_CLI_UINTS(sizeof(EmbeddedCli)) + + BYTES_TO_CLI_UINTS(sizeof(EmbeddedCliImpl)) + + BYTES_TO_CLI_UINTS(config->rxBufferSize * sizeof(char)) + + BYTES_TO_CLI_UINTS(config->cmdBufferSize * sizeof(char)) + + BYTES_TO_CLI_UINTS(config->historyBufferSize * sizeof(char)) + + BYTES_TO_CLI_UINTS(bindingCount * sizeof(CliCommandBinding)) + + BYTES_TO_CLI_UINTS(bindingCount * sizeof(uint8_t)))); +} + +EmbeddedCli *embeddedCliNew(EmbeddedCliConfig *config) { + EmbeddedCli *cli = NULL; + + uint16_t bindingCount = (uint16_t)(config->maxBindingCount + cliInternalBindingCount); + + size_t totalSize = embeddedCliRequiredSize(config); + + bool allocated = false; + if (config->cliBuffer == NULL) { + config->cliBuffer = (CLI_UINT *)malloc(totalSize); // malloc guarantees alignment. + if (config->cliBuffer == NULL) + return NULL; + allocated = true; + } else if (config->cliBufferSize < totalSize) { + return NULL; + } + + CLI_UINT *buf = config->cliBuffer; + + memset(buf, 0, totalSize); + + cli = (EmbeddedCli *)buf; + buf += BYTES_TO_CLI_UINTS(sizeof(EmbeddedCli)); + + cli->_impl = (EmbeddedCliImpl *)buf; + buf += BYTES_TO_CLI_UINTS(sizeof(EmbeddedCliImpl)); + + PREPARE_IMPL(cli); + impl->rxBuffer.buf = (char *)buf; + buf += BYTES_TO_CLI_UINTS(config->rxBufferSize * sizeof(char)); + + impl->cmdBuffer = (char *)buf; + buf += BYTES_TO_CLI_UINTS(config->cmdBufferSize * sizeof(char)); + + impl->bindings = (CliCommandBinding *)buf; + buf += BYTES_TO_CLI_UINTS(bindingCount * sizeof(CliCommandBinding)); + + impl->bindingsFlags = (uint8_t *)buf; + buf += BYTES_TO_CLI_UINTS(bindingCount); + + impl->history.buf = (char *)buf; + impl->history.bufferSize = config->historyBufferSize; + + if (allocated) + SET_FLAG(impl->flags, CLI_FLAG_ALLOCATED); + + if (config->enableAutoComplete) + SET_FLAG(impl->flags, CLI_FLAG_AUTOCOMPLETE_ENABLED); + + impl->rxBuffer.size = config->rxBufferSize; + impl->rxBuffer.front = 0; + impl->rxBuffer.back = 0; + impl->cmdMaxSize = config->cmdBufferSize; + impl->bindingsCount = 0; + impl->maxBindingsCount = (uint16_t)(config->maxBindingCount + cliInternalBindingCount); + impl->lastChar = '\0'; + impl->invitation = config->invitation; + + initInternalBindings(cli); + + return cli; +} + +EmbeddedCli *embeddedCliNewDefault(void) { + return embeddedCliNew(embeddedCliDefaultConfig()); +} + +void embeddedCliReceiveChar(EmbeddedCli *cli, char c) { + PREPARE_IMPL(cli); + + if (!fifoBufPush(&impl->rxBuffer, c)) { + SET_FLAG(impl->flags, CLI_FLAG_OVERFLOW); + } +} + +void embeddedCliProcess(EmbeddedCli *cli) { + if (cli->writeChar == NULL) + return; + + PREPARE_IMPL(cli); + + if (!IS_FLAG_SET(impl->flags, CLI_FLAG_INIT_COMPLETE)) { + SET_FLAG(impl->flags, CLI_FLAG_INIT_COMPLETE); + writeToOutput(cli, impl->invitation); + } + + while (fifoBufAvailable(&impl->rxBuffer)) { + char c = fifoBufPop(&impl->rxBuffer); + + if (IS_FLAG_SET(impl->flags, CLI_FLAG_ESCAPE_MODE)) { + onEscapedInput(cli, c); + } else if (impl->lastChar == 0x1B && c == '[') { + // enter escape mode + SET_FLAG(impl->flags, CLI_FLAG_ESCAPE_MODE); + } else if (isControlChar(c)) { + onControlInput(cli, c); + } else if (isDisplayableChar(c)) { + onCharInput(cli, c); + } + + printLiveAutocompletion(cli); + + impl->lastChar = c; + } + + // discard unfinished command if overflow happened + if (IS_FLAG_SET(impl->flags, CLI_FLAG_OVERFLOW)) { + impl->cmdSize = 0; + impl->cmdBuffer[impl->cmdSize] = '\0'; + UNSET_U8FLAG(impl->flags, CLI_FLAG_OVERFLOW); + } +} + +bool embeddedCliAddBinding(EmbeddedCli *cli, CliCommandBinding binding) { + PREPARE_IMPL(cli); + if (impl->bindingsCount == impl->maxBindingsCount) + return false; + + impl->bindings[impl->bindingsCount] = binding; + + ++impl->bindingsCount; + return true; +} + +void embeddedCliPrint(EmbeddedCli *cli, const char *string) { + if (cli->writeChar == NULL) + return; + + PREPARE_IMPL(cli); + + // remove chars for autocompletion and live command + if (!IS_FLAG_SET(impl->flags, CLI_FLAG_DIRECT_PRINT)) + clearCurrentLine(cli); + + // print provided string + writeToOutput(cli, string); + writeToOutput(cli, lineBreak); + + // print current command back to screen + if (!IS_FLAG_SET(impl->flags, CLI_FLAG_DIRECT_PRINT)) { + writeToOutput(cli, impl->invitation); + writeToOutput(cli, impl->cmdBuffer); + impl->inputLineLength = impl->cmdSize; + + printLiveAutocompletion(cli); + } +} + +void embeddedCliFree(EmbeddedCli *cli) { + PREPARE_IMPL(cli); + if (IS_FLAG_SET(impl->flags, CLI_FLAG_ALLOCATED)) { + // allocation is done in single call to malloc, so need only single free + free(cli); + } +} + +void embeddedCliTokenizeArgs(char *args) { + if (args == NULL) + return; + + // for now only space, but can add more later + const char *separators = " "; + + // indicates that arg is quoted so separators are copied as is + bool quotesEnabled = false; + // indicates that previous char was a slash, so next char is copied as is + bool escapeActivated = false; + int insertPos = 0; + + int i = 0; + char currentChar; + while ((currentChar = args[i]) != '\0') { + ++i; + + if (escapeActivated) { + escapeActivated = false; + } else if (currentChar == '\\') { + escapeActivated = true; + continue; + } else if (currentChar == '"') { + quotesEnabled = !quotesEnabled; + currentChar = '\0'; + } else if (!quotesEnabled && strchr(separators, currentChar) != NULL) { + currentChar = '\0'; + } + + // null chars are only copied once and not copied to the beginning + if (currentChar != '\0' || (insertPos > 0 && args[insertPos - 1] != '\0')) { + args[insertPos] = currentChar; + ++insertPos; + } + } + + // make args double null-terminated source buffer must be big enough to contain extra spaces + args[insertPos] = '\0'; + args[insertPos + 1] = '\0'; +} + +const char *embeddedCliGetToken(const char *tokenizedStr, uint16_t pos) { + uint16_t i = getTokenPosition(tokenizedStr, pos); + + if (i != CLI_TOKEN_NPOS) + return &tokenizedStr[i]; + else + return NULL; +} + +char *embeddedCliGetTokenVariable(char *tokenizedStr, uint16_t pos) { + uint16_t i = getTokenPosition(tokenizedStr, pos); + + if (i != CLI_TOKEN_NPOS) + return &tokenizedStr[i]; + else + return NULL; +} + +uint16_t embeddedCliFindToken(const char *tokenizedStr, const char *token) { + if (tokenizedStr == NULL || token == NULL) + return 0; + + uint16_t size = embeddedCliGetTokenCount(tokenizedStr); + for (uint16_t i = 1; i <= size; ++i) { + if (strcmp(embeddedCliGetToken(tokenizedStr, i), token) == 0) + return i; + } + + return 0; +} + +uint16_t embeddedCliGetTokenCount(const char *tokenizedStr) { + if (tokenizedStr == NULL || tokenizedStr[0] == '\0') + return 0; + + int i = 0; + uint16_t tokenCount = 1; + while (true) { + if (tokenizedStr[i] == '\0') { + if (tokenizedStr[i + 1] == '\0') + break; + ++tokenCount; + } + ++i; + } + + return tokenCount; +} + +static void navigateHistory(EmbeddedCli *cli, bool navigateUp) { + PREPARE_IMPL(cli); + if (impl->history.itemsCount == 0 || + (navigateUp && impl->history.current == impl->history.itemsCount) || + (!navigateUp && impl->history.current == 0)) + return; + + clearCurrentLine(cli); + + writeToOutput(cli, impl->invitation); + + if (navigateUp) + ++impl->history.current; + else + --impl->history.current; + + const char *item = historyGet(&impl->history, impl->history.current); + // simple way to handle empty command the same way as others + if (item == NULL) + item = ""; + uint16_t len = (uint16_t)strlen(item); + memcpy(impl->cmdBuffer, item, len); + impl->cmdBuffer[len] = '\0'; + impl->cmdSize = len; + + writeToOutput(cli, impl->cmdBuffer); + impl->inputLineLength = impl->cmdSize; + + printLiveAutocompletion(cli); +} + +static void onEscapedInput(EmbeddedCli *cli, char c) { + PREPARE_IMPL(cli); + + if (c >= 64 && c <= 126) { + // handle escape sequence + UNSET_U8FLAG(impl->flags, CLI_FLAG_ESCAPE_MODE); + + if (c == 'A' || c == 'B') { + // treat \e[..A as cursor up and \e[..B as cursor down + // there might be extra chars between [ and A/B, just ignore them + navigateHistory(cli, c == 'A'); + } + } +} + +static void onCharInput(EmbeddedCli *cli, char c) { + PREPARE_IMPL(cli); + + // have to reserve two extra chars for command ending (used in tokenization) + if (impl->cmdSize + 2 >= impl->cmdMaxSize) + return; + + impl->cmdBuffer[impl->cmdSize] = c; + ++impl->cmdSize; + impl->cmdBuffer[impl->cmdSize] = '\0'; + + cli->writeChar(cli, c); +} + +static void onControlInput(EmbeddedCli *cli, char c) { + PREPARE_IMPL(cli); + + // process \r\n and \n\r as single \r\n command + if ((impl->lastChar == '\r' && c == '\n') || (impl->lastChar == '\n' && c == '\r')) + return; + + if (c == '\r' || c == '\n') { + // try to autocomplete command and then process it + onAutocompleteRequest(cli); + + writeToOutput(cli, lineBreak); + + if (impl->cmdSize > 0) + parseCommand(cli); + impl->cmdSize = 0; + impl->cmdBuffer[impl->cmdSize] = '\0'; + impl->inputLineLength = 0; + impl->history.current = 0; + + writeToOutput(cli, impl->invitation); + } else if ((c == '\b' || c == 0x7F) && impl->cmdSize > 0) { + // remove char from screen + cli->writeChar(cli, '\b'); + cli->writeChar(cli, ' '); + cli->writeChar(cli, '\b'); + // and from buffer + --impl->cmdSize; + impl->cmdBuffer[impl->cmdSize] = '\0'; + } else if (c == '\t') { + onAutocompleteRequest(cli); + } +} + +static void parseCommand(EmbeddedCli *cli) { + PREPARE_IMPL(cli); + + bool isEmpty = true; + + for (int i = 0; i < impl->cmdSize; ++i) { + if (impl->cmdBuffer[i] != ' ') { + isEmpty = false; + break; + } + } + // do not process empty commands + if (isEmpty) + return; + // push command to history before buffer is modified + historyPut(&impl->history, impl->cmdBuffer); + + char *cmdName = NULL; + char *cmdArgs = NULL; + bool nameFinished = false; + + // find command name and command args inside command buffer + for (int i = 0; i < impl->cmdSize; ++i) { + char c = impl->cmdBuffer[i]; + + if (c == ' ') { + // all spaces between name and args are filled with zeros + // so name is a correct null-terminated string + if (cmdArgs == NULL) + impl->cmdBuffer[i] = '\0'; + if (cmdName != NULL) + nameFinished = true; + + } else if (cmdName == NULL) { + cmdName = &impl->cmdBuffer[i]; + } else if (cmdArgs == NULL && nameFinished) { + cmdArgs = &impl->cmdBuffer[i]; + } + } + + // we keep two last bytes in cmd buffer reserved so cmdSize is always by 2 + // less than cmdMaxSize + impl->cmdBuffer[impl->cmdSize + 1] = '\0'; + + if (cmdName == NULL) + return; + + // try to find command in bindings + for (int i = 0; i < impl->bindingsCount; ++i) { + if (strcmp(cmdName, impl->bindings[i].name) == 0) { + if (impl->bindings[i].binding == NULL) + break; + + if (impl->bindings[i].tokenizeArgs) + embeddedCliTokenizeArgs(cmdArgs); + // currently, output is blank line, so we can just print directly + SET_FLAG(impl->flags, CLI_FLAG_DIRECT_PRINT); + impl->bindings[i].binding(cli, cmdArgs, impl->bindings[i].context); + UNSET_U8FLAG(impl->flags, CLI_FLAG_DIRECT_PRINT); + return; + } + } + + // command not found in bindings or binding was null + // try to call default callback + if (cli->onCommand != NULL) { + CliCommand command; + command.name = cmdName; + command.args = cmdArgs; + + // currently, output is blank line, so we can just print directly + SET_FLAG(impl->flags, CLI_FLAG_DIRECT_PRINT); + cli->onCommand(cli, &command); + UNSET_U8FLAG(impl->flags, CLI_FLAG_DIRECT_PRINT); + } else { + onUnknownCommand(cli, cmdName); + } +} + +static void initInternalBindings(EmbeddedCli *cli) { + CliCommandBinding b = {"help", "Print list of commands", true, NULL, onHelp}; + embeddedCliAddBinding(cli, b); +} + +static void onHelp(EmbeddedCli *cli, char *tokens, void *context) { + UNUSED(context); + PREPARE_IMPL(cli); + + if (impl->bindingsCount == 0) { + writeToOutput(cli, "Help is not available"); + writeToOutput(cli, lineBreak); + return; + } + + uint16_t tokenCount = embeddedCliGetTokenCount(tokens); + if (tokenCount == 0) { + for (int i = 0; i < impl->bindingsCount; ++i) { + writeToOutput(cli, " * "); + writeToOutput(cli, impl->bindings[i].name); + writeToOutput(cli, lineBreak); + if (impl->bindings[i].help != NULL) { + cli->writeChar(cli, '\t'); + writeToOutput(cli, impl->bindings[i].help); + writeToOutput(cli, lineBreak); + } + } + } else if (tokenCount == 1) { + // try find command + const char *helpStr = NULL; + const char *cmdName = embeddedCliGetToken(tokens, 1); + bool found = false; + for (int i = 0; i < impl->bindingsCount; ++i) { + if (strcmp(impl->bindings[i].name, cmdName) == 0) { + helpStr = impl->bindings[i].help; + found = true; + break; + } + } + if (found && helpStr != NULL) { + writeToOutput(cli, " * "); + writeToOutput(cli, cmdName); + writeToOutput(cli, lineBreak); + cli->writeChar(cli, '\t'); + writeToOutput(cli, helpStr); + writeToOutput(cli, lineBreak); + } else if (found) { + writeToOutput(cli, "Help is not available"); + writeToOutput(cli, lineBreak); + } else { + onUnknownCommand(cli, cmdName); + } + } else { + writeToOutput(cli, "Command \"help\" receives one or zero arguments"); + writeToOutput(cli, lineBreak); + } +} + +static void onUnknownCommand(EmbeddedCli *cli, const char *name) { + writeToOutput(cli, "Unknown command: \""); + writeToOutput(cli, name); + writeToOutput(cli, "\". Write \"help\" for a list of available commands"); + writeToOutput(cli, lineBreak); +} + +static AutocompletedCommand getAutocompletedCommand(EmbeddedCli *cli, const char *prefix) { + AutocompletedCommand cmd = {NULL, 0, 0}; + + size_t prefixLen = strlen(prefix); + + PREPARE_IMPL(cli); + if (impl->bindingsCount == 0 || prefixLen == 0) + return cmd; + + for (int i = 0; i < impl->bindingsCount; ++i) { + const char *name = impl->bindings[i].name; + size_t len = strlen(name); + + // unset autocomplete flag + UNSET_U8FLAG(impl->bindingsFlags[i], BINDING_FLAG_AUTOCOMPLETE); + + if (len < prefixLen) + continue; + + // check if this command is candidate for autocomplete + bool isCandidate = true; + for (size_t j = 0; j < prefixLen; ++j) { + if (prefix[j] != name[j]) { + isCandidate = false; + break; + } + } + if (!isCandidate) + continue; + + impl->bindingsFlags[i] |= BINDING_FLAG_AUTOCOMPLETE; + + if (cmd.candidateCount == 0 || len < cmd.autocompletedLen) + cmd.autocompletedLen = (uint16_t)len; + + ++cmd.candidateCount; + + if (cmd.candidateCount == 1) { + cmd.firstCandidate = name; + continue; + } + + for (size_t j = impl->cmdSize; j < cmd.autocompletedLen; ++j) { + if (cmd.firstCandidate[j] != name[j]) { + cmd.autocompletedLen = (uint16_t)j; + break; + } + } + } + + return cmd; +} + +static void printLiveAutocompletion(EmbeddedCli *cli) { + PREPARE_IMPL(cli); + + if (!IS_FLAG_SET(impl->flags, CLI_FLAG_AUTOCOMPLETE_ENABLED)) + return; + + AutocompletedCommand cmd = getAutocompletedCommand(cli, impl->cmdBuffer); + + if (cmd.candidateCount == 0) { + cmd.autocompletedLen = impl->cmdSize; + } + + // print live autocompletion (or nothing, if it doesn't exist) + for (size_t i = impl->cmdSize; i < cmd.autocompletedLen; ++i) { + cli->writeChar(cli, cmd.firstCandidate[i]); + } + // replace with spaces previous autocompletion + for (size_t i = cmd.autocompletedLen; i < impl->inputLineLength; ++i) { + cli->writeChar(cli, ' '); + } + impl->inputLineLength = cmd.autocompletedLen; + cli->writeChar(cli, '\r'); + // print current command again so cursor is moved to initial place + writeToOutput(cli, impl->invitation); + writeToOutput(cli, impl->cmdBuffer); +} + +static void onAutocompleteRequest(EmbeddedCli *cli) { + PREPARE_IMPL(cli); + + AutocompletedCommand cmd = getAutocompletedCommand(cli, impl->cmdBuffer); + + if (cmd.candidateCount == 0) + return; + + if (cmd.candidateCount == 1 || cmd.autocompletedLen > impl->cmdSize) { + // can copy from index cmdSize, but prefix is the same, so copy everything + memcpy(impl->cmdBuffer, cmd.firstCandidate, cmd.autocompletedLen); + if (cmd.candidateCount == 1) { + impl->cmdBuffer[cmd.autocompletedLen] = ' '; + ++cmd.autocompletedLen; + } + impl->cmdBuffer[cmd.autocompletedLen] = '\0'; + + writeToOutput(cli, &impl->cmdBuffer[impl->cmdSize]); + impl->cmdSize = cmd.autocompletedLen; + impl->inputLineLength = impl->cmdSize; + return; + } + + // with multiple candidates when we already completed to common prefix + // we show all candidates and print input again + // we need to completely clear current line since it begins with invitation + clearCurrentLine(cli); + + for (int i = 0; i < impl->bindingsCount; ++i) { + // autocomplete flag is set for all candidates by last call to + // getAutocompletedCommand + if (!(impl->bindingsFlags[i] & BINDING_FLAG_AUTOCOMPLETE)) + continue; + + const char *name = impl->bindings[i].name; + + writeToOutput(cli, name); + writeToOutput(cli, lineBreak); + } + + writeToOutput(cli, impl->invitation); + writeToOutput(cli, impl->cmdBuffer); + + impl->inputLineLength = impl->cmdSize; +} + +static void clearCurrentLine(EmbeddedCli *cli) { + PREPARE_IMPL(cli); + size_t len = impl->inputLineLength + strlen(impl->invitation); + + cli->writeChar(cli, '\r'); + for (size_t i = 0; i < len; ++i) { + cli->writeChar(cli, ' '); + } + cli->writeChar(cli, '\r'); + impl->inputLineLength = 0; +} + +static void writeToOutput(EmbeddedCli *cli, const char *str) { + size_t len = strlen(str); + + for (size_t i = 0; i < len; ++i) { + cli->writeChar(cli, str[i]); + } +} + +static bool isControlChar(char c) { + return c == '\r' || c == '\n' || c == '\b' || c == '\t' || c == 0x7F; +} + +static bool isDisplayableChar(char c) { + return (c >= 32 && c <= 126); +} + +static uint16_t fifoBufAvailable(FifoBuf *buffer) { + if (buffer->back >= buffer->front) + return (uint16_t)(buffer->back - buffer->front); + else + return (uint16_t)(buffer->size - buffer->front + buffer->back); +} + +static char fifoBufPop(FifoBuf *buffer) { + char a = '\0'; + if (buffer->front != buffer->back) { + a = buffer->buf[buffer->front]; + buffer->front = (uint16_t)(buffer->front + 1) % buffer->size; + } + return a; +} + +static bool fifoBufPush(FifoBuf *buffer, char a) { + uint16_t newBack = (uint16_t)(buffer->back + 1) % buffer->size; + if (newBack != buffer->front) { + buffer->buf[buffer->back] = a; + buffer->back = newBack; + return true; + } + return false; +} + +static bool historyPut(CliHistory *history, const char *str) { + size_t len = strlen(str); + // each item is ended with \0 so, need to have that much space at least + if (history->bufferSize < len + 1) + return false; + + // remove str from history (if it's present) so we don't get duplicates + historyRemove(history, str); + + size_t usedSize; + // remove old items if new one can't fit into buffer + while (history->itemsCount > 0) { + const char *item = historyGet(history, history->itemsCount); + size_t itemLen = strlen(item); + usedSize = ((size_t)(item - history->buf)) + itemLen + 1; + + size_t freeSpace = history->bufferSize - usedSize; + + if (freeSpace >= len + 1) + break; + + // space not enough, remove last element + --history->itemsCount; + } + if (history->itemsCount > 0) { + // when history not empty, shift elements so new item is first + memmove(&history->buf[len + 1], history->buf, usedSize); + } + memcpy(history->buf, str, len + 1); + ++history->itemsCount; + + return true; +} + +static const char *historyGet(CliHistory *history, uint16_t item) { + if (item == 0 || item > history->itemsCount) + return NULL; + + // items are stored in the same way (separated by \0 and counted from 1), + // so can use this call + return embeddedCliGetToken(history->buf, item); +} + +static void historyRemove(CliHistory *history, const char *str) { + if (str == NULL || history->itemsCount == 0) + return; + char *item = NULL; + uint16_t itemPosition; + for (itemPosition = 1; itemPosition <= history->itemsCount; ++itemPosition) { + // items are stored in the same way (separated by \0 and counted from 1), + // so can use this call + item = embeddedCliGetTokenVariable(history->buf, itemPosition); + if (strcmp(item, str) == 0) { + break; + } + item = NULL; + } + if (item == NULL) + return; + + --history->itemsCount; + if (itemPosition == (history->itemsCount + 1)) { + // if this is a last element, nothing is remaining to move + return; + } + + size_t len = strlen(item); + size_t remaining = (size_t)(history->bufferSize - (item + len + 1 - history->buf)); + // move everything to the right of found item + memmove(item, &item[len + 1], remaining); +} + +static uint16_t getTokenPosition(const char *tokenizedStr, uint16_t pos) { + if (tokenizedStr == NULL || pos == 0) + return CLI_TOKEN_NPOS; + uint16_t i = 0; + uint16_t tokenCount = 1; + while (true) { + if (tokenCount == pos) + break; + + if (tokenizedStr[i] == '\0') { + ++tokenCount; + if (tokenizedStr[i + 1] == '\0') + break; + } + + ++i; + } + + if (tokenizedStr[i] != '\0') + return i; + else + return CLI_TOKEN_NPOS; +} diff --git a/components/tftpShell/extern/embedded_cli.h b/components/tftpShell/extern/embedded_cli.h new file mode 100644 index 0000000..abfebd4 --- /dev/null +++ b/components/tftpShell/extern/embedded_cli.h @@ -0,0 +1,312 @@ +#ifndef EMBEDDED_CLI_H +#define EMBEDDED_CLI_H + + +#ifdef __cplusplus + +extern "C" { +#else + +#include + +#endif + +// cstdint is available only since C++11, so use C header +#include + +// used for proper alignment of cli buffer +#if UINTPTR_MAX == 0xFFFF +#define CLI_UINT uint16_t +#elif UINTPTR_MAX == 0xFFFFFFFF +#define CLI_UINT uint32_t +#elif UINTPTR_MAX == 0xFFFFFFFFFFFFFFFFu +#define CLI_UINT uint64_t +#else +#error unsupported pointer size +#endif + +#define CLI_UINT_SIZE (sizeof(CLI_UINT)) +// convert size in bytes to size in terms of CLI_UINTs (rounded up +// if bytes is not divisible by size of single CLI_UINT) +#define BYTES_TO_CLI_UINTS(bytes) \ + (((bytes) + CLI_UINT_SIZE - 1)/CLI_UINT_SIZE) + +typedef struct CliCommand CliCommand; +typedef struct CliCommandBinding CliCommandBinding; +typedef struct EmbeddedCli EmbeddedCli; +typedef struct EmbeddedCliConfig EmbeddedCliConfig; + + +struct CliCommand { + /** + * Name of the command. + * In command "set led 1 1" "set" is name + */ + const char *name; + + /** + * String of arguments of the command. + * In command "set led 1 1" "led 1 1" is string of arguments + * Is ended with double 0x00 char + * Use tokenize functions to easily get individual tokens + */ + char *args; +}; + +/** + * Struct to describe binding of command to function and + */ +struct CliCommandBinding { + /** + * Name of command to bind. Should not be NULL. + */ + const char *name; + + /** + * Help string that will be displayed when "help " is executed. + * Can have multiple lines separated with "\r\n" + * Can be NULL if no help is provided. + */ + const char *help; + + /** + * Flag to perform tokenization before calling binding function. + */ + bool tokenizeArgs; + + /** + * Pointer to any specific app context that is required for this binding. + * It will be provided in binding callback. + */ + void *context; + + /** + * Binding function for when command is received. + * If null, default callback (onCommand) will be called. + * @param cli - pointer to cli that is calling this binding + * @param args - string of args (if tokenizeArgs is false) or tokens otherwise + * @param context + */ + void (*binding)(EmbeddedCli *cli, char *args, void *context); +}; + +struct EmbeddedCli { + /** + * Should write char to connection + * @param cli - pointer to cli that executed this function + * @param c - actual character to write + */ + void (*writeChar)(EmbeddedCli *cli, char c); + + /** + * Called when command is received and command not found in list of + * command bindings (or binding function is null). + * @param cli - pointer to cli that executed this function + * @param command - pointer to received command + */ + void (*onCommand)(EmbeddedCli *cli, CliCommand *command); + + /** + * Can be used for any application context + */ + void *appContext; + + /** + * Pointer to actual implementation, do not use. + */ + void *_impl; +}; + +/** + * Configuration to create CLI + */ +struct EmbeddedCliConfig { + /** + * Invitation string. Is printed at the beginning of each line with user + * input + */ + const char *invitation; + + /** + * Size of buffer that is used to store characters until they're processed + */ + uint16_t rxBufferSize; + + /** + * Size of buffer that is used to store current input that is not yet + * sended as command (return not pressed yet) + */ + uint16_t cmdBufferSize; + + /** + * Size of buffer that is used to store previously entered commands + * Only unique commands are stored in buffer. If buffer is smaller than + * entered command (including arguments), command is discarded from history + */ + uint16_t historyBufferSize; + + /** + * Maximum amount of bindings that can be added via addBinding function. + * Cli increases takes extra bindings for internal commands: + * - help + */ + uint16_t maxBindingCount; + + /** + * Buffer to use for cli and all internal structures. If NULL, memory will + * be allocated dynamically. Otherwise this buffer is used and no + * allocations are made + */ + CLI_UINT *cliBuffer; + + /** + * Size of buffer for cli and internal structures (in bytes). + */ + uint16_t cliBufferSize; + + /** + * Whether autocompletion should be enabled. + * If false, autocompletion is disabled but you still can use 'tab' to + * complete current command manually. + */ + bool enableAutoComplete; +}; + +/** + * Returns pointer to default configuration for cli creation. It is safe to + * modify it and then send to embeddedCliNew(). + * Returned structure is always the same so do not free and try to use it + * immediately. + * Default values: + *
    + *
  • rxBufferSize = 64
  • + *
  • cmdBufferSize = 64
  • + *
  • historyBufferSize = 128
  • + *
  • cliBuffer = NULL (use dynamic allocation)
  • + *
  • cliBufferSize = 0
  • + *
  • maxBindingCount = 8
  • + *
  • enableAutoComplete = true
  • + *
+ * @return configuration for cli creation + */ +EmbeddedCliConfig *embeddedCliDefaultConfig(void); + +/** + * Returns how many space in config buffer is required for cli creation + * If you provide buffer with less space, embeddedCliNew will return NULL + * This amount will always be divisible by CLI_UINT_SIZE so allocated buffer + * and internal structures can be properly aligned + * @param config + * @return + */ +uint16_t embeddedCliRequiredSize(EmbeddedCliConfig *config); + +/** + * Create new CLI. + * Memory is allocated dynamically if cliBuffer in config is NULL. + * After CLI is created, override function pointers to start using it + * @param config - config for cli creation + * @return pointer to created CLI + */ +EmbeddedCli *embeddedCliNew(EmbeddedCliConfig *config); + +/** + * Same as calling embeddedCliNew with default config. + * @return + */ +EmbeddedCli *embeddedCliNewDefault(void); + +/** + * Receive character and put it to internal buffer + * Actual processing is done inside embeddedCliProcess + * You can call this function from something like interrupt service routine, + * just make sure that you call it only from single place. Otherwise input + * might get corrupted + * @param cli + * @param c - received char + */ +void embeddedCliReceiveChar(EmbeddedCli *cli, char c); + +/** + * Process rx/tx buffers. Command callbacks are called from here + * @param cli + */ +void embeddedCliProcess(EmbeddedCli *cli); + +/** + * Add specified binding to list of bindings. If list is already full, binding + * is not added and false is returned + * @param cli + * @param binding + * @return true if binding was added, false otherwise + */ +bool embeddedCliAddBinding(EmbeddedCli *cli, CliCommandBinding binding); + +/** + * Print specified string and account for currently entered but not submitted + * command. + * Current command is deleted, provided string is printed (with new line) after + * that current command is printed again, so user can continue typing it. + * @param cli + * @param string + */ +void embeddedCliPrint(EmbeddedCli *cli, const char *string); + +/** + * Free allocated for cli memory + * @param cli + */ +void embeddedCliFree(EmbeddedCli *cli); + +/** + * Perform tokenization of arguments string. Original string is modified and + * should not be used directly (only inside other token functions). + * Individual tokens are separated by single 0x00 char, double 0x00 is put at + * the end of token list. After calling this function, you can use other + * token functions to get individual tokens and token count. + * + * Important: Call this function only once. Otherwise information will be lost if + * more than one token existed + * @param args - string to tokenize (must have extra writable char after 0x00) + * @return + */ +void embeddedCliTokenizeArgs(char *args); + +/** + * Return specific token from tokenized string + * @param tokenizedStr + * @param pos (counted from 1) + * @return token + */ +const char *embeddedCliGetToken(const char *tokenizedStr, uint16_t pos); + +/** + * Same as embeddedCliGetToken but works on non-const buffer + * @param tokenizedStr + * @param pos (counted from 1) + * @return token + */ +char *embeddedCliGetTokenVariable(char *tokenizedStr, uint16_t pos); + +/** + * Find token in provided tokens string and return its position (counted from 1) + * If no such token is found - 0 is returned. + * @param tokenizedStr + * @param token - token to find + * @return position (increased by 1) or zero if no such token found + */ +uint16_t embeddedCliFindToken(const char *tokenizedStr, const char *token); + +/** + * Return number of tokens in tokenized string + * @param tokenizedStr + * @return number of tokens + */ +uint16_t embeddedCliGetTokenCount(const char *tokenizedStr); + +#ifdef __cplusplus +} +#endif + + +#endif //EMBEDDED_CLI_H diff --git a/components/tftpShell/extern/license b/components/tftpShell/extern/license new file mode 100644 index 0000000..8993f81 --- /dev/null +++ b/components/tftpShell/extern/license @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2021 Sviatoslav Kokurin (funbiscuit) + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/components/tftpShell/extern/readme.md b/components/tftpShell/extern/readme.md new file mode 100644 index 0000000..a1cce1f --- /dev/null +++ b/components/tftpShell/extern/readme.md @@ -0,0 +1,7 @@ +# Embedded CLI + +Implementation is taken from `funbiscuit` @ [github](https://github.com/funbiscuit/embedded-cli) + +- Only source and header where integrated. +- Full project is included in `external/embedded-cli` + diff --git a/components/tftpShell/include/cliWrapper.hpp b/components/tftpShell/include/cliWrapper.hpp new file mode 100644 index 0000000..088dc4d --- /dev/null +++ b/components/tftpShell/include/cliWrapper.hpp @@ -0,0 +1,161 @@ +/** + * @file cliWrapper.hpp + * @author Gustice + * @brief Wrapping Command-Line-Interfaces with custom endpoints + * @details Command-Line-Interface is based on the implementation of + * https://github.com/funbiscuit/embedded-cli + * + * @copyright Copyright (c) 2024 + */ + +#pragma once + +#include "embedded_cli.h" +#include +#include +#include +#include "utils.hpp" + +/* Module configuration */ +static constexpr int CLI_RX_BUFFER_SIZE = 64; +static constexpr int CLI_CMD_BUFFER_SIZE = 64; // to also accept URLs for Updates +static constexpr int CLI_HISTORY_SIZE = 16; +static constexpr int CLI_BINDING_COUNT = 8; +static constexpr std::string_view CliWelcomeString{ + "** Smart-DMX-Bridge **\n" + " send 'help' for help-page\n" + " NOTE: Syslog-messages are sent to client of this session\n"}; + +/** + * @brief Wrapper for c-implementation of CLI + */ +class Cli { + public: + /// @brief Type for callback arguments + using CArgs = std::vector &; + /// @brief Type for bind callback + using BindCb = std::function; + /// @brief Type for command callback + using CommandCb = std::function; + + /// @brief Parameter structure for command-binding + struct Binding { + std::string_view name; + std::string_view help_output; + BindCb callback; + }; + + /// @brief Constructor + /// @param port Reference to serial port + /// @param cmd command callback + Cli(CharStream &port, CommandCb cmd) + : _port(port), _commandCb(cmd) { + EmbeddedCliConfig *config = embeddedCliDefaultConfig(); + // NOTE: Note setting cliBuffer allows Cli to allocate needed buffer in correct size + // config->cliBuffer = cliBuffer; + // config->cliBufferSize = CLI_BUFFER_SIZE; + config->rxBufferSize = CLI_RX_BUFFER_SIZE; + config->cmdBufferSize = CLI_CMD_BUFFER_SIZE; + config->historyBufferSize = CLI_HISTORY_SIZE; + config->maxBindingCount = CLI_BINDING_COUNT; + cli = embeddedCliNew(config); + + if (cli == NULL) { + ESP_LOGE("CLI", "Cli was not created, requires %d bytes. Check sizes!", + embeddedCliRequiredSize(config)); + return; + } + cli->onCommand = genericCallback; + cli->writeChar = writeToken; + // appContext is a user-end variable and is bound to current instance + cli->appContext = this; + } + + /// @brief Destructor + ~Cli() { + embeddedCliFree(cli); + } + + /// @brief Create command binding + bool addBinding(Binding bind) { + BindCb *cb = new BindCb{bind.callback}; + + CliCommandBinding cBinding{.name = bind.name.begin(), + .help = bind.help_output.begin(), + .tokenizeArgs = true, + // NOTE: this hard reference only works because we only provide + // an add-function without remove + .context = (void *)(cb), + .binding = boundCallback}; + return embeddedCliAddBinding(cli, cBinding); + } + + /// @brief Static method to generate c-compatible function-pointer for bound-command-callback + /// @details - receives and wraps arguments in string, + /// - determines object according to appContext + /// - calls appropriate method of appropriate object + static void boundCallback(EmbeddedCli *cli, char *args, void *context) { + if (context == nullptr) { + return; + ESP_LOGE("CLI", "No context in callback available"); + } + ESP_LOGD("CLI", "Received command:"); + + BindCb &cb = *((BindCb *)context); + uint len = embeddedCliGetTokenCount(args); + std::vector sArgs; + while (len-- > 0) { + const char *const arg = embeddedCliGetToken(args, 1); + ESP_LOGD("CLI", " arg: %s", arg); + sArgs.emplace_back(arg); + // shift position to end of this element to save time + args += sArgs.back().size() + 1; // needs also to be shifted by null-delimiter + } + cb(sArgs); + } + + /// @brief Static method to generate c-compatible function-pointer for command-callback + /// @details - receives and wraps arguments in string, + /// - determines object according to appContext + /// - calls command-callback of appropriate object + static void genericCallback(EmbeddedCli *cli, CliCommand *command) { + std::string name(command->name); + ESP_LOGI("CLI", "Received generic command: %s", name.c_str()); + + embeddedCliTokenizeArgs(command->args); + std::vector sArgs; + for (int i = 1; i <= embeddedCliGetTokenCount(command->args); ++i) { + sArgs.emplace_back(embeddedCliGetToken(command->args, i)); + ESP_LOGI("CLI", " arg %d : %s", i, sArgs.back().c_str()); + } + auto &commandCb = static_cast(cli->appContext)->_commandCb; + commandCb(name, sArgs); + } + + /// @brief Process buffer and trigger commands + void process() { + auto r = _port.read(); + for (auto &&c : r) { + embeddedCliReceiveChar(cli, c); + } + embeddedCliProcess(cli); + } + + /// @brief publicly exposed write method + void write(std::string output) { + _port.write(output); + } + + private: + /// @brief Reference to hardware port + CharStream &_port; + /// @brief Callback for generic command + CommandCb _commandCb; + EmbeddedCli *cli = nullptr; + + /// @brief Write callback + static void writeToken(EmbeddedCli *embeddedCli, char c) { + static_cast(embeddedCli->appContext)->_port.write(c); + } +}; + diff --git a/main/watchMain.cpp b/main/watchMain.cpp index 2d226a0..7ac2fd0 100644 --- a/main/watchMain.cpp +++ b/main/watchMain.cpp @@ -32,6 +32,9 @@ #include "PwmExpanderPca9685.hpp" #include "StepperDrv8833.hpp" #include "stepperMotor.hpp" +#include "cliWrapper.hpp" +#include "udpSockets.hpp" +#include "tcpSockets.hpp" static const char *TAG = "Watch"; static esp_eth_handle_t s_eth_handle = NULL; @@ -41,6 +44,103 @@ extern "C" { // This switch allows the ROS C-implementation to find this main void app_main(void); } + +std::unique_ptr shellPtr; + +/// @brief Callback for update start +/// @param args callback arguments +void startUpdate(Cli::CArgs args) { + ESP_LOGI(TAG, "Invoking Update"); + if (args.size() != 1) { + shellPtr->write("Need URL to application'\n"); + return; + } + // @ToDo: Start Update here +} + +/// @brief Callback for commands that are not directly wired +/// @param name command name +/// @param args callback arguments +void onCommand(std::string name, Cli::CArgs args) { + ESP_LOGI(TAG, "Invoking: %s", name.c_str()); +} + +void showLogServerConfig(Cli::CArgs args) { + ESP_LOGI(TAG, "on showLogServerConfig:"); + // auto &store = NvsApi::getInstance(); + // auto logger = store.getValue("syslogDst"); + // shellPtr->write(std::string{"syslog destination = '"} + logger + "'\n"); +} + +void setLogServerConfig(Cli::CArgs args) { + ESP_LOGI(TAG, "on showLogServerConfig:"); + if (args.size() != 1) { + shellPtr->write("Need IP address to syslog server'\n"); + return; + } + // auto &store = NvsApi::getInstance(); + // shellPtr->write(std::string{"setting syslog destination to = '"} + args[0] + "'\n"); + // store.setValue("syslogDst", args[0]); +} + +/** + * @brief Telnet socket via TCP/IP + * + * @details The ini-file for TeraTerm only changes CRReceive and CRSend values both to "=CR" + * @param arg ignored task argument + */ +static void telnetSocketTask(int addrFamily) { + ESP_LOGI(TAG, "Starting Telnet task"); + try { + TelnetServer telnet(addrFamily); + telnet.run( + [](TcpSession &session) { + shellPtr = std::make_unique(session, [](std::string n, Cli::CArgs a) { onCommand(n, a); }); + shellPtr->addBinding( + {"update", "Start OTA-Update-Process", + [](Cli::CArgs a) { startUpdate(a); }}); + shellPtr->addBinding( + {"logserver_show", "Shows log server config ip", + [](Cli::CArgs a) { showLogServerConfig(a); }}); + shellPtr->addBinding( + {"logserver_set", "Set log server config ip", + [](Cli::CArgs a) { setLogServerConfig(a); }}); + }, + [](TcpSession &session) { + shellPtr->process(); + Syslog::setupTask(session.getOther()); + }); + } catch (const std::exception &e) { + ESP_LOGE("TCP", "%s", e.what()); + } + + vTaskDelete(NULL); +} + +void startServerTasks() { + static int ipVersion = AF_INET; + static fos::TaskT telnetSocket(telnetSocketTask, ipVersion, "tcpServer", 4096); + // static TaskT dmxNode(dmxSocketTask, ipVersion, "dmxNode", 4096); + // static Task otaUpdate(SystemFunctions::OtaTaskEntry, "otaTask", 8192); + + // initialise_mdns(); + // netbiosns_init(); + // netbiosns_set_name(CONFIG_EXAMPLE_MDNS_HOST_NAME); + // WebServer::init(FileSystem::getInstance()); + // WebServer::getInstance(FileSystem::getInstance()).registerCallback(printSysState); + + // auto &store = NvsApi::getInstance(); + // auto broker = store.getValue("mqttBroker"); + // if (broker != "") { + // MqttClient::init(broker); + // } + + // auto logger = store.getValue("syslogDst"); + // if (logger != "") { + // Syslog::setupTask(IpAdr{logger, 0}); + // } +} + /** Event handler for IP_EVENT_ETH_GOT_IP */ static void got_ip_event_handler(void *arg, esp_event_base_t event_base, int32_t event_id, void *event_data) @@ -54,6 +154,7 @@ static void got_ip_event_handler(void *arg, esp_event_base_t event_base, ESP_LOGI(TAG, "ETHMASK:" IPSTR, IP2STR(&ip_info->netmask)); ESP_LOGI(TAG, "ETHGW:" IPSTR, IP2STR(&ip_info->gw)); ESP_LOGI(TAG, "~~~~~~~~~~~"); + startServerTasks(); } // Event handler for Ethernet diff --git a/sdkconfig b/sdkconfig index 5ec0bf9..e7bdbcb 100644 --- a/sdkconfig +++ b/sdkconfig @@ -585,7 +585,8 @@ CONFIG_COMPILER_FLOAT_LIB_FROM_RVFPLIB=y CONFIG_COMPILER_OPTIMIZATION_ASSERTION_LEVEL=2 # CONFIG_COMPILER_OPTIMIZATION_CHECKS_SILENT is not set CONFIG_COMPILER_HIDE_PATHS_MACROS=y -# CONFIG_COMPILER_CXX_EXCEPTIONS is not set +CONFIG_COMPILER_CXX_EXCEPTIONS=y +CONFIG_COMPILER_CXX_EXCEPTIONS_EMG_POOL_SIZE=0 # CONFIG_COMPILER_CXX_RTTI is not set CONFIG_COMPILER_STACK_CHECK_MODE_NONE=y # CONFIG_COMPILER_STACK_CHECK_MODE_NORM is not set @@ -1412,7 +1413,7 @@ CONFIG_LOG_TIMESTAMP_SOURCE_RTOS=y # LWIP # CONFIG_LWIP_ENABLE=y -CONFIG_LWIP_LOCAL_HOSTNAME="espressif" +CONFIG_LWIP_LOCAL_HOSTNAME="ChaosWatch" # CONFIG_LWIP_NETIF_API is not set CONFIG_LWIP_TCPIP_TASK_PRIO=18 # CONFIG_LWIP_TCPIP_CORE_LOCKING is not set @@ -2001,7 +2002,8 @@ CONFIG_OPTIMIZATION_ASSERTIONS_ENABLED=y # CONFIG_OPTIMIZATION_ASSERTIONS_SILENT is not set # CONFIG_OPTIMIZATION_ASSERTIONS_DISABLED is not set CONFIG_OPTIMIZATION_ASSERTION_LEVEL=2 -# CONFIG_CXX_EXCEPTIONS is not set +CONFIG_CXX_EXCEPTIONS=y +CONFIG_CXX_EXCEPTIONS_EMG_POOL_SIZE=0 CONFIG_STACK_CHECK_NONE=y # CONFIG_STACK_CHECK_NORM is not set # CONFIG_STACK_CHECK_STRONG is not set diff --git a/sdkconfig.defaults b/sdkconfig.defaults index c86f880..2d141da 100644 --- a/sdkconfig.defaults +++ b/sdkconfig.defaults @@ -1,3 +1,9 @@ +# +# Compiler options +# +CONFIG_COMPILER_CXX_EXCEPTIONS=y +CONFIG_LWIP_LOCAL_HOSTNAME="ChaosWatch" + # # Example Ethernet Configuration # @@ -21,15 +27,21 @@ CONFIG_EXAMPLE_ETH_SPI_PHY_ADDR0=1 # # Ethernet # -# CONFIG_ETH_SPI_ETHERNET_W5500 is not set +CONFIG_ETH_SPI_ETHERNET_W5500=y # # Wi-Fi # # CONFIG_ESP_WIFI_GMAC_SUPPORT is not set +# +# LWIP +# +CONFIG_LWIP_LOCAL_HOSTNAME="ChaosWatch" + # # Serial flasher config # CONFIG_ESPTOOLPY_FLASHSIZE_8MB=y CONFIG_ESPTOOLPY_FLASHSIZE="8MB" +