prov/psm,psm2: complete the memory corruption fix for fi_trywait()

Fix a compilation error in commit e74a695 "PSM2: fix memory corruption in fi_trywait() for FI_CLASS_CQ". Apply simiar changes to the psm provider. Signed-off-by: Jianxin Xiong <[email protected]>
hoopoepg · Jun 30, 2016 · 724bd2c · 724bd2c
1 parent 136b7ba
commit 724bd2c
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 3 deletions.
diff --git a/prov/psm/src/psmx.h b/prov/psm/src/psmx.h
@@ -558,6 +558,8 @@ int	psmx_domain_open(struct fid_fabric *fabric, struct fi_info *info,
 			 struct fid_domain **domain, void *context);
 int	psmx_wait_open(struct fid_fabric *fabric, struct fi_wait_attr *attr,
 		       struct fid_wait **waitset);
+int	psmx_wait_trywait(struct fid_fabric *fabric, struct fid **fids,
+			  int count);
 int	psmx_ep_open(struct fid_domain *domain, struct fi_info *info,
 		     struct fid_ep **ep, void *context);
 int	psmx_stx_ctx(struct fid_domain *domain, struct fi_tx_attr *attr,

diff --git a/prov/psm/src/psmx_fabric.c b/prov/psm/src/psmx_fabric.c
@@ -91,7 +91,7 @@ static struct fi_ops_fabric psmx_fabric_ops = {
 	.passive_ep = fi_no_passive_ep,
 	.eq_open = ofi_eq_create,
 	.wait_open = psmx_wait_open,
-	.trywait = ofi_trywait
+	.trywait = psmx_wait_trywait,
 };
 
 static struct fi_fabric_attr psmx_fabric_attr = {

diff --git a/prov/psm/src/psmx_wait.c b/prov/psm/src/psmx_wait.c
@@ -150,3 +150,36 @@ int psmx_wait_open(struct fid_fabric *fabric, struct fi_wait_attr *attr,
 	return 0;
 }
 
+int psmx_wait_trywait(struct fid_fabric *fabric, struct fid **fids, int count)
+{
+	struct psmx_fid_cq *cq_priv;
+	struct util_eq *eq;
+	struct util_wait *wait;
+	int i, ret;
+
+	for (i = 0; i < count; i++) {
+		switch (fids[i]->fclass) {
+			case FI_CLASS_CQ:
+				cq_priv = container_of(fids[i], struct psmx_fid_cq, cq);
+				wait = cq_priv->wait;
+				break;
+			case FI_CLASS_EQ:
+				eq = container_of(fids[i], struct util_eq, eq_fid.fid);
+				wait = eq->wait;
+				break;
+			case FI_CLASS_CNTR:
+				return -FI_ENOSYS;
+			case FI_CLASS_WAIT:
+				wait = container_of(fids[i], struct util_wait, wait_fid.fid);
+				break;
+			default:
+				return -FI_EINVAL;
+		}
+
+		ret = wait->try(wait);
+		if (ret)
+			return ret;
+	}
+	return 0;
+}
+
diff --git a/prov/psm2/src/psmx2.h b/prov/psm2/src/psmx2.h
@@ -689,7 +689,8 @@ int	psmx2_cntr_open(struct fid_domain *domain, struct fi_cntr_attr *attr,
 		       struct fid_cntr **cntr, void *context);
 int	psmx2_wait_open(struct fid_fabric *fabric, struct fi_wait_attr *attr,
 			struct fid_wait **waitset);
-int psmx2_wait_trywait(struct fid_fabric *fabric, struct fid **fids, int count);
+int	psmx2_wait_trywait(struct fid_fabric *fabric, struct fid **fids,
+			   int count);
 
 static inline void psmx2_fabric_acquire(struct psmx2_fid_fabric *fabric)
 {

diff --git a/prov/psm2/src/psmx2_wait.c b/prov/psm2/src/psmx2_wait.c
@@ -164,7 +164,7 @@ int psmx2_wait_trywait(struct fid_fabric *fabric, struct fid **fids, int count)
 				wait = cq_priv->wait;
 				break;
 			case FI_CLASS_EQ:
-				eq = container_of(fids[i], struct util_eq, eq_fid.fid)
+				eq = container_of(fids[i], struct util_eq, eq_fid.fid);
 				wait = eq->wait;
 				break;
 			case FI_CLASS_CNTR: