dialParameters_test.go 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276
  1. /*
  2. * Copyright (c) 2018, Psiphon Inc.
  3. * All rights reserved.
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. */
  19. package psiphon
  20. import (
  21. "bytes"
  22. "context"
  23. "encoding/json"
  24. "fmt"
  25. "io"
  26. "io/ioutil"
  27. "os"
  28. "reflect"
  29. "testing"
  30. "time"
  31. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
  32. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/errors"
  33. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/parameters"
  34. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/prng"
  35. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/protocol"
  36. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/transforms"
  37. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/values"
  38. lrucache "github.com/cognusion/go-cache-lru"
  39. )
  40. func TestDialParametersAndReplay(t *testing.T) {
  41. for _, tunnelProtocol := range protocol.SupportedTunnelProtocols {
  42. if !common.Contains(protocol.DefaultDisabledTunnelProtocols, tunnelProtocol) {
  43. runDialParametersAndReplay(t, tunnelProtocol)
  44. }
  45. }
  46. }
  47. var testNetworkID = prng.HexString(8)
  48. type testNetworkGetter struct {
  49. }
  50. func (t *testNetworkGetter) GetNetworkID() string {
  51. return testNetworkID
  52. }
  53. func runDialParametersAndReplay(t *testing.T, tunnelProtocol string) {
  54. t.Logf("Test %s...", tunnelProtocol)
  55. testDataDirName, err := ioutil.TempDir("", "psiphon-dial-parameters-test")
  56. if err != nil {
  57. t.Fatalf("TempDir failed: %s", err)
  58. }
  59. defer os.RemoveAll(testDataDirName)
  60. err = SetNoticeWriter(io.Discard)
  61. if err != nil {
  62. t.Fatalf("error setting notice writer: %s", err)
  63. }
  64. defer ResetNoticeWriter()
  65. clientConfig := &Config{
  66. PropagationChannelId: "0",
  67. SponsorId: "0",
  68. DataRootDirectory: testDataDirName,
  69. NetworkIDGetter: new(testNetworkGetter),
  70. }
  71. err = clientConfig.Commit(false)
  72. if err != nil {
  73. t.Fatalf("error committing configuration file: %s", err)
  74. }
  75. holdOffTunnelProtocolNames := protocol.TunnelProtocols{protocol.TUNNEL_PROTOCOL_OBFUSCATED_SSH}
  76. providerID := prng.HexString(8)
  77. frontingProviderID := prng.HexString(8)
  78. var holdOffDirectTunnelProviderRegions parameters.KeyStrings
  79. if tunnelProtocol == protocol.TUNNEL_PROTOCOL_UNFRONTED_MEEK {
  80. holdOffDirectTunnelProviderRegions = map[string][]string{providerID: {""}}
  81. }
  82. var holdOffInproxyTunnelProviderRegions parameters.KeyStrings
  83. if protocol.TunnelProtocolUsesInproxy(tunnelProtocol) &&
  84. protocol.TunnelProtocolMinusInproxy(tunnelProtocol) == protocol.TUNNEL_PROTOCOL_OBFUSCATED_SSH {
  85. holdOffInproxyTunnelProviderRegions = map[string][]string{providerID: {""}}
  86. }
  87. applyParameters := make(map[string]interface{})
  88. applyParameters[parameters.TransformHostNameProbability] = 1.0
  89. applyParameters[parameters.PickUserAgentProbability] = 1.0
  90. applyParameters[parameters.HoldOffTunnelProtocolMinDuration] = "1ms"
  91. applyParameters[parameters.HoldOffTunnelProtocolMaxDuration] = "10ms"
  92. applyParameters[parameters.HoldOffTunnelProtocolNames] = holdOffTunnelProtocolNames
  93. applyParameters[parameters.HoldOffTunnelProtocolProbability] = 1.0
  94. applyParameters[parameters.HoldOffFrontingTunnelMinDuration] = "1ms"
  95. applyParameters[parameters.HoldOffFrontingTunnelMaxDuration] = "10ms"
  96. applyParameters[parameters.HoldOffFrontingTunnelProviderIDs] = []string{frontingProviderID}
  97. applyParameters[parameters.HoldOffFrontingTunnelProbability] = 1.0
  98. applyParameters[parameters.HoldOffDirectTunnelMinDuration] = "1ms"
  99. applyParameters[parameters.HoldOffDirectTunnelMaxDuration] = "10ms"
  100. applyParameters[parameters.HoldOffDirectTunnelProviderRegions] = holdOffDirectTunnelProviderRegions
  101. applyParameters[parameters.HoldOffInproxyTunnelProbability] = 1.0
  102. applyParameters[parameters.HoldOffInproxyTunnelMinDuration] = "1ms"
  103. applyParameters[parameters.HoldOffInproxyTunnelMaxDuration] = "10ms"
  104. applyParameters[parameters.HoldOffInproxyTunnelProviderRegions] = holdOffInproxyTunnelProviderRegions
  105. applyParameters[parameters.HoldOffDirectTunnelProbability] = 1.0
  106. applyParameters[parameters.DNSResolverAlternateServers] = []string{"127.0.0.1", "127.0.0.2", "127.0.0.3"}
  107. applyParameters[parameters.DirectHTTPProtocolTransformProbability] = 1.0
  108. applyParameters[parameters.DirectHTTPProtocolTransformSpecs] = transforms.Specs{"spec": transforms.Spec{{"", ""}}}
  109. applyParameters[parameters.DirectHTTPProtocolTransformScopedSpecNames] = transforms.ScopedSpecNames{"": {"spec"}}
  110. applyParameters[parameters.OSSHObfuscatorSeedTransformProbability] = 1.0
  111. applyParameters[parameters.OSSHObfuscatorSeedTransformSpecs] = transforms.Specs{"spec": transforms.Spec{{"", ""}}}
  112. applyParameters[parameters.OSSHObfuscatorSeedTransformScopedSpecNames] = transforms.ScopedSpecNames{"": {"spec"}}
  113. applyParameters[parameters.ObfuscatedQUICNonceTransformProbability] = 1.0
  114. applyParameters[parameters.ObfuscatedQUICNonceTransformSpecs] = transforms.Specs{"spec": transforms.Spec{{"", ""}}}
  115. applyParameters[parameters.ObfuscatedQUICNonceTransformScopedSpecNames] = transforms.ScopedSpecNames{"": {"spec"}}
  116. applyParameters[parameters.ShadowsocksPrefixProbability] = 1.0
  117. applyParameters[parameters.ShadowsocksPrefixSpecs] = transforms.Specs{"spec": transforms.Spec{{"", ""}}}
  118. applyParameters[parameters.ShadowsocksPrefixScopedSpecNames] = transforms.ScopedSpecNames{"": {"spec"}}
  119. err = clientConfig.SetParameters("tag1", false, applyParameters)
  120. if err != nil {
  121. t.Fatalf("SetParameters failed: %s", err)
  122. }
  123. resolver := NewResolver(clientConfig, true)
  124. defer resolver.Stop()
  125. clientConfig.SetResolver(resolver)
  126. err = OpenDataStore(clientConfig)
  127. if err != nil {
  128. t.Fatalf("error initializing client datastore: %s", err)
  129. }
  130. defer CloseDataStore()
  131. serverEntries := makeMockServerEntries(tunnelProtocol, "CA", providerID, frontingProviderID, 100)
  132. canReplay := func(serverEntry *protocol.ServerEntry, replayProtocol string) bool {
  133. return replayProtocol == tunnelProtocol
  134. }
  135. selectProtocol := func(serverEntry *protocol.ServerEntry) (string, bool) {
  136. return tunnelProtocol, true
  137. }
  138. values.SetSSHClientVersionsSpec(
  139. values.NewPickOneSpec([]string{"SSH-2.0-A", "SSH-2.0-B", "SSH-2.0-C"}))
  140. values.SetUserAgentsSpec(
  141. values.NewPickOneSpec([]string{"ua1", "ua2", "ua3"}))
  142. // Test: expected dial parameter fields set
  143. steeringIPCache := lrucache.NewWithLRU(1*time.Hour, 1*time.Hour, 0)
  144. upstreamProxyErrorCallback := func(_ error) {}
  145. dialParams, err := MakeDialParameters(
  146. clientConfig, steeringIPCache, nil, nil, upstreamProxyErrorCallback, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  147. if err != nil {
  148. t.Fatalf("MakeDialParameters failed: %s", err)
  149. }
  150. if dialParams.ServerEntry != serverEntries[0] {
  151. t.Fatalf("unexpected server entry")
  152. }
  153. if dialParams.NetworkID != testNetworkID {
  154. t.Fatalf("unexpected network ID")
  155. }
  156. if dialParams.IsReplay {
  157. t.Fatalf("unexpected replay")
  158. }
  159. if dialParams.TunnelProtocol != tunnelProtocol {
  160. t.Fatalf("unexpected tunnel protocol")
  161. }
  162. if !protocol.TunnelProtocolUsesMeek(tunnelProtocol) &&
  163. dialParams.DirectDialAddress == "" {
  164. t.Fatalf("missing direct dial fields")
  165. }
  166. if dialParams.DialPortNumber == "" {
  167. t.Fatalf("missing port number fields")
  168. }
  169. if !dialParams.SelectedSSHClientVersion || dialParams.SSHClientVersion == "" || dialParams.SSHKEXSeed == nil {
  170. t.Fatalf("missing SSH fields")
  171. }
  172. if protocol.TunnelProtocolUsesObfuscatedSSH(tunnelProtocol) &&
  173. dialParams.ObfuscatorPaddingSeed == nil {
  174. t.Fatalf("missing obfuscator fields")
  175. }
  176. if dialParams.FragmentorSeed == nil {
  177. t.Fatalf("missing fragmentor field")
  178. }
  179. if protocol.TunnelProtocolUsesMeek(tunnelProtocol) &&
  180. (dialParams.MeekDialAddress == "" ||
  181. dialParams.MeekHostHeader == "" ||
  182. dialParams.MeekObfuscatorPaddingSeed == nil) {
  183. t.Fatalf("missing meek fields")
  184. }
  185. if protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol) &&
  186. (dialParams.MeekFrontingDialAddress == "" ||
  187. dialParams.MeekFrontingHost == "" ||
  188. dialParams.ResolveParameters == nil) {
  189. t.Fatalf("missing meek fronting fields")
  190. }
  191. if protocol.TunnelProtocolUsesMeekHTTP(tunnelProtocol) &&
  192. dialParams.UserAgent == "" {
  193. t.Fatalf("missing meek HTTP fields")
  194. }
  195. if protocol.TunnelProtocolUsesMeekHTTPS(tunnelProtocol) &&
  196. (dialParams.MeekSNIServerName == "" ||
  197. !dialParams.SelectedTLSProfile ||
  198. dialParams.TLSProfile == "") {
  199. t.Fatalf("missing meek HTTPS fields")
  200. }
  201. if protocol.TunnelProtocolUsesQUIC(tunnelProtocol) {
  202. if dialParams.QUICVersion == "" {
  203. t.Fatalf("missing QUIC version field")
  204. }
  205. if protocol.TunnelProtocolUsesFrontedMeekQUIC(tunnelProtocol) {
  206. if dialParams.MeekFrontingDialAddress == "" ||
  207. dialParams.MeekFrontingHost == "" ||
  208. dialParams.MeekSNIServerName == "" {
  209. t.Fatalf("missing fronted QUIC fields")
  210. }
  211. } else {
  212. if dialParams.QUICDialSNIAddress == "" {
  213. t.Fatalf("missing QUIC SNI field")
  214. }
  215. }
  216. }
  217. if dialParams.LivenessTestSeed == nil {
  218. t.Fatalf("missing liveness test fields")
  219. }
  220. if dialParams.APIRequestPaddingSeed == nil {
  221. t.Fatalf("missing API request fields")
  222. }
  223. expectHoldOffTunnelProtocolNames := common.Contains(holdOffTunnelProtocolNames, tunnelProtocol)
  224. expectHoldOffFrontingTunnelProviderIDs := protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol)
  225. expectHoldOffDirectTunnelProviderRegion := protocol.TunnelProtocolIsDirect(tunnelProtocol) &&
  226. common.ContainsAny(
  227. holdOffDirectTunnelProviderRegions[dialParams.ServerEntry.ProviderID],
  228. []string{"", dialParams.ServerEntry.Region})
  229. expectHoldOffInproxyTunnelProviderRegion := protocol.TunnelProtocolUsesInproxy(tunnelProtocol) &&
  230. common.ContainsAny(
  231. holdOffInproxyTunnelProviderRegions[dialParams.ServerEntry.ProviderID],
  232. []string{"", dialParams.ServerEntry.Region})
  233. if expectHoldOffTunnelProtocolNames ||
  234. expectHoldOffFrontingTunnelProviderIDs ||
  235. expectHoldOffDirectTunnelProviderRegion ||
  236. expectHoldOffInproxyTunnelProviderRegion {
  237. if dialParams.HoldOffTunnelDuration < 1*time.Millisecond ||
  238. dialParams.HoldOffTunnelDuration > 10*time.Millisecond {
  239. t.Fatalf("unexpected hold-off duration: %v", dialParams.HoldOffTunnelDuration)
  240. }
  241. } else {
  242. if dialParams.HoldOffTunnelDuration != 0 {
  243. t.Fatalf("unexpected hold-off duration: %v", dialParams.HoldOffTunnelDuration)
  244. }
  245. }
  246. dialConfig := dialParams.GetDialConfig()
  247. if dialConfig.UpstreamProxyErrorCallback == nil {
  248. t.Fatalf("missing upstreamProxyErrorCallback")
  249. }
  250. // Test: no replay after dial reported to fail
  251. dialErr := errors.TraceNew("dial error")
  252. dialParams.Failed(clientConfig, dialErr)
  253. dialParams, err = MakeDialParameters(
  254. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  255. if err != nil {
  256. t.Fatalf("MakeDialParameters failed: %s", err)
  257. }
  258. if dialParams.IsReplay {
  259. t.Fatalf("unexpected replay")
  260. }
  261. if dialParams.ReplayIgnoredChange {
  262. t.Fatalf("unexpected replay ignored change")
  263. }
  264. // Test: no replay after network ID changes
  265. dialParams.Succeeded()
  266. testNetworkID = prng.HexString(8)
  267. clientConfig.networkIDGetter.FlushCache()
  268. dialParams, err = MakeDialParameters(
  269. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  270. if err != nil {
  271. t.Fatalf("MakeDialParameters failed: %s", err)
  272. }
  273. if dialParams.NetworkID != testNetworkID {
  274. t.Fatalf("unexpected network ID")
  275. }
  276. if dialParams.IsReplay {
  277. t.Fatalf("unexpected replay")
  278. }
  279. if dialParams.ReplayIgnoredChange {
  280. t.Fatalf("unexpected replay ignored change")
  281. }
  282. // Test: replay after dial reported to succeed, and replay fields match previous dial parameters
  283. dialParams.Succeeded()
  284. replayDialParams, err := MakeDialParameters(
  285. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  286. if err != nil {
  287. t.Fatalf("MakeDialParameters failed: %s", err)
  288. }
  289. if !replayDialParams.IsReplay {
  290. t.Fatalf("unexpected non-replay")
  291. }
  292. if replayDialParams.ReplayIgnoredChange {
  293. t.Fatalf("unexpected replay ignored change")
  294. }
  295. if !replayDialParams.LastUsedTimestamp.After(dialParams.LastUsedTimestamp) {
  296. t.Fatalf("unexpected non-updated timestamp")
  297. }
  298. if replayDialParams.TunnelProtocol != dialParams.TunnelProtocol {
  299. t.Fatalf("mismatching tunnel protocol")
  300. }
  301. if replayDialParams.DirectDialAddress != dialParams.DirectDialAddress ||
  302. replayDialParams.DialPortNumber != dialParams.DialPortNumber {
  303. t.Fatalf("mismatching dial fields")
  304. }
  305. identicalSeeds := func(seed1, seed2 *prng.Seed) bool {
  306. if seed1 == nil {
  307. return seed2 == nil
  308. }
  309. return bytes.Equal(seed1[:], seed2[:])
  310. }
  311. if replayDialParams.SelectedSSHClientVersion != dialParams.SelectedSSHClientVersion ||
  312. replayDialParams.SSHClientVersion != dialParams.SSHClientVersion ||
  313. !identicalSeeds(replayDialParams.SSHKEXSeed, dialParams.SSHKEXSeed) {
  314. t.Fatalf("mismatching SSH fields")
  315. }
  316. if !identicalSeeds(replayDialParams.ObfuscatorPaddingSeed, dialParams.ObfuscatorPaddingSeed) {
  317. t.Fatalf("mismatching obfuscator fields")
  318. }
  319. if !identicalSeeds(replayDialParams.FragmentorSeed, dialParams.FragmentorSeed) {
  320. t.Fatalf("mismatching fragmentor fields")
  321. }
  322. if replayDialParams.MeekFrontingDialAddress != dialParams.MeekFrontingDialAddress ||
  323. replayDialParams.MeekFrontingHost != dialParams.MeekFrontingHost ||
  324. replayDialParams.MeekDialAddress != dialParams.MeekDialAddress ||
  325. replayDialParams.MeekTransformedHostName != dialParams.MeekTransformedHostName ||
  326. replayDialParams.MeekSNIServerName != dialParams.MeekSNIServerName ||
  327. replayDialParams.MeekHostHeader != dialParams.MeekHostHeader ||
  328. !identicalSeeds(replayDialParams.MeekObfuscatorPaddingSeed, dialParams.MeekObfuscatorPaddingSeed) {
  329. t.Fatalf("mismatching meek fields")
  330. }
  331. if replayDialParams.SelectedUserAgent != dialParams.SelectedUserAgent ||
  332. replayDialParams.UserAgent != dialParams.UserAgent {
  333. t.Fatalf("mismatching user agent fields")
  334. }
  335. if replayDialParams.SelectedTLSProfile != dialParams.SelectedTLSProfile ||
  336. replayDialParams.TLSProfile != dialParams.TLSProfile ||
  337. !identicalSeeds(replayDialParams.RandomizedTLSProfileSeed, dialParams.RandomizedTLSProfileSeed) {
  338. t.Fatalf("mismatching TLS fields")
  339. }
  340. if replayDialParams.QUICVersion != dialParams.QUICVersion ||
  341. replayDialParams.QUICDialSNIAddress != dialParams.QUICDialSNIAddress ||
  342. !identicalSeeds(replayDialParams.ObfuscatedQUICPaddingSeed, dialParams.ObfuscatedQUICPaddingSeed) {
  343. t.Fatalf("mismatching QUIC fields")
  344. }
  345. if !identicalSeeds(replayDialParams.LivenessTestSeed, dialParams.LivenessTestSeed) {
  346. t.Fatalf("mismatching liveness test fields")
  347. }
  348. if !identicalSeeds(replayDialParams.APIRequestPaddingSeed, dialParams.APIRequestPaddingSeed) {
  349. t.Fatalf("mismatching API request fields")
  350. }
  351. if (replayDialParams.ResolveParameters == nil) != (dialParams.ResolveParameters == nil) ||
  352. (replayDialParams.ResolveParameters != nil &&
  353. !reflect.DeepEqual(replayDialParams.ResolveParameters, dialParams.ResolveParameters)) {
  354. t.Fatalf("mismatching ResolveParameters fields")
  355. }
  356. if (replayDialParams.HTTPTransformerParameters == nil) != (dialParams.HTTPTransformerParameters == nil) ||
  357. (replayDialParams.HTTPTransformerParameters != nil &&
  358. !reflect.DeepEqual(replayDialParams.HTTPTransformerParameters, dialParams.HTTPTransformerParameters)) {
  359. t.Fatalf("mismatching HTTPTransformerParameters fields")
  360. }
  361. if (replayDialParams.OSSHObfuscatorSeedTransformerParameters == nil) != (dialParams.OSSHObfuscatorSeedTransformerParameters == nil) ||
  362. (replayDialParams.OSSHObfuscatorSeedTransformerParameters != nil &&
  363. !reflect.DeepEqual(replayDialParams.OSSHObfuscatorSeedTransformerParameters, dialParams.OSSHObfuscatorSeedTransformerParameters)) {
  364. t.Fatalf("mismatching ObfuscatorSeedTransformerParameters fields")
  365. }
  366. if (replayDialParams.ObfuscatedQUICNonceTransformerParameters == nil) != (dialParams.ObfuscatedQUICNonceTransformerParameters == nil) ||
  367. (replayDialParams.ObfuscatedQUICNonceTransformerParameters != nil &&
  368. !reflect.DeepEqual(replayDialParams.ObfuscatedQUICNonceTransformerParameters, dialParams.ObfuscatedQUICNonceTransformerParameters)) {
  369. t.Fatalf("mismatching ObfuscatedQUICNonceTransformerParameters fields")
  370. }
  371. if (replayDialParams.ShadowsocksPrefixSpec == nil) != (dialParams.ShadowsocksPrefixSpec == nil) ||
  372. (replayDialParams.ShadowsocksPrefixSpec != nil &&
  373. !reflect.DeepEqual(replayDialParams.ShadowsocksPrefixSpec, dialParams.ShadowsocksPrefixSpec)) {
  374. t.Fatalf("mismatching ShadowsocksPrefixSpec fields")
  375. }
  376. // Test: replay after change tactics, with ReplayIgnoreChangedConfigStateProbability = 1.0
  377. applyParameters[parameters.ReplayDialParametersTTL] = "1s"
  378. applyParameters[parameters.ReplayIgnoreChangedConfigStateProbability] = 1.0
  379. err = clientConfig.SetParameters("tag2a", false, applyParameters)
  380. if err != nil {
  381. t.Fatalf("SetParameters failed: %s", err)
  382. }
  383. dialParams, err = MakeDialParameters(
  384. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  385. if err != nil {
  386. t.Fatalf("MakeDialParameters failed: %s", err)
  387. }
  388. if !dialParams.IsReplay {
  389. t.Fatalf("unexpected non-replay")
  390. }
  391. if !dialParams.ReplayIgnoredChange {
  392. t.Fatalf("unexpected replay ignored change")
  393. }
  394. // Test: no replay after change tactics
  395. applyParameters[parameters.ReplayDialParametersTTL] = "1s"
  396. applyParameters[parameters.ReplayIgnoreChangedConfigStateProbability] = 0.0
  397. err = clientConfig.SetParameters("tag2", false, applyParameters)
  398. if err != nil {
  399. t.Fatalf("SetParameters failed: %s", err)
  400. }
  401. dialParams, err = MakeDialParameters(
  402. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  403. if err != nil {
  404. t.Fatalf("MakeDialParameters failed: %s", err)
  405. }
  406. if dialParams.IsReplay {
  407. t.Fatalf("unexpected replay")
  408. }
  409. if dialParams.ReplayIgnoredChange {
  410. t.Fatalf("unexpected replay ignored change")
  411. }
  412. // Test: no replay after dial parameters expired
  413. dialParams.Succeeded()
  414. time.Sleep(1 * time.Second)
  415. dialParams, err = MakeDialParameters(
  416. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  417. if err != nil {
  418. t.Fatalf("MakeDialParameters failed: %s", err)
  419. }
  420. if dialParams.IsReplay {
  421. t.Fatalf("unexpected replay")
  422. }
  423. if dialParams.ReplayIgnoredChange {
  424. t.Fatalf("unexpected replay ignored change")
  425. }
  426. // Test: no replay after server entry changes
  427. dialParams.Succeeded()
  428. serverEntries[0].ConfigurationVersion += 1
  429. dialParams, err = MakeDialParameters(
  430. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  431. if err != nil {
  432. t.Fatalf("MakeDialParameters failed: %s", err)
  433. }
  434. if dialParams.IsReplay {
  435. t.Fatalf("unexpected replay")
  436. }
  437. if dialParams.ReplayIgnoredChange {
  438. t.Fatalf("unexpected replay ignored change")
  439. }
  440. // Test: disable replay elements (partial coverage)
  441. applyParameters[parameters.ReplayDialParametersTTL] = "24h"
  442. applyParameters[parameters.ReplaySSH] = false
  443. applyParameters[parameters.ReplayObfuscatorPadding] = false
  444. applyParameters[parameters.ReplayFragmentor] = false
  445. applyParameters[parameters.ReplayObfuscatedQUIC] = false
  446. applyParameters[parameters.ReplayLivenessTest] = false
  447. applyParameters[parameters.ReplayAPIRequestPadding] = false
  448. err = clientConfig.SetParameters("tag3", false, applyParameters)
  449. if err != nil {
  450. t.Fatalf("SetParameters failed: %s", err)
  451. }
  452. dialParams, err = MakeDialParameters(
  453. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  454. if err != nil {
  455. t.Fatalf("MakeDialParameters failed: %s", err)
  456. }
  457. dialParams.Succeeded()
  458. replayDialParams, err = MakeDialParameters(
  459. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  460. if err != nil {
  461. t.Fatalf("MakeDialParameters failed: %s", err)
  462. }
  463. if !replayDialParams.IsReplay {
  464. t.Fatalf("unexpected non-replay")
  465. }
  466. if identicalSeeds(replayDialParams.SSHKEXSeed, dialParams.SSHKEXSeed) ||
  467. (protocol.TunnelProtocolUsesObfuscatedSSH(tunnelProtocol) &&
  468. identicalSeeds(replayDialParams.ObfuscatorPaddingSeed, dialParams.ObfuscatorPaddingSeed)) ||
  469. identicalSeeds(replayDialParams.FragmentorSeed, dialParams.FragmentorSeed) ||
  470. (protocol.TunnelProtocolUsesMeek(tunnelProtocol) &&
  471. identicalSeeds(replayDialParams.MeekObfuscatorPaddingSeed, dialParams.MeekObfuscatorPaddingSeed)) ||
  472. (protocol.TunnelProtocolUsesQUIC(tunnelProtocol) &&
  473. identicalSeeds(replayDialParams.ObfuscatedQUICPaddingSeed, dialParams.ObfuscatedQUICPaddingSeed) &&
  474. replayDialParams.ObfuscatedQUICPaddingSeed != nil) ||
  475. identicalSeeds(replayDialParams.LivenessTestSeed, dialParams.LivenessTestSeed) ||
  476. identicalSeeds(replayDialParams.APIRequestPaddingSeed, dialParams.APIRequestPaddingSeed) {
  477. t.Fatalf("unexpected replayed fields")
  478. }
  479. // Test: client-side restrict fronting provider ID
  480. applyParameters[parameters.RestrictFrontingProviderIDs] = []string{frontingProviderID}
  481. applyParameters[parameters.RestrictFrontingProviderIDsClientProbability] = 1.0
  482. err = clientConfig.SetParameters("tag4", false, applyParameters)
  483. if err != nil {
  484. t.Fatalf("SetParameters failed: %s", err)
  485. }
  486. dialParams, err = MakeDialParameters(
  487. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  488. if protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol) {
  489. if err == nil {
  490. if dialParams != nil {
  491. t.Fatalf("unexpected MakeDialParameters success")
  492. }
  493. }
  494. } else {
  495. if err != nil {
  496. t.Fatalf("MakeDialParameters failed: %s", err)
  497. }
  498. }
  499. applyParameters[parameters.RestrictFrontingProviderIDsClientProbability] = 0.0
  500. err = clientConfig.SetParameters("tag5", false, applyParameters)
  501. if err != nil {
  502. t.Fatalf("SetParameters failed: %s", err)
  503. }
  504. // Test: client-side restrict provider ID by region for direct protocols
  505. applyParameters[parameters.RestrictDirectProviderRegions] = map[string][]string{providerID: {"CA"}}
  506. applyParameters[parameters.RestrictDirectProviderIDsClientProbability] = 1.0
  507. err = clientConfig.SetParameters("tag6", false, applyParameters)
  508. if err != nil {
  509. t.Fatalf("SetParameters failed: %s", err)
  510. }
  511. dialParams, err = MakeDialParameters(
  512. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  513. if protocol.TunnelProtocolIsDirect(tunnelProtocol) {
  514. if err == nil {
  515. if dialParams != nil {
  516. t.Fatalf("unexpected MakeDialParameters success")
  517. }
  518. }
  519. } else {
  520. if err != nil {
  521. t.Fatalf("MakeDialParameters failed: %s", err)
  522. }
  523. }
  524. applyParameters[parameters.RestrictDirectProviderRegions] = map[string][]string{}
  525. applyParameters[parameters.RestrictDirectProviderIDsClientProbability] = 0.0
  526. err = clientConfig.SetParameters("tag7", false, applyParameters)
  527. if err != nil {
  528. t.Fatalf("SetParameters failed: %s", err)
  529. }
  530. // Test: client-side restrict provider ID by region for inproxy protocols
  531. applyParameters[parameters.RestrictInproxyProviderRegions] = map[string][]string{providerID: {"CA"}}
  532. applyParameters[parameters.RestrictInproxyProviderIDsClientProbability] = 1.0
  533. err = clientConfig.SetParameters("tag8", false, applyParameters)
  534. if err != nil {
  535. t.Fatalf("SetParameters failed: %s", err)
  536. }
  537. dialParams, err = MakeDialParameters(
  538. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  539. if protocol.TunnelProtocolUsesInproxy(tunnelProtocol) {
  540. if err == nil {
  541. if dialParams != nil {
  542. t.Fatalf("unexpected MakeDialParameters success")
  543. }
  544. }
  545. } else {
  546. if err != nil {
  547. t.Fatalf("MakeDialParameters failed: %s", err)
  548. }
  549. }
  550. applyParameters[parameters.RestrictInproxyProviderRegions] = map[string][]string{}
  551. applyParameters[parameters.RestrictInproxyProviderIDsClientProbability] = 0.0
  552. err = clientConfig.SetParameters("tag9", false, applyParameters)
  553. if err != nil {
  554. t.Fatalf("SetParameters failed: %s", err)
  555. }
  556. if protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol) {
  557. steeringIPCache.Flush()
  558. // Test: steering IP used in non-replay case
  559. dialParams, err = MakeDialParameters(
  560. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  561. if err != nil {
  562. t.Fatalf("MakeDialParameters failed: %s", err)
  563. }
  564. dialParams.Failed(clientConfig, dialErr)
  565. getCacheKey := func() string {
  566. return fmt.Sprintf("%s %s %s", testNetworkID, frontingProviderID, tunnelProtocol)
  567. }
  568. setCacheEntry := func(steeringIP string) {
  569. steeringIPCache.Set(getCacheKey(), steeringIP, lrucache.DefaultExpiration)
  570. }
  571. setCacheEntry("127.0.0.1")
  572. dialParams, err = MakeDialParameters(
  573. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  574. if err != nil {
  575. t.Fatalf("MakeDialParameters failed: %s", err)
  576. }
  577. if dialParams.IsReplay {
  578. t.Fatalf("unexpected replay")
  579. }
  580. checkSteeringIP := func(expectedSteeringIP string) {
  581. ctx, cancelFunc := context.WithTimeout(context.Background(), 1*time.Microsecond)
  582. defer cancelFunc()
  583. IPs, err := dialParams.dialConfig.ResolveIP(ctx, "example.com")
  584. if err != nil {
  585. t.Fatalf("ResolveIP failed: %s", err)
  586. }
  587. if IPs[0].String() != expectedSteeringIP {
  588. t.Fatalf("missing expected steering IP")
  589. }
  590. }
  591. checkSteeringIP("127.0.0.1")
  592. // Test: steering IP used in replay case
  593. dialParams.Succeeded()
  594. dialParams, err = MakeDialParameters(
  595. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  596. if err != nil {
  597. t.Fatalf("MakeDialParameters failed: %s", err)
  598. }
  599. if !dialParams.IsReplay {
  600. t.Fatalf("unexpected non-replay")
  601. }
  602. checkSteeringIP("127.0.0.1")
  603. // Test: different steering IP clears replay flag
  604. dialParams.Succeeded()
  605. setCacheEntry("127.0.0.2")
  606. dialParams, err = MakeDialParameters(
  607. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  608. if err != nil {
  609. t.Fatalf("MakeDialParameters failed: %s", err)
  610. }
  611. if dialParams.IsReplay {
  612. t.Fatalf("unexpected replay")
  613. }
  614. checkSteeringIP("127.0.0.2")
  615. // Test: newly present steering IP clears replay flag
  616. steeringIPCache.Flush()
  617. dialParams, err = MakeDialParameters(
  618. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  619. if err != nil {
  620. t.Fatalf("MakeDialParameters failed: %s", err)
  621. }
  622. dialParams.Succeeded()
  623. setCacheEntry("127.0.0.3")
  624. dialParams, err = MakeDialParameters(
  625. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[0], nil, nil, false, 0, 0)
  626. if err != nil {
  627. t.Fatalf("MakeDialParameters failed: %s", err)
  628. }
  629. if dialParams.IsReplay {
  630. t.Fatalf("unexpected replay")
  631. }
  632. // Test: steering IP cleared from cache after failure
  633. dialParams.Failed(clientConfig, dialErr)
  634. _, ok := steeringIPCache.Get(getCacheKey())
  635. if ok {
  636. t.Fatalf("unexpected steering IP cache entry")
  637. }
  638. }
  639. // Test: DSLPendingPrioritizeDial placeholder transformed to full dial parameters
  640. networkID := clientConfig.GetNetworkID()
  641. err = datastoreUpdate(func(tx *datastoreTx) error {
  642. return dslPrioritizeDialServerEntry(
  643. tx, networkID, []byte(serverEntries[1].IpAddress))
  644. })
  645. if err != nil {
  646. t.Fatalf("dslPrioritizeDialServerEntry failed: %s", err)
  647. }
  648. dialParams, err = MakeDialParameters(
  649. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[1], nil, nil, false, 0, 0)
  650. if err != nil {
  651. t.Fatalf("MakeDialParameters failed: %s", err)
  652. }
  653. if dialParams.DSLPendingPrioritizeDial || !dialParams.DSLPrioritizedDial {
  654. t.Fatalf("unexpected DSL prioritize state")
  655. }
  656. if dialParams.IsReplay {
  657. t.Fatalf("unexpected replay")
  658. }
  659. dialParams.Succeeded()
  660. dialParams, err = MakeDialParameters(
  661. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[1], nil, nil, false, 0, 0)
  662. if err != nil {
  663. t.Fatalf("MakeDialParameters failed: %s", err)
  664. }
  665. if dialParams.DSLPendingPrioritizeDial || !dialParams.DSLPrioritizedDial {
  666. t.Fatalf("unexpected DSL prioritize state")
  667. }
  668. if !dialParams.IsReplay {
  669. t.Fatalf("unexpected non-replay")
  670. }
  671. // Test: DSLPendingPrioritizeDial placeholder doesn't replace full dial parameters
  672. err = datastoreUpdate(func(tx *datastoreTx) error {
  673. return dslPrioritizeDialServerEntry(
  674. tx, networkID, []byte(serverEntries[1].IpAddress))
  675. })
  676. if err != nil {
  677. t.Fatalf("dslPrioritizeDialServerEntry failed: %s", err)
  678. }
  679. dialParams, err = MakeDialParameters(
  680. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntries[1], nil, nil, false, 0, 0)
  681. if err != nil {
  682. t.Fatalf("MakeDialParameters failed: %s", err)
  683. }
  684. if dialParams.DSLPendingPrioritizeDial || !dialParams.DSLPrioritizedDial {
  685. t.Fatalf("unexpected DSL prioritize state")
  686. }
  687. if !dialParams.IsReplay {
  688. t.Fatalf("unexpected non-replay")
  689. }
  690. // Test: iterator shuffles
  691. for i, serverEntry := range serverEntries {
  692. data, err := json.Marshal(serverEntry)
  693. if err != nil {
  694. t.Fatalf("json.Marshal failed: %s", err)
  695. }
  696. var serverEntryFields protocol.ServerEntryFields
  697. err = json.Unmarshal(data, &serverEntryFields)
  698. if err != nil {
  699. t.Fatalf("json.Unmarshal failed: %s", err)
  700. }
  701. err = StoreServerEntry(serverEntryFields, false)
  702. if err != nil {
  703. t.Fatalf("StoreServerEntry failed: %s", err)
  704. }
  705. if i%10 == 0 {
  706. dialParams, err := MakeDialParameters(
  707. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntry, nil, nil, false, 0, 0)
  708. if err != nil {
  709. t.Fatalf("MakeDialParameters failed: %s", err)
  710. }
  711. dialParams.Succeeded()
  712. }
  713. }
  714. for i := 0; i < 5; i++ {
  715. hasAffinity, iterator, err := NewServerEntryIterator(clientConfig)
  716. if err != nil {
  717. t.Fatalf("NewServerEntryIterator failed: %s", err)
  718. }
  719. if hasAffinity {
  720. t.Fatalf("unexpected affinity server")
  721. }
  722. // Test: the first shuffle should move the replay candidates to the front
  723. for j := 0; j < 10; j++ {
  724. serverEntry, err := iterator.Next()
  725. if err != nil {
  726. t.Fatalf("ServerEntryIterator.Next failed: %s", err)
  727. }
  728. dialParams, err := MakeDialParameters(
  729. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntry, nil, nil, false, 0, 0)
  730. if err != nil {
  731. t.Fatalf("MakeDialParameters failed: %s", err)
  732. }
  733. if !dialParams.IsReplay {
  734. t.Fatalf("unexpected non-replay")
  735. }
  736. }
  737. iterator.Reset()
  738. // Test: subsequent shuffles should not move the replay candidates
  739. allReplay := true
  740. for j := 0; j < 10; j++ {
  741. serverEntry, err := iterator.Next()
  742. if err != nil {
  743. t.Fatalf("ServerEntryIterator.Next failed: %s", err)
  744. }
  745. dialParams, err := MakeDialParameters(
  746. clientConfig, steeringIPCache, nil, nil, nil, canReplay, selectProtocol, serverEntry, nil, nil, false, 0, 0)
  747. if err != nil {
  748. t.Fatalf("MakeDialParameters failed: %s", err)
  749. }
  750. if !dialParams.IsReplay {
  751. allReplay = false
  752. }
  753. }
  754. if allReplay {
  755. t.Fatalf("unexpected all replay")
  756. }
  757. iterator.Close()
  758. }
  759. }
  760. func TestLimitTunnelDialPortNumbers(t *testing.T) {
  761. testDataDirName, err := ioutil.TempDir("", "psiphon-limit-tunnel-dial-port-numbers-test")
  762. if err != nil {
  763. t.Fatalf("TempDir failed: %s", err)
  764. }
  765. defer os.RemoveAll(testDataDirName)
  766. err = SetNoticeWriter(io.Discard)
  767. if err != nil {
  768. t.Fatalf("error setting notice writer: %s", err)
  769. }
  770. defer ResetNoticeWriter()
  771. clientConfig := &Config{
  772. PropagationChannelId: "0",
  773. SponsorId: "0",
  774. DataRootDirectory: testDataDirName,
  775. NetworkIDGetter: new(testNetworkGetter),
  776. }
  777. err = clientConfig.Commit(false)
  778. if err != nil {
  779. t.Fatalf("error committing configuration file: %s", err)
  780. }
  781. jsonLimitDialPortNumbers := `
  782. {
  783. "SSH" : [[10,11]],
  784. "OSSH" : [[20,21]],
  785. "QUIC-OSSH" : [[30,31]],
  786. "TAPDANCE-OSSH" : [[40,41]],
  787. "CONJURE-OSSH" : [[50,51]],
  788. "SHADOWSOCKS-OSSH" : [[60,61]],
  789. "All" : [[70,71],80,443]
  790. }
  791. `
  792. var limitTunnelDialPortNumbers parameters.TunnelProtocolPortLists
  793. err = json.Unmarshal([]byte(jsonLimitDialPortNumbers), &limitTunnelDialPortNumbers)
  794. if err != nil {
  795. t.Fatalf("Unmarshal failed: %s", err)
  796. }
  797. applyParameters := make(map[string]interface{})
  798. applyParameters[parameters.LimitTunnelDialPortNumbers] = limitTunnelDialPortNumbers
  799. applyParameters[parameters.LimitTunnelDialPortNumbersProbability] = 1.0
  800. err = clientConfig.SetParameters("tag1", false, applyParameters)
  801. if err != nil {
  802. t.Fatalf("SetParameters failed: %s", err)
  803. }
  804. constraints := &protocolSelectionConstraints{
  805. config: clientConfig,
  806. limitTunnelDialPortNumbers: protocol.TunnelProtocolPortLists(
  807. clientConfig.GetParameters().Get().TunnelProtocolPortLists(parameters.LimitTunnelDialPortNumbers)),
  808. }
  809. selectProtocol := func(serverEntry *protocol.ServerEntry) (string, bool) {
  810. protocol, _, ok := constraints.selectProtocol(0, false, false, serverEntry)
  811. return protocol, ok
  812. }
  813. for _, tunnelProtocol := range protocol.SupportedTunnelProtocols {
  814. if common.Contains(protocol.DefaultDisabledTunnelProtocols, tunnelProtocol) {
  815. continue
  816. }
  817. serverEntries := makeMockServerEntries(tunnelProtocol, "", "", "", 100)
  818. selected := false
  819. skipped := false
  820. for _, serverEntry := range serverEntries {
  821. selectedProtocol, ok := selectProtocol(serverEntry)
  822. if ok {
  823. if selectedProtocol != tunnelProtocol {
  824. t.Fatalf("unexpected selected protocol: %s", selectedProtocol)
  825. }
  826. port, err := serverEntry.GetDialPortNumber(selectedProtocol)
  827. if err != nil {
  828. t.Fatalf("GetDialPortNumber failed: %s", err)
  829. }
  830. if port%10 != 0 && port%10 != 1 && !protocol.TunnelProtocolUsesFrontedMeek(selectedProtocol) {
  831. t.Fatalf("unexpected dial port number: %d", port)
  832. }
  833. selected = true
  834. } else {
  835. skipped = true
  836. }
  837. }
  838. if !selected {
  839. t.Fatalf("expected at least one selected server entry: %s", tunnelProtocol)
  840. }
  841. if !skipped && !protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol) {
  842. t.Fatalf("expected at least one skipped server entry: %s", tunnelProtocol)
  843. }
  844. }
  845. }
  846. func makeMockServerEntries(
  847. tunnelProtocol string,
  848. region string,
  849. providerID string,
  850. frontingProviderID string,
  851. count int) []*protocol.ServerEntry {
  852. serverEntries := make([]*protocol.ServerEntry, count)
  853. for i := 0; i < count; i++ {
  854. serverEntries[i] = &protocol.ServerEntry{
  855. IpAddress: fmt.Sprintf("192.168.0.%d", i),
  856. SshPort: prng.Range(10, 19),
  857. SshObfuscatedPort: prng.Range(20, 29),
  858. SshObfuscatedQUICPort: prng.Range(30, 39),
  859. SshObfuscatedTapDancePort: prng.Range(40, 49),
  860. SshObfuscatedConjurePort: prng.Range(50, 59),
  861. SshShadowsocksPort: prng.Range(60, 69),
  862. MeekServerPort: prng.Range(70, 79),
  863. MeekFrontingHosts: []string{"www1.example.org", "www2.example.org", "www3.example.org"},
  864. MeekFrontingAddressesRegex: "[a-z0-9]{1,64}.example.org",
  865. Region: region,
  866. ProviderID: providerID,
  867. FrontingProviderID: frontingProviderID,
  868. LocalSource: protocol.SERVER_ENTRY_SOURCE_EMBEDDED,
  869. LocalTimestamp: common.TruncateTimestampToHour(common.GetCurrentTimestamp()),
  870. Capabilities: []string{protocol.GetCapability(tunnelProtocol)},
  871. }
  872. }
  873. return serverEntries
  874. }
  875. func TestMakeHTTPTransformerParameters(t *testing.T) {
  876. type test struct {
  877. name string
  878. frontingProviderID string
  879. isFronted bool
  880. paramValues map[string]interface{}
  881. expectedTransformName string
  882. expectedTransformSpec transforms.Spec
  883. }
  884. tests := []test{
  885. {
  886. name: "unfronted",
  887. frontingProviderID: "",
  888. isFronted: false,
  889. paramValues: map[string]interface{}{
  890. "DirectHTTPProtocolTransformProbability": 1,
  891. "DirectHTTPProtocolTransformSpecs": transforms.Specs{
  892. "spec1": {{"A", "B"}},
  893. },
  894. "DirectHTTPProtocolTransformScopedSpecNames": transforms.ScopedSpecNames{
  895. "": {"spec1"},
  896. },
  897. },
  898. expectedTransformName: "spec1",
  899. expectedTransformSpec: [][2]string{{"A", "B"}},
  900. },
  901. {
  902. name: "fronted",
  903. frontingProviderID: "frontingProvider",
  904. isFronted: true,
  905. paramValues: map[string]interface{}{
  906. "FrontedHTTPProtocolTransformProbability": 1,
  907. "FrontedHTTPProtocolTransformSpecs": transforms.Specs{
  908. "spec1": {{"A", "B"}},
  909. },
  910. "FrontedHTTPProtocolTransformScopedSpecNames": transforms.ScopedSpecNames{
  911. "frontingProvider": {"spec1"},
  912. },
  913. },
  914. expectedTransformName: "spec1",
  915. expectedTransformSpec: [][2]string{{"A", "B"}},
  916. },
  917. {
  918. name: "no transform, coinflip false",
  919. frontingProviderID: "frontingProvider",
  920. isFronted: false,
  921. paramValues: map[string]interface{}{
  922. "DirectHTTPProtocolTransformProbability": 0,
  923. "DirectHTTPProtocolTransformSpecs": transforms.Specs{
  924. "spec1": {{"A", "B"}},
  925. },
  926. "DirectHTTPProtocolTransformScopedSpecNames": transforms.ScopedSpecNames{
  927. "frontingProvider": {"spec1"},
  928. },
  929. },
  930. expectedTransformName: "",
  931. expectedTransformSpec: nil,
  932. },
  933. }
  934. for _, tt := range tests {
  935. t.Run(tt.name, func(t *testing.T) {
  936. params, err := parameters.NewParameters(nil)
  937. if err != nil {
  938. t.Fatalf("parameters.NewParameters failed %v", err)
  939. }
  940. _, err = params.Set("", 0, tt.paramValues)
  941. if err != nil {
  942. t.Fatalf("params.Set failed %v", err)
  943. }
  944. httpTransformerParams, err := makeHTTPTransformerParameters(params.Get(), tt.frontingProviderID, tt.isFronted)
  945. if err != nil {
  946. t.Fatalf("MakeHTTPTransformerParameters failed %v", err)
  947. }
  948. if httpTransformerParams.ProtocolTransformName != tt.expectedTransformName {
  949. t.Fatalf("expected ProtocolTransformName \"%s\" but got \"%s\"", tt.expectedTransformName, httpTransformerParams.ProtocolTransformName)
  950. }
  951. if !reflect.DeepEqual(httpTransformerParams.ProtocolTransformSpec, tt.expectedTransformSpec) {
  952. t.Fatalf("expected ProtocolTransformSpec %v but got %v", tt.expectedTransformSpec, httpTransformerParams.ProtocolTransformSpec)
  953. }
  954. if httpTransformerParams.ProtocolTransformSpec != nil {
  955. if httpTransformerParams.ProtocolTransformSeed == nil {
  956. t.Fatalf("expected non-nil seed")
  957. }
  958. }
  959. })
  960. }
  961. }
  962. func TestMakeOSSHObfuscatorSeedTranformerParameters(t *testing.T) {
  963. type test struct {
  964. name string
  965. paramValues map[string]interface{}
  966. expectedTransformName string
  967. expectedTransformSpec transforms.Spec
  968. }
  969. tests := []test{
  970. {
  971. name: "transform",
  972. paramValues: map[string]interface{}{
  973. "OSSHObfuscatorSeedTransformProbability": 1,
  974. "OSSHObfuscatorSeedTransformSpecs": transforms.Specs{
  975. "spec1": {{"A", "B"}},
  976. },
  977. "OSSHObfuscatorSeedTransformScopedSpecNames": transforms.ScopedSpecNames{
  978. "": {"spec1"},
  979. },
  980. },
  981. expectedTransformName: "spec1",
  982. expectedTransformSpec: [][2]string{{"A", "B"}},
  983. },
  984. {
  985. name: "no transform, coinflip false",
  986. paramValues: map[string]interface{}{
  987. "OSSHObfuscatorSeedTransformProbability": 0,
  988. "OSSHObfuscatorSeedTransformSpecs": transforms.Specs{
  989. "spec1": {{"A", "B"}},
  990. },
  991. "OSSHObfuscatorSeedTransformScopedSpecNames": transforms.ScopedSpecNames{
  992. "": {"spec1"},
  993. },
  994. },
  995. expectedTransformName: "",
  996. expectedTransformSpec: nil,
  997. },
  998. }
  999. for _, tt := range tests {
  1000. t.Run(tt.name, func(t *testing.T) {
  1001. params, err := parameters.NewParameters(nil)
  1002. if err != nil {
  1003. t.Fatalf("parameters.NewParameters failed: %v", err)
  1004. }
  1005. _, err = params.Set("", 0, tt.paramValues)
  1006. if err != nil {
  1007. t.Fatalf("params.Set failed: %v", err)
  1008. }
  1009. transformerParams, err := makeSeedTransformerParameters(
  1010. params.Get(),
  1011. parameters.OSSHObfuscatorSeedTransformProbability,
  1012. parameters.OSSHObfuscatorSeedTransformSpecs,
  1013. parameters.OSSHObfuscatorSeedTransformScopedSpecNames)
  1014. if err != nil {
  1015. t.Fatalf("makeSeedTransformerParameters failed: %v", err)
  1016. }
  1017. if transformerParams.TransformName != tt.expectedTransformName {
  1018. t.Fatalf("expected TransformName \"%s\" but got \"%s\"", tt.expectedTransformName, transformerParams.TransformName)
  1019. }
  1020. if !reflect.DeepEqual(transformerParams.TransformSpec, tt.expectedTransformSpec) {
  1021. t.Fatalf("expected TransformSpec %v but got %v", tt.expectedTransformSpec, transformerParams.TransformSpec)
  1022. }
  1023. if transformerParams.TransformSpec != nil {
  1024. if transformerParams.TransformSeed == nil {
  1025. t.Fatalf("expected non-nil seed")
  1026. }
  1027. }
  1028. })
  1029. }
  1030. }