From e4f7cf7bc5678d3ad26818625fb5b3e49c7af94d Mon Sep 17 00:00:00 2001
From: Pali Rohár <pali.rohar@gmail.com>
Date: Sun, 23 Sep 2012 16:48:24 +0200
Subject: main: Fix loading images

---
 src/main.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/main.c')

diff --git a/src/main.c b/src/main.c
index 6c279b1..3e75bf9 100644
--- a/src/main.c
+++ b/src/main.c
@@ -980,7 +980,7 @@ int main(int argc, char **argv) {
 						if ( have_kernel == 0 ) {
 							image_kernel = image_ptr->image;
 							have_kernel = 1;
-						} else if ( have_kernel == 1 ) {
+						} else if ( have_kernel == 1 && image_kernel != image_ptr->image ) {
 							image_kernel = NULL;
 							have_kernel = 2;
 						}
@@ -995,7 +995,7 @@ int main(int argc, char **argv) {
 						if ( have_initfs == 0 ) {
 							image_initfs = image_ptr->image;
 							have_initfs = 1;
-						} else if ( have_initfs == 1 ) {
+						} else if ( have_initfs == 1 && image_initfs != image_ptr->image ) {
 							image_initfs = NULL;
 							have_initfs = 2;
 						}
-- 
cgit v1.2.3