|
|
@ -64,12 +64,6 @@ struct client_instance { |
|
|
|
/* Is this the parent passthrough client */ |
|
|
|
/* Is this the parent passthrough client */ |
|
|
|
bool passthrough; |
|
|
|
bool passthrough; |
|
|
|
|
|
|
|
|
|
|
|
/* Does this client expect gz compression? */ |
|
|
|
|
|
|
|
bool gz; |
|
|
|
|
|
|
|
bool compressed; /* Currently receiving a compressed message */ |
|
|
|
|
|
|
|
unsigned long compsize; /* Expected compressed data size */ |
|
|
|
|
|
|
|
unsigned long decompsize; /* Expected decompressed data size */ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Linked list of shares in redirector mode.*/ |
|
|
|
/* Linked list of shares in redirector mode.*/ |
|
|
|
share_t *shares; |
|
|
|
share_t *shares; |
|
|
|
|
|
|
|
|
|
|
@ -485,34 +479,6 @@ retry: |
|
|
|
return; |
|
|
|
return; |
|
|
|
} |
|
|
|
} |
|
|
|
client->bufofs += ret; |
|
|
|
client->bufofs += ret; |
|
|
|
compressed: |
|
|
|
|
|
|
|
if (client->compressed) { |
|
|
|
|
|
|
|
unsigned long res; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (client->bufofs < client->compsize) |
|
|
|
|
|
|
|
goto retry; |
|
|
|
|
|
|
|
res = PAGESIZE - 4; |
|
|
|
|
|
|
|
if (unlikely(client->decompsize > res)) { |
|
|
|
|
|
|
|
LOGNOTICE("Client attempting to send oversize compressed message, disconnecting"); |
|
|
|
|
|
|
|
invalidate_client(ckp, cdata, client); |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
ret = uncompress((Bytef *)msg, &res, (Bytef *)client->buf, client->compsize); |
|
|
|
|
|
|
|
if (ret != Z_OK || res != client->decompsize) { |
|
|
|
|
|
|
|
LOGNOTICE("Failed to decompress %lu from %lu bytes in parse_client_msg, got %d", |
|
|
|
|
|
|
|
client->decompsize, client->compsize, ret); |
|
|
|
|
|
|
|
invalidate_client(ckp, cdata, client); |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
LOGDEBUG("Received client message compressed %lu from %lu", |
|
|
|
|
|
|
|
client->compsize, client->decompsize); |
|
|
|
|
|
|
|
msg[res] = '\0'; |
|
|
|
|
|
|
|
client->bufofs -= client->compsize; |
|
|
|
|
|
|
|
if (client->bufofs) |
|
|
|
|
|
|
|
memmove(client->buf, client->buf + buflen, client->bufofs); |
|
|
|
|
|
|
|
client->compressed = false; |
|
|
|
|
|
|
|
goto parse; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
reparse: |
|
|
|
reparse: |
|
|
|
eol = memchr(client->buf, '\n', client->bufofs); |
|
|
|
eol = memchr(client->buf, '\n', client->bufofs); |
|
|
|
if (!eol) |
|
|
|
if (!eol) |
|
|
@ -526,39 +492,11 @@ reparse: |
|
|
|
return; |
|
|
|
return; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/* Look for a compression header */ |
|
|
|
|
|
|
|
if (!strncmp(client->buf, gzip_magic, 3)) { |
|
|
|
|
|
|
|
uint32_t msglen; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Do we have the whole header? If not, keep reading */ |
|
|
|
|
|
|
|
if (client->bufofs < 12) |
|
|
|
|
|
|
|
goto retry; |
|
|
|
|
|
|
|
memcpy(&msglen, client->buf + 4, 4); |
|
|
|
|
|
|
|
client->compsize = le32toh(msglen); |
|
|
|
|
|
|
|
memcpy(&msglen, client->buf + 8, 4); |
|
|
|
|
|
|
|
client->decompsize = le32toh(msglen); |
|
|
|
|
|
|
|
if (unlikely(!client->compsize || !client->decompsize || |
|
|
|
|
|
|
|
client->compsize > MAX_MSGSIZE || client->decompsize > MAX_MSGSIZE)) { |
|
|
|
|
|
|
|
LOGNOTICE("Client id %"PRId64" invalid compressed message size %lu/%lu, disconnecting", |
|
|
|
|
|
|
|
client->id, client->compsize, client->decompsize); |
|
|
|
|
|
|
|
invalidate_client(ckp, cdata, client); |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
client->bufofs -= 12; |
|
|
|
|
|
|
|
if (client->bufofs > 0) |
|
|
|
|
|
|
|
memmove(client->buf, client->buf + 12, client->bufofs); |
|
|
|
|
|
|
|
client->compressed = true; |
|
|
|
|
|
|
|
if (client->bufofs >= client->compsize) |
|
|
|
|
|
|
|
goto compressed; |
|
|
|
|
|
|
|
goto retry; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
memcpy(msg, client->buf, buflen); |
|
|
|
memcpy(msg, client->buf, buflen); |
|
|
|
msg[buflen] = '\0'; |
|
|
|
msg[buflen] = '\0'; |
|
|
|
client->bufofs -= buflen; |
|
|
|
client->bufofs -= buflen; |
|
|
|
memmove(client->buf, client->buf + buflen, client->bufofs); |
|
|
|
memmove(client->buf, client->buf + buflen, client->bufofs); |
|
|
|
client->buf[client->bufofs] = '\0'; |
|
|
|
client->buf[client->bufofs] = '\0'; |
|
|
|
parse: |
|
|
|
|
|
|
|
if (!(val = json_loads(msg, 0, NULL))) { |
|
|
|
if (!(val = json_loads(msg, 0, NULL))) { |
|
|
|
char *buf = strdup("Invalid JSON, disconnecting\n"); |
|
|
|
char *buf = strdup("Invalid JSON, disconnecting\n"); |
|
|
|
|
|
|
|
|
|
|
@ -1006,36 +944,6 @@ static void send_client(cdata_t *cdata, const int64_t id, char *buf) |
|
|
|
test_redirector_shares(ckp, client, buf); |
|
|
|
test_redirector_shares(ckp, client, buf); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/* Does this client accept compressed data? Only compress if it's
|
|
|
|
|
|
|
|
* larger than one MTU. */ |
|
|
|
|
|
|
|
if (client->gz && len > 1492) { |
|
|
|
|
|
|
|
unsigned long compsize, decompsize = len; |
|
|
|
|
|
|
|
uint32_t msglen; |
|
|
|
|
|
|
|
Bytef *dest; |
|
|
|
|
|
|
|
int ret; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
compsize = round_up_page(len); |
|
|
|
|
|
|
|
dest = alloca(compsize); |
|
|
|
|
|
|
|
ret = compress(dest, &compsize, (Bytef *)buf, len); |
|
|
|
|
|
|
|
if (unlikely(ret != Z_OK)) { |
|
|
|
|
|
|
|
LOGWARNING("Failed to gz compress in send_client, got %d sending uncompressed", ret); |
|
|
|
|
|
|
|
goto out; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
if (unlikely(compsize + 12 >= decompsize)) |
|
|
|
|
|
|
|
goto out; |
|
|
|
|
|
|
|
/* Copy gz magic header */ |
|
|
|
|
|
|
|
memcpy(buf, gzip_magic, 4); |
|
|
|
|
|
|
|
/* Copy compressed message length */ |
|
|
|
|
|
|
|
msglen = htole32(compsize); |
|
|
|
|
|
|
|
memcpy(buf + 4, &msglen, 4); |
|
|
|
|
|
|
|
/* Copy decompressed message length */ |
|
|
|
|
|
|
|
msglen = htole32(decompsize); |
|
|
|
|
|
|
|
memcpy(buf + 8, &msglen, 4); |
|
|
|
|
|
|
|
memcpy(buf + 12, dest, compsize); |
|
|
|
|
|
|
|
len = compsize + 12; |
|
|
|
|
|
|
|
LOGDEBUG("Sending client message compressed %d from %lu", len, decompsize); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
out: |
|
|
|
|
|
|
|
sender_send = ckzalloc(sizeof(sender_send_t)); |
|
|
|
sender_send = ckzalloc(sizeof(sender_send_t)); |
|
|
|
sender_send->client = client; |
|
|
|
sender_send->client = client; |
|
|
|
sender_send->buf = buf; |
|
|
|
sender_send->buf = buf; |
|
|
@ -1065,7 +973,7 @@ static void passthrough_client(cdata_t *cdata, client_instance_t *client) |
|
|
|
|
|
|
|
|
|
|
|
LOGINFO("Connector adding passthrough client %"PRId64, client->id); |
|
|
|
LOGINFO("Connector adding passthrough client %"PRId64, client->id); |
|
|
|
client->passthrough = true; |
|
|
|
client->passthrough = true; |
|
|
|
ASPRINTF(&buf, "{\"result\": true, \"gz\": true}\n"); |
|
|
|
ASPRINTF(&buf, "{\"result\": true}\n"); |
|
|
|
send_client(cdata, client->id, buf); |
|
|
|
send_client(cdata, client->id, buf); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
@ -1242,15 +1150,10 @@ retry: |
|
|
|
sscanf(buf, "loglevel=%d", &ckp->loglevel); |
|
|
|
sscanf(buf, "loglevel=%d", &ckp->loglevel); |
|
|
|
} else if (cmdmatch(buf, "shutdown")) { |
|
|
|
} else if (cmdmatch(buf, "shutdown")) { |
|
|
|
goto out; |
|
|
|
goto out; |
|
|
|
} else if (cmdmatch(buf, "pass")) { |
|
|
|
} else if (cmdmatch(buf, "passthrough")) { |
|
|
|
client_instance_t *client; |
|
|
|
client_instance_t *client; |
|
|
|
bool gz = false; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (strstr(buf, "gz")) { |
|
|
|
ret = sscanf(buf, "passthrough=%"PRId64, &client_id); |
|
|
|
gz = true; |
|
|
|
|
|
|
|
ret = sscanf(buf, "passgz=%"PRId64, &client_id); |
|
|
|
|
|
|
|
} else |
|
|
|
|
|
|
|
ret = sscanf(buf, "passthrough=%"PRId64, &client_id); |
|
|
|
|
|
|
|
if (ret < 0) { |
|
|
|
if (ret < 0) { |
|
|
|
LOGDEBUG("Connector failed to parse passthrough command: %s", buf); |
|
|
|
LOGDEBUG("Connector failed to parse passthrough command: %s", buf); |
|
|
|
goto retry; |
|
|
|
goto retry; |
|
|
@ -1260,7 +1163,6 @@ retry: |
|
|
|
LOGINFO("Connector failed to find client id %"PRId64" to pass through", client_id); |
|
|
|
LOGINFO("Connector failed to find client id %"PRId64" to pass through", client_id); |
|
|
|
goto retry; |
|
|
|
goto retry; |
|
|
|
} |
|
|
|
} |
|
|
|
client->gz = gz; |
|
|
|
|
|
|
|
passthrough_client(cdata, client); |
|
|
|
passthrough_client(cdata, client); |
|
|
|
dec_instance_ref(cdata, client); |
|
|
|
dec_instance_ref(cdata, client); |
|
|
|
} else if (cmdmatch(buf, "getxfd")) { |
|
|
|
} else if (cmdmatch(buf, "getxfd")) { |
|
|
|