1 files changed, 8 insertions, 2 deletions
diff --git a/tests/src/util.c b/tests/src/util.c
index 69f5c34..0d8040d 100644
--- a/tests/src/util.c
+++ b/tests/src/util.c
@@ -1,6 +1,7 @@
 #include "util.h"
 
 #include <stdio.h>
+#include <stdlib.h>
 
 #include "libfdt.h"
 #include "libufdt_sysdeps.h"
@@ -29,7 +30,8 @@ char *load_file(const char *filename, size_t *pLen) {
   return buf;
 }
 
-int write_fdt_to_file(const char *filename, void *fdt) {
+int write_buf_to_file(const char *filename,
+                      const void *buf, size_t buf_size) {
   int ret = 0;
   FILE *fout = NULL;
 
@@ -38,7 +40,7 @@ int write_fdt_to_file(const char *filename, void *fdt) {
     ret = 1;
     goto end;
   }
-  if (fwrite(fdt, 1, fdt_totalsize(fdt), fout) < 1) {
+  if (fwrite(buf, 1, buf_size, fout) < 1) {
     ret = 2;
     goto end;
   }
@@ -48,3 +50,7 @@ end:
 
   return ret;
 }
+
+int write_fdt_to_file(const char *filename, const void *fdt) {
+  return write_buf_to_file(filename, fdt, fdt_totalsize(fdt));
+}