idr:New idr_find interface

Different from idr_get_next, if idr_find directly when could not find the node id return an error

Signed-off-by: chenrun1 <chenrun1@xiaomi.com>

include/nuttx/idr.h

@@ -40,6 +40,7 @@ struct idr_s;
  * Public Function Prototypes
  ****************************************************************************/
 
+FAR void *idr_find(FAR struct idr_s *idr, unsigned int id);
 FAR void *idr_get_next(FAR struct idr_s *idr, FAR int *id);
 FAR struct idr_s *idr_init_base(int base);
 void idr_destroy(FAR struct idr_s *idr);

libs/libc/misc/lib_idr.c

@@ -93,6 +93,32 @@ RB_GENERATE_STATIC(idr_tree_s, idr_node_s, link, idr_compare)
  * Public Functions
  ****************************************************************************/
 
+/****************************************************************************
+ * Name: idr_find
+ ****************************************************************************/
+
+FAR void *idr_find(FAR struct idr_s *idr, unsigned int id)
+{
+  FAR struct idr_node_s *node;
+  struct idr_node_s search;
+
+  search.id = id;
+  nxmutex_lock(&idr->lock);
+  node = RB_FIND(idr_tree_s, &idr->alloced, &search);
+  if (node == NULL)
+    {
+      nxmutex_unlock(&idr->lock);
+      return NULL;
+    }
+
+  nxmutex_unlock(&idr->lock);
+  return node->data;
+}
+
+/****************************************************************************
+ * Name: idr_get_next
+ ****************************************************************************/
+
 FAR void *idr_get_next(FAR struct idr_s *idr, FAR int *id)
 {
   FAR struct idr_node_s *node;