summaryrefslogtreecommitdiffstats
path: root/rpc/rpc-transport/socket
diff options
context:
space:
mode:
authorGu Feng <flygoast@126.com>2014-09-09 18:00:22 +0800
committerRaghavendra G <rgowdapp@redhat.com>2014-09-19 05:37:05 -0700
commitfb6702b7f8ba19333b7ba4af543d908e3f5e1923 (patch)
treed3fdeb770bb7adb2f20f6ba579bc7139e54d1a2c /rpc/rpc-transport/socket
parent0ee0cfae80f7b17115b78d76610b995edb1bfa8b (diff)
socket: Fixed parsing RPC records containing multi fragments
In __socket_proto_state_machine(), when parsing RPC records containing multi fragments, just change the state of parsing process, had not processed the memory to coalesce the multi fragments. Change-Id: I5583e578603bd7290814a5d26885b31759c73115 BUG: 1139598 Signed-off-by: Gu Feng <flygoast@126.com> Reviewed-on: http://review.gluster.org/8662 Tested-by: Gluster Build System <jenkins@build.gluster.com> Reviewed-by: Niels de Vos <ndevos@redhat.com> Reviewed-by: Raghavendra G <rgowdapp@redhat.com> Tested-by: Raghavendra G <rgowdapp@redhat.com>
Diffstat (limited to 'rpc/rpc-transport/socket')
-rw-r--r--rpc/rpc-transport/socket/src/socket.c23
1 files changed, 20 insertions, 3 deletions
diff --git a/rpc/rpc-transport/socket/src/socket.c b/rpc/rpc-transport/socket/src/socket.c
index bd3f271cf1a..c1ad8ec9278 100644
--- a/rpc/rpc-transport/socket/src/socket.c
+++ b/rpc/rpc-transport/socket/src/socket.c
@@ -383,10 +383,11 @@ __socket_cached_read (rpc_transport_t *this, struct iovec *opvector, int opcount
/* first call after passing SP_STATE_READING_FRAGHDR */
in->ra_max = min (RPC_FRAGSIZE (in->fraghdr), GF_SOCKET_RA_MAX);
/* Note that the in->iobuf is the primary iobuf into which
- headers are read into. By using this itself as our
+ headers are read into, and in->frag.fragcurrent points to
+ some position in the buffer. By using this itself as our
read-ahead cache, we can avoid memory copies in iov_load
*/
- in->ra_buf = iobuf_ptr (in->iobuf);
+ in->ra_buf = in->frag.fragcurrent;
}
/* fill read-ahead */
@@ -2002,9 +2003,22 @@ __socket_proto_state_machine (rpc_transport_t *this,
goto out;
}
+ if (in->iobuf == NULL) {
+ /* first fragment */
+ frag->fragcurrent = iobuf_ptr (iobuf);
+ } else {
+ /* second or further fragment */
+ memcpy(iobuf_ptr (iobuf), iobuf_ptr (in->iobuf),
+ in->total_bytes_read - RPC_FRAGSIZE(in->fraghdr));
+ iobuf_unref (in->iobuf);
+ frag->fragcurrent = (char *) iobuf_ptr (iobuf) +
+ in->total_bytes_read - RPC_FRAGSIZE(in->fraghdr);
+ frag->pending_vector->iov_base = frag->fragcurrent;
+ in->pending_vector = frag->pending_vector;
+ }
+
in->iobuf = iobuf;
in->iobuf_size = 0;
- frag->fragcurrent = iobuf_ptr (iobuf);
in->record_state = SP_STATE_READING_FRAG;
/* fall through */
@@ -2019,6 +2033,9 @@ __socket_proto_state_machine (rpc_transport_t *this,
frag->bytes_read = 0;
if (!RPC_LASTFRAG (in->fraghdr)) {
+ in->pending_vector = in->vector;
+ in->pending_vector->iov_base = &in->fraghdr;
+ in->pending_vector->iov_len = sizeof(in->fraghdr);
in->record_state = SP_STATE_READING_FRAGHDR;
break;
}