From 367e12597617b2581eb72b3676c6cb86822268b5 Mon Sep 17 00:00:00 2001
From: Joe Hershberger <joe.hershberger@ni.com>
Date: Fri, 17 Aug 2012 10:34:35 +0000
Subject: [PATCH] fdt: Check error codes returned from fdtlib when loading ITB

Before this patch, error codes returned from fdtlib were ignored and
continued access would cause a crash. Now just check if the image is
truncated and error if so.

Signed-off-by: Joe Hershberger <joe.hershberger@ni.com>
---
 common/image.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/common/image.c b/common/image.c
index f084d2baed3..852a96fab8b 100644
--- a/common/image.c
+++ b/common/image.c
@@ -2820,6 +2820,11 @@ int fit_image_check_hashes(const void *fit, int image_noffset)
 		}
 	}
 
+	if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) {
+		err_msg = " error!\nCorrupted or truncated tree";
+		goto error;
+	}
+
 	return 1;
 
 error:
-- 
GitLab