Function TestSpeaker_CorruptMessage

vpn/speaker_internal_test.go:226–268

TestSpeaker_RawPeer tests the speaker with a peer that we simulate by directly making reads and writes to the other end of the pipe. There should be at least one test that does this, rather than use 2 speakers so that we don't have a bug where we don't adhere to the stated protocol, but both sides h

(t *testing.T)

// than use 2 speakers so that we don't have a bug where we don't adhere to the stated protocol, but
// both sides have the bug and can still communicate.
func TestSpeaker_CorruptMessage(t *testing.T) {
	t.Parallel()
	mp, tp := net.Pipe()
	defer mp.Close()
	defer tp.Close()
	ctx := testutil.Context(t, testutil.WaitShort)
	// We're going to use deadlines for this test so that we don't hang the main test thread if
	// the speaker misbehaves.
	err := mp.SetReadDeadline(time.Now().Add(testutil.WaitShort))
	require.NoError(t, err)
	err = mp.SetWriteDeadline(time.Now().Add(testutil.WaitShort))
	require.NoError(t, err)
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).Leveled(slog.LevelDebug)
	var tun *speaker[*TunnelMessage, *ManagerMessage, ManagerMessage]
	errCh := make(chan error, 1)
	go func() {
		s, err := newSpeaker[*TunnelMessage, *ManagerMessage](ctx, logger, tp, SpeakerRoleTunnel, SpeakerRoleManager)
		tun = s
		errCh <- err
	}()

	b := make([]byte, 256)
	n, err := mp.Read(b)
	require.NoError(t, err)
	require.Equal(t, expectedHandshake, string(b[:n]))

	_, err = mp.Write([]byte("codervpn manager 1.0\n"))
	require.NoError(t, err)

	err = testutil.TryReceive(ctx, t, errCh)
	require.NoError(t, err)
	tun.start()

	err = binary.Write(mp, binary.BigEndian, uint32(10))
	require.NoError(t, err)
	n, err = mp.Write([]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0})
	require.NoError(t, err)
	require.EqualValues(t, 10, n)

	// it should hang up on us if we write nonsense
	_, err = mp.Read(b)
	require.ErrorIs(t, err, io.EOF)
}

func TestSpeaker_unaryRPC_mainline(t *testing.T) {
	t.Parallel()