runc/syncpipe/sync_pipe.go

package syncpipe

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
	"syscall"
)

// SyncPipe allows communication to and from the child processes
// to it's parent and allows the two independent processes to
// syncronize their state.
type SyncPipe struct {
	parent, child *os.File
}

func NewSyncPipeFromFd(parentFd, childFd uintptr) (*SyncPipe, error) {
	s := &SyncPipe{}

	if parentFd > 0 {
		s.parent = os.NewFile(parentFd, "parentPipe")
	} else if childFd > 0 {
		s.child = os.NewFile(childFd, "childPipe")
	} else {
		return nil, fmt.Errorf("no valid sync pipe fd specified")
	}

	return s, nil
}

func (s *SyncPipe) Child() *os.File {
	return s.child
}

func (s *SyncPipe) Parent() *os.File {
	return s.parent
}

func (s *SyncPipe) SendToChild(v interface{}) error {
	data, err := json.Marshal(v)
	if err != nil {
		return err
	}

	s.parent.Write(data)

	return syscall.Shutdown(int(s.parent.Fd()), syscall.SHUT_WR)
}

func (s *SyncPipe) ReadFromChild() error {
	data, err := ioutil.ReadAll(s.parent)
	if err != nil {
		return err
	}

	if len(data) > 0 {
		return fmt.Errorf("%s", data)
	}

	return nil
}

func (s *SyncPipe) ReadFromParent(v interface{}) error {
	data, err := ioutil.ReadAll(s.child)
	if err != nil {
		return fmt.Errorf("error reading from sync pipe %s", err)
	}

	if len(data) > 0 {
		if err := json.Unmarshal(data, v); err != nil {
			return err
		}
	}

	return nil
}

func (s *SyncPipe) ReportChildError(err error) {
	// ensure that any data sent from the parent is consumed so it doesn't
	// receive ECONNRESET when the child writes to the pipe.
	ioutil.ReadAll(s.child)

	s.child.Write([]byte(err.Error()))
	s.CloseChild()
}

func (s *SyncPipe) Close() error {
	if s.parent != nil {
		s.parent.Close()
	}

	if s.child != nil {
		s.child.Close()
	}

	return nil
}

func (s *SyncPipe) CloseChild() {
	if s.child != nil {
		s.child.Close()
		s.child = nil
	}
}