benchmark_test.c

#include "deks.h"
#include "peks.h"
#include "sa_peks.h"
#include <sys/time.h>

static double time_cost(struct timeval A, struct timeval B) {
    double res;
    res = (B.tv_sec - A.tv_sec) * 1000000 + B.tv_usec - A.tv_usec;
    return res / 1000; 
}

int main(int argc, char * argv[]) {
    
	char * ec_param = "type f\nq 16283262548997601220198008118239886027035269286659395419233331082106632227801\nr 16283262548997601220198008118239886026907663399064043451383740756301306087801\nb 7322757890651446833342173470888950103129198494728585151431956701557392549679\nbeta 9776063510951480907546829895586341975790848099009256891110816835958685274282\nalpha0 1776307819061905444848005605541335123416176649043693774569737527341520482926\nalpha1 4922472003107175314522406564324183657125055707028154503359780540299732014568";
    //Type-f BN256
    pairing_t pairing;
    element_t g, pk, sk, Tw, Ca;
    mpz_t N, pi, e, d, phi_N;
    int g2_len, zr_len;
    FILE *fp;
    char * g2_buf, * zr_buf;
    char w[64], Cb[HASH_LEN];
    struct timeval start, end;
    double cost1, cost2, cost3;
    
    pairing_init_set_str(pairing, ec_param);
    g2_len = pairing_length_in_bytes_compressed_G2(pairing);
    zr_len = pairing_length_in_bytes_Zr(pairing);
    g2_buf = (char *)malloc(g2_len);
    zr_buf = (char *)malloc(zr_len);

    element_init_G1(Tw, pairing);
    element_init_G2(Ca, pairing);
    element_init_G2(g, pairing);
    element_init_G2(pk, pairing);
    element_init_Zr(sk, pairing);
    mpz_init(N);
    mpz_init(pi);
    mpz_init(e);
    mpz_init(d);
    mpz_init(phi_N);

    if ((fp = fopen("params.txt", "rb")) == NULL) {
        printf("Please execute benchmark_params_ciphers first!\n");
        return STS_ERR;
    }
    fread(g2_buf, g2_len, 1, fp);
    element_from_bytes_compressed(g, g2_buf);
    fread(g2_buf, g2_len, 1, fp);
    element_from_bytes_compressed(pk, g2_buf);
    fread(zr_buf, zr_len, 1, fp);
    element_from_bytes(sk, zr_buf);
    mpz_inp_raw(N, fp);
    mpz_inp_raw(e, fp);
    mpz_inp_raw(d, fp);
    mpz_inp_raw(phi_N, fp);
    fclose(fp);

    strcpy(w, "Henwood");

    cost1 = 0.0;
    printf("Time cost of DEKS Test with T=2^0:(ms)\n");
    catalyst_gen(0, phi_N, pi);
    deks_trapdoor(Tw, sk, pi, N, w);
    if ((fp = fopen("CipherDEKS-0.txt", "rb")) == NULL) {
        printf("Please execute benchmark_params_ciphers first!\n");
        return STS_ERR;
    }
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        fread(g2_buf, g2_len, 1, fp);
        element_from_bytes_compressed(Ca, g2_buf);
        fread(Cb, HASH_LEN, 1, fp);
        deks_test(pairing, Tw, Ca, Cb);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost1 += time_cost(start, end);
            printf("%9.3f\n", cost1);
            gettimeofday(&start, NULL);
        }
    }
    fclose(fp);
    printf("avg-%.3f\n", cost1/1000);

    cost1 = 0.0;
    printf("Time cost of DEKS Test with T=2^12:(ms)\n");
    catalyst_gen(12, phi_N, pi);
    deks_trapdoor(Tw, sk, pi, N, w);
    if ((fp = fopen("CipherDEKS-12.txt", "rb")) == NULL) {
        printf("Please execute benchmark_params_ciphers first!\n");
        return STS_ERR;
    }
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        fread(g2_buf, g2_len, 1, fp);
        element_from_bytes_compressed(Ca, g2_buf);
        fread(Cb, HASH_LEN, 1, fp);
        deks_test(pairing, Tw, Ca, Cb);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost1 += time_cost(start, end);
            printf("%9.3f\n", cost1);
            gettimeofday(&start, NULL);
        }
    }
    fclose(fp);
    printf("avg-%.3f\n", cost1/1000);

    cost1 = 0.0;
    printf("Time cost of DEKS Test with T=2^24:(ms)\n");
    catalyst_gen(24, phi_N, pi);
    deks_trapdoor(Tw, sk, pi, N, w);
    if ((fp = fopen("CipherDEKS-24.txt", "rb")) == NULL) {
        printf("Please execute benchmark_params_ciphers first!\n");
        return STS_ERR;
    }
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        fread(g2_buf, g2_len, 1, fp);
        element_from_bytes_compressed(Ca, g2_buf);
        fread(Cb, HASH_LEN, 1, fp);
        deks_test(pairing, Tw, Ca, Cb);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost1 += time_cost(start, end);
            printf("%9.3f\n", cost1);
            gettimeofday(&start, NULL);
        }
    }
    fclose(fp);
    printf("avg-%.3f\n", cost1/1000);

    cost2 = 0.0;
    printf("Time cost of PEKS Test:(ms)\n");
	peks_trapdoor(Tw, sk, w);
    if ((fp = fopen("CipherPEKS.txt", "rb")) == NULL) {
        printf("Please execute benchmark_params_ciphers first!\n");
        return STS_ERR;
    }
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        fread(g2_buf, g2_len, 1, fp);
        element_from_bytes_compressed(Ca, g2_buf);
        fread(Cb, HASH_LEN, 1, fp);
        peks_test(pairing, Tw, Ca, Cb);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost2 += time_cost(start, end);
            printf("%9.3f\n", cost2);
            gettimeofday(&start, NULL);
        }
    }
    fclose(fp);
    printf("avg-%.3f\n", cost2/1000);

    cost3 = 0.0;
    printf("Time cost of SA-PEKS Test:(ms)\n");
	sa_peks_trapdoor(Tw, sk, N, e, d, w);
    if ((fp = fopen("CipherSAPEKS.txt", "rb")) == NULL) {
        printf("Please execute benchmark_params_ciphers first!\n");
        return STS_ERR;
    }
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        fread(g2_buf, g2_len, 1, fp);
        element_from_bytes_compressed(Ca, g2_buf);
        fread(Cb, HASH_LEN, 1, fp);
        sa_peks_test(pairing, Tw, Ca, Cb);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost3 += time_cost(start, end);
            printf("%9.3f\n", cost3);
            gettimeofday(&start, NULL);
        }
    }
    fclose(fp);
    printf("avg-%.3f\n", cost3/1000);

    element_clear(g);
    element_clear(pk);
    element_clear(sk);
    element_clear(Tw);
    element_clear(Ca);
    mpz_clear(N);
    mpz_clear(pi);
    mpz_clear(e);
    mpz_clear(d);
    mpz_clear(phi_N);
    pairing_clear(pairing);
    return 0;
}