Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20;
From:   Liam Howlett <liam.howlett@oracle.com>
To:     "maple-tree@lists.infradead.org" <maple-tree@lists.infradead.org>,
        "linux-mm@kvack.org" <linux-mm@kvack.org>,
        "linux-kernel@vger.kernel.org" <linux-kernel@vger.kernel.org>,
        Andrew Morton <akpm@linux-foundation.org>
Subject: [PATCH v5 04/70] radix tree test suite: Add allocation counts and
 size to kmem_cache
Thread-Topic: [PATCH v5 04/70] radix tree test suite: Add allocation counts
 and size to kmem_cache
Thread-Index: AQHYF95wlrsCjKk/YUOwmA1eSmMkSA==
Date:   Wed, 2 Feb 2022 02:41:54 +0000
Message-ID: <20220202024137.2516438-5-Liam.Howlett@oracle.com>
References: <20220202024137.2516438-1-Liam.Howlett@oracle.com>
In-Reply-To: <20220202024137.2516438-1-Liam.Howlett@oracle.com>
Accept-Language: en-US
Content-Language: en-US
x-ms-exchange-antispam-messagedata-chunkcount: 1
x-ms-exchange-antispam-messagedata-0: =?iso-8859-1?Q?BByAOdsipsZc00wBthYN+peZDzKVqLgz4upXPjTRlgIE8mXpf8gt34AkfL?=
 =?iso-8859-1?Q?hYShrEL3bPvA2189J+OswS2BB34mMKgmzXRmw9muuBVtpXjtpaZJvpzBWG?=
 =?iso-8859-1?Q?k8PCt2w+Qbxnnw9iegtvkrdf8LzelRKXCtYGL7vCrLo+q4EVRr8lmdyOaL?=
 =?iso-8859-1?Q?/8Yasfb7YatwA/dXEpmIyqHsXAeOXsJS5v4xZDzMY9D5sB0262CPpy0ikD?=
 =?iso-8859-1?Q?Y7MKmBywQZf3wDBJxbRrfmoeZABM0mLnrYsx7q4fmnQ4MHRovlfAfL64A/?=
 =?iso-8859-1?Q?438cLSty+iChdR2hrIyKfT2vSpNom4xz5VvzuqQwqQl2xg1HVjYpJoKrU2?=
 =?iso-8859-1?Q?5VlPBScpTYuuSxkNAdFyL3ikDHcvbi5h6ds1DlqxQailnthPz963/lZB0O?=
 =?iso-8859-1?Q?CI99ifLmFu76B0GjxyfaGeMhQFRSAHs1xBPjQOcjFDC8CgQFuDHwU1lRg8?=
 =?iso-8859-1?Q?N9FAENkUcCGQBN2nMlyh5P0w9TD3laIY1I1cWEw+aTj++dZPSOpS7boCdM?=
 =?iso-8859-1?Q?kJpifLEVSPByvRhXTmnZIpqzY6PoZXPHBdyT0k2+hYC7KQM0VKgcDAWA7q?=
 =?iso-8859-1?Q?uNxtas/MRzpz3rtJLJQ+3fPyvvyqp55TmdpJVx7Rvp7aDqWJmos7TwCf/q?=
 =?iso-8859-1?Q?r+1SbJsUzYTbhgiH/VMNMkpzyz3Yhy7TMyPHmjMrcarNuafa5IUKtkaZxS?=
 =?iso-8859-1?Q?fQ6yl8pg1yIcuMV6mqN/05a/qasUljRgb5KdMAnmUb94QUqUYvgC0bmGUZ?=
 =?iso-8859-1?Q?ZJA/qwkwycL/oA3nJm9HEscmKJ2zqIpmIVGUaktuZwopHxU35bwPIBFPED?=
 =?iso-8859-1?Q?RwWKoVON0iGfKECUQpThBcKPL+os8ozXIGWCqwU4/CfE3fT9PAdEV9zvEN?=
 =?iso-8859-1?Q?nsjHNfJrWY4BB0PDWVkYFojp2XaKtSHxUSzipFXxdBbg1MIq5a82Vd3fQ/?=
 =?iso-8859-1?Q?gw4ZpiqPMrpo5BMfBnUtFvH3G9qHCz4zxa8vGXMuWrNW0SW5Lubq3+YhRf?=
 =?iso-8859-1?Q?E+Fbcro3tTytwtD7ODOyVBszVC4T3puBI88D/aeWPPdrBJbw6Vnlup6YzY?=
 =?iso-8859-1?Q?tm37B1t4wvLe+k/WpSeJhqWWbrIoLsQodx7XgqMm0CxdzR8xdcluDgJz/k?=
 =?iso-8859-1?Q?gK1F7+vV2C3kg4urD5xrC+JpkKdL7JmJZpsPxxH6Z8yNhJG5n3ZHG/y3JI?=
 =?iso-8859-1?Q?/4jW7WhSehteC8yA2hZwdMpoAqdPOl8eZoUGgVd9iACM7ZRs3oaPAzK2z0?=
 =?iso-8859-1?Q?SGmTKnTZImyiAmVb8FUspd5Xcz6KpICHJroEJT90/csuMD8VmJkdWNsL3l?=
 =?iso-8859-1?Q?DancdfQ9u63rIyySgoos/PZmPtAxXP4GSUouZkR3mvla0qM137heDIbeeN?=
 =?iso-8859-1?Q?cobBW0iBrq7cRyYiUznI0V2Qckznq1KEAl3aOcw2q+6iPlXyPufmI+YlW/?=
 =?iso-8859-1?Q?sW5cHKouSXQ4TNF7LmpjzHWjgQoQsIb+N6V6Pxesr0RYLu9re4sh/9HIbI?=
 =?iso-8859-1?Q?OWqhh5scqrSEj5O/Qks5zfcuQ9WXYDgP4jWS1cXKq+3xI/HSfEhtnzFKhg?=
 =?iso-8859-1?Q?pnPebVAQEWqBYcvPZPLSrh8+R6d4oCm8ZeWGe1zjBGfhf0w3iObswaCXAM?=
 =?iso-8859-1?Q?IOj0ZKPzqhFM7YpOCRh/QwCm/3cs1iVcurZeUjVhMKNlw8LUhLqylIorsc?=
 =?iso-8859-1?Q?fQN1IsVfdrhQ8XAoPAM=3D?=
Content-Type: text/plain; charset="iso-8859-1"
Content-Transfer-Encoding: quoted-printable
MIME-Version: 1.0
X-OriginatorOrg: oracle.com
X-MS-Exchange-CrossTenant-AuthAs: Internal
X-MS-Exchange-CrossTenant-AuthSource: SN6PR10MB3022.namprd10.prod.outlook.com
X-MS-Exchange-CrossTenant-Network-Message-Id: a103ec47-e3b4-488e-7386-08d9e5f59377
X-MS-Exchange-CrossTenant-originalarrivaltime: 02 Feb 2022 02:41:54.4557
 (UTC)
X-MS-Exchange-CrossTenant-fromentityheader: Hosted
X-MS-Exchange-CrossTenant-id: 4e2c6054-71cb-48f1-bd6c-3a9705aca71b
X-MS-Exchange-CrossTenant-mailboxtype: HOSTED
X-MS-Exchange-CrossTenant-userprincipalname: 4x3mRERUmpPmUoBRbNvucaPoHHFWv9lk5oKkat4Gv7wmx42ygGr+VK9wvJwYdbqNI0Uj2Ynav4pOdMfmbLjOMA==
X-MS-Exchange-Transport-CrossTenantHeadersStamped: MWHPR10MB1325
X-Proofpoint-Virus-Version: vendor=nai engine=6300 definitions=10245 signatures=673430
X-Proofpoint-Spam-Details: rule=notspam policy=default score=0 mlxlogscore=999 adultscore=0
 suspectscore=0 malwarescore=0 bulkscore=0 mlxscore=0 spamscore=0
 phishscore=0 classifier=spam adjust=0 reason=mlx scancount=1
 engine=8.12.0-2201110000 definitions=main-2202020010
X-Proofpoint-GUID: NlJBB6dJW4OIGjNshW4VTM-3AJ_kka85
X-Proofpoint-ORIG-GUID: NlJBB6dJW4OIGjNshW4VTM-3AJ_kka85
Precedence: bulk
List-ID: <linux-kernel.vger.kernel.org>
X-Mailing-List: linux-kernel@vger.kernel.org

From: "Liam R. Howlett" <Liam.Howlett@Oracle.com>

Add functions to get the number of allocations, and total allocations
from a kmem_cache.  Also add a function to get the allocated size and a
way to zero the total allocations.

Signed-off-by: Liam R. Howlett <Liam.Howlett@Oracle.com>
---
 tools/testing/radix-tree/linux.c | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/tools/testing/radix-tree/linux.c b/tools/testing/radix-tree/li=
nux.c
index 00ee01a14652..f95e71d65f00 100644
--- a/tools/testing/radix-tree/linux.c
+++ b/tools/testing/radix-tree/linux.c
@@ -25,6 +25,8 @@ struct kmem_cache {
 	void *objs;
 	void (*ctor)(void *);
 	unsigned int non_kernel;
+	unsigned long nr_allocated;
+	unsigned long nr_tallocated;
 };
=20
 void kmem_cache_set_non_kernel(struct kmem_cache *cachep, unsigned int val=
)
@@ -32,6 +34,26 @@ void kmem_cache_set_non_kernel(struct kmem_cache *cachep=
, unsigned int val)
 	cachep->non_kernel =3D val;
 }
=20
+unsigned long kmem_cache_get_alloc(struct kmem_cache *cachep)
+{
+	return cachep->size * cachep->nr_allocated;
+}
+
+unsigned long kmem_cache_nr_allocated(struct kmem_cache *cachep)
+{
+	return cachep->nr_allocated;
+}
+
+unsigned long kmem_cache_nr_tallocated(struct kmem_cache *cachep)
+{
+	return cachep->nr_tallocated;
+}
+
+void kmem_cache_zero_nr_tallocated(struct kmem_cache *cachep)
+{
+	cachep->nr_tallocated =3D 0;
+}
+
 void *kmem_cache_alloc(struct kmem_cache *cachep, int gfp)
 {
 	void *p;
@@ -63,7 +85,9 @@ void *kmem_cache_alloc(struct kmem_cache *cachep, int gfp=
)
 			memset(p, 0, cachep->size);
 	}
=20
+	uatomic_inc(&cachep->nr_allocated);
 	uatomic_inc(&nr_allocated);
+	uatomic_inc(&cachep->nr_tallocated);
 	if (kmalloc_verbose)
 		printf("Allocating %p from slab\n", p);
 	return p;
@@ -73,6 +97,7 @@ void kmem_cache_free(struct kmem_cache *cachep, void *obj=
p)
 {
 	assert(objp);
 	uatomic_dec(&nr_allocated);
+	uatomic_dec(&cachep->nr_allocated);
 	if (kmalloc_verbose)
 		printf("Freeing %p to slab\n", objp);
 	pthread_mutex_lock(&cachep->lock);
@@ -124,6 +149,8 @@ kmem_cache_create(const char *name, unsigned int size, =
unsigned int align,
 	ret->size =3D size;
 	ret->align =3D align;
 	ret->nr_objs =3D 0;
+	ret->nr_allocated =3D 0;
+	ret->nr_tallocated =3D 0;
 	ret->objs =3D NULL;
 	ret->ctor =3D ctor;
 	ret->non_kernel =3D 0;
--=20
2.34.1