extmod/modbtree: Check __bt_open() return value for error.

pfalcon · pfalcon · commit 25df419c6703 · 2016-07-24T00:29:32.000+03:00
diff --git a/extmod/modbtree.c b/extmod/modbtree.c
@@ -331,6 +331,9 @@ STATIC mp_obj_t mod_btree_open(size_t n_args, const mp_obj_t *pos_args, mp_map_t
         MP_ARRAY_SIZE(allowed_args), allowed_args, (mp_arg_val_t*)&args);
 
     DB *db = __bt_open(fname, /*flags*/O_CREAT | O_RDWR, /*mode*/0770, /*openinfo*/NULL, /*dflags*/0);
+    if (db == NULL) {
+        nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(errno)));
+    }
     return MP_OBJ_FROM_PTR(btree_new(db));
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_KW(mod_btree_open_obj, 1, mod_btree_open);

Original file line number	Diff line number	Diff line change
`@@ -331,6 +331,9 @@ STATIC mp_obj_t mod_btree_open(size_t n_args, const mp_obj_t *pos_args, mp_map_t`
`331`	`331`	`MP_ARRAY_SIZE(allowed_args), allowed_args, (mp_arg_val_t*)&args);`
`332`	`332`
`333`	`333`	`DB db = __bt_open(fname, /flags/O_CREAT \| O_RDWR, /mode/0770, /openinfo/NULL, /dflags*/0);`
	`334`	`+ if (db == NULL) {`
	`335`	`+ nlr_raise(mp_obj_new_exception_arg1(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(errno)));`
	`336`	`+ }`
`334`	`337`	`return MP_OBJ_FROM_PTR(btree_new(db));`
`335`	`338`	`}`
`336`	`339`	`STATIC MP_DEFINE_CONST_FUN_OBJ_KW(mod_btree_open_obj, 1, mod_btree_open);`