Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 2 | #include "mbedtls/bignum.h" |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 3 | #include "mbedtls/entropy.h" |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 4 | |
Chris Jones | e64a46f | 2020-12-03 17:44:03 +0000 | [diff] [blame] | 5 | #if MBEDTLS_MPI_MAX_BITS > 792 |
| 6 | #define MPI_MAX_BITS_LARGER_THAN_792 |
Chris Jones | 4592bd8 | 2020-12-03 14:24:33 +0000 | [diff] [blame] | 7 | #endif |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 8 | |
Gilles Peskine | b53b218 | 2021-06-10 15:34:15 +0200 | [diff] [blame] | 9 | /* Check the validity of the sign bit in an MPI object. Reject representations |
| 10 | * that are not supported by the rest of the library and indicate a bug when |
| 11 | * constructing the value. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 12 | static int sign_is_valid(const mbedtls_mpi *X) |
Gilles Peskine | b53b218 | 2021-06-10 15:34:15 +0200 | [diff] [blame] | 13 | { |
Gilles Peskine | 53a7206 | 2022-11-09 21:08:44 +0100 | [diff] [blame] | 14 | /* Only +1 and -1 are valid sign bits, not e.g. 0 */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 15 | if (X->s != 1 && X->s != -1) { |
| 16 | return 0; |
| 17 | } |
Gilles Peskine | 53a7206 | 2022-11-09 21:08:44 +0100 | [diff] [blame] | 18 | |
| 19 | /* The value 0 must be represented with the sign +1. A "negative zero" |
| 20 | * with s=-1 is an invalid representation. Forbid that. As an exception, |
| 21 | * we sometimes test the robustness of library functions when given |
| 22 | * a negative zero input. If a test case has a negative zero as input, |
| 23 | * we don't mind if the function has a negative zero output. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 24 | if (!mbedtls_test_case_uses_negative_0 && |
| 25 | mbedtls_mpi_bitlen(X) == 0 && X->s != 1) { |
| 26 | return 0; |
Gilles Peskine | 53a7206 | 2022-11-09 21:08:44 +0100 | [diff] [blame] | 27 | } |
| 28 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 29 | return 1; |
Gilles Peskine | b53b218 | 2021-06-10 15:34:15 +0200 | [diff] [blame] | 30 | } |
| 31 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 32 | typedef struct mbedtls_test_mpi_random { |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 33 | data_t *data; |
| 34 | size_t pos; |
| 35 | size_t chunk_len; |
| 36 | } mbedtls_test_mpi_random; |
| 37 | |
| 38 | /* |
| 39 | * This function is called by the Miller-Rabin primality test each time it |
| 40 | * chooses a random witness. The witnesses (or non-witnesses as provided by the |
| 41 | * test) are stored in the data member of the state structure. Each number is in |
| 42 | * the format that mbedtls_mpi_read_string understands and is chunk_len long. |
| 43 | */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 44 | int mbedtls_test_mpi_miller_rabin_determinizer(void *state, |
| 45 | unsigned char *buf, |
| 46 | size_t len) |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 47 | { |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 48 | mbedtls_test_mpi_random *random = (mbedtls_test_mpi_random *) state; |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 49 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 50 | if (random == NULL || random->data->x == NULL || buf == NULL) { |
| 51 | return -1; |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 52 | } |
| 53 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 54 | if (random->pos + random->chunk_len > random->data->len |
| 55 | || random->chunk_len > len) { |
| 56 | return -1; |
| 57 | } |
| 58 | |
| 59 | memset(buf, 0, len); |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 60 | |
| 61 | /* The witness is written to the end of the buffer, since the buffer is |
| 62 | * used as big endian, unsigned binary data in mbedtls_mpi_read_binary. |
| 63 | * Writing the witness to the start of the buffer would result in the |
| 64 | * buffer being 'witness 000...000', which would be treated as |
| 65 | * witness * 2^n for some n. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 66 | memcpy(buf + len - random->chunk_len, &random->data->x[random->pos], |
| 67 | random->chunk_len); |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 68 | |
| 69 | random->pos += random->chunk_len; |
| 70 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 71 | return 0; |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 72 | } |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 73 | |
| 74 | /* Random generator that is told how many bytes to return. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 75 | static int f_rng_bytes_left(void *state, unsigned char *buf, size_t len) |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 76 | { |
| 77 | size_t *bytes_left = state; |
| 78 | size_t i; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 79 | for (i = 0; i < len; i++) { |
| 80 | if (*bytes_left == 0) { |
| 81 | return MBEDTLS_ERR_ENTROPY_SOURCE_FAILED; |
| 82 | } |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 83 | buf[i] = *bytes_left & 0xff; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 84 | --(*bytes_left); |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 85 | } |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 86 | return 0; |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 87 | } |
| 88 | |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 89 | /* Test whether bytes represents (in big-endian base 256) a number b that |
| 90 | * is significantly above a power of 2. That is, b must not have a long run |
| 91 | * of unset bits after the most significant bit. |
| 92 | * |
| 93 | * Let n be the bit-size of b, i.e. the integer such that 2^n <= b < 2^{n+1}. |
| 94 | * This function returns 1 if, when drawing a number between 0 and b, |
| 95 | * the probability that this number is at least 2^n is not negligible. |
| 96 | * This probability is (b - 2^n) / b and this function checks that this |
| 97 | * number is above some threshold A. The threshold value is heuristic and |
| 98 | * based on the needs of mpi_random_many(). |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 99 | */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 100 | static int is_significantly_above_a_power_of_2(data_t *bytes) |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 101 | { |
| 102 | const uint8_t *p = bytes->x; |
| 103 | size_t len = bytes->len; |
| 104 | unsigned x; |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 105 | |
| 106 | /* Skip leading null bytes */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 107 | while (len > 0 && p[0] == 0) { |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 108 | ++p; |
| 109 | --len; |
| 110 | } |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 111 | /* 0 is not significantly above a power of 2 */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 112 | if (len == 0) { |
| 113 | return 0; |
| 114 | } |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 115 | /* Extract the (up to) 2 most significant bytes */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 116 | if (len == 1) { |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 117 | x = p[0]; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 118 | } else { |
| 119 | x = (p[0] << 8) | p[1]; |
| 120 | } |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 121 | |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 122 | /* Shift the most significant bit of x to position 8 and mask it out */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 123 | while ((x & 0xfe00) != 0) { |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 124 | x >>= 1; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 125 | } |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 126 | x &= 0x00ff; |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 127 | |
Gilles Peskine | 3b05615 | 2021-04-13 19:50:04 +0200 | [diff] [blame] | 128 | /* At this point, x = floor((b - 2^n) / 2^(n-8)). b is significantly above |
| 129 | * a power of 2 iff x is significantly above 0 compared to 2^8. |
| 130 | * Testing x >= 2^4 amounts to picking A = 1/16 in the function |
| 131 | * description above. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 132 | return x >= 0x10; |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 133 | } |
| 134 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 135 | /* END_HEADER */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 136 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 137 | /* BEGIN_DEPENDENCIES |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 138 | * depends_on:MBEDTLS_BIGNUM_C |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 139 | * END_DEPENDENCIES |
| 140 | */ |
Paul Bakker | 5690efc | 2011-05-26 13:16:06 +0000 | [diff] [blame] | 141 | |
Hanno Becker | b48e1aa | 2018-12-18 23:25:01 +0000 | [diff] [blame] | 142 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 143 | void mpi_valid_param() |
Hanno Becker | b48e1aa | 2018-12-18 23:25:01 +0000 | [diff] [blame] | 144 | { |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 145 | TEST_VALID_PARAM(mbedtls_mpi_free(NULL)); |
Hanno Becker | b48e1aa | 2018-12-18 23:25:01 +0000 | [diff] [blame] | 146 | } |
| 147 | /* END_CASE */ |
| 148 | |
Hanno Becker | afb607b | 2018-12-11 14:27:08 +0000 | [diff] [blame] | 149 | /* BEGIN_CASE depends_on:MBEDTLS_CHECK_PARAMS:!MBEDTLS_PARAM_FAILED_ALT */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 150 | void mpi_invalid_param() |
Hanno Becker | afb607b | 2018-12-11 14:27:08 +0000 | [diff] [blame] | 151 | { |
| 152 | mbedtls_mpi X; |
| 153 | const char *s_in = "00101000101010"; |
| 154 | char s_out[16] = { 0 }; |
| 155 | unsigned char u_out[16] = { 0 }; |
| 156 | unsigned char u_in[16] = { 0 }; |
| 157 | size_t olen; |
| 158 | mbedtls_mpi_uint mpi_uint; |
| 159 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 160 | TEST_INVALID_PARAM(mbedtls_mpi_init(NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 161 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 162 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 163 | mbedtls_mpi_grow(NULL, 42)); |
| 164 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 165 | mbedtls_mpi_copy(NULL, &X)); |
| 166 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 167 | mbedtls_mpi_copy(&X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 168 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 169 | TEST_INVALID_PARAM(mbedtls_mpi_swap(NULL, &X)); |
| 170 | TEST_INVALID_PARAM(mbedtls_mpi_swap(&X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 171 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 172 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 173 | mbedtls_mpi_safe_cond_assign(NULL, &X, 0)); |
| 174 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 175 | mbedtls_mpi_safe_cond_assign(&X, NULL, 0)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 176 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 177 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 178 | mbedtls_mpi_safe_cond_swap(NULL, &X, 0)); |
| 179 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 180 | mbedtls_mpi_safe_cond_swap(&X, NULL, 0)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 181 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 182 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 183 | mbedtls_mpi_lset(NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 184 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 185 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 186 | mbedtls_mpi_get_bit(NULL, 42)); |
| 187 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 188 | mbedtls_mpi_set_bit(NULL, 42, 0)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 189 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 190 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 191 | mbedtls_mpi_read_string(NULL, 2, s_in)); |
| 192 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 193 | mbedtls_mpi_read_string(&X, 2, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 194 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 195 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 196 | mbedtls_mpi_write_string(NULL, 2, |
| 197 | s_out, sizeof(s_out), |
| 198 | &olen)); |
| 199 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 200 | mbedtls_mpi_write_string(&X, 2, |
| 201 | NULL, sizeof(s_out), |
| 202 | &olen)); |
| 203 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 204 | mbedtls_mpi_write_string(&X, 2, |
| 205 | s_out, sizeof(s_out), |
| 206 | NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 207 | |
Hanno Becker | afb607b | 2018-12-11 14:27:08 +0000 | [diff] [blame] | 208 | #if defined(MBEDTLS_FS_IO) |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 209 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 210 | mbedtls_mpi_read_file(NULL, 2, stdin)); |
| 211 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 212 | mbedtls_mpi_read_file(&X, 2, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 213 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 214 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 215 | mbedtls_mpi_write_file("", NULL, 2, NULL)); |
Hanno Becker | afb607b | 2018-12-11 14:27:08 +0000 | [diff] [blame] | 216 | #endif /* MBEDTLS_FS_IO */ |
| 217 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 218 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 219 | mbedtls_mpi_read_binary(NULL, u_in, |
| 220 | sizeof(u_in))); |
| 221 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 222 | mbedtls_mpi_read_binary(&X, NULL, |
| 223 | sizeof(u_in))); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 224 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 225 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 226 | mbedtls_mpi_write_binary(NULL, u_out, |
| 227 | sizeof(u_out))); |
| 228 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 229 | mbedtls_mpi_write_binary(&X, NULL, |
| 230 | sizeof(u_out))); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 231 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 232 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 233 | mbedtls_mpi_shift_l(NULL, 42)); |
| 234 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 235 | mbedtls_mpi_shift_r(NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 236 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 237 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 238 | mbedtls_mpi_cmp_abs(NULL, &X)); |
| 239 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 240 | mbedtls_mpi_cmp_abs(&X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 241 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 242 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 243 | mbedtls_mpi_cmp_mpi(NULL, &X)); |
| 244 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 245 | mbedtls_mpi_cmp_mpi(&X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 246 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 247 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 248 | mbedtls_mpi_cmp_int(NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 249 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 250 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 251 | mbedtls_mpi_add_abs(NULL, &X, &X)); |
| 252 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 253 | mbedtls_mpi_add_abs(&X, NULL, &X)); |
| 254 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 255 | mbedtls_mpi_add_abs(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 256 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 257 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 258 | mbedtls_mpi_sub_abs(NULL, &X, &X)); |
| 259 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 260 | mbedtls_mpi_sub_abs(&X, NULL, &X)); |
| 261 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 262 | mbedtls_mpi_sub_abs(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 263 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 264 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 265 | mbedtls_mpi_add_mpi(NULL, &X, &X)); |
| 266 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 267 | mbedtls_mpi_add_mpi(&X, NULL, &X)); |
| 268 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 269 | mbedtls_mpi_add_mpi(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 270 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 271 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 272 | mbedtls_mpi_sub_mpi(NULL, &X, &X)); |
| 273 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 274 | mbedtls_mpi_sub_mpi(&X, NULL, &X)); |
| 275 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 276 | mbedtls_mpi_sub_mpi(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 277 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 278 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 279 | mbedtls_mpi_add_int(NULL, &X, 42)); |
| 280 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 281 | mbedtls_mpi_add_int(&X, NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 282 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 283 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 284 | mbedtls_mpi_sub_int(NULL, &X, 42)); |
| 285 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 286 | mbedtls_mpi_sub_int(&X, NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 287 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 288 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 289 | mbedtls_mpi_mul_mpi(NULL, &X, &X)); |
| 290 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 291 | mbedtls_mpi_mul_mpi(&X, NULL, &X)); |
| 292 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 293 | mbedtls_mpi_mul_mpi(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 294 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 295 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 296 | mbedtls_mpi_mul_int(NULL, &X, 42)); |
| 297 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 298 | mbedtls_mpi_mul_int(&X, NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 299 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 300 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 301 | mbedtls_mpi_div_mpi(&X, &X, NULL, &X)); |
| 302 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 303 | mbedtls_mpi_div_mpi(&X, &X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 304 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 305 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 306 | mbedtls_mpi_div_int(&X, &X, NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 307 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 308 | TEST_INVALID_PARAM_RET(0, mbedtls_mpi_lsb(NULL)); |
Hanno Becker | f25ee7f | 2018-12-19 16:51:02 +0000 | [diff] [blame] | 309 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 310 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 311 | mbedtls_mpi_mod_mpi(NULL, &X, &X)); |
| 312 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 313 | mbedtls_mpi_mod_mpi(&X, NULL, &X)); |
| 314 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 315 | mbedtls_mpi_mod_mpi(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 316 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 317 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 318 | mbedtls_mpi_mod_int(NULL, &X, 42)); |
| 319 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 320 | mbedtls_mpi_mod_int(&mpi_uint, NULL, 42)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 321 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 322 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 323 | mbedtls_mpi_exp_mod(NULL, &X, &X, &X, NULL)); |
| 324 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 325 | mbedtls_mpi_exp_mod(&X, NULL, &X, &X, NULL)); |
| 326 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 327 | mbedtls_mpi_exp_mod(&X, &X, NULL, &X, NULL)); |
| 328 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 329 | mbedtls_mpi_exp_mod(&X, &X, &X, NULL, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 330 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 331 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 332 | mbedtls_mpi_fill_random(NULL, 42, |
| 333 | mbedtls_test_rnd_std_rand, |
| 334 | NULL)); |
| 335 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 336 | mbedtls_mpi_fill_random(&X, 42, NULL, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 337 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 338 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 339 | mbedtls_mpi_gcd(NULL, &X, &X)); |
| 340 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 341 | mbedtls_mpi_gcd(&X, NULL, &X)); |
| 342 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 343 | mbedtls_mpi_gcd(&X, &X, NULL)); |
Hanno Becker | e118504 | 2018-12-13 14:31:46 +0000 | [diff] [blame] | 344 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 345 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 346 | mbedtls_mpi_inv_mod(NULL, &X, &X)); |
| 347 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 348 | mbedtls_mpi_inv_mod(&X, NULL, &X)); |
| 349 | TEST_INVALID_PARAM_RET(MBEDTLS_ERR_MPI_BAD_INPUT_DATA, |
| 350 | mbedtls_mpi_inv_mod(&X, &X, NULL)); |
Hanno Becker | afb607b | 2018-12-11 14:27:08 +0000 | [diff] [blame] | 351 | |
| 352 | exit: |
| 353 | return; |
Hanno Becker | afb607b | 2018-12-11 14:27:08 +0000 | [diff] [blame] | 354 | } |
| 355 | /* END_CASE */ |
| 356 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 357 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 358 | void mpi_null() |
Manuel Pégourié-Gonnard | 770b5e1 | 2015-04-29 17:02:01 +0200 | [diff] [blame] | 359 | { |
Manuel Pégourié-Gonnard | da61ed3 | 2015-04-30 10:28:51 +0200 | [diff] [blame] | 360 | mbedtls_mpi X, Y, Z; |
Manuel Pégourié-Gonnard | 770b5e1 | 2015-04-29 17:02:01 +0200 | [diff] [blame] | 361 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 362 | mbedtls_mpi_init(&X); |
| 363 | mbedtls_mpi_init(&Y); |
| 364 | mbedtls_mpi_init(&Z); |
Manuel Pégourié-Gonnard | 770b5e1 | 2015-04-29 17:02:01 +0200 | [diff] [blame] | 365 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 366 | TEST_ASSERT(mbedtls_mpi_get_bit(&X, 42) == 0); |
| 367 | TEST_ASSERT(mbedtls_mpi_lsb(&X) == 0); |
| 368 | TEST_ASSERT(mbedtls_mpi_bitlen(&X) == 0); |
| 369 | TEST_ASSERT(mbedtls_mpi_size(&X) == 0); |
Manuel Pégourié-Gonnard | 770b5e1 | 2015-04-29 17:02:01 +0200 | [diff] [blame] | 370 | |
| 371 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 372 | mbedtls_mpi_free(&X); |
Manuel Pégourié-Gonnard | 770b5e1 | 2015-04-29 17:02:01 +0200 | [diff] [blame] | 373 | } |
| 374 | /* END_CASE */ |
| 375 | |
| 376 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 377 | void mpi_read_write_string(int radix_X, char *input_X, int radix_A, |
| 378 | char *input_A, int output_size, int result_read, |
| 379 | int result_write) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 380 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 381 | mbedtls_mpi X; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 382 | char str[1000]; |
Manuel Pégourié-Gonnard | f79b425 | 2015-06-02 15:41:48 +0100 | [diff] [blame] | 383 | size_t len; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 384 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 385 | mbedtls_mpi_init(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 386 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 387 | memset(str, '!', sizeof(str)); |
Janos Follath | 04dadb7 | 2019-03-06 12:29:37 +0000 | [diff] [blame] | 388 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 389 | TEST_ASSERT(mbedtls_mpi_read_string(&X, radix_X, input_X) == result_read); |
| 390 | if (result_read == 0) { |
| 391 | TEST_ASSERT(sign_is_valid(&X)); |
| 392 | TEST_ASSERT(mbedtls_mpi_write_string(&X, radix_A, str, output_size, &len) == result_write); |
| 393 | if (result_write == 0) { |
| 394 | TEST_ASSERT(strcasecmp(str, input_A) == 0); |
| 395 | TEST_ASSERT(str[len] == '!'); |
Paul Bakker | ba48cb2 | 2009-07-12 11:01:32 +0000 | [diff] [blame] | 396 | } |
| 397 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 398 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 399 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 400 | mbedtls_mpi_free(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 401 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 402 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 403 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 404 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 405 | void mpi_read_binary(data_t *buf, char *input_A) |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 406 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 407 | mbedtls_mpi X; |
Janos Follath | e5670f2 | 2019-02-25 16:11:58 +0000 | [diff] [blame] | 408 | char str[1000]; |
Manuel Pégourié-Gonnard | f79b425 | 2015-06-02 15:41:48 +0100 | [diff] [blame] | 409 | size_t len; |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 410 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 411 | mbedtls_mpi_init(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 412 | |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 413 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 414 | TEST_ASSERT(mbedtls_mpi_read_binary(&X, buf->x, buf->len) == 0); |
| 415 | TEST_ASSERT(sign_is_valid(&X)); |
| 416 | TEST_ASSERT(mbedtls_mpi_write_string(&X, 16, str, sizeof(str), &len) == 0); |
| 417 | TEST_ASSERT(strcmp((char *) str, input_A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 418 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 419 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 420 | mbedtls_mpi_free(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 421 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 422 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 423 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 424 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 425 | void mpi_read_binary_le(data_t *buf, char *input_A) |
Janos Follath | a778a94 | 2019-02-13 10:28:28 +0000 | [diff] [blame] | 426 | { |
| 427 | mbedtls_mpi X; |
Janos Follath | e5670f2 | 2019-02-25 16:11:58 +0000 | [diff] [blame] | 428 | char str[1000]; |
Janos Follath | a778a94 | 2019-02-13 10:28:28 +0000 | [diff] [blame] | 429 | size_t len; |
| 430 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 431 | mbedtls_mpi_init(&X); |
Janos Follath | a778a94 | 2019-02-13 10:28:28 +0000 | [diff] [blame] | 432 | |
| 433 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 434 | TEST_ASSERT(mbedtls_mpi_read_binary_le(&X, buf->x, buf->len) == 0); |
| 435 | TEST_ASSERT(sign_is_valid(&X)); |
| 436 | TEST_ASSERT(mbedtls_mpi_write_string(&X, 16, str, sizeof(str), &len) == 0); |
| 437 | TEST_ASSERT(strcmp((char *) str, input_A) == 0); |
Janos Follath | a778a94 | 2019-02-13 10:28:28 +0000 | [diff] [blame] | 438 | |
| 439 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 440 | mbedtls_mpi_free(&X); |
Janos Follath | a778a94 | 2019-02-13 10:28:28 +0000 | [diff] [blame] | 441 | } |
| 442 | /* END_CASE */ |
| 443 | |
| 444 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 445 | void mpi_write_binary(char *input_X, data_t *input_A, |
| 446 | int output_size, int result) |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 447 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 448 | mbedtls_mpi X; |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 449 | unsigned char buf[1000]; |
Paul Bakker | f4a3f30 | 2011-04-24 15:53:29 +0000 | [diff] [blame] | 450 | size_t buflen; |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 451 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 452 | memset(buf, 0x00, 1000); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 453 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 454 | mbedtls_mpi_init(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 455 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 456 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Manuel Pégourié-Gonnard | e670f90 | 2015-10-30 09:23:19 +0100 | [diff] [blame] | 457 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 458 | buflen = mbedtls_mpi_size(&X); |
| 459 | if (buflen > (size_t) output_size) { |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 460 | buflen = (size_t) output_size; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 461 | } |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 462 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 463 | TEST_ASSERT(mbedtls_mpi_write_binary(&X, buf, buflen) == result); |
| 464 | if (result == 0) { |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 465 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 466 | TEST_ASSERT(mbedtls_test_hexcmp(buf, input_A->x, |
| 467 | buflen, input_A->len) == 0); |
Paul Bakker | ba48cb2 | 2009-07-12 11:01:32 +0000 | [diff] [blame] | 468 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 469 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 470 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 471 | mbedtls_mpi_free(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 472 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 473 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 474 | |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 475 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 476 | void mpi_write_binary_le(char *input_X, data_t *input_A, |
| 477 | int output_size, int result) |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 478 | { |
| 479 | mbedtls_mpi X; |
| 480 | unsigned char buf[1000]; |
| 481 | size_t buflen; |
| 482 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 483 | memset(buf, 0x00, 1000); |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 484 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 485 | mbedtls_mpi_init(&X); |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 486 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 487 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 488 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 489 | buflen = mbedtls_mpi_size(&X); |
| 490 | if (buflen > (size_t) output_size) { |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 491 | buflen = (size_t) output_size; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 492 | } |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 493 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 494 | TEST_ASSERT(mbedtls_mpi_write_binary_le(&X, buf, buflen) == result); |
| 495 | if (result == 0) { |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 496 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 497 | TEST_ASSERT(mbedtls_test_hexcmp(buf, input_A->x, |
| 498 | buflen, input_A->len) == 0); |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 499 | } |
| 500 | |
| 501 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 502 | mbedtls_mpi_free(&X); |
Janos Follath | e344d0f | 2019-02-19 16:17:40 +0000 | [diff] [blame] | 503 | } |
| 504 | /* END_CASE */ |
| 505 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 506 | /* BEGIN_CASE depends_on:MBEDTLS_FS_IO */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 507 | void mpi_read_file(char *input_file, data_t *input_A, int result) |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 508 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 509 | mbedtls_mpi X; |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 510 | unsigned char buf[1000]; |
Paul Bakker | f4a3f30 | 2011-04-24 15:53:29 +0000 | [diff] [blame] | 511 | size_t buflen; |
Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 512 | FILE *file; |
Manuel Pégourié-Gonnard | e43187d | 2015-02-14 16:01:34 +0000 | [diff] [blame] | 513 | int ret; |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 514 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 515 | memset(buf, 0x00, 1000); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 516 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 517 | mbedtls_mpi_init(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 518 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 519 | file = fopen(input_file, "r"); |
| 520 | TEST_ASSERT(file != NULL); |
| 521 | ret = mbedtls_mpi_read_file(&X, 16, file); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 522 | fclose(file); |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 523 | TEST_ASSERT(ret == result); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 524 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 525 | if (result == 0) { |
| 526 | TEST_ASSERT(sign_is_valid(&X)); |
| 527 | buflen = mbedtls_mpi_size(&X); |
| 528 | TEST_ASSERT(mbedtls_mpi_write_binary(&X, buf, buflen) == 0); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 529 | |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 530 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 531 | TEST_ASSERT(mbedtls_test_hexcmp(buf, input_A->x, |
| 532 | buflen, input_A->len) == 0); |
Paul Bakker | ba48cb2 | 2009-07-12 11:01:32 +0000 | [diff] [blame] | 533 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 534 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 535 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 536 | mbedtls_mpi_free(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 537 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 538 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 539 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 540 | /* BEGIN_CASE depends_on:MBEDTLS_FS_IO */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 541 | void mpi_write_file(char *input_X, char *output_file) |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 542 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 543 | mbedtls_mpi X, Y; |
Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 544 | FILE *file_out, *file_in; |
Manuel Pégourié-Gonnard | ac5361f | 2015-06-24 01:08:09 +0200 | [diff] [blame] | 545 | int ret; |
Paul Bakker | 69998dd | 2009-07-11 19:15:20 +0000 | [diff] [blame] | 546 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 547 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 548 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 549 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 550 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 551 | file_out = fopen(output_file, "w"); |
| 552 | TEST_ASSERT(file_out != NULL); |
| 553 | ret = mbedtls_mpi_write_file(NULL, &X, 16, file_out); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 554 | fclose(file_out); |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 555 | TEST_ASSERT(ret == 0); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 556 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 557 | file_in = fopen(output_file, "r"); |
| 558 | TEST_ASSERT(file_in != NULL); |
| 559 | ret = mbedtls_mpi_read_file(&Y, 16, file_in); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 560 | fclose(file_in); |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 561 | TEST_ASSERT(ret == 0); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 562 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 563 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &Y) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 564 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 565 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 566 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 567 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 568 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 569 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 570 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 571 | void mpi_get_bit(char *input_X, int pos, int val) |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 572 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 573 | mbedtls_mpi X; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 574 | mbedtls_mpi_init(&X); |
| 575 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 576 | TEST_ASSERT(mbedtls_mpi_get_bit(&X, pos) == val); |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 577 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 578 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 579 | mbedtls_mpi_free(&X); |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 580 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 581 | /* END_CASE */ |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 582 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 583 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 584 | void mpi_set_bit(char *input_X, int pos, int val, |
| 585 | char *output_Y, int result) |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 586 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 587 | mbedtls_mpi X, Y; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 588 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 589 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 590 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 591 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, output_Y) == 0); |
| 592 | TEST_ASSERT(mbedtls_mpi_set_bit(&X, pos, val) == result); |
Paul Bakker | ec5ceb6 | 2016-07-14 12:47:07 +0100 | [diff] [blame] | 593 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 594 | if (result == 0) { |
| 595 | TEST_ASSERT(sign_is_valid(&X)); |
| 596 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &Y) == 0); |
Paul Bakker | ec5ceb6 | 2016-07-14 12:47:07 +0100 | [diff] [blame] | 597 | } |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 598 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 599 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 600 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 601 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 602 | /* END_CASE */ |
Paul Bakker | 2f5947e | 2011-05-18 15:47:11 +0000 | [diff] [blame] | 603 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 604 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 605 | void mpi_lsb(char *input_X, int nr_bits) |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 606 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 607 | mbedtls_mpi X; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 608 | mbedtls_mpi_init(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 609 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 610 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 611 | TEST_ASSERT(mbedtls_mpi_lsb(&X) == (size_t) nr_bits); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 612 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 613 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 614 | mbedtls_mpi_free(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 615 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 616 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 617 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 618 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 619 | void mpi_bitlen(char *input_X, int nr_bits) |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 620 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 621 | mbedtls_mpi X; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 622 | mbedtls_mpi_init(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 623 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 624 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 625 | TEST_ASSERT(mbedtls_mpi_bitlen(&X) == (size_t) nr_bits); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 626 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 627 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 628 | mbedtls_mpi_free(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 629 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 630 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 631 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 632 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 633 | void mpi_gcd(char *input_X, char *input_Y, |
| 634 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 635 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 636 | mbedtls_mpi A, X, Y, Z; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 637 | mbedtls_mpi_init(&A); mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 638 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 639 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 640 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 641 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 642 | TEST_ASSERT(mbedtls_mpi_gcd(&Z, &X, &Y) == 0); |
| 643 | TEST_ASSERT(sign_is_valid(&Z)); |
| 644 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 645 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 646 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 647 | mbedtls_mpi_free(&A); mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 648 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 649 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 650 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 651 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 652 | void mpi_cmp_int(int input_X, int input_A, int result_CMP) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 653 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 654 | mbedtls_mpi X; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 655 | mbedtls_mpi_init(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 656 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 657 | TEST_ASSERT(mbedtls_mpi_lset(&X, input_X) == 0); |
| 658 | TEST_ASSERT(mbedtls_mpi_cmp_int(&X, input_A) == result_CMP); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 659 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 660 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 661 | mbedtls_mpi_free(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 662 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 663 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 664 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 665 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 666 | void mpi_cmp_mpi(char *input_X, char *input_Y, |
| 667 | int input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 668 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 669 | mbedtls_mpi X, Y; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 670 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 671 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 672 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 673 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 674 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &Y) == input_A); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 675 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 676 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 677 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 678 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 679 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 680 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 681 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 682 | void mpi_lt_mpi_ct(int size_X, char *input_X, |
| 683 | int size_Y, char *input_Y, |
| 684 | int input_ret, int input_err) |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 685 | { |
Gilles Peskine | 0deccf1 | 2020-09-02 15:18:07 +0200 | [diff] [blame] | 686 | unsigned ret = -1; |
Janos Follath | 0e5532d | 2019-10-11 14:21:53 +0100 | [diff] [blame] | 687 | unsigned input_uret = input_ret; |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 688 | mbedtls_mpi X, Y; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 689 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 690 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 691 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 692 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 693 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 694 | TEST_ASSERT(mbedtls_mpi_grow(&X, size_X) == 0); |
| 695 | TEST_ASSERT(mbedtls_mpi_grow(&Y, size_Y) == 0); |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 696 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 697 | TEST_ASSERT(mbedtls_mpi_lt_mpi_ct(&X, &Y, &ret) == input_err); |
| 698 | if (input_err == 0) { |
| 699 | TEST_ASSERT(ret == input_uret); |
| 700 | } |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 701 | |
| 702 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 703 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); |
Janos Follath | 385d5b8 | 2019-09-11 16:07:14 +0100 | [diff] [blame] | 704 | } |
| 705 | /* END_CASE */ |
| 706 | |
| 707 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 708 | void mpi_cmp_abs(char *input_X, char *input_Y, |
| 709 | int input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 710 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 711 | mbedtls_mpi X, Y; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 712 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 713 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 714 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 715 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 716 | TEST_ASSERT(mbedtls_mpi_cmp_abs(&X, &Y) == input_A); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 717 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 718 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 719 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 720 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 721 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 722 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 723 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 724 | void mpi_copy(char *src_hex, char *dst_hex) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 725 | { |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 726 | mbedtls_mpi src, dst, ref; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 727 | mbedtls_mpi_init(&src); |
| 728 | mbedtls_mpi_init(&dst); |
| 729 | mbedtls_mpi_init(&ref); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 730 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 731 | TEST_ASSERT(mbedtls_test_read_mpi(&src, src_hex) == 0); |
| 732 | TEST_ASSERT(mbedtls_test_read_mpi(&ref, dst_hex) == 0); |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 733 | |
| 734 | /* mbedtls_mpi_copy() */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 735 | TEST_ASSERT(mbedtls_test_read_mpi(&dst, dst_hex) == 0); |
| 736 | TEST_ASSERT(mbedtls_mpi_copy(&dst, &src) == 0); |
| 737 | TEST_ASSERT(sign_is_valid(&dst)); |
| 738 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&dst, &src) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 739 | |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 740 | /* mbedtls_mpi_safe_cond_assign(), assignment done */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 741 | mbedtls_mpi_free(&dst); |
| 742 | TEST_ASSERT(mbedtls_test_read_mpi(&dst, dst_hex) == 0); |
| 743 | TEST_ASSERT(mbedtls_mpi_safe_cond_assign(&dst, &src, 1) == 0); |
| 744 | TEST_ASSERT(sign_is_valid(&dst)); |
| 745 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&dst, &src) == 0); |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 746 | |
| 747 | /* mbedtls_mpi_safe_cond_assign(), assignment not done */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 748 | mbedtls_mpi_free(&dst); |
| 749 | TEST_ASSERT(mbedtls_test_read_mpi(&dst, dst_hex) == 0); |
| 750 | TEST_ASSERT(mbedtls_mpi_safe_cond_assign(&dst, &src, 0) == 0); |
| 751 | TEST_ASSERT(sign_is_valid(&dst)); |
| 752 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&dst, &ref) == 0); |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 753 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 754 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 755 | mbedtls_mpi_free(&src); |
| 756 | mbedtls_mpi_free(&dst); |
| 757 | mbedtls_mpi_free(&ref); |
Gilles Peskine | 7428b45 | 2020-01-20 21:01:51 +0100 | [diff] [blame] | 758 | } |
| 759 | /* END_CASE */ |
| 760 | |
| 761 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 762 | void mpi_copy_self(char *input_X) |
Gilles Peskine | 7428b45 | 2020-01-20 21:01:51 +0100 | [diff] [blame] | 763 | { |
Gilles Peskine | 77f55c9 | 2021-06-10 15:17:30 +0200 | [diff] [blame] | 764 | mbedtls_mpi X, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 765 | mbedtls_mpi_init(&A); |
| 766 | mbedtls_mpi_init(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 767 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 768 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 769 | TEST_ASSERT(mbedtls_mpi_copy(&X, &X) == 0); |
Gilles Peskine | 77f55c9 | 2021-06-10 15:17:30 +0200 | [diff] [blame] | 770 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 771 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_X) == 0); |
| 772 | TEST_ASSERT(sign_is_valid(&X)); |
| 773 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 774 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 775 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 776 | mbedtls_mpi_free(&A); |
| 777 | mbedtls_mpi_free(&X); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 778 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 779 | /* END_CASE */ |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 780 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 781 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 782 | void mpi_swap(char *X_hex, char *Y_hex) |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 783 | { |
| 784 | mbedtls_mpi X, Y, X0, Y0; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 785 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); |
| 786 | mbedtls_mpi_init(&X0); mbedtls_mpi_init(&Y0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 787 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 788 | TEST_ASSERT(mbedtls_test_read_mpi(&X0, X_hex) == 0); |
| 789 | TEST_ASSERT(mbedtls_test_read_mpi(&Y0, Y_hex) == 0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 790 | |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 791 | /* mbedtls_mpi_swap() */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 792 | TEST_ASSERT(mbedtls_test_read_mpi(&X, X_hex) == 0); |
| 793 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, Y_hex) == 0); |
| 794 | mbedtls_mpi_swap(&X, &Y); |
| 795 | TEST_ASSERT(sign_is_valid(&X)); |
| 796 | TEST_ASSERT(sign_is_valid(&Y)); |
| 797 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &Y0) == 0); |
| 798 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &X0) == 0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 799 | |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 800 | /* mbedtls_mpi_safe_cond_swap(), swap done */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 801 | mbedtls_mpi_free(&X); |
| 802 | mbedtls_mpi_free(&Y); |
| 803 | TEST_ASSERT(mbedtls_test_read_mpi(&X, X_hex) == 0); |
| 804 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, Y_hex) == 0); |
| 805 | TEST_ASSERT(mbedtls_mpi_safe_cond_swap(&X, &Y, 1) == 0); |
| 806 | TEST_ASSERT(sign_is_valid(&X)); |
| 807 | TEST_ASSERT(sign_is_valid(&Y)); |
| 808 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &Y0) == 0); |
| 809 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &X0) == 0); |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 810 | |
| 811 | /* mbedtls_mpi_safe_cond_swap(), swap not done */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 812 | mbedtls_mpi_free(&X); |
| 813 | mbedtls_mpi_free(&Y); |
| 814 | TEST_ASSERT(mbedtls_test_read_mpi(&X, X_hex) == 0); |
| 815 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, Y_hex) == 0); |
| 816 | TEST_ASSERT(mbedtls_mpi_safe_cond_swap(&X, &Y, 0) == 0); |
| 817 | TEST_ASSERT(sign_is_valid(&X)); |
| 818 | TEST_ASSERT(sign_is_valid(&Y)); |
| 819 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &X0) == 0); |
| 820 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &Y0) == 0); |
Gilles Peskine | 5023167 | 2021-06-10 23:00:33 +0200 | [diff] [blame] | 821 | |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 822 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 823 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); |
| 824 | mbedtls_mpi_free(&X0); mbedtls_mpi_free(&Y0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 825 | } |
| 826 | /* END_CASE */ |
| 827 | |
| 828 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 829 | void mpi_swap_self(char *X_hex) |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 830 | { |
| 831 | mbedtls_mpi X, X0; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 832 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&X0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 833 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 834 | TEST_ASSERT(mbedtls_test_read_mpi(&X, X_hex) == 0); |
| 835 | TEST_ASSERT(mbedtls_test_read_mpi(&X0, X_hex) == 0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 836 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 837 | mbedtls_mpi_swap(&X, &X); |
| 838 | TEST_ASSERT(sign_is_valid(&X)); |
| 839 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &X0) == 0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 840 | |
| 841 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 842 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&X0); |
Gilles Peskine | d382c28 | 2021-06-10 22:29:57 +0200 | [diff] [blame] | 843 | } |
| 844 | /* END_CASE */ |
| 845 | |
| 846 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 847 | void mpi_shrink(int before, int used, int min, int after) |
Manuel Pégourié-Gonnard | 5868163 | 2013-11-21 10:39:37 +0100 | [diff] [blame] | 848 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 849 | mbedtls_mpi X; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 850 | mbedtls_mpi_init(&X); |
Manuel Pégourié-Gonnard | 5868163 | 2013-11-21 10:39:37 +0100 | [diff] [blame] | 851 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 852 | TEST_ASSERT(mbedtls_mpi_grow(&X, before) == 0); |
| 853 | if (used > 0) { |
| 854 | size_t used_bit_count = used * 8 * sizeof(mbedtls_mpi_uint); |
| 855 | TEST_ASSERT(mbedtls_mpi_set_bit(&X, used_bit_count - 1, 1) == 0); |
Gilles Peskine | 399c8fa | 2021-06-15 21:19:18 +0200 | [diff] [blame] | 856 | } |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 857 | TEST_EQUAL(X.n, (size_t) before); |
| 858 | TEST_ASSERT(mbedtls_mpi_shrink(&X, min) == 0); |
| 859 | TEST_EQUAL(X.n, (size_t) after); |
Manuel Pégourié-Gonnard | 5868163 | 2013-11-21 10:39:37 +0100 | [diff] [blame] | 860 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 861 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 862 | mbedtls_mpi_free(&X); |
Manuel Pégourié-Gonnard | 5868163 | 2013-11-21 10:39:37 +0100 | [diff] [blame] | 863 | } |
| 864 | /* END_CASE */ |
| 865 | |
| 866 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 867 | void mpi_add_mpi(char *input_X, char *input_Y, |
| 868 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 869 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 870 | mbedtls_mpi X, Y, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 871 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 872 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 873 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 874 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 875 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 876 | TEST_ASSERT(mbedtls_mpi_add_mpi(&Z, &X, &Y) == 0); |
| 877 | TEST_ASSERT(sign_is_valid(&Z)); |
| 878 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 879 | |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 880 | /* result == first operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 881 | TEST_ASSERT(mbedtls_mpi_add_mpi(&X, &X, &Y) == 0); |
| 882 | TEST_ASSERT(sign_is_valid(&X)); |
| 883 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
| 884 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 885 | |
| 886 | /* result == second operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 887 | TEST_ASSERT(mbedtls_mpi_add_mpi(&Y, &X, &Y) == 0); |
| 888 | TEST_ASSERT(sign_is_valid(&Y)); |
| 889 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &A) == 0); |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 890 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 891 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 892 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 893 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 894 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 895 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 896 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 897 | void mpi_add_mpi_inplace(char *input_X, char *input_A) |
Janos Follath | 044a86b | 2015-10-25 10:58:03 +0100 | [diff] [blame] | 898 | { |
| 899 | mbedtls_mpi X, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 900 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&A); |
Janos Follath | 044a86b | 2015-10-25 10:58:03 +0100 | [diff] [blame] | 901 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 902 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
Janos Follath | 6cbacec | 2015-10-25 12:29:13 +0100 | [diff] [blame] | 903 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 904 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 905 | TEST_ASSERT(mbedtls_mpi_sub_abs(&X, &X, &X) == 0); |
| 906 | TEST_ASSERT(mbedtls_mpi_cmp_int(&X, 0) == 0); |
| 907 | TEST_ASSERT(sign_is_valid(&X)); |
Janos Follath | 6cbacec | 2015-10-25 12:29:13 +0100 | [diff] [blame] | 908 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 909 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 910 | TEST_ASSERT(mbedtls_mpi_add_abs(&X, &X, &X) == 0); |
| 911 | TEST_ASSERT(sign_is_valid(&X)); |
| 912 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
Janos Follath | 6cbacec | 2015-10-25 12:29:13 +0100 | [diff] [blame] | 913 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 914 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 915 | TEST_ASSERT(mbedtls_mpi_add_mpi(&X, &X, &X) == 0); |
| 916 | TEST_ASSERT(sign_is_valid(&X)); |
| 917 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
Janos Follath | 044a86b | 2015-10-25 10:58:03 +0100 | [diff] [blame] | 918 | |
| 919 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 920 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&A); |
Janos Follath | 044a86b | 2015-10-25 10:58:03 +0100 | [diff] [blame] | 921 | } |
| 922 | /* END_CASE */ |
| 923 | |
| 924 | |
| 925 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 926 | void mpi_add_abs(char *input_X, char *input_Y, |
| 927 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 928 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 929 | mbedtls_mpi X, Y, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 930 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 931 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 932 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 933 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 934 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 935 | TEST_ASSERT(mbedtls_mpi_add_abs(&Z, &X, &Y) == 0); |
| 936 | TEST_ASSERT(sign_is_valid(&Z)); |
| 937 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 938 | |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 939 | /* result == first operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 940 | TEST_ASSERT(mbedtls_mpi_add_abs(&X, &X, &Y) == 0); |
| 941 | TEST_ASSERT(sign_is_valid(&X)); |
| 942 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
| 943 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 944 | |
| 945 | /* result == second operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 946 | TEST_ASSERT(mbedtls_mpi_add_abs(&Y, &X, &Y) == 0); |
| 947 | TEST_ASSERT(sign_is_valid(&Y)); |
| 948 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 949 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 950 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 951 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | ba48cb2 | 2009-07-12 11:01:32 +0000 | [diff] [blame] | 952 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 953 | /* END_CASE */ |
Paul Bakker | ba48cb2 | 2009-07-12 11:01:32 +0000 | [diff] [blame] | 954 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 955 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 956 | void mpi_add_int(char *input_X, int input_Y, |
| 957 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 958 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 959 | mbedtls_mpi X, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 960 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 961 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 962 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 963 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 964 | TEST_ASSERT(mbedtls_mpi_add_int(&Z, &X, input_Y) == 0); |
| 965 | TEST_ASSERT(sign_is_valid(&Z)); |
| 966 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 967 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 968 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 969 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 970 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 971 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 972 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 973 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 974 | void mpi_sub_mpi(char *input_X, char *input_Y, |
| 975 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 976 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 977 | mbedtls_mpi X, Y, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 978 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 979 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 980 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 981 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 982 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 983 | TEST_ASSERT(mbedtls_mpi_sub_mpi(&Z, &X, &Y) == 0); |
| 984 | TEST_ASSERT(sign_is_valid(&Z)); |
| 985 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 986 | |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 987 | /* result == first operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 988 | TEST_ASSERT(mbedtls_mpi_sub_mpi(&X, &X, &Y) == 0); |
| 989 | TEST_ASSERT(sign_is_valid(&X)); |
| 990 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
| 991 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 992 | |
| 993 | /* result == second operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 994 | TEST_ASSERT(mbedtls_mpi_sub_mpi(&Y, &X, &Y) == 0); |
| 995 | TEST_ASSERT(sign_is_valid(&Y)); |
| 996 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &A) == 0); |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 997 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 998 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 999 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1000 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1001 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1002 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1003 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1004 | void mpi_sub_abs(char *input_X, char *input_Y, |
| 1005 | char *input_A, int sub_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1006 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1007 | mbedtls_mpi X, Y, Z, A; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1008 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1009 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1010 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1011 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1012 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 1013 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
Manuel Pégourié-Gonnard | e670f90 | 2015-10-30 09:23:19 +0100 | [diff] [blame] | 1014 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1015 | res = mbedtls_mpi_sub_abs(&Z, &X, &Y); |
| 1016 | TEST_ASSERT(res == sub_result); |
| 1017 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1018 | if (res == 0) { |
| 1019 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
| 1020 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1021 | |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 1022 | /* result == first operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1023 | TEST_ASSERT(mbedtls_mpi_sub_abs(&X, &X, &Y) == sub_result); |
| 1024 | TEST_ASSERT(sign_is_valid(&X)); |
| 1025 | if (sub_result == 0) { |
| 1026 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
| 1027 | } |
| 1028 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 1029 | |
| 1030 | /* result == second operand */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1031 | TEST_ASSERT(mbedtls_mpi_sub_abs(&Y, &X, &Y) == sub_result); |
| 1032 | TEST_ASSERT(sign_is_valid(&Y)); |
| 1033 | if (sub_result == 0) { |
| 1034 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Y, &A) == 0); |
| 1035 | } |
Gilles Peskine | 56f943a | 2020-07-23 01:18:11 +0200 | [diff] [blame] | 1036 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1037 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1038 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1039 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1040 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1041 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1042 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1043 | void mpi_sub_int(char *input_X, int input_Y, |
| 1044 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1045 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1046 | mbedtls_mpi X, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1047 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1048 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1049 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1050 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1051 | TEST_ASSERT(mbedtls_mpi_sub_int(&Z, &X, input_Y) == 0); |
| 1052 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1053 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1054 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1055 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1056 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1057 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1058 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1059 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1060 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1061 | void mpi_mul_mpi(char *input_X, char *input_Y, |
| 1062 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1063 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1064 | mbedtls_mpi X, Y, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1065 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1066 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1067 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1068 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 1069 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1070 | TEST_ASSERT(mbedtls_mpi_mul_mpi(&Z, &X, &Y) == 0); |
| 1071 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1072 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1073 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1074 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1075 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1076 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1077 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1078 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1079 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1080 | void mpi_mul_int(char *input_X, int input_Y, |
| 1081 | char *input_A, char *result_comparison) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1082 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1083 | mbedtls_mpi X, Z, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1084 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1085 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1086 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1087 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1088 | TEST_ASSERT(mbedtls_mpi_mul_int(&Z, &X, input_Y) == 0); |
| 1089 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1090 | if (strcmp(result_comparison, "==") == 0) { |
| 1091 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
| 1092 | } else if (strcmp(result_comparison, "!=") == 0) { |
| 1093 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) != 0); |
| 1094 | } else { |
| 1095 | TEST_ASSERT("unknown operator" == 0); |
| 1096 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1097 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1098 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1099 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1100 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1101 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1102 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1103 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1104 | void mpi_div_mpi(char *input_X, char *input_Y, |
| 1105 | char *input_A, char *input_B, |
| 1106 | int div_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1107 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1108 | mbedtls_mpi X, Y, Q, R, A, B; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1109 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1110 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Q); mbedtls_mpi_init(&R); |
| 1111 | mbedtls_mpi_init(&A); mbedtls_mpi_init(&B); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1112 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1113 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1114 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 1115 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1116 | TEST_ASSERT(mbedtls_test_read_mpi(&B, input_B) == 0); |
| 1117 | res = mbedtls_mpi_div_mpi(&Q, &R, &X, &Y); |
| 1118 | TEST_ASSERT(res == div_result); |
| 1119 | if (res == 0) { |
| 1120 | TEST_ASSERT(sign_is_valid(&Q)); |
| 1121 | TEST_ASSERT(sign_is_valid(&R)); |
| 1122 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Q, &A) == 0); |
| 1123 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&R, &B) == 0); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1124 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1125 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1126 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1127 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Q); mbedtls_mpi_free(&R); |
| 1128 | mbedtls_mpi_free(&A); mbedtls_mpi_free(&B); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1129 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1130 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1131 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1132 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1133 | void mpi_div_int(char *input_X, int input_Y, |
| 1134 | char *input_A, char *input_B, |
| 1135 | int div_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1136 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1137 | mbedtls_mpi X, Q, R, A, B; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1138 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1139 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Q); mbedtls_mpi_init(&R); mbedtls_mpi_init(&A); |
| 1140 | mbedtls_mpi_init(&B); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1141 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1142 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1143 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1144 | TEST_ASSERT(mbedtls_test_read_mpi(&B, input_B) == 0); |
| 1145 | res = mbedtls_mpi_div_int(&Q, &R, &X, input_Y); |
| 1146 | TEST_ASSERT(res == div_result); |
| 1147 | if (res == 0) { |
| 1148 | TEST_ASSERT(sign_is_valid(&Q)); |
| 1149 | TEST_ASSERT(sign_is_valid(&R)); |
| 1150 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Q, &A) == 0); |
| 1151 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&R, &B) == 0); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1152 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1153 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1154 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1155 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Q); mbedtls_mpi_free(&R); mbedtls_mpi_free(&A); |
| 1156 | mbedtls_mpi_free(&B); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1157 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1158 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1159 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1160 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1161 | void mpi_mod_mpi(char *input_X, char *input_Y, |
| 1162 | char *input_A, int div_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1163 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1164 | mbedtls_mpi X, Y, A; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1165 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1166 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1167 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1168 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1169 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 1170 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1171 | res = mbedtls_mpi_mod_mpi(&X, &X, &Y); |
| 1172 | TEST_ASSERT(res == div_result); |
| 1173 | if (res == 0) { |
| 1174 | TEST_ASSERT(sign_is_valid(&X)); |
| 1175 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1176 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1177 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1178 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1179 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1180 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1181 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1182 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1183 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1184 | void mpi_mod_int(char *input_X, char *input_Y, |
| 1185 | char *input_A, int mod_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1186 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1187 | mbedtls_mpi X; |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1188 | mbedtls_mpi Y; |
| 1189 | mbedtls_mpi A; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1190 | int res; |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1191 | mbedtls_mpi_uint r; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1192 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1193 | mbedtls_mpi_init(&X); |
| 1194 | mbedtls_mpi_init(&Y); |
| 1195 | mbedtls_mpi_init(&A); |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1196 | |
| 1197 | /* We use MPIs to read Y and A since the test framework limits us to |
| 1198 | * ints, so we can't have 64-bit values */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1199 | TEST_EQUAL(mbedtls_test_read_mpi(&X, input_X), 0); |
| 1200 | TEST_EQUAL(mbedtls_test_read_mpi(&Y, input_Y), 0); |
| 1201 | TEST_EQUAL(mbedtls_test_read_mpi(&A, input_A), 0); |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1202 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1203 | TEST_EQUAL(Y.n, 1); |
| 1204 | TEST_EQUAL(A.n, 1); |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1205 | |
Tom Cosgrove | 5c307b9 | 2022-11-10 12:05:55 +0000 | [diff] [blame] | 1206 | /* Convert the MPIs for Y and A to (signed) mbedtls_mpi_sints */ |
| 1207 | |
| 1208 | /* Since we're converting sign+magnitude to two's complement, we lose one |
| 1209 | * bit of value in the output. This means there are some values we can't |
| 1210 | * represent, e.g. (hex) -A0000000 on 32-bit systems. These are technically |
| 1211 | * invalid test cases, so could be considered "won't happen", but they are |
| 1212 | * easy to test for, and this helps guard against human error. */ |
| 1213 | |
| 1214 | mbedtls_mpi_sint y = (mbedtls_mpi_sint) Y.p[0]; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1215 | TEST_ASSERT(y >= 0); /* If y < 0 here, we can't make negative y */ |
| 1216 | if (Y.s == -1) { |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1217 | y = -y; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1218 | } |
Tom Cosgrove | 5c307b9 | 2022-11-10 12:05:55 +0000 | [diff] [blame] | 1219 | |
| 1220 | mbedtls_mpi_sint a = (mbedtls_mpi_sint) A.p[0]; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1221 | TEST_ASSERT(a >= 0); /* Same goes for a */ |
| 1222 | if (A.s == -1) { |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1223 | a = -a; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1224 | } |
Tom Cosgrove | c2c6fcb | 2022-11-09 12:59:33 +0000 | [diff] [blame] | 1225 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1226 | res = mbedtls_mpi_mod_int(&r, &X, y); |
| 1227 | TEST_EQUAL(res, mod_result); |
| 1228 | if (res == 0) { |
| 1229 | TEST_EQUAL(r, a); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1230 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1231 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1232 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1233 | mbedtls_mpi_free(&X); |
| 1234 | mbedtls_mpi_free(&Y); |
| 1235 | mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1236 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1237 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1238 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1239 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1240 | void mpi_exp_mod(char *input_A, char *input_E, |
| 1241 | char *input_N, char *input_X, |
| 1242 | int exp_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1243 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1244 | mbedtls_mpi A, E, N, RR, Z, X; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1245 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1246 | mbedtls_mpi_init(&A); mbedtls_mpi_init(&E); mbedtls_mpi_init(&N); |
| 1247 | mbedtls_mpi_init(&RR); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1248 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1249 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1250 | TEST_ASSERT(mbedtls_test_read_mpi(&E, input_E) == 0); |
| 1251 | TEST_ASSERT(mbedtls_test_read_mpi(&N, input_N) == 0); |
| 1252 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1253 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1254 | res = mbedtls_mpi_exp_mod(&Z, &A, &E, &N, NULL); |
| 1255 | TEST_ASSERT(res == exp_result); |
| 1256 | if (res == 0) { |
| 1257 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1258 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &X) == 0); |
Gilles Peskine | 4cc8021 | 2021-06-09 18:31:35 +0200 | [diff] [blame] | 1259 | } |
| 1260 | |
| 1261 | /* Now test again with the speed-up parameter supplied as an output. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1262 | res = mbedtls_mpi_exp_mod(&Z, &A, &E, &N, &RR); |
| 1263 | TEST_ASSERT(res == exp_result); |
| 1264 | if (res == 0) { |
| 1265 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1266 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &X) == 0); |
Gilles Peskine | 4cc8021 | 2021-06-09 18:31:35 +0200 | [diff] [blame] | 1267 | } |
| 1268 | |
| 1269 | /* Now test again with the speed-up parameter supplied in calculated form. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1270 | res = mbedtls_mpi_exp_mod(&Z, &A, &E, &N, &RR); |
| 1271 | TEST_ASSERT(res == exp_result); |
| 1272 | if (res == 0) { |
| 1273 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1274 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &X) == 0); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1275 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1276 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1277 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1278 | mbedtls_mpi_free(&A); mbedtls_mpi_free(&E); mbedtls_mpi_free(&N); |
| 1279 | mbedtls_mpi_free(&RR); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1280 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1281 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1282 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1283 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1284 | void mpi_exp_mod_size(int A_bytes, int E_bytes, int N_bytes, |
| 1285 | char *input_RR, int exp_result) |
Chris Jones | d10b331 | 2020-12-02 10:41:50 +0000 | [diff] [blame] | 1286 | { |
| 1287 | mbedtls_mpi A, E, N, RR, Z; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1288 | mbedtls_mpi_init(&A); mbedtls_mpi_init(&E); mbedtls_mpi_init(&N); |
| 1289 | mbedtls_mpi_init(&RR); mbedtls_mpi_init(&Z); |
Chris Jones | d10b331 | 2020-12-02 10:41:50 +0000 | [diff] [blame] | 1290 | |
Chris Jones | aa850cd | 2020-12-03 11:35:41 +0000 | [diff] [blame] | 1291 | /* Set A to 2^(A_bytes - 1) + 1 */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1292 | TEST_ASSERT(mbedtls_mpi_lset(&A, 1) == 0); |
| 1293 | TEST_ASSERT(mbedtls_mpi_shift_l(&A, (A_bytes * 8) - 1) == 0); |
| 1294 | TEST_ASSERT(mbedtls_mpi_set_bit(&A, 0, 1) == 0); |
Chris Jones | aa850cd | 2020-12-03 11:35:41 +0000 | [diff] [blame] | 1295 | |
| 1296 | /* Set E to 2^(E_bytes - 1) + 1 */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1297 | TEST_ASSERT(mbedtls_mpi_lset(&E, 1) == 0); |
| 1298 | TEST_ASSERT(mbedtls_mpi_shift_l(&E, (E_bytes * 8) - 1) == 0); |
| 1299 | TEST_ASSERT(mbedtls_mpi_set_bit(&E, 0, 1) == 0); |
Chris Jones | aa850cd | 2020-12-03 11:35:41 +0000 | [diff] [blame] | 1300 | |
| 1301 | /* Set N to 2^(N_bytes - 1) + 1 */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1302 | TEST_ASSERT(mbedtls_mpi_lset(&N, 1) == 0); |
| 1303 | TEST_ASSERT(mbedtls_mpi_shift_l(&N, (N_bytes * 8) - 1) == 0); |
| 1304 | TEST_ASSERT(mbedtls_mpi_set_bit(&N, 0, 1) == 0); |
Chris Jones | d10b331 | 2020-12-02 10:41:50 +0000 | [diff] [blame] | 1305 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1306 | if (strlen(input_RR)) { |
| 1307 | TEST_ASSERT(mbedtls_test_read_mpi(&RR, input_RR) == 0); |
| 1308 | } |
Chris Jones | d10b331 | 2020-12-02 10:41:50 +0000 | [diff] [blame] | 1309 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1310 | TEST_ASSERT(mbedtls_mpi_exp_mod(&Z, &A, &E, &N, &RR) == exp_result); |
Chris Jones | d10b331 | 2020-12-02 10:41:50 +0000 | [diff] [blame] | 1311 | |
| 1312 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1313 | mbedtls_mpi_free(&A); mbedtls_mpi_free(&E); mbedtls_mpi_free(&N); |
| 1314 | mbedtls_mpi_free(&RR); mbedtls_mpi_free(&Z); |
Chris Jones | d10b331 | 2020-12-02 10:41:50 +0000 | [diff] [blame] | 1315 | } |
| 1316 | /* END_CASE */ |
| 1317 | |
| 1318 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1319 | void mpi_inv_mod(char *input_X, char *input_Y, |
| 1320 | char *input_A, int div_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1321 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1322 | mbedtls_mpi X, Y, Z, A; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1323 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1324 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&Y); mbedtls_mpi_init(&Z); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1325 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1326 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1327 | TEST_ASSERT(mbedtls_test_read_mpi(&Y, input_Y) == 0); |
| 1328 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1329 | res = mbedtls_mpi_inv_mod(&Z, &X, &Y); |
| 1330 | TEST_ASSERT(res == div_result); |
| 1331 | if (res == 0) { |
| 1332 | TEST_ASSERT(sign_is_valid(&Z)); |
| 1333 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&Z, &A) == 0); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1334 | } |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1335 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1336 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1337 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&Y); mbedtls_mpi_free(&Z); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1338 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1339 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1340 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1341 | /* BEGIN_CASE depends_on:MBEDTLS_GENPRIME */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1342 | void mpi_is_prime(char *input_X, int div_result) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1343 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1344 | mbedtls_mpi X; |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1345 | int res; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1346 | mbedtls_mpi_init(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1347 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1348 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1349 | res = mbedtls_mpi_is_prime_ext(&X, 40, mbedtls_test_rnd_std_rand, NULL); |
| 1350 | TEST_ASSERT(res == div_result); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1351 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1352 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1353 | mbedtls_mpi_free(&X); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1354 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1355 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1356 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1357 | /* BEGIN_CASE depends_on:MBEDTLS_GENPRIME */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1358 | void mpi_is_prime_det(data_t *input_X, data_t *witnesses, |
| 1359 | int chunk_len, int rounds) |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 1360 | { |
| 1361 | mbedtls_mpi X; |
| 1362 | int res; |
| 1363 | mbedtls_test_mpi_random rand; |
| 1364 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1365 | mbedtls_mpi_init(&X); |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 1366 | rand.data = witnesses; |
| 1367 | rand.pos = 0; |
| 1368 | rand.chunk_len = chunk_len; |
| 1369 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1370 | TEST_ASSERT(mbedtls_mpi_read_binary(&X, input_X->x, input_X->len) == 0); |
| 1371 | res = mbedtls_mpi_is_prime_ext(&X, rounds - 1, |
| 1372 | mbedtls_test_mpi_miller_rabin_determinizer, |
| 1373 | &rand); |
| 1374 | TEST_ASSERT(res == 0); |
Darryl Green | ac2ead0 | 2018-10-02 15:30:39 +0100 | [diff] [blame] | 1375 | |
| 1376 | rand.data = witnesses; |
| 1377 | rand.pos = 0; |
| 1378 | rand.chunk_len = chunk_len; |
| 1379 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1380 | res = mbedtls_mpi_is_prime_ext(&X, rounds, |
| 1381 | mbedtls_test_mpi_miller_rabin_determinizer, |
| 1382 | &rand); |
| 1383 | TEST_ASSERT(res == MBEDTLS_ERR_MPI_NOT_ACCEPTABLE); |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 1384 | |
| 1385 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1386 | mbedtls_mpi_free(&X); |
Janos Follath | 64eca05 | 2018-09-05 17:04:49 +0100 | [diff] [blame] | 1387 | } |
| 1388 | /* END_CASE */ |
| 1389 | |
| 1390 | /* BEGIN_CASE depends_on:MBEDTLS_GENPRIME */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1391 | void mpi_gen_prime(int bits, int flags, int ref_ret) |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1392 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1393 | mbedtls_mpi X; |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1394 | int my_ret; |
| 1395 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1396 | mbedtls_mpi_init(&X); |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1397 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1398 | my_ret = mbedtls_mpi_gen_prime(&X, bits, flags, |
| 1399 | mbedtls_test_rnd_std_rand, NULL); |
| 1400 | TEST_ASSERT(my_ret == ref_ret); |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1401 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1402 | if (ref_ret == 0) { |
| 1403 | size_t actual_bits = mbedtls_mpi_bitlen(&X); |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1404 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1405 | TEST_ASSERT(actual_bits >= (size_t) bits); |
| 1406 | TEST_ASSERT(actual_bits <= (size_t) bits + 1); |
| 1407 | TEST_ASSERT(sign_is_valid(&X)); |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1408 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1409 | TEST_ASSERT(mbedtls_mpi_is_prime_ext(&X, 40, |
| 1410 | mbedtls_test_rnd_std_rand, |
| 1411 | NULL) == 0); |
| 1412 | if (flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH) { |
Hanno Becker | d4d6057 | 2018-01-10 07:12:01 +0000 | [diff] [blame] | 1413 | /* X = ( X - 1 ) / 2 */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1414 | TEST_ASSERT(mbedtls_mpi_shift_r(&X, 1) == 0); |
| 1415 | TEST_ASSERT(mbedtls_mpi_is_prime_ext(&X, 40, |
| 1416 | mbedtls_test_rnd_std_rand, |
| 1417 | NULL) == 0); |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1418 | } |
| 1419 | } |
| 1420 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1421 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1422 | mbedtls_mpi_free(&X); |
Manuel Pégourié-Gonnard | 15f58a8 | 2014-06-16 17:12:40 +0200 | [diff] [blame] | 1423 | } |
| 1424 | /* END_CASE */ |
| 1425 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1426 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1427 | void mpi_shift_l(char *input_X, int shift_X, |
| 1428 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1429 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1430 | mbedtls_mpi X, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1431 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1432 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1433 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1434 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1435 | TEST_ASSERT(mbedtls_mpi_shift_l(&X, shift_X) == 0); |
| 1436 | TEST_ASSERT(sign_is_valid(&X)); |
| 1437 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1438 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1439 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1440 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1441 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1442 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1443 | |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1444 | /* BEGIN_CASE */ |
Gilles Peskine | 366e685 | 2023-04-26 22:43:54 +0200 | [diff] [blame^] | 1445 | void mpi_shift_r(char *input_X, int shift_X, |
| 1446 | char *input_A) |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1447 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1448 | mbedtls_mpi X, A; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1449 | mbedtls_mpi_init(&X); mbedtls_mpi_init(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1450 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1451 | TEST_ASSERT(mbedtls_test_read_mpi(&X, input_X) == 0); |
| 1452 | TEST_ASSERT(mbedtls_test_read_mpi(&A, input_A) == 0); |
| 1453 | TEST_ASSERT(mbedtls_mpi_shift_r(&X, shift_X) == 0); |
| 1454 | TEST_ASSERT(sign_is_valid(&X)); |
| 1455 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&X, &A) == 0); |
Paul Bakker | 6c591fa | 2011-05-05 11:49:20 +0000 | [diff] [blame] | 1456 | |
Paul Bakker | bd51b26 | 2014-07-10 15:26:12 +0200 | [diff] [blame] | 1457 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1458 | mbedtls_mpi_free(&X); mbedtls_mpi_free(&A); |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1459 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1460 | /* END_CASE */ |
Paul Bakker | 367dae4 | 2009-06-28 21:50:27 +0000 | [diff] [blame] | 1461 | |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1462 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1463 | void mpi_fill_random(int wanted_bytes, int rng_bytes, |
| 1464 | int before, int expected_ret) |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1465 | { |
| 1466 | mbedtls_mpi X; |
| 1467 | int ret; |
| 1468 | size_t bytes_left = rng_bytes; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1469 | mbedtls_mpi_init(&X); |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1470 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1471 | if (before != 0) { |
Gilles Peskine | f467e1a | 2021-04-02 00:02:27 +0200 | [diff] [blame] | 1472 | /* Set X to sign(before) * 2^(|before|-1) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1473 | TEST_ASSERT(mbedtls_mpi_lset(&X, before > 0 ? 1 : -1) == 0); |
| 1474 | if (before < 0) { |
| 1475 | before = -before; |
| 1476 | } |
| 1477 | TEST_ASSERT(mbedtls_mpi_shift_l(&X, before - 1) == 0); |
Gilles Peskine | f467e1a | 2021-04-02 00:02:27 +0200 | [diff] [blame] | 1478 | } |
| 1479 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1480 | ret = mbedtls_mpi_fill_random(&X, wanted_bytes, |
| 1481 | f_rng_bytes_left, &bytes_left); |
| 1482 | TEST_ASSERT(ret == expected_ret); |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1483 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1484 | if (expected_ret == 0) { |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1485 | /* mbedtls_mpi_fill_random is documented to use bytes from the RNG |
| 1486 | * as a big-endian representation of the number. We know when |
| 1487 | * our RNG function returns null bytes, so we know how many |
| 1488 | * leading zero bytes the number has. */ |
| 1489 | size_t leading_zeros = 0; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1490 | if (wanted_bytes > 0 && rng_bytes % 256 == 0) { |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1491 | leading_zeros = 1; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1492 | } |
| 1493 | TEST_ASSERT(mbedtls_mpi_size(&X) + leading_zeros == |
| 1494 | (size_t) wanted_bytes); |
| 1495 | TEST_ASSERT((int) bytes_left == rng_bytes - wanted_bytes); |
| 1496 | TEST_ASSERT(sign_is_valid(&X)); |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1497 | } |
| 1498 | |
| 1499 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1500 | mbedtls_mpi_free(&X); |
Gilles Peskine | 3cb1e29 | 2020-11-25 15:37:20 +0100 | [diff] [blame] | 1501 | } |
| 1502 | /* END_CASE */ |
| 1503 | |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1504 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1505 | void mpi_random_many(int min, data_t *bound_bytes, int iterations) |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1506 | { |
| 1507 | /* Generate numbers in the range 1..bound-1. Do it iterations times. |
| 1508 | * This function assumes that the value of bound is at least 2 and |
| 1509 | * that iterations is large enough that a one-in-2^iterations chance |
| 1510 | * effectively never occurs. |
| 1511 | */ |
| 1512 | |
| 1513 | mbedtls_mpi upper_bound; |
| 1514 | size_t n_bits; |
| 1515 | mbedtls_mpi result; |
| 1516 | size_t b; |
| 1517 | /* If upper_bound is small, stats[b] is the number of times the value b |
| 1518 | * has been generated. Otherwise stats[b] is the number of times a |
| 1519 | * value with bit b set has been generated. */ |
| 1520 | size_t *stats = NULL; |
| 1521 | size_t stats_len; |
| 1522 | int full_stats; |
| 1523 | size_t i; |
| 1524 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1525 | mbedtls_mpi_init(&upper_bound); |
| 1526 | mbedtls_mpi_init(&result); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1527 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1528 | TEST_EQUAL(0, mbedtls_mpi_read_binary(&upper_bound, |
| 1529 | bound_bytes->x, bound_bytes->len)); |
| 1530 | n_bits = mbedtls_mpi_bitlen(&upper_bound); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1531 | /* Consider a bound "small" if it's less than 2^5. This value is chosen |
| 1532 | * to be small enough that the probability of missing one value is |
| 1533 | * negligible given the number of iterations. It must be less than |
| 1534 | * 256 because some of the code below assumes that "small" values |
| 1535 | * fit in a byte. */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1536 | if (n_bits <= 5) { |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1537 | full_stats = 1; |
| 1538 | stats_len = bound_bytes->x[bound_bytes->len - 1]; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1539 | } else { |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1540 | full_stats = 0; |
| 1541 | stats_len = n_bits; |
| 1542 | } |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1543 | ASSERT_ALLOC(stats, stats_len); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1544 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1545 | for (i = 0; i < (size_t) iterations; i++) { |
| 1546 | mbedtls_test_set_step(i); |
| 1547 | TEST_EQUAL(0, mbedtls_mpi_random(&result, min, &upper_bound, |
| 1548 | mbedtls_test_rnd_std_rand, NULL)); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1549 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1550 | TEST_ASSERT(sign_is_valid(&result)); |
| 1551 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&result, &upper_bound) < 0); |
| 1552 | TEST_ASSERT(mbedtls_mpi_cmp_int(&result, min) >= 0); |
| 1553 | if (full_stats) { |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1554 | uint8_t value; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1555 | TEST_EQUAL(0, mbedtls_mpi_write_binary(&result, &value, 1)); |
| 1556 | TEST_ASSERT(value < stats_len); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1557 | ++stats[value]; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1558 | } else { |
| 1559 | for (b = 0; b < n_bits; b++) { |
| 1560 | stats[b] += mbedtls_mpi_get_bit(&result, b); |
| 1561 | } |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1562 | } |
| 1563 | } |
| 1564 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1565 | if (full_stats) { |
| 1566 | for (b = min; b < stats_len; b++) { |
| 1567 | mbedtls_test_set_step(1000000 + b); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1568 | /* Assert that each value has been reached at least once. |
| 1569 | * This is almost guaranteed if the iteration count is large |
| 1570 | * enough. This is a very crude way of checking the distribution. |
| 1571 | */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1572 | TEST_ASSERT(stats[b] > 0); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1573 | } |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1574 | } else { |
Gilles Peskine | e4f937f | 2021-06-02 21:24:04 +0200 | [diff] [blame] | 1575 | int statistically_safe_all_the_way = |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1576 | is_significantly_above_a_power_of_2(bound_bytes); |
| 1577 | for (b = 0; b < n_bits; b++) { |
| 1578 | mbedtls_test_set_step(1000000 + b); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1579 | /* Assert that each bit has been set in at least one result and |
| 1580 | * clear in at least one result. Provided that iterations is not |
| 1581 | * too small, it would be extremely unlikely for this not to be |
| 1582 | * the case if the results are uniformly distributed. |
| 1583 | * |
| 1584 | * As an exception, the top bit may legitimately never be set |
| 1585 | * if bound is a power of 2 or only slightly above. |
| 1586 | */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1587 | if (statistically_safe_all_the_way || b != n_bits - 1) { |
| 1588 | TEST_ASSERT(stats[b] > 0); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1589 | } |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1590 | TEST_ASSERT(stats[b] < (size_t) iterations); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1591 | } |
| 1592 | } |
| 1593 | |
| 1594 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1595 | mbedtls_mpi_free(&upper_bound); |
| 1596 | mbedtls_mpi_free(&result); |
| 1597 | mbedtls_free(stats); |
Gilles Peskine | 4699fa4 | 2021-03-29 22:02:55 +0200 | [diff] [blame] | 1598 | } |
| 1599 | /* END_CASE */ |
| 1600 | |
Gilles Peskine | 9312ba5 | 2021-03-29 22:14:51 +0200 | [diff] [blame] | 1601 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1602 | void mpi_random_sizes(int min, data_t *bound_bytes, int nlimbs, int before) |
Gilles Peskine | 8f45470 | 2021-04-01 15:57:18 +0200 | [diff] [blame] | 1603 | { |
| 1604 | mbedtls_mpi upper_bound; |
| 1605 | mbedtls_mpi result; |
| 1606 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1607 | mbedtls_mpi_init(&upper_bound); |
| 1608 | mbedtls_mpi_init(&result); |
Gilles Peskine | 8f45470 | 2021-04-01 15:57:18 +0200 | [diff] [blame] | 1609 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1610 | if (before != 0) { |
Gilles Peskine | f467e1a | 2021-04-02 00:02:27 +0200 | [diff] [blame] | 1611 | /* Set result to sign(before) * 2^(|before|-1) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1612 | TEST_ASSERT(mbedtls_mpi_lset(&result, before > 0 ? 1 : -1) == 0); |
| 1613 | if (before < 0) { |
| 1614 | before = -before; |
| 1615 | } |
| 1616 | TEST_ASSERT(mbedtls_mpi_shift_l(&result, before - 1) == 0); |
Gilles Peskine | f467e1a | 2021-04-02 00:02:27 +0200 | [diff] [blame] | 1617 | } |
| 1618 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1619 | TEST_EQUAL(0, mbedtls_mpi_grow(&result, nlimbs)); |
| 1620 | TEST_EQUAL(0, mbedtls_mpi_read_binary(&upper_bound, |
| 1621 | bound_bytes->x, bound_bytes->len)); |
| 1622 | TEST_EQUAL(0, mbedtls_mpi_random(&result, min, &upper_bound, |
| 1623 | mbedtls_test_rnd_std_rand, NULL)); |
| 1624 | TEST_ASSERT(sign_is_valid(&result)); |
| 1625 | TEST_ASSERT(mbedtls_mpi_cmp_mpi(&result, &upper_bound) < 0); |
| 1626 | TEST_ASSERT(mbedtls_mpi_cmp_int(&result, min) >= 0); |
Gilles Peskine | 8f45470 | 2021-04-01 15:57:18 +0200 | [diff] [blame] | 1627 | |
| 1628 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1629 | mbedtls_mpi_free(&upper_bound); |
| 1630 | mbedtls_mpi_free(&result); |
Gilles Peskine | 8f45470 | 2021-04-01 15:57:18 +0200 | [diff] [blame] | 1631 | } |
| 1632 | /* END_CASE */ |
| 1633 | |
| 1634 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1635 | void mpi_random_fail(int min, data_t *bound_bytes, int expected_ret) |
Gilles Peskine | 9312ba5 | 2021-03-29 22:14:51 +0200 | [diff] [blame] | 1636 | { |
| 1637 | mbedtls_mpi upper_bound; |
| 1638 | mbedtls_mpi result; |
| 1639 | int actual_ret; |
| 1640 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1641 | mbedtls_mpi_init(&upper_bound); |
| 1642 | mbedtls_mpi_init(&result); |
Gilles Peskine | 9312ba5 | 2021-03-29 22:14:51 +0200 | [diff] [blame] | 1643 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1644 | TEST_EQUAL(0, mbedtls_mpi_read_binary(&upper_bound, |
| 1645 | bound_bytes->x, bound_bytes->len)); |
| 1646 | actual_ret = mbedtls_mpi_random(&result, min, &upper_bound, |
| 1647 | mbedtls_test_rnd_std_rand, NULL); |
| 1648 | TEST_EQUAL(expected_ret, actual_ret); |
Gilles Peskine | 9312ba5 | 2021-03-29 22:14:51 +0200 | [diff] [blame] | 1649 | |
| 1650 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1651 | mbedtls_mpi_free(&upper_bound); |
| 1652 | mbedtls_mpi_free(&result); |
Gilles Peskine | 9312ba5 | 2021-03-29 22:14:51 +0200 | [diff] [blame] | 1653 | } |
| 1654 | /* END_CASE */ |
| 1655 | |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1656 | /* BEGIN_CASE */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1657 | void most_negative_mpi_sint() |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1658 | { |
| 1659 | /* Ad hoc tests for n = -p = -2^(biL-1) as a mbedtls_mpi_sint. We |
| 1660 | * guarantee that mbedtls_mpi_sint is a two's complement type, so this |
| 1661 | * is a valid value. However, negating it (`-n`) has undefined behavior |
| 1662 | * (although in practice `-n` evaluates to the value n). |
| 1663 | * |
| 1664 | * This function has ad hoc tests for this value. It's separated from other |
| 1665 | * functions because the test framework makes it hard to pass this value |
| 1666 | * into test cases. |
| 1667 | * |
| 1668 | * In the comments here: |
| 1669 | * - biL = number of bits in limbs |
| 1670 | * - p = 2^(biL-1) (smallest positive value not in mbedtls_mpi_sint range) |
| 1671 | * - n = -2^(biL-1) (largest negative value in mbedtls_mpi_sint range) |
| 1672 | */ |
| 1673 | |
| 1674 | mbedtls_mpi A, R, X; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1675 | mbedtls_mpi_init(&A); |
| 1676 | mbedtls_mpi_init(&R); |
| 1677 | mbedtls_mpi_init(&X); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1678 | |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1679 | const size_t biL = 8 * sizeof(mbedtls_mpi_sint); |
| 1680 | mbedtls_mpi_uint most_positive_plus_1 = (mbedtls_mpi_uint) 1 << (biL - 1); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1681 | const mbedtls_mpi_sint most_positive = most_positive_plus_1 - 1; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1682 | const mbedtls_mpi_sint most_negative = -most_positive - 1; |
| 1683 | TEST_EQUAL((mbedtls_mpi_uint) most_negative, |
| 1684 | (mbedtls_mpi_uint) 1 << (biL - 1)); |
| 1685 | TEST_EQUAL((mbedtls_mpi_uint) most_negative << 1, 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1686 | |
| 1687 | /* Test mbedtls_mpi_lset() */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1688 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_negative), 0); |
| 1689 | TEST_EQUAL(A.s, -1); |
| 1690 | TEST_EQUAL(A.n, 1); |
| 1691 | TEST_EQUAL(A.p[0], most_positive_plus_1); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1692 | |
| 1693 | /* Test mbedtls_mpi_cmp_int(): -p == -p */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1694 | TEST_EQUAL(mbedtls_mpi_cmp_int(&A, most_negative), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1695 | |
| 1696 | /* Test mbedtls_mpi_cmp_int(): -(p+1) < -p */ |
| 1697 | A.p[0] = most_positive_plus_1 + 1; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1698 | TEST_EQUAL(mbedtls_mpi_cmp_int(&A, most_negative), -1); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1699 | |
| 1700 | /* Test mbedtls_mpi_cmp_int(): -(p-1) > -p */ |
| 1701 | A.p[0] = most_positive_plus_1 - 1; |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1702 | TEST_EQUAL(mbedtls_mpi_cmp_int(&A, most_negative), 1); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1703 | |
| 1704 | /* Test mbedtls_mpi_add_int(): (p-1) + (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1705 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_positive), 0); |
| 1706 | TEST_EQUAL(mbedtls_mpi_add_int(&X, &A, most_negative), 0); |
| 1707 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, -1), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1708 | |
| 1709 | /* Test mbedtls_mpi_add_int(): (0) + (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1710 | TEST_EQUAL(mbedtls_mpi_lset(&A, 0), 0); |
| 1711 | TEST_EQUAL(mbedtls_mpi_add_int(&X, &A, most_negative), 0); |
| 1712 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, most_negative), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1713 | |
| 1714 | /* Test mbedtls_mpi_add_int(): (-p) + (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1715 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_negative), 0); |
| 1716 | TEST_EQUAL(mbedtls_mpi_add_int(&X, &A, most_negative), 0); |
| 1717 | TEST_EQUAL(X.s, -1); |
| 1718 | TEST_EQUAL(X.n, 2); |
| 1719 | TEST_EQUAL(X.p[0], 0); |
| 1720 | TEST_EQUAL(X.p[1], 1); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1721 | |
| 1722 | /* Test mbedtls_mpi_sub_int(): (p) - (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1723 | mbedtls_mpi_free(&X); |
| 1724 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_positive), 0); |
| 1725 | TEST_EQUAL(mbedtls_mpi_sub_int(&X, &A, most_negative), 0); |
| 1726 | TEST_EQUAL(X.s, 1); |
| 1727 | TEST_EQUAL(X.n, 1); |
| 1728 | TEST_EQUAL(X.p[0], ~(mbedtls_mpi_uint) 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1729 | |
| 1730 | /* Test mbedtls_mpi_sub_int(): (0) - (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1731 | TEST_EQUAL(mbedtls_mpi_lset(&A, 0), 0); |
| 1732 | TEST_EQUAL(mbedtls_mpi_sub_int(&X, &A, most_negative), 0); |
| 1733 | TEST_EQUAL(X.s, 1); |
| 1734 | TEST_EQUAL(X.n, 1); |
| 1735 | TEST_EQUAL(X.p[0], most_positive_plus_1); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1736 | |
| 1737 | /* Test mbedtls_mpi_sub_int(): (-p) - (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1738 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_negative), 0); |
| 1739 | TEST_EQUAL(mbedtls_mpi_sub_int(&X, &A, most_negative), 0); |
| 1740 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, 0), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1741 | |
| 1742 | /* Test mbedtls_mpi_div_int(): (-p+1) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1743 | TEST_EQUAL(mbedtls_mpi_lset(&A, -most_positive), 0); |
| 1744 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1745 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, 0), 0); |
| 1746 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, -most_positive), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1747 | |
| 1748 | /* Test mbedtls_mpi_div_int(): (-p) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1749 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_negative), 0); |
| 1750 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1751 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, 1), 0); |
| 1752 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, 0), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1753 | |
| 1754 | /* Test mbedtls_mpi_div_int(): (-2*p) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1755 | TEST_EQUAL(mbedtls_mpi_shift_l(&A, 1), 0); |
| 1756 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1757 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, 2), 0); |
| 1758 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, 0), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1759 | |
| 1760 | /* Test mbedtls_mpi_div_int(): (-2*p+1) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1761 | TEST_EQUAL(mbedtls_mpi_add_int(&A, &A, 1), 0); |
| 1762 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1763 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, 1), 0); |
| 1764 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, -most_positive), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1765 | |
| 1766 | /* Test mbedtls_mpi_div_int(): (p-1) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1767 | TEST_EQUAL(mbedtls_mpi_lset(&A, most_positive), 0); |
| 1768 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1769 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, 0), 0); |
| 1770 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, most_positive), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1771 | |
| 1772 | /* Test mbedtls_mpi_div_int(): (p) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1773 | TEST_EQUAL(mbedtls_mpi_add_int(&A, &A, 1), 0); |
| 1774 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1775 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, -1), 0); |
| 1776 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, 0), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1777 | |
| 1778 | /* Test mbedtls_mpi_div_int(): (2*p) / (-p) */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1779 | TEST_EQUAL(mbedtls_mpi_shift_l(&A, 1), 0); |
| 1780 | TEST_EQUAL(mbedtls_mpi_div_int(&X, &R, &A, most_negative), 0); |
| 1781 | TEST_EQUAL(mbedtls_mpi_cmp_int(&X, -2), 0); |
| 1782 | TEST_EQUAL(mbedtls_mpi_cmp_int(&R, 0), 0); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1783 | |
| 1784 | /* Test mbedtls_mpi_mod_int(): never valid */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1785 | TEST_EQUAL(mbedtls_mpi_mod_int(X.p, &A, most_negative), |
| 1786 | MBEDTLS_ERR_MPI_NEGATIVE_VALUE); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1787 | |
| 1788 | /* Test mbedtls_mpi_random(): never valid */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1789 | TEST_EQUAL(mbedtls_mpi_random(&X, most_negative, &A, |
| 1790 | mbedtls_test_rnd_std_rand, NULL), |
| 1791 | MBEDTLS_ERR_MPI_BAD_INPUT_DATA); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1792 | |
| 1793 | exit: |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1794 | mbedtls_mpi_free(&A); |
| 1795 | mbedtls_mpi_free(&R); |
| 1796 | mbedtls_mpi_free(&X); |
Gilles Peskine | de1629a | 2022-11-15 23:02:14 +0100 | [diff] [blame] | 1797 | } |
| 1798 | /* END_CASE */ |
| 1799 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 1800 | /* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */ |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1801 | void mpi_selftest() |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 1802 | { |
Gilles Peskine | 1b6c09a | 2023-01-11 14:52:35 +0100 | [diff] [blame] | 1803 | TEST_ASSERT(mbedtls_mpi_self_test(1) == 0); |
Paul Bakker | e896fea | 2009-07-06 06:40:23 +0000 | [diff] [blame] | 1804 | } |
Paul Bakker | 33b43f1 | 2013-08-20 11:48:36 +0200 | [diff] [blame] | 1805 | /* END_CASE */ |