teems/Enclave/OblivAlgs/ORExpand.cpp

#include "ORExpand.hpp"

#ifdef TEST_OREXPAND
void test_ORExpand()
{
    size_t block_size = 48;
    uint32_t N = 16;

    uint32_t dest_specified[] = {0, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14};
    size_t dest_specified_len = sizeof(dest_specified)/sizeof(uint32_t);
    unsigned char *buf = new unsigned char[N*block_size];
    uint32_t *dest = new uint32_t[N];
    for (size_t i=0;i<dest_specified_len;++i) {
        dest[i] = dest_specified[i];
    }
    for (size_t i=dest_specified_len;i<N;++i) {
        dest[i] = 0xffffffff;
    }

    for (size_t i=0;i<N;++i) {
        for (size_t j=0;j<block_size;j+=2) {
            buf[i*block_size+j] = (unsigned char)i;
            buf[i*block_size+j+1] = (unsigned char)j/2;
        }
    }

    ORExpand<OSWAP_16X>(buf, dest, block_size, N);

    for(size_t i=0;i<N;++i) {
        printf("%2d ", i);
        if (dest[i] == 0xffffffff) {
            printf("PD ");
        } else {
            printf("%2d ", dest[i]);
        }
        for (size_t j=0;j<block_size;++j) {
            printf("%02x", buf[i*block_size+j]);
        }
        printf("\n");
    }

    delete[] buf;
    delete[] dest;
}
#endif