Removing memory leaks in rsa setter test
[folly.git] / folly / portability / test / OpenSSLPortabilityTest.cpp
1 /*
2  * Copyright 2017 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (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  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <folly/portability/GTest.h>
18 #include <folly/ssl/OpenSSLPtrTypes.h>
19
20 using namespace folly;
21 using namespace folly::ssl;
22 using namespace testing;
23
24 TEST(OpenSSLPortabilityTest, TestRSASetter) {
25   RsaUniquePtr r(RSA_new());
26   BIGNUM* n = BN_new();
27   BIGNUM* e = BN_new();
28   BIGNUM* d = BN_new();
29   BIGNUM* n_actual;
30   BIGNUM* e_actual;
31   BIGNUM* d_actual;
32   EXPECT_TRUE(BN_set_bit(n, 1));
33   EXPECT_TRUE(BN_set_bit(e, 3));
34   EXPECT_TRUE(BN_set_bit(d, 2));
35   RSA_set0_key(r.get(), n, e, d);
36   RSA_get0_key(
37       r.get(),
38       (const BIGNUM**)&n_actual,
39       (const BIGNUM**)&e_actual,
40       (const BIGNUM**)&d_actual);
41   // BN_cmp returns 0 if the two BIGNUMs are equal
42   EXPECT_FALSE(BN_cmp(n, n_actual));
43   EXPECT_FALSE(BN_cmp(e, e_actual));
44   EXPECT_FALSE(BN_cmp(d, d_actual));
45
46   RsaUniquePtr public_key(RSA_new());
47   BIGNUM* n_public = BN_new();
48   BIGNUM* e_public = BN_new();
49   EXPECT_TRUE(BN_set_bit(n_public, 1));
50   EXPECT_TRUE(BN_set_bit(e_public, 3));
51   RSA_set0_key(public_key.get(), n_public, e_public, nullptr);
52   BIGNUM* n_public_actual;
53   BIGNUM* e_public_actual;
54   RSA_get0_key(
55       public_key.get(),
56       (const BIGNUM**)&n_public_actual,
57       (const BIGNUM**)&e_public_actual,
58       nullptr);
59   EXPECT_FALSE(BN_cmp(n_public, n_public_actual));
60   EXPECT_FALSE(BN_cmp(e_public, e_public_actual));
61 }