From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id 6B854C433FE for ; Wed, 5 Oct 2022 09:03:56 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 377876B0072; Wed, 5 Oct 2022 05:03:55 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 3271B6B0073; Wed, 5 Oct 2022 05:03:55 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 1EE598E0001; Wed, 5 Oct 2022 05:03:55 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0011.hostedemail.com [216.40.44.11]) by kanga.kvack.org (Postfix) with ESMTP id 0DAE16B0072 for ; Wed, 5 Oct 2022 05:03:55 -0400 (EDT) Received: from smtpin08.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay01.hostedemail.com (Postfix) with ESMTP id C44A21C6798 for ; Wed, 5 Oct 2022 09:03:54 +0000 (UTC) X-FDA: 79986308388.08.E2B5620 Received: from relay.virtuozzo.com (relay.virtuozzo.com [130.117.225.111]) by imf26.hostedemail.com (Postfix) with ESMTP id 2FD1A140023 for ; Wed, 5 Oct 2022 09:03:53 +0000 (UTC) Received: from dev011.ch-qa.sw.ru ([172.29.1.16]) by relay.virtuozzo.com with esmtp (Exim 4.95) (envelope-from ) id 1og0GL-007ckN-6V; Wed, 05 Oct 2022 11:02:45 +0200 From: Alexander Atanasov To: "Michael S. Tsirkin" , David Hildenbrand , Andrew Morton Cc: kernel@openvz.org, Alexander Atanasov , virtualization@lists.linux-foundation.org, linux-kernel@vger.kernel.org, linux-mm@kvack.org Subject: [PATCH v4 2/7] Enable balloon drivers to report inflated memory Date: Wed, 5 Oct 2022 12:01:52 +0300 Message-Id: <20221005090158.2801592-3-alexander.atanasov@virtuozzo.com> X-Mailer: git-send-email 2.31.1 In-Reply-To: <20221005090158.2801592-1-alexander.atanasov@virtuozzo.com> References: <20221005090158.2801592-1-alexander.atanasov@virtuozzo.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1664960634; a=rsa-sha256; cv=none; b=ZNU0lNrq0pZ3Hl6FIoc1C2idHqhl7HE3km/3aEbSC/Kenl8sCYOEptVCcaYCpEt7AyzKKw gdYndEeUU9yOOXkVkomydApb2hSZfnulKTPGUrDJQgUryqwwTzLStmztMskZ6sYk/BnwcB huyfFXai1tFVITsH02hsZoXQaE69ZDQ= ARC-Authentication-Results: i=1; imf26.hostedemail.com; dkim=none; spf=pass (imf26.hostedemail.com: domain of alexander.atanasov@virtuozzo.com designates 130.117.225.111 as permitted sender) smtp.mailfrom=alexander.atanasov@virtuozzo.com; dmarc=pass (policy=quarantine) header.from=virtuozzo.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1664960634; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=W1Qualv07jAb7hBW1fXyEkKq4YxXnRaD0fxYQrBWXEw=; b=6+w0hSh80dGUnjDWN9tzOmhNR7J5x0bdPap6Tm74k9oXPkNOkG7TfMY+OveETuesxj3f2Y BbTu4Ei7vdYPMiNtzbSZdSmJCZIRdhMMyoxhYSwmYYhhbD2sOmDHN6kCdHpC7lvca2ckmt ry4i0/pfQ/TvEjggigNBEaoKePvePqE= X-Rspamd-Server: rspam03 X-Rspamd-Queue-Id: 2FD1A140023 X-Rspam-User: Authentication-Results: imf26.hostedemail.com; dkim=none; spf=pass (imf26.hostedemail.com: domain of alexander.atanasov@virtuozzo.com designates 130.117.225.111 as permitted sender) smtp.mailfrom=alexander.atanasov@virtuozzo.com; dmarc=pass (policy=quarantine) header.from=virtuozzo.com X-Stat-Signature: 35idasyif5caxhxktrpknwyugjr7z4kk X-HE-Tag: 1664960633-862903 X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: Add counters to be updated by the balloon drivers. Create balloon notifier to propagate changes. Signed-off-by: Alexander Atanasov --- include/linux/balloon.h | 18 ++++++++++++++++++ mm/balloon.c | 36 ++++++++++++++++++++++++++++++++++++ 2 files changed, 54 insertions(+) diff --git a/include/linux/balloon.h b/include/linux/balloon.h index 46ac8f61f607..59657da77d95 100644 --- a/include/linux/balloon.h +++ b/include/linux/balloon.h @@ -57,6 +57,24 @@ struct balloon_dev_info { struct page *page, enum migrate_mode mode); }; +extern atomic_long_t mem_balloon_inflated_total_kb; +extern atomic_long_t mem_balloon_inflated_free_kb; + +void balloon_set_inflated_total(long inflated_kb); +void balloon_set_inflated_free(long inflated_kb); + +#define BALLOON_CHANGED_TOTAL 0 +#define BALLOON_CHANGED_FREE 1 + +int register_balloon_notifier(struct notifier_block *nb); +void unregister_balloon_notifier(struct notifier_block *nb); + +#define balloon_notifier(fn, pri) ({ \ + static struct notifier_block fn##_mem_nb __meminitdata =\ + { .notifier_call = fn, .priority = pri }; \ + register_balloon_notifier(&fn##_mem_nb); \ +}) + struct page *balloon_page_alloc(void); void balloon_page_enqueue(struct balloon_dev_info *b_dev_info, struct page *page); diff --git a/mm/balloon.c b/mm/balloon.c index 22b3e876bc78..8e1d5855fef8 100644 --- a/mm/balloon.c +++ b/mm/balloon.c @@ -7,8 +7,44 @@ #include #include #include +#include #include +atomic_long_t mem_balloon_inflated_total_kb = ATOMIC_LONG_INIT(0); +atomic_long_t mem_balloon_inflated_free_kb = ATOMIC_LONG_INIT(0); +SRCU_NOTIFIER_HEAD_STATIC(balloon_chain); + +int register_balloon_notifier(struct notifier_block *nb) +{ + return srcu_notifier_chain_register(&balloon_chain, nb); +} +EXPORT_SYMBOL(register_balloon_notifier); + +void unregister_balloon_notifier(struct notifier_block *nb) +{ + srcu_notifier_chain_unregister(&balloon_chain, nb); +} +EXPORT_SYMBOL(unregister_balloon_notifier); + +static int balloon_notify(unsigned long val) +{ + return srcu_notifier_call_chain(&balloon_chain, val, NULL); +} + +void balloon_set_inflated_total(long inflated_kb) +{ + atomic_long_set(&mem_balloon_inflated_total_kb, inflated_kb); + balloon_notify(BALLOON_CHANGED_TOTAL); +} +EXPORT_SYMBOL(balloon_set_inflated_total); + +void balloon_set_inflated_free(long inflated_kb) +{ + atomic_long_set(&mem_balloon_inflated_free_kb, inflated_kb); + balloon_notify(BALLOON_CHANGED_FREE); +} +EXPORT_SYMBOL(balloon_set_inflated_free); + static void balloon_page_enqueue_one(struct balloon_dev_info *b_dev_info, struct page *page) { -- 2.31.1