tile: add null check for kzalloc in tile/kernel/setup.c

Should check the return value of kzalloc first to avoid the null pointer.
Then can dereference the non-null pointer to access the fields of struct
resource.

Signed-off-by: Wang Sheng-Hui <shhuiw@gmail.com>
Signed-off-by: Chris Metcalf <cmetcalf@tilera.com>

1 files changed, 4 insertions, 0 deletions

diff --git a/arch/tile/kernel/setup.c b/arch/tile/kernel/setup.c
index 128a2d0b8650..c2c42f329c1a 100644
--- a/arch/tile/kernel/setup.c
+++ b/arch/tile/kernel/setup.c
@@ -1643,6 +1643,8 @@ insert_non_bus_resource(void)
 {
         struct resource *res =
                 kzalloc(sizeof(struct resource), GFP_ATOMIC);
+        if (!res)
+                return NULL;
         res->name = "Non-Bus Physical Address Space";
         res->start = (1ULL << 32);
         res->end = -1LL;
@@ -1660,6 +1662,8 @@ insert_ram_resource(u64 start_pfn, u64 end_pfn, bool reserved)
 {
         struct resource *res =
                 kzalloc(sizeof(struct resource), GFP_ATOMIC);
+        if (!res)
+                return NULL;
         res->name = reserved ? "Reserved" : "System RAM";
         res->start = start_pfn << PAGE_SHIFT;
         res->end = (end_pfn << PAGE_SHIFT) - 1;