Method Serve

agent/reconnectingpty/server.go:65–139

(ctx, hardCtx context.Context, l net.Listener)

}

func (s *Server) Serve(ctx, hardCtx context.Context, l net.Listener) (retErr error) {
	var wg sync.WaitGroup
	for {
		if ctx.Err() != nil {
			break
		}
		conn, err := l.Accept()
		if err != nil {
			s.logger.Debug(ctx, "accept pty failed", slog.Error(err))
			retErr = err
			break
		}
		clog := s.logger.With(
			slog.F("remote", conn.RemoteAddr()),
			slog.F("local", conn.LocalAddr()))
		clog.Info(ctx, "accepted conn")

		// It's not safe to assume RemoteAddr() returns a non-nil value. slog.F usage is fine because it correctly
		// handles nil.
		// c.f. https://github.com/coder/internal/issues/1143
		remoteAddr := conn.RemoteAddr()
		remoteAddrString := ""
		if remoteAddr != nil {
			remoteAddrString = remoteAddr.String()
		}

		wg.Add(1)
		disconnected := s.reportConnection(uuid.New(), remoteAddrString)
		closed := make(chan struct{})
		go func() {
			defer wg.Done()
			select {
			case <-closed:
			case <-hardCtx.Done():
				clog.Info(hardCtx, "reconnecting pty closed",
					codersdk.ConnectionDirectionAgentToClient.SlogField(),
					codersdk.DisconnectReasonServerShutdown.SlogField(),
					codersdk.DisconnectReasonServerShutdown.SlogExpectedField(),
				)
				disconnected(1, "server shut down")
				_ = conn.Close()
			}
		}()
		wg.Add(1)
		go func() {
			defer close(closed)
			defer wg.Done()
			err := s.handleConn(ctx, clog, conn)
			var reason codersdk.DisconnectReason
			var code int
			var detail string
			switch {
			case err != nil && ctx.Err() != nil:
				reason = codersdk.DisconnectReasonServerShutdown
				code = 1
			case err != nil:
				reason = codersdk.DisconnectReasonNetworkError
				detail = err.Error()