/test-definitions/blob/master/auto-test/crypto/crypto.sh

#!/bin/sh

set -x
#进入到目录执行脚本，然后通过cd -返回当前目录
cd ../../utils
. ./sys_info.sh
cd -

#Test user id
#检查是否是root 用户
if [ `whoami` != 'root' ]; then
echo " You must be the superuser to run this script" >&2
exit 1
fi
#distro=`cat /etc/redhat-release | cut -b 1-6`
#如果是centos的话，则分别安装下面几个包
case $distro in
"centos")
yum install gcc -y
yum install gcc-c++ -y
yum install make -y
yum install unzip -y
#下载zip,并解压zip
wget http://192.168.1.107/cryptopp-CRYPTOPP_5_6_5.zip unzip cryptopp-CRYPTOPP_5_6_5.zip
;;
esac
#这段的目的是编译一个so
cd cryptopp-CRYPTOPP_5_6_5
make
make libcryptopp.so
make install
#动态生成一个c++ 文件，
cat << EOF >> ./Cryptopp_test.cc
#include <cryptopp/randpool.h>
#include <cryptopp/rsa.h>
#include <cryptopp/hex.h>
#include <cryptopp/files.h>
#include <iostream>
using namespace std;
using namespace CryptoPP;
#pragma comment(lib, "cryptlib.lib")
//------------------------
// 函数声明
//------------------------
void GenerateRSAKey(unsigned int keyLength, const char *privFilename, const char *pubFilename, const char *seed);
string RSAEncryptString(const char *pubFilename, const char *seed, const char *message);
string RSADecryptString(const char *privFilename, const char *ciphertext);
RandomPool & GlobalRNG();
//------------------------
// 主程序
//------------------------
int main()
{
char priKey[128] = {0};
char pubKey[128] = {0};
char seed[1024] = {0};
// 生成 RSA 密钥对
strcpy(priKey, "pri"); // 生成的私钥文件
strcpy(pubKey, "pub"); // 生成的公钥文件
strcpy(seed, "seed");
GenerateRSAKey(1024, priKey, pubKey, seed);
//RSA 加解密
char message[1024] = {0};
cout<<"Origin Text:\t"<<"just a test!"<<endl<<endl;
strcpy(message, "just a test!");
string encryptedText = RSAEncryptString(pubKey, seed, message); // RSA 加密
cout<<"Encrypted Text:\t"<<encryptedText<<endl<<endl;
string decryptedText = RSADecryptString(priKey, encryptedText.c_str()); // RSA  解密
cout<<"Decrypted Text:\t"<<decryptedText<<endl<<endl;
return 0;
}
//------------------------
//生成 RSA 密钥对
//------------------------
void GenerateRSAKey(unsigned int keyLength, const char *privFilename, const char *pubFilename, const char *seed)
{
RandomPool randPool;
randPool.Put((byte *)seed, strlen(seed));
RSAES_OAEP_SHA_Decryptor priv(randPool, keyLength);
HexEncoder privFile(new FileSink(privFilename));
priv.DEREncode(privFile);
privFile.MessageEnd();
RSAES_OAEP_SHA_Encryptor pub(priv);
HexEncoder pubFile(new FileSink(pubFilename));
pub.DEREncode(pubFile);
pubFile.MessageEnd();
}
//------------------------
// RSA 加密
//------------------------
string RSAEncryptString(const char *pubFilename, const char *seed, const char *message)
{
FileSource pubFile(pubFilename, true, new HexDecoder);
RSAES_OAEP_SHA_Encryptor pub(pubFile);
RandomPool randPool;
randPool.Put((byte *)seed, strlen(seed));
string result;
StringSource(message, true, new PK_EncryptorFilter(randPool, pub, new HexEncoder(new StringSink(result))));
return result;
}
//------------------------
// RSA  解密
//------------------------
string RSADecryptString(const char *privFilename, const char *ciphertext)
{
FileSource privFile(privFilename, true, new HexDecoder);
RSAES_OAEP_SHA_Decryptor priv(privFile);
string result;
StringSource(ciphertext, true, new HexDecoder(new PK_DecryptorFilter(GlobalRNG(), priv, new StringSink(result))));
return result;
}
//------------------------
// 定义全局的随机数池
//------------------------
RandomPool & GlobalRNG()
{
static RandomPool randomPool;
return randomPool;
}
EOF
#编译c++ 程序，这里用到前面编译的libcryptopp
g++ -lcryptopp -lpthread Cryptopp_test.cc -o Cryptopp_test
#更新so后，需要将so的路径加进去，并执行ldconfig这个命令
export LD_LIBRARY_PATH=/lib:$LD_LIBRARY_PATH
sudo ldconfig
#之前前面编译的c++函数
./Cryptopp_test >> crytest.log
#通过飘键保存grep的结果，来检查c++ 是否执行成功
str=`grep -Po "Encrypted Text" crytest.log`
TCID="crypto-policies-test"
if [ "$str" != "" ];then
lava-test-case $TCID --result pass
else
lava-test-case $TCID --result fail
fi