lywsvip/cloudpods
1
0
Fork 0
mirror of https://github.com/yunionio/cloudpods.git synced 2026-05-06 21:52:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

cloudproxy: agent: fix possible invalid memory access

Browse Source
This commit is contained in:
Yousong Zhou
2021-04-02 17:26:59 +08:00
parent 8ee5978185
commit d57e2d2495
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/cloudproxy/agent/ssh/clientset.go
View File

@@ -70,6 +70,7 @@ func (cs *ClientSet) ResetIfChanged(ctx context.Context, epKey string, cc ssh_ut
if epcs.cc != cc {
epcs.stop(ctx)
delete(cs.epClients, epKey)
cs.AddIfNotExist(ctx, epKey, cc)
return true
}
epcs.setMark()
@@ -172,7 +173,7 @@ func (cs *ClientSet) getClient_(epKey string, typ string, create bool) (*Client,
clients, ok := cs.epClients[epKey]
if !ok || len(clients.clients) == 0 {
if !create {
if !ok || !create {
return nil, false
}
client = NewClient(&clients.cc)