Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[fed] Fixes for the encrypted GRPC backend. (#10503)

Browse Source
This commit is contained in:
Jiaming Yuan
2024-07-02 15:15:12 +08:00
committed by GitHub
parent 5f0c1e902b
commit a39fef2c67
9 changed files with 192 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
plugin/federated/federated_comm.cc
View File

@@ -1,5 +1,5 @@
/**
* Copyright 2023, XGBoost contributors
* Copyright 2023-2024, XGBoost contributors
*/
#include "federated_comm.h"
@@ -11,6 +11,7 @@
#include <string> // for string, stoi
#include "../../src/common/common.h" // for Split
#include "../../src/common/io.h" // for ReadAll
#include "../../src/common/json_utils.h" // for OptionalArg
#include "xgboost/json.h" // for Json
#include "xgboost/logging.h"
@@ -46,9 +47,9 @@ void FederatedComm::Init(std::string const& host, std::int32_t port, std::int32_
} else {
stub_ = [&] {
grpc::SslCredentialsOptions options;
options.pem_root_certs = server_cert;
options.pem_private_key = client_key;
options.pem_cert_chain = client_cert;
options.pem_root_certs = common::ReadAll(server_cert);
options.pem_private_key = common::ReadAll(client_key);
options.pem_cert_chain = common::ReadAll(client_cert);
grpc::ChannelArguments args;
args.SetMaxReceiveMessageSize(std::numeric_limits<std::int32_t>::max());
auto channel = grpc::CreateCustomChannel(host + ":" + std::to_string(port),