[PATCH v1] s390/dasd: Fix memory leak in FBA driver initialization

[Volkan Unal <vunal@linux.ibm.com>]

Fix memory leak in dasd_fba_init() where the allocated zero page
is not freed if ccw_driver_register() fails.

Reviewed-by: Jan Höppner <hoeppner@linux.ibm.com>
Reviewed-by: Stefan Haberland <sth@linux.ibm.com>
Signed-off-by: Volkan Unal <vunal@linux.ibm.com>
---
 drivers/s390/block/dasd_fba.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/drivers/s390/block/dasd_fba.c b/drivers/s390/block/dasd_fba.c
index 3f56b353b868..3b8d15e9a2de 100644
--- a/drivers/s390/block/dasd_fba.c
+++ b/drivers/s390/block/dasd_fba.c
@@ -804,10 +804,13 @@ dasd_fba_init(void)
 		return -ENOMEM;
 
 	ret = ccw_driver_register(&dasd_fba_driver);
-	if (!ret)
-		wait_for_device_probe();
+	if (ret) {
+		free_page((unsigned long)dasd_fba_zero_page);
+		return ret;
+	}
 
-	return ret;
+	wait_for_device_probe();
+	return 0;
 }
 
 static void __exit
-- 
2.55.0