From: Kent Overstreet <kent.overstr...@linux.dev>
Memory allocation profiling is turning krealloc() into a nontrivial
macro - so for now, we need a helper for it.
Until we have proper support on the rust side for memory allocation
profiling this does mean that all Rust allocations will be accounted to
the helper.
Signed-off-by: Kent Overstreet <kent.overstr...@linux.dev>
Cc: Miguel Ojeda <oj...@kernel.org>
Cc: Alex Gaynor <alex.gay...@gmail.com>
Cc: Wedson Almeida Filho <wedso...@gmail.com>
Cc: Boqun Feng <boqun.f...@gmail.com>
Cc: Gary Guo <g...@garyguo.net>
Cc: "Björn Roy Baron" <bjorn3...@protonmail.com>
Cc: Benno Lossin <benno.los...@proton.me>
Cc: Andreas Hindborg <a.hindb...@samsung.com>
Cc: Alice Ryhl <alicer...@google.com>
Cc: rust-for-li...@vger.kernel.org
Signed-off-by: Suren Baghdasaryan <sur...@google.com>
Reviewed-by: Alice Ryhl <alicer...@google.com>
---
rust/helpers.c | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/rust/helpers.c b/rust/helpers.c
index 70e59efd92bc..ad62eaf604b3 100644
--- a/rust/helpers.c
+++ b/rust/helpers.c
@@ -28,6 +28,7 @@
#include <linux/mutex.h>
#include <linux/refcount.h>
#include <linux/sched/signal.h>
+#include <linux/slab.h>
#include <linux/spinlock.h>
#include <linux/wait.h>
#include <linux/workqueue.h>
@@ -157,6 +158,13 @@ void rust_helper_init_work_with_key(struct work_struct
*work, work_func_t func,
}
EXPORT_SYMBOL_GPL(rust_helper_init_work_with_key);
+void * __must_check rust_helper_krealloc(const void *objp, size_t new_size,
+ gfp_t flags) __realloc_size(2)
+{
+ return krealloc(objp, new_size, flags);
+}
+EXPORT_SYMBOL_GPL(rust_helper_krealloc);
+
/*
* `bindgen` binds the C `size_t` type as the Rust `usize` type, so we can
* use it in contexts where Rust expects a `usize` like slice (array) indices.
--
2.44.0.278.ge034bb2e1d-goog
