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

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 {