Address CodeRabbit VNC review feedback

This commit is contained in:
Viktor Liu
2026-05-21 18:09:07 +02:00
parent e2127b45f8
commit d06ed64e59
9 changed files with 59 additions and 29 deletions
+17
View File
@@ -0,0 +1,17 @@
//go:build freebsd
package server
import "fmt"
// UInputInjector is a freebsd placeholder; the linux uinput implementation
// uses Linux-only ioctls (UI_DEV_CREATE etc.) and is not portable.
type UInputInjector struct {
StubInputInjector
}
// NewUInputInjector always returns an error on freebsd so callers fall back
// to a stub or platform-appropriate injector.
func NewUInputInjector(_, _ int) (*UInputInjector, error) {
return nil, fmt.Errorf("uinput not implemented on freebsd")
}
-4
View File
@@ -169,10 +169,6 @@ func (w *WindowsInputInjector) Close() {
func (w *WindowsInputInjector) tryEnqueue(cmd inputCmd) {
select {
case <-w.closed:
return
default:
}
select {
case w.ch <- cmd:
default:
}
+7 -6
View File
@@ -268,19 +268,20 @@ func TestNoise_TruncatedMsg1_ClosesConnection(t *testing.T) {
conn, err := net.Dial("tcp", addr.String())
require.NoError(t, err)
defer conn.Close()
writeHeaderPrefix(t, conn, ModeAttach)
_, err = conn.Write([]byte("NBV3"))
require.NoError(t, err)
_, err = conn.Write(make([]byte, 8))
require.NoError(t, err)
require.NoError(t, conn.Close())
require.NoError(t, conn.(*net.TCPConn).CloseWrite())
// Re-dial just to confirm the listener is alive (the previous
// connection terminated server-side without affecting the listener).
probe, err := net.Dial("tcp", addr.String())
require.NoError(t, err)
require.NoError(t, probe.Close())
require.NoError(t, conn.SetReadDeadline(time.Now().Add(2*time.Second)))
buf := make([]byte, 64)
n, err := conn.Read(buf)
require.Equal(t, 0, n, "server must not emit RFB bytes after a truncated handshake")
require.ErrorIs(t, err, io.EOF, "server must close the connection on truncated msg1")
}
// TestNoise_AuthEnabled_NoHandshake_Rejected proves that with auth on,