X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fmeta.c;h=e56c86f9124d585966759fb63b9a8ae5a4b6f267;hb=54c900e961de6065f607f5661edeb7c84be29ea5;hp=a3f7ef3cfa1f38c7fae38badbfcc11f28eb641a9;hpb=886a6f61a1f4cc48a77b42d10f34f9126377d904;p=tinc diff --git a/src/meta.c b/src/meta.c index a3f7ef3c..e56c86f9 100644 --- a/src/meta.c +++ b/src/meta.c @@ -42,24 +42,20 @@ bool send_meta(connection_t *c, const char *buffer, int length) { /* Add our data to buffer */ if(c->status.encryptout) { - char outbuf[length]; size_t outlen = length; - if(!cipher_encrypt(&c->outcipher, buffer, length, outbuf, &outlen, false) || outlen != length) { + if(!cipher_encrypt(&c->outcipher, buffer, length, buffer_prepare(&c->outbuf, length), &outlen, false) || outlen != length) { logger(LOG_ERR, "Error while encrypting metadata to %s (%s)", c->name, c->hostname); return false; } - - ifdebug(META) logger(LOG_DEBUG, "Encrypted write %p %p %p %d", c, c->buffer, outbuf, length); - bufferevent_write(c->buffer, (void *)outbuf, length); - ifdebug(META) logger(LOG_DEBUG, "Done."); + } else { - ifdebug(META) logger(LOG_DEBUG, "Unencrypted write %p %p %p %d", c, c->buffer, buffer, length); - bufferevent_write(c->buffer, (void *)buffer, length); - ifdebug(META) logger(LOG_DEBUG, "Done."); + buffer_add(&c->outbuf, buffer, length); } + event_add(&c->outevent, NULL); + return true; } @@ -89,7 +85,14 @@ bool receive_meta(connection_t *c) { - If not, keep stuff in buffer and exit. */ - inlen = recv(c->socket, inbuf, sizeof inbuf, 0); + buffer_compact(&c->inbuf, MAXBUFSIZE); + + if(sizeof inbuf <= c->inbuf.len) { + logger(LOG_ERR, "Input buffer full for %s (%s)\n", c, c->hostname); + return false; + } + + inlen = recv(c->socket, inbuf, sizeof inbuf - c->inbuf.len, 0); if(inlen <= 0) { if(!inlen || !errno) { @@ -111,33 +114,31 @@ bool receive_meta(connection_t *c) { else endp = bufp + inlen; - evbuffer_add(c->buffer->input, bufp, endp - bufp); + buffer_add(&c->inbuf, bufp, endp - bufp); inlen -= endp - bufp; bufp = endp; } else { size_t outlen = inlen; ifdebug(META) logger(LOG_DEBUG, "Received encrypted %d bytes", inlen); - evbuffer_expand(c->buffer->input, c->buffer->input->off + inlen); - if(!cipher_decrypt(&c->incipher, bufp, inlen, c->buffer->input->buffer + c->buffer->input->off, &outlen, false) || inlen != outlen) { + if(!cipher_decrypt(&c->incipher, bufp, inlen, buffer_prepare(&c->inbuf, inlen), &outlen, false) || inlen != outlen) { logger(LOG_ERR, "Error while decrypting metadata from %s (%s)", c->name, c->hostname); return false; } - c->buffer->input->off += inlen; inlen = 0; } - while(c->buffer->input->off) { + while(c->inbuf.len) { /* Are we receiving a TCPpacket? */ if(c->tcplen) { - if(c->tcplen <= c->buffer->input->off) { - receive_tcppacket(c, (char *)c->buffer->input->buffer, c->tcplen); - evbuffer_drain(c->buffer->input, c->tcplen); + char *tcpbuffer = buffer_read(&c->inbuf, c->tcplen); + if(tcpbuffer) { c->tcplen = 0; + receive_tcppacket(c, tcpbuffer, c->tcplen); continue; } else { break; @@ -146,10 +147,9 @@ bool receive_meta(connection_t *c) { /* Otherwise we are waiting for a request */ - char *request = evbuffer_readline(c->buffer->input); + char *request = buffer_readline(&c->inbuf); if(request) { bool result = receive_request(c, request); - free(request); if(!result) return false; continue;