From 53b41316a9900eeb58c6e7d4d41631a1c880da63 Mon Sep 17 00:00:00 2001
From: Sean Hefty <sean.hefty@intel.com>
Date: Tue, 26 Jun 2012 16:57:50 -0700
Subject: [PATCH] Refresh of shut_wr

---
 src/rsocket.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/rsocket.c b/src/rsocket.c
index 5fc54046..4eb1ce1e 100644
--- a/src/rsocket.c
+++ b/src/rsocket.c
@@ -888,6 +888,8 @@ static int rs_poll_cq(struct rsocket *rs)
 				if (rs_msg_data(imm_data) == RS_CTRL_DISCONNECT) {
 					rs->state = rs_disconnected;
 					return ERR(ECONNRESET);
+				} else if (rs_msg_data(imm_data) == RS_CTRL_SHUTDOWN) {
+					rs->state &= ~rs_connect_rd;
 				}
 				break;
 			default:
-- 
2.45.2