2021-01-27 15:32:13 +00:00
|
|
|
package dnsforward
|
|
|
|
|
|
|
|
import (
|
|
|
|
"crypto/tls"
|
|
|
|
"net"
|
|
|
|
"net/http"
|
|
|
|
"net/url"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/AdguardTeam/dnsproxy/proxy"
|
|
|
|
"github.com/lucas-clemente/quic-go"
|
|
|
|
"github.com/stretchr/testify/assert"
|
2021-02-11 12:20:30 +00:00
|
|
|
"github.com/stretchr/testify/require"
|
2021-01-27 15:32:13 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// testTLSConn is a tlsConn for tests.
|
|
|
|
type testTLSConn struct {
|
|
|
|
// Conn is embedded here simply to make testTLSConn a net.Conn without
|
2021-04-13 14:23:36 +01:00
|
|
|
// actually implementing all methods.
|
2021-01-27 15:32:13 +00:00
|
|
|
net.Conn
|
|
|
|
|
|
|
|
serverName string
|
|
|
|
}
|
|
|
|
|
|
|
|
// ConnectionState implements the tlsConn interface for testTLSConn.
|
|
|
|
func (c testTLSConn) ConnectionState() (cs tls.ConnectionState) {
|
|
|
|
cs.ServerName = c.serverName
|
|
|
|
|
|
|
|
return cs
|
|
|
|
}
|
|
|
|
|
|
|
|
// testQUICSession is a quicSession for tests.
|
|
|
|
type testQUICSession struct {
|
|
|
|
// Session is embedded here simply to make testQUICSession
|
|
|
|
// a quic.Session without acctually implementing all methods.
|
|
|
|
quic.Session
|
|
|
|
|
|
|
|
serverName string
|
|
|
|
}
|
|
|
|
|
|
|
|
// ConnectionState implements the quicSession interface for testQUICSession.
|
|
|
|
func (c testQUICSession) ConnectionState() (cs quic.ConnectionState) {
|
2021-04-21 10:49:41 +01:00
|
|
|
cs.TLS.ServerName = c.serverName
|
2021-01-27 15:32:13 +00:00
|
|
|
|
|
|
|
return cs
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestProcessClientID(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
proto string
|
|
|
|
hostSrvName string
|
|
|
|
cliSrvName string
|
|
|
|
wantClientID string
|
|
|
|
wantErrMsg string
|
|
|
|
wantRes resultCode
|
2021-02-11 12:20:30 +00:00
|
|
|
strictSNI bool
|
2021-01-27 15:32:13 +00:00
|
|
|
}{{
|
|
|
|
name: "udp",
|
|
|
|
proto: proxy.ProtoUDP,
|
|
|
|
hostSrvName: "",
|
|
|
|
cliSrvName: "",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
2021-02-11 12:20:30 +00:00
|
|
|
strictSNI: false,
|
2021-01-27 15:32:13 +00:00
|
|
|
}, {
|
|
|
|
name: "tls_no_client_id",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "example.com",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
2021-02-11 12:20:30 +00:00
|
|
|
strictSNI: true,
|
|
|
|
}, {
|
|
|
|
name: "tls_no_client_server_name",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: `client id check: client server name "" ` +
|
|
|
|
`doesn't match host server name "example.com"`,
|
|
|
|
wantRes: resultCodeError,
|
|
|
|
strictSNI: true,
|
|
|
|
}, {
|
|
|
|
name: "tls_no_client_server_name_no_strict",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
|
|
|
strictSNI: false,
|
2021-01-27 15:32:13 +00:00
|
|
|
}, {
|
|
|
|
name: "tls_client_id",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "cli.example.com",
|
|
|
|
wantClientID: "cli",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
2021-02-11 12:20:30 +00:00
|
|
|
strictSNI: true,
|
2021-01-27 15:32:13 +00:00
|
|
|
}, {
|
|
|
|
name: "tls_client_id_hostname_error",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "cli.example.net",
|
|
|
|
wantClientID: "",
|
2021-02-11 12:20:30 +00:00
|
|
|
wantErrMsg: `client id check: client server name "cli.example.net" ` +
|
|
|
|
`doesn't match host server name "example.com"`,
|
|
|
|
wantRes: resultCodeError,
|
|
|
|
strictSNI: true,
|
2021-01-27 15:32:13 +00:00
|
|
|
}, {
|
|
|
|
name: "tls_invalid_client_id",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "!!!.example.com",
|
|
|
|
wantClientID: "",
|
2021-02-11 12:20:30 +00:00
|
|
|
wantErrMsg: `client id check: invalid client id: invalid char '!' ` +
|
2021-03-25 13:00:27 +00:00
|
|
|
`at index 0 in "!!!"`,
|
2021-02-11 12:20:30 +00:00
|
|
|
wantRes: resultCodeError,
|
|
|
|
strictSNI: true,
|
2021-01-27 15:32:13 +00:00
|
|
|
}, {
|
2021-02-11 12:20:30 +00:00
|
|
|
name: "tls_client_id_too_long",
|
|
|
|
proto: proxy.ProtoTLS,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: `abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmno` +
|
|
|
|
`pqrstuvwxyz0123456789.example.com`,
|
2021-01-27 15:32:13 +00:00
|
|
|
wantClientID: "",
|
2021-03-25 13:00:27 +00:00
|
|
|
wantErrMsg: `client id check: invalid client id: "abcdefghijklmno` +
|
2021-02-11 12:20:30 +00:00
|
|
|
`pqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789" ` +
|
2021-03-25 13:00:27 +00:00
|
|
|
`is too long, max: 63`,
|
2021-02-11 12:20:30 +00:00
|
|
|
wantRes: resultCodeError,
|
|
|
|
strictSNI: true,
|
2021-01-27 15:32:13 +00:00
|
|
|
}, {
|
|
|
|
name: "quic_client_id",
|
|
|
|
proto: proxy.ProtoQUIC,
|
|
|
|
hostSrvName: "example.com",
|
|
|
|
cliSrvName: "cli.example.com",
|
|
|
|
wantClientID: "cli",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
2021-02-11 12:20:30 +00:00
|
|
|
strictSNI: true,
|
2021-01-27 15:32:13 +00:00
|
|
|
}}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
2021-02-11 12:20:30 +00:00
|
|
|
tlsConf := TLSConfig{
|
|
|
|
ServerName: tc.hostSrvName,
|
|
|
|
StrictSNICheck: tc.strictSNI,
|
|
|
|
}
|
2021-01-27 15:32:13 +00:00
|
|
|
srv := &Server{
|
2021-02-11 12:20:30 +00:00
|
|
|
conf: ServerConfig{TLSConfig: tlsConf},
|
2021-01-27 15:32:13 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
var conn net.Conn
|
|
|
|
if tc.proto == proxy.ProtoTLS {
|
|
|
|
conn = testTLSConn{
|
|
|
|
serverName: tc.cliSrvName,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var qs quic.Session
|
|
|
|
if tc.proto == proxy.ProtoQUIC {
|
|
|
|
qs = testQUICSession{
|
|
|
|
serverName: tc.cliSrvName,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
dctx := &dnsContext{
|
|
|
|
srv: srv,
|
|
|
|
proxyCtx: &proxy.DNSContext{
|
|
|
|
Proto: tc.proto,
|
|
|
|
Conn: conn,
|
|
|
|
QUICSession: qs,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
res := processClientID(dctx)
|
|
|
|
assert.Equal(t, tc.wantRes, res)
|
|
|
|
assert.Equal(t, tc.wantClientID, dctx.clientID)
|
|
|
|
|
2021-02-11 12:20:30 +00:00
|
|
|
if tc.wantErrMsg == "" {
|
2021-03-25 13:00:27 +00:00
|
|
|
assert.NoError(t, dctx.err)
|
2021-02-11 12:20:30 +00:00
|
|
|
} else {
|
2021-03-25 13:00:27 +00:00
|
|
|
require.Error(t, dctx.err)
|
2021-02-11 12:20:30 +00:00
|
|
|
assert.Equal(t, tc.wantErrMsg, dctx.err.Error())
|
2021-01-27 15:32:13 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestProcessClientID_https(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
path string
|
|
|
|
wantClientID string
|
|
|
|
wantErrMsg string
|
|
|
|
wantRes resultCode
|
|
|
|
}{{
|
|
|
|
name: "no_client_id",
|
|
|
|
path: "/dns-query",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
|
|
|
}, {
|
|
|
|
name: "no_client_id_slash",
|
|
|
|
path: "/dns-query/",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
|
|
|
}, {
|
|
|
|
name: "client_id",
|
|
|
|
path: "/dns-query/cli",
|
|
|
|
wantClientID: "cli",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
|
|
|
}, {
|
|
|
|
name: "client_id_slash",
|
|
|
|
path: "/dns-query/cli/",
|
|
|
|
wantClientID: "cli",
|
|
|
|
wantErrMsg: "",
|
|
|
|
wantRes: resultCodeSuccess,
|
|
|
|
}, {
|
|
|
|
name: "bad_url",
|
|
|
|
path: "/foo",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: `client id check: invalid path "/foo"`,
|
|
|
|
wantRes: resultCodeError,
|
|
|
|
}, {
|
|
|
|
name: "extra",
|
|
|
|
path: "/dns-query/cli/foo",
|
|
|
|
wantClientID: "",
|
|
|
|
wantErrMsg: `client id check: invalid path "/dns-query/cli/foo": extra parts`,
|
|
|
|
wantRes: resultCodeError,
|
|
|
|
}, {
|
|
|
|
name: "invalid_client_id",
|
|
|
|
path: "/dns-query/!!!",
|
|
|
|
wantClientID: "",
|
2021-04-07 14:36:38 +01:00
|
|
|
wantErrMsg: `client id check: invalid client id: invalid char '!' ` +
|
|
|
|
`at index 0 in "!!!"`,
|
2021-02-11 12:20:30 +00:00
|
|
|
wantRes: resultCodeError,
|
2021-01-27 15:32:13 +00:00
|
|
|
}}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
r := &http.Request{
|
|
|
|
URL: &url.URL{
|
|
|
|
Path: tc.path,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
dctx := &dnsContext{
|
|
|
|
proxyCtx: &proxy.DNSContext{
|
|
|
|
Proto: proxy.ProtoHTTPS,
|
|
|
|
HTTPRequest: r,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
res := processClientID(dctx)
|
|
|
|
assert.Equal(t, tc.wantRes, res)
|
|
|
|
assert.Equal(t, tc.wantClientID, dctx.clientID)
|
|
|
|
|
2021-02-11 12:20:30 +00:00
|
|
|
if tc.wantErrMsg == "" {
|
2021-03-25 13:00:27 +00:00
|
|
|
assert.NoError(t, dctx.err)
|
2021-02-11 12:20:30 +00:00
|
|
|
} else {
|
2021-03-25 13:00:27 +00:00
|
|
|
require.Error(t, dctx.err)
|
|
|
|
|
2021-02-11 12:20:30 +00:00
|
|
|
assert.Equal(t, tc.wantErrMsg, dctx.err.Error())
|
2021-01-27 15:32:13 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|