summaryrefslogtreecommitdiff
path: root/tests/mangle-layout.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/mangle-layout.c')
-rw-r--r--tests/mangle-layout.c8
1 files changed, 2 insertions, 6 deletions
diff --git a/tests/mangle-layout.c b/tests/mangle-layout.c
index a76e51e..ae01a55 100644
--- a/tests/mangle-layout.c
+++ b/tests/mangle-layout.c
@@ -44,13 +44,9 @@ static void expand_buf(struct bufstate *buf, int newsize)
static void new_header(struct bufstate *buf, int version, const void *fdt)
{
- int hdrsize;
+ int hdrsize = fdt_header_size_(version);
- if (version == 16)
- hdrsize = FDT_V16_SIZE;
- else if (version == 17)
- hdrsize = FDT_V17_SIZE;
- else
+ if ((version != 16) && (version != 17))
CONFIG("Bad version %d", version);
expand_buf(buf, hdrsize);