From 8e3bebc736c224db4d7fc81ab9fa841c7015318a Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Mon, 30 Dec 2019 14:18:35 -0800 Subject: fixes #1075 WebSocket heap use after free This also introduces a new atomic boolean type, so we can use that to trigger whether we've added the HTTP handler or not. --- src/core/platform.h | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/core/platform.h') diff --git a/src/core/platform.h b/src/core/platform.h index b1d76899..b56f9a68 100644 --- a/src/core/platform.h +++ b/src/core/platform.h @@ -169,6 +169,15 @@ typedef struct nni_atomic_flag nni_atomic_flag; extern bool nni_atomic_flag_test_and_set(nni_atomic_flag *); extern void nni_atomic_flag_reset(nni_atomic_flag *); +// nni_atomic_bool is for boolean flags that need to be checked without +// changing their value. This might require a lock on some systems. +typedef struct nni_atomic_bool nni_atomic_bool; + +extern void nni_atomic_init_bool(nni_atomic_bool *); +extern void nni_atomic_set_bool(nni_atomic_bool *, bool); +extern bool nni_atomic_get_bool(nni_atomic_bool *); +extern bool nni_atomic_swap_bool(nni_atomic_bool *, bool); + typedef struct nni_atomic_u64 nni_atomic_u64; extern void nni_atomic_init64(nni_atomic_u64 *); -- cgit v1.2.3-70-g09d2