rewrite curl.go
parent
d0122a9771
commit
1de78f2f06
@ -0,0 +1,464 @@
|
|||||||
|
package starnet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestUrlEncodeRaw(t *testing.T) {
|
||||||
|
input := "hello world!@#$%^&*()_+-=~`"
|
||||||
|
expected := "hello%20world%21%40%23%24%25%5E%26%2A%28%29_%2B-%3D~%60"
|
||||||
|
result := UrlEncodeRaw(input)
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("UrlEncodeRaw(%q) = %q; want %q", input, result, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUrlEncode(t *testing.T) {
|
||||||
|
input := "hello world!@#$%^&*()_+-=~`"
|
||||||
|
expected := `hello+world%21%40%23%24%25%5E%26%2A%28%29_%2B-%3D~%60`
|
||||||
|
result := UrlEncode(input)
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("UrlEncode(%q) = %q; want %q", input, result, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUrlDecode(t *testing.T) {
|
||||||
|
input := "hello%20world%21%40%23%24%25%5E%26*%28%29_%2B-%3D~%60"
|
||||||
|
expected := "hello world!@#$%^&*()_+-=~`"
|
||||||
|
result, err := UrlDecode(input)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("UrlDecode(%q) returned error: %v", input, err)
|
||||||
|
}
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("UrlDecode(%q) = %q; want %q", input, result, expected)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test for error case
|
||||||
|
invalidInput := "%zz"
|
||||||
|
_, err = UrlDecode(invalidInput)
|
||||||
|
if err == nil {
|
||||||
|
t.Errorf("UrlDecode(%q) expected error, got nil", invalidInput)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildPostForm_WithValidInput(t *testing.T) {
|
||||||
|
input := map[string]string{
|
||||||
|
"key1": "value1",
|
||||||
|
"key2": "value2",
|
||||||
|
}
|
||||||
|
|
||||||
|
expected := []byte("key1=value1&key2=value2")
|
||||||
|
|
||||||
|
result := BuildPostForm(input)
|
||||||
|
|
||||||
|
if string(result) != string(expected) {
|
||||||
|
t.Errorf("BuildPostForm(%v) = %v; want %v", input, result, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildPostForm_WithEmptyInput(t *testing.T) {
|
||||||
|
input := map[string]string{}
|
||||||
|
|
||||||
|
expected := []byte("")
|
||||||
|
|
||||||
|
result := BuildPostForm(input)
|
||||||
|
|
||||||
|
if string(result) != string(expected) {
|
||||||
|
t.Errorf("BuildPostForm(%v) = %v; want %v", input, result, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildPostForm_WithNilInput(t *testing.T) {
|
||||||
|
var input map[string]string
|
||||||
|
|
||||||
|
expected := []byte("")
|
||||||
|
|
||||||
|
result := BuildPostForm(input)
|
||||||
|
|
||||||
|
if string(result) != string(expected) {
|
||||||
|
t.Errorf("BuildPostForm(%v) = %v; want %v", input, result, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetRequest(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Get(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPostRequest(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodPost {
|
||||||
|
t.Errorf("Expected 'POST', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Post(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestOptionsRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodOptions {
|
||||||
|
t.Errorf("Expected 'OPTIONS', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Options(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPutRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodPut {
|
||||||
|
t.Errorf("Expected 'PUT', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Put(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodDelete {
|
||||||
|
t.Errorf("Expected 'DELETE', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Delete(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHeadRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodHead {
|
||||||
|
t.Errorf("Expected 'HEAD', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Head(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body == "OK" {
|
||||||
|
t.Errorf("Expected , got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPatchRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodPatch {
|
||||||
|
t.Errorf("Expected 'PATCH', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Patch(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTraceRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodTrace {
|
||||||
|
t.Errorf("Expected 'TRACE', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Trace(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestConnectRequestWithValidInput(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodConnect {
|
||||||
|
t.Errorf("Expected 'CONNECT', got %v", req.Method)
|
||||||
|
}
|
||||||
|
rw.Write([]byte(`OK`))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
resp, err := Connect(server.URL)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body := resp.Body().String()
|
||||||
|
if body != "OK" {
|
||||||
|
t.Errorf("Expected OK, got %v", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func TestMethodReturnsCorrectValue(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
req.SetMethodNoError("GET")
|
||||||
|
if req.Method() != "GET" {
|
||||||
|
t.Errorf("Expected 'GET', got %v", req.Method())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetMethodHandlesInvalidInput(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
err := req.SetMethod("我是谁")
|
||||||
|
if err == nil {
|
||||||
|
t.Errorf("Expected error, got nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetMethodNoErrorSetsMethodCorrectly(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
req.SetMethodNoError("POST")
|
||||||
|
if req.Method() != "POST" {
|
||||||
|
t.Errorf("Expected 'POST', got %v", req.Method())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetMethodNoErrorIgnoresInvalidInput(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
req.SetMethodNoError("你是谁")
|
||||||
|
if req.Method() != "GET" {
|
||||||
|
t.Errorf("Expected '', got %v", req.Method())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUriReturnsCorrectValue(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
if req.Uri() != "https://example.com" {
|
||||||
|
t.Errorf("Expected 'https://example.com', got %v", req.Uri())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetUriHandlesValidInput(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
err := req.SetUri("https://newexample.com")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if req.Uri() != "https://newexample.com" {
|
||||||
|
t.Errorf("Expected 'https://newexample.com', got %v", req.Uri())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetUriHandlesInvalidInput(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
err := req.SetUri("://invalidurl")
|
||||||
|
if err == nil {
|
||||||
|
t.Errorf("Expected error, got nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetUriNoErrorSetsUriCorrectly(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
req.SetUriNoError("https://newexample.com")
|
||||||
|
if req.Uri() != "https://newexample.com" {
|
||||||
|
t.Errorf("Expected 'https://newexample.com', got %v", req.Uri())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetUriNoErrorIgnoresInvalidInput(t *testing.T) {
|
||||||
|
req := NewReq("https://example.com")
|
||||||
|
req.SetUriNoError("://invalidurl")
|
||||||
|
if req.Uri() != "https://example.com" {
|
||||||
|
t.Errorf("Expected 'https://example.com', got %v", req.Uri())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type postmanReply struct {
|
||||||
|
Args struct {
|
||||||
|
} `json:"args"`
|
||||||
|
Form map[string]string `json:"form"`
|
||||||
|
Headers map[string]string `json:"headers"`
|
||||||
|
Url string `json:"url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGet(t *testing.T) {
|
||||||
|
var reply postmanReply
|
||||||
|
resp, err := NewReq("https://postman-echo.com/get").
|
||||||
|
AddHeader("hello", "nononmo").
|
||||||
|
SetAutoCalcContentLength(true).
|
||||||
|
Do()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
err = resp.Body().Unmarshal(&reply)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
fmt.Println(resp.Body().String())
|
||||||
|
fmt.Println(reply.Headers)
|
||||||
|
fmt.Println(resp.Cookies())
|
||||||
|
}
|
||||||
|
|
||||||
|
type testData struct {
|
||||||
|
name string
|
||||||
|
args *Request
|
||||||
|
want func(*Response) error
|
||||||
|
wantErr bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func headerTestData() []testData {
|
||||||
|
return []testData{
|
||||||
|
{
|
||||||
|
name: "addHeader",
|
||||||
|
args: NewReq("https://postman-echo.com/get").
|
||||||
|
AddHeader("b612", "test-data").
|
||||||
|
AddHeader("b612", "test-header").
|
||||||
|
AddSimpleCookie("b612", "test-cookie").
|
||||||
|
SetHeader("User-Agent", "starnet test"),
|
||||||
|
want: func(resp *Response) error {
|
||||||
|
//fmt.Println(resp.Body().String())
|
||||||
|
if resp == nil {
|
||||||
|
return fmt.Errorf("response is nil")
|
||||||
|
}
|
||||||
|
if resp.StatusCode != 200 {
|
||||||
|
return fmt.Errorf("status code is %d", resp.StatusCode)
|
||||||
|
}
|
||||||
|
var reply postmanReply
|
||||||
|
err := resp.Body().Unmarshal(&reply)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if reply.Headers["b612"] != "test-data, test-header" {
|
||||||
|
return fmt.Errorf("header not found")
|
||||||
|
}
|
||||||
|
if reply.Headers["user-agent"] != "starnet test" {
|
||||||
|
return fmt.Errorf("user-agent not found")
|
||||||
|
}
|
||||||
|
if reply.Headers["cookie"] != "b612=test-cookie" {
|
||||||
|
return fmt.Errorf("cookie not found")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "postForm",
|
||||||
|
args: NewSimpleRequest("https://postman-echo.com/post", "POST").
|
||||||
|
AddHeader("b612", "test-data").
|
||||||
|
AddHeader("b612", "test-header").
|
||||||
|
AddSimpleCookie("b612", "test-cookie").
|
||||||
|
SetHeader("User-Agent", "starnet test").
|
||||||
|
//SetHeader("Content-Type", "application/x-www-form-urlencoded").
|
||||||
|
AddFormData("hello", "world").
|
||||||
|
AddFormData("hello2", "world2").
|
||||||
|
SetMethodNoError("POST"),
|
||||||
|
want: func(resp *Response) error {
|
||||||
|
//fmt.Println(resp.Body().String())
|
||||||
|
if resp == nil {
|
||||||
|
return fmt.Errorf("response is nil")
|
||||||
|
}
|
||||||
|
if resp.StatusCode != 200 {
|
||||||
|
return fmt.Errorf("status code is %d", resp.StatusCode)
|
||||||
|
}
|
||||||
|
var reply postmanReply
|
||||||
|
err := resp.Body().Unmarshal(&reply)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if reply.Headers["b612"] != "test-data, test-header" {
|
||||||
|
return fmt.Errorf("header not found")
|
||||||
|
}
|
||||||
|
if reply.Headers["user-agent"] != "starnet test" {
|
||||||
|
return fmt.Errorf("user-agent not found")
|
||||||
|
}
|
||||||
|
if reply.Headers["cookie"] != "b612=test-cookie" {
|
||||||
|
return fmt.Errorf("cookie not found")
|
||||||
|
}
|
||||||
|
if reply.Form["hello"] != "world" {
|
||||||
|
return fmt.Errorf("form data not found")
|
||||||
|
}
|
||||||
|
if reply.Form["hello2"] != "world2" {
|
||||||
|
return fmt.Errorf("form data not found")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func TestCurl(t *testing.T) {
|
||||||
|
for _, tt := range headerTestData() {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got, err := Curl(tt.args)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("Curl() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if tt.want != nil {
|
||||||
|
if err := tt.want(got); err != nil {
|
||||||
|
t.Errorf("Curl() = %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,47 +0,0 @@
|
|||||||
b612.me/stario v0.0.9 h1:bFDlejUJMwZ12a09snZJspQsOlkqpDAl9qKPEYOGWCk=
|
|
||||||
b612.me/stario v0.0.9/go.mod h1:x4D/x8zA5SC0pj/uJAi4FyG5p4j5UZoMEZfvuRR6VNw=
|
|
||||||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
|
||||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
|
||||||
golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
|
|
||||||
golang.org/x/crypto v0.21.0 h1:X31++rzVUdKhX5sWmSOFZxx8UW/ldWx55cbf08iNAMA=
|
|
||||||
golang.org/x/crypto v0.21.0/go.mod h1:0BP7YvVV9gBbVKyeTG0Gyn+gZm94bibOW5BjDEYAOMs=
|
|
||||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
|
||||||
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
|
||||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
|
||||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
|
||||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
|
||||||
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
|
||||||
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
|
|
||||||
golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=
|
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
|
||||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
|
||||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
|
||||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
|
||||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
|
||||||
golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
|
|
||||||
golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
|
||||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
|
||||||
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
|
||||||
golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
|
|
||||||
golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk=
|
|
||||||
golang.org/x/term v0.18.0 h1:FcHjZXDMxI8mM3nwhX9HlKop4C0YQvCVCdwYl2wOtE8=
|
|
||||||
golang.org/x/term v0.18.0/go.mod h1:ILwASektA3OnRv7amZ1xhE/KTR+u50pbXfZ03+6Nx58=
|
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
|
||||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
|
||||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
|
||||||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
|
||||||
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
|
|
||||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
|
||||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
|
||||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
|
||||||
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
|
||||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
|
||||||
@ -0,0 +1,120 @@
|
|||||||
|
package starnet
|
||||||
|
|
||||||
|
import "strings"
|
||||||
|
|
||||||
|
var isTokenTable = [127]bool{
|
||||||
|
'!': true,
|
||||||
|
'#': true,
|
||||||
|
'$': true,
|
||||||
|
'%': true,
|
||||||
|
'&': true,
|
||||||
|
'\'': true,
|
||||||
|
'*': true,
|
||||||
|
'+': true,
|
||||||
|
'-': true,
|
||||||
|
'.': true,
|
||||||
|
'0': true,
|
||||||
|
'1': true,
|
||||||
|
'2': true,
|
||||||
|
'3': true,
|
||||||
|
'4': true,
|
||||||
|
'5': true,
|
||||||
|
'6': true,
|
||||||
|
'7': true,
|
||||||
|
'8': true,
|
||||||
|
'9': true,
|
||||||
|
'A': true,
|
||||||
|
'B': true,
|
||||||
|
'C': true,
|
||||||
|
'D': true,
|
||||||
|
'E': true,
|
||||||
|
'F': true,
|
||||||
|
'G': true,
|
||||||
|
'H': true,
|
||||||
|
'I': true,
|
||||||
|
'J': true,
|
||||||
|
'K': true,
|
||||||
|
'L': true,
|
||||||
|
'M': true,
|
||||||
|
'N': true,
|
||||||
|
'O': true,
|
||||||
|
'P': true,
|
||||||
|
'Q': true,
|
||||||
|
'R': true,
|
||||||
|
'S': true,
|
||||||
|
'T': true,
|
||||||
|
'U': true,
|
||||||
|
'W': true,
|
||||||
|
'V': true,
|
||||||
|
'X': true,
|
||||||
|
'Y': true,
|
||||||
|
'Z': true,
|
||||||
|
'^': true,
|
||||||
|
'_': true,
|
||||||
|
'`': true,
|
||||||
|
'a': true,
|
||||||
|
'b': true,
|
||||||
|
'c': true,
|
||||||
|
'd': true,
|
||||||
|
'e': true,
|
||||||
|
'f': true,
|
||||||
|
'g': true,
|
||||||
|
'h': true,
|
||||||
|
'i': true,
|
||||||
|
'j': true,
|
||||||
|
'k': true,
|
||||||
|
'l': true,
|
||||||
|
'm': true,
|
||||||
|
'n': true,
|
||||||
|
'o': true,
|
||||||
|
'p': true,
|
||||||
|
'q': true,
|
||||||
|
'r': true,
|
||||||
|
's': true,
|
||||||
|
't': true,
|
||||||
|
'u': true,
|
||||||
|
'v': true,
|
||||||
|
'w': true,
|
||||||
|
'x': true,
|
||||||
|
'y': true,
|
||||||
|
'z': true,
|
||||||
|
'|': true,
|
||||||
|
'~': true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsTokenRune(r rune) bool {
|
||||||
|
i := int(r)
|
||||||
|
return i < len(isTokenTable) && isTokenTable[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
func validMethod(method string) bool {
|
||||||
|
/*
|
||||||
|
Method = "OPTIONS" ; Section 9.2
|
||||||
|
| "GET" ; Section 9.3
|
||||||
|
| "HEAD" ; Section 9.4
|
||||||
|
| "POST" ; Section 9.5
|
||||||
|
| "PUT" ; Section 9.6
|
||||||
|
| "DELETE" ; Section 9.7
|
||||||
|
| "TRACE" ; Section 9.8
|
||||||
|
| "CONNECT" ; Section 9.9
|
||||||
|
| extension-method
|
||||||
|
extension-method = token
|
||||||
|
token = 1*<any CHAR except CTLs or separators>
|
||||||
|
*/
|
||||||
|
return len(method) > 0 && strings.IndexFunc(method, isNotToken) == -1
|
||||||
|
}
|
||||||
|
|
||||||
|
func isNotToken(r rune) bool {
|
||||||
|
return !IsTokenRune(r)
|
||||||
|
}
|
||||||
|
|
||||||
|
func hasPort(s string) bool { return strings.LastIndex(s, ":") > strings.LastIndex(s, "]") }
|
||||||
|
|
||||||
|
// removeEmptyPort strips the empty port in ":port" to ""
|
||||||
|
// as mandated by RFC 3986 Section 6.2.3.
|
||||||
|
func removeEmptyPort(host string) string {
|
||||||
|
if hasPort(host) {
|
||||||
|
return strings.TrimSuffix(host, ":")
|
||||||
|
}
|
||||||
|
return host
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue