[PATCH 2/2] librdmacm/rstream: Use MSG_WAITALL for blocking test

[Hefty, Sean]

Signed-off-by: Sean Hefty <sean.he...@intel.com>
---
 examples/rstream.c |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/rstream.c b/examples/rstream.c
index befb7c6..1d221d0 100644
--- a/examples/rstream.c
+++ b/examples/rstream.c
@@ -607,7 +607,7 @@ static int set_test_opt(char *optarg)
                        use_async = 1;
                        break;
                case 'b':
-                       flags &= ~MSG_DONTWAIT;
+                       flags = (flags & ~MSG_DONTWAIT) | MSG_WAITALL;
                        break;
                case 'f':
                        use_fork = 1;
@@ -628,7 +628,7 @@ static int set_test_opt(char *optarg)
                } else if (!strncasecmp("async", optarg, 5)) {
                        use_async = 1;
                } else if (!strncasecmp("block", optarg, 5)) {
-                       flags &= ~MSG_DONTWAIT;
+                       flags = (flags & ~MSG_DONTWAIT) | MSG_WAITALL;
                } else if (!strncasecmp("nonblock", optarg, 8)) {
                        flags |= MSG_DONTWAIT;
                } else if (!strncasecmp("verify", optarg, 6)) {


N�����r��y����b�X��ǧv�^�)޺{.n�+����{��ٚ�{ay�ʇڙ�,j��f���h���z��w���
���j:+v���w�j�m��������zZ+�����ݢj"��!�i