1 files changed, 41 insertions, 36 deletions

diff --git a/src/reader.h b/src/reader.h
index abae5d92..92baac3f 100644
--- a/src/reader.h
+++ b/src/reader.h
@@ -35,18 +35,13 @@
 #    define SERD_STACK_ASSERT_TOP(reader, ref)
 #endif
 
-/* Reference to a node in the stack (we can not use pointers since the
-   stack may be reallocated, invalidating any pointers to elements).
-*/
-typedef size_t Ref;
-
 typedef struct {
-	Ref                 graph;
-	Ref                 subject;
-	Ref                 predicate;
-	Ref                 object;
-	Ref                 datatype;
-	Ref                 lang;
+	SerdNode*           graph;
+	SerdNode*           subject;
+	SerdNode*           predicate;
+	SerdNode*           object;
+	SerdNode*           datatype;
+	SerdNode*           lang;
 	SerdStatementFlags* flags;
 } ReadContext;
 
@@ -55,9 +50,9 @@ struct SerdReaderImpl {
 	const SerdSinkInterface* sink;
 	SerdErrorSink            error_sink;
 	void*                    error_handle;
-	Ref                      rdf_first;
-	Ref                      rdf_rest;
-	Ref                      rdf_nil;
+	SerdNode*                rdf_first;
+	SerdNode*                rdf_rest;
+	SerdNode*                rdf_nil;
 	SerdNode*                default_graph;
 	SerdByteSource           source;
 	SerdStack                stack;
@@ -70,33 +65,35 @@ struct SerdReaderImpl {
 	bool                     strict; ///< True iff strict parsing
 	bool                     seen_genid;
 #ifdef SERD_STACK_CHECK
-	Ref*                     allocs; ///< Stack of push offsets
+	SerdNode**               allocs; ///< Stack of push offsets
 	size_t                   n_allocs; ///< Number of stack pushes
 #endif
 };
 
 int r_err(SerdReader* reader, SerdStatus st, const char* fmt, ...);
 
-Ref push_node_padded(SerdReader* reader,
-                     size_t      maxlen,
-                     SerdType    type,
-                     const char* str,
-                     size_t      n_bytes);
-
-Ref push_node(SerdReader* reader,
-              SerdType    type,
-              const char* str,
-              size_t      n_bytes);
+SerdNode* push_node_padded(SerdReader* reader,
+                           size_t      maxlen,
+                           SerdType    type,
+                           const char* str,
+                           size_t      n_bytes);
 
-size_t genid_size(SerdReader* reader);
-Ref    blank_id(SerdReader* reader);
-void   set_blank_id(SerdReader* reader, Ref ref, size_t buf_size);
+SerdNode* push_node(SerdReader* reader,
+                    SerdType    type,
+                    const char* str,
+                    size_t      n_bytes);
 
-SerdNode* deref(SerdReader* reader, Ref ref);
+size_t    genid_size(SerdReader* reader);
+SerdNode* blank_id(SerdReader* reader);
+void      set_blank_id(SerdReader* reader, SerdNode* node, size_t buf_size);
 
-Ref pop_node(SerdReader* reader, Ref ref);
+SerdNode* pop_node(SerdReader* reader, const SerdNode* node);
 
-bool emit_statement(SerdReader* reader, ReadContext ctx, Ref o, Ref d, Ref l);
+bool emit_statement(SerdReader* reader,
+                    ReadContext ctx,
+                    SerdNode*   o,
+                    SerdNode*   d,
+                    SerdNode*   l);
 
 bool read_n3_statement(SerdReader* reader);
 SerdStatus read_nquadsDoc(SerdReader* reader);
@@ -149,23 +146,31 @@ eat_string(SerdReader* reader, const char* str, unsigned n)
 }
 
 static inline SerdStatus
-push_byte(SerdReader* reader, Ref ref, const uint8_t c)
+push_byte(SerdReader* reader, SerdNode* node, const uint8_t c)
 {
 	SERD_STACK_ASSERT_TOP(reader, ref);
-	char* const     s    = (char*)serd_stack_push(&reader->stack, 1);
-	SerdNode* const node = (SerdNode*)(reader->stack.buf + ref);
+	char* const s = (char*)serd_stack_push(&reader->stack, 1);
+	if (!s) {
+		return SERD_ERR_OVERFLOW;
+	}
+
 	++node->n_bytes;
 	*(s - 1) = c;
 	*s       = '\0';
 	return SERD_SUCCESS;
 }
 
-static inline void
-push_bytes(SerdReader* reader, Ref ref, const uint8_t* bytes, unsigned len)
+static inline SerdStatus
+push_bytes(SerdReader* reader, SerdNode* ref, const uint8_t* bytes, unsigned len)
 {
+	if (reader->stack.buf_size < reader->stack.size + len) {
+		return SERD_ERR_OVERFLOW;
+	}
+
 	for (unsigned i = 0; i < len; ++i) {
 		push_byte(reader, ref, bytes[i]);
 	}
+	return SERD_SUCCESS;
 }
 
 #endif // SERD_READER_H