From 9bf5f2857d2015e3d440feb03c102ed9730c8f4d Mon Sep 17 00:00:00 2001
From: Damien George
Date: Thu, 9 Oct 2014 16:53:37 +0100
Subject: py: Add further checks for failed malloc in lexer init functions.

---
 stmhal/lexerfatfs.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'stmhal')

diff --git a/stmhal/lexerfatfs.c b/stmhal/lexerfatfs.c
index 21e3a2007..ea47aaf88 100644
--- a/stmhal/lexerfatfs.c
+++ b/stmhal/lexerfatfs.c
@@ -64,7 +64,10 @@ STATIC void file_buf_close(mp_lexer_file_buf_t *fb) {
 }
 
 mp_lexer_t *mp_lexer_new_from_file(const char *filename) {
-    mp_lexer_file_buf_t *fb = m_new_obj(mp_lexer_file_buf_t);
+    mp_lexer_file_buf_t *fb = m_new_obj_maybe(mp_lexer_file_buf_t);
+    if (fb == NULL) {
+        return NULL;
+    }
     FRESULT res = f_open(&fb->fp, filename, FA_READ);
     if (res != FR_OK) {
         m_del_obj(mp_lexer_file_buf_t, fb);
-- 
cgit v1.2.3