benchmark_trapgen.c

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

char dataset[1024][64];

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;
    mpz_t N, pi, e, d, phi_N;
    int g2_len, zr_len;
    FILE *fp;
    char * g2_buf, * zr_buf;
    char w[64];
    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(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("keyword.txt", "r")) == NULL) {
        printf("Please provide at least 1000 keywords in a file named \'keyword.txt\'.\n");
        return STS_ERR;
    }
    for(int i = 0; i < 1000; i++) {
        if(fscanf(fp, "%s", w) != EOF) {
            strcpy(dataset[i], w);
        }else {
            printf("Error!\n");
            break;
        }
    }
    fclose(fp);

    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);
    
    cost1 = 0.0;
    printf("Time cost of DEKS TrapGen with T=2^0:(ms)\n");
    catalyst_gen(0, phi_N, pi);
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        deks_trapdoor(Tw, sk, pi, N, dataset[i]);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost1 += time_cost(start, end);
            printf("%7.3f\n", cost1);
            gettimeofday(&start, NULL);
        }
    }
    printf("avg-%.3f\n", cost1/1000);

    cost1 = 0.0;
    printf("Time cost of DEKS TrapGen with T=2^12:(ms)\n");
    catalyst_gen(12, phi_N, pi);
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        deks_trapdoor(Tw, sk, pi, N, dataset[i]);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost1 += time_cost(start, end);
            printf("%9.3f\n", cost1);
            gettimeofday(&start, NULL);
        }
    }
    printf("avg-%.3f\n", cost1/1000);

    cost1 = 0.0;
    printf("Time cost of DEKS TrapGen with T=2^24:(ms)\n");
    catalyst_gen(24, phi_N, pi);
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        deks_trapdoor(Tw, sk, pi, N, dataset[i]);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost1 += time_cost(start, end);
            printf("%9.3f\n", cost1);
            gettimeofday(&start, NULL);
        }
    }
    printf("avg-%.3f\n", cost1/1000);

    cost2 = 0.0;
    printf("Time cost of PEKS TrapGen:(ms)\n");
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        peks_trapdoor(Tw, sk, dataset[i]);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost2 += time_cost(start, end);
            printf("%7.3f\n", cost2);
            gettimeofday(&start, NULL);
        }
    }
    printf("avg-%.3f\n", cost2/1000);

    cost3 = 0.0;
    printf("Time cost of SA-PEKS TrapGen:(ms)\n");
    gettimeofday(&start, NULL);
    for(int i = 0; i < 1000; i++) {
        sa_peks_trapdoor(Tw, sk, N, e, d, dataset[i]);
        if(i % 50 == 49) {
            gettimeofday(&end, NULL);
            cost3 += time_cost(start, end);
            printf("%9.3f\n", cost3);
            gettimeofday(&start, NULL);
        }
    }
    printf("avg-%.3f\n", cost3/1000);

    printf("Time cost of generating catalyst with various T:(us)\n");
    for (int i = 0; i < 29; ++i) {
        gettimeofday(&start, NULL);
        for (int j = 0; j < 1000; ++j) {
            catalyst_gen(i, phi_N, pi);
        }
        gettimeofday(&end, NULL);
        cost1 = time_cost(start, end);
        printf("%2d\t%6.3f\n", i, cost1);
    }

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

    return STS_OK;
}