瀏覽代碼

auto update script executed

Toby Chui 1 年之前
父節點
當前提交
fc35965df3
共有 1 個文件被更改,包括 2 次插入5 次删除
  1. 2 5
      mod/sshprox/utils.go

+ 2 - 5
mod/sshprox/utils.go

@@ -4,16 +4,13 @@ import (
 	"fmt"
 	"fmt"
 	"net"
 	"net"
 	"net/url"
 	"net/url"
+	"strings"
 	"time"
 	"time"
 )
 )
 
 
 //Rewrite url based on proxy root
 //Rewrite url based on proxy root
 func RewriteURL(rooturl string, requestURL string) (*url.URL, error) {
 func RewriteURL(rooturl string, requestURL string) (*url.URL, error) {
-	rewrittenURL := requestURL
-	if len(requestURL) > len(rooturl) {
-		rewrittenURL = requestURL[len(rooturl):]
-	}
-
+	rewrittenURL := strings.TrimPrefix(requestURL, rooturl)
 	return url.Parse(rewrittenURL)
 	return url.Parse(rewrittenURL)
 }
 }