-int receive_meta(connection_t *cl)
-{
- int x, l = sizeof(x);
- int oldlen, i;
- int lenin = 0;
- char inbuf[MAXBUFSIZE];
- int decrypted = 0;
-cp
- if(getsockopt(cl->meta_socket, SOL_SOCKET, SO_ERROR, &x, &l) < 0)
- {
- syslog(LOG_ERR, _("This is a bug: %s:%d: %d:%m %s (%s)"), __FILE__, __LINE__, cl->meta_socket,
- cl->name, cl->hostname);
- return -1;
- }
- if(x)
- {
- syslog(LOG_ERR, _("Metadata socket error for %s (%s): %s"),
- cl->name, cl->hostname, strerror(x));
- return -1;
- }
-
- lenin = read(cl->meta_socket, cl->buffer + cl->buflen, MAXBUFSIZE - cl->buflen);
-
- if(lenin<=0)
- {
- if(lenin==0)
- {
- if(debug_lvl >= DEBUG_CONNECTIONS)
- syslog(LOG_NOTICE, _("Connection closed by %s (%s)"),
- cl->name, cl->hostname);
- }
- else
- if(errno==EINTR)
- return 0;
- else
- syslog(LOG_ERR, _("Metadata socket read error for %s (%s): %m"),
- cl->name, cl->hostname);
-
- return -1;
- }
-
- oldlen = cl->buflen;
- cl->buflen += lenin;
-
- while(lenin)
- {
- if(cl->status.decryptin && !decrypted)
- {
- EVP_DecryptUpdate(cl->cipher_inctx, inbuf, &lenin, cl->buffer + oldlen, lenin);
- memcpy(cl->buffer + oldlen, inbuf, lenin);
- decrypted = 1;
- }
-
- cl->reqlen = 0;
-
- for(i = oldlen; i < cl->buflen; i++)
- {
- if(cl->buffer[i] == '\n')
- {
- cl->buffer[i] = 0; /* replace end-of-line by end-of-string so we can use sscanf */
- cl->reqlen = i + 1;
- break;
- }
- }
-
- if(cl->reqlen)
- {
- if(debug_lvl >= DEBUG_META)
- syslog(LOG_DEBUG, _("Got request from %s (%s): %s"),
- cl->name, cl->hostname, cl->buffer);
-
- if(receive_request(cl))
- return -1;
-
- cl->buflen -= cl->reqlen;
- lenin -= cl->reqlen;
- memmove(cl->buffer, cl->buffer + cl->reqlen, cl->buflen);
- oldlen = 0;
- }
- else
- {
- break;
- }
- }
-
- if(cl->buflen >= MAXBUFSIZE)
- {
- syslog(LOG_ERR, _("Metadata read buffer overflow for %s (%s)"),
- cl->name, cl->hostname);
- return -1;
- }
-
- cl->last_ping_time = time(NULL);
-cp
- return 0;
+static bool process_meta(connection_t *c, char *reqbuf, int *len) {
+ while(*len) {
+ if(c->tcplen) {
+ if(c->tcplen > *len)
+ break;
+
+ mutex_lock(&mutex);
+ receive_tcppacket(c, reqbuf, c->tcplen);
+ mutex_unlock(&mutex);
+
+ memmove(reqbuf, reqbuf, *len - c->tcplen);
+ *len -= c->tcplen;
+ } else {
+ char *end = memchr(reqbuf, '\n', *len);
+ if(!end)
+ break;
+ else
+ *end++ = 0;
+
+ mutex_lock(&mutex);
+ bool success = receive_request(c, reqbuf);
+ mutex_unlock(&mutex);
+
+ if(!success)
+ return false;
+
+ memmove(reqbuf, end, *len - (end - reqbuf));
+ *len -= end - reqbuf;
+ }
+ }
+
+ return true;
+}
+
+bool receive_meta(connection_t *c) {
+ int inlen;
+ int reqlen = 0;
+ char inbuf[MAXBUFSIZE];
+ char reqbuf[MAXBUFSIZE];
+
+ /* Strategy:
+ - Read as much as possible from the TCP socket in one go.
+ - Decrypt it if necessary.
+ - Check if a full request is in the request buffer.
+ - If yes, process request and remove it from the buffer, then check again.
+ - If not, try to read more.
+ */
+
+ while(true) {
+ inlen = recv(c->socket, inbuf, sizeof inbuf - reqlen, 0);
+
+ if(inlen <= 0) {
+ if(!inlen || !errno) {
+ ifdebug(CONNECTIONS) logger(LOG_NOTICE, "Connection closed by %s (%s)",
+ c->name, c->hostname);
+ } else if(sockwouldblock(sockerrno))
+ continue;
+ else
+ logger(LOG_ERR, "Metadata socket read error for %s (%s): %s",
+ c->name, c->hostname, sockstrerror(sockerrno));
+ return false;
+ }
+
+ while(inlen) {
+ if(!c->status.decryptin) {
+ char *end = memchr(inbuf, '\n', inlen);
+ if(!end)
+ end = inbuf + inlen;
+ else
+ end++;
+ memcpy(reqbuf + reqlen, inbuf, end - inbuf);
+ reqlen += end - inbuf;
+
+ if(!process_meta(c, reqbuf, &reqlen))
+ return false;
+
+ memmove(inbuf, end, inlen - (end - inbuf));
+ inlen -= end - inbuf;
+ } else {
+ size_t outlen = inlen;
+
+ if(!cipher_decrypt(&c->incipher, inbuf, inlen, reqbuf + reqlen, &outlen, false) || inlen != outlen) {
+ logger(LOG_ERR, "Error while decrypting metadata from %s (%s)", c->name, c->hostname);
+ return false;
+ }
+
+ reqlen += inlen;
+ inlen = 0;
+
+ if(!process_meta(c, reqbuf, &reqlen))
+ return false;
+ }
+ }
+ }