WIP extmod/modussl_axtls: Check NULL return from mp_obj_str_get_data.

dpgeorge · dpgeorge · commit 456169491ba2 · 2019-12-13T22:42:45.000+11:00
diff --git a/extmod/modussl_axtls.c b/extmod/modussl_axtls.c
@@ -86,6 +86,9 @@ STATIC mp_obj_t ussl_socket_new(mp_obj_t sock, struct ssl_args *args) {
         }
 
         data = (const byte*)mp_obj_str_get_data(args->cert.u_obj, &len);
+        if (data == NULL) {
+            return MP_OBJ_NULL;
+        }
         res = ssl_obj_memory_load(o->ssl_ctx, SSL_OBJ_X509_CERT, data, len, NULL);
         if (res != SSL_OK) {
             return mp_raise_ValueError_o("invalid cert");

Original file line number	Diff line number	Diff line change
`@@ -86,6 +86,9 @@ STATIC mp_obj_t ussl_socket_new(mp_obj_t sock, struct ssl_args *args) {`
`86`	`86`	`}`
`87`	`87`
`88`	`88`	`data = (const byte*)mp_obj_str_get_data(args->cert.u_obj, &len);`
	`89`	`+ if (data == NULL) {`
	`90`	`+ return MP_OBJ_NULL;`
	`91`	`+ }`
`89`	`92`	`res = ssl_obj_memory_load(o->ssl_ctx, SSL_OBJ_X509_CERT, data, len, NULL);`
`90`	`93`	`if (res != SSL_OK) {`
`91`	`94`	`return mp_raise_ValueError_o("invalid cert");`