From 77666d12ab3333ad42448e7a525a3d4fcdc1ba35 Mon Sep 17 00:00:00 2001 From: Yossi Itigin Date: Mon, 27 Apr 2020 00:42:49 +0300 Subject: [PATCH] UCP/EAGER: Check ep_ptr for incoming messages --- src/ucp/tag/eager_rcv.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/ucp/tag/eager_rcv.c b/src/ucp/tag/eager_rcv.c index 60b46f53f0f..bbf563caed1 100644 --- a/src/ucp/tag/eager_rcv.c +++ b/src/ucp/tag/eager_rcv.c @@ -87,6 +87,10 @@ ucp_eager_tagged_handler(void *arg, void *data, size_t length, unsigned am_flags ucs_assert(length >= hdr_len); ucs_assert(flags & UCP_RECV_DESC_FLAG_EAGER); + if (ucp_worker_get_ep_by_ptr(worker, eager_hdr->ep_ptr) == NULL) { + return UCS_OK; + } + recv_tag = eager_hdr->super.tag; recv_len = length - hdr_len; @@ -164,6 +168,10 @@ ucp_eager_common_middle_handler(ucp_worker_t *worker, void *data, size_t length, khiter_t iter; int ret; + if (ucp_worker_get_ep_by_ptr(worker, hdr->ep_ptr) == NULL) { + return UCS_OK; + } + iter = kh_put(ucp_tag_frag_hash, &worker->tm.frag_hash, hdr->msg_id, &ret); ucs_assert(ret >= 0); matchq = &kh_value(&worker->tm.frag_hash, iter);