From: Sean Hefty <sean.hefty@intel.com>
Date: Thu, 16 Aug 2012 22:44:18 +0000 (-0700)
Subject: Refresh of waitall-ex
X-Git-Url: https://openfabrics.org/gitweb/?a=commitdiff_plain;h=f5723089e884738bcad1189898bcffca68d5450c;p=~shefty%2Flibrdmacm.git

Refresh of waitall-ex
---

diff --git a/examples/rstream.c b/examples/rstream.c
index befb7c66..1d221d0f 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)) {