stream_client.cpp.hpp

Go to the documentation of this file.

#pragma once
 
#ifdef FASTLED_HAS_NETWORKING
 
#include "fl/net/http/stream_client.h"
#include "fl/stl/stdio.h"  // fl::snprintf — avoids _svfprintf_r (#2773 item 1.1)
#include "fl/stl/string.h"
#include "fl/stl/stdint.h"
#include "fl/stl/chrono.h"
// Note: fl/stl/cstdio.h intentionally NOT included — workaround for
// zackees/zccache#619 (Windows PCH path-spelling drift). This TU doesn't
// reference any cstdio.h symbol directly; the include was dead.
#include "fl/stl/thread.h"
#include "fl/stl/noexcept.h"
namespace fl {
namespace net {
namespace http {
 
HttpStreamClient::HttpStreamClient(const fl::string& host, u16 port, u32 heartbeatIntervalMs)
    : HttpStreamTransport(host, port, heartbeatIntervalMs)
    , mHttpHeaderSent(false)
    , mHttpHeaderReceived(false)
    , mHost(host)
    , mPort(port) {
    // Create native client with default connection config
    ConnectionConfig config;
    mNativeClient = fl::make_unique<NativeHttpClient>(mHost, mPort, config);
}
 
HttpStreamClient::~HttpStreamClient() FL_NOEXCEPT {
    disconnect();
}
 
bool HttpStreamClient::connect() {
    // If already connected, return true
    if (isConnected()) {
        return true;
    }
 
    // Reset HTTP state
    mHttpHeaderSent = false;
    mHttpHeaderReceived = false;
 
    // Connect native socket
    if (!mNativeClient->connect()) {
        return false;
    }
 
    // Send HTTP POST request header
    if (!sendHttpRequestHeader()) {
        mNativeClient->disconnect();
        return false;
    }
 
    // Read HTTP response header
    if (!readHttpResponseHeader()) {
        mNativeClient->disconnect();
        return false;
    }
 
    // Mark connection as established in base class
    mConnection.onConnected();
 
    return true;
}
 
void HttpStreamClient::disconnect() {
    if (mNativeClient) {
        mNativeClient->disconnect();
    }
    mHttpHeaderSent = false;
    mHttpHeaderReceived = false;
    mConnection.onDisconnected();
}
 
bool HttpStreamClient::isConnected() const {
    return mNativeClient && mNativeClient->isConnected() && mHttpHeaderSent && mHttpHeaderReceived;
}
 
int HttpStreamClient::sendData(fl::span<const u8> data) {
    if (!isConnected()) {
        return -1;
    }
    return mNativeClient->send(data);
}
 
int HttpStreamClient::recvData(fl::span<u8> buffer) {
    if (!isConnected()) {
        return -1;
    }
    return mNativeClient->recv(buffer);
}
 
void HttpStreamClient::triggerReconnect() {
    // Disconnect and let the base class reconnection logic handle it
    disconnect();
}
 
bool HttpStreamClient::sendHttpRequestHeader() {
    // Build HTTP POST request header
    // Format:
    // POST /rpc HTTP/1.1
    // Host: <host>:<port>
    // Content-Type: application/json
    // Transfer-Encoding: chunked
    // Connection: keep-alive
    // \r\n
 
    fl::string header;
    header.append("POST /rpc HTTP/1.1\r\n");
    header.append("Host: ");
    header.append(mHost);
    if (mPort != 80) {
        header.append(":");
        char portStr[8];
        fl::snprintf(portStr, sizeof(portStr), "%u", mPort);
        header.append(portStr);
    }
    header.append("\r\n");
    header.append("Content-Type: application/json\r\n");
    header.append("Transfer-Encoding: chunked\r\n");
    header.append("Connection: keep-alive\r\n");
    header.append("\r\n");
 
    // Send header
    int sent = mNativeClient->send(fl::span<const u8>(reinterpret_cast<const u8*>(header.c_str()), header.size())); // ok reinterpret cast
    if (sent != static_cast<int>(header.size())) {
        return false;
    }
 
    mHttpHeaderSent = true;
    return true;
}
 
bool HttpStreamClient::readHttpResponseHeader() {
    // Read HTTP response header
    // Expected format:
    // HTTP/1.1 200 OK
    // Content-Type: application/json
    // Transfer-Encoding: chunked
    // Connection: keep-alive
    // \r\n
 
    // Read until we find \r\n\r\n (end of headers)
    fl::string headerBuffer;
    u8 buffer[256];  // Read in chunks instead of byte-by-byte
 
    // Maximum header size: 4KB, max retries when no data (yield to server thread)
    const size_t MAX_HEADER_SIZE = 4096;
    const int MAX_READ_ATTEMPTS = 500;  // 500 yields before giving up
    int readAttempts = 0;
 
    while (headerBuffer.size() < MAX_HEADER_SIZE) {
        int received = mNativeClient->recv(buffer);
 
        if (received < 0) {
            return false;
        }
 
        if (received == 0) {
            // No data yet - server may still be processing the request
            // Yield to allow other threads (e.g. server thread) to run
            readAttempts++;
            if (readAttempts >= MAX_READ_ATTEMPTS) {
                return false;
            }
            // Sleep briefly to allow server thread to run.
            // The server thread sleeps 10ms between iterations, so we need
            // a real sleep (not just yield) to give it time to process.
            fl::this_thread::sleep_for(fl::chrono::milliseconds(10));  // ok sleep for
            continue;
        }
 
        headerBuffer.append(reinterpret_cast<const char*>(buffer), received); // ok reinterpret cast
 
        // Check for \r\n\r\n pattern
        if (headerBuffer.size() >= 4) {
            size_t pos = headerBuffer.find("\r\n\r\n");
            if (pos != fl::string::npos) {
                break;
            }
        }
    }
 
    // Validate response
    // Must start with "HTTP/1.1 200"
    if (headerBuffer.size() < 12) {
        return false;
    }
 
    if (headerBuffer.substr(0, 12) != "HTTP/1.1 200") {
        return false;
    }
 
    // Check for required headers (case-insensitive)
    // We're being lenient here - just check that the connection is valid
    // The chunked encoding will handle the actual data parsing
 
    // Look for Transfer-Encoding: chunked
    bool hasChunked = false;
    if (headerBuffer.find("Transfer-Encoding: chunked") != fl::string::npos ||
        headerBuffer.find("transfer-encoding: chunked") != fl::string::npos) {
        hasChunked = true;
    }
 
    if (!hasChunked) {
        return false;
    }
 
    mHttpHeaderReceived = true;
    return true;
}
 
}  // namespace http
}  // namespace net
}  // namespace fl
 
#endif // FASTLED_HAS_NETWORKING