sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-10-25 22:09:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

add semaphore on grpc

Browse Source
This commit is contained in:
Pascal Fischer
2025-05-07 18:06:10 +02:00
parent b43def9164
commit 7b8929e8ba
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
management/server/grpcserver.go
+10
View File
@@ -444,6 +444,16 @@ func (s *GRPCServer) Login(ctx context.Context, req *proto.EncryptedMessage) (*p
//nolint
ctx = context.WithValue(ctx, nbContext.AccountIDKey, accountID)
err = TryAcquire(loginReq.GetPeerKeys().String())
if err != nil {
log.WithContext(ctx).Debugf("error while acquiring grpc semaphore for %s: %v", loginReq.GetPeerKeys().String(), err)
return nil, err
}
defer func() {
Release(loginReq.GetPeerKeys().String())
}()
if loginReq.GetMeta() == nil {
msg := status.Errorf(codes.FailedPrecondition,
"peer system meta has to be provided to log in. Peer %s, remote addr %s", peerKey.String(), realIP)