Merge pull request #9766 from opensourcerouting/typesafe-member-nhrp-zap

lib: add typesafe membership-test functions

1 files changed, 40 insertions, 0 deletions

diff --git a/lib/typesafe.c b/lib/typesafe.c
index 76705fad0d..f90b59daf0 100644
--- a/lib/typesafe.c
+++ b/lib/typesafe.c
@@ -29,6 +29,46 @@ DEFINE_MTYPE_STATIC(LIB, TYPEDHASH_BUCKET, "Typed-hash bucket");
 DEFINE_MTYPE_STATIC(LIB, SKIPLIST_OFLOW, "Skiplist overflow");
 DEFINE_MTYPE_STATIC(LIB, HEAP_ARRAY, "Typed-heap array");
 
+struct slist_item typesafe_slist_sentinel = { NULL };
+
+bool typesafe_list_member(const struct slist_head *head,
+			  const struct slist_item *item)
+{
+	struct slist_item *fromhead = head->first;
+	struct slist_item **fromnext = (struct slist_item **)&item->next;
+
+	while (fromhead != _SLIST_LAST) {
+		if (fromhead == item || fromnext == head->last_next)
+			return true;
+
+		fromhead = fromhead->next;
+		if (!*fromnext || *fromnext == _SLIST_LAST)
+			break;
+		fromnext = &(*fromnext)->next;
+	}
+
+	return false;
+}
+
+bool typesafe_dlist_member(const struct dlist_head *head,
+			   const struct dlist_item *item)
+{
+	const struct dlist_item *fromhead = head->hitem.next;
+	const struct dlist_item *fromitem = item->next;
+
+	if (!item->prev || !item->next)
+		return false;
+
+	while (fromhead != &head->hitem && fromitem != item) {
+		if (fromitem == &head->hitem || fromhead == item)
+			return true;
+		fromhead = fromhead->next;
+		fromitem = fromitem->next;
+	}
+
+	return false;
+}
+
 #if 0
 static void hash_consistency_check(struct thash_head *head)
 {