Jiahu235
commented
1 year ago 为了便于排错，我补充了我的相关操作:
- 新建文件夹 "cc/modules/protocol/zk/mystique/tests", 在此文件夹中写了一个名为"relu.cpp" ，代码如下:
#include <string>
#include <unordered_map>
#include <vector>
#include <iostream>
#include <algorithm>
#include "cc/modules/protocol/utility/include/_test_common.h"
#include "cc/modules/protocol/utility/include/version_compat_utils.h"
#include "cc/modules/protocol/zk/mystique/include/mystique_impl.h"
#include "cc/modules/protocol/zk/mystique/include/mystique_ops_impl.h"
#include "cc/modules/protocol/utility/include/util.h"
#include "cc/modules/protocol/zk/mystique/include/wvr_util.h"
#include "cc/modules/protocol/zk/mystique/include/mystique_internal.h"
using namespace emp;
using namespace rosetta;
using namespace std;
int party = 0;
int port = 48000;
void *wvr_impl;
std::string node_id;
std::string config_json;
std::string task_id = "1";
std::string msgID = "1";
std::string protocol_name("mystique");
rosetta::MystiqueProtocol* myProt;
rosetta::MystiqueProtocolFactory* myProtFact;
attr_type info = {
{"m", 0}, {"k", 0}, {"n", 0}, {"transpose_a", 0}, {"transpose_b", 0}, {"rh_is_const", 0}
};
attr_type* attr_info = &info;
msg_id_t msg_id(msgID.c_str(), msgID.size());
shared_ptr<ProtocolContext> context = myProt->GetMpcContext();
rosetta::MystiqueOpsImpl* myOpImpl = new rosetta::MystiqueOpsImpl(msg_id, context);
ZkIntFp ZkZero = ZkIntFp(int64_t(0), party);
void Reveal(std::vector<ZkIntFp>& inputArr, std::vector<int64_t>& expected){
int64_t size = inputArr.size();
// for (int64_t i = 0; i < size; i++){
// inputArr[i].reveal(expected[i]);
// }
ZkIntFp* input = new ZkIntFp[size];
for (int64_t i = 0; i < size; i++) input[i] = inputArr[i];
ZkIntFp tmp;
tmp.batch_reveal(input, size);
}
void test_Relu()
{
int64_t size = 1*32*32*16;
std::vector<ZkIntFp> input(size, ZkIntFp(int64_t(-1), party));
vector<ZkIntFp> output(size, ZkIntFp(int64_t(1), party));
vector<int64_t> expected(size, 0);
std::vector<string> inputArr(size);
std::vector<string> outputArr(size);
convert_mac_to_string(input, inputArr, true, 1);
myOpImpl->Relu(inputArr, outputArr, attr_info);
convert_string_to_mac(outputArr, output, true, wvr_impl);
Reveal(output, expected);
}
int main(int argc, char** argv)
{
emp::parse_party_and_port(argv, &party, &port);
string logfile = "cc/modules/protocol/zk/mystique/log/relu-" + to_string(party);
Logger::Get().log_to_stdout(false);
Logger::Get().set_filename(logfile + "-backend.log");
Logger::Get().set_level(0);
rosetta_old_conf_parse(node_id, config_json, party, "cc/conf/zk-CONFIG.json");
myProt = new rosetta::MystiqueProtocol(task_id);
IOManager::Instance()->CreateChannel(task_id, node_id, config_json);
myProt->Init(logfile + "-console.log");
shared_ptr<NET_IO> net_io = myProt->GetNetHandler();
string node_id_0 = net_io->GetNodeId(0);
string node_id_1 = net_io->GetNodeId(1);
vector<string> reveal_receivers = {"P0", "P1"};
rosetta::attr_type reveal_attr;
reveal_attr["receive_parties"] = receiver_parties_pack(reveal_receivers);
test_Relu();
myProt->Uninit();
return 0;
}- 对"cc/modules/protocol/zk/mystique/CMakeLists.txt"进行了修改，在文件末尾增加了一段代码：
function(compile_ex_zkp category)
file(GLOB EXAMPLE_SOURCE_FILES RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}/${category}" "${category}/*.cpp")
foreach(EXAMPLE_SOURCE_FILE ${EXAMPLE_SOURCE_FILES})
string(REGEX REPLACE "(.*)\\.cpp" "\\1" EXAMPLE_NAME ${EXAMPLE_SOURCE_FILE})
set(EXAMPLE_TARGET "protocol_zkp_mystique_${category}_${EXAMPLE_NAME}")
add_executable(${EXAMPLE_TARGET} ${category}/${EXAMPLE_SOURCE_FILE})
target_link_libraries(${EXAMPLE_TARGET} zkp-mystique protocol-utility ${EMP-TOOL_LIBRARIES} ${EMP-OT_LIBRARIES} ${EMP-ZK_LIBRARIES})
endforeach()
endfunction()
compile_ex_zkp(tests)- 编译命令：
./rosetta.sh compile --phase=modules --enable-protocol-zk
joyoFeng
wercsrsr
rosetta中没有zk(mystique)协议的测试文件，我们想测试基于zk(mystique)协议的操作，例如relu操作 测试文件编译时报错：首先，测试文件中并没有调用与CheatRecord相关的内容；其次，'CheatRecord::put(std::string const&)' 和'CheatRecord::message' 在'emp-zk-bool.cpp'文件中有定义。报错的可能原因是什么？