1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
/*
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
#pragma once
#define s2n_likely(x) __builtin_expect(!!(x), 1)
#define s2n_unlikely(x) __builtin_expect(!!(x), 0)
/**
* s2n_ensure provides low-level safety check functionality
*
* This should only consumed directly by s2n_safety.
*
* Note: This module can be replaced by static analyzer implementation
* to insert additional safety checks.
*/
/**
* Ensures `cond` is true, otherwise `action` will be performed
*/
#define __S2N_ENSURE( cond, action ) do {if ( !(cond) ) { action; }} while (0)
#define __S2N_ENSURE_LIKELY( cond, action ) do {if ( s2n_unlikely( !(cond) ) ) { action; }} while (0)
#ifdef NDEBUG
#define __S2N_ENSURE_DEBUG( cond, action ) do {} while (0)
#else
#define __S2N_ENSURE_DEBUG( cond, action ) __S2N_ENSURE_LIKELY((cond), action)
#endif
#define __S2N_ENSURE_PRECONDITION( result ) (s2n_likely(s2n_result_is_ok(result)) ? S2N_RESULT_OK : S2N_RESULT_ERROR)
#ifdef NDEBUG
#define __S2N_ENSURE_POSTCONDITION( result ) (S2N_RESULT_OK)
#else
#define __S2N_ENSURE_POSTCONDITION( result ) (s2n_likely(s2n_result_is_ok(result)) ? S2N_RESULT_OK : S2N_RESULT_ERROR)
#endif
#define __S2N_ENSURE_SAFE_MEMCPY( d , s , n , guard ) \
do { \
__typeof( n ) __tmp_n = ( n ); \
if ( s2n_likely( __tmp_n ) ) { \
void *r = s2n_ensure_memcpy_trace( (d), (s) , (__tmp_n), _S2N_DEBUG_LINE); \
guard(r); \
} \
} while(0)
#define __S2N_ENSURE_SAFE_MEMSET( d , c , n , guard ) \
do { \
__typeof( n ) __tmp_n = ( n ); \
if ( s2n_likely( __tmp_n ) ) { \
__typeof( d ) __tmp_d = ( d ); \
guard( __tmp_d ); \
memset( __tmp_d, (c), __tmp_n); \
} \
} while(0)
extern void* s2n_ensure_memcpy_trace(void *restrict to, const void *restrict from, size_t size, const char *debug_str);
|
