From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: X-Cyrus-Session-Id: sloti22d1t05-362987-1516223218-2-17784169487750982518 X-Sieve: CMU Sieve 3.0 X-Spam-known-sender: no X-Spam-score: 0.0 X-Spam-hits: BAYES_00 -1.9, HEADER_FROM_DIFFERENT_DOMAINS 0.25, RCVD_IN_DNSWL_HI -5, T_RP_MATCHES_RCVD -0.01, LANGUAGES en, BAYES_USED global, SA_VERSION 3.4.0 X-Spam-source: IP='209.132.180.67', Host='vger.kernel.org', Country='US', FromHeader='org', MailFrom='org' X-Spam-charsets: X-Resolved-to: greg@kroah.com X-Delivered-to: greg@kroah.com X-Mail-from: linux-usb-owner@vger.kernel.org ARC-Seal: i=1; a=rsa-sha256; cv=none; d=messagingengine.com; s=arctest; t=1516223218; b=HEJujeFirg91Tma6a6mICd0dQnnYVG+FnL/9QADAjzBccS+ XYzrBT/T1D9TmqYStf1gCFjTJ+9ZiGMbN1r0uEFkep7bwQZHn3BJiisfx6VmlQPM Hiu5gBwNkFT1PD0uGFQueWDq+1r78RvEEDX3RsgRvao1cwBj7B7CPncm7FuY6TJN F610zWNE2ebuRp3y2gjqVBMR0fXqYOiZyhPjXU4ceFUKPfwF+xbjarQnWCkKMyGc QVx3FjgXNMh+IrJDMGLzbMDL9IaMag4pc45qNKQ0PB3oMirC4KQKrmeZY+Hc/RMv dMElg8hkS7GfACtd5wBGC2ygWHmj3JP0+M/jQwg== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d= messagingengine.com; h=from:to:cc:subject:date:message-id :in-reply-to:references:sender:list-id; s=arctest; t=1516223218; bh=CEvJArdIR87M0tjATwD9ttwg2hTxtUtymitkYTAG3LQ=; b=ZBzYTozdg2Zd 8mEYPbxRKbCPMM5tV2HA1SywV2ZUFNzY0f8k1AnS52X3m6dO+JlrpBS64/R0jFqG i3g7YeRPFPQRNcVh6UV1GpVmbJ7lVHsFYhXTwyUS2tmMsjZ6ciBcpzzjm7W3CmSx 8vP+Q4ZxucYfWK1XIF7nhkPVSn+Nw3Lv3DQrd5/wZqy1ydLqNeW5FJJNApvaOoul Vu7/lZfizcdnLZqAyNMUsNRPqMCzDBIfSqT/RylWmMlFVabf1EOri1qG0M1eV/LQ fgVu9V1D0bS/pKDUTYCtuiCD2wuWfMsgNm3L6zrnH22Vv30QoetRqs5Vy+4cfAyx m7YnchW2PQ== ARC-Authentication-Results: i=1; mx4.messagingengine.com; arc=none (no signatures found); dkim=fail (message has been altered; 2048-bit rsa key sha256) header.d=infradead.org header.i=@infradead.org header.b=Nted86jA x-bits=2048 x-keytype=rsa x-algorithm=sha256 x-selector=bombadil.20170209; dmarc=none (p=none,has-list-id=yes,d=none) header.from=infradead.org; iprev=pass policy.iprev=209.132.180.67 (vger.kernel.org); spf=none smtp.mailfrom=linux-usb-owner@vger.kernel.org smtp.helo=vger.kernel.org; x-aligned-from=fail; x-ptr=pass x-ptr-helo=vger.kernel.org x-ptr-lookup=vger.kernel.org; x-return-mx=pass smtp.domain=vger.kernel.org smtp.result=pass smtp_org.domain=kernel.org smtp_org.result=pass smtp_is_org_domain=no header.domain=infradead.org header.result=pass header_is_org_domain=yes Authentication-Results: mx4.messagingengine.com; arc=none (no signatures found); dkim=fail (message has been altered; 2048-bit rsa key sha256) header.d=infradead.org header.i=@infradead.org header.b=Nted86jA x-bits=2048 x-keytype=rsa x-algorithm=sha256 x-selector=bombadil.20170209; dmarc=none (p=none,has-list-id=yes,d=none) header.from=infradead.org; iprev=pass policy.iprev=209.132.180.67 (vger.kernel.org); spf=none smtp.mailfrom=linux-usb-owner@vger.kernel.org smtp.helo=vger.kernel.org; x-aligned-from=fail; x-ptr=pass x-ptr-helo=vger.kernel.org x-ptr-lookup=vger.kernel.org; x-return-mx=pass smtp.domain=vger.kernel.org smtp.result=pass smtp_org.domain=kernel.org smtp_org.result=pass smtp_is_org_domain=no header.domain=infradead.org header.result=pass header_is_org_domain=yes Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S1754591AbeAQVGh (ORCPT ); Wed, 17 Jan 2018 16:06:37 -0500 Received: from bombadil.infradead.org ([65.50.211.133]:37363 "EHLO bombadil.infradead.org" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S1753522AbeAQUWe (ORCPT ); Wed, 17 Jan 2018 15:22:34 -0500 From: Matthew Wilcox To: linux-kernel@vger.kernel.org Cc: Matthew Wilcox , linux-mm@kvack.org, linux-fsdevel@vger.kernel.org, linux-f2fs-devel@lists.sourceforge.net, linux-nilfs@vger.kernel.org, linux-btrfs@vger.kernel.org, linux-xfs@vger.kernel.org, linux-usb@vger.kernel.org, Bjorn Andersson , Stefano Stabellini , iommu@lists.linux-foundation.org, linux-remoteproc@vger.kernel.org, linux-s390@vger.kernel.org, intel-gfx@lists.freedesktop.org, cgroups@vger.kernel.org, linux-sh@vger.kernel.org, David Howells Subject: [PATCH v6 21/99] xarray: Add xa_reserve and xa_release Date: Wed, 17 Jan 2018 12:20:45 -0800 Message-Id: <20180117202203.19756-22-willy@infradead.org> X-Mailer: git-send-email 2.14.3 In-Reply-To: <20180117202203.19756-1-willy@infradead.org> References: <20180117202203.19756-1-willy@infradead.org> Sender: linux-usb-owner@vger.kernel.org X-Mailing-List: linux-usb@vger.kernel.org X-getmail-retrieved-from-mailbox: INBOX X-Mailing-List: linux-kernel@vger.kernel.org List-ID: From: Matthew Wilcox This function simply creates a slot in the XArray for users which need to acquire multiple locks before storing their entry in the tree and so cannot use a plain xa_store(). Signed-off-by: Matthew Wilcox --- include/linux/xarray.h | 14 ++++++++++ lib/xarray.c | 51 ++++++++++++++++++++++++++++++++++ tools/testing/radix-tree/xarray-test.c | 25 +++++++++++++++++ 3 files changed, 90 insertions(+) diff --git a/include/linux/xarray.h b/include/linux/xarray.h index 6f59f1f60205..c3f7405c5517 100644 --- a/include/linux/xarray.h +++ b/include/linux/xarray.h @@ -259,6 +259,7 @@ void *xa_load(struct xarray *, unsigned long index); void *xa_store(struct xarray *, unsigned long index, void *entry, gfp_t); void *xa_cmpxchg(struct xarray *, unsigned long index, void *old, void *entry, gfp_t); +int xa_reserve(struct xarray *, unsigned long index, gfp_t); bool xa_get_tag(struct xarray *, unsigned long index, xa_tag_t); void xa_set_tag(struct xarray *, unsigned long index, xa_tag_t); void xa_clear_tag(struct xarray *, unsigned long index, xa_tag_t); @@ -373,6 +374,19 @@ static inline int xa_insert(struct xarray *xa, unsigned long index, return -EEXIST; } +/** + * xa_release() - Release a reserved entry. + * @xa: XArray. + * @index: Index of entry. + * + * After calling xa_reserve(), you can call this function to release the + * reservation. It is harmless to call this function if the entry was used. + */ +static inline void xa_release(struct xarray *xa, unsigned long index) +{ + xa_cmpxchg(xa, index, NULL, NULL, 0); +} + #define xa_trylock(xa) spin_trylock(&(xa)->xa_lock) #define xa_lock(xa) spin_lock(&(xa)->xa_lock) #define xa_unlock(xa) spin_unlock(&(xa)->xa_lock) diff --git a/lib/xarray.c b/lib/xarray.c index ace309cc9253..b4dec8e2d202 100644 --- a/lib/xarray.c +++ b/lib/xarray.c @@ -1275,6 +1275,8 @@ void *xa_cmpxchg(struct xarray *xa, unsigned long index, do { xas_lock(&xas); curr = xas_load(&xas); + if (curr == XA_ZERO_ENTRY) + curr = NULL; if (curr == old) xas_store(&xas, entry); xas_unlock(&xas); @@ -1310,6 +1312,8 @@ void *__xa_cmpxchg(struct xarray *xa, unsigned long index, do { curr = xas_load(&xas); + if (curr == XA_ZERO_ENTRY) + curr = NULL; if (curr == old) xas_store(&xas, entry); } while (__xas_nomem(&xas, gfp)); @@ -1318,6 +1322,53 @@ void *__xa_cmpxchg(struct xarray *xa, unsigned long index, } EXPORT_SYMBOL(__xa_cmpxchg); +/** + * xa_reserve() - Reserve this index in the XArray. + * @xa: XArray. + * @index: Index into array. + * @gfp: Memory allocation flags. + * + * Ensures there is somewhere to store an entry at @index in the array. + * If there is already something stored at @index, this function does + * nothing. If there was nothing there, the entry is marked as reserved. + * Loads from @index will continue to see a %NULL pointer until a + * subsequent store to @index. + * + * If you do not use the entry that you have reserved, call xa_release() + * or xa_erase() to free any unnecessary memory. + * + * Return: 0 if the reservation succeeded or -ENOMEM if it failed. + */ +int xa_reserve(struct xarray *xa, unsigned long index, gfp_t gfp) +{ + XA_STATE(xas, xa, index); + unsigned int lock_type = xa_lock_type(xa); + void *curr; + + do { + if (lock_type == XA_LOCK_IRQ) + xas_lock_irq(&xas); + else if (lock_type == XA_LOCK_BH) + xas_lock_bh(&xas); + else + xas_lock(&xas); + + curr = xas_create(&xas); + if (!curr) + xas_store(&xas, XA_ZERO_ENTRY); + + if (lock_type == XA_LOCK_IRQ) + xas_unlock_irq(&xas); + else if (lock_type == XA_LOCK_BH) + xas_unlock_bh(&xas); + else + xas_unlock(&xas); + } while (xas_nomem(&xas, gfp)); + + return xas_error(&xas); +} +EXPORT_SYMBOL(xa_reserve); + /** * __xa_set_tag() - Set this tag on this entry while locked. * @xa: XArray. diff --git a/tools/testing/radix-tree/xarray-test.c b/tools/testing/radix-tree/xarray-test.c index 4d3541ac31e9..fe38b53df2ab 100644 --- a/tools/testing/radix-tree/xarray-test.c +++ b/tools/testing/radix-tree/xarray-test.c @@ -502,6 +502,29 @@ void check_move(struct xarray *xa) } while (i < (1 << 16)); } +void check_reserve(struct xarray *xa) +{ + assert(xa_empty(xa)); + xa_reserve(xa, 12345678, GFP_KERNEL); + assert(!xa_empty(xa)); + assert(!xa_load(xa, 12345678)); + xa_release(xa, 12345678); + assert(xa_empty(xa)); + + xa_reserve(xa, 12345678, GFP_KERNEL); + assert(!xa_store(xa, 12345678, xa_mk_value(12345678), GFP_NOWAIT)); + xa_release(xa, 12345678); + assert(xa_erase(xa, 12345678) == xa_mk_value(12345678)); + assert(xa_empty(xa)); + + xa_reserve(xa, 12345678, GFP_KERNEL); + assert(!xa_cmpxchg(xa, 12345678, NULL, xa_mk_value(12345678), + GFP_NOWAIT)); + xa_release(xa, 12345678); + assert(xa_erase(xa, 12345678) == xa_mk_value(12345678)); + assert(xa_empty(xa)); +} + void xarray_checks(void) { DEFINE_XARRAY(array); @@ -548,6 +571,8 @@ void xarray_checks(void) check_move(&array); item_kill_tree(&array); + + check_reserve(&array); } int __weak main(void) -- 2.15.1