commit 409730e2d6eb8090561ae6f295d361b58e7c756f
author Kazuho Oku <kazuhooku@gmail.com> Mon Jun 15 10:55:57 2020 +0900
committer Kazuho Oku <kazuhooku@gmail.com> Mon Jun 15 10:55:57 2020 +0900
tree 20d9e4443d69f44cb5f2d8e693773072c5cd4237
parent 7a38b09537f33248718f12322042b532bac06785

handle malloc failure

lib/picotls.c

diff --git a/lib/picotls.c b/lib/picotls.c
index 764befa..f74fd28 100644
--- a/lib/picotls.c
+++ b/lib/picotls.c
@@ -2481,7 +2481,10 @@
         default:
             if (should_collect_unknown_extension(tls, properties, type)) {
                 if (unknown_extensions == &no_unknown_extensions) {
-                    unknown_extensions = malloc(sizeof(*unknown_extensions) * (MAX_UNKNOWN_EXTENSIONS + 1));
+                    if ((unknown_extensions = malloc(sizeof(*unknown_extensions) * (MAX_UNKNOWN_EXTENSIONS + 1))) == NULL) {
+                        ret = PTLS_ERROR_NO_MEMORY;
+                        goto Exit;
+                    }
                     unknown_extensions[0].type = UINT16_MAX;
                 }
                 if ((ret = collect_unknown_extension(tls, type, src, end, unknown_extensions)) != 0)