controller.go 57 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600
  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. timer := time.NewTimer(retryPeriod)
  350. select {
  351. case <-timer.C:
  352. case <-controller.shutdownBroadcast:
  353. timer.Stop()
  354. break fetcherLoop
  355. }
  356. }
  357. }
  358. NoticeInfo("exiting %s remote server list fetcher", name)
  359. }
  360. // establishTunnelWatcher terminates the controller if a tunnel
  361. // has not been established in the configured time period. This
  362. // is regardless of how many tunnels are presently active -- meaning
  363. // that if an active tunnel was established and lost the controller
  364. // is left running (to re-establish).
  365. func (controller *Controller) establishTunnelWatcher() {
  366. defer controller.runWaitGroup.Done()
  367. timer := time.NewTimer(
  368. time.Duration(*controller.config.EstablishTunnelTimeoutSeconds) * time.Second)
  369. defer timer.Stop()
  370. select {
  371. case <-timer.C:
  372. if !controller.hasEstablishedOnce() {
  373. NoticeAlert("failed to establish tunnel before timeout")
  374. controller.SignalComponentFailure()
  375. }
  376. case <-controller.shutdownBroadcast:
  377. }
  378. NoticeInfo("exiting establish tunnel watcher")
  379. }
  380. // connectedReporter sends periodic "connected" requests to the Psiphon API.
  381. // These requests are for server-side unique user stats calculation. See the
  382. // comment in DoConnectedRequest for a description of the request mechanism.
  383. // To ensure we don't over- or under-count unique users, only one connected
  384. // request is made across all simultaneous multi-tunnels; and the connected
  385. // request is repeated periodically for very long-lived tunnels.
  386. // The signalReportConnected mechanism is used to trigger another connected
  387. // request immediately after a reconnect.
  388. func (controller *Controller) connectedReporter() {
  389. defer controller.runWaitGroup.Done()
  390. loop:
  391. for {
  392. // Pick any active tunnel and make the next connected request. No error
  393. // is logged if there's no active tunnel, as that's not an unexpected condition.
  394. reported := false
  395. tunnel := controller.getNextActiveTunnel()
  396. if tunnel != nil {
  397. err := tunnel.serverContext.DoConnectedRequest()
  398. if err == nil {
  399. reported = true
  400. } else {
  401. NoticeAlert("failed to make connected request: %s", err)
  402. }
  403. }
  404. // Schedule the next connected request and wait.
  405. var duration time.Duration
  406. if reported {
  407. duration = PSIPHON_API_CONNECTED_REQUEST_PERIOD
  408. } else {
  409. duration = PSIPHON_API_CONNECTED_REQUEST_RETRY_PERIOD
  410. }
  411. timer := time.NewTimer(duration)
  412. doBreak := false
  413. select {
  414. case <-controller.signalReportConnected:
  415. case <-timer.C:
  416. // Make another connected request
  417. case <-controller.shutdownBroadcast:
  418. doBreak = true
  419. }
  420. timer.Stop()
  421. if doBreak {
  422. break loop
  423. }
  424. }
  425. NoticeInfo("exiting connected reporter")
  426. }
  427. func (controller *Controller) startOrSignalConnectedReporter() {
  428. // session is nil when DisableApi is set
  429. if controller.config.DisableApi {
  430. return
  431. }
  432. // Start the connected reporter after the first tunnel is established.
  433. // Concurrency note: only the runTunnels goroutine may access startedConnectedReporter.
  434. if !controller.startedConnectedReporter {
  435. controller.startedConnectedReporter = true
  436. controller.runWaitGroup.Add(1)
  437. go controller.connectedReporter()
  438. } else {
  439. select {
  440. case controller.signalReportConnected <- *new(struct{}):
  441. default:
  442. }
  443. }
  444. }
  445. // upgradeDownloader makes periodic attempts to complete a client upgrade
  446. // download. DownloadUpgrade() is resumable, so each attempt has potential for
  447. // getting closer to completion, even in conditions where the download or
  448. // tunnel is repeatedly interrupted.
  449. // An upgrade download is triggered by either a handshake response indicating
  450. // that a new version is available; or after failing to connect, in which case
  451. // it's useful to check, out-of-band, for an upgrade with new circumvention
  452. // capabilities.
  453. // Once the download operation completes successfully, the downloader exits
  454. // and is not run again: either there is not a newer version, or the upgrade
  455. // has been downloaded and is ready to be applied.
  456. // We're assuming that the upgrade will be applied and the entire system
  457. // restarted before another upgrade is to be downloaded.
  458. //
  459. // TODO: refactor upgrade downloader and remote server list fetcher to use
  460. // common code (including the resumable download routines).
  461. //
  462. func (controller *Controller) upgradeDownloader() {
  463. defer controller.runWaitGroup.Done()
  464. var lastDownloadTime monotime.Time
  465. downloadLoop:
  466. for {
  467. // Wait for a signal before downloading
  468. var handshakeVersion string
  469. select {
  470. case handshakeVersion = <-controller.signalDownloadUpgrade:
  471. case <-controller.shutdownBroadcast:
  472. break downloadLoop
  473. }
  474. // Unless handshake is explicitly advertizing a new version, skip
  475. // checking entirely when a recent download was successful.
  476. if handshakeVersion == "" &&
  477. lastDownloadTime != 0 &&
  478. lastDownloadTime.Add(DOWNLOAD_UPGRADE_STALE_PERIOD).After(monotime.Now()) {
  479. continue
  480. }
  481. retryLoop:
  482. for attempt := 0; ; attempt++ {
  483. // Don't attempt to download while there is no network connectivity,
  484. // to avoid alert notice noise.
  485. if !WaitForNetworkConnectivity(
  486. controller.config.NetworkConnectivityChecker,
  487. controller.shutdownBroadcast) {
  488. break downloadLoop
  489. }
  490. // Pick any active tunnel and make the next download attempt. If there's
  491. // no active tunnel, the untunneledDialConfig will be used.
  492. tunnel := controller.getNextActiveTunnel()
  493. err := DownloadUpgrade(
  494. controller.config,
  495. attempt,
  496. handshakeVersion,
  497. tunnel,
  498. controller.untunneledDialConfig)
  499. if err == nil {
  500. lastDownloadTime = monotime.Now()
  501. break retryLoop
  502. }
  503. NoticeAlert("failed to download upgrade: %s", err)
  504. timer := time.NewTimer(
  505. time.Duration(*controller.config.DownloadUpgradeRetryPeriodSeconds) * time.Second)
  506. select {
  507. case <-timer.C:
  508. case <-controller.shutdownBroadcast:
  509. timer.Stop()
  510. break downloadLoop
  511. }
  512. }
  513. }
  514. NoticeInfo("exiting upgrade downloader")
  515. }
  516. // runTunnels is the controller tunnel management main loop. It starts and stops
  517. // establishing tunnels based on the target tunnel pool size and the current size
  518. // of the pool. Tunnels are established asynchronously using worker goroutines.
  519. //
  520. // When there are no server entries for the target region/protocol, the
  521. // establishCandidateGenerator will yield no candidates and wait before
  522. // trying again. In the meantime, a remote server entry fetch may supply
  523. // valid candidates.
  524. //
  525. // When a tunnel is established, it's added to the active pool. The tunnel's
  526. // operateTunnel goroutine monitors the tunnel.
  527. //
  528. // When a tunnel fails, it's removed from the pool and the establish process is
  529. // restarted to fill the pool.
  530. func (controller *Controller) runTunnels() {
  531. defer controller.runWaitGroup.Done()
  532. var clientVerificationPayload string
  533. // Start running
  534. controller.startEstablishing()
  535. loop:
  536. for {
  537. select {
  538. case failedTunnel := <-controller.failedTunnels:
  539. NoticeAlert("tunnel failed: %s", failedTunnel.serverEntry.IpAddress)
  540. controller.terminateTunnel(failedTunnel)
  541. // Note: we make this extra check to ensure the shutdown signal takes priority
  542. // and that we do not start establishing. Critically, startEstablishing() calls
  543. // establishPendingConns.Reset() which clears the closed flag in
  544. // establishPendingConns; this causes the pendingConns.Add() within
  545. // interruptibleTCPDial to succeed instead of aborting, and the result
  546. // is that it's possible for establish goroutines to run all the way through
  547. // NewServerContext before being discarded... delaying shutdown.
  548. select {
  549. case <-controller.shutdownBroadcast:
  550. break loop
  551. default:
  552. }
  553. controller.classifyImpairedProtocol(failedTunnel)
  554. // Clear the reference to this tunnel before calling startEstablishing,
  555. // which will invoke a garbage collection.
  556. failedTunnel = nil
  557. // Concurrency note: only this goroutine may call startEstablishing/stopEstablishing,
  558. // which reference controller.isEstablishing.
  559. controller.startEstablishing()
  560. case connectedTunnel := <-controller.connectedTunnels:
  561. if controller.isImpairedProtocol(connectedTunnel.protocol) {
  562. // Protocol was classified as impaired while this tunnel established.
  563. // This is most likely to occur with TunnelPoolSize > 0. We log the
  564. // event but take no action. Discarding the tunnel would break the
  565. // impaired logic unless we did that (a) only if there are other
  566. // unimpaired protocols; (b) only during the first iteration of the
  567. // ESTABLISH_TUNNEL_WORK_TIME loop. By not discarding here, a true
  568. // impaired protocol may require an extra reconnect.
  569. NoticeAlert("connected tunnel with impaired protocol: %s", connectedTunnel.protocol)
  570. }
  571. // Tunnel establishment has two phases: connection and activation.
  572. //
  573. // Connection is run concurrently by the establishTunnelWorkers, to minimize
  574. // delay when it's not yet known which server and protocol will be available
  575. // and unblocked.
  576. //
  577. // Activation is run serially, here, to minimize the overhead of making a
  578. // handshake request and starting the operateTunnel management worker for a
  579. // tunnel which may be discarded.
  580. //
  581. // When the active tunnel will complete establishment, establishment is
  582. // stopped before activation. This interrupts all connecting tunnels and
  583. // garbage collects their memory. The purpose is to minimize memory
  584. // pressure when the handshake request is made. In the unlikely case that the
  585. // handshake fails, establishment is restarted.
  586. //
  587. // Any delays in stopEstablishing will delay the handshake for the last
  588. // active tunnel.
  589. //
  590. // In the typical case of TunnelPoolSize of 1, only a single handshake is
  591. // performed and the homepages notices file, when used, will not be modifed
  592. // after the NoticeTunnels(1) [i.e., connected] until NoticeTunnels(0) [i.e.,
  593. // disconnected]. For TunnelPoolSize > 1, serial handshakes only ensures that
  594. // each set of emitted NoticeHomepages is contiguous.
  595. active, outstanding := controller.numTunnels()
  596. // discardTunnel will be true here when already fully established.
  597. discardTunnel := (outstanding <= 0)
  598. isFirstTunnel := (active == 0)
  599. isLastTunnel := (outstanding == 1)
  600. if !discardTunnel {
  601. if isLastTunnel {
  602. controller.stopEstablishing()
  603. }
  604. err := connectedTunnel.Activate(controller, controller.shutdownBroadcast)
  605. if err != nil {
  606. // Assume the Activate failed due to a broken tunnel connection,
  607. // currently the most likely case, and classify as impaired, as in
  608. // the failed tunnel case above.
  609. // TODO: distinguish between network and other errors
  610. controller.classifyImpairedProtocol(connectedTunnel)
  611. NoticeAlert("failed to activate %s: %s", connectedTunnel.serverEntry.IpAddress, err)
  612. discardTunnel = true
  613. } else {
  614. // It's unlikely that registerTunnel will fail, since only this goroutine
  615. // calls registerTunnel -- and after checking numTunnels; so failure is not
  616. // expected.
  617. if !controller.registerTunnel(connectedTunnel) {
  618. NoticeAlert("failed to register %s: %s", connectedTunnel.serverEntry.IpAddress)
  619. discardTunnel = true
  620. }
  621. }
  622. // May need to replace this tunnel
  623. if isLastTunnel && discardTunnel {
  624. controller.startEstablishing()
  625. }
  626. }
  627. if discardTunnel {
  628. controller.discardTunnel(connectedTunnel)
  629. // Clear the reference to this discarded tunnel and immediately run
  630. // a garbage collection to reclaim its memory.
  631. connectedTunnel = nil
  632. aggressiveGarbageCollection()
  633. // Skip the rest of this case
  634. break
  635. }
  636. NoticeActiveTunnel(
  637. connectedTunnel.serverEntry.IpAddress,
  638. connectedTunnel.protocol,
  639. connectedTunnel.serverEntry.SupportsSSHAPIRequests())
  640. if isFirstTunnel {
  641. // The split tunnel classifier is started once the first tunnel is
  642. // established. This first tunnel is passed in to be used to make
  643. // the routes data request.
  644. // A long-running controller may run while the host device is present
  645. // in different regions. In this case, we want the split tunnel logic
  646. // to switch to routes for new regions and not classify traffic based
  647. // on routes installed for older regions.
  648. // We assume that when regions change, the host network will also
  649. // change, and so all tunnels will fail and be re-established. Under
  650. // that assumption, the classifier will be re-Start()-ed here when
  651. // the region has changed.
  652. controller.splitTunnelClassifier.Start(connectedTunnel)
  653. // Signal a connected request on each 1st tunnel establishment. For
  654. // multi-tunnels, the session is connected as long as at least one
  655. // tunnel is established.
  656. controller.startOrSignalConnectedReporter()
  657. // If the handshake indicated that a new client version is available,
  658. // trigger an upgrade download.
  659. // Note: serverContext is nil when DisableApi is set
  660. if connectedTunnel.serverContext != nil &&
  661. connectedTunnel.serverContext.clientUpgradeVersion != "" {
  662. handshakeVersion := connectedTunnel.serverContext.clientUpgradeVersion
  663. select {
  664. case controller.signalDownloadUpgrade <- handshakeVersion:
  665. default:
  666. }
  667. }
  668. }
  669. // Set the new tunnel as the transport for the packet tunnel. The packet tunnel
  670. // client remains up when reestablishing, but no packets are relayed while there
  671. // is no connected tunnel. UseTunnel will establish a new packet tunnel SSH
  672. // channel over the new SSH tunnel and configure the packet tunnel client to use
  673. // the new SSH channel as its transport.
  674. //
  675. // Note: as is, this logic is suboptimal for TunnelPoolSize > 1, as this would
  676. // continuously initialize new packet tunnel sessions for each established
  677. // server. For now, config validation requires TunnelPoolSize == 1 when
  678. // the packet tunnel is used.
  679. if controller.packetTunnelTransport != nil {
  680. controller.packetTunnelTransport.UseTunnel(connectedTunnel)
  681. }
  682. // TODO: design issue -- might not be enough server entries with region/caps to ever fill tunnel slots;
  683. // possible solution is establish target MIN(CountServerEntries(region, protocol), TunnelPoolSize)
  684. if controller.isFullyEstablished() {
  685. controller.stopEstablishing()
  686. }
  687. case clientVerificationPayload = <-controller.newClientVerificationPayload:
  688. controller.setClientVerificationPayloadForActiveTunnels(clientVerificationPayload)
  689. case <-controller.shutdownBroadcast:
  690. break loop
  691. }
  692. }
  693. // Stop running
  694. controller.stopEstablishing()
  695. controller.terminateAllTunnels()
  696. // Drain tunnel channels
  697. close(controller.connectedTunnels)
  698. for tunnel := range controller.connectedTunnels {
  699. controller.discardTunnel(tunnel)
  700. }
  701. close(controller.failedTunnels)
  702. for tunnel := range controller.failedTunnels {
  703. controller.discardTunnel(tunnel)
  704. }
  705. NoticeInfo("exiting run tunnels")
  706. }
  707. // TerminateNextActiveTunnel is a support routine for
  708. // test code that must terminate the active tunnel and
  709. // restart establishing. This function is not guaranteed
  710. // to be safe for use in other cases.
  711. func (controller *Controller) TerminateNextActiveTunnel() {
  712. tunnel := controller.getNextActiveTunnel()
  713. if tunnel != nil {
  714. controller.SignalTunnelFailure(tunnel)
  715. NoticeInfo("terminated tunnel: %s", tunnel.serverEntry.IpAddress)
  716. }
  717. }
  718. // classifyImpairedProtocol tracks "impaired" protocol classifications for failed
  719. // tunnels. A protocol is classified as impaired if a tunnel using that protocol
  720. // fails, repeatedly, shortly after the start of the connection. During tunnel
  721. // establishment, impaired protocols are briefly skipped.
  722. //
  723. // One purpose of this measure is to defend against an attack where the adversary,
  724. // for example, tags an OSSH TCP connection as an "unidentified" protocol; allows
  725. // it to connect; but then kills the underlying TCP connection after a short time.
  726. // Since OSSH has less latency than other protocols that may bypass an "unidentified"
  727. // filter, these other protocols might never be selected for use.
  728. //
  729. // Concurrency note: only the runTunnels() goroutine may call classifyImpairedProtocol
  730. func (controller *Controller) classifyImpairedProtocol(failedTunnel *Tunnel) {
  731. // If the tunnel failed while activating, its establishedTime will be 0.
  732. if failedTunnel.establishedTime == 0 ||
  733. failedTunnel.establishedTime.Add(IMPAIRED_PROTOCOL_CLASSIFICATION_DURATION).After(monotime.Now()) {
  734. controller.impairedProtocolClassification[failedTunnel.protocol] += 1
  735. } else {
  736. controller.impairedProtocolClassification[failedTunnel.protocol] = 0
  737. }
  738. // Reset classification once all known protocols are classified as impaired, as
  739. // there is now no way to proceed with only unimpaired protocols. The network
  740. // situation (or attack) resulting in classification may not be protocol-specific.
  741. //
  742. // Note: with controller.config.TunnelProtocol set, this will always reset once
  743. // that protocol has reached IMPAIRED_PROTOCOL_CLASSIFICATION_THRESHOLD.
  744. if CountNonImpairedProtocols(
  745. controller.config.EgressRegion,
  746. controller.config.TunnelProtocol,
  747. controller.getImpairedProtocols()) == 0 {
  748. controller.impairedProtocolClassification = make(map[string]int)
  749. }
  750. }
  751. // getImpairedProtocols returns a list of protocols that have sufficient
  752. // classifications to be considered impaired protocols.
  753. //
  754. // Concurrency note: only the runTunnels() goroutine may call getImpairedProtocols
  755. func (controller *Controller) getImpairedProtocols() []string {
  756. NoticeImpairedProtocolClassification(controller.impairedProtocolClassification)
  757. impairedProtocols := make([]string, 0)
  758. for protocol, count := range controller.impairedProtocolClassification {
  759. if count >= IMPAIRED_PROTOCOL_CLASSIFICATION_THRESHOLD {
  760. impairedProtocols = append(impairedProtocols, protocol)
  761. }
  762. }
  763. return impairedProtocols
  764. }
  765. // isImpairedProtocol checks if the specified protocol is classified as impaired.
  766. //
  767. // Concurrency note: only the runTunnels() goroutine may call isImpairedProtocol
  768. func (controller *Controller) isImpairedProtocol(protocol string) bool {
  769. count, ok := controller.impairedProtocolClassification[protocol]
  770. return ok && count >= IMPAIRED_PROTOCOL_CLASSIFICATION_THRESHOLD
  771. }
  772. // SignalSeededNewSLOK implements the TunnelOwner interface. This function
  773. // is called by Tunnel.operateTunnel when the tunnel has received a new,
  774. // previously unknown SLOK from the server. The Controller triggers an OSL
  775. // fetch, as the new SLOK may be sufficient to access new OSLs.
  776. func (controller *Controller) SignalSeededNewSLOK() {
  777. select {
  778. case controller.signalFetchObfuscatedServerLists <- *new(struct{}):
  779. default:
  780. }
  781. }
  782. // SignalTunnelFailure implements the TunnelOwner interface. This function
  783. // is called by Tunnel.operateTunnel when the tunnel has detected that it
  784. // has failed. The Controller will signal runTunnels to create a new
  785. // tunnel and/or remove the tunnel from the list of active tunnels.
  786. func (controller *Controller) SignalTunnelFailure(tunnel *Tunnel) {
  787. // Don't block. Assumes the receiver has a buffer large enough for
  788. // the typical number of operated tunnels. In case there's no room,
  789. // terminate the tunnel (runTunnels won't get a signal in this case,
  790. // but the tunnel will be removed from the list of active tunnels).
  791. select {
  792. case controller.failedTunnels <- tunnel:
  793. default:
  794. controller.terminateTunnel(tunnel)
  795. }
  796. }
  797. // discardTunnel disposes of a successful connection that is no longer required.
  798. func (controller *Controller) discardTunnel(tunnel *Tunnel) {
  799. NoticeInfo("discard tunnel: %s", tunnel.serverEntry.IpAddress)
  800. // TODO: not calling PromoteServerEntry, since that would rank the
  801. // discarded tunnel before fully active tunnels. Can a discarded tunnel
  802. // be promoted (since it connects), but with lower rank than all active
  803. // tunnels?
  804. tunnel.Close(true)
  805. }
  806. // registerTunnel adds the connected tunnel to the pool of active tunnels
  807. // which are candidates for port forwarding. Returns true if the pool has an
  808. // empty slot and false if the pool is full (caller should discard the tunnel).
  809. func (controller *Controller) registerTunnel(tunnel *Tunnel) bool {
  810. controller.tunnelMutex.Lock()
  811. defer controller.tunnelMutex.Unlock()
  812. if len(controller.tunnels) >= controller.config.TunnelPoolSize {
  813. return false
  814. }
  815. // Perform a final check just in case we've established
  816. // a duplicate connection.
  817. for _, activeTunnel := range controller.tunnels {
  818. if activeTunnel.serverEntry.IpAddress == tunnel.serverEntry.IpAddress {
  819. NoticeAlert("duplicate tunnel: %s", tunnel.serverEntry.IpAddress)
  820. return false
  821. }
  822. }
  823. controller.establishedOnce = true
  824. controller.tunnels = append(controller.tunnels, tunnel)
  825. NoticeTunnels(len(controller.tunnels))
  826. // Promote this successful tunnel to first rank so it's one
  827. // of the first candidates next time establish runs.
  828. // Connecting to a TargetServerEntry does not change the
  829. // ranking.
  830. if controller.config.TargetServerEntry == "" {
  831. PromoteServerEntry(tunnel.serverEntry.IpAddress)
  832. }
  833. return true
  834. }
  835. // hasEstablishedOnce indicates if at least one active tunnel has
  836. // been established up to this point. This is regardeless of how many
  837. // tunnels are presently active.
  838. func (controller *Controller) hasEstablishedOnce() bool {
  839. controller.tunnelMutex.Lock()
  840. defer controller.tunnelMutex.Unlock()
  841. return controller.establishedOnce
  842. }
  843. // isFullyEstablished indicates if the pool of active tunnels is full.
  844. func (controller *Controller) isFullyEstablished() bool {
  845. controller.tunnelMutex.Lock()
  846. defer controller.tunnelMutex.Unlock()
  847. return len(controller.tunnels) >= controller.config.TunnelPoolSize
  848. }
  849. // numTunnels returns the number of active and outstanding tunnels.
  850. // Oustanding is the number of tunnels required to fill the pool of
  851. // active tunnels.
  852. func (controller *Controller) numTunnels() (int, int) {
  853. controller.tunnelMutex.Lock()
  854. defer controller.tunnelMutex.Unlock()
  855. active := len(controller.tunnels)
  856. outstanding := controller.config.TunnelPoolSize - len(controller.tunnels)
  857. return active, outstanding
  858. }
  859. // terminateTunnel removes a tunnel from the pool of active tunnels
  860. // and closes the tunnel. The next-tunnel state used by getNextActiveTunnel
  861. // is adjusted as required.
  862. func (controller *Controller) terminateTunnel(tunnel *Tunnel) {
  863. controller.tunnelMutex.Lock()
  864. defer controller.tunnelMutex.Unlock()
  865. for index, activeTunnel := range controller.tunnels {
  866. if tunnel == activeTunnel {
  867. controller.tunnels = append(
  868. controller.tunnels[:index], controller.tunnels[index+1:]...)
  869. if controller.nextTunnel > index {
  870. controller.nextTunnel--
  871. }
  872. if controller.nextTunnel >= len(controller.tunnels) {
  873. controller.nextTunnel = 0
  874. }
  875. activeTunnel.Close(false)
  876. NoticeTunnels(len(controller.tunnels))
  877. break
  878. }
  879. }
  880. }
  881. // terminateAllTunnels empties the tunnel pool, closing all active tunnels.
  882. // This is used when shutting down the controller.
  883. func (controller *Controller) terminateAllTunnels() {
  884. controller.tunnelMutex.Lock()
  885. defer controller.tunnelMutex.Unlock()
  886. // Closing all tunnels in parallel. In an orderly shutdown, each tunnel
  887. // may take a few seconds to send a final status request. We only want
  888. // to wait as long as the single slowest tunnel.
  889. closeWaitGroup := new(sync.WaitGroup)
  890. closeWaitGroup.Add(len(controller.tunnels))
  891. for _, activeTunnel := range controller.tunnels {
  892. tunnel := activeTunnel
  893. go func() {
  894. defer closeWaitGroup.Done()
  895. tunnel.Close(false)
  896. }()
  897. }
  898. closeWaitGroup.Wait()
  899. controller.tunnels = make([]*Tunnel, 0)
  900. controller.nextTunnel = 0
  901. NoticeTunnels(len(controller.tunnels))
  902. }
  903. // getNextActiveTunnel returns the next tunnel from the pool of active
  904. // tunnels. Currently, tunnel selection order is simple round-robin.
  905. func (controller *Controller) getNextActiveTunnel() (tunnel *Tunnel) {
  906. controller.tunnelMutex.Lock()
  907. defer controller.tunnelMutex.Unlock()
  908. for i := len(controller.tunnels); i > 0; i-- {
  909. tunnel = controller.tunnels[controller.nextTunnel]
  910. controller.nextTunnel =
  911. (controller.nextTunnel + 1) % len(controller.tunnels)
  912. return tunnel
  913. }
  914. return nil
  915. }
  916. // isActiveTunnelServerEntry is used to check if there's already
  917. // an existing tunnel to a candidate server.
  918. func (controller *Controller) isActiveTunnelServerEntry(
  919. serverEntry *protocol.ServerEntry) bool {
  920. controller.tunnelMutex.Lock()
  921. defer controller.tunnelMutex.Unlock()
  922. for _, activeTunnel := range controller.tunnels {
  923. if activeTunnel.serverEntry.IpAddress == serverEntry.IpAddress {
  924. return true
  925. }
  926. }
  927. return false
  928. }
  929. // setClientVerificationPayloadForActiveTunnels triggers the client verification
  930. // request for all active tunnels.
  931. func (controller *Controller) setClientVerificationPayloadForActiveTunnels(
  932. clientVerificationPayload string) {
  933. controller.tunnelMutex.Lock()
  934. defer controller.tunnelMutex.Unlock()
  935. for _, activeTunnel := range controller.tunnels {
  936. activeTunnel.SetClientVerificationPayload(clientVerificationPayload)
  937. }
  938. }
  939. // Dial selects an active tunnel and establishes a port forward
  940. // connection through the selected tunnel. Failure to connect is considered
  941. // a port forward failure, for the purpose of monitoring tunnel health.
  942. func (controller *Controller) Dial(
  943. remoteAddr string, alwaysTunnel bool, downstreamConn net.Conn) (conn net.Conn, err error) {
  944. tunnel := controller.getNextActiveTunnel()
  945. if tunnel == nil {
  946. return nil, common.ContextError(errors.New("no active tunnels"))
  947. }
  948. // Perform split tunnel classification when feature is enabled, and if the remote
  949. // address is classified as untunneled, dial directly.
  950. if !alwaysTunnel && controller.config.SplitTunnelDnsServer != "" {
  951. host, _, err := net.SplitHostPort(remoteAddr)
  952. if err != nil {
  953. return nil, common.ContextError(err)
  954. }
  955. // Note: a possible optimization, when split tunnel is active and IsUntunneled performs
  956. // a DNS resolution in order to make its classification, is to reuse that IP address in
  957. // the following Dials so they do not need to make their own resolutions. However, the
  958. // way this is currently implemented ensures that, e.g., DNS geo load balancing occurs
  959. // relative to the outbound network.
  960. if controller.splitTunnelClassifier.IsUntunneled(host) {
  961. // TODO: track downstreamConn and close it when the DialTCP conn closes, as with tunnel.Dial conns?
  962. return DialTCP(remoteAddr, controller.untunneledDialConfig)
  963. }
  964. }
  965. tunneledConn, err := tunnel.Dial(remoteAddr, alwaysTunnel, downstreamConn)
  966. if err != nil {
  967. return nil, common.ContextError(err)
  968. }
  969. return tunneledConn, nil
  970. }
  971. // startEstablishing creates a pool of worker goroutines which will
  972. // attempt to establish tunnels to candidate servers. The candidates
  973. // are generated by another goroutine.
  974. func (controller *Controller) startEstablishing() {
  975. if controller.isEstablishing {
  976. return
  977. }
  978. NoticeInfo("start establishing")
  979. controller.concurrentEstablishTunnelsMutex.Lock()
  980. controller.concurrentEstablishTunnels = 0
  981. controller.concurrentMeekEstablishTunnels = 0
  982. controller.peakConcurrentEstablishTunnels = 0
  983. controller.peakConcurrentMeekEstablishTunnels = 0
  984. controller.concurrentEstablishTunnelsMutex.Unlock()
  985. aggressiveGarbageCollection()
  986. emitMemoryMetrics()
  987. controller.isEstablishing = true
  988. controller.establishWaitGroup = new(sync.WaitGroup)
  989. controller.stopEstablishingBroadcast = make(chan struct{})
  990. controller.candidateServerEntries = make(chan *candidateServerEntry)
  991. controller.establishPendingConns.Reset()
  992. // The server affinity mechanism attempts to favor the previously
  993. // used server when reconnecting. This is beneficial for user
  994. // applications which expect consistency in user IP address (for
  995. // example, a web site which prompts for additional user
  996. // authentication when the IP address changes).
  997. //
  998. // Only the very first server, as determined by
  999. // datastore.PromoteServerEntry(), is the server affinity candidate.
  1000. // Concurrent connections attempts to many servers are launched
  1001. // without delay, in case the affinity server connection fails.
  1002. // While the affinity server connection is outstanding, when any
  1003. // other connection is established, there is a short grace period
  1004. // delay before delivering the established tunnel; this allows some
  1005. // time for the affinity server connection to succeed first.
  1006. // When the affinity server connection fails, any other established
  1007. // tunnel is registered without delay.
  1008. //
  1009. // Note: the establishTunnelWorker that receives the affinity
  1010. // candidate is solely resonsible for closing
  1011. // controller.serverAffinityDoneBroadcast.
  1012. //
  1013. // Note: if config.EgressRegion or config.TunnelProtocol has changed
  1014. // since the top server was promoted, the first server may not actually
  1015. // be the last connected server.
  1016. // TODO: should not favor the first server in this case
  1017. controller.serverAffinityDoneBroadcast = make(chan struct{})
  1018. for i := 0; i < controller.config.ConnectionWorkerPoolSize; i++ {
  1019. controller.establishWaitGroup.Add(1)
  1020. go controller.establishTunnelWorker()
  1021. }
  1022. controller.establishWaitGroup.Add(1)
  1023. go controller.establishCandidateGenerator(
  1024. controller.getImpairedProtocols())
  1025. }
  1026. // stopEstablishing signals the establish goroutines to stop and waits
  1027. // for the group to halt. pendingConns is used to interrupt any worker
  1028. // blocked on a socket connect.
  1029. func (controller *Controller) stopEstablishing() {
  1030. if !controller.isEstablishing {
  1031. return
  1032. }
  1033. NoticeInfo("stop establishing")
  1034. close(controller.stopEstablishingBroadcast)
  1035. // Note: interruptibleTCPClose doesn't really interrupt socket connects
  1036. // and may leave goroutines running for a time after the Wait call.
  1037. controller.establishPendingConns.CloseAll()
  1038. // Note: establishCandidateGenerator closes controller.candidateServerEntries
  1039. // (as it may be sending to that channel).
  1040. controller.establishWaitGroup.Wait()
  1041. NoticeInfo("stopped establishing")
  1042. controller.isEstablishing = false
  1043. controller.establishWaitGroup = nil
  1044. controller.stopEstablishingBroadcast = nil
  1045. controller.candidateServerEntries = nil
  1046. controller.serverAffinityDoneBroadcast = nil
  1047. controller.concurrentEstablishTunnelsMutex.Lock()
  1048. peakConcurrent := controller.peakConcurrentEstablishTunnels
  1049. peakConcurrentMeek := controller.peakConcurrentMeekEstablishTunnels
  1050. controller.concurrentEstablishTunnels = 0
  1051. controller.concurrentMeekEstablishTunnels = 0
  1052. controller.peakConcurrentEstablishTunnels = 0
  1053. controller.peakConcurrentMeekEstablishTunnels = 0
  1054. controller.concurrentEstablishTunnelsMutex.Unlock()
  1055. NoticeInfo("peak concurrent establish tunnels: %d", peakConcurrent)
  1056. NoticeInfo("peak concurrent meek establish tunnels: %d", peakConcurrentMeek)
  1057. emitMemoryMetrics()
  1058. standardGarbageCollection()
  1059. }
  1060. // establishCandidateGenerator populates the candidate queue with server entries
  1061. // from the data store. Server entries are iterated in rank order, so that promoted
  1062. // servers with higher rank are priority candidates.
  1063. func (controller *Controller) establishCandidateGenerator(impairedProtocols []string) {
  1064. defer controller.establishWaitGroup.Done()
  1065. defer close(controller.candidateServerEntries)
  1066. // establishStartTime is used to calculate and report the
  1067. // client's tunnel establishment duration.
  1068. //
  1069. // networkWaitDuration is the elapsed time spent waiting
  1070. // for network connectivity. This duration will be excluded
  1071. // from reported tunnel establishment duration.
  1072. establishStartTime := monotime.Now()
  1073. var networkWaitDuration time.Duration
  1074. iterator, err := NewServerEntryIterator(controller.config)
  1075. if err != nil {
  1076. NoticeAlert("failed to iterate over candidates: %s", err)
  1077. controller.SignalComponentFailure()
  1078. return
  1079. }
  1080. defer iterator.Close()
  1081. isServerAffinityCandidate := true
  1082. // TODO: reconcile server affinity scheme with multi-tunnel mode
  1083. if controller.config.TunnelPoolSize > 1 {
  1084. isServerAffinityCandidate = false
  1085. close(controller.serverAffinityDoneBroadcast)
  1086. }
  1087. loop:
  1088. // Repeat until stopped
  1089. for i := 0; ; i++ {
  1090. networkWaitStartTime := monotime.Now()
  1091. if !WaitForNetworkConnectivity(
  1092. controller.config.NetworkConnectivityChecker,
  1093. controller.stopEstablishingBroadcast,
  1094. controller.shutdownBroadcast) {
  1095. break loop
  1096. }
  1097. networkWaitDuration += monotime.Since(networkWaitStartTime)
  1098. // Send each iterator server entry to the establish workers
  1099. startTime := monotime.Now()
  1100. for {
  1101. serverEntry, err := iterator.Next()
  1102. if err != nil {
  1103. NoticeAlert("failed to get next candidate: %s", err)
  1104. controller.SignalComponentFailure()
  1105. break loop
  1106. }
  1107. if serverEntry == nil {
  1108. // Completed this iteration
  1109. break
  1110. }
  1111. if controller.config.TargetApiProtocol == protocol.PSIPHON_SSH_API_PROTOCOL &&
  1112. !serverEntry.SupportsSSHAPIRequests() {
  1113. continue
  1114. }
  1115. // Disable impaired protocols. This is only done for the
  1116. // first iteration of the ESTABLISH_TUNNEL_WORK_TIME
  1117. // loop since (a) one iteration should be sufficient to
  1118. // evade the attack; (b) there's a good chance of false
  1119. // positives (such as short tunnel durations due to network
  1120. // hopping on a mobile device).
  1121. // The edited serverEntry is temporary copy which is not
  1122. // stored or reused.
  1123. if i == 0 {
  1124. serverEntry.DisableImpairedProtocols(impairedProtocols)
  1125. if len(serverEntry.GetSupportedProtocols(false)) == 0 {
  1126. // Skip this server entry, as it has no supported
  1127. // protocols after disabling the impaired ones
  1128. // TODO: modify ServerEntryIterator to skip these?
  1129. continue
  1130. }
  1131. }
  1132. // adjustedEstablishStartTime is establishStartTime shifted
  1133. // to exclude time spent waiting for network connectivity.
  1134. candidate := &candidateServerEntry{
  1135. serverEntry: serverEntry,
  1136. isServerAffinityCandidate: isServerAffinityCandidate,
  1137. adjustedEstablishStartTime: establishStartTime.Add(networkWaitDuration),
  1138. }
  1139. wasServerAffinityCandidate := isServerAffinityCandidate
  1140. // Note: there must be only one server affinity candidate, as it
  1141. // closes the serverAffinityDoneBroadcast channel.
  1142. isServerAffinityCandidate = false
  1143. // TODO: here we could generate multiple candidates from the
  1144. // server entry when there are many MeekFrontingAddresses.
  1145. select {
  1146. case controller.candidateServerEntries <- candidate:
  1147. case <-controller.stopEstablishingBroadcast:
  1148. break loop
  1149. case <-controller.shutdownBroadcast:
  1150. break loop
  1151. }
  1152. if startTime.Add(ESTABLISH_TUNNEL_WORK_TIME).Before(monotime.Now()) {
  1153. // Start over, after a brief pause, with a new shuffle of the server
  1154. // entries, and potentially some newly fetched server entries.
  1155. break
  1156. }
  1157. if wasServerAffinityCandidate {
  1158. // Don't start the next candidate until either the server affinity
  1159. // candidate has completed (success or failure) or is still working
  1160. // and the grace period has elapsed.
  1161. timer := time.NewTimer(ESTABLISH_TUNNEL_SERVER_AFFINITY_GRACE_PERIOD)
  1162. doBreak := false
  1163. select {
  1164. case <-timer.C:
  1165. case <-controller.serverAffinityDoneBroadcast:
  1166. case <-controller.stopEstablishingBroadcast:
  1167. doBreak = true
  1168. case <-controller.shutdownBroadcast:
  1169. doBreak = true
  1170. }
  1171. timer.Stop()
  1172. if doBreak {
  1173. break loop
  1174. }
  1175. } else if controller.config.StaggerConnectionWorkersMilliseconds != 0 {
  1176. // Stagger concurrent connection workers.
  1177. timer := time.NewTimer(time.Millisecond * time.Duration(
  1178. controller.config.StaggerConnectionWorkersMilliseconds))
  1179. doBreak := false
  1180. select {
  1181. case <-timer.C:
  1182. case <-controller.stopEstablishingBroadcast:
  1183. doBreak = true
  1184. case <-controller.shutdownBroadcast:
  1185. doBreak = true
  1186. }
  1187. timer.Stop()
  1188. if doBreak {
  1189. break loop
  1190. }
  1191. }
  1192. }
  1193. // Free up resources now, but don't reset until after the pause.
  1194. iterator.Close()
  1195. // Trigger a common remote server list fetch, since we may have failed
  1196. // to connect with all known servers. Don't block sending signal, since
  1197. // this signal may have already been sent.
  1198. // Don't wait for fetch remote to succeed, since it may fail and
  1199. // enter a retry loop and we're better off trying more known servers.
  1200. // TODO: synchronize the fetch response, so it can be incorporated
  1201. // into the server entry iterator as soon as available.
  1202. select {
  1203. case controller.signalFetchCommonRemoteServerList <- *new(struct{}):
  1204. default:
  1205. }
  1206. // Trigger an OSL fetch in parallel. Both fetches are run in parallel
  1207. // so that if one out of the common RLS and OSL set is large, it doesn't
  1208. // doesn't entirely block fetching the other.
  1209. select {
  1210. case controller.signalFetchObfuscatedServerLists <- *new(struct{}):
  1211. default:
  1212. }
  1213. // Trigger an out-of-band upgrade availability check and download.
  1214. // Since we may have failed to connect, we may benefit from upgrading
  1215. // to a new client version with new circumvention capabilities.
  1216. select {
  1217. case controller.signalDownloadUpgrade <- "":
  1218. default:
  1219. }
  1220. // After a complete iteration of candidate servers, pause before iterating again.
  1221. // This helps avoid some busy wait loop conditions, and also allows some time for
  1222. // network conditions to change. Also allows for fetch remote to complete,
  1223. // in typical conditions (it isn't strictly necessary to wait for this, there will
  1224. // be more rounds if required).
  1225. timer := time.NewTimer(
  1226. time.Duration(*controller.config.EstablishTunnelPausePeriodSeconds) * time.Second)
  1227. doBreak := false
  1228. select {
  1229. case <-timer.C:
  1230. // Retry iterating
  1231. case <-controller.stopEstablishingBroadcast:
  1232. doBreak = true
  1233. case <-controller.shutdownBroadcast:
  1234. doBreak = true
  1235. }
  1236. timer.Stop()
  1237. if doBreak {
  1238. break loop
  1239. }
  1240. iterator.Reset()
  1241. }
  1242. }
  1243. // establishTunnelWorker pulls candidates from the candidate queue, establishes
  1244. // a connection to the tunnel server, and delivers the connected tunnel to a channel.
  1245. func (controller *Controller) establishTunnelWorker() {
  1246. defer controller.establishWaitGroup.Done()
  1247. loop:
  1248. for candidateServerEntry := range controller.candidateServerEntries {
  1249. // Note: don't receive from candidateServerEntries and stopEstablishingBroadcast
  1250. // in the same select, since we want to prioritize receiving the stop signal
  1251. if controller.isStopEstablishingBroadcast() {
  1252. break loop
  1253. }
  1254. // There may already be a tunnel to this candidate. If so, skip it.
  1255. if controller.isActiveTunnelServerEntry(candidateServerEntry.serverEntry) {
  1256. continue
  1257. }
  1258. // ConnectTunnel will allocate significant memory, so first attempt to
  1259. // reclaim as much as possible.
  1260. aggressiveGarbageCollection()
  1261. // Select the tunnel protocol. Unless config.TunnelProtocol is set, the
  1262. // selection will be made at random from protocols supported by the
  1263. // server entry.
  1264. //
  1265. // When limiting concurrent meek connection workers, and at the limit,
  1266. // do not select meek since otherwise the candidate must be skipped.
  1267. //
  1268. // If at the limit and unabled to select a non-meek protocol, skip the
  1269. // candidate entirely and move on to the next. Since candidates are shuffled
  1270. // it's probable that the next candidate is not meek. In this case, a
  1271. // StaggerConnectionWorkersMilliseconds delay may still be incurred.
  1272. excludeMeek := false
  1273. controller.concurrentEstablishTunnelsMutex.Lock()
  1274. if controller.config.LimitMeekConnectionWorkers > 0 &&
  1275. controller.concurrentMeekEstablishTunnels >=
  1276. controller.config.LimitMeekConnectionWorkers {
  1277. excludeMeek = true
  1278. }
  1279. controller.concurrentEstablishTunnelsMutex.Unlock()
  1280. selectedProtocol, err := selectProtocol(
  1281. controller.config, candidateServerEntry.serverEntry, excludeMeek)
  1282. if err == errProtocolNotSupported {
  1283. // selectProtocol returns errProtocolNotSupported when excludeMeek
  1284. // is set and the server entry only supports meek protocols.
  1285. // Skip this candidate.
  1286. continue
  1287. }
  1288. var tunnel *Tunnel
  1289. if err == nil {
  1290. isMeek := protocol.TunnelProtocolUsesMeek(selectedProtocol) ||
  1291. protocol.TunnelProtocolUsesMeek(selectedProtocol)
  1292. controller.concurrentEstablishTunnelsMutex.Lock()
  1293. if isMeek {
  1294. // Recheck the limit now that we know we're selecting meek and
  1295. // adjusting concurrentMeekEstablishTunnels.
  1296. if controller.config.LimitMeekConnectionWorkers > 0 &&
  1297. controller.concurrentMeekEstablishTunnels >=
  1298. controller.config.LimitMeekConnectionWorkers {
  1299. // Skip this candidate.
  1300. controller.concurrentEstablishTunnelsMutex.Unlock()
  1301. continue
  1302. }
  1303. controller.concurrentMeekEstablishTunnels += 1
  1304. if controller.concurrentMeekEstablishTunnels > controller.peakConcurrentMeekEstablishTunnels {
  1305. controller.peakConcurrentMeekEstablishTunnels = controller.concurrentMeekEstablishTunnels
  1306. }
  1307. }
  1308. controller.concurrentEstablishTunnels += 1
  1309. if controller.concurrentEstablishTunnels > controller.peakConcurrentEstablishTunnels {
  1310. controller.peakConcurrentEstablishTunnels = controller.concurrentEstablishTunnels
  1311. }
  1312. controller.concurrentEstablishTunnelsMutex.Unlock()
  1313. tunnel, err = ConnectTunnel(
  1314. controller.config,
  1315. controller.untunneledDialConfig,
  1316. controller.sessionId,
  1317. controller.establishPendingConns,
  1318. candidateServerEntry.serverEntry,
  1319. selectedProtocol,
  1320. candidateServerEntry.adjustedEstablishStartTime)
  1321. controller.concurrentEstablishTunnelsMutex.Lock()
  1322. if isMeek {
  1323. controller.concurrentMeekEstablishTunnels -= 1
  1324. }
  1325. controller.concurrentEstablishTunnels -= 1
  1326. controller.concurrentEstablishTunnelsMutex.Unlock()
  1327. }
  1328. // Periodically emit memory metrics during the establishment cycle.
  1329. if !controller.isStopEstablishingBroadcast() {
  1330. emitMemoryMetrics()
  1331. }
  1332. // Immediately reclaim memory allocated by the establishment. In the case
  1333. // of failure, first clear the reference to the tunnel. In the case of
  1334. // success, the garbage collection may still be effective as the initial
  1335. // phases of some protocols involve significant memory allocation that
  1336. // could now be reclaimed.
  1337. if err != nil {
  1338. tunnel = nil
  1339. }
  1340. aggressiveGarbageCollection()
  1341. if err != nil {
  1342. // Unblock other candidates immediately when
  1343. // server affinity candidate fails.
  1344. if candidateServerEntry.isServerAffinityCandidate {
  1345. close(controller.serverAffinityDoneBroadcast)
  1346. }
  1347. // Before emitting error, check if establish interrupted, in which
  1348. // case the error is noise.
  1349. if controller.isStopEstablishingBroadcast() {
  1350. break loop
  1351. }
  1352. NoticeInfo("failed to connect to %s: %s", candidateServerEntry.serverEntry.IpAddress, err)
  1353. continue
  1354. }
  1355. // Deliver connected tunnel.
  1356. // Don't block. Assumes the receiver has a buffer large enough for
  1357. // the number of desired tunnels. If there's no room, the tunnel must
  1358. // not be required so it's discarded.
  1359. select {
  1360. case controller.connectedTunnels <- tunnel:
  1361. default:
  1362. controller.discardTunnel(tunnel)
  1363. // Clear the reference to this discarded tunnel and immediately run
  1364. // a garbage collection to reclaim its memory.
  1365. tunnel = nil
  1366. aggressiveGarbageCollection()
  1367. }
  1368. // Unblock other candidates only after delivering when
  1369. // server affinity candidate succeeds.
  1370. if candidateServerEntry.isServerAffinityCandidate {
  1371. close(controller.serverAffinityDoneBroadcast)
  1372. }
  1373. }
  1374. }
  1375. func (controller *Controller) isStopEstablishingBroadcast() bool {
  1376. select {
  1377. case <-controller.stopEstablishingBroadcast:
  1378. return true
  1379. default:
  1380. }
  1381. return false
  1382. }