mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
44 lines
947 B
Go
44 lines
947 B
Go
package libcontainerd
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
|
|
"github.com/Microsoft/hcsshim"
|
|
"github.com/docker/docker/pkg/ioutils"
|
|
)
|
|
|
|
type autoClosingReader struct {
|
|
io.ReadCloser
|
|
sync.Once
|
|
}
|
|
|
|
func (r *autoClosingReader) Read(b []byte) (n int, err error) {
|
|
n, err = r.ReadCloser.Read(b)
|
|
if err != nil {
|
|
r.Once.Do(func() { r.ReadCloser.Close() })
|
|
}
|
|
return
|
|
}
|
|
|
|
func createStdInCloser(pipe io.WriteCloser, process hcsshim.Process) io.WriteCloser {
|
|
return ioutils.NewWriteCloserWrapper(pipe, func() error {
|
|
if err := pipe.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
err := process.CloseStdin()
|
|
if err != nil && !hcsshim.IsNotExist(err) && !hcsshim.IsAlreadyClosed(err) {
|
|
// This error will occur if the compute system is currently shutting down
|
|
if perr, ok := err.(*hcsshim.ProcessError); ok && perr.Err != hcsshim.ErrVmcomputeOperationInvalidState {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (p *process) Cleanup() error {
|
|
return nil
|
|
}
|