feature_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853
  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. Tag: "blocked",
  442. PortRange: net.SinglePortRange(dest2.Port),
  443. },
  444. },
  445. }),
  446. },
  447. }
  448. servers, err := InitializeServerConfigs(serverConfig)
  449. assert(err, IsNil)
  450. conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
  451. IP: []byte{127, 0, 0, 1},
  452. Port: int(serverPort2),
  453. })
  454. assert(err, IsNil)
  455. payload := "dokodemo request."
  456. {
  457. nBytes, err := conn.Write([]byte(payload))
  458. assert(err, IsNil)
  459. assert(nBytes, Equals, len(payload))
  460. }
  461. {
  462. response := make([]byte, 1024)
  463. _, err := conn.Read(response)
  464. assert(err, IsNotNil)
  465. }
  466. assert(conn.Close(), IsNil)
  467. CloseAllServers(servers)
  468. }
  469. func TestForward(t *testing.T) {
  470. assert := With(t)
  471. tcpServer := tcp.Server{
  472. MsgProcessor: xor,
  473. }
  474. dest, err := tcpServer.Start()
  475. assert(err, IsNil)
  476. defer tcpServer.Close()
  477. serverPort := tcp.PickPort()
  478. serverConfig := &core.Config{
  479. Inbound: []*core.InboundHandlerConfig{
  480. {
  481. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  482. PortRange: net.SinglePortRange(serverPort),
  483. Listen: net.NewIPOrDomain(net.LocalHostIP),
  484. }),
  485. ProxySettings: serial.ToTypedMessage(&socks.ServerConfig{
  486. AuthType: socks.AuthType_NO_AUTH,
  487. Accounts: map[string]string{
  488. "Test Account": "Test Password",
  489. },
  490. Address: net.NewIPOrDomain(net.LocalHostIP),
  491. UdpEnabled: false,
  492. }),
  493. },
  494. },
  495. Outbound: []*core.OutboundHandlerConfig{
  496. {
  497. ProxySettings: serial.ToTypedMessage(&freedom.Config{
  498. DestinationOverride: &freedom.DestinationOverride{
  499. Server: &protocol.ServerEndpoint{
  500. Address: net.NewIPOrDomain(net.LocalHostIP),
  501. Port: uint32(dest.Port),
  502. },
  503. },
  504. }),
  505. },
  506. },
  507. }
  508. servers, err := InitializeServerConfigs(serverConfig)
  509. assert(err, IsNil)
  510. {
  511. noAuthDialer, err := xproxy.SOCKS5("tcp", net.TCPDestination(net.LocalHostIP, serverPort).NetAddr(), nil, xproxy.Direct)
  512. assert(err, IsNil)
  513. conn, err := noAuthDialer.Dial("tcp", "google.com:80")
  514. assert(err, IsNil)
  515. payload := "test payload"
  516. nBytes, err := conn.Write([]byte(payload))
  517. assert(err, IsNil)
  518. assert(nBytes, Equals, len(payload))
  519. response := make([]byte, 1024)
  520. nBytes, err = conn.Read(response)
  521. assert(err, IsNil)
  522. assert(response[:nBytes], Equals, xor([]byte(payload)))
  523. assert(conn.Close(), IsNil)
  524. }
  525. CloseAllServers(servers)
  526. }
  527. func TestUDPConnection(t *testing.T) {
  528. assert := With(t)
  529. udpServer := udp.Server{
  530. MsgProcessor: xor,
  531. }
  532. dest, err := udpServer.Start()
  533. assert(err, IsNil)
  534. defer udpServer.Close()
  535. clientPort := tcp.PickPort()
  536. clientConfig := &core.Config{
  537. Inbound: []*core.InboundHandlerConfig{
  538. {
  539. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  540. PortRange: net.SinglePortRange(clientPort),
  541. Listen: net.NewIPOrDomain(net.LocalHostIP),
  542. }),
  543. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  544. Address: net.NewIPOrDomain(dest.Address),
  545. Port: uint32(dest.Port),
  546. NetworkList: &net.NetworkList{
  547. Network: []net.Network{net.Network_UDP},
  548. },
  549. }),
  550. },
  551. },
  552. Outbound: []*core.OutboundHandlerConfig{
  553. {
  554. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  555. },
  556. },
  557. }
  558. servers, err := InitializeServerConfigs(clientConfig)
  559. assert(err, IsNil)
  560. {
  561. conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
  562. IP: []byte{127, 0, 0, 1},
  563. Port: int(clientPort),
  564. })
  565. assert(err, IsNil)
  566. payload := "dokodemo request."
  567. for i := 0; i < 5; i++ {
  568. nBytes, err := conn.Write([]byte(payload))
  569. assert(err, IsNil)
  570. assert(nBytes, Equals, len(payload))
  571. response := make([]byte, 1024)
  572. nBytes, err = conn.Read(response)
  573. assert(err, IsNil)
  574. assert(response[:nBytes], Equals, xor([]byte(payload)))
  575. }
  576. assert(conn.Close(), IsNil)
  577. }
  578. time.Sleep(20 * time.Second)
  579. {
  580. conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
  581. IP: []byte{127, 0, 0, 1},
  582. Port: int(clientPort),
  583. })
  584. assert(err, IsNil)
  585. payload := "dokodemo request."
  586. nBytes, err := conn.Write([]byte(payload))
  587. assert(err, IsNil)
  588. assert(nBytes, Equals, len(payload))
  589. response := make([]byte, 1024)
  590. nBytes, err = conn.Read(response)
  591. assert(err, IsNil)
  592. assert(response[:nBytes], Equals, xor([]byte(payload)))
  593. assert(conn.Close(), IsNil)
  594. }
  595. CloseAllServers(servers)
  596. }
  597. func TestDomainSniffing(t *testing.T) {
  598. assert := With(t)
  599. sniffingPort := tcp.PickPort()
  600. httpPort := tcp.PickPort()
  601. serverConfig := &core.Config{
  602. Inbound: []*core.InboundHandlerConfig{
  603. {
  604. Tag: "snif",
  605. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  606. PortRange: net.SinglePortRange(sniffingPort),
  607. Listen: net.NewIPOrDomain(net.LocalHostIP),
  608. DomainOverride: []proxyman.KnownProtocols{
  609. proxyman.KnownProtocols_TLS,
  610. },
  611. }),
  612. ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
  613. Address: net.NewIPOrDomain(net.LocalHostIP),
  614. Port: 443,
  615. NetworkList: &net.NetworkList{
  616. Network: []net.Network{net.Network_TCP},
  617. },
  618. }),
  619. },
  620. {
  621. Tag: "http",
  622. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  623. PortRange: net.SinglePortRange(httpPort),
  624. Listen: net.NewIPOrDomain(net.LocalHostIP),
  625. }),
  626. ProxySettings: serial.ToTypedMessage(&v2http.ServerConfig{}),
  627. },
  628. },
  629. Outbound: []*core.OutboundHandlerConfig{
  630. {
  631. Tag: "redir",
  632. ProxySettings: serial.ToTypedMessage(&freedom.Config{
  633. DestinationOverride: &freedom.DestinationOverride{
  634. Server: &protocol.ServerEndpoint{
  635. Address: net.NewIPOrDomain(net.LocalHostIP),
  636. Port: uint32(sniffingPort),
  637. },
  638. },
  639. }),
  640. },
  641. {
  642. Tag: "direct",
  643. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  644. },
  645. },
  646. App: []*serial.TypedMessage{
  647. serial.ToTypedMessage(&router.Config{
  648. Rule: []*router.RoutingRule{
  649. {
  650. Tag: "direct",
  651. InboundTag: []string{"snif"},
  652. }, {
  653. Tag: "redir",
  654. InboundTag: []string{"http"},
  655. },
  656. },
  657. }),
  658. serial.ToTypedMessage(&log.Config{
  659. ErrorLogLevel: clog.Severity_Debug,
  660. ErrorLogType: log.LogType_Console,
  661. }),
  662. },
  663. }
  664. servers, err := InitializeServerConfigs(serverConfig)
  665. assert(err, IsNil)
  666. {
  667. transport := &http.Transport{
  668. Proxy: func(req *http.Request) (*url.URL, error) {
  669. return url.Parse("http://127.0.0.1:" + httpPort.String())
  670. },
  671. }
  672. client := &http.Client{
  673. Transport: transport,
  674. }
  675. resp, err := client.Get("https://www.github.com/")
  676. assert(err, IsNil)
  677. assert(resp.StatusCode, Equals, 200)
  678. assert(resp.Write(ioutil.Discard), IsNil)
  679. }
  680. CloseAllServers(servers)
  681. }
  682. func TestDialV2Ray(t *testing.T) {
  683. assert := With(t)
  684. tcpServer := tcp.Server{
  685. MsgProcessor: xor,
  686. }
  687. dest, err := tcpServer.Start()
  688. assert(err, IsNil)
  689. defer tcpServer.Close()
  690. userID := protocol.NewID(uuid.New())
  691. serverPort := tcp.PickPort()
  692. serverConfig := &core.Config{
  693. App: []*serial.TypedMessage{
  694. serial.ToTypedMessage(&log.Config{
  695. ErrorLogLevel: clog.Severity_Debug,
  696. ErrorLogType: log.LogType_Console,
  697. }),
  698. },
  699. Inbound: []*core.InboundHandlerConfig{
  700. {
  701. ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
  702. PortRange: net.SinglePortRange(serverPort),
  703. Listen: net.NewIPOrDomain(net.LocalHostIP),
  704. }),
  705. ProxySettings: serial.ToTypedMessage(&inbound.Config{
  706. User: []*protocol.User{
  707. {
  708. Account: serial.ToTypedMessage(&vmess.Account{
  709. Id: userID.String(),
  710. AlterId: 64,
  711. }),
  712. },
  713. },
  714. }),
  715. },
  716. },
  717. Outbound: []*core.OutboundHandlerConfig{
  718. {
  719. ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
  720. },
  721. },
  722. }
  723. clientConfig := &core.Config{
  724. App: []*serial.TypedMessage{
  725. serial.ToTypedMessage(&dispatcher.Config{}),
  726. serial.ToTypedMessage(&proxyman.InboundConfig{}),
  727. serial.ToTypedMessage(&proxyman.OutboundConfig{}),
  728. },
  729. Inbound: []*core.InboundHandlerConfig{},
  730. Outbound: []*core.OutboundHandlerConfig{
  731. {
  732. ProxySettings: serial.ToTypedMessage(&outbound.Config{
  733. Receiver: []*protocol.ServerEndpoint{
  734. {
  735. Address: net.NewIPOrDomain(net.LocalHostIP),
  736. Port: uint32(serverPort),
  737. User: []*protocol.User{
  738. {
  739. Account: serial.ToTypedMessage(&vmess.Account{
  740. Id: userID.String(),
  741. AlterId: 64,
  742. SecuritySettings: &protocol.SecurityConfig{
  743. Type: protocol.SecurityType_AES128_GCM,
  744. },
  745. }),
  746. },
  747. },
  748. },
  749. },
  750. }),
  751. },
  752. },
  753. }
  754. servers, err := InitializeServerConfigs(serverConfig)
  755. assert(err, IsNil)
  756. client, err := core.New(clientConfig)
  757. assert(err, IsNil)
  758. conn, err := core.Dial(context.Background(), client, dest)
  759. assert(err, IsNil)
  760. payload := "commander request."
  761. nBytes, err := conn.Write([]byte(payload))
  762. assert(err, IsNil)
  763. assert(nBytes, Equals, len(payload))
  764. response := make([]byte, 1024)
  765. nBytes, err = conn.Read(response)
  766. assert(err, IsNil)
  767. assert(response[:nBytes], Equals, xor([]byte(payload)))
  768. assert(conn.Close(), IsNil)
  769. CloseAllServers(servers)
  770. }