py: Add further checks for failed malloc in lexer init functions.

1 files changed, 4 insertions, 1 deletions

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);