Fix fd leak of shim log

darfux · darfux · commit cf6e0085423a · 2019-05-09T20:21:57.000+08:00
Open shim v2 log with the flag `O_RDWR` will cause the `Read()` block
forever even if the pipe has been closed on the shim side. Then the
`io.Copy()` would never return and lead to a fd leak.
Fix typo when closing shim v1 log which causes the `stdouLog` leak.
Update `numPipes` function in test case to get the opened FIFO
correctly.

Signed-off-by: Li Yuxuan &lt;liyuxuan04@baidu.com&gt;
diff --git a/container_linux_test.go b/container_linux_test.go
@@ -330,7 +330,7 @@ func TestShimDoesNotLeakPipes(t *testing.T) {
 }
 
 func numPipes(pid int) (int, error) {
-	cmd := exec.Command("sh", "-c", fmt.Sprintf("lsof -p %d | grep pipe", pid))
+	cmd := exec.Command("sh", "-c", fmt.Sprintf("lsof -p %d | grep FIFO", pid))
 
 	var stdout bytes.Buffer
 	cmd.Stdout = &stdout
diff --git a/runtime/v1/shim/client/client.go b/runtime/v1/shim/client/client.go
@@ -98,9 +98,9 @@ func WithStart(binary, address, daemonAddress, cgroup string, debug bool, exitHa
 			cmd.Wait()
 			exitHandler()
 			if stdoutLog != nil {
-				stderrLog.Close()
+				stdoutLog.Close()
 			}
-			if stdoutLog != nil {
+			if stderrLog != nil {
 				stderrLog.Close()
 			}
 		}()
diff --git a/runtime/v2/shim_unix.go b/runtime/v2/shim_unix.go
@@ -28,5 +28,5 @@ import (
 )
 
 func openShimLog(ctx context.Context, bundle *Bundle) (io.ReadCloser, error) {
-	return fifo.OpenFifo(ctx, filepath.Join(bundle.Path, "log"), unix.O_RDWR|unix.O_CREAT, 0700)
+	return fifo.OpenFifo(ctx, filepath.Join(bundle.Path, "log"), unix.O_RDONLY|unix.O_CREAT|unix.O_NONBLOCK, 0700)
 }

Original file line number	Diff line number	Diff line change
`@@ -330,7 +330,7 @@ func TestShimDoesNotLeakPipes(t *testing.T) {`
`330`	`330`	`}`
`331`	`331`
`332`	`332`	`func numPipes(pid int) (int, error) {`
`333`		`- cmd := exec.Command("sh", "-c", fmt.Sprintf("lsof -p %d \| grep pipe", pid))`
	`333`	`+ cmd := exec.Command("sh", "-c", fmt.Sprintf("lsof -p %d \| grep FIFO", pid))`
`334`	`334`
`335`	`335`	`var stdout bytes.Buffer`
`336`	`336`	`cmd.Stdout = &stdout`
Original file line number	Diff line number	Diff line change
`@@ -98,9 +98,9 @@ func WithStart(binary, address, daemonAddress, cgroup string, debug bool, exitHa`
`98`	`98`	`cmd.Wait()`
`99`	`99`	`exitHandler()`
`100`	`100`	`if stdoutLog != nil {`
`101`		`- stderrLog.Close()`
	`101`	`+ stdoutLog.Close()`
`102`	`102`	`}`
`103`		`- if stdoutLog != nil {`
	`103`	`+ if stderrLog != nil {`
`104`	`104`	`stderrLog.Close()`
`105`	`105`	`}`
`106`	`106`	`}()`
Original file line number	Diff line number	Diff line change
`@@ -28,5 +28,5 @@ import (`
`28`	`28`	`)`
`29`	`29`
`30`	`30`	`func openShimLog(ctx context.Context, bundle *Bundle) (io.ReadCloser, error) {`
`31`		`- return fifo.OpenFifo(ctx, filepath.Join(bundle.Path, "log"), unix.O_RDWR\|unix.O_CREAT, 0700)`
	`31`	`+ return fifo.OpenFifo(ctx, filepath.Join(bundle.Path, "log"), unix.O_RDONLY\|unix.O_CREAT\|unix.O_NONBLOCK, 0700)`
`32`	`32`	`}`