aljaz/ttyd
1
1
Fork 0
mirror of https://github.com/tsl0922/ttyd.git synced 2025-12-25 05:04:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

protocol: fix auth token check

Browse Source
This commit is contained in:
Shuanglei Tao
2021-03-06 17:37:30 +08:00
parent 96e88fb4ce
commit f67feebcc2
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/protocol.c
View File

@@ -284,16 +284,17 @@ int callback_tty(struct lws *wsi, enum lws_callback_reasons reason, void *user,
if (pss->process != NULL) break;
uint16_t columns = 0;
uint16_t rows = 0;
char *token = NULL;
json_object *obj = parse_window_size(pss->buffer, pss->len, &columns, &rows);
if (server->credential != NULL) {
struct json_object *o = NULL;
if (json_object_object_get_ex(obj, "AuthToken", &o)) {
const char *token = json_object_get_string(o);
pss->authenticated = token != NULL && !strcmp(token, server->credential);
if (token != NULL && !strcmp(token, server->credential))
pss->authenticated = true;
else
lwsl_warn("WS authentication failed with token: %s\n", token);
}
if (!pss->authenticated) {
lwsl_warn("WS authentication failed with token: %s\n", token);
json_object_put(obj);
lws_close_reason(wsi, LWS_CLOSE_STATUS_POLICY_VIOLATION, NULL, 0);
return -1;