Say whether an approval was refused, unanswered, or never shown

This commit is contained in:
Viktor Liu
2026-08-31 15:15:37 +02:00
parent 3af7764b09
commit bad63c14b8
5 changed files with 200 additions and 10 deletions
+19 -1
View File
@@ -321,11 +321,29 @@ func (a *vncApprover) Request(ctx context.Context, info vncserver.ApprovalInfo)
Metadata: meta, Metadata: meta,
}) })
if err != nil { if err != nil {
return vncserver.ApprovalDecision{}, err return vncserver.ApprovalDecision{}, approvalCause(err)
} }
return vncserver.ApprovalDecision{ViewOnly: d.ViewOnly}, nil return vncserver.ApprovalDecision{ViewOnly: d.ViewOnly}, nil
} }
// approvalCause restates a broker failure as the cause the VNC server
// classifies, so the peer that dialled learns whether the user refused or
// never answered. The broker's denial and timeout carry nothing beyond the
// sentinel, so those are returned as the server's own; the two unavailable
// cases differ in a way worth keeping, so they are wrapped.
func approvalCause(err error) error {
switch {
case errors.Is(err, approval.ErrDenied):
return vncserver.ErrApprovalDenied
case errors.Is(err, approval.ErrTimeout):
return vncserver.ErrApprovalTimeout
case errors.Is(err, approval.ErrNoSubscriber), errors.Is(err, approval.ErrPromptNotShown):
return fmt.Errorf("%w: %w", vncserver.ErrApprovalUnavailable, err)
default:
return err
}
}
func displayPeer(info vncserver.ApprovalInfo) string { func displayPeer(info vncserver.ApprovalInfo) string {
if info.Initiator != "" { if info.Initiator != "" {
return info.Initiator return info.Initiator
+60
View File
@@ -0,0 +1,60 @@
//go:build !js && !ios && !android
package internal
import (
"context"
"errors"
"fmt"
"testing"
"github.com/stretchr/testify/assert"
"github.com/netbirdio/netbird/client/internal/approval"
vncserver "github.com/netbirdio/netbird/client/vnc/server"
)
// TestApprovalCause checks that each way the broker can refuse arrives at
// the VNC server as the matching cause. Without this the server sees one
// opaque error and tells every caller it was denied, which reads as "the
// user said no" even when no prompt was ever answered or shown.
func TestApprovalCause(t *testing.T) {
cases := []struct {
name string
in error
want error
}{
{"denied", approval.ErrDenied, vncserver.ErrApprovalDenied},
{"timeout", approval.ErrTimeout, vncserver.ErrApprovalTimeout},
{"no_subscriber", approval.ErrNoSubscriber, vncserver.ErrApprovalUnavailable},
{"prompt_not_shown", approval.ErrPromptNotShown, vncserver.ErrApprovalUnavailable},
// Wrapped on the way in: the broker may add context, and the
// classification must survive it.
{"wrapped_timeout", fmt.Errorf("request: %w", approval.ErrTimeout), vncserver.ErrApprovalTimeout},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
got := approvalCause(tc.in)
assert.ErrorIs(t, got, tc.want, "broker error %v must classify as %v", tc.in, tc.want)
})
}
}
// TestApprovalCausePreservesUnavailableDetail: the two unavailable causes
// mean different things to whoever reads the daemon log ("nobody was
// listening" versus "somebody was, and never got it"), so collapsing them
// onto one sentinel must not discard which one occurred.
func TestApprovalCausePreservesUnavailableDetail(t *testing.T) {
got := approvalCause(approval.ErrPromptNotShown)
assert.ErrorIs(t, got, vncserver.ErrApprovalUnavailable)
assert.ErrorIs(t, got, approval.ErrPromptNotShown, "the specific cause must stay readable in the log")
}
// TestApprovalCausePassesThroughUnknown keeps an error the mapping does not
// recognise intact, so the daemon log still shows what actually happened.
// The server rejects it as a denial either way.
func TestApprovalCausePassesThroughUnknown(t *testing.T) {
for _, err := range []error{context.Canceled, errors.New("something else")} {
assert.ErrorIs(t, approvalCause(err), err, "unknown cause must pass through unchanged")
}
}
+44 -9
View File
@@ -33,14 +33,16 @@ const (
// stable so clients can branch on them without parsing free text. // stable so clients can branch on them without parsing free text.
// Format: "CODE: human message". // Format: "CODE: human message".
const ( const (
RejectCodeAuthForbidden = "AUTH_FORBIDDEN" RejectCodeAuthForbidden = "AUTH_FORBIDDEN"
RejectCodeSessionError = "SESSION_ERROR" RejectCodeSessionError = "SESSION_ERROR"
RejectCodeCapturerError = "CAPTURER_ERROR" RejectCodeCapturerError = "CAPTURER_ERROR"
RejectCodeUnsupportedOS = "UNSUPPORTED" RejectCodeUnsupportedOS = "UNSUPPORTED"
RejectCodeBadRequest = "BAD_REQUEST" RejectCodeBadRequest = "BAD_REQUEST"
RejectCodeNoConsoleUser = "NO_CONSOLE_USER" RejectCodeNoConsoleUser = "NO_CONSOLE_USER"
RejectCodeApprovalDenied = "APPROVAL_DENIED" RejectCodeApprovalDenied = "APPROVAL_DENIED"
RejectCodeNoApprover = "NO_APPROVER" RejectCodeApprovalTimeout = "APPROVAL_TIMEOUT"
RejectCodeApprovalUnavailable = "APPROVAL_UNAVAILABLE"
RejectCodeNoApprover = "NO_APPROVER"
) )
// EnvVNCDisableDownscale disables any platform-specific framebuffer // EnvVNCDisableDownscale disables any platform-specific framebuffer
@@ -355,6 +357,23 @@ type Approver interface {
Request(ctx context.Context, info ApprovalInfo) (ApprovalDecision, error) Request(ctx context.Context, info ApprovalInfo) (ApprovalDecision, error)
} }
// Causes an Approver can report so the rejection names why the connection
// was refused rather than blaming the user for every outcome. An error
// matching none of these still rejects, reported as a denial.
var (
// ErrApprovalDenied means the user answered and said no.
ErrApprovalDenied = errors.New("approval denied")
// ErrApprovalTimeout means the prompt was raised and nobody answered
// it in time. Distinct from a denial because the two call for
// different things from whoever is dialling: try again later, versus
// stop asking.
ErrApprovalTimeout = errors.New("approval timed out")
// ErrApprovalUnavailable means the prompt never reached a user, so
// there was no answer to wait for. A device with no way to display
// one lands here, as does a daemon with no UI attached.
ErrApprovalUnavailable = errors.New("no approval prompt reached the user")
)
// ApprovalDecision carries the parts of the user's response the VNC // ApprovalDecision carries the parts of the user's response the VNC
// server acts on. Accept is implicit (errors signal deny). ViewOnly puts // server acts on. Accept is implicit (errors signal deny). ViewOnly puts
// the session into read-only mode: the server drops input events. // the session into read-only mode: the server drops input events.
@@ -618,7 +637,8 @@ func (s *Server) gateApproval(conn net.Conn, header *connectionHeader) (Approval
} }
decision, err := s.approver.Request(s.ctx, info) decision, err := s.approver.Request(s.ctx, info)
if err != nil { if err != nil {
rejectConnection(conn, codeMessage(RejectCodeApprovalDenied, "approval denied")) code, message := approvalRejection(err)
rejectConnection(conn, codeMessage(code, message))
return ApprovalDecision{}, fmt.Errorf("approval: %w", err) return ApprovalDecision{}, fmt.Errorf("approval: %w", err)
} }
return decision, nil return decision, nil
@@ -1259,6 +1279,21 @@ func modeString(m byte) string {
} }
} }
// approvalRejection maps an approver's error onto the code and message the
// client is told. An error matching no known cause is reported as a denial:
// the connection is refused either way, and describing a cause the server
// could not identify would put a wrong explanation in front of the caller.
func approvalRejection(err error) (string, string) {
switch {
case errors.Is(err, ErrApprovalTimeout):
return RejectCodeApprovalTimeout, "approval timed out"
case errors.Is(err, ErrApprovalUnavailable):
return RejectCodeApprovalUnavailable, "no approval prompt reached the user"
default:
return RejectCodeApprovalDenied, "approval denied"
}
}
// acceptRetryable reports whether an Accept error can plausibly clear on its // acceptRetryable reports whether an Accept error can plausibly clear on its
// own. Running out of file descriptors can, once open connections close, and an // own. Running out of file descriptors can, once open connections close, and an
// aborted handshake concerns one client rather than the listener. Anything else // aborted handshake concerns one client rather than the listener. Anything else
+75
View File
@@ -8,10 +8,12 @@ import (
"encoding/binary" "encoding/binary"
"encoding/hex" "encoding/hex"
"errors" "errors"
"fmt"
"image" "image"
"io" "io"
"net" "net"
"net/netip" "net/netip"
"strings"
"sync/atomic" "sync/atomic"
"testing" "testing"
"time" "time"
@@ -443,6 +445,37 @@ func newGateConn(t *testing.T) net.Conn {
return srv return srv
} }
// newGateConnWithReason is newGateConn plus the reject reason the server
// wrote, so a test can assert on the code the client would read.
func newGateConnWithReason(t *testing.T) (net.Conn, <-chan string) {
t.Helper()
srv, cli := net.Pipe()
t.Cleanup(func() { _ = srv.Close() })
reasons := make(chan string, 1)
go func() {
defer cli.Close()
var srvVer [12]byte
if _, err := io.ReadFull(cli, srvVer[:]); err != nil {
return
}
if _, err := cli.Write([]byte("RFB 003.008\n")); err != nil {
return
}
// Server sends numTypes=0, then a 4-byte reason length, then the reason.
var head [5]byte
if _, err := io.ReadFull(cli, head[:]); err != nil {
return
}
reason := make([]byte, binary.BigEndian.Uint32(head[1:5]))
if _, err := io.ReadFull(cli, reason); err != nil {
return
}
reasons <- string(reason)
}()
return srv, reasons
}
func gateTestServer(requireApproval bool, approver Approver) *Server { func gateTestServer(requireApproval bool, approver Approver) *Server {
return &Server{ return &Server{
log: log.WithField("test", "gate"), log: log.WithField("test", "gate"),
@@ -540,6 +573,48 @@ func TestGateApproval_ApproverDenies(t *testing.T) {
} }
} }
// TestGateApproval_RejectCodeNamesCause pins the reject code the client
// reads for each cause an approver can report. A device that cannot show a
// prompt at all, or a user who walked away, must not be reported as a user
// who said no: the caller acts on that difference, and a peer told "denied"
// when nobody was ever asked has no reason to retry. Causes are wrapped in
// the cases that carry one so the mapping cannot regress to == comparison.
func TestGateApproval_RejectCodeNamesCause(t *testing.T) {
cases := []struct {
name string
err error
code string
}{
{"denied", ErrApprovalDenied, RejectCodeApprovalDenied},
{"denied_wrapped", fmt.Errorf("broker: %w", ErrApprovalDenied), RejectCodeApprovalDenied},
{"timeout", ErrApprovalTimeout, RejectCodeApprovalTimeout},
{"timeout_wrapped", fmt.Errorf("broker: %w", ErrApprovalTimeout), RejectCodeApprovalTimeout},
{"unavailable", ErrApprovalUnavailable, RejectCodeApprovalUnavailable},
{"unavailable_wrapped", fmt.Errorf("no UI: %w", ErrApprovalUnavailable), RejectCodeApprovalUnavailable},
// An unclassified cause still rejects, reported as a denial.
{"ctx_canceled", context.Canceled, RejectCodeApprovalDenied},
{"unknown", errors.New("anything else"), RejectCodeApprovalDenied},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
app := &recordingApprover{respond: tc.err}
srv := gateTestServer(true, app)
conn, reasons := newGateConnWithReason(t)
_, err := srv.gateApproval(conn, &connectionHeader{mode: ModeAttach})
require.Error(t, err, "approver error %v must deny", tc.err)
select {
case reason := <-reasons:
assert.True(t, strings.HasPrefix(reason, tc.code+":"),
"reject reason %q must name the cause with code %s", reason, tc.code)
case <-time.After(2 * time.Second):
t.Fatal("did not observe rejection reason")
}
})
}
}
// TestGateApproval_ApproverAccepts confirms the happy path actually // TestGateApproval_ApproverAccepts confirms the happy path actually
// returns true so we know the deny path is not the only outcome the // returns true so we know the deny path is not the only outcome the
// gate can produce. // gate can produce.
+2
View File
@@ -557,6 +557,8 @@ var vncRejectCodes = [...]string{
"BAD_REQUEST", "BAD_REQUEST",
"NO_CONSOLE_USER", "NO_CONSOLE_USER",
"APPROVAL_DENIED", "APPROVAL_DENIED",
"APPROVAL_TIMEOUT",
"APPROVAL_UNAVAILABLE",
"NO_APPROVER", "NO_APPROVER",
} }