feature_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859
  1. package scenarios
  2. import (
  3. "context"
  4. "io/ioutil"
  5. "net/http"
  6. "net/url"
  7. "testing"
  8. "time"
  9. xproxy "golang.org/x/net/proxy"
  10. "v2ray.com/core"
  11. "v2ray.com/core/app/dispatcher"
  12. "v2ray.com/core/app/log"
  13. "v2ray.com/core/app/proxyman"
  14. _ "v2ray.com/core/app/proxyman/inbound"
  15. _ "v2ray.com/core/app/proxyman/outbound"
  16. "v2ray.com/core/app/router"
  17. clog "v2ray.com/core/common/log"
  18. "v2ray.com/core/common/net"
  19. "v2ray.com/core/common/protocol"
  20. "v2ray.com/core/common/serial"
  21. "v2ray.com/core/common/uuid"
  22. "v2ray.com/core/proxy/blackhole"
  23. "v2ray.com/core/proxy/dokodemo"
  24. "v2ray.com/core/proxy/freedom"
  25. v2http "v2ray.com/core/proxy/http"
  26. "v2ray.com/core/proxy/socks"
  27. "v2ray.com/core/proxy/vmess"
  28. "v2ray.com/core/proxy/vmess/inbound"
  29. "v2ray.com/core/proxy/vmess/outbound"
  30. "v2ray.com/core/testing/servers/tcp"
  31. "v2ray.com/core/testing/servers/udp"
  32. "v2ray.com/core/transport/internet"
  33. . "v2ray.com/ext/assert"
  34. )
  35. func TestPassiveConnection(t *testing.T) {
  36. assert := With(t)
  37. tcpServer := tcp.Server{
  38. MsgProcessor: xor,
  39. SendFirst: []byte("send first"),
  40. }
  41. dest, err := tcpServer.Start()
  42. assert(err, IsNil)
  43. defer tcpServer.Close()
  44. serverPort := tcp.PickPort()
  45. serverConfig := &core.Config{
  46. Inbound: []*core.InboundHandlerConfig{
  47. {
  48. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  49. PortRange: net.SinglePortRange(serverPort),
  50. Listen: net.NewIPOrDomain(net.LocalHostIP),
  51. }),
  52. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  53. Address: net.NewIPOrDomain(dest.Address),
  54. Port: uint32(dest.Port),
  55. NetworkList: &net.NetworkList{
  56. Network: []net.Network{net.Network_TCP},
  57. },
  58. }),
  59. },
  60. },
  61. Outbound: []*core.OutboundHandlerConfig{
  62. {
  63. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  64. },
  65. },
  66. }
  67. servers, err := InitializeServerConfigs(serverConfig)
  68. assert(err, IsNil)
  69. conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
  70. IP: []byte{127, 0, 0, 1},
  71. Port: int(serverPort),
  72. })
  73. assert(err, IsNil)
  74. {
  75. response := make([]byte, 1024)
  76. nBytes, err := conn.Read(response)
  77. assert(err, IsNil)
  78. assert(string(response[:nBytes]), Equals, "send first")
  79. }
  80. payload := "dokodemo request."
  81. {
  82. nBytes, err := conn.Write([]byte(payload))
  83. assert(err, IsNil)
  84. assert(nBytes, Equals, len(payload))
  85. }
  86. {
  87. response := make([]byte, 1024)
  88. nBytes, err := conn.Read(response)
  89. assert(err, IsNil)
  90. assert(response[:nBytes], Equals, xor([]byte(payload)))
  91. }
  92. assert(conn.Close(), IsNil)
  93. CloseAllServers(servers)
  94. }
  95. func TestProxy(t *testing.T) {
  96. assert := With(t)
  97. tcpServer := tcp.Server{
  98. MsgProcessor: xor,
  99. }
  100. dest, err := tcpServer.Start()
  101. assert(err, IsNil)
  102. defer tcpServer.Close()
  103. serverUserID := protocol.NewID(uuid.New())
  104. serverPort := tcp.PickPort()
  105. serverConfig := &core.Config{
  106. Inbound: []*core.InboundHandlerConfig{
  107. {
  108. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  109. PortRange: net.SinglePortRange(serverPort),
  110. Listen: net.NewIPOrDomain(net.LocalHostIP),
  111. }),
  112. ProxySettings: serial.ToTypedMessage(&inbound.Config{
  113. User: []*protocol.User{
  114. {
  115. Account: serial.ToTypedMessage(&vmess.Account{
  116. Id: serverUserID.String(),
  117. }),
  118. },
  119. },
  120. }),
  121. },
  122. },
  123. Outbound: []*core.OutboundHandlerConfig{
  124. {
  125. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  126. },
  127. },
  128. }
  129. proxyUserID := protocol.NewID(uuid.New())
  130. proxyPort := tcp.PickPort()
  131. proxyConfig := &core.Config{
  132. Inbound: []*core.InboundHandlerConfig{
  133. {
  134. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  135. PortRange: net.SinglePortRange(proxyPort),
  136. Listen: net.NewIPOrDomain(net.LocalHostIP),
  137. }),
  138. ProxySettings: serial.ToTypedMessage(&inbound.Config{
  139. User: []*protocol.User{
  140. {
  141. Account: serial.ToTypedMessage(&vmess.Account{
  142. Id: proxyUserID.String(),
  143. }),
  144. },
  145. },
  146. }),
  147. },
  148. },
  149. Outbound: []*core.OutboundHandlerConfig{
  150. {
  151. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  152. },
  153. },
  154. }
  155. clientPort := tcp.PickPort()
  156. clientConfig := &core.Config{
  157. Inbound: []*core.InboundHandlerConfig{
  158. {
  159. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  160. PortRange: net.SinglePortRange(clientPort),
  161. Listen: net.NewIPOrDomain(net.LocalHostIP),
  162. }),
  163. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  164. Address: net.NewIPOrDomain(dest.Address),
  165. Port: uint32(dest.Port),
  166. NetworkList: &net.NetworkList{
  167. Network: []net.Network{net.Network_TCP},
  168. },
  169. }),
  170. },
  171. },
  172. Outbound: []*core.OutboundHandlerConfig{
  173. {
  174. ProxySettings: serial.ToTypedMessage(&outbound.Config{
  175. Receiver: []*protocol.ServerEndpoint{
  176. {
  177. Address: net.NewIPOrDomain(net.LocalHostIP),
  178. Port: uint32(serverPort),
  179. User: []*protocol.User{
  180. {
  181. Account: serial.ToTypedMessage(&vmess.Account{
  182. Id: serverUserID.String(),
  183. }),
  184. },
  185. },
  186. },
  187. },
  188. }),
  189. SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
  190. ProxySettings: &internet.ProxyConfig{
  191. Tag: "proxy",
  192. },
  193. }),
  194. },
  195. {
  196. Tag: "proxy",
  197. ProxySettings: serial.ToTypedMessage(&outbound.Config{
  198. Receiver: []*protocol.ServerEndpoint{
  199. {
  200. Address: net.NewIPOrDomain(net.LocalHostIP),
  201. Port: uint32(proxyPort),
  202. User: []*protocol.User{
  203. {
  204. Account: serial.ToTypedMessage(&vmess.Account{
  205. Id: proxyUserID.String(),
  206. }),
  207. },
  208. },
  209. },
  210. },
  211. }),
  212. },
  213. },
  214. }
  215. servers, err := InitializeServerConfigs(serverConfig, proxyConfig, clientConfig)
  216. assert(err, IsNil)
  217. conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
  218. IP: []byte{127, 0, 0, 1},
  219. Port: int(clientPort),
  220. })
  221. assert(err, IsNil)
  222. payload := "dokodemo request."
  223. nBytes, err := conn.Write([]byte(payload))
  224. assert(err, IsNil)
  225. assert(nBytes, Equals, len(payload))
  226. response := make([]byte, 1024)
  227. nBytes, err = conn.Read(response)
  228. assert(err, IsNil)
  229. assert(response[:nBytes], Equals, xor([]byte(payload)))
  230. assert(conn.Close(), IsNil)
  231. CloseAllServers(servers)
  232. }
  233. func TestProxyOverKCP(t *testing.T) {
  234. assert := With(t)
  235. tcpServer := tcp.Server{
  236. MsgProcessor: xor,
  237. }
  238. dest, err := tcpServer.Start()
  239. assert(err, IsNil)
  240. defer tcpServer.Close()
  241. serverUserID := protocol.NewID(uuid.New())
  242. serverPort := tcp.PickPort()
  243. serverConfig := &core.Config{
  244. Inbound: []*core.InboundHandlerConfig{
  245. {
  246. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  247. PortRange: net.SinglePortRange(serverPort),
  248. Listen: net.NewIPOrDomain(net.LocalHostIP),
  249. StreamSettings: &internet.StreamConfig{
  250. Protocol: internet.TransportProtocol_MKCP,
  251. },
  252. }),
  253. ProxySettings: serial.ToTypedMessage(&inbound.Config{
  254. User: []*protocol.User{
  255. {
  256. Account: serial.ToTypedMessage(&vmess.Account{
  257. Id: serverUserID.String(),
  258. }),
  259. },
  260. },
  261. }),
  262. },
  263. },
  264. Outbound: []*core.OutboundHandlerConfig{
  265. {
  266. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  267. },
  268. },
  269. }
  270. proxyUserID := protocol.NewID(uuid.New())
  271. proxyPort := tcp.PickPort()
  272. proxyConfig := &core.Config{
  273. Inbound: []*core.InboundHandlerConfig{
  274. {
  275. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  276. PortRange: net.SinglePortRange(proxyPort),
  277. Listen: net.NewIPOrDomain(net.LocalHostIP),
  278. }),
  279. ProxySettings: serial.ToTypedMessage(&inbound.Config{
  280. User: []*protocol.User{
  281. {
  282. Account: serial.ToTypedMessage(&vmess.Account{
  283. Id: proxyUserID.String(),
  284. }),
  285. },
  286. },
  287. }),
  288. },
  289. },
  290. Outbound: []*core.OutboundHandlerConfig{
  291. {
  292. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  293. SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
  294. StreamSettings: &internet.StreamConfig{
  295. Protocol: internet.TransportProtocol_MKCP,
  296. },
  297. }),
  298. },
  299. },
  300. }
  301. clientPort := tcp.PickPort()
  302. clientConfig := &core.Config{
  303. Inbound: []*core.InboundHandlerConfig{
  304. {
  305. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  306. PortRange: net.SinglePortRange(clientPort),
  307. Listen: net.NewIPOrDomain(net.LocalHostIP),
  308. }),
  309. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  310. Address: net.NewIPOrDomain(dest.Address),
  311. Port: uint32(dest.Port),
  312. NetworkList: &net.NetworkList{
  313. Network: []net.Network{net.Network_TCP},
  314. },
  315. }),
  316. },
  317. },
  318. Outbound: []*core.OutboundHandlerConfig{
  319. {
  320. ProxySettings: serial.ToTypedMessage(&outbound.Config{
  321. Receiver: []*protocol.ServerEndpoint{
  322. {
  323. Address: net.NewIPOrDomain(net.LocalHostIP),
  324. Port: uint32(serverPort),
  325. User: []*protocol.User{
  326. {
  327. Account: serial.ToTypedMessage(&vmess.Account{
  328. Id: serverUserID.String(),
  329. }),
  330. },
  331. },
  332. },
  333. },
  334. }),
  335. SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
  336. ProxySettings: &internet.ProxyConfig{
  337. Tag: "proxy",
  338. },
  339. StreamSettings: &internet.StreamConfig{
  340. Protocol: internet.TransportProtocol_MKCP,
  341. },
  342. }),
  343. },
  344. {
  345. Tag: "proxy",
  346. ProxySettings: serial.ToTypedMessage(&outbound.Config{
  347. Receiver: []*protocol.ServerEndpoint{
  348. {
  349. Address: net.NewIPOrDomain(net.LocalHostIP),
  350. Port: uint32(proxyPort),
  351. User: []*protocol.User{
  352. {
  353. Account: serial.ToTypedMessage(&vmess.Account{
  354. Id: proxyUserID.String(),
  355. }),
  356. },
  357. },
  358. },
  359. },
  360. }),
  361. },
  362. },
  363. }
  364. servers, err := InitializeServerConfigs(serverConfig, proxyConfig, clientConfig)
  365. assert(err, IsNil)
  366. conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
  367. IP: []byte{127, 0, 0, 1},
  368. Port: int(clientPort),
  369. })
  370. assert(err, IsNil)
  371. payload := "dokodemo request."
  372. nBytes, err := conn.Write([]byte(payload))
  373. assert(err, IsNil)
  374. assert(nBytes, Equals, len(payload))
  375. response := make([]byte, 1024)
  376. nBytes, err = conn.Read(response)
  377. assert(err, IsNil)
  378. assert(response[:nBytes], Equals, xor([]byte(payload)))
  379. assert(conn.Close(), IsNil)
  380. CloseAllServers(servers)
  381. }
  382. func TestBlackhole(t *testing.T) {
  383. assert := With(t)
  384. tcpServer := tcp.Server{
  385. MsgProcessor: xor,
  386. }
  387. dest, err := tcpServer.Start()
  388. assert(err, IsNil)
  389. defer tcpServer.Close()
  390. tcpServer2 := tcp.Server{
  391. MsgProcessor: xor,
  392. }
  393. dest2, err := tcpServer2.Start()
  394. assert(err, IsNil)
  395. defer tcpServer2.Close()
  396. serverPort := tcp.PickPort()
  397. serverPort2 := tcp.PickPort()
  398. serverConfig := &core.Config{
  399. Inbound: []*core.InboundHandlerConfig{
  400. {
  401. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  402. PortRange: net.SinglePortRange(serverPort),
  403. Listen: net.NewIPOrDomain(net.LocalHostIP),
  404. }),
  405. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  406. Address: net.NewIPOrDomain(dest.Address),
  407. Port: uint32(dest.Port),
  408. NetworkList: &net.NetworkList{
  409. Network: []net.Network{net.Network_TCP},
  410. },
  411. }),
  412. },
  413. {
  414. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  415. PortRange: net.SinglePortRange(serverPort2),
  416. Listen: net.NewIPOrDomain(net.LocalHostIP),
  417. }),
  418. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  419. Address: net.NewIPOrDomain(dest2.Address),
  420. Port: uint32(dest2.Port),
  421. NetworkList: &net.NetworkList{
  422. Network: []net.Network{net.Network_TCP},
  423. },
  424. }),
  425. },
  426. },
  427. Outbound: []*core.OutboundHandlerConfig{
  428. {
  429. Tag: "direct",
  430. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  431. },
  432. {
  433. Tag: "blocked",
  434. ProxySettings: serial.ToTypedMessage(&blackhole.Config{}),
  435. },
  436. },
  437. App: []*serial.TypedMessage{
  438. serial.ToTypedMessage(&router.Config{
  439. Rule: []*router.RoutingRule{
  440. {
  441. TargetTag: &router.RoutingRule_Tag{
  442. Tag: "blocked",
  443. },
  444. PortRange: net.SinglePortRange(dest2.Port),
  445. },
  446. },
  447. }),
  448. },
  449. }
  450. servers, err := InitializeServerConfigs(serverConfig)
  451. assert(err, IsNil)
  452. conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
  453. IP: []byte{127, 0, 0, 1},
  454. Port: int(serverPort2),
  455. })
  456. assert(err, IsNil)
  457. payload := "dokodemo request."
  458. {
  459. nBytes, err := conn.Write([]byte(payload))
  460. assert(err, IsNil)
  461. assert(nBytes, Equals, len(payload))
  462. }
  463. {
  464. response := make([]byte, 1024)
  465. _, err := conn.Read(response)
  466. assert(err, IsNotNil)
  467. }
  468. assert(conn.Close(), IsNil)
  469. CloseAllServers(servers)
  470. }
  471. func TestForward(t *testing.T) {
  472. assert := With(t)
  473. tcpServer := tcp.Server{
  474. MsgProcessor: xor,
  475. }
  476. dest, err := tcpServer.Start()
  477. assert(err, IsNil)
  478. defer tcpServer.Close()
  479. serverPort := tcp.PickPort()
  480. serverConfig := &core.Config{
  481. Inbound: []*core.InboundHandlerConfig{
  482. {
  483. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  484. PortRange: net.SinglePortRange(serverPort),
  485. Listen: net.NewIPOrDomain(net.LocalHostIP),
  486. }),
  487. ProxySettings: serial.ToTypedMessage(&socks.ServerConfig{
  488. AuthType: socks.AuthType_NO_AUTH,
  489. Accounts: map[string]string{
  490. "Test Account": "Test Password",
  491. },
  492. Address: net.NewIPOrDomain(net.LocalHostIP),
  493. UdpEnabled: false,
  494. }),
  495. },
  496. },
  497. Outbound: []*core.OutboundHandlerConfig{
  498. {
  499. ProxySettings: serial.ToTypedMessage(&freedom.Config{
  500. DestinationOverride: &freedom.DestinationOverride{
  501. Server: &protocol.ServerEndpoint{
  502. Address: net.NewIPOrDomain(net.LocalHostIP),
  503. Port: uint32(dest.Port),
  504. },
  505. },
  506. }),
  507. },
  508. },
  509. }
  510. servers, err := InitializeServerConfigs(serverConfig)
  511. assert(err, IsNil)
  512. {
  513. noAuthDialer, err := xproxy.SOCKS5("tcp", net.TCPDestination(net.LocalHostIP, serverPort).NetAddr(), nil, xproxy.Direct)
  514. assert(err, IsNil)
  515. conn, err := noAuthDialer.Dial("tcp", "google.com:80")
  516. assert(err, IsNil)
  517. payload := "test payload"
  518. nBytes, err := conn.Write([]byte(payload))
  519. assert(err, IsNil)
  520. assert(nBytes, Equals, len(payload))
  521. response := make([]byte, 1024)
  522. nBytes, err = conn.Read(response)
  523. assert(err, IsNil)
  524. assert(response[:nBytes], Equals, xor([]byte(payload)))
  525. assert(conn.Close(), IsNil)
  526. }
  527. CloseAllServers(servers)
  528. }
  529. func TestUDPConnection(t *testing.T) {
  530. assert := With(t)
  531. udpServer := udp.Server{
  532. MsgProcessor: xor,
  533. }
  534. dest, err := udpServer.Start()
  535. assert(err, IsNil)
  536. defer udpServer.Close()
  537. clientPort := tcp.PickPort()
  538. clientConfig := &core.Config{
  539. Inbound: []*core.InboundHandlerConfig{
  540. {
  541. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  542. PortRange: net.SinglePortRange(clientPort),
  543. Listen: net.NewIPOrDomain(net.LocalHostIP),
  544. }),
  545. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  546. Address: net.NewIPOrDomain(dest.Address),
  547. Port: uint32(dest.Port),
  548. NetworkList: &net.NetworkList{
  549. Network: []net.Network{net.Network_UDP},
  550. },
  551. }),
  552. },
  553. },
  554. Outbound: []*core.OutboundHandlerConfig{
  555. {
  556. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  557. },
  558. },
  559. }
  560. servers, err := InitializeServerConfigs(clientConfig)
  561. assert(err, IsNil)
  562. {
  563. conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
  564. IP: []byte{127, 0, 0, 1},
  565. Port: int(clientPort),
  566. })
  567. assert(err, IsNil)
  568. payload := "dokodemo request."
  569. for i := 0; i < 5; i++ {
  570. nBytes, err := conn.Write([]byte(payload))
  571. assert(err, IsNil)
  572. assert(nBytes, Equals, len(payload))
  573. response := make([]byte, 1024)
  574. nBytes, err = conn.Read(response)
  575. assert(err, IsNil)
  576. assert(response[:nBytes], Equals, xor([]byte(payload)))
  577. }
  578. assert(conn.Close(), IsNil)
  579. }
  580. time.Sleep(20 * time.Second)
  581. {
  582. conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
  583. IP: []byte{127, 0, 0, 1},
  584. Port: int(clientPort),
  585. })
  586. assert(err, IsNil)
  587. payload := "dokodemo request."
  588. nBytes, err := conn.Write([]byte(payload))
  589. assert(err, IsNil)
  590. assert(nBytes, Equals, len(payload))
  591. response := make([]byte, 1024)
  592. nBytes, err = conn.Read(response)
  593. assert(err, IsNil)
  594. assert(response[:nBytes], Equals, xor([]byte(payload)))
  595. assert(conn.Close(), IsNil)
  596. }
  597. CloseAllServers(servers)
  598. }
  599. func TestDomainSniffing(t *testing.T) {
  600. assert := With(t)
  601. sniffingPort := tcp.PickPort()
  602. httpPort := tcp.PickPort()
  603. serverConfig := &core.Config{
  604. Inbound: []*core.InboundHandlerConfig{
  605. {
  606. Tag: "snif",
  607. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  608. PortRange: net.SinglePortRange(sniffingPort),
  609. Listen: net.NewIPOrDomain(net.LocalHostIP),
  610. DomainOverride: []proxyman.KnownProtocols{
  611. proxyman.KnownProtocols_TLS,
  612. },
  613. }),
  614. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  615. Address: net.NewIPOrDomain(net.LocalHostIP),
  616. Port: 443,
  617. NetworkList: &net.NetworkList{
  618. Network: []net.Network{net.Network_TCP},
  619. },
  620. }),
  621. },
  622. {
  623. Tag: "http",
  624. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  625. PortRange: net.SinglePortRange(httpPort),
  626. Listen: net.NewIPOrDomain(net.LocalHostIP),
  627. }),
  628. ProxySettings: serial.ToTypedMessage(&v2http.ServerConfig{}),
  629. },
  630. },
  631. Outbound: []*core.OutboundHandlerConfig{
  632. {
  633. Tag: "redir",
  634. ProxySettings: serial.ToTypedMessage(&freedom.Config{
  635. DestinationOverride: &freedom.DestinationOverride{
  636. Server: &protocol.ServerEndpoint{
  637. Address: net.NewIPOrDomain(net.LocalHostIP),
  638. Port: uint32(sniffingPort),
  639. },
  640. },
  641. }),
  642. },
  643. {
  644. Tag: "direct",
  645. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  646. },
  647. },
  648. App: []*serial.TypedMessage{
  649. serial.ToTypedMessage(&router.Config{
  650. Rule: []*router.RoutingRule{
  651. {
  652. TargetTag: &router.RoutingRule_Tag{
  653. Tag: "direct",
  654. },
  655. InboundTag: []string{"snif"},
  656. }, {
  657. TargetTag: &router.RoutingRule_Tag{
  658. Tag: "redir",
  659. },
  660. InboundTag: []string{"http"},
  661. },
  662. },
  663. }),
  664. serial.ToTypedMessage(&log.Config{
  665. ErrorLogLevel: clog.Severity_Debug,
  666. ErrorLogType: log.LogType_Console,
  667. }),
  668. },
  669. }
  670. servers, err := InitializeServerConfigs(serverConfig)
  671. assert(err, IsNil)
  672. {
  673. transport := &http.Transport{
  674. Proxy: func(req *http.Request) (*url.URL, error) {
  675. return url.Parse("http://127.0.0.1:" + httpPort.String())
  676. },
  677. }
  678. client := &http.Client{
  679. Transport: transport,
  680. }
  681. resp, err := client.Get("https://www.github.com/")
  682. assert(err, IsNil)
  683. assert(resp.StatusCode, Equals, 200)
  684. assert(resp.Write(ioutil.Discard), IsNil)
  685. }
  686. CloseAllServers(servers)
  687. }
  688. func TestDialV2Ray(t *testing.T) {
  689. assert := With(t)
  690. tcpServer := tcp.Server{
  691. MsgProcessor: xor,
  692. }
  693. dest, err := tcpServer.Start()
  694. assert(err, IsNil)
  695. defer tcpServer.Close()
  696. userID := protocol.NewID(uuid.New())
  697. serverPort := tcp.PickPort()
  698. serverConfig := &core.Config{
  699. App: []*serial.TypedMessage{
  700. serial.ToTypedMessage(&log.Config{
  701. ErrorLogLevel: clog.Severity_Debug,
  702. ErrorLogType: log.LogType_Console,
  703. }),
  704. },
  705. Inbound: []*core.InboundHandlerConfig{
  706. {
  707. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  708. PortRange: net.SinglePortRange(serverPort),
  709. Listen: net.NewIPOrDomain(net.LocalHostIP),
  710. }),
  711. ProxySettings: serial.ToTypedMessage(&inbound.Config{
  712. User: []*protocol.User{
  713. {
  714. Account: serial.ToTypedMessage(&vmess.Account{
  715. Id: userID.String(),
  716. AlterId: 64,
  717. }),
  718. },
  719. },
  720. }),
  721. },
  722. },
  723. Outbound: []*core.OutboundHandlerConfig{
  724. {
  725. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  726. },
  727. },
  728. }
  729. clientConfig := &core.Config{
  730. App: []*serial.TypedMessage{
  731. serial.ToTypedMessage(&dispatcher.Config{}),
  732. serial.ToTypedMessage(&proxyman.InboundConfig{}),
  733. serial.ToTypedMessage(&proxyman.OutboundConfig{}),
  734. },
  735. Inbound: []*core.InboundHandlerConfig{},
  736. Outbound: []*core.OutboundHandlerConfig{
  737. {
  738. ProxySettings: serial.ToTypedMessage(&outbound.Config{
  739. Receiver: []*protocol.ServerEndpoint{
  740. {
  741. Address: net.NewIPOrDomain(net.LocalHostIP),
  742. Port: uint32(serverPort),
  743. User: []*protocol.User{
  744. {
  745. Account: serial.ToTypedMessage(&vmess.Account{
  746. Id: userID.String(),
  747. AlterId: 64,
  748. SecuritySettings: &protocol.SecurityConfig{
  749. Type: protocol.SecurityType_AES128_GCM,
  750. },
  751. }),
  752. },
  753. },
  754. },
  755. },
  756. }),
  757. },
  758. },
  759. }
  760. servers, err := InitializeServerConfigs(serverConfig)
  761. assert(err, IsNil)
  762. client, err := core.New(clientConfig)
  763. assert(err, IsNil)
  764. conn, err := core.Dial(context.Background(), client, dest)
  765. assert(err, IsNil)
  766. payload := "commander request."
  767. nBytes, err := conn.Write([]byte(payload))
  768. assert(err, IsNil)
  769. assert(nBytes, Equals, len(payload))
  770. response := make([]byte, 1024)
  771. nBytes, err = conn.Read(response)
  772. assert(err, IsNil)
  773. assert(response[:nBytes], Equals, xor([]byte(payload)))
  774. assert(conn.Close(), IsNil)
  775. CloseAllServers(servers)
  776. }