Function HandleStdIOClient

pkg/wshutil/wshutil.go:280–325

blocking, returns if there is an error, or on EOF of input

(logName string, input chan utilfn.LineOutput, output io.Writer)

// blocking, returns if there is an error, or on EOF of input
func HandleStdIOClient(logName string, input chan utilfn.LineOutput, output io.Writer) {
	proxy := MakeRpcProxy(logName)
	linkId := DefaultRouter.RegisterTrustedRouter(proxy)
	rawCh := make(chan []byte, DefaultInputChSize)
	go func() {
		defer func() {
			panichandler.PanicHandler("HandleStdIOClient:ParseWithLinesChan", recover())
		}()
		packetparser.ParseWithLinesChan(input, proxy.FromRemoteCh, rawCh)
	}()
	doneCh := make(chan struct{})
	var doneOnce sync.Once
	closeDoneCh := func() {
		doneOnce.Do(func() {
			close(doneCh)
			DefaultRouter.UnregisterLink(linkId)
			close(proxy.FromRemoteCh)
		})
	}
	go func() {
		defer func() {
			panichandler.PanicHandler("HandleStdIOClient:ToRemoteChLoop", recover())
		}()
		defer closeDoneCh()
		for msg := range proxy.ToRemoteCh {
			err := packetparser.WritePacket(output, msg)
			if err != nil {
				log.Printf("[%s] error writing to output: %v\n", logName, err)
				break
			}
		}
	}()
	go func() {
		defer func() {
			panichandler.PanicHandler("HandleStdIOClient:RawChLoop", recover())
		}()
		defer closeDoneCh()
		for msg := range rawCh {
			if !bytes.HasSuffix(msg, []byte{'\n'}) {
				msg = append(msg, '\n')
			}
			log.Printf("[%s:stdout] %s", logName, msg)
		}
	}()
	<-doneCh
}

func handleDomainSocketClient(conn net.Conn, readCallback func()) {
	var linkIdContainer atomic.Int32