From d69f340d498042fd0a4979259c73511d884ccd9a Mon Sep 17 00:00:00 2001 From: Razvan Deaconescu Date: Sun, 24 Oct 2010 17:33:41 +0300 Subject: [PATCH] test-socket-signal: update receive process --- Utils/test-socket-signal/peer.c | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/Utils/test-socket-signal/peer.c b/Utils/test-socket-signal/peer.c index 649660d..8895c19 100644 --- a/Utils/test-socket-signal/peer.c +++ b/Utils/test-socket-signal/peer.c @@ -69,7 +69,7 @@ static struct { .frequency = DEFAULT_FREQUENCY }; -static char rcv_buf[RECEIVE_BUFFER_SIZE]; +static char *rcv_buf; static char *snd_buf; /* connection socket */ @@ -90,12 +90,13 @@ static void init_buffer_random(char *buf, size_t len) static void init_buffers(void) { - init_buffer_random(rcv_buf, RECEIVE_BUFFER_SIZE); + init_buffer_random(rcv_buf, cmd_args.peer_buffer_size); init_buffer_random(snd_buf, PACKET_PAYLOAD_SIZE); } static void init(void) { + rcv_buf = malloc(cmd_args.peer_buffer_size * sizeof(char)); snd_buf = malloc(PACKET_SIZE * sizeof(char)); DIE(snd_buf == NULL, "malloc"); @@ -104,6 +105,7 @@ static void init(void) static void cleanup(void) { + free(rcv_buf); free(snd_buf); } @@ -135,7 +137,7 @@ static ssize_t receive_buffer(int sockfd) ssize_t n; while (nbytes < (ssize_t) cmd_args.peer_buffer_size) { - n = recv(sockfd, rcv_buf, RECEIVE_BUFFER_SIZE, 0); + n = recv(sockfd, rcv_buf, cmd_args.peer_buffer_size - nbytes, 0); if (n <= 0) break; nbytes += n; -- 2.20.1