controller.go 56 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567
  1. /*
  2. * Copyright (c) 2015, 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 implements the core tunnel functionality of a Psiphon client.
  20. // The main function is RunForever, which runs a Controller that obtains lists of
  21. // servers, establishes tunnel connections, and runs local proxies through which
  22. // tunneled traffic may be sent.
  23. package psiphon
  24. import (
  25. "errors"
  26. "fmt"
  27. "math/rand"
  28. "net"
  29. "sync"
  30. "time"
  31. "github.com/Psiphon-Inc/goarista/monotime"
  32. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
  33. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/protocol"
  34. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/tun"
  35. )
  36. // Controller is a tunnel lifecycle coordinator. It manages lists of servers to
  37. // connect to; establishes and monitors tunnels; and runs local proxies which
  38. // route traffic through the tunnels.
  39. type Controller struct {
  40. config *Config
  41. sessionId string
  42. componentFailureSignal chan struct{}
  43. shutdownBroadcast chan struct{}
  44. runWaitGroup *sync.WaitGroup
  45. connectedTunnels chan *Tunnel
  46. failedTunnels chan *Tunnel
  47. tunnelMutex sync.Mutex
  48. establishedOnce bool
  49. tunnels []*Tunnel
  50. nextTunnel int
  51. startedConnectedReporter bool
  52. isEstablishing bool
  53. concurrentEstablishTunnelsMutex sync.Mutex
  54. concurrentEstablishTunnels int
  55. concurrentMeekEstablishTunnels int
  56. peakConcurrentEstablishTunnels int
  57. peakConcurrentMeekEstablishTunnels int
  58. establishWaitGroup *sync.WaitGroup
  59. stopEstablishingBroadcast chan struct{}
  60. candidateServerEntries chan *candidateServerEntry
  61. establishPendingConns *common.Conns
  62. untunneledPendingConns *common.Conns
  63. untunneledDialConfig *DialConfig
  64. splitTunnelClassifier *SplitTunnelClassifier
  65. signalFetchCommonRemoteServerList chan struct{}
  66. signalFetchObfuscatedServerLists chan struct{}
  67. signalDownloadUpgrade chan string
  68. impairedProtocolClassification map[string]int
  69. signalReportConnected chan struct{}
  70. serverAffinityDoneBroadcast chan struct{}
  71. newClientVerificationPayload chan string
  72. packetTunnelClient *tun.Client
  73. packetTunnelTransport *PacketTunnelTransport
  74. }
  75. type candidateServerEntry struct {
  76. serverEntry *protocol.ServerEntry
  77. isServerAffinityCandidate bool
  78. adjustedEstablishStartTime monotime.Time
  79. }
  80. // NewController initializes a new controller.
  81. func NewController(config *Config) (controller *Controller, err error) {
  82. // Needed by regen, at least
  83. rand.Seed(int64(time.Now().Nanosecond()))
  84. // The session ID for the Psiphon server API is used across all
  85. // tunnels established by the controller.
  86. NoticeSessionId(config.SessionID)
  87. // untunneledPendingConns may be used to interrupt the fetch remote server list
  88. // request and other untunneled connection establishments. BindToDevice may be
  89. // used to exclude these requests and connection from VPN routing.
  90. // TODO: fetch remote server list and untunneled upgrade download should remove
  91. // their completed conns from untunneledPendingConns.
  92. untunneledPendingConns := new(common.Conns)
  93. untunneledDialConfig := &DialConfig{
  94. UpstreamProxyUrl: config.UpstreamProxyUrl,
  95. CustomHeaders: config.CustomHeaders,
  96. PendingConns: untunneledPendingConns,
  97. DeviceBinder: config.DeviceBinder,
  98. DnsServerGetter: config.DnsServerGetter,
  99. IPv6Synthesizer: config.IPv6Synthesizer,
  100. UseIndistinguishableTLS: config.UseIndistinguishableTLS,
  101. TrustedCACertificatesFilename: config.TrustedCACertificatesFilename,
  102. DeviceRegion: config.DeviceRegion,
  103. }
  104. controller = &Controller{
  105. config: config,
  106. sessionId: config.SessionID,
  107. // componentFailureSignal receives a signal from a component (including socks and
  108. // http local proxies) if they unexpectedly fail. Senders should not block.
  109. // Buffer allows at least one stop signal to be sent before there is a receiver.
  110. componentFailureSignal: make(chan struct{}, 1),
  111. shutdownBroadcast: make(chan struct{}),
  112. runWaitGroup: new(sync.WaitGroup),
  113. // connectedTunnels and failedTunnels buffer sizes are large enough to
  114. // receive full pools of tunnels without blocking. Senders should not block.
  115. connectedTunnels: make(chan *Tunnel, config.TunnelPoolSize),
  116. failedTunnels: make(chan *Tunnel, config.TunnelPoolSize),
  117. tunnels: make([]*Tunnel, 0),
  118. establishedOnce: false,
  119. startedConnectedReporter: false,
  120. isEstablishing: false,
  121. establishPendingConns: new(common.Conns),
  122. untunneledPendingConns: untunneledPendingConns,
  123. untunneledDialConfig: untunneledDialConfig,
  124. impairedProtocolClassification: make(map[string]int),
  125. // TODO: Add a buffer of 1 so we don't miss a signal while receiver is
  126. // starting? Trade-off is potential back-to-back fetch remotes. As-is,
  127. // establish will eventually signal another fetch remote.
  128. signalFetchCommonRemoteServerList: make(chan struct{}),
  129. signalFetchObfuscatedServerLists: make(chan struct{}),
  130. signalDownloadUpgrade: make(chan string),
  131. signalReportConnected: make(chan struct{}),
  132. // Buffer allows SetClientVerificationPayloadForActiveTunnels to submit one
  133. // new payload without blocking or dropping it.
  134. newClientVerificationPayload: make(chan string, 1),
  135. }
  136. controller.splitTunnelClassifier = NewSplitTunnelClassifier(config, controller)
  137. if config.PacketTunnelTunFileDescriptor > 0 {
  138. // Run a packet tunnel client. The lifetime of the tun.Client is the
  139. // lifetime of the Controller, so it exists across tunnel establishments
  140. // and reestablishments. The PacketTunnelTransport provides a layer
  141. // that presents a continuosuly existing transport to the tun.Client;
  142. // it's set to use new SSH channels after new SSH tunnel establishes.
  143. packetTunnelTransport := NewPacketTunnelTransport()
  144. packetTunnelClient, err := tun.NewClient(&tun.ClientConfig{
  145. Logger: NoticeCommonLogger(),
  146. TunFileDescriptor: config.PacketTunnelTunFileDescriptor,
  147. Transport: packetTunnelTransport,
  148. })
  149. if err != nil {
  150. return nil, common.ContextError(err)
  151. }
  152. controller.packetTunnelClient = packetTunnelClient
  153. controller.packetTunnelTransport = packetTunnelTransport
  154. }
  155. return controller, nil
  156. }
  157. // Run executes the controller. It launches components and then monitors
  158. // for a shutdown signal; after receiving the signal it shuts down the
  159. // controller.
  160. // The components include:
  161. // - the periodic remote server list fetcher
  162. // - the connected reporter
  163. // - the tunnel manager
  164. // - a local SOCKS proxy that port forwards through the pool of tunnels
  165. // - a local HTTP proxy that port forwards through the pool of tunnels
  166. func (controller *Controller) Run(shutdownBroadcast <-chan struct{}) {
  167. ReportAvailableRegions()
  168. // Start components
  169. // TODO: IPv6 support
  170. var listenIP string
  171. if controller.config.ListenInterface == "" {
  172. listenIP = "127.0.0.1"
  173. } else if controller.config.ListenInterface == "any" {
  174. listenIP = "0.0.0.0"
  175. } else {
  176. IPv4Address, _, err := common.GetInterfaceIPAddresses(controller.config.ListenInterface)
  177. if err == nil && IPv4Address == nil {
  178. err = fmt.Errorf("no IPv4 address for interface %s", controller.config.ListenInterface)
  179. }
  180. if err != nil {
  181. NoticeError("error getting listener IP: %s", err)
  182. return
  183. }
  184. listenIP = IPv4Address.String()
  185. }
  186. if !controller.config.DisableLocalSocksProxy {
  187. socksProxy, err := NewSocksProxy(controller.config, controller, listenIP)
  188. if err != nil {
  189. NoticeAlert("error initializing local SOCKS proxy: %s", err)
  190. return
  191. }
  192. defer socksProxy.Close()
  193. }
  194. if !controller.config.DisableLocalHTTPProxy {
  195. httpProxy, err := NewHttpProxy(
  196. controller.config, controller.untunneledDialConfig, controller, listenIP)
  197. if err != nil {
  198. NoticeAlert("error initializing local HTTP proxy: %s", err)
  199. return
  200. }
  201. defer httpProxy.Close()
  202. }
  203. if !controller.config.DisableRemoteServerListFetcher {
  204. retryPeriod := time.Duration(
  205. *controller.config.FetchRemoteServerListRetryPeriodSeconds) * time.Second
  206. if controller.config.RemoteServerListURLs != nil {
  207. controller.runWaitGroup.Add(1)
  208. go controller.remoteServerListFetcher(
  209. "common",
  210. FetchCommonRemoteServerList,
  211. controller.signalFetchCommonRemoteServerList,
  212. retryPeriod,
  213. FETCH_REMOTE_SERVER_LIST_STALE_PERIOD)
  214. }
  215. if controller.config.ObfuscatedServerListRootURLs != nil {
  216. controller.runWaitGroup.Add(1)
  217. go controller.remoteServerListFetcher(
  218. "obfuscated",
  219. FetchObfuscatedServerLists,
  220. controller.signalFetchObfuscatedServerLists,
  221. retryPeriod,
  222. FETCH_REMOTE_SERVER_LIST_STALE_PERIOD)
  223. }
  224. }
  225. if controller.config.UpgradeDownloadURLs != nil {
  226. controller.runWaitGroup.Add(1)
  227. go controller.upgradeDownloader()
  228. }
  229. /// Note: the connected reporter isn't started until a tunnel is
  230. // established
  231. controller.runWaitGroup.Add(1)
  232. go controller.runTunnels()
  233. if *controller.config.EstablishTunnelTimeoutSeconds != 0 {
  234. controller.runWaitGroup.Add(1)
  235. go controller.establishTunnelWatcher()
  236. }
  237. if controller.packetTunnelClient != nil {
  238. controller.packetTunnelClient.Start()
  239. }
  240. // Wait while running
  241. select {
  242. case <-shutdownBroadcast:
  243. NoticeInfo("controller shutdown by request")
  244. case <-controller.componentFailureSignal:
  245. NoticeAlert("controller shutdown due to component failure")
  246. }
  247. close(controller.shutdownBroadcast)
  248. if controller.packetTunnelClient != nil {
  249. controller.packetTunnelClient.Stop()
  250. }
  251. // Interrupts and stops establish workers blocking on
  252. // tunnel establishment network operations.
  253. controller.establishPendingConns.CloseAll()
  254. // Interrupts and stops workers blocking on untunneled
  255. // network operations. This includes fetch remote server
  256. // list and untunneled uprade download.
  257. // Note: this doesn't interrupt the final, untunneled status
  258. // requests started in operateTunnel after shutdownBroadcast.
  259. // This is by design -- we want to give these requests a short
  260. // timer period to succeed and deliver stats. These particular
  261. // requests opt out of untunneledPendingConns and use the
  262. // PSIPHON_API_SHUTDOWN_SERVER_TIMEOUT timeout (see
  263. // doUntunneledStatusRequest).
  264. controller.untunneledPendingConns.CloseAll()
  265. // Now with all workers signaled to stop and with all
  266. // blocking network operations interrupted, wait for
  267. // all workers to terminate.
  268. controller.runWaitGroup.Wait()
  269. controller.splitTunnelClassifier.Shutdown()
  270. NoticeInfo("exiting controller")
  271. NoticeExiting()
  272. }
  273. // SignalComponentFailure notifies the controller that an associated component has failed.
  274. // This will terminate the controller.
  275. func (controller *Controller) SignalComponentFailure() {
  276. select {
  277. case controller.componentFailureSignal <- *new(struct{}):
  278. default:
  279. }
  280. }
  281. // SetClientVerificationPayloadForActiveTunnels sets the client verification
  282. // payload that is to be sent in client verification requests to all established
  283. // tunnels.
  284. //
  285. // Client verification is used to verify that the client is a
  286. // valid Psiphon client, which will determine how the server treats
  287. // the client traffic. The proof-of-validity is platform-specific
  288. // and the payload is opaque to this function but assumed to be JSON.
  289. //
  290. // Since, in some cases, verification payload cannot be determined until
  291. // after tunnel-core starts, the payload cannot be simply specified in
  292. // the Config.
  293. //
  294. // SetClientVerificationPayloadForActiveTunnels will not block enqueuing a new verification
  295. // payload. One new payload can be enqueued, after which additional payloads
  296. // will be dropped if a payload is still enqueued.
  297. func (controller *Controller) SetClientVerificationPayloadForActiveTunnels(clientVerificationPayload string) {
  298. select {
  299. case controller.newClientVerificationPayload <- clientVerificationPayload:
  300. default:
  301. }
  302. }
  303. // remoteServerListFetcher fetches an out-of-band list of server entries
  304. // for more tunnel candidates. It fetches when signalled, with retries
  305. // on failure.
  306. func (controller *Controller) remoteServerListFetcher(
  307. name string,
  308. fetcher RemoteServerListFetcher,
  309. signal <-chan struct{},
  310. retryPeriod, stalePeriod time.Duration) {
  311. defer controller.runWaitGroup.Done()
  312. var lastFetchTime monotime.Time
  313. fetcherLoop:
  314. for {
  315. // Wait for a signal before fetching
  316. select {
  317. case <-signal:
  318. case <-controller.shutdownBroadcast:
  319. break fetcherLoop
  320. }
  321. // Skip fetch entirely (i.e., send no request at all, even when ETag would save
  322. // on response size) when a recent fetch was successful
  323. if lastFetchTime != 0 &&
  324. lastFetchTime.Add(stalePeriod).After(monotime.Now()) {
  325. continue
  326. }
  327. retryLoop:
  328. for attempt := 0; ; attempt++ {
  329. // Don't attempt to fetch while there is no network connectivity,
  330. // to avoid alert notice noise.
  331. if !WaitForNetworkConnectivity(
  332. controller.config.NetworkConnectivityChecker,
  333. controller.shutdownBroadcast) {
  334. break fetcherLoop
  335. }
  336. // Pick any active tunnel and make the next fetch attempt. If there's
  337. // no active tunnel, the untunneledDialConfig will be used.
  338. tunnel := controller.getNextActiveTunnel()
  339. err := fetcher(
  340. controller.config,
  341. attempt,
  342. tunnel,
  343. controller.untunneledDialConfig)
  344. if err == nil {
  345. lastFetchTime = monotime.Now()
  346. break retryLoop
  347. }
  348. NoticeAlert("failed to fetch %s remote server list: %s", name, err)
  349. timeout := time.After(retryPeriod)
  350. select {
  351. case <-timeout:
  352. case <-controller.shutdownBroadcast:
  353. break fetcherLoop
  354. }
  355. }
  356. }
  357. NoticeInfo("exiting %s remote server list fetcher", name)
  358. }
  359. // establishTunnelWatcher terminates the controller if a tunnel
  360. // has not been established in the configured time period. This
  361. // is regardless of how many tunnels are presently active -- meaning
  362. // that if an active tunnel was established and lost the controller
  363. // is left running (to re-establish).
  364. func (controller *Controller) establishTunnelWatcher() {
  365. defer controller.runWaitGroup.Done()
  366. timeout := time.After(
  367. time.Duration(*controller.config.EstablishTunnelTimeoutSeconds) * time.Second)
  368. select {
  369. case <-timeout:
  370. if !controller.hasEstablishedOnce() {
  371. NoticeAlert("failed to establish tunnel before timeout")
  372. controller.SignalComponentFailure()
  373. }
  374. case <-controller.shutdownBroadcast:
  375. }
  376. NoticeInfo("exiting establish tunnel watcher")
  377. }
  378. // connectedReporter sends periodic "connected" requests to the Psiphon API.
  379. // These requests are for server-side unique user stats calculation. See the
  380. // comment in DoConnectedRequest for a description of the request mechanism.
  381. // To ensure we don't over- or under-count unique users, only one connected
  382. // request is made across all simultaneous multi-tunnels; and the connected
  383. // request is repeated periodically for very long-lived tunnels.
  384. // The signalReportConnected mechanism is used to trigger another connected
  385. // request immediately after a reconnect.
  386. func (controller *Controller) connectedReporter() {
  387. defer controller.runWaitGroup.Done()
  388. loop:
  389. for {
  390. // Pick any active tunnel and make the next connected request. No error
  391. // is logged if there's no active tunnel, as that's not an unexpected condition.
  392. reported := false
  393. tunnel := controller.getNextActiveTunnel()
  394. if tunnel != nil {
  395. err := tunnel.serverContext.DoConnectedRequest()
  396. if err == nil {
  397. reported = true
  398. } else {
  399. NoticeAlert("failed to make connected request: %s", err)
  400. }
  401. }
  402. // Schedule the next connected request and wait.
  403. var duration time.Duration
  404. if reported {
  405. duration = PSIPHON_API_CONNECTED_REQUEST_PERIOD
  406. } else {
  407. duration = PSIPHON_API_CONNECTED_REQUEST_RETRY_PERIOD
  408. }
  409. timeout := time.After(duration)
  410. select {
  411. case <-controller.signalReportConnected:
  412. case <-timeout:
  413. // Make another connected request
  414. case <-controller.shutdownBroadcast:
  415. break loop
  416. }
  417. }
  418. NoticeInfo("exiting connected reporter")
  419. }
  420. func (controller *Controller) startOrSignalConnectedReporter() {
  421. // session is nil when DisableApi is set
  422. if controller.config.DisableApi {
  423. return
  424. }
  425. // Start the connected reporter after the first tunnel is established.
  426. // Concurrency note: only the runTunnels goroutine may access startedConnectedReporter.
  427. if !controller.startedConnectedReporter {
  428. controller.startedConnectedReporter = true
  429. controller.runWaitGroup.Add(1)
  430. go controller.connectedReporter()
  431. } else {
  432. select {
  433. case controller.signalReportConnected <- *new(struct{}):
  434. default:
  435. }
  436. }
  437. }
  438. // upgradeDownloader makes periodic attempts to complete a client upgrade
  439. // download. DownloadUpgrade() is resumable, so each attempt has potential for
  440. // getting closer to completion, even in conditions where the download or
  441. // tunnel is repeatedly interrupted.
  442. // An upgrade download is triggered by either a handshake response indicating
  443. // that a new version is available; or after failing to connect, in which case
  444. // it's useful to check, out-of-band, for an upgrade with new circumvention
  445. // capabilities.
  446. // Once the download operation completes successfully, the downloader exits
  447. // and is not run again: either there is not a newer version, or the upgrade
  448. // has been downloaded and is ready to be applied.
  449. // We're assuming that the upgrade will be applied and the entire system
  450. // restarted before another upgrade is to be downloaded.
  451. //
  452. // TODO: refactor upgrade downloader and remote server list fetcher to use
  453. // common code (including the resumable download routines).
  454. //
  455. func (controller *Controller) upgradeDownloader() {
  456. defer controller.runWaitGroup.Done()
  457. var lastDownloadTime monotime.Time
  458. downloadLoop:
  459. for {
  460. // Wait for a signal before downloading
  461. var handshakeVersion string
  462. select {
  463. case handshakeVersion = <-controller.signalDownloadUpgrade:
  464. case <-controller.shutdownBroadcast:
  465. break downloadLoop
  466. }
  467. // Unless handshake is explicitly advertizing a new version, skip
  468. // checking entirely when a recent download was successful.
  469. if handshakeVersion == "" &&
  470. lastDownloadTime != 0 &&
  471. lastDownloadTime.Add(DOWNLOAD_UPGRADE_STALE_PERIOD).After(monotime.Now()) {
  472. continue
  473. }
  474. retryLoop:
  475. for attempt := 0; ; attempt++ {
  476. // Don't attempt to download while there is no network connectivity,
  477. // to avoid alert notice noise.
  478. if !WaitForNetworkConnectivity(
  479. controller.config.NetworkConnectivityChecker,
  480. controller.shutdownBroadcast) {
  481. break downloadLoop
  482. }
  483. // Pick any active tunnel and make the next download attempt. If there's
  484. // no active tunnel, the untunneledDialConfig will be used.
  485. tunnel := controller.getNextActiveTunnel()
  486. err := DownloadUpgrade(
  487. controller.config,
  488. attempt,
  489. handshakeVersion,
  490. tunnel,
  491. controller.untunneledDialConfig)
  492. if err == nil {
  493. lastDownloadTime = monotime.Now()
  494. break retryLoop
  495. }
  496. NoticeAlert("failed to download upgrade: %s", err)
  497. timeout := time.After(
  498. time.Duration(*controller.config.DownloadUpgradeRetryPeriodSeconds) * time.Second)
  499. select {
  500. case <-timeout:
  501. case <-controller.shutdownBroadcast:
  502. break downloadLoop
  503. }
  504. }
  505. }
  506. NoticeInfo("exiting upgrade downloader")
  507. }
  508. // runTunnels is the controller tunnel management main loop. It starts and stops
  509. // establishing tunnels based on the target tunnel pool size and the current size
  510. // of the pool. Tunnels are established asynchronously using worker goroutines.
  511. //
  512. // When there are no server entries for the target region/protocol, the
  513. // establishCandidateGenerator will yield no candidates and wait before
  514. // trying again. In the meantime, a remote server entry fetch may supply
  515. // valid candidates.
  516. //
  517. // When a tunnel is established, it's added to the active pool. The tunnel's
  518. // operateTunnel goroutine monitors the tunnel.
  519. //
  520. // When a tunnel fails, it's removed from the pool and the establish process is
  521. // restarted to fill the pool.
  522. func (controller *Controller) runTunnels() {
  523. defer controller.runWaitGroup.Done()
  524. var clientVerificationPayload string
  525. // Start running
  526. controller.startEstablishing()
  527. loop:
  528. for {
  529. select {
  530. case failedTunnel := <-controller.failedTunnels:
  531. NoticeAlert("tunnel failed: %s", failedTunnel.serverEntry.IpAddress)
  532. controller.terminateTunnel(failedTunnel)
  533. // Note: we make this extra check to ensure the shutdown signal takes priority
  534. // and that we do not start establishing. Critically, startEstablishing() calls
  535. // establishPendingConns.Reset() which clears the closed flag in
  536. // establishPendingConns; this causes the pendingConns.Add() within
  537. // interruptibleTCPDial to succeed instead of aborting, and the result
  538. // is that it's possible for establish goroutines to run all the way through
  539. // NewServerContext before being discarded... delaying shutdown.
  540. select {
  541. case <-controller.shutdownBroadcast:
  542. break loop
  543. default:
  544. }
  545. controller.classifyImpairedProtocol(failedTunnel)
  546. // Clear the reference to this tunnel before calling startEstablishing,
  547. // which will invoke a garbage collection.
  548. failedTunnel = nil
  549. // Concurrency note: only this goroutine may call startEstablishing/stopEstablishing,
  550. // which reference controller.isEstablishing.
  551. controller.startEstablishing()
  552. case connectedTunnel := <-controller.connectedTunnels:
  553. if controller.isImpairedProtocol(connectedTunnel.protocol) {
  554. // Protocol was classified as impaired while this tunnel established.
  555. // This is most likely to occur with TunnelPoolSize > 0. We log the
  556. // event but take no action. Discarding the tunnel would break the
  557. // impaired logic unless we did that (a) only if there are other
  558. // unimpaired protocols; (b) only during the first iteration of the
  559. // ESTABLISH_TUNNEL_WORK_TIME loop. By not discarding here, a true
  560. // impaired protocol may require an extra reconnect.
  561. NoticeAlert("connected tunnel with impaired protocol: %s", connectedTunnel.protocol)
  562. }
  563. // Tunnel establishment has two phases: connection and activation.
  564. //
  565. // Connection is run concurrently by the establishTunnelWorkers, to minimize
  566. // delay when it's not yet known which server and protocol will be available
  567. // and unblocked.
  568. //
  569. // Activation is run serially, here, to minimize the overhead of making a
  570. // handshake request and starting the operateTunnel management worker for a
  571. // tunnel which may be discarded.
  572. //
  573. // When the active tunnel will complete establishment, establishment is
  574. // stopped before activation. This interrupts all connecting tunnels and
  575. // garbage collects their memory. The purpose is to minimize memory
  576. // pressure when the handshake request is made. In the unlikely case that the
  577. // handshake fails, establishment is restarted.
  578. //
  579. // Any delays in stopEstablishing will delay the handshake for the last
  580. // active tunnel.
  581. //
  582. // In the typical case of TunnelPoolSize of 1, only a single handshake is
  583. // performed and the homepages notices file, when used, will not be modifed
  584. // after the NoticeTunnels(1) [i.e., connected] until NoticeTunnels(0) [i.e.,
  585. // disconnected]. For TunnelPoolSize > 1, serial handshakes only ensures that
  586. // each set of emitted NoticeHomepages is contiguous.
  587. active, outstanding := controller.numTunnels()
  588. // discardTunnel will be true here when already fully established.
  589. discardTunnel := (outstanding <= 0)
  590. isFirstTunnel := (active == 0)
  591. isLastTunnel := (outstanding == 1)
  592. if !discardTunnel {
  593. if isLastTunnel {
  594. controller.stopEstablishing()
  595. }
  596. err := connectedTunnel.Activate(controller, controller.shutdownBroadcast)
  597. if err != nil {
  598. NoticeAlert("failed to activate %s: %s", connectedTunnel.serverEntry.IpAddress, err)
  599. discardTunnel = true
  600. } else {
  601. // It's unlikely that registerTunnel will fail, since only this goroutine
  602. // calls registerTunnel -- and after checking numTunnels; so failure is not
  603. // expected.
  604. if !controller.registerTunnel(connectedTunnel) {
  605. NoticeAlert("failed to register %s: %s", connectedTunnel.serverEntry.IpAddress)
  606. discardTunnel = true
  607. }
  608. }
  609. // May need to replace this tunnel
  610. if isLastTunnel && discardTunnel {
  611. controller.startEstablishing()
  612. }
  613. }
  614. if discardTunnel {
  615. controller.discardTunnel(connectedTunnel)
  616. // Clear the reference to this discarded tunnel and immediately run
  617. // a garbage collection to reclaim its memory.
  618. connectedTunnel = nil
  619. aggressiveGarbageCollection()
  620. // Skip the rest of this case
  621. break
  622. }
  623. NoticeActiveTunnel(
  624. connectedTunnel.serverEntry.IpAddress,
  625. connectedTunnel.protocol,
  626. connectedTunnel.serverEntry.SupportsSSHAPIRequests())
  627. if isFirstTunnel {
  628. // The split tunnel classifier is started once the first tunnel is
  629. // established. This first tunnel is passed in to be used to make
  630. // the routes data request.
  631. // A long-running controller may run while the host device is present
  632. // in different regions. In this case, we want the split tunnel logic
  633. // to switch to routes for new regions and not classify traffic based
  634. // on routes installed for older regions.
  635. // We assume that when regions change, the host network will also
  636. // change, and so all tunnels will fail and be re-established. Under
  637. // that assumption, the classifier will be re-Start()-ed here when
  638. // the region has changed.
  639. controller.splitTunnelClassifier.Start(connectedTunnel)
  640. // Signal a connected request on each 1st tunnel establishment. For
  641. // multi-tunnels, the session is connected as long as at least one
  642. // tunnel is established.
  643. controller.startOrSignalConnectedReporter()
  644. // If the handshake indicated that a new client version is available,
  645. // trigger an upgrade download.
  646. // Note: serverContext is nil when DisableApi is set
  647. if connectedTunnel.serverContext != nil &&
  648. connectedTunnel.serverContext.clientUpgradeVersion != "" {
  649. handshakeVersion := connectedTunnel.serverContext.clientUpgradeVersion
  650. select {
  651. case controller.signalDownloadUpgrade <- handshakeVersion:
  652. default:
  653. }
  654. }
  655. }
  656. // Set the new tunnel as the transport for the packet tunnel. The packet tunnel
  657. // client remains up when reestablishing, but no packets are relayed while there
  658. // is no connected tunnel. UseTunnel will establish a new packet tunnel SSH
  659. // channel over the new SSH tunnel and configure the packet tunnel client to use
  660. // the new SSH channel as its transport.
  661. //
  662. // Note: as is, this logic is suboptimal for TunnelPoolSize > 1, as this would
  663. // continuously initialize new packet tunnel sessions for each established
  664. // server. For now, config validation requires TunnelPoolSize == 1 when
  665. // the packet tunnel is used.
  666. if controller.packetTunnelTransport != nil {
  667. controller.packetTunnelTransport.UseTunnel(connectedTunnel)
  668. }
  669. // TODO: design issue -- might not be enough server entries with region/caps to ever fill tunnel slots;
  670. // possible solution is establish target MIN(CountServerEntries(region, protocol), TunnelPoolSize)
  671. if controller.isFullyEstablished() {
  672. controller.stopEstablishing()
  673. }
  674. case clientVerificationPayload = <-controller.newClientVerificationPayload:
  675. controller.setClientVerificationPayloadForActiveTunnels(clientVerificationPayload)
  676. case <-controller.shutdownBroadcast:
  677. break loop
  678. }
  679. }
  680. // Stop running
  681. controller.stopEstablishing()
  682. controller.terminateAllTunnels()
  683. // Drain tunnel channels
  684. close(controller.connectedTunnels)
  685. for tunnel := range controller.connectedTunnels {
  686. controller.discardTunnel(tunnel)
  687. }
  688. close(controller.failedTunnels)
  689. for tunnel := range controller.failedTunnels {
  690. controller.discardTunnel(tunnel)
  691. }
  692. NoticeInfo("exiting run tunnels")
  693. }
  694. // TerminateNextActiveTunnel is a support routine for
  695. // test code that must terminate the active tunnel and
  696. // restart establishing. This function is not guaranteed
  697. // to be safe for use in other cases.
  698. func (controller *Controller) TerminateNextActiveTunnel() {
  699. tunnel := controller.getNextActiveTunnel()
  700. if tunnel != nil {
  701. controller.SignalTunnelFailure(tunnel)
  702. NoticeInfo("terminated tunnel: %s", tunnel.serverEntry.IpAddress)
  703. }
  704. }
  705. // classifyImpairedProtocol tracks "impaired" protocol classifications for failed
  706. // tunnels. A protocol is classified as impaired if a tunnel using that protocol
  707. // fails, repeatedly, shortly after the start of the connection. During tunnel
  708. // establishment, impaired protocols are briefly skipped.
  709. //
  710. // One purpose of this measure is to defend against an attack where the adversary,
  711. // for example, tags an OSSH TCP connection as an "unidentified" protocol; allows
  712. // it to connect; but then kills the underlying TCP connection after a short time.
  713. // Since OSSH has less latency than other protocols that may bypass an "unidentified"
  714. // filter, these other protocols might never be selected for use.
  715. //
  716. // Concurrency note: only the runTunnels() goroutine may call classifyImpairedProtocol
  717. func (controller *Controller) classifyImpairedProtocol(failedTunnel *Tunnel) {
  718. if failedTunnel.establishedTime.Add(IMPAIRED_PROTOCOL_CLASSIFICATION_DURATION).After(monotime.Now()) {
  719. controller.impairedProtocolClassification[failedTunnel.protocol] += 1
  720. } else {
  721. controller.impairedProtocolClassification[failedTunnel.protocol] = 0
  722. }
  723. // Reset classification once all known protocols are classified as impaired, as
  724. // there is now no way to proceed with only unimpaired protocols. The network
  725. // situation (or attack) resulting in classification may not be protocol-specific.
  726. //
  727. // Note: with controller.config.TunnelProtocol set, this will always reset once
  728. // that protocol has reached IMPAIRED_PROTOCOL_CLASSIFICATION_THRESHOLD.
  729. if CountNonImpairedProtocols(
  730. controller.config.EgressRegion,
  731. controller.config.TunnelProtocol,
  732. controller.getImpairedProtocols()) == 0 {
  733. controller.impairedProtocolClassification = make(map[string]int)
  734. }
  735. }
  736. // getImpairedProtocols returns a list of protocols that have sufficient
  737. // classifications to be considered impaired protocols.
  738. //
  739. // Concurrency note: only the runTunnels() goroutine may call getImpairedProtocols
  740. func (controller *Controller) getImpairedProtocols() []string {
  741. NoticeImpairedProtocolClassification(controller.impairedProtocolClassification)
  742. impairedProtocols := make([]string, 0)
  743. for protocol, count := range controller.impairedProtocolClassification {
  744. if count >= IMPAIRED_PROTOCOL_CLASSIFICATION_THRESHOLD {
  745. impairedProtocols = append(impairedProtocols, protocol)
  746. }
  747. }
  748. return impairedProtocols
  749. }
  750. // isImpairedProtocol checks if the specified protocol is classified as impaired.
  751. //
  752. // Concurrency note: only the runTunnels() goroutine may call isImpairedProtocol
  753. func (controller *Controller) isImpairedProtocol(protocol string) bool {
  754. count, ok := controller.impairedProtocolClassification[protocol]
  755. return ok && count >= IMPAIRED_PROTOCOL_CLASSIFICATION_THRESHOLD
  756. }
  757. // SignalSeededNewSLOK implements the TunnelOwner interface. This function
  758. // is called by Tunnel.operateTunnel when the tunnel has received a new,
  759. // previously unknown SLOK from the server. The Controller triggers an OSL
  760. // fetch, as the new SLOK may be sufficient to access new OSLs.
  761. func (controller *Controller) SignalSeededNewSLOK() {
  762. select {
  763. case controller.signalFetchObfuscatedServerLists <- *new(struct{}):
  764. default:
  765. }
  766. }
  767. // SignalTunnelFailure implements the TunnelOwner interface. This function
  768. // is called by Tunnel.operateTunnel when the tunnel has detected that it
  769. // has failed. The Controller will signal runTunnels to create a new
  770. // tunnel and/or remove the tunnel from the list of active tunnels.
  771. func (controller *Controller) SignalTunnelFailure(tunnel *Tunnel) {
  772. // Don't block. Assumes the receiver has a buffer large enough for
  773. // the typical number of operated tunnels. In case there's no room,
  774. // terminate the tunnel (runTunnels won't get a signal in this case,
  775. // but the tunnel will be removed from the list of active tunnels).
  776. select {
  777. case controller.failedTunnels <- tunnel:
  778. default:
  779. controller.terminateTunnel(tunnel)
  780. }
  781. }
  782. // discardTunnel disposes of a successful connection that is no longer required.
  783. func (controller *Controller) discardTunnel(tunnel *Tunnel) {
  784. NoticeInfo("discard tunnel: %s", tunnel.serverEntry.IpAddress)
  785. // TODO: not calling PromoteServerEntry, since that would rank the
  786. // discarded tunnel before fully active tunnels. Can a discarded tunnel
  787. // be promoted (since it connects), but with lower rank than all active
  788. // tunnels?
  789. tunnel.Close(true)
  790. }
  791. // registerTunnel adds the connected tunnel to the pool of active tunnels
  792. // which are candidates for port forwarding. Returns true if the pool has an
  793. // empty slot and false if the pool is full (caller should discard the tunnel).
  794. func (controller *Controller) registerTunnel(tunnel *Tunnel) bool {
  795. controller.tunnelMutex.Lock()
  796. defer controller.tunnelMutex.Unlock()
  797. if len(controller.tunnels) >= controller.config.TunnelPoolSize {
  798. return false
  799. }
  800. // Perform a final check just in case we've established
  801. // a duplicate connection.
  802. for _, activeTunnel := range controller.tunnels {
  803. if activeTunnel.serverEntry.IpAddress == tunnel.serverEntry.IpAddress {
  804. NoticeAlert("duplicate tunnel: %s", tunnel.serverEntry.IpAddress)
  805. return false
  806. }
  807. }
  808. controller.establishedOnce = true
  809. controller.tunnels = append(controller.tunnels, tunnel)
  810. NoticeTunnels(len(controller.tunnels))
  811. // Promote this successful tunnel to first rank so it's one
  812. // of the first candidates next time establish runs.
  813. // Connecting to a TargetServerEntry does not change the
  814. // ranking.
  815. if controller.config.TargetServerEntry == "" {
  816. PromoteServerEntry(tunnel.serverEntry.IpAddress)
  817. }
  818. return true
  819. }
  820. // hasEstablishedOnce indicates if at least one active tunnel has
  821. // been established up to this point. This is regardeless of how many
  822. // tunnels are presently active.
  823. func (controller *Controller) hasEstablishedOnce() bool {
  824. controller.tunnelMutex.Lock()
  825. defer controller.tunnelMutex.Unlock()
  826. return controller.establishedOnce
  827. }
  828. // isFullyEstablished indicates if the pool of active tunnels is full.
  829. func (controller *Controller) isFullyEstablished() bool {
  830. controller.tunnelMutex.Lock()
  831. defer controller.tunnelMutex.Unlock()
  832. return len(controller.tunnels) >= controller.config.TunnelPoolSize
  833. }
  834. // numTunnels returns the number of active and outstanding tunnels.
  835. // Oustanding is the number of tunnels required to fill the pool of
  836. // active tunnels.
  837. func (controller *Controller) numTunnels() (int, int) {
  838. controller.tunnelMutex.Lock()
  839. defer controller.tunnelMutex.Unlock()
  840. active := len(controller.tunnels)
  841. outstanding := controller.config.TunnelPoolSize - len(controller.tunnels)
  842. return active, outstanding
  843. }
  844. // terminateTunnel removes a tunnel from the pool of active tunnels
  845. // and closes the tunnel. The next-tunnel state used by getNextActiveTunnel
  846. // is adjusted as required.
  847. func (controller *Controller) terminateTunnel(tunnel *Tunnel) {
  848. controller.tunnelMutex.Lock()
  849. defer controller.tunnelMutex.Unlock()
  850. for index, activeTunnel := range controller.tunnels {
  851. if tunnel == activeTunnel {
  852. controller.tunnels = append(
  853. controller.tunnels[:index], controller.tunnels[index+1:]...)
  854. if controller.nextTunnel > index {
  855. controller.nextTunnel--
  856. }
  857. if controller.nextTunnel >= len(controller.tunnels) {
  858. controller.nextTunnel = 0
  859. }
  860. activeTunnel.Close(false)
  861. NoticeTunnels(len(controller.tunnels))
  862. break
  863. }
  864. }
  865. }
  866. // terminateAllTunnels empties the tunnel pool, closing all active tunnels.
  867. // This is used when shutting down the controller.
  868. func (controller *Controller) terminateAllTunnels() {
  869. controller.tunnelMutex.Lock()
  870. defer controller.tunnelMutex.Unlock()
  871. // Closing all tunnels in parallel. In an orderly shutdown, each tunnel
  872. // may take a few seconds to send a final status request. We only want
  873. // to wait as long as the single slowest tunnel.
  874. closeWaitGroup := new(sync.WaitGroup)
  875. closeWaitGroup.Add(len(controller.tunnels))
  876. for _, activeTunnel := range controller.tunnels {
  877. tunnel := activeTunnel
  878. go func() {
  879. defer closeWaitGroup.Done()
  880. tunnel.Close(false)
  881. }()
  882. }
  883. closeWaitGroup.Wait()
  884. controller.tunnels = make([]*Tunnel, 0)
  885. controller.nextTunnel = 0
  886. NoticeTunnels(len(controller.tunnels))
  887. }
  888. // getNextActiveTunnel returns the next tunnel from the pool of active
  889. // tunnels. Currently, tunnel selection order is simple round-robin.
  890. func (controller *Controller) getNextActiveTunnel() (tunnel *Tunnel) {
  891. controller.tunnelMutex.Lock()
  892. defer controller.tunnelMutex.Unlock()
  893. for i := len(controller.tunnels); i > 0; i-- {
  894. tunnel = controller.tunnels[controller.nextTunnel]
  895. controller.nextTunnel =
  896. (controller.nextTunnel + 1) % len(controller.tunnels)
  897. return tunnel
  898. }
  899. return nil
  900. }
  901. // isActiveTunnelServerEntry is used to check if there's already
  902. // an existing tunnel to a candidate server.
  903. func (controller *Controller) isActiveTunnelServerEntry(
  904. serverEntry *protocol.ServerEntry) bool {
  905. controller.tunnelMutex.Lock()
  906. defer controller.tunnelMutex.Unlock()
  907. for _, activeTunnel := range controller.tunnels {
  908. if activeTunnel.serverEntry.IpAddress == serverEntry.IpAddress {
  909. return true
  910. }
  911. }
  912. return false
  913. }
  914. // setClientVerificationPayloadForActiveTunnels triggers the client verification
  915. // request for all active tunnels.
  916. func (controller *Controller) setClientVerificationPayloadForActiveTunnels(
  917. clientVerificationPayload string) {
  918. controller.tunnelMutex.Lock()
  919. defer controller.tunnelMutex.Unlock()
  920. for _, activeTunnel := range controller.tunnels {
  921. activeTunnel.SetClientVerificationPayload(clientVerificationPayload)
  922. }
  923. }
  924. // Dial selects an active tunnel and establishes a port forward
  925. // connection through the selected tunnel. Failure to connect is considered
  926. // a port forward failure, for the purpose of monitoring tunnel health.
  927. func (controller *Controller) Dial(
  928. remoteAddr string, alwaysTunnel bool, downstreamConn net.Conn) (conn net.Conn, err error) {
  929. tunnel := controller.getNextActiveTunnel()
  930. if tunnel == nil {
  931. return nil, common.ContextError(errors.New("no active tunnels"))
  932. }
  933. // Perform split tunnel classification when feature is enabled, and if the remote
  934. // address is classified as untunneled, dial directly.
  935. if !alwaysTunnel && controller.config.SplitTunnelDnsServer != "" {
  936. host, _, err := net.SplitHostPort(remoteAddr)
  937. if err != nil {
  938. return nil, common.ContextError(err)
  939. }
  940. // Note: a possible optimization, when split tunnel is active and IsUntunneled performs
  941. // a DNS resolution in order to make its classification, is to reuse that IP address in
  942. // the following Dials so they do not need to make their own resolutions. However, the
  943. // way this is currently implemented ensures that, e.g., DNS geo load balancing occurs
  944. // relative to the outbound network.
  945. if controller.splitTunnelClassifier.IsUntunneled(host) {
  946. // TODO: track downstreamConn and close it when the DialTCP conn closes, as with tunnel.Dial conns?
  947. return DialTCP(remoteAddr, controller.untunneledDialConfig)
  948. }
  949. }
  950. tunneledConn, err := tunnel.Dial(remoteAddr, alwaysTunnel, downstreamConn)
  951. if err != nil {
  952. return nil, common.ContextError(err)
  953. }
  954. return tunneledConn, nil
  955. }
  956. // startEstablishing creates a pool of worker goroutines which will
  957. // attempt to establish tunnels to candidate servers. The candidates
  958. // are generated by another goroutine.
  959. func (controller *Controller) startEstablishing() {
  960. if controller.isEstablishing {
  961. return
  962. }
  963. NoticeInfo("start establishing")
  964. controller.concurrentEstablishTunnelsMutex.Lock()
  965. controller.concurrentEstablishTunnels = 0
  966. controller.concurrentMeekEstablishTunnels = 0
  967. controller.peakConcurrentEstablishTunnels = 0
  968. controller.peakConcurrentMeekEstablishTunnels = 0
  969. controller.concurrentEstablishTunnelsMutex.Unlock()
  970. aggressiveGarbageCollection()
  971. emitMemoryMetrics()
  972. controller.isEstablishing = true
  973. controller.establishWaitGroup = new(sync.WaitGroup)
  974. controller.stopEstablishingBroadcast = make(chan struct{})
  975. controller.candidateServerEntries = make(chan *candidateServerEntry)
  976. controller.establishPendingConns.Reset()
  977. // The server affinity mechanism attempts to favor the previously
  978. // used server when reconnecting. This is beneficial for user
  979. // applications which expect consistency in user IP address (for
  980. // example, a web site which prompts for additional user
  981. // authentication when the IP address changes).
  982. //
  983. // Only the very first server, as determined by
  984. // datastore.PromoteServerEntry(), is the server affinity candidate.
  985. // Concurrent connections attempts to many servers are launched
  986. // without delay, in case the affinity server connection fails.
  987. // While the affinity server connection is outstanding, when any
  988. // other connection is established, there is a short grace period
  989. // delay before delivering the established tunnel; this allows some
  990. // time for the affinity server connection to succeed first.
  991. // When the affinity server connection fails, any other established
  992. // tunnel is registered without delay.
  993. //
  994. // Note: the establishTunnelWorker that receives the affinity
  995. // candidate is solely resonsible for closing
  996. // controller.serverAffinityDoneBroadcast.
  997. //
  998. // Note: if config.EgressRegion or config.TunnelProtocol has changed
  999. // since the top server was promoted, the first server may not actually
  1000. // be the last connected server.
  1001. // TODO: should not favor the first server in this case
  1002. controller.serverAffinityDoneBroadcast = make(chan struct{})
  1003. for i := 0; i < controller.config.ConnectionWorkerPoolSize; i++ {
  1004. controller.establishWaitGroup.Add(1)
  1005. go controller.establishTunnelWorker()
  1006. }
  1007. controller.establishWaitGroup.Add(1)
  1008. go controller.establishCandidateGenerator(
  1009. controller.getImpairedProtocols())
  1010. }
  1011. // stopEstablishing signals the establish goroutines to stop and waits
  1012. // for the group to halt. pendingConns is used to interrupt any worker
  1013. // blocked on a socket connect.
  1014. func (controller *Controller) stopEstablishing() {
  1015. if !controller.isEstablishing {
  1016. return
  1017. }
  1018. NoticeInfo("stop establishing")
  1019. close(controller.stopEstablishingBroadcast)
  1020. // Note: interruptibleTCPClose doesn't really interrupt socket connects
  1021. // and may leave goroutines running for a time after the Wait call.
  1022. controller.establishPendingConns.CloseAll()
  1023. // Note: establishCandidateGenerator closes controller.candidateServerEntries
  1024. // (as it may be sending to that channel).
  1025. controller.establishWaitGroup.Wait()
  1026. NoticeInfo("stopped establishing")
  1027. controller.isEstablishing = false
  1028. controller.establishWaitGroup = nil
  1029. controller.stopEstablishingBroadcast = nil
  1030. controller.candidateServerEntries = nil
  1031. controller.serverAffinityDoneBroadcast = nil
  1032. controller.concurrentEstablishTunnelsMutex.Lock()
  1033. peakConcurrent := controller.peakConcurrentEstablishTunnels
  1034. peakConcurrentMeek := controller.peakConcurrentMeekEstablishTunnels
  1035. controller.concurrentEstablishTunnels = 0
  1036. controller.concurrentMeekEstablishTunnels = 0
  1037. controller.peakConcurrentEstablishTunnels = 0
  1038. controller.peakConcurrentMeekEstablishTunnels = 0
  1039. controller.concurrentEstablishTunnelsMutex.Unlock()
  1040. NoticeInfo("peak concurrent establish tunnels: %d", peakConcurrent)
  1041. NoticeInfo("peak concurrent meek establish tunnels: %d", peakConcurrentMeek)
  1042. emitMemoryMetrics()
  1043. standardGarbageCollection()
  1044. }
  1045. // establishCandidateGenerator populates the candidate queue with server entries
  1046. // from the data store. Server entries are iterated in rank order, so that promoted
  1047. // servers with higher rank are priority candidates.
  1048. func (controller *Controller) establishCandidateGenerator(impairedProtocols []string) {
  1049. defer controller.establishWaitGroup.Done()
  1050. defer close(controller.candidateServerEntries)
  1051. // establishStartTime is used to calculate and report the
  1052. // client's tunnel establishment duration.
  1053. //
  1054. // networkWaitDuration is the elapsed time spent waiting
  1055. // for network connectivity. This duration will be excluded
  1056. // from reported tunnel establishment duration.
  1057. establishStartTime := monotime.Now()
  1058. var networkWaitDuration time.Duration
  1059. iterator, err := NewServerEntryIterator(controller.config)
  1060. if err != nil {
  1061. NoticeAlert("failed to iterate over candidates: %s", err)
  1062. controller.SignalComponentFailure()
  1063. return
  1064. }
  1065. defer iterator.Close()
  1066. isServerAffinityCandidate := true
  1067. // TODO: reconcile server affinity scheme with multi-tunnel mode
  1068. if controller.config.TunnelPoolSize > 1 {
  1069. isServerAffinityCandidate = false
  1070. close(controller.serverAffinityDoneBroadcast)
  1071. }
  1072. loop:
  1073. // Repeat until stopped
  1074. for i := 0; ; i++ {
  1075. networkWaitStartTime := monotime.Now()
  1076. if !WaitForNetworkConnectivity(
  1077. controller.config.NetworkConnectivityChecker,
  1078. controller.stopEstablishingBroadcast,
  1079. controller.shutdownBroadcast) {
  1080. break loop
  1081. }
  1082. networkWaitDuration += monotime.Since(networkWaitStartTime)
  1083. // Send each iterator server entry to the establish workers
  1084. startTime := monotime.Now()
  1085. for {
  1086. serverEntry, err := iterator.Next()
  1087. if err != nil {
  1088. NoticeAlert("failed to get next candidate: %s", err)
  1089. controller.SignalComponentFailure()
  1090. break loop
  1091. }
  1092. if serverEntry == nil {
  1093. // Completed this iteration
  1094. break
  1095. }
  1096. if controller.config.TargetApiProtocol == protocol.PSIPHON_SSH_API_PROTOCOL &&
  1097. !serverEntry.SupportsSSHAPIRequests() {
  1098. continue
  1099. }
  1100. // Disable impaired protocols. This is only done for the
  1101. // first iteration of the ESTABLISH_TUNNEL_WORK_TIME
  1102. // loop since (a) one iteration should be sufficient to
  1103. // evade the attack; (b) there's a good chance of false
  1104. // positives (such as short tunnel durations due to network
  1105. // hopping on a mobile device).
  1106. // The edited serverEntry is temporary copy which is not
  1107. // stored or reused.
  1108. if i == 0 {
  1109. serverEntry.DisableImpairedProtocols(impairedProtocols)
  1110. if len(serverEntry.GetSupportedProtocols(false)) == 0 {
  1111. // Skip this server entry, as it has no supported
  1112. // protocols after disabling the impaired ones
  1113. // TODO: modify ServerEntryIterator to skip these?
  1114. continue
  1115. }
  1116. }
  1117. // adjustedEstablishStartTime is establishStartTime shifted
  1118. // to exclude time spent waiting for network connectivity.
  1119. candidate := &candidateServerEntry{
  1120. serverEntry: serverEntry,
  1121. isServerAffinityCandidate: isServerAffinityCandidate,
  1122. adjustedEstablishStartTime: establishStartTime.Add(networkWaitDuration),
  1123. }
  1124. wasServerAffinityCandidate := isServerAffinityCandidate
  1125. // Note: there must be only one server affinity candidate, as it
  1126. // closes the serverAffinityDoneBroadcast channel.
  1127. isServerAffinityCandidate = false
  1128. // TODO: here we could generate multiple candidates from the
  1129. // server entry when there are many MeekFrontingAddresses.
  1130. select {
  1131. case controller.candidateServerEntries <- candidate:
  1132. case <-controller.stopEstablishingBroadcast:
  1133. break loop
  1134. case <-controller.shutdownBroadcast:
  1135. break loop
  1136. }
  1137. if startTime.Add(ESTABLISH_TUNNEL_WORK_TIME).Before(monotime.Now()) {
  1138. // Start over, after a brief pause, with a new shuffle of the server
  1139. // entries, and potentially some newly fetched server entries.
  1140. break
  1141. }
  1142. if wasServerAffinityCandidate {
  1143. // Don't start the next candidate until either the server affinity
  1144. // candidate has completed (success or failure) or is still working
  1145. // and the grace period has elapsed.
  1146. timer := time.NewTimer(ESTABLISH_TUNNEL_SERVER_AFFINITY_GRACE_PERIOD)
  1147. select {
  1148. case <-timer.C:
  1149. case <-controller.serverAffinityDoneBroadcast:
  1150. case <-controller.stopEstablishingBroadcast:
  1151. break loop
  1152. case <-controller.shutdownBroadcast:
  1153. break loop
  1154. }
  1155. } else if controller.config.StaggerConnectionWorkersMilliseconds != 0 {
  1156. // Stagger concurrent connection workers.
  1157. timer := time.NewTimer(time.Millisecond * time.Duration(
  1158. controller.config.StaggerConnectionWorkersMilliseconds))
  1159. select {
  1160. case <-timer.C:
  1161. case <-controller.stopEstablishingBroadcast:
  1162. break loop
  1163. case <-controller.shutdownBroadcast:
  1164. break loop
  1165. }
  1166. }
  1167. }
  1168. // Free up resources now, but don't reset until after the pause.
  1169. iterator.Close()
  1170. // Trigger a common remote server list fetch, since we may have failed
  1171. // to connect with all known servers. Don't block sending signal, since
  1172. // this signal may have already been sent.
  1173. // Don't wait for fetch remote to succeed, since it may fail and
  1174. // enter a retry loop and we're better off trying more known servers.
  1175. // TODO: synchronize the fetch response, so it can be incorporated
  1176. // into the server entry iterator as soon as available.
  1177. select {
  1178. case controller.signalFetchCommonRemoteServerList <- *new(struct{}):
  1179. default:
  1180. }
  1181. // Trigger an OSL fetch in parallel. Both fetches are run in parallel
  1182. // so that if one out of the common RLS and OSL set is large, it doesn't
  1183. // doesn't entirely block fetching the other.
  1184. select {
  1185. case controller.signalFetchObfuscatedServerLists <- *new(struct{}):
  1186. default:
  1187. }
  1188. // Trigger an out-of-band upgrade availability check and download.
  1189. // Since we may have failed to connect, we may benefit from upgrading
  1190. // to a new client version with new circumvention capabilities.
  1191. select {
  1192. case controller.signalDownloadUpgrade <- "":
  1193. default:
  1194. }
  1195. // After a complete iteration of candidate servers, pause before iterating again.
  1196. // This helps avoid some busy wait loop conditions, and also allows some time for
  1197. // network conditions to change. Also allows for fetch remote to complete,
  1198. // in typical conditions (it isn't strictly necessary to wait for this, there will
  1199. // be more rounds if required).
  1200. timeout := time.After(
  1201. time.Duration(*controller.config.EstablishTunnelPausePeriodSeconds) * time.Second)
  1202. select {
  1203. case <-timeout:
  1204. // Retry iterating
  1205. case <-controller.stopEstablishingBroadcast:
  1206. break loop
  1207. case <-controller.shutdownBroadcast:
  1208. break loop
  1209. }
  1210. iterator.Reset()
  1211. }
  1212. }
  1213. // establishTunnelWorker pulls candidates from the candidate queue, establishes
  1214. // a connection to the tunnel server, and delivers the connected tunnel to a channel.
  1215. func (controller *Controller) establishTunnelWorker() {
  1216. defer controller.establishWaitGroup.Done()
  1217. loop:
  1218. for candidateServerEntry := range controller.candidateServerEntries {
  1219. // Note: don't receive from candidateServerEntries and stopEstablishingBroadcast
  1220. // in the same select, since we want to prioritize receiving the stop signal
  1221. if controller.isStopEstablishingBroadcast() {
  1222. break loop
  1223. }
  1224. // There may already be a tunnel to this candidate. If so, skip it.
  1225. if controller.isActiveTunnelServerEntry(candidateServerEntry.serverEntry) {
  1226. continue
  1227. }
  1228. // ConnectTunnel will allocate significant memory, so first attempt to
  1229. // reclaim as much as possible.
  1230. aggressiveGarbageCollection()
  1231. // Select the tunnel protocol. Unless config.TunnelProtocol is set, the
  1232. // selection will be made at random from protocols supported by the
  1233. // server entry.
  1234. //
  1235. // When limiting concurrent meek connection workers, and at the limit,
  1236. // do not select meek since otherwise the candidate must be skipped.
  1237. //
  1238. // If at the limit and unabled to select a non-meek protocol, skip the
  1239. // candidate entirely and move on to the next. Since candidates are shuffled
  1240. // it's probable that the next candidate is not meek. In this case, a
  1241. // StaggerConnectionWorkersMilliseconds delay may still be incurred.
  1242. excludeMeek := false
  1243. controller.concurrentEstablishTunnelsMutex.Lock()
  1244. if controller.config.LimitMeekConnectionWorkers > 0 &&
  1245. controller.concurrentMeekEstablishTunnels >=
  1246. controller.config.LimitMeekConnectionWorkers {
  1247. excludeMeek = true
  1248. }
  1249. controller.concurrentEstablishTunnelsMutex.Unlock()
  1250. selectedProtocol, err := selectProtocol(
  1251. controller.config, candidateServerEntry.serverEntry, excludeMeek)
  1252. if err == errProtocolNotSupported {
  1253. // selectProtocol returns errProtocolNotSupported when excludeMeek
  1254. // is set and the server entry only supports meek protocols.
  1255. // Skip this candidate.
  1256. continue
  1257. }
  1258. var tunnel *Tunnel
  1259. if err == nil {
  1260. isMeek := protocol.TunnelProtocolUsesMeek(selectedProtocol) ||
  1261. protocol.TunnelProtocolUsesMeek(selectedProtocol)
  1262. controller.concurrentEstablishTunnelsMutex.Lock()
  1263. if isMeek {
  1264. // Recheck the limit now that we know we're selecting meek and
  1265. // adjusting concurrentMeekEstablishTunnels.
  1266. if controller.config.LimitMeekConnectionWorkers > 0 &&
  1267. controller.concurrentMeekEstablishTunnels >=
  1268. controller.config.LimitMeekConnectionWorkers {
  1269. // Skip this candidate.
  1270. controller.concurrentEstablishTunnelsMutex.Unlock()
  1271. continue
  1272. }
  1273. controller.concurrentMeekEstablishTunnels += 1
  1274. if controller.concurrentMeekEstablishTunnels > controller.peakConcurrentMeekEstablishTunnels {
  1275. controller.peakConcurrentMeekEstablishTunnels = controller.concurrentMeekEstablishTunnels
  1276. }
  1277. }
  1278. controller.concurrentEstablishTunnels += 1
  1279. if controller.concurrentEstablishTunnels > controller.peakConcurrentEstablishTunnels {
  1280. controller.peakConcurrentEstablishTunnels = controller.concurrentEstablishTunnels
  1281. }
  1282. controller.concurrentEstablishTunnelsMutex.Unlock()
  1283. tunnel, err = ConnectTunnel(
  1284. controller.config,
  1285. controller.untunneledDialConfig,
  1286. controller.sessionId,
  1287. controller.establishPendingConns,
  1288. candidateServerEntry.serverEntry,
  1289. selectedProtocol,
  1290. candidateServerEntry.adjustedEstablishStartTime)
  1291. controller.concurrentEstablishTunnelsMutex.Lock()
  1292. if isMeek {
  1293. controller.concurrentMeekEstablishTunnels -= 1
  1294. }
  1295. controller.concurrentEstablishTunnels -= 1
  1296. controller.concurrentEstablishTunnelsMutex.Unlock()
  1297. }
  1298. // Periodically emit memory metrics during the establishment cycle.
  1299. if !controller.isStopEstablishingBroadcast() {
  1300. emitMemoryMetrics()
  1301. }
  1302. // Immediately reclaim memory allocated by the establishment. In the case
  1303. // of failure, first clear the reference to the tunnel. In the case of
  1304. // success, the garbage collection may still be effective as the initial
  1305. // phases of some protocols involve significant memory allocation that
  1306. // could now be reclaimed.
  1307. if err != nil {
  1308. tunnel = nil
  1309. }
  1310. aggressiveGarbageCollection()
  1311. if err != nil {
  1312. // Unblock other candidates immediately when
  1313. // server affinity candidate fails.
  1314. if candidateServerEntry.isServerAffinityCandidate {
  1315. close(controller.serverAffinityDoneBroadcast)
  1316. }
  1317. // Before emitting error, check if establish interrupted, in which
  1318. // case the error is noise.
  1319. if controller.isStopEstablishingBroadcast() {
  1320. break loop
  1321. }
  1322. NoticeInfo("failed to connect to %s: %s", candidateServerEntry.serverEntry.IpAddress, err)
  1323. continue
  1324. }
  1325. // Deliver connected tunnel.
  1326. // Don't block. Assumes the receiver has a buffer large enough for
  1327. // the number of desired tunnels. If there's no room, the tunnel must
  1328. // not be required so it's discarded.
  1329. select {
  1330. case controller.connectedTunnels <- tunnel:
  1331. default:
  1332. controller.discardTunnel(tunnel)
  1333. // Clear the reference to this discarded tunnel and immediately run
  1334. // a garbage collection to reclaim its memory.
  1335. tunnel = nil
  1336. aggressiveGarbageCollection()
  1337. }
  1338. // Unblock other candidates only after delivering when
  1339. // server affinity candidate succeeds.
  1340. if candidateServerEntry.isServerAffinityCandidate {
  1341. close(controller.serverAffinityDoneBroadcast)
  1342. }
  1343. }
  1344. }
  1345. func (controller *Controller) isStopEstablishingBroadcast() bool {
  1346. select {
  1347. case <-controller.stopEstablishingBroadcast:
  1348. return true
  1349. default:
  1350. }
  1351. return false
  1352. }