From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from mail-pj1-f47.google.com (mail-pj1-f47.google.com [209.85.216.47]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 4A2FC388E50 for ; Tue, 9 Jun 2026 11:35:58 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.216.47 ARC-Seal:i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1781004959; cv=none; b=ut2nxR+6s31rt3iecmzodqzO32zUAO3IcdYub21bXSigcmR6b8ng/lh257LSQxWcfujLrLQdVOmk+jb0yAi0kibmJ3+alYSVGPoSAp91A3DglMVeJ4mT6WmpJ7KkfXQ5rjrJ0sXuPveE5QKWRNQ2jApcuhROR2H+AE5sbee+KV8= ARC-Message-Signature:i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1781004959; c=relaxed/simple; bh=hOTHzfv9Dw42L56/OwesQ5Xt6mIL60jUnFpFaT2TKnc=; h=From:To:Cc:Subject:Date:Message-Id:In-Reply-To:References: MIME-Version; b=PLStrZTSB922n3XP2I47xpPWjel2Pqrr2iDwaoZ5GnSstiqEWAwRtXN3a1dbyhdBq5hANWcoMPpq+TWt+BfsBzqNdp0NEYSs8AzUrvWsdSQYtJ8fxQNWSRug8FNpC4/8V2xu568uGDt8Fqbn9D4HQefvoET9fj7fJO9+PJXVTLY= ARC-Authentication-Results:i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=gmail.com; spf=pass smtp.mailfrom=gmail.com; dkim=pass (2048-bit key) header.d=gmail.com header.i=@gmail.com header.b=EGVH6R9E; arc=none smtp.client-ip=209.85.216.47 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=gmail.com Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=gmail.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=gmail.com header.i=@gmail.com header.b="EGVH6R9E" Received: by mail-pj1-f47.google.com with SMTP id 98e67ed59e1d1-36b9033d230so2806439a91.1 for ; Tue, 09 Jun 2026 04:35:58 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20251104; t=1781004957; x=1781609757; darn=vger.kernel.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=2a1iC93mF2361F98xckcBOo8JlNpMzrV7XxWUy9iDrk=; b=EGVH6R9E+EKkhljVMX61Xc+B/bdSVyu6yrvjpSpcjCErGj3vYqDJoRNYfNx54oPOxQ MYIcdLjZBhGfNRXvK6/gzcM+zLBpTqD+7FO6AfI9lD0eYS8xa4mJoW2NMC9HWfPKnXr/ TQTqlX1zO2c7Ax8Bs0ZQ20vi3rQ8u5fr2B978RuNm24vEKdeQWre4nCnXRkV7MHPG+xA thQVyGDbZYdx4wt1yWv57oKnkER7dcyuL2MqMfyoI5m/p+vCQy91Fo8Pk4qgg8Bp1lBp 1eeWCxVn/b/V7efO7DpTek+wTbEqXwM66avO/XXbsdWpz9U7+2UWJhtVR6W7ExJviPQk Hc2Q== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20251104; t=1781004958; x=1781609758; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-gg:x-gm-message-state:from :to:cc:subject:date:message-id:reply-to; bh=2a1iC93mF2361F98xckcBOo8JlNpMzrV7XxWUy9iDrk=; b=MBtvOwMPHg1tQDbppvd02/sucRrA+n8nhKCKJW/nKoSp70h7vgDhBtwjeHpDxNKk/N lqEAi/HvPcBOeC2GOkcGuxW/4gvBJWST38ocWdU1sVGnV4WW64Ka+8gcd/NNByoeJvXB kDjtnvmNrm4AwZfO8r/UWPAjJg5SqEsTFN23uC0dYrjeavDIKkL7n2230NF2qCRjYhr5 3FALjOUeGfXudjkOnDC74O1Dbs94q/Glps05ta8iGocgiEh0KaqERR5SFaxmqo4OkYLa tffQaomNhKBTw/WdescUF90sNJJkehX5iUme+VQ0ZlxtBAHrZgLq8pxmkP4V0EjQkEaX K03A== X-Forwarded-Encrypted: i=1; AFNElJ8pqxb5YfqQaGBdm6L8bXmgt6llqsHPsmspchw2h/SrePGFGe8tzP9YcUFuTFseMzB1cwjtgVcAIoUY7eI=@vger.kernel.org X-Gm-Message-State: AOJu0YyYp7Dmso1Tv3hFQ+o5YI40BmQ9EFYhAiaVeHxf6hjC2KHjkHTM 0BElHbNdI1yQi/ltqLA5gJXwnGKRtI2v5s7LzlmDV7Vdqa9gAx75aVea X-Gm-Gg: Acq92OHhvZ9s87ztucsE62WQr6xR2ZljnVrJ4CTWrcQCzY035J1xBA2XWy1JlSuceO5 BOms1O9lc69fho+6YURuZf7q79aA5dGW6n8sp2RGkErDbjB0ok6igdr6bHmXqSKZ0YyNzSJLD7c tdrNlBACrldLkRP88yBjitmQiqP9jElPBRtWmYNpIDrXhdQy0mh3zLQAe+oEeFKoOrtaj+1dvA3 bVQQipVxFKXU+VOsN1gFhHpU7Mn498VamoPcJS8MmHL0ZySQP6FdzeQz+cUGbjhZVYQdUZ3rHeS ImXZF+kXNjOnNfJ+HP9YMinGfrsSmepRZKPfizdy0lb65DlQQo8vs99QYa2YrOiNLf4brRhJ7C5 mBm4Eak58N1O52NSjZVMbhc++eO01bWE1esm5xDsfhEzcP1Qhs7sxXwrVQcrx2nneRYJh7WMNwk nslKZ1Oli6CWqKxx5/3iR9ojGotxgz1dHFKLt6dO53yv6J9PKdSw== X-Received: by 2002:a17:90b:3f46:b0:35f:b9f1:fded with SMTP id 98e67ed59e1d1-3713336f91fmr14812189a91.12.1781004957417; Tue, 09 Jun 2026 04:35:57 -0700 (PDT) Received: from ubuntu22.mioffice.cn ([43.224.245.232]) by smtp.gmail.com with ESMTPSA id 98e67ed59e1d1-36f6c668e98sm19655292a91.5.2026.06.09.04.35.54 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Tue, 09 Jun 2026 04:35:56 -0700 (PDT) From: Wenchao Hao To: Andrew Morton , linux-kernel@vger.kernel.org, linux-mm@kvack.org, Minchan Kim , Sergey Senozhatsky Cc: Nhat Pham , Joshua Hahn , Barry Song , Wenchao Hao Subject: [PATCH v4 1/4] mm/zsmalloc: encode class index in obj value for lockless class lookup Date: Tue, 9 Jun 2026 19:35:17 +0800 Message-Id: <20260609113520.3507783-2-haowenchao22@gmail.com> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20260609113520.3507783-1-haowenchao22@gmail.com> References: <20260609113520.3507783-1-haowenchao22@gmail.com> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: 8bit From: Wenchao Hao Encode the size_class index (class_idx) into the obj value so that zs_free() can determine the correct size_class without dereferencing the handle->obj->PFN->zpdesc->zspage->class chain under pool->lock. class_idx is invariant across page migration (only PFN is rewritten), so a lockless read of obj always yields a valid class_idx. Where obj has more bits below the PFN field than obj_idx alone needs, split that space into class_idx and obj_idx subfields: |<-- _PFN_BITS -->|<-- ZS_OBJ_CLASS_BITS -->|<-- ZS_OBJ_IDX_BITS -->| +-----------------+-------------------------+-----------------------+ | PFN | class_idx | obj_idx | +-----------------+-------------------------+-----------------------+ MSB ^ LSB | +-- ZS_OBJ_PFN_SHIFT The macro layout changes as follows: Before After Meaning ---------------- ------------------ ---------------------------- OBJ_INDEX_BITS ZS_OBJ_IDX_BITS width of obj_idx subfield OBJ_INDEX_MASK ZS_OBJ_IDX_MASK mask of obj_idx subfield (n/a) ZS_OBJ_CLASS_BITS width of class_idx subfield (n/a) ZS_OBJ_CLASS_MASK mask of class_idx subfield (n/a) ZS_OBJ_PFN_SHIFT bit offset of PFN in obj ZS_OBJ_CLASS_BITS folds to 0 (and the layout collapses to [PFN | obj_idx]) when obj has no spare bits, i.e. on 32-bit or on 64-bit fallback paths where MAX_POSSIBLE_PHYSMEM_BITS == BITS_PER_LONG (e.g. UML); zs_free() then falls back to pool->lock. Signed-off-by: Wenchao Hao --- mm/zsmalloc.c | 99 +++++++++++++++++++++++++++++++++++++++++++-------- 1 file changed, 85 insertions(+), 14 deletions(-) diff --git a/mm/zsmalloc.c b/mm/zsmalloc.c index 63128ddb7959..f84258d63917 100644 --- a/mm/zsmalloc.c +++ b/mm/zsmalloc.c @@ -67,8 +67,8 @@ #define MAX_POSSIBLE_PHYSMEM_BITS MAX_PHYSMEM_BITS #else /* - * If this definition of MAX_PHYSMEM_BITS is used, OBJ_INDEX_BITS will just - * be PAGE_SHIFT + * If this definition of MAX_PHYSMEM_BITS is used, ZS_OBJ_PFN_SHIFT will + * just be PAGE_SHIFT */ #define MAX_POSSIBLE_PHYSMEM_BITS BITS_PER_LONG #endif @@ -88,8 +88,23 @@ #define OBJ_TAG_BITS 1 #define OBJ_TAG_MASK OBJ_ALLOCATED_TAG -#define OBJ_INDEX_BITS (BITS_PER_LONG - _PFN_BITS) -#define OBJ_INDEX_MASK ((_AC(1, UL) << OBJ_INDEX_BITS) - 1) +/* + * obj is encoded as [PFN | class_idx | obj_idx] within an unsigned long: + * + * |<-- _PFN_BITS -->|<-- ZS_OBJ_CLASS_BITS -->|<-- ZS_OBJ_IDX_BITS -->| + * +-----------------+-------------------------+-----------------------+ + * | PFN | class_idx | obj_idx | + * +-----------------+-------------------------+-----------------------+ + * MSB ^ LSB + * | + * +-- ZS_OBJ_PFN_SHIFT + * + * Encoding class_idx into obj lets zs_free() locate the size_class + * without holding pool->lock; class_idx is invariant across page + * migration (only PFN changes), so a lockless read of the obj value + * always yields a valid class_idx. + */ +#define ZS_OBJ_PFN_SHIFT (BITS_PER_LONG - _PFN_BITS) #define HUGE_BITS 1 #define FULLNESS_BITS 4 @@ -98,9 +113,55 @@ #define ZS_MAX_PAGES_PER_ZSPAGE (_AC(CONFIG_ZSMALLOC_CHAIN_SIZE, UL)) +/* + * ceil(log2(ZS_MAX_PAGES_PER_ZSPAGE)) at preprocessor time, for use + * in #if below. Kconfig restricts ZSMALLOC_CHAIN_SIZE to [4, 16]. + */ +#if ZS_MAX_PAGES_PER_ZSPAGE <= 4 +#define ZS_CHAIN_LOG2 2 +#elif ZS_MAX_PAGES_PER_ZSPAGE <= 8 +#define ZS_CHAIN_LOG2 3 +#elif ZS_MAX_PAGES_PER_ZSPAGE <= 16 +#define ZS_CHAIN_LOG2 4 +#else +#error "ZSMALLOC_CHAIN_SIZE out of expected range [4,16]" +#endif + +/* PAGE_SHIFT - 5 = log2(PAGE_SIZE / 32); 32 = ZS_MIN_ALLOC_SIZE floor. */ +#define ZS_MAX_OBJ_PER_PAGE_LOG2 (PAGE_SHIFT - 5) + +/* + * obj_idx width that keeps ZS_MIN_ALLOC_SIZE at its 32-byte floor. + * Below this, ZS_MIN_ALLOC_SIZE is auto-raised by the MAX(32, ...) + * formula -- still correct, but objects are coarser. + */ +#define ZS_OBJ_IDX_DENSE_BITS (ZS_CHAIN_LOG2 + ZS_MAX_OBJ_PER_PAGE_LOG2) + +/* + * Encode class_idx only when obj has spare bits; otherwise + * ZS_OBJ_CLASS_BITS folds to 0 (32-bit, or 64-bit UML/fallback). + */ +#if BITS_PER_LONG >= 64 && \ + ZS_OBJ_PFN_SHIFT >= (CLASS_BITS + 1) + ZS_OBJ_IDX_DENSE_BITS +#define ZS_OBJ_CLASS_BITS (CLASS_BITS + 1) +#else +#define ZS_OBJ_CLASS_BITS 0 +#endif +#define ZS_OBJ_CLASS_MASK ((_AC(1, UL) << ZS_OBJ_CLASS_BITS) - 1) + +#define ZS_OBJ_IDX_BITS (ZS_OBJ_PFN_SHIFT - ZS_OBJ_CLASS_BITS) +#define ZS_OBJ_IDX_MASK ((_AC(1, UL) << ZS_OBJ_IDX_BITS) - 1) + +/* + * Belt-and-suspenders: the #if above already guarantees this when + * class_idx is enabled. Catches future tweaks that bypass it. + */ +static_assert(ZS_OBJ_IDX_BITS >= ZS_CHAIN_LOG2, + "zsmalloc: ZS_MIN_ALLOC_SIZE would exceed ZS_MAX_ALLOC_SIZE"); + /* ZS_MIN_ALLOC_SIZE must be multiple of ZS_ALIGN */ #define ZS_MIN_ALLOC_SIZE \ - MAX(32, (ZS_MAX_PAGES_PER_ZSPAGE << PAGE_SHIFT >> OBJ_INDEX_BITS)) + MAX(32, (ZS_MAX_PAGES_PER_ZSPAGE << PAGE_SHIFT >> ZS_OBJ_IDX_BITS)) /* each chunk includes extra space to keep handle */ #define ZS_MAX_ALLOC_SIZE PAGE_SIZE @@ -721,26 +782,35 @@ static struct zpdesc *get_next_zpdesc(struct zpdesc *zpdesc) static void obj_to_location(unsigned long obj, struct zpdesc **zpdesc, unsigned int *obj_idx) { - *zpdesc = pfn_zpdesc(obj >> OBJ_INDEX_BITS); - *obj_idx = (obj & OBJ_INDEX_MASK); + *zpdesc = pfn_zpdesc(obj >> ZS_OBJ_PFN_SHIFT); + *obj_idx = (obj & ZS_OBJ_IDX_MASK); } static void obj_to_zpdesc(unsigned long obj, struct zpdesc **zpdesc) { - *zpdesc = pfn_zpdesc(obj >> OBJ_INDEX_BITS); + *zpdesc = pfn_zpdesc(obj >> ZS_OBJ_PFN_SHIFT); +} + +/* Folds to 0 when ZS_OBJ_CLASS_BITS == 0; no ifdef needed at callers. */ +static unsigned int obj_to_class_idx(unsigned long obj) +{ + return (obj >> ZS_OBJ_IDX_BITS) & ZS_OBJ_CLASS_MASK; } /** - * location_to_obj - get obj value encoded from (, ) + * location_to_obj - encode (, , ) into obj value * @zpdesc: zpdesc object resides in zspage * @obj_idx: object index + * @class_idx: size class index; ignored when ZS_OBJ_CLASS_BITS == 0 */ -static unsigned long location_to_obj(struct zpdesc *zpdesc, unsigned int obj_idx) +static unsigned long location_to_obj(struct zpdesc *zpdesc, unsigned int obj_idx, + unsigned int class_idx) { unsigned long obj; - obj = zpdesc_pfn(zpdesc) << OBJ_INDEX_BITS; - obj |= obj_idx & OBJ_INDEX_MASK; + obj = zpdesc_pfn(zpdesc) << ZS_OBJ_PFN_SHIFT; + obj |= (unsigned long)(class_idx & ZS_OBJ_CLASS_MASK) << ZS_OBJ_IDX_BITS; + obj |= obj_idx & ZS_OBJ_IDX_MASK; return obj; } @@ -1276,7 +1346,7 @@ static unsigned long obj_malloc(struct zs_pool *pool, kunmap_local(vaddr); mod_zspage_inuse(zspage, 1); - obj = location_to_obj(m_zpdesc, obj); + obj = location_to_obj(m_zpdesc, obj, zspage->class); record_obj(handle, obj); return obj; @@ -1762,7 +1832,8 @@ static int zs_page_migrate(struct page *newpage, struct page *page, old_obj = handle_to_obj(handle); obj_to_location(old_obj, &dummy, &obj_idx); - new_obj = (unsigned long)location_to_obj(newzpdesc, obj_idx); + new_obj = location_to_obj(newzpdesc, obj_idx, + obj_to_class_idx(old_obj)); record_obj(handle, new_obj); } } -- 2.34.1