summaryrefslogtreecommitdiffstats
path: root/transport/ib-verbs
diff options
context:
space:
mode:
authorAnand V. Avati <avati@amp.gluster.com>2009-04-18 01:09:31 +0530
committerAnand V. Avati <avati@amp.gluster.com>2009-04-18 01:09:31 +0530
commit307528ef9a33a4d294b3ea49bbf5f0b16b27aa0f (patch)
tree08aecf95d4615ea05dc1c78aacef5950dd647901 /transport/ib-verbs
parent50b2291e3b046ea6b8fdfcb7e206c772eae64f0e (diff)
disconnect transport peer if system is out of memory while allocating iobuf
Diffstat (limited to 'transport/ib-verbs')
-rw-r--r--transport/ib-verbs/src/ib-verbs.c14
1 files changed, 14 insertions, 0 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;
}