mirror of
https://github.com/sle118/squeezelite-esp32.git
synced 2025-12-12 22:47:15 +03:00
436 lines
14 KiB
C++
436 lines
14 KiB
C++
#include "HTTPServer.h"
|
|
|
|
bell::HTTPServer::HTTPServer(int serverPort) { this->serverPort = serverPort; }
|
|
|
|
unsigned char bell::HTTPServer::h2int(char c) {
|
|
if (c >= '0' && c <= '9') {
|
|
return ((unsigned char)c - '0');
|
|
}
|
|
if (c >= 'a' && c <= 'f') {
|
|
return ((unsigned char)c - 'a' + 10);
|
|
}
|
|
if (c >= 'A' && c <= 'F') {
|
|
return ((unsigned char)c - 'A' + 10);
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
std::string bell::HTTPServer::urlDecode(std::string str) {
|
|
std::string encodedString = "";
|
|
char c;
|
|
char code0;
|
|
char code1;
|
|
for (int i = 0; i < str.length(); i++) {
|
|
c = str[i];
|
|
if (c == '+') {
|
|
encodedString += ' ';
|
|
} else if (c == '%') {
|
|
i++;
|
|
code0 = str[i];
|
|
i++;
|
|
code1 = str[i];
|
|
c = (h2int(code0) << 4) | h2int(code1);
|
|
encodedString += c;
|
|
} else {
|
|
|
|
encodedString += c;
|
|
}
|
|
}
|
|
|
|
return encodedString;
|
|
}
|
|
|
|
std::vector<std::string> bell::HTTPServer::splitUrl(const std::string &url,
|
|
char delimiter) {
|
|
std::stringstream ssb(url);
|
|
std::string segment;
|
|
std::vector<std::string> seglist;
|
|
|
|
while (std::getline(ssb, segment, delimiter)) {
|
|
seglist.push_back(segment);
|
|
}
|
|
return seglist;
|
|
}
|
|
|
|
void bell::HTTPServer::registerHandler(RequestType requestType,
|
|
const std::string &routeUrl,
|
|
httpHandler handler) {
|
|
if (routes.find(routeUrl) == routes.end()) {
|
|
routes.insert({routeUrl, std::vector<HTTPRoute>()});
|
|
}
|
|
this->routes[routeUrl].push_back(HTTPRoute{
|
|
.requestType = requestType,
|
|
.handler = handler,
|
|
});
|
|
}
|
|
|
|
void bell::HTTPServer::listen() {
|
|
BELL_LOG(info, "http", "Starting configuration server at port %d",
|
|
this->serverPort);
|
|
|
|
// setup address
|
|
struct addrinfo hints, *server;
|
|
memset(&hints, 0, sizeof hints);
|
|
hints.ai_family = AF_INET;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
hints.ai_flags = AI_PASSIVE;
|
|
getaddrinfo(NULL, std::to_string(serverPort).c_str(), &hints, &server);
|
|
|
|
int sockfd =
|
|
socket(server->ai_family, server->ai_socktype, server->ai_protocol);
|
|
struct sockaddr_in clientname;
|
|
socklen_t incomingSockSize;
|
|
int i;
|
|
int yes = true;
|
|
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
|
|
bind(sockfd, server->ai_addr, server->ai_addrlen);
|
|
::listen(sockfd, 10);
|
|
|
|
FD_ZERO(&activeFdSet);
|
|
FD_SET(sockfd, &activeFdSet);
|
|
|
|
for (;;) {
|
|
/* Block until input arrives on one or more active sockets. */
|
|
readFdSet = activeFdSet;
|
|
struct timeval tv = {0, 100000};
|
|
|
|
if (select(FD_SETSIZE, &readFdSet, NULL, NULL, &tv) < 0) {
|
|
BELL_LOG(error, "http", "Error in select");
|
|
perror("select");
|
|
// exit(EXIT_FAILURE);
|
|
}
|
|
|
|
/* Service all the sockets with input pending. */
|
|
for (i = 0; i < FD_SETSIZE; ++i)
|
|
if (FD_ISSET(i, &readFdSet)) {
|
|
if (i == sockfd) {
|
|
/* Connection request on original socket. */
|
|
int newFd;
|
|
incomingSockSize = sizeof(clientname);
|
|
newFd = accept(sockfd, (struct sockaddr *)&clientname,
|
|
&incomingSockSize);
|
|
if (newFd < 0) {
|
|
perror("accept");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
FD_SET(newFd, &activeFdSet);
|
|
|
|
HTTPConnection conn = {.buffer = std::vector<uint8_t>(128),
|
|
.httpMethod = ""};
|
|
|
|
this->connections.insert({newFd, conn});
|
|
} else {
|
|
/* Data arriving on an already-connected socket. */
|
|
readFromClient(i);
|
|
}
|
|
}
|
|
|
|
for (auto it = this->connections.cbegin();
|
|
it != this->connections.cend() /* not hoisted */;
|
|
/* no increment */) {
|
|
if ((*it).second.toBeClosed) {
|
|
close((*it).first);
|
|
FD_CLR((*it).first, &activeFdSet);
|
|
this->connections.erase(
|
|
it++); // or "it = m.erase(it)" since C++11
|
|
} else {
|
|
++it;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void bell::HTTPServer::readFromClient(int clientFd) {
|
|
HTTPConnection &conn = this->connections[clientFd];
|
|
|
|
int nbytes = recv(clientFd, &conn.buffer[0], conn.buffer.size(), 0);
|
|
if (nbytes < 0) {
|
|
BELL_LOG(error, "http", "Error reading from client");
|
|
perror("recv");
|
|
this->closeConnection(clientFd);
|
|
} else if (nbytes == 0) {
|
|
this->closeConnection(clientFd);
|
|
} else {
|
|
conn.currentLine +=
|
|
std::string(conn.buffer.data(), conn.buffer.data() + nbytes);
|
|
READBODY:
|
|
if (!conn.isReadingBody) {
|
|
while (conn.currentLine.find("\r\n") != std::string::npos) {
|
|
auto line =
|
|
conn.currentLine.substr(0, conn.currentLine.find("\r\n"));
|
|
conn.currentLine = conn.currentLine.substr(
|
|
conn.currentLine.find("\r\n") + 2, conn.currentLine.size());
|
|
if (line.find("GET ") != std::string::npos ||
|
|
line.find("POST ") != std::string::npos ||
|
|
line.find("OPTIONS ") != std::string::npos) {
|
|
conn.httpMethod = line;
|
|
}
|
|
|
|
if (line.find("Content-Length: ") != std::string::npos) {
|
|
conn.contentLength =
|
|
std::stoi(line.substr(16, line.size() - 1));
|
|
BELL_LOG(info, "http", "Content-Length: %d",
|
|
conn.contentLength);
|
|
}
|
|
if (line.size() == 0) {
|
|
if (conn.contentLength != 0) {
|
|
conn.isReadingBody = true;
|
|
goto READBODY;
|
|
} else {
|
|
findAndHandleRoute(conn.httpMethod, conn.currentLine,
|
|
clientFd);
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if (conn.currentLine.size() >= conn.contentLength) {
|
|
findAndHandleRoute(conn.httpMethod, conn.currentLine, clientFd);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void bell::HTTPServer::closeConnection(int connection) {
|
|
|
|
this->connections[connection].toBeClosed = true;
|
|
}
|
|
|
|
void bell::HTTPServer::writeResponseEvents(int connFd) {
|
|
std::lock_guard lock(this->responseMutex);
|
|
|
|
std::stringstream stream;
|
|
stream << "HTTP/1.1 200 OK\r\n";
|
|
stream << "Server: EUPHONIUM\r\n";
|
|
stream << "Connection: keep-alive\r\n";
|
|
stream << "Content-type: text/event-stream\r\n";
|
|
stream << "Cache-Control: no-cache\r\n";
|
|
stream << "Access-Control-Allow-Origin: *\r\n";
|
|
stream << "Access-Control-Allow-Methods: GET, POST, PATCH, PUT, DELETE, "
|
|
"OPTIONS\r\n";
|
|
stream << "Access-Control-Allow-Headers: Origin, Content-Type, "
|
|
"X-Auth-Token\r\n";
|
|
stream << "\r\n";
|
|
|
|
auto responseStr = stream.str();
|
|
|
|
write(connFd, responseStr.c_str(), responseStr.size());
|
|
this->connections[connFd].isEventConnection = true;
|
|
}
|
|
|
|
void bell::HTTPServer::writeResponse(const HTTPResponse &response) {
|
|
std::lock_guard lock(this->responseMutex);
|
|
|
|
auto fileSize = response.body.size();
|
|
|
|
if (response.responseReader != nullptr) {
|
|
fileSize = response.responseReader->getTotalSize();
|
|
}
|
|
|
|
std::stringstream stream;
|
|
stream << "HTTP/1.1 " << response.status << " OK\r\n";
|
|
stream << "Server: EUPHONIUM\r\n";
|
|
stream << "Connection: close\r\n";
|
|
stream << "Content-type: " << response.contentType << "\r\n";
|
|
|
|
if (response.useGzip) {
|
|
stream << "Content-encoding: gzip"
|
|
<< "\r\n";
|
|
}
|
|
|
|
stream << "Access-Control-Allow-Origin: *\r\n";
|
|
stream << "Access-Control-Allow-Methods: GET, POST, DELETE, OPTIONS\r\n";
|
|
stream << "Access-Control-Expose-Headers: Location\r\n";
|
|
stream << "Access-Control-Allow-Headers: Origin, Content-Type, "
|
|
"X-Auth-Token\r\n";
|
|
|
|
// go over every item in request->extraHeaders
|
|
for (auto &header : response.extraHeaders) {
|
|
stream << header << "\r\n";
|
|
}
|
|
|
|
stream << "\r\n";
|
|
|
|
if (response.body.size() > 0) {
|
|
stream << response.body;
|
|
}
|
|
|
|
auto responseStr = stream.str();
|
|
|
|
write(response.connectionFd, responseStr.c_str(), responseStr.size());
|
|
|
|
if (response.responseReader != nullptr) {
|
|
size_t read;
|
|
do {
|
|
read = response.responseReader->read(responseBuffer.data(),
|
|
responseBuffer.size());
|
|
if (read > 0) {
|
|
write(response.connectionFd, responseBuffer.data(), read);
|
|
}
|
|
} while (read > 0);
|
|
}
|
|
|
|
BELL_LOG(info, "HTTP", "Closing connection");
|
|
this->closeConnection(response.connectionFd);
|
|
}
|
|
|
|
void bell::HTTPServer::respond(const HTTPResponse &response) {
|
|
writeResponse(response);
|
|
}
|
|
|
|
void bell::HTTPServer::redirectTo(const std::string & url, int connectionFd) {
|
|
std::lock_guard lock(this->responseMutex);
|
|
std::stringstream stream;
|
|
stream << "HTTP/1.1 301 Moved Permanently\r\n";
|
|
stream << "Server: EUPHONIUM\r\n";
|
|
stream << "Connection: close\r\n";
|
|
stream << "Location: " << url << "\r\n\r\n";
|
|
auto responseStr = stream.str();
|
|
|
|
write(connectionFd, responseStr.c_str(), responseStr.size());
|
|
this->closeConnection(connectionFd);
|
|
}
|
|
|
|
void bell::HTTPServer::publishEvent(std::string eventName,
|
|
std::string eventData) {
|
|
std::lock_guard lock(this->responseMutex);
|
|
BELL_LOG(info, "http", "Publishing event");
|
|
|
|
std::stringstream stream;
|
|
stream << "event: " << eventName << "\n";
|
|
stream << "data: " << eventData << "\n\n";
|
|
auto responseStr = stream.str();
|
|
|
|
// Reply to all event-connections
|
|
for (auto it = this->connections.cbegin(); it != this->connections.cend();
|
|
++it) {
|
|
if ((*it).second.isEventConnection) {
|
|
write(it->first, responseStr.c_str(), responseStr.size());
|
|
}
|
|
}
|
|
}
|
|
|
|
std::map<std::string, std::string>
|
|
bell::HTTPServer::parseQueryString(const std::string &queryString) {
|
|
std::map<std::string, std::string> query;
|
|
auto prefixedString = "&" + queryString;
|
|
while (prefixedString.find("&") != std::string::npos) {
|
|
auto keyStart = prefixedString.find("&");
|
|
auto keyEnd = prefixedString.find("=");
|
|
// Find second occurence of "&" in prefixedString
|
|
auto valueEnd = prefixedString.find("&", keyStart + 1);
|
|
if (valueEnd == std::string::npos) {
|
|
valueEnd = prefixedString.size();
|
|
}
|
|
|
|
auto key = prefixedString.substr(keyStart + 1, keyEnd - 1);
|
|
auto value = prefixedString.substr(keyEnd + 1, valueEnd - keyEnd - 1);
|
|
query[key] = urlDecode(value);
|
|
prefixedString = prefixedString.substr(valueEnd);
|
|
}
|
|
|
|
return query;
|
|
}
|
|
|
|
void bell::HTTPServer::findAndHandleRoute(std::string &url, std::string &body,
|
|
int connectionFd) {
|
|
std::map<std::string, std::string> pathParams;
|
|
std::map<std::string, std::string> queryParams;
|
|
BELL_LOG(info, "http", "URL %s", url.c_str());
|
|
|
|
if (url.find("OPTIONS /") != std::string::npos) {
|
|
std::stringstream stream;
|
|
stream << "HTTP/1.1 200 OK\r\n";
|
|
stream << "Server: EUPHONIUM\r\n";
|
|
stream << "Allow: OPTIONS, GET, HEAD, POST\r\n";
|
|
stream << "Connection: close\r\n";
|
|
stream << "Access-Control-Allow-Origin: *\r\n";
|
|
stream << "Access-Control-Allow-Methods: GET, POST, OPTIONS\r\n";
|
|
stream << "Access-Control-Allow-Headers: Origin, Content-Type, "
|
|
"X-Auth-Token\r\n";
|
|
stream << "\r\n";
|
|
|
|
auto responseStr = stream.str();
|
|
|
|
write(connectionFd, responseStr.c_str(), responseStr.size());
|
|
closeConnection(connectionFd);
|
|
return;
|
|
}
|
|
|
|
if (url.find("GET /events") != std::string::npos) {
|
|
// Handle SSE endpoint here
|
|
writeResponseEvents(connectionFd);
|
|
return;
|
|
}
|
|
|
|
for (const auto &routeSet : this->routes) {
|
|
for (const auto &route : routeSet.second) {
|
|
|
|
std::string path = url;
|
|
if (url.find("GET ") != std::string::npos &&
|
|
route.requestType == RequestType::GET) {
|
|
path = path.substr(4);
|
|
} else if (url.find("POST ") != std::string::npos &&
|
|
route.requestType == RequestType::POST) {
|
|
path = path.substr(5);
|
|
} else {
|
|
continue;
|
|
}
|
|
|
|
path = path.substr(0, path.find(" "));
|
|
|
|
if (path.find("?") != std::string::npos) {
|
|
auto urlEncodedSplit = splitUrl(path, '?');
|
|
path = urlEncodedSplit[0];
|
|
queryParams = this->parseQueryString(urlEncodedSplit[1]);
|
|
}
|
|
|
|
auto routeSplit = splitUrl(routeSet.first, '/');
|
|
auto urlSplit = splitUrl(path, '/');
|
|
bool matches = true;
|
|
|
|
pathParams.clear();
|
|
|
|
if (routeSplit.size() == urlSplit.size()) {
|
|
for (int x = 0; x < routeSplit.size(); x++) {
|
|
if (routeSplit[x] != urlSplit[x]) {
|
|
if (routeSplit[x][0] == ':') {
|
|
pathParams.insert(
|
|
{routeSplit[x].substr(1), urlSplit[x]});
|
|
} else {
|
|
matches = false;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
matches = false;
|
|
}
|
|
|
|
if (routeSplit.back().find("*") != std::string::npos &&
|
|
urlSplit[1] == routeSplit[1]) {
|
|
matches = true;
|
|
}
|
|
|
|
if (matches) {
|
|
if (body.find("&") != std::string::npos) {
|
|
queryParams = this->parseQueryString(body);
|
|
}
|
|
|
|
HTTPRequest req = {.urlParams = pathParams,
|
|
.queryParams = queryParams,
|
|
.body = body,
|
|
.handlerId = 0,
|
|
.connection = connectionFd};
|
|
|
|
route.handler(req);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
writeResponse(HTTPResponse{
|
|
.connectionFd = connectionFd,
|
|
.status = 404,
|
|
.body = "Not found",
|
|
});
|
|
}
|