mimiclaw/main/proxy/http_proxy.c

#include "http_proxy.h"
#include "mimi_config.h"

#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>

#include "esp_log.h"
#include "nvs.h"
#include "esp_tls.h"
#include "esp_crt_bundle.h"

static const char *TAG = "proxy";

/* ── Config (cached from NVS) ─────────────────────────────────── */

static char     s_proxy_host[64] = {0};
static uint16_t s_proxy_port     = 0;

esp_err_t http_proxy_init(void)
{
    nvs_handle_t nvs;
    esp_err_t err = nvs_open(MIMI_NVS_PROXY, NVS_READONLY, &nvs);
    if (err == ESP_OK) {
        size_t len = sizeof(s_proxy_host);
        nvs_get_str(nvs, MIMI_NVS_KEY_PROXY_HOST, s_proxy_host, &len);
        nvs_get_u16(nvs, MIMI_NVS_KEY_PROXY_PORT, &s_proxy_port);
        nvs_close(nvs);
    }

    if (s_proxy_host[0] && s_proxy_port) {
        ESP_LOGI(TAG, "Proxy configured: %s:%d", s_proxy_host, s_proxy_port);
    } else {
        ESP_LOGI(TAG, "No proxy configured (direct connection)");
    }
    return ESP_OK;
}

bool http_proxy_is_enabled(void)
{
    return s_proxy_host[0] != '\0' && s_proxy_port != 0;
}

esp_err_t http_proxy_set(const char *host, uint16_t port)
{
    nvs_handle_t nvs;
    ESP_ERROR_CHECK(nvs_open(MIMI_NVS_PROXY, NVS_READWRITE, &nvs));
    ESP_ERROR_CHECK(nvs_set_str(nvs, MIMI_NVS_KEY_PROXY_HOST, host));
    ESP_ERROR_CHECK(nvs_set_u16(nvs, MIMI_NVS_KEY_PROXY_PORT, port));
    ESP_ERROR_CHECK(nvs_commit(nvs));
    nvs_close(nvs);

    strncpy(s_proxy_host, host, sizeof(s_proxy_host) - 1);
    s_proxy_port = port;
    ESP_LOGI(TAG, "Proxy set to %s:%d", s_proxy_host, s_proxy_port);
    return ESP_OK;
}

esp_err_t http_proxy_clear(void)
{
    nvs_handle_t nvs;
    ESP_ERROR_CHECK(nvs_open(MIMI_NVS_PROXY, NVS_READWRITE, &nvs));
    nvs_erase_key(nvs, MIMI_NVS_KEY_PROXY_HOST);
    nvs_erase_key(nvs, MIMI_NVS_KEY_PROXY_PORT);
    nvs_commit(nvs);
    nvs_close(nvs);

    s_proxy_host[0] = '\0';
    s_proxy_port = 0;
    ESP_LOGI(TAG, "Proxy cleared");
    return ESP_OK;
}

/* ── Proxied TLS connection ───────────────────────────────────── */

struct proxy_conn {
    int         sock;   /* raw TCP socket (for timeout control) */
    esp_tls_t  *tls;    /* esp_tls handle owns TLS + socket lifecycle */
};

/* Read a line from socket (up to CR-LF). Returns length or -1. */
static int sock_read_line(int fd, char *buf, int max, int timeout_ms)
{
    int pos = 0;
    struct timeval tv = { .tv_sec = timeout_ms / 1000, .tv_usec = (timeout_ms % 1000) * 1000 };
    setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));

    while (pos < max - 1) {
        char c;
        int r = recv(fd, &c, 1, 0);
        if (r <= 0) return -1;
        if (c == '\n') { buf[pos] = '\0'; return pos; }
        if (c != '\r') buf[pos++] = c;
    }
    buf[pos] = '\0';
    return pos;
}

/* Open TCP + CONNECT tunnel, returns socket fd or -1 */
static int open_connect_tunnel(const char *host, int port, int timeout_ms)
{
    struct addrinfo hints = { .ai_family = AF_INET, .ai_socktype = SOCK_STREAM };
    struct addrinfo *res = NULL;
    char port_str[8];
    snprintf(port_str, sizeof(port_str), "%d", s_proxy_port);

    if (getaddrinfo(s_proxy_host, port_str, &hints, &res) != 0 || !res) {
        ESP_LOGE(TAG, "DNS resolve failed for proxy %s", s_proxy_host);
        return -1;
    }

    int sock = socket(AF_INET, SOCK_STREAM, 0);
    if (sock < 0) { freeaddrinfo(res); return -1; }

    struct timeval tv = { .tv_sec = timeout_ms / 1000, .tv_usec = (timeout_ms % 1000) * 1000 };
    setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv));
    setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));

    if (connect(sock, res->ai_addr, res->ai_addrlen) != 0) {
        ESP_LOGE(TAG, "TCP connect to proxy %s:%d failed", s_proxy_host, s_proxy_port);
        freeaddrinfo(res); close(sock); return -1;
    }
    freeaddrinfo(res);
    ESP_LOGI(TAG, "Connected to proxy %s:%d", s_proxy_host, s_proxy_port);

    char req[256];
    int len = snprintf(req, sizeof(req),
        "CONNECT %s:%d HTTP/1.1\r\nHost: %s:%d\r\n\r\n", host, port, host, port);

    if (send(sock, req, len, 0) != len) {
        ESP_LOGE(TAG, "Failed to send CONNECT"); close(sock); return -1;
    }

    char line[256];
    if (sock_read_line(sock, line, sizeof(line), timeout_ms) < 0) {
        ESP_LOGE(TAG, "No response from proxy"); close(sock); return -1;
    }
    if (strstr(line, "200") == NULL) {
        ESP_LOGE(TAG, "CONNECT rejected: %s", line); close(sock); return -1;
    }

    /* Consume remaining response headers */
    while (sock_read_line(sock, line, sizeof(line), timeout_ms) > 0) { }

    ESP_LOGI(TAG, "CONNECT tunnel established to %s:%d", host, port);
    return sock;
}

proxy_conn_t *proxy_conn_open(const char *host, int port, int timeout_ms)
{
    if (!http_proxy_is_enabled()) {
        ESP_LOGE(TAG, "proxy_conn_open called but no proxy configured");
        return NULL;
    }

    int sock = open_connect_tunnel(host, port, timeout_ms);
    if (sock < 0) return NULL;

    proxy_conn_t *conn = calloc(1, sizeof(*conn));
    if (!conn) { close(sock); return NULL; }
    conn->sock = sock;

    /* ── TLS handshake via esp_tls over tunnel ───────────────── */
    conn->tls = esp_tls_init();
    if (!conn->tls) {
        ESP_LOGE(TAG, "esp_tls_init failed");
        close(sock); free(conn); return NULL;
    }

    /* Inject our CONNECT-tunnel socket and skip TCP connect phase */
    esp_tls_set_conn_sockfd(conn->tls, sock);
    esp_tls_set_conn_state(conn->tls, ESP_TLS_CONNECTING);

    esp_tls_cfg_t cfg = {
        .crt_bundle_attach = esp_crt_bundle_attach,
        .timeout_ms = timeout_ms,
    };

    int ret = esp_tls_conn_new_sync(host, strlen(host), port, &cfg, conn->tls);
    if (ret <= 0) {
        ESP_LOGE(TAG, "TLS handshake failed over proxy tunnel");
        esp_tls_conn_destroy(conn->tls);
        /* esp_tls_conn_destroy closes the socket */
        free(conn);
        return NULL;
    }

    ESP_LOGI(TAG, "TLS handshake OK with %s:%d via proxy", host, port);
    return conn;
}

int proxy_conn_write(proxy_conn_t *conn, const char *data, int len)
{
    int written = 0;
    while (written < len) {
        ssize_t ret = esp_tls_conn_write(conn->tls, data + written, len - written);
        if (ret > 0) {
            written += (int)ret;
        } else if (ret == ESP_TLS_ERR_SSL_WANT_WRITE) {
            continue;
        } else {
            ESP_LOGE(TAG, "esp_tls_conn_write error: %d", (int)ret);
            return -1;
        }
    }
    return written;
}

int proxy_conn_read(proxy_conn_t *conn, char *buf, int len, int timeout_ms)
{
    struct timeval tv = { .tv_sec = timeout_ms / 1000, .tv_usec = (timeout_ms % 1000) * 1000 };
    setsockopt(conn->sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));

    ssize_t ret = esp_tls_conn_read(conn->tls, buf, len);
    if (ret == ESP_TLS_ERR_SSL_WANT_READ) return 0;
    if (ret == 0) return 0;
    if (ret < 0) {
        ESP_LOGE(TAG, "esp_tls_conn_read error: %d", (int)ret);
        return -1;
    }
    return (int)ret;
}

void proxy_conn_close(proxy_conn_t *conn)
{
    if (!conn) return;
    if (conn->tls) {
        esp_tls_conn_destroy(conn->tls);
    }
    free(conn);
}