bool test_3(size_t thread_count,size_t loop_peer_thread)
{
const unsigned char test_data[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char key[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
sm::sm4::SM4 sm4(key);
bool status=true;
boost::thread_group tg;
for (size_t i=0;i<thread_count;++i){
tg.add_thread(new boost::thread(crypt_func,&sm4,test_data,16,loop_peer_thread,&status));
}
tg.join_all();
return status;
}
// test memory overwrite
bool test_4()
{
unsigned char test_data[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char key[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char right_e_data[16]={0x68,0x1e,0xdf,0x34,0xd2,0x06,0x96,0x5e,0x86,0xb3,0xe9,0x4f,0x53,0x6e,0x42,0x46};
sm::sm4::sm4_data_writor edata_writor(test_data,16);
sm::sm4::SM4 sm4(key);
// test_data == edata_writor.ptr()
sm4.crypt(sm::sm4::kSM4Encrypt,test_data,16,edata_writor);
bool e_data_ok=::memcmp(edata_writor.ptr(),right_e_data,edata_writor.has_writen())==0;
return e_data_ok;
}
// test 1000000 times coumpting
bool test_2()
{
unsigned char test_data[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char key[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char right_e_data[16]={0x59 ,0x52 ,0x98 ,0xc7 ,0xc6 ,0xfd ,0x27 ,0x1f ,0x04 ,0x02 ,0xf8 ,0x04 ,0xc3 ,0x3d ,0x3f ,0x66};
unsigned char e_data[16];
sm::sm4::SM4 sm4(key);
for (int i=0;i<1000000;++i){
if(i!=0){
memcpy(test_data,e_data,16);
}
sm4.crypt(sm::sm4::kSM4Encrypt,test_data,16,e_data,16);
}
return (::memcmp(e_data,right_e_data,16)==0);
}
// test single round
bool test_1()
{
const unsigned char test_data[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char key[16]={0x01,0x23,0x45,0x67,0x89,0xab,0xcd,0xef,0xfe,0xdc,0xba,0x98,0x76,0x54,0x32,0x10};
const unsigned char right_e_data[16]={0x68,0x1e,0xdf,0x34,0xd2,0x06,0x96,0x5e,0x86,0xb3,0xe9,0x4f,0x53,0x6e,0x42,0x46};
unsigned char c_data[16];
sm::sm4::sm4_crypt_data edata(16);
sm::sm4::sm4_data_writor cdata_writor(c_data,16);
sm::sm4::SM4 sm4(key);
sm4.crypt(sm::sm4::kSM4Encrypt,test_data,16,edata);
sm4.crypt(sm::sm4::kSM4Decrypt,edata.ptr(),edata.bytes(),cdata_writor);
bool c_data_ok=::memcmp(test_data,cdata_writor.ptr(),cdata_writor.has_writen())==0;
bool e_data_ok=::memcmp(edata.ptr(),right_e_data,edata.bytes())==0;
return ( c_data_ok && e_data_ok);
}
int main(int argc, char *argv[])
{
// bench_sort();
int rows = SIZE;
int cols = SIZE;
float density = DENSITY;
EigenSparseMatrix sm1(rows,cols), sm2(rows,cols), sm3(rows,cols), sm4(rows,cols);
BenchTimer timer;
for (int nnzPerCol = NNZPERCOL; nnzPerCol>1; nnzPerCol/=1.1)
{
sm1.setZero();
sm2.setZero();
fillMatrix2(nnzPerCol, rows, cols, sm1);
fillMatrix2(nnzPerCol, rows, cols, sm2);
// std::cerr << "filling OK\n";
// dense matrices
#ifdef DENSEMATRIX
{
std::cout << "Eigen Dense\t" << nnzPerCol << "%\n";
DenseMatrix m1(rows,cols), m2(rows,cols), m3(rows,cols);
eiToDense(sm1, m1);
eiToDense(sm2, m2);
timer.reset();
timer.start();
for (int k=0; k<REPEAT; ++k)
m3 = m1 * m2;
timer.stop();
std::cout << " a * b:\t" << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<REPEAT; ++k)
m3 = m1.transpose() * m2;
timer.stop();
std::cout << " a' * b:\t" << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<REPEAT; ++k)
m3 = m1.transpose() * m2.transpose();
timer.stop();
std::cout << " a' * b':\t" << timer.value() << endl;
timer.reset();
timer.start();
for (int k=0; k<REPEAT; ++k)
m3 = m1 * m2.transpose();
timer.stop();
std::cout << " a * b':\t" << timer.value() << endl;
}
#endif
// eigen sparse matrices
{
std::cout << "Eigen sparse\t" << sm1.nonZeros()/(float(sm1.rows())*float(sm1.cols()))*100 << "% * "
<< sm2.nonZeros()/(float(sm2.rows())*float(sm2.cols()))*100 << "%\n";
BENCH(sm3 = sm1 * sm2; )
std::cout << " a * b:\t" << timer.value() << endl;
// BENCH(sm3 = sm1.transpose() * sm2; )
// std::cout << " a' * b:\t" << timer.value() << endl;
// //
// BENCH(sm3 = sm1.transpose() * sm2.transpose(); )
// std::cout << " a' * b':\t" << timer.value() << endl;
// //
// BENCH(sm3 = sm1 * sm2.transpose(); )
// std::cout << " a * b' :\t" << timer.value() << endl;
// std::cout << "\n";
//
// BENCH( sm3._experimentalNewProduct(sm1, sm2); )
// std::cout << " a * b:\t" << timer.value() << endl;
//
// BENCH(sm3._experimentalNewProduct(sm1.transpose(),sm2); )
// std::cout << " a' * b:\t" << timer.value() << endl;
// //
// BENCH(sm3._experimentalNewProduct(sm1.transpose(),sm2.transpose()); )
// std::cout << " a' * b':\t" << timer.value() << endl;
// //
// BENCH(sm3._experimentalNewProduct(sm1, sm2.transpose());)
// std::cout << " a * b' :\t" << timer.value() << endl;
}
// eigen dyn-sparse matrices
/*{
DynamicSparseMatrix<Scalar> m1(sm1), m2(sm2), m3(sm3);
std::cout << "Eigen dyn-sparse\t" << m1.nonZeros()/(float(m1.rows())*float(m1.cols()))*100 << "% * "
<< m2.nonZeros()/(float(m2.rows())*float(m2.cols()))*100 << "%\n";
// timer.reset();
// timer.start();
BENCH(for (int k=0; k<REPEAT; ++k) m3 = m1 * m2;)
// timer.stop();
std::cout << " a * b:\t" << timer.value() << endl;
// std::cout << sm3 << "\n";
timer.reset();
timer.start();
// std::cerr << "transpose...\n";
// EigenSparseMatrix sm4 = sm1.transpose();
// std::cout << sm4.nonZeros() << " == " << sm1.nonZeros() << "\n";
// exit(1);
// std::cerr << "transpose OK\n";
// std::cout << sm1 << "\n\n" << sm1.transpose() << "\n\n" << sm4.transpose() << "\n\n";
BENCH(for (int k=0; k<REPEAT; ++k) m3 = m1.transpose() * m2;)
// timer.stop();
std::cout << " a' * b:\t" << timer.value() << endl;
// timer.reset();
// timer.start();
BENCH( for (int k=0; k<REPEAT; ++k) m3 = m1.transpose() * m2.transpose(); )
// timer.stop();
std::cout << " a' * b':\t" << timer.value() << endl;
// timer.reset();
// timer.start();
BENCH( for (int k=0; k<REPEAT; ++k) m3 = m1 * m2.transpose(); )
// timer.stop();
std::cout << " a * b' :\t" << timer.value() << endl;
}*/
// CSparse
#ifdef CSPARSE
{
std::cout << "CSparse \t" << nnzPerCol << "%\n";
cs *m1, *m2, *m3;
eiToCSparse(sm1, m1);
eiToCSparse(sm2, m2);
// timer.reset();
// timer.start();
// for (int k=0; k<REPEAT; ++k)
BENCH(
{
m3 = cs_sorted_multiply(m1, m2);
if (!m3)
{
std::cerr << "cs_multiply failed\n";
// break;
}
// cs_print(m3, 0);
cs_spfree(m3);
}
);
// timer.stop();
std::cout << " a * b:\t" << timer.value() << endl;
// BENCH( { m3 = cs_sorted_multiply2(m1, m2); cs_spfree(m3); } );
// std::cout << " a * b:\t" << timer.value() << endl;
}
