bf0286e fix(ports): reject command argument count overflow · vev-vt

M internal/app/cmd.go +9 -1


@@ -157,13 +157,21 @@ func runCmdWithDeps(ctx context.Context, invocation cmdInvocation, deps cmdDeps)
157	157		return usagef("--self requires running inside a vev pane")
158	158		}
159	159
	160	+	payload, err := ports.MarshalCommandRequest(request)
	161	+	if errors.Is(err, ports.ErrTooManyCommandArgs) {
	162	+	return usagef("too many command arguments")
	163	+	}
	164	+	if err != nil {
	165	+	return fmt.Errorf("encoding command: %w", err)
	166	+	}
	167	+
160	168		transport, err := deps.dial(ctx, ipc.SocketDir())
161	169		if err != nil {
162	170		return &exitCoded{code: 3, err: fmt.Errorf("%w: %w", errDaemonUnreachable, err)}
163	171		}
164	172		defer func() { _ = transport.Close() }()
165	173
166		-	frame := ports.Frame{Type: ports.MsgCommand, Payload: ports.MarshalCommandRequest(request)}
	174	+	frame := ports.Frame{Type: ports.MsgCommand, Payload: payload}
167	175		if err := transport.Send(frame); err != nil {
168	176		return &exitCoded{code: 3, err: fmt.Errorf("sending command: %w", err)}
169	177		}

 		return usagef("--self requires running inside a vev pane")
 	}
+	payload, err := ports.MarshalCommandRequest(request)
+	if errors.Is(err, ports.ErrTooManyCommandArgs) {
+		return usagef("too many command arguments")
+	}
+	if err != nil {
+		return fmt.Errorf("encoding command: %w", err)
+	}
 	transport, err := deps.dial(ctx, ipc.SocketDir())
 	if err != nil {
 		return &exitCoded{code: 3, err: fmt.Errorf("%w: %w", errDaemonUnreachable, err)}
 	}
 	defer func() { _ = transport.Close() }()
-	frame := ports.Frame{Type: ports.MsgCommand, Payload: ports.MarshalCommandRequest(request)}
+	frame := ports.Frame{Type: ports.MsgCommand, Payload: payload}
 	if err := transport.Send(frame); err != nil {
 		return &exitCoded{code: 3, err: fmt.Errorf("sending command: %w", err)}
 	}

M internal/app/cmd_test.go +23 -0


@@ -4,6 +4,7 @@ import (
4	4		"context"
5	5		"errors"
6	6		"io"
	7	+	"math"
7	8		"reflect"
8	9		"strings"
9	10		"testing"

@@ -163,6 +164,28 @@ func TestRunCmdExplicitSessionDoesNotUseEnvIDsWithoutSelf(t *testing.T) {
163	164		}
164	165		}
165	166
	167	+	func TestRunCmdRejectsTooManyArgumentsBeforeDialing(t *testing.T) {
	168	+	dialed := false
	169	+	err := runCmdWithDeps(context.Background(), cmdInvocation{
	170	+	slug: "toast",
	171	+	args: make([]string, math.MaxUint16+1),
	172	+	}, cmdDeps{
	173	+	stdout: io.Discard,
	174	+	getenv: func(string) string { return "" },
	175	+	dial: func(context.Context, string) (ports.Transport, error) {
	176	+	dialed = true
	177	+	return nil, errors.New("must not dial")
	178	+	},
	179	+	})
	180	+
	181	+	if ExitCode(err) != 2 {
	182	+	t.Fatalf("error=%v code=%d, want usage error/2", err, ExitCode(err))
	183	+	}
	184	+	if dialed {
	185	+	t.Fatal("oversized command dialed daemon")
	186	+	}
	187	+	}
	188	+
166	189		func TestRunCmdDoesNotAutostartAndClassifiesDialFailure(t *testing.T) {
167	190		dials := 0
168	191		dialErr := errors.New("unreachable")


@@ -4,6 +4,7 @@ import (
4	"context"	4		"context"
5	"errors"	5		"errors"
6	"io"	6		"io"
		7	+	"math"
7	"reflect"	8		"reflect"
8	"strings"	9		"strings"
9	"testing"	10		"testing"

@@ -163,6 +164,28 @@ func TestRunCmdExplicitSessionDoesNotUseEnvIDsWithoutSelf(t *testing.T) {
163	}	164		}
164	}	165		}
165		166
		167	+	func TestRunCmdRejectsTooManyArgumentsBeforeDialing(t *testing.T) {
		168	+	dialed := false
		169	+	err := runCmdWithDeps(context.Background(), cmdInvocation{
		170	+	slug: "toast",
		171	+	args: make([]string, math.MaxUint16+1),
		172	+	}, cmdDeps{
		173	+	stdout: io.Discard,
		174	+	getenv: func(string) string { return "" },
		175	+	dial: func(context.Context, string) (ports.Transport, error) {
		176	+	dialed = true
		177	+	return nil, errors.New("must not dial")
		178	+	},
		179	+	})
		180	+
		181	+	if ExitCode(err) != 2 {
		182	+	t.Fatalf("error=%v code=%d, want usage error/2", err, ExitCode(err))
		183	+	}
		184	+	if dialed {
		185	+	t.Fatal("oversized command dialed daemon")
		186	+	}
		187	+	}
		188	+
166	func TestRunCmdDoesNotAutostartAndClassifiesDialFailure(t *testing.T) {	189		func TestRunCmdDoesNotAutostartAndClassifiesDialFailure(t *testing.T) {
167	dials := 0	190		dials := 0
168	dialErr := errors.New("unreachable")	191		dialErr := errors.New("unreachable")

M internal/ports/wire_command.go +14 -3

@@ -1,6 +1,13 @@
1	1		package ports
2	2
3		-	import "encoding/binary"
	3	+	import (
	4	+	"encoding/binary"
	5	+	"errors"
	6	+	"math"
	7	+	)
	8	+
	9	+	// ErrTooManyCommandArgs reports that a request exceeds the wire count limit.
	10	+	var ErrTooManyCommandArgs = errors.New("ports: too many command arguments")
4	11
5	12		// CommandRequest asks the daemon to run one control command. Version must
6	13		// stay first so a future payload layout can still be rejected cleanly.

@@ -33,7 +40,11 @@ func PeekCommandVersion(b []byte) (uint16, bool) {
33	40		}
34	41
35	42		// MarshalCommandRequest encodes m into a CommandRequest payload.
36		-	func MarshalCommandRequest(m CommandRequest) []byte {
	43	+	func MarshalCommandRequest(m CommandRequest) ([]byte, error) {
	44	+	if len(m.Args) > math.MaxUint16 {
	45	+	return nil, ErrTooManyCommandArgs
	46	+	}
	47	+
37	48		w := payloadWriter{}
38	49		w.putUint16(m.Version)
39	50		if m.Self {

@@ -54,7 +65,7 @@ func MarshalCommandRequest(m CommandRequest) []byte {
54	65		} else {
55	66		w.putUint8(0)
56	67		}
57		-	return w.b
	68	+	return w.b, nil
58	69		}
59	70
60	71		// UnmarshalCommandRequest decodes a strict CommandRequest payload.

 package ports
-import "encoding/binary"
+import (
+	"encoding/binary"
+	"errors"
+	"math"
+)
+// ErrTooManyCommandArgs reports that a request exceeds the wire count limit.
+var ErrTooManyCommandArgs = errors.New("ports: too many command arguments")
 // CommandRequest asks the daemon to run one control command. Version must
 // stay first so a future payload layout can still be rejected cleanly.
 }
 // MarshalCommandRequest encodes m into a CommandRequest payload.
-func MarshalCommandRequest(m CommandRequest) []byte {
+func MarshalCommandRequest(m CommandRequest) ([]byte, error) {
+	if len(m.Args) > math.MaxUint16 {
+		return nil, ErrTooManyCommandArgs
+	}
 	w := payloadWriter{}
 	w.putUint16(m.Version)
 	if m.Self {
 	} else {
 		w.putUint8(0)
 	}
-	return w.b
+	return w.b, nil
 }
 // UnmarshalCommandRequest decodes a strict CommandRequest payload.

M internal/ports/wire_test.go +16 -2


@@ -3,6 +3,7 @@ package ports
3	3		import (
4	4		"bytes"
5	5		"fmt"
	6	+	"math"
6	7		"reflect"
7	8		"testing"
8	9

@@ -498,7 +499,8 @@ func TestCommandRequestGoldenAndRoundTrip(t *testing.T) {
498	499
499	500		for _, tt := range tests {
500	501		t.Run(tt.name, func(t *testing.T) {
501		-	got := MarshalCommandRequest(tt.msg)
	502	+	got, err := MarshalCommandRequest(tt.msg)
	503	+	require.NoError(t, err)
502	504		require.Equal(t, tt.want, got)
503	505		back, err := UnmarshalCommandRequest(got)
504	506		require.NoError(t, err)

@@ -516,8 +518,20 @@ func TestCommandRequestGoldenAndRoundTrip(t *testing.T) {
516	518		}
517	519		}
518	520
	521	+	func TestMarshalCommandRequestRejectsTooManyArguments(t *testing.T) {
	522	+	payload, err := MarshalCommandRequest(CommandRequest{
	523	+	Version: ProtocolVersion,
	524	+	Slug: "toast",
	525	+	Args: make([]string, math.MaxUint16+1),
	526	+	})
	527	+
	528	+	require.Nil(t, payload)
	529	+	require.ErrorIs(t, err, ErrTooManyCommandArgs)
	530	+	}
	531	+
519	532		func TestCommandRequestRejectsImpossibleArgumentCount(t *testing.T) {
520		-	payload := MarshalCommandRequest(CommandRequest{Version: ProtocolVersion, Slug: "toast"})
	533	+	payload, err := MarshalCommandRequest(CommandRequest{Version: ProtocolVersion, Slug: "toast"})
	534	+	require.NoError(t, err)
521	535		argCountOffset := 2 + 1 + 2 + len("toast")
522	536		payload[argCountOffset] = 0xff
523	537		payload[argCountOffset+1] = 0xff

 import (
 	"bytes"
 	"fmt"
+	"math"
 	"reflect"
 	"testing"
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			got := MarshalCommandRequest(tt.msg)
+			got, err := MarshalCommandRequest(tt.msg)
+			require.NoError(t, err)
 			require.Equal(t, tt.want, got)
 			back, err := UnmarshalCommandRequest(got)
 			require.NoError(t, err)
 	}
 }
+func TestMarshalCommandRequestRejectsTooManyArguments(t *testing.T) {
+	payload, err := MarshalCommandRequest(CommandRequest{
+		Version: ProtocolVersion,
+		Slug:    "toast",
+		Args:    make([]string, math.MaxUint16+1),
+	})
+	require.Nil(t, payload)
+	require.ErrorIs(t, err, ErrTooManyCommandArgs)
+}
 func TestCommandRequestRejectsImpossibleArgumentCount(t *testing.T) {
-	payload := MarshalCommandRequest(CommandRequest{Version: ProtocolVersion, Slug: "toast"})
+	payload, err := MarshalCommandRequest(CommandRequest{Version: ProtocolVersion, Slug: "toast"})
+	require.NoError(t, err)
 	argCountOffset := 2 + 1 + 2 + len("toast")
 	payload[argCountOffset] = 0xff
 	payload[argCountOffset+1] = 0xff

M internal/usecase/daemon/control_test.go +11 -6


@@ -66,7 +66,9 @@ func TestHandleCommandRejectsVersionBeforeDecodeOrDispatch(t *testing.T) {
66	66		func TestHandleConnRoutesCommand(t *testing.T) {
67	67		d := newTestDaemon(t, nil, stubClock{})
68	68		request := ports.CommandRequest{Version: ports.ProtocolVersion, Slug: "list-sessions"}
69		-	frame := ports.Frame{Type: ports.MsgCommand, Payload: ports.MarshalCommandRequest(request)}
	69	+	payload, err := ports.MarshalCommandRequest(request)
	70	+	require.NoError(t, err)
	71	+	frame := ports.Frame{Type: ports.MsgCommand, Payload: payload}
70	72		tr, sends, _ := newConn(t, frame)
71	73
72	74		d.handleConn(tr)

@@ -463,7 +465,7 @@ func TestHandleCommandSerializesSelfTargetOnNonActiveTab(t *testing.T) {
463	465		sess.mu.Unlock()
464	466		original := target.tree.Focus
465	467
466		-	firstFrame := commandFrame(ports.CommandRequest{
	468	+	firstFrame := commandFrame(t, ports.CommandRequest{
467	469		Slug: "split-right", Self: true, TargetSession: "work",
468	470		TargetTab: "t_invoking", TargetPane: "p_invoking",
469	471		})

@@ -479,7 +481,7 @@ func TestHandleCommandSerializesSelfTargetOnNonActiveTab(t *testing.T) {
479	481		t.Fatal("first command did not enter its blocked action")
480	482		}
481	483
482		-	secondFrame := commandFrame(ports.CommandRequest{
	484	+	secondFrame := commandFrame(t, ports.CommandRequest{
483	485		Slug: "focus-pane-right", Self: true, TargetSession: "work",
484	486		TargetTab: "t_invoking", TargetPane: "p_invoking",
485	487		})

@@ -509,16 +511,19 @@ func TestHandleCommandSerializesSelfTargetOnNonActiveTab(t *testing.T) {
509	511		require.NotEqual(t, original, target.tree.Focus, "second command must execute after the split creates its right neighbor")
510	512		}
511	513
512		-	func commandFrame(request ports.CommandRequest) ports.Frame {
	514	+	func commandFrame(t *testing.T, request ports.CommandRequest) ports.Frame {
	515	+	t.Helper()
513	516		if request.Version == 0 {
514	517		request.Version = ports.ProtocolVersion
515	518		}
516		-	return ports.Frame{Type: ports.MsgCommand, Payload: ports.MarshalCommandRequest(request)}
	519	+	payload, err := ports.MarshalCommandRequest(request)
	520	+	require.NoError(t, err)
	521	+	return ports.Frame{Type: ports.MsgCommand, Payload: payload}
517	522		}
518	523
519	524		func sendCommand(t testing.T, d Daemon, request ports.CommandRequest) ports.CommandResult {
520	525		t.Helper()
521		-	frame := commandFrame(request)
	526	+	frame := commandFrame(t, request)
522	527		tr, sends, _ := newConn(t, frame)
523	528		d.handleCommand(tr, frame)
524	529		return awaitCommandResult(t, sends)

 func TestHandleConnRoutesCommand(t *testing.T) {
 	d := newTestDaemon(t, nil, stubClock{})
 	request := ports.CommandRequest{Version: ports.ProtocolVersion, Slug: "list-sessions"}
-	frame := ports.Frame{Type: ports.MsgCommand, Payload: ports.MarshalCommandRequest(request)}
+	payload, err := ports.MarshalCommandRequest(request)
+	require.NoError(t, err)
+	frame := ports.Frame{Type: ports.MsgCommand, Payload: payload}
 	tr, sends, _ := newConn(t, frame)
 	d.handleConn(tr)
 	sess.mu.Unlock()
 	original := target.tree.Focus
-	firstFrame := commandFrame(ports.CommandRequest{
+	firstFrame := commandFrame(t, ports.CommandRequest{
 		Slug: "split-right", Self: true, TargetSession: "work",
 		TargetTab: "t_invoking", TargetPane: "p_invoking",
 	})
 		t.Fatal("first command did not enter its blocked action")
 	}
-	secondFrame := commandFrame(ports.CommandRequest{
+	secondFrame := commandFrame(t, ports.CommandRequest{
 		Slug: "focus-pane-right", Self: true, TargetSession: "work",
 		TargetTab: "t_invoking", TargetPane: "p_invoking",
 	})
 	require.NotEqual(t, original, target.tree.Focus, "second command must execute after the split creates its right neighbor")
 }
-func commandFrame(request ports.CommandRequest) ports.Frame {
+func commandFrame(t *testing.T, request ports.CommandRequest) ports.Frame {
+	t.Helper()
 	if request.Version == 0 {
 		request.Version = ports.ProtocolVersion
 	}
-	return ports.Frame{Type: ports.MsgCommand, Payload: ports.MarshalCommandRequest(request)}
+	payload, err := ports.MarshalCommandRequest(request)
+	require.NoError(t, err)
+	return ports.Frame{Type: ports.MsgCommand, Payload: payload}
 }
 func sendCommand(t *testing.T, d *Daemon, request ports.CommandRequest) ports.CommandResult {
 	t.Helper()
-	frame := commandFrame(request)
+	frame := commandFrame(t, request)
 	tr, sends, _ := newConn(t, frame)
 	d.handleCommand(tr, frame)
 	return awaitCommandResult(t, sends)