2016-02-20 17:27:06 -05:00
|
|
|
package http_test
|
2015-12-31 18:34:08 -05:00
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
|
|
|
"net/http"
|
|
|
|
"strings"
|
|
|
|
"testing"
|
2016-02-20 17:27:06 -05:00
|
|
|
|
|
|
|
testdispatcher "github.com/v2ray/v2ray-core/app/dispatcher/testing"
|
2016-08-15 16:12:11 -04:00
|
|
|
"github.com/v2ray/v2ray-core/common/dice"
|
2016-05-29 10:37:52 -04:00
|
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
2016-06-04 08:25:13 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy"
|
2016-02-20 17:27:06 -05:00
|
|
|
. "github.com/v2ray/v2ray-core/proxy/http"
|
|
|
|
"github.com/v2ray/v2ray-core/testing/assert"
|
2016-06-14 16:54:08 -04:00
|
|
|
"github.com/v2ray/v2ray-core/transport/internet"
|
|
|
|
|
|
|
|
_ "github.com/v2ray/v2ray-core/transport/internet/tcp"
|
2015-12-31 18:34:08 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestHopByHopHeadersStrip(t *testing.T) {
|
2016-05-24 15:55:46 -04:00
|
|
|
assert := assert.On(t)
|
2016-02-20 17:27:06 -05:00
|
|
|
|
2016-02-03 15:36:52 -05:00
|
|
|
rawRequest := `GET /pkg/net/http/ HTTP/1.1
|
2015-12-31 18:34:08 -05:00
|
|
|
Host: golang.org
|
|
|
|
Connection: keep-alive,Foo, Bar
|
|
|
|
Foo: foo
|
|
|
|
Bar: bar
|
|
|
|
Proxy-Connection: keep-alive
|
|
|
|
Proxy-Authenticate: abc
|
|
|
|
User-Agent: Mozilla/5.0 (Macintosh; U; Intel Mac OS X; de-de) AppleWebKit/523.10.3 (KHTML, like Gecko) Version/3.0.4 Safari/523.10
|
|
|
|
Accept-Encoding: gzip
|
|
|
|
Accept-Charset: ISO-8859-1,UTF-8;q=0.7,*;q=0.7
|
|
|
|
Cache-Control: no-cache
|
|
|
|
Accept-Language: de,en;q=0.7,en-us;q=0.3
|
|
|
|
|
|
|
|
`
|
|
|
|
b := bufio.NewReader(strings.NewReader(rawRequest))
|
|
|
|
req, err := http.ReadRequest(b)
|
|
|
|
assert.Error(err).IsNil()
|
2016-05-24 15:55:46 -04:00
|
|
|
assert.String(req.Header.Get("Foo")).Equals("foo")
|
|
|
|
assert.String(req.Header.Get("Bar")).Equals("bar")
|
|
|
|
assert.String(req.Header.Get("Connection")).Equals("keep-alive,Foo, Bar")
|
|
|
|
assert.String(req.Header.Get("Proxy-Connection")).Equals("keep-alive")
|
|
|
|
assert.String(req.Header.Get("Proxy-Authenticate")).Equals("abc")
|
2015-12-31 18:34:08 -05:00
|
|
|
|
2016-02-20 17:27:06 -05:00
|
|
|
StripHopByHopHeaders(req)
|
2016-05-24 15:55:46 -04:00
|
|
|
assert.String(req.Header.Get("Connection")).Equals("close")
|
|
|
|
assert.String(req.Header.Get("Foo")).Equals("")
|
|
|
|
assert.String(req.Header.Get("Bar")).Equals("")
|
|
|
|
assert.String(req.Header.Get("Proxy-Connection")).Equals("")
|
|
|
|
assert.String(req.Header.Get("Proxy-Authenticate")).Equals("")
|
2015-12-31 18:34:08 -05:00
|
|
|
}
|
2016-02-20 17:27:06 -05:00
|
|
|
|
|
|
|
func TestNormalGetRequest(t *testing.T) {
|
2016-05-24 15:55:46 -04:00
|
|
|
assert := assert.On(t)
|
2016-02-20 17:27:06 -05:00
|
|
|
|
|
|
|
testPacketDispatcher := testdispatcher.NewTestPacketDispatcher(nil)
|
|
|
|
|
2016-08-15 16:12:11 -04:00
|
|
|
port := v2net.Port(dice.Roll(20000) + 10000)
|
2016-06-14 16:54:08 -04:00
|
|
|
httpProxy := NewServer(
|
|
|
|
&Config{},
|
|
|
|
testPacketDispatcher,
|
|
|
|
&proxy.InboundHandlerMeta{
|
|
|
|
Address: v2net.LocalHostIP,
|
|
|
|
Port: port,
|
|
|
|
StreamSettings: &internet.StreamSettings{
|
|
|
|
Type: internet.StreamConnectionTypeRawTCP,
|
|
|
|
}})
|
2016-02-20 17:27:06 -05:00
|
|
|
defer httpProxy.Close()
|
|
|
|
|
2016-06-03 18:38:22 -04:00
|
|
|
err := httpProxy.Start()
|
2016-02-20 17:27:06 -05:00
|
|
|
assert.Error(err).IsNil()
|
2016-05-24 09:29:08 -04:00
|
|
|
assert.Port(port).Equals(httpProxy.Port())
|
2016-02-20 17:27:06 -05:00
|
|
|
|
|
|
|
httpClient := &http.Client{}
|
|
|
|
resp, err := httpClient.Get("http://127.0.0.1:" + port.String() + "/")
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
assert.Int(resp.StatusCode).Equals(400)
|
|
|
|
}
|