lorenz.stechauner/sesimos
1
1
Fork 0
Code Issues Pull Requests Projects Releases Activity

Try #1 Websocket

Browse Source
This commit is contained in:
Lorenz Stechauner
2020-12-01 19:48:34 +01:00
parent a1f02dba49
commit ce9e3c33f8
1 changed files with 37 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
src/client.cpp
View File

@ -209,6 +209,21 @@ long getPosition(std::string str, char c, int occurence) {
return -1; return -1;
} }
int websocket_handler(Socket *socket, pid_t *childpid, stds *pipes) {
fd_set readfd;
int maxfd = (socket->getFd() > pipes->stdout->_fileno) ? socket->getFd() : pipes->stdout->_fileno;
FD_ZERO(&readfd);
FD_SET(socket->getFd(), &readfd);
FD_SET(pipes->stdout->_fileno, &readfd);
while (true) {
int ret = ::select(maxfd + 1, &readfd, nullptr, nullptr, nullptr);
if (ret < 0) {
throw (char *) strerror(errno);
}
}
}
/** /**
* Handles (keep-alive) HTTP connections * Handles (keep-alive) HTTP connections
* @param prefix The connection prefix * @param prefix The connection prefix
@ -245,7 +260,7 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
} }
try { try {
bool noRedirect, redir, invalidMethod, etag, compress; bool noRedirect, redir, invalidMethod, etag, compress, websocket;
URI path; URI path;
pid_t childpid; pid_t childpid;
FILE *file; FILE *file;
@ -253,6 +268,7 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
string hash, type, host; string hash, type, host;
thread *t; thread *t;
long pos; long pos;
stds pipes;
if (req.isExistingField("Connection") && req.getField("Connection") == "keep-alive") { if (req.isExistingField("Connection") && req.getField("Connection") == "keep-alive") {
req.setField("Connection", "keep-alive"); req.setField("Connection", "keep-alive");
@ -400,7 +416,7 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
" GATEWAY_INTERFACE=" + cli_encode("CGI/1.1") + " GATEWAY_INTERFACE=" + cli_encode("CGI/1.1") +
" /usr/bin/php-cgi"; " /usr/bin/php-cgi";
stds pipes = procopen(cmd.c_str()); pipes = procopen(cmd.c_str());
childpid = pipes.pid; childpid = pipes.pid;
long len = req.isExistingField("Content-Length") long len = req.isExistingField("Content-Length")
@ -408,8 +424,6 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
: ((req.getMethod() == "POST" || req.getMethod() == "PUT") ? -1 : 0); : ((req.getMethod() == "POST" || req.getMethod() == "PUT") ? -1 : 0);
socket->receive(pipes.stdin, len); socket->receive(pipes.stdin, len);
fclose(pipes.stdin);
t = new thread(php_error_handler, prefix, pipes.stderr); t = new thread(php_error_handler, prefix, pipes.stderr);
string line; string line;
@ -435,15 +449,23 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
} }
} }
websocket = statuscode == 101 && req.isExistingResponseField("Connection") && req.getResponseField("Connection") == "upgrade";
fclose(file); fclose(file);
file = pipes.stdout; file = pipes.stdout;
int c = fgetc(pipes.stdout); if (websocket) {
if (c == -1) { req.respond(statuscode);
// No Data -> Error
req.respond((statuscode == 0) ? 500 : statuscode);
goto respond; goto respond;
} else { } else {
ungetc(c, pipes.stdout); fclose(pipes.stdin);
int c = fgetc(pipes.stdout);
if (c == -1) {
// No Data -> Error
req.respond((statuscode == 0) ? 500 : statuscode);
goto respond;
} else {
ungetc(c, pipes.stdout);
}
} }
} }
@ -508,8 +530,8 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
int code = status.code; int code = status.code;
string color; string color;
string comment; string comment;
if ((code >= 200 && code < 300) || code == 304) { if ((code >= 200 && code < 300) || code == 304 || code == 101) {
color = "\x1B[1;32m"; // Success (Cached): Green color = "\x1B[1;32m"; // Success (Cached, Switching Protocols): Green
} else if (code >= 100 && code < 200) { } else if (code >= 100 && code < 200) {
color = "\x1B[1;93m"; // Continue: Yellow color = "\x1B[1;93m"; // Continue: Yellow
} else if (code >= 300 && code < 400) { } else if (code >= 300 && code < 400) {
@ -528,6 +550,10 @@ bool connection_handler(const char *preprefix, const char *col1, const char *col
if (!host.empty()) { if (!host.empty()) {
log_to_file(prefix, msg, host); log_to_file(prefix, msg, host);
} }
if (websocket) {
websocket_handler(socket, &childpid, &pipes);
}
} catch (char *msg) { } catch (char *msg) {
HttpStatusCode status = req.getStatusCode(); HttpStatusCode status = req.getStatusCode();
log(prefix, to_string(status.code) + " " + status.message + " (" + formatTime(req.getDuration()) + ")"); log(prefix, to_string(status.code) + " " + status.message + " (" + formatTime(req.getDuration()) + ")");