Handle malloc returning null

Locks are required for multi-threading. If memory can't be allocated, exit the program with memory error. If we let the program continue, it will deadlock in the next part of the code anyway so better end it before.

1 files changed, 4 insertions, 2 deletions

diff --git a/crypto/compat/crypto_lock_win.c b/crypto/compat/crypto_lock_win.c
index e995726..b3b1858 100644
--- a/crypto/compat/crypto_lock_win.c
+++ b/crypto/compat/crypto_lock_win.c
@@ -1,6 +1,7 @@
 /* $OpenBSD: crypto_lock.c,v 1.1 2018/11/11 06:41:28 bcook Exp $ */
 /*
- * Copyright (c) 2018 Brent Cook <bcook@openbsd.org>
+ * Copyright (c) 2019 Brent Cook <bcook@openbsd.org>
+ * Copyright (c) 2019 John Norrbin <jlnorrbin@johnex.se>
  *
  * Permission to use, copy, modify, and distribute this software for any
  * purpose with or without fee is hereby granted, provided that the above
@@ -29,8 +30,9 @@ CRYPTO_lock(int mode, int type, const char *file, int line)
 
         if (locks[type] == NULL) {
                 LPCRITICAL_SECTION lcs = malloc(sizeof(CRITICAL_SECTION));
+                if (lcs == NULL) exit(ENOMEM);
                 InitializeCriticalSection(lcs);
-                if (InterlockedCompareExchangePointer((void **)&locks[type], (void *)lcs, NULL) != NULL) {
+                if (InterlockedCompareExchangePointer((PVOID*)&locks[type], (PVOID)lcs, NULL) != NULL) {
                         DeleteCriticalSection(lcs);
                         free(lcs);
                 }