librdmacm: Add rsocket to connect service on success too

Previously we only added it when rconnect() returned EINPROGRESS.
Now also add when connect succeeds so the progress thread can
drive state and handle disconnects.

Signed-off-by: Batsheva Black <bblack@nvidia.com>

Author: BatshevaBlack

librdmacm/rsocket.c

@@ -1727,9 +1727,11 @@ int rconnect(int socket, const struct sockaddr *addr, socklen_t addrlen)
 	if (rs->type == SOCK_STREAM) {
 		memcpy(&rs->cm_id->route.addr.dst_addr, addr, addrlen);
 		ret = rs_do_connect(rs);
-		if (ret == -1 && errno == EINPROGRESS) {
+		if (ret == 0 || (ret == -1 && errno == EINPROGRESS)) {
 			save_errno = errno;
-			/* The app can still drive the CM state on failure */
+			/* Add rsocket to internal thread that drives CM progress
+			 * so the app can drive state and respond to disconnect requests.
+			 */
 			rs_notify_svc(&connect_svc, rs, RS_SVC_ADD_CM);
 			errno = save_errno;
 		}