rpmsg: zero out rdev field in rpmsg_unregister_endpoint

so is_rpmsg_ept_ready can check the validity more easier Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com>
OpenAMP · May 29, 2020 · 94ca6c2 · 94ca6c2
1 parent be5b3b3
commit 94ca6c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/include/openamp/rpmsg.h b/lib/include/openamp/rpmsg.h
@@ -343,8 +343,7 @@ void rpmsg_destroy_ept(struct rpmsg_endpoint *ept);
  */
 static inline unsigned int is_rpmsg_ept_ready(struct rpmsg_endpoint *ept)
 {
-	return (ept->dest_addr != RPMSG_ADDR_ANY) &&
-		(ept->addr != RPMSG_ADDR_ANY);
+	return ept && ept->rdev && ept->dest_addr != RPMSG_ADDR_ANY;
 }
 
 #if defined __cplusplus

diff --git a/lib/rpmsg/rpmsg.c b/lib/rpmsg/rpmsg.c
@@ -186,6 +186,7 @@ static void rpmsg_unregister_endpoint(struct rpmsg_endpoint *ept)
 		rpmsg_release_address(rdev->bitmap, RPMSG_ADDR_BMP_SIZE,
 				      ept->addr);
 	metal_list_del(&ept->node);
+	ept->rdev = NULL;
 	metal_mutex_release(&rdev->lock);
 }