mirror of
https://github.com/coder/coder.git
synced 2026-09-22 13:10:21 +08:00
118 lines
3.8 KiB
Go
118 lines
3.8 KiB
Go
package workspaceapps
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Test_originLocalURL checks that originLocalURL produces a redirect target that
|
|
// stays on the current origin.
|
|
func Test_originLocalURL(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("RejectsOffOrigin", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// Each path models an already-percent-decoded r.URL.Path that tries to
|
|
// smuggle a separate host into the redirect.
|
|
cases := []struct {
|
|
name string
|
|
path string
|
|
}{
|
|
{name: "DoubleSlash", path: "//evil.com/phish"},
|
|
{name: "TripleSlash", path: "///evil.com/phish"},
|
|
{name: "SlashBackslash", path: "/\\evil.com/phish"},
|
|
{name: "SlashBackslashSlash", path: "/\\/evil.com/phish"},
|
|
{name: "DoubleBackslash", path: "\\\\evil.com/phish"},
|
|
{name: "SlashTab", path: "/\t/evil.com/phish"},
|
|
{name: "SlashTabBackslash", path: "/\t\\evil.com/phish"},
|
|
{name: "SlashNewline", path: "/\n/evil.com/phish"},
|
|
{name: "SlashCarriageReturn", path: "/\r/evil.com/phish"},
|
|
{name: "SlashTabDoubleSlash", path: "/\t//evil.com/phish"},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
loc := originLocalURL(tc.path).String()
|
|
|
|
// The Location must parse as a relative, same-origin reference.
|
|
require.Falsef(t, strings.HasPrefix(loc, "//"),
|
|
"path %q produced scheme-relative Location %q", tc.path, loc)
|
|
parsed, err := url.Parse(loc)
|
|
require.NoErrorf(t, err, "path %q produced unparseable Location %q", tc.path, loc)
|
|
require.Emptyf(t, parsed.Scheme, "path %q produced Location %q with a scheme", tc.path, loc)
|
|
require.Emptyf(t, parsed.Host, "path %q produced Location %q with a host", tc.path, loc)
|
|
|
|
// It must also be free of raw bytes a browser would normalize back
|
|
// into an authority before resolving (a backslash becomes "/", and
|
|
// tab/newline/CR are stripped, either of which could re-form
|
|
// "//host"). url.URL.String() guarantees this by percent-encoding
|
|
// them; we assert it here rather than reproducing browser
|
|
// normalization in the code.
|
|
for _, raw := range []string{`\`, "\t", "\n", "\r"} {
|
|
require.NotContainsf(t, loc, raw,
|
|
"path %q produced Location %q containing a raw %q", tc.path, loc, raw)
|
|
}
|
|
})
|
|
}
|
|
})
|
|
|
|
t.Run("EscapesControlCharacters", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// A redirect built from a path containing a raw control character is an
|
|
// open redirect: http.Redirect emits it verbatim (url.Parse rejects the
|
|
// control byte and skips cleaning) and browsers strip tab/newline/CR
|
|
// before resolving, re-forming "//evil.com". originLocalURL percent-encodes
|
|
// each one. Assert every class is escaped so a future change that breaks
|
|
// encoding for only one class is caught.
|
|
cases := []struct {
|
|
name string
|
|
in string
|
|
want string
|
|
}{
|
|
{name: "Tab", in: "/\t/evil.com", want: "/%09/evil.com"},
|
|
{name: "Newline", in: "/\n/evil.com", want: "/%0A/evil.com"},
|
|
{name: "CarriageReturn", in: "/\r/evil.com", want: "/%0D/evil.com"},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Equalf(t, tc.want, originLocalURL(tc.in).String(),
|
|
"originLocalURL(%q) must percent-encode the control character", tc.in)
|
|
})
|
|
}
|
|
})
|
|
|
|
t.Run("PreservesLegitPaths", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cases := []struct {
|
|
name string
|
|
in string
|
|
want string
|
|
}{
|
|
{name: "Empty", in: "", want: "/"},
|
|
{name: "Root", in: "/", want: "/"},
|
|
{name: "Simple", in: "/test", want: "/test"},
|
|
{name: "Nested", in: "/app/sub/page", want: "/app/sub/page"},
|
|
{name: "PathApp", in: "/@user/ws/apps/app", want: "/@user/ws/apps/app"},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Equalf(t, tc.want, originLocalURL(tc.in).String(), "originLocalURL(%q)", tc.in)
|
|
})
|
|
}
|
|
})
|
|
}
|