+TEST(DipeTest, DipeDecryptSuccessMultiKey) {
+ size_t ptxt_len;
+ size_t overhead;
+ dipe_master_publickey_t pk[3];
+ dipe_master_secretkey_t msk[3];
+ dipe_secretkey_t sk[3];
+
+ element_t y[6];
+ element_t x[6];
+
+ for (size_t i = 0; i < 6; i+=2) {
+ element_init_Zr(y[i], *dipe_get_pairing(param));
+ element_init_Zr(x[i], *dipe_get_pairing(param));
+ element_init_Zr(y[i+1], *dipe_get_pairing(param));
+ element_init_Zr(x[i+1], *dipe_get_pairing(param));
+
+ element_set1(y[i]);
+ element_set1(x[i+1]);
+ element_random(y[i+1]);
+ element_neg(x[i], y[i+1]);
+ }
+
+ for (size_t i = 0; i < 3; ++i) {
+ dipe_master_keygen(param, 6, &pk[i], &msk[i]);
+ dipe_keygen(param, msk[i], "1234567890abcdef", y, &sk[i]);
+ }
+ overhead = dipe_ciphertext_overhead(param, 6, 0);
+
+ char ctxt[overhead + 35];
+ char ptxt[overhead + 35];
+
+
+ for (size_t clen = overhead; clen <= overhead + 35; ++clen) {
+ memset(ctxt, 0, overhead + 35);
+ memset(ptxt, 0, overhead + 35);
+ memcpy(ptxt, "aaaaaaaaaaaaaaaabbbbbbbbbbbbbbbbccc", 35);
+
+ dipe_encrypt(param, 3, pk, 3, x, 35, ptxt, clen, ctxt);
+ ptxt_len = dipe_decrypt(param, 3, sk, 0, "1234567890abcdef", y, clen, ctxt, ptxt);
+
+ if (clen >= overhead + 35) {
+ EXPECT_EQ(ptxt_len, 35);
+ EXPECT_STREQ(ptxt, "aaaaaaaaaaaaaaaabbbbbbbbbbbbbbbbccc");
+ }
+ else {
+ EXPECT_EQ(ptxt_len, 0);
+ }
+ }
+
+ for (size_t i = 0; i < 6; ++i) {
+ element_clear(y[i]);
+ element_clear(x[i]);
+ }
+
+ for (size_t i = 0; i < 3; ++i) {
+ dipe_free_master_secretkey(msk[i]);
+ dipe_free_master_publickey(pk[i]);
+ dipe_free_secretkey(sk[i]);
+ }
+}
+
+TEST(DipeTest, DipeDecryptSuccessThreshold) {
+ size_t ptxt_len;
+ size_t overhead;
+ dipe_master_publickey_t pk[3];
+ dipe_master_secretkey_t msk[3];
+ dipe_secretkey_t sk[3];
+
+ element_t y[6];
+ element_t x[6];
+
+ for (size_t i = 0; i < 6; i+=2) {
+ element_init_Zr(y[i], *dipe_get_pairing(param));
+ element_init_Zr(x[i], *dipe_get_pairing(param));
+ element_init_Zr(y[i+1], *dipe_get_pairing(param));
+ element_init_Zr(x[i+1], *dipe_get_pairing(param));
+
+ element_set1(y[i]);
+ element_set1(x[i+1]);
+ element_random(y[i+1]);
+ element_neg(x[i], y[i+1]);
+ }
+
+ for (size_t i = 0; i < 3; ++i) {
+ dipe_master_keygen(param, 6, &pk[i], &msk[i]);
+ dipe_keygen(param, msk[i], "1234567890abcdef", y, &sk[i]);
+ }
+ overhead = dipe_ciphertext_overhead(param, 6, 2);
+
+ char ctxt[overhead + 35];
+ char ptxt[overhead + 35];
+
+
+ for (size_t clen = overhead+20; clen <= overhead + 35; ++clen) {
+ for (size_t kid = 0; kid < 3; ++kid) {
+ memset(ctxt, 0, overhead + 35);
+ memset(ptxt, 0, overhead + 35);
+ memcpy(ptxt, "aaaaaaaaaaaaaaaabbbbbbbbbbbbbbbbccc", 35);
+
+ dipe_encrypt(param, 3, pk, 1, x, 35, ptxt, clen, ctxt);
+ ptxt_len = dipe_decrypt(param, 1, &sk[kid], 2, "1234567890abcdef", y, clen, ctxt, ptxt);
+
+ if (clen >= overhead + 35) {
+ EXPECT_EQ(ptxt_len, 35);
+ EXPECT_STREQ(ptxt, "aaaaaaaaaaaaaaaabbbbbbbbbbbbbbbbccc");
+ }
+ else {
+ EXPECT_EQ(ptxt_len, 0);
+ }
+ }
+ }
+
+ for (size_t i = 0; i < 6; ++i) {
+ element_clear(y[i]);
+ element_clear(x[i]);
+ }
+
+ for (size_t i = 0; i < 3; ++i) {
+ dipe_free_master_secretkey(msk[i]);
+ dipe_free_master_publickey(pk[i]);
+ dipe_free_secretkey(sk[i]);
+ }
+}
+