Skip to content

Commit f56656a

Browse files
authored
Merge f9fffcf into 791558d
2 parents 791558d + f9fffcf commit f56656a

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

43 files changed

+117
-28
lines changed

ydb/core/client/server/msgbus_blobstorage_config.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ class TMessageBusBlobStorageConfig
2727
{
2828
SetSecurityToken(token);
2929
SetRequireAdminAccess(true);
30+
SetPeerName(msg.GetPeerName());
3031
}
3132

3233
void Handle(TEvBlobStorage::TEvControllerConfigResponse::TPtr &ev, const TActorContext &ctx) {

ydb/core/client/server/msgbus_server.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ class TBusMessageContext::TImpl : public TThrRefBase {
1616
virtual void SendReplyMove(NBus::TBusMessageAutoPtr response) = 0;
1717
virtual TVector<TStringBuf> FindClientCert() const = 0;
1818
virtual THolder<TMessageBusSessionIdentHolder::TImpl> CreateSessionIdentHolder() = 0;
19+
virtual TString GetPeerName() const = 0;
1920
};
2021

2122
class TBusMessageContext::TImplMessageBus
@@ -61,6 +62,9 @@ class TBusMessageContext::TImplMessageBus
6162
return {};
6263
}
6364

65+
TString GetPeerName() const override {
66+
return TStringBuilder() << GetPeerAddrNetAddr();
67+
}
6468

6569
THolder<TMessageBusSessionIdentHolder::TImpl> CreateSessionIdentHolder() override;
6670
};
@@ -175,6 +179,10 @@ class TBusMessageContext::TImplGRpc
175179
};
176180

177181
THolder<TMessageBusSessionIdentHolder::TImpl> CreateSessionIdentHolder() override;
182+
183+
TString GetPeerName() const override {
184+
return RequestContext->GetPeer();
185+
}
178186
};
179187

180188
TBusMessageContext::TBusMessageContext()
@@ -221,6 +229,8 @@ void TBusMessageContext::Swap(TBusMessageContext &msg) {
221229

222230
TVector<TStringBuf> TBusMessageContext::FindClientCert() const { return Impl->FindClientCert(); }
223231

232+
TString TBusMessageContext::GetPeerName() const { return Impl->GetPeerName(); }
233+
224234
THolder<TMessageBusSessionIdentHolder::TImpl> TBusMessageContext::CreateSessionIdentHolder() {
225235
Y_ABORT_UNLESS(Impl);
226236
return Impl->CreateSessionIdentHolder();

ydb/core/client/server/msgbus_server.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ class TBusMessageContext {
8787
void SendReplyMove(NBus::TBusMessageAutoPtr response);
8888
void Swap(TBusMessageContext& msg);
8989
TVector<TStringBuf> FindClientCert() const;
90+
TString GetPeerName() const;
9091

9192
private:
9293
friend class TMessageBusSessionIdentHolder;

ydb/core/client/server/msgbus_server_cms.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ class TCmsRequestActor : public TMessageBusSecureRequest<TMessageBusServerReques
3131
, Request(request)
3232
{
3333
TBase::SetSecurityToken(request.GetSecurityToken());
34+
TBase::SetPeerName(msg.GetPeerName());
3435
}
3536

3637
void Bootstrap(const TActorContext &ctx)

ydb/core/client/server/msgbus_server_console.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ class TConsoleRequestActor : public TMessageBusSecureRequest<TMessageBusServerRe
4444
TBase::SetRequireAdminAccess(true);
4545
}
4646

47+
TBase::SetPeerName(msg.GetPeerName());
4748
}
4849

4950
void Bootstrap(const TActorContext &ctx)

ydb/core/client/server/msgbus_server_drain_node.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ class TMessageBusDrainNode : public TMessageBusSecureRequest<TMessageBusTabletRe
2020
{
2121
SetSecurityToken(Request->Record.GetSecurityToken());
2222
SetRequireAdminAccess(true);
23+
SetPeerName(msg.GetPeerName());
2324
}
2425

2526
std::pair<ui64, TAutoPtr<IEventBase>> MakeReqPair(const TActorContext& ctx) {

ydb/core/client/server/msgbus_server_fill_node.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ class TMessageBusFillNode : public TMessageBusSecureRequest<TMessageBusTabletReq
2020
{
2121
SetSecurityToken(Request->Record.GetSecurityToken());
2222
SetRequireAdminAccess(true);
23+
SetPeerName(msg.GetPeerName());
2324
}
2425

2526
std::pair<ui64, TAutoPtr<IEventBase>> MakeReqPair(const TActorContext& ctx) {

ydb/core/client/server/msgbus_server_local_minikql.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ class TMessageBusLocalMKQL : public TMessageBusSecureRequest<TMessageBusSimpleTa
1717
{
1818
SetSecurityToken(static_cast<TBusTabletLocalMKQL*>(msg.GetMessage())->Record.GetSecurityToken());
1919
SetRequireAdminAccess(true);
20+
SetPeerName(msg.GetPeerName());
2021
}
2122

2223
void Handle(TEvTablet::TEvLocalMKQLResponse::TPtr &ev, const TActorContext &ctx) {

ydb/core/client/server/msgbus_server_local_scheme_tx.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ class TMessageBusLocalSchemeTx : public TMessageBusSecureRequest<TMessageBusSimp
2121
Request.Swap(&request);
2222
TBase::SetSecurityToken(Request.GetSecurityToken());
2323
TBase::SetRequireAdminAccess(true);
24+
TBase::SetPeerName(msg.GetPeerName());
2425
}
2526

2627
void Handle(TEvTablet::TEvLocalSchemeTxResponse::TPtr &ev, const TActorContext &ctx) {

ydb/core/client/server/msgbus_server_node_registration.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ class TNodeRegistrationActor : public TMessageBusSecureRequest<TMessageBusServer
4040
} else {
4141
TBase::SetSecurityToken(BUILTIN_ACL_ROOT); // NBS compatibility
4242
}
43+
TBase::SetPeerName(msg.GetPeerName());
4344
}
4445

4546
void Bootstrap(const TActorContext &ctx)

0 commit comments

Comments
 (0)