Integrate J-PAKE and TLS-PSK. Increase PSK buffer size. Fix memory leaks.
[openssl.git] / crypto / jpake / jpaketest.c
1 #ifndef OPENSSL_EXPERIMENTAL_JPAKE
2
3 #include <stdio.h>
4
5 int main(int argc, char *argv[])
6 {
7     printf("No J-PAKE support\n");
8     return(0);
9 }
10
11 #else
12
13 #include <openssl/jpake.h>
14 #include <openssl/err.h>
15
16 static void showbn(const char *name, const BIGNUM *bn)
17     {
18     fputs(name, stdout);
19     fputs(" = ", stdout);
20     BN_print_fp(stdout, bn);
21     putc('\n', stdout);
22     }
23
24 static int run_jpake(JPAKE_CTX *alice, JPAKE_CTX *bob)
25     {
26     JPAKE_STEP1 alice_s1;
27     JPAKE_STEP1 bob_s1;
28     JPAKE_STEP2 alice_s2;
29     JPAKE_STEP2 bob_s2;
30     JPAKE_STEP3A alice_s3a;
31     JPAKE_STEP3B bob_s3b;
32
33    /* Alice -> Bob: step 1 */
34     puts("A->B s1");
35     JPAKE_STEP1_init(&alice_s1);
36     JPAKE_STEP1_generate(&alice_s1, alice);
37     if(!JPAKE_STEP1_process(bob, &alice_s1))
38         {
39         printf("Bob fails to process Alice's step 1\n");
40         ERR_print_errors_fp(stdout);
41         return 1;
42         }
43     JPAKE_STEP1_release(&alice_s1);
44
45    /* Bob -> Alice: step 1 */
46     puts("B->A s1");
47     JPAKE_STEP1_init(&bob_s1);
48     JPAKE_STEP1_generate(&bob_s1, bob);
49     if(!JPAKE_STEP1_process(alice, &bob_s1))
50         {
51         printf("Alice fails to process Bob's step 1\n");
52         ERR_print_errors_fp(stdout);
53         return 2;
54         }
55     JPAKE_STEP1_release(&bob_s1);
56
57    /* Alice -> Bob: step 2 */
58     puts("A->B s2");
59     JPAKE_STEP2_init(&alice_s2);
60     JPAKE_STEP2_generate(&alice_s2, alice);
61     if(!JPAKE_STEP2_process(bob, &alice_s2))
62         {
63         printf("Bob fails to process Alice's step 2\n");
64         ERR_print_errors_fp(stdout);
65         return 3;
66         }
67     JPAKE_STEP2_release(&alice_s2);
68
69    /* Bob -> Alice: step 2 */
70     puts("B->A s2");
71     JPAKE_STEP2_init(&bob_s2);
72     JPAKE_STEP2_generate(&bob_s2, bob);
73     if(!JPAKE_STEP2_process(alice, &bob_s2))
74         {
75         printf("Alice fails to process Bob's step 2\n");
76         ERR_print_errors_fp(stdout);
77         return 4;
78         }
79     JPAKE_STEP2_release(&bob_s2);
80
81     showbn("Alice's key", JPAKE_get_shared_key(alice));
82     showbn("Bob's key  ", JPAKE_get_shared_key(bob));
83
84    /* Alice -> Bob: step 3a */
85     puts("A->B s3a");
86     JPAKE_STEP3A_init(&alice_s3a);
87     JPAKE_STEP3A_generate(&alice_s3a, alice);
88     if(!JPAKE_STEP3A_process(bob, &alice_s3a))
89         {
90         printf("Bob fails to process Alice's step 3a\n");
91         ERR_print_errors_fp(stdout);
92         return 5;
93         }
94     JPAKE_STEP3A_release(&alice_s3a);
95     
96    /* Bob -> Alice: step 3b */
97     puts("B->A s3b");
98     JPAKE_STEP3B_init(&bob_s3b);
99     JPAKE_STEP3B_generate(&bob_s3b, bob);
100     if(!JPAKE_STEP3B_process(alice, &bob_s3b))
101         {
102         printf("Alice fails to process Bob's step 3b\n");
103         ERR_print_errors_fp(stdout);
104         return 6;
105         }
106     JPAKE_STEP3B_release(&bob_s3b);
107
108     return 0;
109     }
110
111 int main(int argc, char **argv)
112     {
113     JPAKE_CTX *alice;
114     JPAKE_CTX *bob;
115     BIGNUM *p = NULL;
116     BIGNUM *g = NULL;
117     BIGNUM *q = NULL;
118     BIGNUM *secret = BN_new();
119     BIO *bio_err;
120
121     bio_err = BIO_new_fp(stderr, BIO_NOCLOSE);
122
123     CRYPTO_malloc_debug_init();
124     CRYPTO_dbg_set_options(V_CRYPTO_MDEBUG_ALL);
125     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
126
127     ERR_load_crypto_strings();
128
129     /*
130     BN_hex2bn(&p, "fd7f53811d75122952df4a9c2eece4e7f611b7523cef4400c31e3f80b6512669455d402251fb593d8d58fabfc5f5ba30f6cb9b556cd7813b801d346ff26660b76b9950a5a49f9fe8047b1022c24fbba9d7feb7c61bf83b57e7c6a8a6150f04fb83f6d3c51ec3023554135a169132f675f3ae2b61d72aeff22203199dd14801c7");
131     BN_hex2bn(&g, "f7e1a085d69b3ddecbbcab5c36b857b97994afbbfa3aea82f9574c0b3d0782675159578ebad4594fe67107108180b449167123e84c281613b7cf09328cc8a6e13c167a8b547c8d28e0a3ae1e2bb3a675916ea37f0bfa213562f1fb627a01243bcca4f1bea8519089a883dfe15ae59f06928b665e807b552564014c3bfecf492a");
132     BN_hex2bn(&q, "9760508f15230bccb292b982a2eb840bf0581cf5");
133     */
134     /*
135     p = BN_new();
136     BN_generate_prime(p, 1024, 1, NULL, NULL, NULL, NULL);
137     */
138    /* Use a safe prime for p (that we found earlier) */
139     BN_hex2bn(&p, "F9E5B365665EA7A05A9C534502780FEE6F1AB5BD4F49947FD036DBD7E905269AF46EF28B0FC07487EE4F5D20FB3C0AF8E700F3A2FA3414970CBED44FEDFF80CE78D800F184BB82435D137AADA2C6C16523247930A63B85661D1FC817A51ACD96168E95898A1F83A79FFB529368AA7833ABD1B0C3AEDDB14D2E1A2F71D99F763F");
140     showbn("p", p);
141     g = BN_new();
142     BN_set_word(g, 2);
143     showbn("g", g);
144     q = BN_new();
145     BN_rshift1(q, p);
146     showbn("q", q);
147
148     BN_rand(secret, 32, -1, 0);
149
150    /* A normal run, expect this to work... */
151     alice = JPAKE_CTX_new("Alice", "Bob", p, g, q, secret);
152     bob = JPAKE_CTX_new("Bob", "Alice", p, g, q, secret);
153
154     if(run_jpake(alice, bob) != 0)
155         {
156         fprintf(stderr, "Plain JPAKE run failed\n");
157         return 1;
158         }
159
160     JPAKE_CTX_free(bob);
161     JPAKE_CTX_free(alice);
162
163    /* Now give Alice and Bob different secrets */
164     alice = JPAKE_CTX_new("Alice", "Bob", p, g, q, secret);
165     BN_add_word(secret, 1);
166     bob = JPAKE_CTX_new("Bob", "Alice", p, g, q, secret);
167
168     if(run_jpake(alice, bob) != 5)
169         {
170         fprintf(stderr, "Mismatched secret JPAKE run failed\n");
171         return 1;
172         }
173
174     JPAKE_CTX_free(bob);
175     JPAKE_CTX_free(alice);
176
177     BN_free(secret);
178     BN_free(q);
179     BN_free(g);
180     BN_free(p);
181
182     CRYPTO_cleanup_all_ex_data();
183     ERR_remove_state(0);
184     ERR_free_strings();
185     CRYPTO_mem_leaks(bio_err);
186
187     return 0;
188     }
189
190 #endif