mirror of
https://github.com/emsesp/EMS-ESP32.git
synced 2025-12-06 15:59:52 +03:00
json is always chunked
This commit is contained in:
Proddy
243
lib/PsychicHttp/src_old/PsychicWebSocket.cpp
Normal file
243
lib/PsychicHttp/src_old/PsychicWebSocket.cpp
Normal file
@@ -0,0 +1,243 @@
|
||||
#include "PsychicWebSocket.h"
|
||||
|
||||
/*************************************/
|
||||
/* PsychicWebSocketRequest */
|
||||
/*************************************/
|
||||
|
||||
PsychicWebSocketRequest::PsychicWebSocketRequest(PsychicRequest * req)
|
||||
: PsychicRequest(req->server(), req->request())
|
||||
, _client(req->client()) {
|
||||
}
|
||||
|
||||
PsychicWebSocketRequest::~PsychicWebSocketRequest() {
|
||||
}
|
||||
|
||||
PsychicWebSocketClient * PsychicWebSocketRequest::client() {
|
||||
return &_client;
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketRequest::reply(httpd_ws_frame_t * ws_pkt) {
|
||||
return httpd_ws_send_frame(this->_req, ws_pkt);
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketRequest::reply(httpd_ws_type_t op, const void * data, size_t len) {
|
||||
httpd_ws_frame_t ws_pkt;
|
||||
memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
|
||||
|
||||
ws_pkt.payload = (uint8_t *)data;
|
||||
ws_pkt.len = len;
|
||||
ws_pkt.type = op;
|
||||
|
||||
return this->reply(&ws_pkt);
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketRequest::reply(const char * buf) {
|
||||
return this->reply(HTTPD_WS_TYPE_TEXT, buf, strlen(buf));
|
||||
}
|
||||
|
||||
/*************************************/
|
||||
/* PsychicWebSocketClient */
|
||||
/*************************************/
|
||||
|
||||
PsychicWebSocketClient::PsychicWebSocketClient(PsychicClient * client)
|
||||
: PsychicClient(client->server(), client->socket()) {
|
||||
}
|
||||
|
||||
PsychicWebSocketClient::~PsychicWebSocketClient() {
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketClient::sendMessage(httpd_ws_frame_t * ws_pkt) {
|
||||
return httpd_ws_send_frame_async(this->server(), this->socket(), ws_pkt);
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketClient::sendMessage(httpd_ws_type_t op, const void * data, size_t len) {
|
||||
httpd_ws_frame_t ws_pkt;
|
||||
memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
|
||||
|
||||
ws_pkt.payload = (uint8_t *)data;
|
||||
ws_pkt.len = len;
|
||||
ws_pkt.type = op;
|
||||
|
||||
return this->sendMessage(&ws_pkt);
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketClient::sendMessage(const char * buf) {
|
||||
return this->sendMessage(HTTPD_WS_TYPE_TEXT, buf, strlen(buf));
|
||||
}
|
||||
|
||||
PsychicWebSocketHandler::PsychicWebSocketHandler()
|
||||
: PsychicHandler()
|
||||
, _onOpen(NULL)
|
||||
, _onFrame(NULL)
|
||||
, _onClose(NULL) {
|
||||
}
|
||||
|
||||
PsychicWebSocketHandler::~PsychicWebSocketHandler() {
|
||||
}
|
||||
|
||||
PsychicWebSocketClient * PsychicWebSocketHandler::getClient(int socket) {
|
||||
PsychicClient * client = PsychicHandler::getClient(socket);
|
||||
if (client == NULL)
|
||||
return NULL;
|
||||
|
||||
if (client->_friend == NULL) {
|
||||
DUMP(socket);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return (PsychicWebSocketClient *)client->_friend;
|
||||
}
|
||||
|
||||
PsychicWebSocketClient * PsychicWebSocketHandler::getClient(PsychicClient * client) {
|
||||
return getClient(client->socket());
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::addClient(PsychicClient * client) {
|
||||
client->_friend = new PsychicWebSocketClient(client);
|
||||
PsychicHandler::addClient(client);
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::removeClient(PsychicClient * client) {
|
||||
PsychicHandler::removeClient(client);
|
||||
delete (PsychicWebSocketClient *)client->_friend;
|
||||
client->_friend = NULL;
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::openCallback(PsychicClient * client) {
|
||||
PsychicWebSocketClient * buddy = getClient(client);
|
||||
if (buddy == NULL) {
|
||||
TRACE();
|
||||
return;
|
||||
}
|
||||
|
||||
if (_onOpen != NULL)
|
||||
_onOpen(getClient(buddy));
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::closeCallback(PsychicClient * client) {
|
||||
PsychicWebSocketClient * buddy = getClient(client);
|
||||
if (buddy == NULL) {
|
||||
TRACE();
|
||||
return;
|
||||
}
|
||||
|
||||
if (_onClose != NULL)
|
||||
_onClose(getClient(buddy));
|
||||
}
|
||||
|
||||
bool PsychicWebSocketHandler::isWebSocket() {
|
||||
return true;
|
||||
}
|
||||
|
||||
esp_err_t PsychicWebSocketHandler::handleRequest(PsychicRequest * request) {
|
||||
//lookup our client
|
||||
PsychicClient * client = checkForNewClient(request->client());
|
||||
|
||||
// beginning of the ws URI handler and our onConnect hook
|
||||
if (request->method() == HTTP_GET) {
|
||||
if (client->isNew)
|
||||
openCallback(client);
|
||||
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
//prep our request
|
||||
PsychicWebSocketRequest wsRequest(request);
|
||||
|
||||
//init our memory for storing the packet
|
||||
httpd_ws_frame_t ws_pkt;
|
||||
memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
|
||||
ws_pkt.type = HTTPD_WS_TYPE_TEXT;
|
||||
uint8_t * buf = NULL;
|
||||
|
||||
/* Set max_len = 0 to get the frame len */
|
||||
esp_err_t ret = httpd_ws_recv_frame(wsRequest.request(), &ws_pkt, 0);
|
||||
if (ret != ESP_OK) {
|
||||
ESP_LOGE(PH_TAG, "httpd_ws_recv_frame failed to get frame len with %s", esp_err_to_name(ret));
|
||||
return ret;
|
||||
}
|
||||
|
||||
//okay, now try to load the packet
|
||||
ESP_LOGI(PH_TAG, "frame len is %d", ws_pkt.len);
|
||||
if (ws_pkt.len) {
|
||||
/* ws_pkt.len + 1 is for NULL termination as we are expecting a string */
|
||||
buf = (uint8_t *)calloc(1, ws_pkt.len + 1);
|
||||
if (buf == NULL) {
|
||||
ESP_LOGE(PH_TAG, "Failed to calloc memory for buf");
|
||||
return ESP_ERR_NO_MEM;
|
||||
}
|
||||
ws_pkt.payload = buf;
|
||||
/* Set max_len = ws_pkt.len to get the frame payload */
|
||||
ret = httpd_ws_recv_frame(wsRequest.request(), &ws_pkt, ws_pkt.len);
|
||||
if (ret != ESP_OK) {
|
||||
ESP_LOGE(PH_TAG, "httpd_ws_recv_frame failed with %s", esp_err_to_name(ret));
|
||||
free(buf);
|
||||
return ret;
|
||||
}
|
||||
ESP_LOGI(PH_TAG, "Got packet with message: %s", ws_pkt.payload);
|
||||
}
|
||||
|
||||
// Text messages are our payload.
|
||||
if (ws_pkt.type == HTTPD_WS_TYPE_TEXT) {
|
||||
if (this->_onFrame != NULL)
|
||||
ret = this->_onFrame(&wsRequest, &ws_pkt);
|
||||
}
|
||||
|
||||
//logging housekeeping
|
||||
if (ret != ESP_OK)
|
||||
ESP_LOGE(PH_TAG, "httpd_ws_send_frame failed with %s", esp_err_to_name(ret));
|
||||
ESP_LOGI(PH_TAG,
|
||||
"ws_handler: httpd_handle_t=%p, sockfd=%d, client_info:%d",
|
||||
request->server(),
|
||||
httpd_req_to_sockfd(request->request()),
|
||||
httpd_ws_get_fd_info(request->server(), httpd_req_to_sockfd(request->request())));
|
||||
|
||||
//dont forget to release our buffer memory
|
||||
free(buf);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
PsychicWebSocketHandler * PsychicWebSocketHandler::onOpen(PsychicWebSocketClientCallback fn) {
|
||||
_onOpen = fn;
|
||||
return this;
|
||||
}
|
||||
|
||||
PsychicWebSocketHandler * PsychicWebSocketHandler::onFrame(PsychicWebSocketFrameCallback fn) {
|
||||
_onFrame = fn;
|
||||
return this;
|
||||
}
|
||||
|
||||
PsychicWebSocketHandler * PsychicWebSocketHandler::onClose(PsychicWebSocketClientCallback fn) {
|
||||
_onClose = fn;
|
||||
return this;
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::sendAll(httpd_ws_frame_t * ws_pkt) {
|
||||
for (PsychicClient * client : _clients) {
|
||||
ESP_LOGI(PH_TAG, "Active client (fd=%d) -> sending async message", client->socket());
|
||||
|
||||
if (client->_friend == NULL) {
|
||||
TRACE();
|
||||
return;
|
||||
}
|
||||
|
||||
if (((PsychicWebSocketClient *)client->_friend)->sendMessage(ws_pkt) != ESP_OK)
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::sendAll(httpd_ws_type_t op, const void * data, size_t len) {
|
||||
httpd_ws_frame_t ws_pkt;
|
||||
memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
|
||||
|
||||
ws_pkt.payload = (uint8_t *)data;
|
||||
ws_pkt.len = len;
|
||||
ws_pkt.type = op;
|
||||
|
||||
this->sendAll(&ws_pkt);
|
||||
}
|
||||
|
||||
void PsychicWebSocketHandler::sendAll(const char * buf) {
|
||||
this->sendAll(HTTPD_WS_TYPE_TEXT, buf, strlen(buf));
|
||||
}
|
||||
Reference in New Issue
Block a user