Dave Rodgman | 4413b66 | 2022-11-22 17:31:50 +0000 | [diff] [blame^] | 1 | /* BEGIN_HEADER */ |
| 2 | #include "../library/common.h" |
| 3 | |
| 4 | void fill_arrays( unsigned char *a, unsigned char *b, unsigned char *r1, unsigned char *r2, size_t n ) |
| 5 | { |
| 6 | for ( size_t i = 0; i < n; i++ ) |
| 7 | { |
| 8 | a[i] = (unsigned char) i * 3; |
| 9 | b[i] = (unsigned char) i * 3 + 1; |
| 10 | r1[i] = (unsigned char) i * 3 + 2; |
| 11 | r2[i] = r1[i]; |
| 12 | } |
| 13 | } |
| 14 | /* END_HEADER */ |
| 15 | |
| 16 | /* BEGIN_CASE */ |
| 17 | void mbedtls_xor( int len ) |
| 18 | { |
| 19 | size_t n = (size_t) len; |
| 20 | unsigned char *a = NULL, *b = NULL, *r1 = NULL, *r2 = NULL; |
| 21 | ASSERT_ALLOC( a, n ); |
| 22 | ASSERT_ALLOC( b, n ); |
| 23 | ASSERT_ALLOC( r1, n ); |
| 24 | ASSERT_ALLOC( r2, n ); |
| 25 | |
| 26 | /* Test non-overlapping */ |
| 27 | fill_arrays( a, b, r1, r2, n ); |
| 28 | for ( size_t i = 0; i < n; i++ ) |
| 29 | { |
| 30 | r1[i] = a[i] ^ b[i]; |
| 31 | } |
| 32 | mbedtls_xor( r2, a, b, n ); |
| 33 | ASSERT_COMPARE( r1, n, r2, n ); |
| 34 | |
| 35 | /* Test r == a */ |
| 36 | fill_arrays( a, b, r1, r2, n ); |
| 37 | for ( size_t i = 0; i < n; i++ ) |
| 38 | { |
| 39 | r1[i] = r1[i] ^ b[i]; |
| 40 | } |
| 41 | mbedtls_xor( r2, r2, b, n ); |
| 42 | ASSERT_COMPARE( r1, n, r2, n ); |
| 43 | |
| 44 | /* Test r == b */ |
| 45 | fill_arrays( a, b, r1, r2, n ); |
| 46 | for ( size_t i = 0; i < n; i++ ) |
| 47 | { |
| 48 | r1[i] = a[i] ^ r1[i]; |
| 49 | } |
| 50 | mbedtls_xor( r2, a, r2, n ); |
| 51 | ASSERT_COMPARE( r1, n, r2, n ); |
| 52 | |
| 53 | /* Test a == b */ |
| 54 | fill_arrays( a, b, r1, r2, n ); |
| 55 | for ( size_t i = 0; i < n; i++ ) |
| 56 | { |
| 57 | r1[i] = a[i] ^ a[i]; |
| 58 | } |
| 59 | mbedtls_xor( r2, a, a, n ); |
| 60 | ASSERT_COMPARE( r1, n, r2, n ); |
| 61 | |
| 62 | /* Test a == b == r */ |
| 63 | fill_arrays( a, b, r1, r2, n ); |
| 64 | for ( size_t i = 0; i < n; i++ ) |
| 65 | { |
| 66 | r1[i] = r1[i] ^ r1[i]; |
| 67 | } |
| 68 | mbedtls_xor( r2, r2, r2, n ); |
| 69 | ASSERT_COMPARE( r1, n, r2, n ); |
| 70 | exit: |
| 71 | if ( a != NULL ) mbedtls_free( a ); |
| 72 | if ( a != NULL ) mbedtls_free( b ); |
| 73 | if ( r1 != NULL ) mbedtls_free( r1 ); |
| 74 | if ( r2 != NULL ) mbedtls_free( r2 ); |
| 75 | } |
| 76 | /* END_CASE */ |