mirror of
https://github.com/cloudflare/cloudflared.git
synced 2025-05-10 23:26:35 +00:00
d8bee0b4d9
Also changed the socks test code so that it binds to localhost, so that we don't get popups saying "would you like to allow socks.test to use the network"
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package socks
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"golang.org/x/net/proxy"
|
|
)
|
|
|
|
type successResponse struct {
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
func sendSocksRequest(t *testing.T) []byte {
|
|
dialer, err := proxy.SOCKS5("tcp", "127.0.0.1:8086", nil, proxy.Direct)
|
|
assert.NoError(t, err)
|
|
|
|
httpTransport := &http.Transport{}
|
|
httpClient := &http.Client{Transport: httpTransport}
|
|
// set our socks5 as the dialer
|
|
httpTransport.Dial = dialer.Dial
|
|
|
|
req, err := http.NewRequest("GET", "http://127.0.0.1:8085", nil)
|
|
assert.NoError(t, err)
|
|
|
|
resp, err := httpClient.Do(req)
|
|
assert.NoError(t, err)
|
|
defer resp.Body.Close()
|
|
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
assert.NoError(t, err)
|
|
|
|
return b
|
|
}
|
|
|
|
func startTestServer(t *testing.T, httpHandler func(w http.ResponseWriter, r *http.Request)) {
|
|
// create a socks server
|
|
requestHandler := NewRequestHandler(NewNetDialer())
|
|
socksServer := NewConnectionHandler(requestHandler)
|
|
listener, err := net.Listen("tcp", "localhost:8086")
|
|
assert.NoError(t, err)
|
|
|
|
go func() {
|
|
defer listener.Close()
|
|
for {
|
|
conn, _ := listener.Accept()
|
|
go socksServer.Serve(conn)
|
|
}
|
|
}()
|
|
|
|
// create an http server
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/", httpHandler)
|
|
|
|
// start the servers
|
|
go http.ListenAndServe("localhost:8085", mux)
|
|
}
|
|
|
|
func respondWithJSON(w http.ResponseWriter, v interface{}, status int) {
|
|
data, _ := json.Marshal(v)
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(status)
|
|
w.Write(data)
|
|
}
|
|
|
|
func OkJSONResponseHandler(w http.ResponseWriter, r *http.Request) {
|
|
resp := successResponse{
|
|
Status: "ok",
|
|
}
|
|
respondWithJSON(w, resp, http.StatusOK)
|
|
}
|
|
|
|
func TestSocksConnection(t *testing.T) {
|
|
startTestServer(t, OkJSONResponseHandler)
|
|
time.Sleep(100 * time.Millisecond)
|
|
b := sendSocksRequest(t)
|
|
assert.True(t, len(b) > 0, "no data returned!")
|
|
|
|
var resp successResponse
|
|
json.Unmarshal(b, &resp)
|
|
|
|
assert.True(t, resp.Status == "ok", "response didn't return ok")
|
|
}
|