github/kata-containers
1
0
Fork 1
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-04-29 12:14:48 +00:00
Code Issues Projects Releases Wiki Activity

runtime: Provide protection for shared data

Browse Source 
The k.reqHandlers should be protected by locks when used

Fixes #3440

Signed-off-by: liangxianlong <liang.xianlong@zte.com.cn>
This commit is contained in:
liangxianlong 2022-01-13 14:48:10 +08:00
parent ef835b5948
commit 878ab93c15
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/runtime/virtcontainers/kata_agent.go
View File

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