Merge pull request #3441 from liangxianlong/main

runtime: Provide protection for shared data
This commit is contained in:
James O. D. Hunt 2022-01-13 17:46:35 +00:00 committed by GitHub
commit 6387a3d323
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2044,10 +2044,20 @@ func (k *kataAgent) sendReq(spanCtx context.Context, request interface{}) (inter
}
msgName := proto.MessageName(request.(proto.Message))
k.Lock()
if k.reqHandlers == nil {
return nil, errors.New("Client has already disconnected")
}
handler := k.reqHandlers[msgName]
if msgName == "" || handler == nil {
return nil, errors.New("Invalid request type")
}
k.Unlock()
message := request.(proto.Message)
ctx, cancel := k.getReqContext(spanCtx, msgName)
if cancel != nil {