add msgb_set_talloc_ctx() to set the talloc context for msgb allocations

2 files changed, 7 insertions, 0 deletions

diff --git a/include/osmocom/core/msgb.h b/include/osmocom/core/msgb.h
index 8665c2bf..58976f7f 100644
--- a/include/osmocom/core/msgb.h
+++ b/include/osmocom/core/msgb.h
@@ -193,5 +193,7 @@ static inline struct msgb *msgb_alloc_headroom(int size, int headroom,
 uint8_t *msgb_data(const struct msgb *msg);
 uint16_t msgb_length(const struct msgb *msg);
 
+/* set the talloc context for msgb_alloc[_headroom] */
+void msgb_set_talloc_ctx(void *ctx);
 
 #endif /* _MSGB_H */
diff --git a/src/msgb.c b/src/msgb.c
index d2c167aa..f9841ed1 100644
--- a/src/msgb.c
+++ b/src/msgb.c
@@ -98,3 +98,8 @@ uint16_t msgb_length(const struct msgb *msg)
 {
 	return msg->len;
 }
+
+void msgb_set_talloc_ctx(void *ctx)
+{
+	tall_msgb_ctx = ctx;
+}