go-socks5/ruleset_test.go

package socks5

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"

	"git.tcp.direct/kayos/go-socks5/statute"
)

func TestPermitCommand(t *testing.T) {
	var r RuleSet
	var ok bool

	ctx := context.Background()

	r = NewPermitAll()
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandConnect}})
	require.True(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandBind}})
	require.True(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandAssociate}})
	require.True(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: 0x00}})
	require.False(t, ok)

	r = NewPermitConnAndAss()
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandConnect}})
	require.True(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandBind}})
	require.False(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandAssociate}})
	require.True(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: 0x00}})
	require.False(t, ok)

	r = NewPermitNone()
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandConnect}})
	require.False(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandBind}})
	require.False(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: statute.CommandAssociate}})
	require.False(t, ok)
	_, ok = r.Allow(ctx, &Request{Request: statute.Request{Command: 0x00}})
	require.False(t, ok)
}