Fix no-tls1_2
[openssl.git] / test / ecstresstest.c
1 /*
2  * Copyright 2017-2018 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL licenses, (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  * https://www.openssl.org/source/license.html
8  * or in the file LICENSE in the source distribution.
9  */
10
11 #include "internal/nelem.h"
12 #include "testutil.h"
13
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <string.h>
17 #include <ctype.h>
18
19 #define NUM_REPEATS "1000000"
20
21 static int64_t num_repeats;
22 static int print_mode = 0;
23
24 #ifndef OPENSSL_NO_EC
25 # include <openssl/ec.h>
26 # include <openssl/err.h>
27 # include <openssl/obj_mac.h>
28 # include <openssl/objects.h>
29 # include <openssl/rand.h>
30 # include <openssl/bn.h>
31 # include <openssl/opensslconf.h>
32
33 static const char *kP256DefaultResult =
34     "A1E24B223B8E81BC1FFF99BAFB909EDB895FACDE7D6DA5EF5E7B3255FB378E0F";
35
36 /*
37  * Perform a deterministic walk on the curve, by starting from |point| and
38  * using the X-coordinate of the previous point as the next scalar for
39  * point multiplication.
40  * Returns the X-coordinate of the end result or NULL on error.
41  */
42 static BIGNUM *walk_curve(const EC_GROUP *group, EC_POINT *point, int64_t num)
43 {
44     BIGNUM *scalar = NULL;
45     int64_t i;
46
47     if (!TEST_ptr(scalar = BN_new())
48             || !TEST_true(EC_POINT_get_affine_coordinates(group, point, scalar,
49                                                           NULL, NULL)))
50         goto err;
51
52     for (i = 0; i < num; i++) {
53         if (!TEST_true(EC_POINT_mul(group, point, NULL, point, scalar, NULL))
54                 || !TEST_true(EC_POINT_get_affine_coordinates(group, point,
55                                                               scalar,
56                                                               NULL, NULL)))
57             goto err;
58     }
59     return scalar;
60
61 err:
62     BN_free(scalar);
63     return NULL;
64 }
65
66 static int test_curve(void)
67 {
68     EC_GROUP *group = NULL;
69     EC_POINT *point = NULL;
70     BIGNUM *result = NULL, *expected_result = NULL;
71     int ret = 0;
72
73     /*
74      * We currently hard-code P-256, though adaptation to other curves.
75      * would be straightforward.
76      */
77     if (!TEST_ptr(group = EC_GROUP_new_by_curve_name(NID_X9_62_prime256v1))
78             || !TEST_ptr(point = EC_POINT_dup(EC_GROUP_get0_generator(group),
79                                               group))
80             || !TEST_ptr(result = walk_curve(group, point, num_repeats)))
81         return 0;
82
83     if (print_mode) {
84         BN_print(bio_out, result);
85         BIO_printf(bio_out, "\n");
86         ret = 1;
87     } else {
88         if (!TEST_true(BN_hex2bn(&expected_result, kP256DefaultResult))
89                 || !TEST_ptr(expected_result)
90                 || !TEST_BN_eq(result, expected_result))
91             goto err;
92         ret = 1;
93     }
94
95 err:
96     EC_GROUP_free(group);
97     EC_POINT_free(point);
98     BN_free(result);
99     BN_free(expected_result);
100     return ret;
101 }
102 #endif
103
104 static int atoi64(const char *in, int64_t *result)
105 {
106     int64_t ret = 0;
107
108     for ( ; *in != '\0'; in++) {
109         char c = *in;
110
111         if (!isdigit((unsigned char)c))
112             return 0;
113         ret *= 10;
114         ret += (c - '0');
115     }
116     *result = ret;
117     return 1;
118 }
119
120 /*
121  * Stress test the curve. If the '-num' argument is given, runs the loop
122  * |num| times and prints the resulting X-coordinate. Otherwise runs the test
123  * the default number of times and compares against the expected result.
124  */
125 int setup_tests(void)
126 {
127     const char *p;
128
129     if (!atoi64(NUM_REPEATS, &num_repeats)) {
130         TEST_error("Cannot parse " NUM_REPEATS);
131         return 0;
132     }
133     /*
134      * TODO(openssl-team): code under test/ should be able to reuse the option
135      * parsing framework currently in apps/.
136      */
137     p = test_get_option_argument("-num");
138     if (p != NULL) {
139         if (!atoi64(p, &num_repeats)
140                 || num_repeats < 0)
141             return 0;
142         print_mode = 1;
143     }
144
145 #ifndef OPENSSL_NO_EC
146     ADD_TEST(test_curve);
147 #endif
148     return 1;
149 }