Brendan Le Glaunec
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
package attack
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDetectIncrementalRoute_ChannelID(t *testing.T) {
|
|
route := "/StreamingSetting?version=1.0&action=getRTSPStream&ChannelID=01&ChannelName=Channel1"
|
|
|
|
match, ok := detectIncrementalRoute(route)
|
|
require.True(t, ok)
|
|
assert.True(t, match.isChannel)
|
|
assert.Equal(t, 1, match.number)
|
|
assert.Equal(t, 2, match.width)
|
|
assert.Equal(t, "/StreamingSetting?version=1.0&action=getRTSPStream&ChannelID=", match.prefix)
|
|
assert.Equal(t, "&ChannelName=Channel1", match.suffix)
|
|
|
|
next := buildIncrementedRoute(match, match.number+1)
|
|
assert.Equal(t, "/StreamingSetting?version=1.0&action=getRTSPStream&ChannelID=02&ChannelName=Channel1", next)
|
|
}
|
|
|
|
func TestDetectIncrementalRoute_ChannelSuffix(t *testing.T) {
|
|
route := "/path/to/channel2/stream"
|
|
|
|
match, ok := detectIncrementalRoute(route)
|
|
require.True(t, ok)
|
|
assert.True(t, match.isChannel)
|
|
assert.Equal(t, 2, match.number)
|
|
assert.Equal(t, "/path/to/channel", match.prefix)
|
|
assert.Equal(t, "/stream", match.suffix)
|
|
}
|
|
|
|
func TestDetectIncrementalRoute_LastNumber(t *testing.T) {
|
|
route := "/foo/bar12/baz"
|
|
|
|
match, ok := detectIncrementalRoute(route)
|
|
require.True(t, ok)
|
|
assert.False(t, match.isChannel)
|
|
assert.Equal(t, 12, match.number)
|
|
assert.Equal(t, 2, match.width)
|
|
assert.Equal(t, "/foo/bar", match.prefix)
|
|
assert.Equal(t, "/baz", match.suffix)
|
|
|
|
next := buildIncrementedRoute(match, 13)
|
|
assert.Equal(t, "/foo/bar13/baz", next)
|
|
}
|
|
|
|
func TestDetectIncrementalRoute_NoNumber(t *testing.T) {
|
|
match, ok := detectIncrementalRoute("/no/number/here")
|
|
assert.False(t, ok)
|
|
assert.Equal(t, incrementalRoute{}, match)
|
|
}
|
|
|
|
func TestDetectIncrementalRoute_OverflowAtEndFallsBack(t *testing.T) {
|
|
// The trailing token overflows strconv.Atoi, so we fall back to earlier numbers.
|
|
route := "/foo1/bar999999999999999999999999999999"
|
|
|
|
match, ok := detectIncrementalRoute(route)
|
|
require.True(t, ok)
|
|
assert.False(t, match.isChannel)
|
|
assert.Equal(t, 1, match.number)
|
|
assert.Equal(t, "/foo", match.prefix)
|
|
assert.Equal(t, "/bar999999999999999999999999999999", match.suffix)
|
|
}
|
|
|
|
func TestBuildIncrementedRoute_ZeroPadding(t *testing.T) {
|
|
match := incrementalRoute{
|
|
prefix: "/channel",
|
|
suffix: "/stream",
|
|
number: 1,
|
|
width: 3,
|
|
}
|
|
|
|
assert.Equal(t, "/channel002/stream", buildIncrementedRoute(match, 2))
|
|
}
|