Function ReceiveData

helper/tunnel/server.go:90–141

(stream remote.Tunnel_InitTunnelServer, closeChan chan struct{})

}

func ReceiveData(stream remote.Tunnel_InitTunnelServer, closeChan chan struct{}) {
	for {
		select {
		case <-closeChan:
			return
		case <-stream.Context().Done():
			return
		default:
			message, err := stream.Recv()
			if err != nil {
				stderrlog.Errorf("failed receiving message from stream, exiting: %v", err)
				close(closeChan)
				continue
			}

			reqID, err := uuid.Parse(message.GetRequestId())
			if err != nil {
				stderrlog.Errorf(" %s; failed to parse requestId, %v", message.GetRequestId(), err)
				continue
			}

			session, ok := GetSession(reqID)
			if !ok && !message.ShouldClose {
				stderrlog.Errorf("%s; session not found in openRequests", reqID)
				continue
			}

			data := message.GetData()
			br := len(data)

			stderrlog.Debugf("received %d bytes from client", len(data))

			// send data if we received any
			if br > 0 && session.Open {
				stderrlog.Debugf("writing %d bytes to conn", br)
				_, err := session.Conn.Write(data)
				if err != nil {
					stderrlog.Errorf("%s; failed writing data to socket", reqID)
					message.ShouldClose = true
				} else {
					stderrlog.Debugf("wrote %d bytes to conn", br)
				}
			}

			if message.ShouldClose {
				stderrlog.Debugf("closing session")
				session.Close()
				stderrlog.Debugf("closed session")
			}
		}
	}
}

func readConn(ctx context.Context, session *Session, sessions chan<- *Session) {
	for {