plugins: don't panic on Close if PluginServer nil

Signed-off-by: Laura Brehm <laurabrehm@hey.com>
(cherry picked from commit 9c4480604e)
Signed-off-by: Paweł Gronowski <pawel.gronowski@docker.com>
This commit is contained in:
Laura Brehm 2024-08-07 14:04:31 +01:00 committed by Paweł Gronowski
parent 30c7951192
commit 984ef9072c
No known key found for this signature in database
GPG Key ID: B85EFCFE26DEF92A
2 changed files with 15 additions and 0 deletions

View File

@ -95,6 +95,9 @@ func (pl *PluginServer) Addr() net.Addr {
// //
// The error value is that of the underlying [net.Listner.Close] call. // The error value is that of the underlying [net.Listner.Close] call.
func (pl *PluginServer) Close() error { func (pl *PluginServer) Close() error {
if pl == nil {
return nil
}
logrus.Trace("Closing plugin server") logrus.Trace("Closing plugin server")
// Close connections first to ensure the connections get io.EOF instead // Close connections first to ensure the connections get io.EOF instead
// of a connection reset. // of a connection reset.

View File

@ -117,6 +117,18 @@ func TestPluginServer(t *testing.T) {
assert.NilError(t, err, "failed to dial returned server") assert.NilError(t, err, "failed to dial returned server")
checkDirNoNewPluginServer(t) checkDirNoNewPluginServer(t)
}) })
t.Run("does not panic on Close if server is nil", func(t *testing.T) {
var srv *PluginServer
defer func() {
if r := recover(); r != nil {
t.Errorf("panicked on Close")
}
}()
err := srv.Close()
assert.NilError(t, err)
})
} }
func checkDirNoNewPluginServer(t *testing.T) { func checkDirNoNewPluginServer(t *testing.T) {