Return-Path: Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S935493AbdIYM5V (ORCPT ); Mon, 25 Sep 2017 08:57:21 -0400 Received: from mail-pg0-f67.google.com ([74.125.83.67]:33362 "EHLO mail-pg0-f67.google.com" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S935053AbdIYM4q (ORCPT ); Mon, 25 Sep 2017 08:56:46 -0400 X-Google-Smtp-Source: AOwi7QDURKO4f8iqwoXXakzl6h0YTa+igEy2Ke3X4CbULv/0IvRRJyIT0lz2c9H0F2oyH25VxtOlng== From: Gargi Sharma To: linux-kernel@vger.kernel.org Cc: riel@surriel.com, julia.lawall@lip6.fr, akpm@linux-foundation.org, mingo@kernel.org, pasha.tatashin@oracle.com, ktkhai@virtuozzo.com, oleg@redhat.com, Gargi Sharma Subject: [PATCH 2/4] idr: Add a function idr_get() Date: Mon, 25 Sep 2017 08:56:16 -0400 Message-Id: <6d2d9a99d9047a38ae3b8442e537e76c54511c51.1506342921.git.gs051095@gmail.com> X-Mailer: git-send-email 2.7.4 In-Reply-To: References: In-Reply-To: References: Sender: linux-kernel-owner@vger.kernel.org List-ID: X-Mailing-List: linux-kernel@vger.kernel.org Content-Length: 1522 Lines: 48 idr_get(namespace, id) returns a NULL if id is not present in the idr tree or returns the pointer to the struct if id is present in the idr tree. With this function in the idr library, code for pid allocation can be simplified by calling this function instead of looking through the pidhash. Signed-off-by: Gargi Sharma --- include/linux/idr.h | 1 + lib/idr.c | 11 +++++++++++ 2 files changed, 12 insertions(+) diff --git a/include/linux/idr.h b/include/linux/idr.h index 7c3a365..e12b174 100644 --- a/include/linux/idr.h +++ b/include/linux/idr.h @@ -135,6 +135,7 @@ int idr_for_each(const struct idr *, int (*fn)(int id, void *p, void *data), void *data); void *idr_get_next(struct idr *, int *nextid); void *idr_get_next_ext(struct idr *idr, unsigned long *nextid); +void *idr_get(struct idr *idr, int *id); void *idr_replace(struct idr *, void *, int id); void *idr_replace_ext(struct idr *idr, void *ptr, unsigned long id); void idr_destroy(struct idr *); diff --git a/lib/idr.c b/lib/idr.c index f9adf48..bb76400 100644 --- a/lib/idr.c +++ b/lib/idr.c @@ -135,6 +135,17 @@ void *idr_get_next_ext(struct idr *idr, unsigned long *nextid) } EXPORT_SYMBOL(idr_get_next_ext); +void * idr_get(struct idr *idr, int *id) +{ + struct radix_tree_node *node; + void __rcu **slot = NULL; + + __radix_tree_lookup(&idr->idr_rt, *id, &node, &slot); + if (!slot) + return NULL; + return node; +} + /** * idr_replace - replace pointer for given id * @idr: idr handle -- 2.7.4