diff options
Diffstat (limited to 'transport')
| -rw-r--r-- | transport/ib-verbs/src/ib-verbs.c | 14 | ||||
| -rw-r--r-- | transport/socket/src/socket.c | 16 |
2 files changed, 26 insertions, 4 deletions
diff --git a/transport/ib-verbs/src/ib-verbs.c b/transport/ib-verbs/src/ib-verbs.c index 5933e50fa3f..dc5ae11379c 100644 --- a/transport/ib-verbs/src/ib-verbs.c +++ b/transport/ib-verbs/src/ib-verbs.c @@ -525,6 +525,13 @@ ib_verbs_receive (transport_t *this, char **hdr_p, size_t *hdrlen_p, if (size1) { hdr = CALLOC (1, size1); + if (!hdr) { + gf_log (this->xl->name, GF_LOG_ERROR, + "unable to allocate header for peer %s", + this->peerinfo.identifier); + ret = -ENOMEM; + goto err; + } memcpy (hdr, copy_from, size1); copy_from += size1; *hdr_p = hdr; @@ -533,6 +540,13 @@ ib_verbs_receive (transport_t *this, char **hdr_p, size_t *hdrlen_p, if (size2) { iobuf = iobuf_get (this->xl->ctx->iobuf_pool); + if (!iobuf) { + gf_log (this->xl->name, GF_LOG_ERROR, + "unable to allocate IO buffer for peer %s", + this->peerinfo.identifier); + ret = -ENOMEM; + goto err; + } memcpy (iobuf->ptr, copy_from, size2); *iobuf_p = iobuf; } diff --git a/transport/socket/src/socket.c b/transport/socket/src/socket.c index 4c9338bfe20..b37c49090bf 100644 --- a/transport/socket/src/socket.c +++ b/transport/socket/src/socket.c @@ -510,6 +510,7 @@ __socket_proto_state_machine (transport_t *this) size_t size2 = 0; int previous_state = -1; struct socket_header *hdr = NULL; + struct iobuf *iobuf = NULL; priv = this->private; @@ -592,10 +593,17 @@ __socket_proto_state_machine (transport_t *this) if (size2) { /* TODO: sanity check size2 < page size */ - priv->incoming.iobuf = - iobuf_get (this->xl->ctx->iobuf_pool); - priv->incoming.buf_p = - priv->incoming.iobuf->ptr; + iobuf = iobuf_get (this->xl->ctx->iobuf_pool); + if (!iobuf) { + gf_log (this->xl->name, GF_LOG_ERROR, + "unable to allocate IO buffer " + "for peer %s", + this->peerinfo.identifier); + ret = -ENOMEM; + goto unlock; + } + priv->incoming.iobuf = iobuf; + priv->incoming.buf_p = iobuf->ptr; } priv->incoming.vector[0].iov_base = |