mirror of https://github.com/gogits/gogs.git
Unknwon
6 years ago
8 changed files with 57 additions and 17 deletions
@ -0,0 +1,12 @@ |
|||||||
|
// Copyright 2018 The Gogs Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package tool |
||||||
|
|
||||||
|
// IsSameSiteURLPath returns true if the URL path belongs to the same site, false otherwise.
|
||||||
|
// False: //url, http://url, /\url
|
||||||
|
// True: /url
|
||||||
|
func IsSameSiteURLPath(url string) bool { |
||||||
|
return len(url) >= 2 && url[0] == '/' && url[1] != '/' && url[1] != '\\' |
||||||
|
} |
@ -0,0 +1,32 @@ |
|||||||
|
// Copyright 2018 The Gogs Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package tool |
||||||
|
|
||||||
|
import ( |
||||||
|
"testing" |
||||||
|
|
||||||
|
. "github.com/smartystreets/goconvey/convey" |
||||||
|
) |
||||||
|
|
||||||
|
func Test_IsSameSiteURLPath(t *testing.T) { |
||||||
|
Convey("Check if a path belongs to the same site", t, func() { |
||||||
|
testCases := []struct { |
||||||
|
url string |
||||||
|
expect bool |
||||||
|
}{ |
||||||
|
{"//github.com", false}, |
||||||
|
{"http://github.com", false}, |
||||||
|
{"https://github.com", false}, |
||||||
|
{"/\\github.com", false}, |
||||||
|
|
||||||
|
{"/admin", true}, |
||||||
|
{"/user/repo", true}, |
||||||
|
} |
||||||
|
|
||||||
|
for _, tc := range testCases { |
||||||
|
So(IsSameSiteURLPath(tc.url), ShouldEqual, tc.expect) |
||||||
|
} |
||||||
|
}) |
||||||
|
} |
Loading…
Reference in new issue