WIP.

3 files changed, 10 insertions, 17 deletions

diff --git a/upb/decode.c b/upb/decode.c
index 96622f9..79bda54 100644
--- a/upb/decode.c
+++ b/upb/decode.c
@@ -33,6 +33,7 @@ typedef struct {
   /* Current decoding pointer.  Points to the beginning of a field until we
    * have finished decoding the whole field. */
   const char *ptr;
+  upb_arena *arena;
 } upb_decstate;
 
 /* Data pertaining to a single message frame. */
@@ -557,10 +558,11 @@ static bool upb_decode_message(upb_decstate *d, const char *limit,
   return true;
 }
 
-bool upb_decode(const char *buf, size_t size, void *msg,
-                const upb_msglayout *l) {
+bool upb_decode(const char *buf, size_t size, void *msg, const upb_msglayout *l,
+                upb_arena *arena) {
   upb_decstate state;
   state.ptr = buf;
+  state.arena = arena;
 
   return upb_decode_message(&state, buf + size, 0, msg, l);
 }
diff --git a/upb/msg.c b/upb/msg.c
index ca2766c..31607b7 100644
--- a/upb/msg.c
+++ b/upb/msg.c
@@ -13,9 +13,6 @@
 
 /* Used when a message is not extendable. */
 typedef struct {
-  /* TODO(haberman): use pointer tagging so we we are slim when known unknown
-   * fields are not present. */
-  upb_arena *arena;
   char *unknown;
   size_t unknown_len;
   size_t unknown_size;
@@ -49,10 +46,6 @@ static upb_msg_internal_withext *upb_msg_getinternalwithext(
   return VOIDPTR_AT(msg, -sizeof(upb_msg_internal_withext));
 }
 
-upb_arena *upb_msg_arena(const upb_msg *msg) {
-  return upb_msg_getinternal_const(msg)->arena;
-}
-
 upb_msg *upb_msg_new(const upb_msglayout *l, upb_arena *a) {
   upb_alloc *alloc = upb_arena_alloc(a);
   void *mem = upb_malloc(alloc, upb_msg_sizeof(l));
@@ -70,7 +63,6 @@ upb_msg *upb_msg_new(const upb_msglayout *l, upb_arena *a) {
 
   /* Initialize internal members. */
   in = upb_msg_getinternal(msg);
-  in->arena = a;
   in->unknown = NULL;
   in->unknown_len = 0;
   in->unknown_size = 0;
@@ -99,10 +91,11 @@ upb_array *upb_array_new(upb_fieldtype_t type, upb_arena *a) {
   return ret;
 }
 
-void upb_msg_addunknown(upb_msg *msg, const char *data, size_t len) {
-  upb_msg_internal* in = upb_msg_getinternal(msg);
+void upb_msg_addunknown(upb_msg *msg, const char *data, size_t len,
+                        upb_arena *arena) {
+  upb_msg_internal *in = upb_msg_getinternal(msg);
   if (len > in->unknown_size - in->unknown_len) {
-    upb_alloc *alloc = upb_arena_alloc(in->arena);
+    upb_alloc *alloc = upb_arena_alloc(arena);
     size_t need = in->unknown_size + len;
     size_t newsize = UPB_MAX(in->unknown_size * 2, need);
     in->unknown = upb_realloc(alloc, in->unknown, in->unknown_size, newsize);
diff --git a/upb/msg.h b/upb/msg.h
index e46733f..7f1a6e7 100644
--- a/upb/msg.h
+++ b/upb/msg.h
@@ -58,10 +58,8 @@ typedef struct {
 upb_msg *upb_msg_new(const upb_msglayout *l, upb_arena *a);
 upb_msg *upb_msg_new(const upb_msglayout *l, upb_arena *a);
 
-/* Returns the arena for the given message. */
-upb_arena *upb_msg_arena(const upb_msg *msg);
-
-void upb_msg_addunknown(upb_msg *msg, const char *data, size_t len);
+void upb_msg_addunknown(upb_msg *msg, const char *data, size_t len,
+                        upb_arena *arena);
 const char *upb_msg_getunknown(const upb_msg *msg, size_t *len);
 
 upb_array *upb_array_new(upb_fieldtype_t type, upb_arena *a);