Merge pull request #5051 from wzshiming/fix/missing-close

Fix missing close
This commit is contained in:
Derek McGowan 2021-02-26 14:59:43 -08:00 committed by GitHub
commit 10bbd1a462
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -126,7 +126,7 @@ func newCommand(ctx context.Context, id, containerdBinary, containerdAddress, co
return cmd, nil return cmd, nil
} }
func (s *service) StartShim(ctx context.Context, id, containerdBinary, containerdAddress, containerdTTRPCAddress string) (string, error) { func (s *service) StartShim(ctx context.Context, id, containerdBinary, containerdAddress, containerdTTRPCAddress string) (_ string, retErr error) {
cmd, err := newCommand(ctx, id, containerdBinary, containerdAddress, containerdTTRPCAddress) cmd, err := newCommand(ctx, id, containerdBinary, containerdAddress, containerdTTRPCAddress)
if err != nil { if err != nil {
return "", err return "", err
@ -147,6 +147,12 @@ func (s *service) StartShim(ctx context.Context, id, containerdBinary, container
return "", err return "", err
} }
} }
defer func() {
if retErr != nil {
socket.Close()
_ = shim.RemoveSocket(address)
}
}()
f, err := socket.File() f, err := socket.File()
if err != nil { if err != nil {
return "", err return "", err
@ -155,11 +161,11 @@ func (s *service) StartShim(ctx context.Context, id, containerdBinary, container
cmd.ExtraFiles = append(cmd.ExtraFiles, f) cmd.ExtraFiles = append(cmd.ExtraFiles, f)
if err := cmd.Start(); err != nil { if err := cmd.Start(); err != nil {
f.Close()
return "", err return "", err
} }
defer func() { defer func() {
if err != nil { if retErr != nil {
_ = shim.RemoveSocket(address)
cmd.Process.Kill() cmd.Process.Kill()
} }
}() }()