serverApi.go 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170
  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
  20. import (
  21. "bytes"
  22. "context"
  23. "encoding/base64"
  24. "encoding/hex"
  25. "encoding/json"
  26. "fmt"
  27. "io"
  28. "io/ioutil"
  29. "net"
  30. "net/http"
  31. "net/url"
  32. "strconv"
  33. "strings"
  34. "time"
  35. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
  36. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/buildinfo"
  37. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/errors"
  38. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/fragmentor"
  39. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/parameters"
  40. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/prng"
  41. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/protocol"
  42. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/tactics"
  43. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/transferstats"
  44. )
  45. // ServerContext is a utility struct which holds all of the data associated
  46. // with a Psiphon server connection. In addition to the established tunnel, this
  47. // includes data and transport mechanisms for Psiphon API requests. Legacy servers
  48. // offer the Psiphon API through a web service; newer servers offer the Psiphon
  49. // API through SSH requests made directly through the tunnel's SSH client.
  50. type ServerContext struct {
  51. tunnel *Tunnel
  52. psiphonHttpsClient *http.Client
  53. statsRegexps *transferstats.Regexps
  54. clientRegion string
  55. clientUpgradeVersion string
  56. serverHandshakeTimestamp string
  57. paddingPRNG *prng.PRNG
  58. }
  59. // MakeSessionId creates a new session ID. The same session ID is used across
  60. // multi-tunnel controller runs, where each tunnel has its own ServerContext
  61. // instance.
  62. // In server-side stats, we now consider a "session" to be the lifetime of the
  63. // Controller (e.g., the user's commanded start and stop) and we measure this
  64. // duration as well as the duration of each tunnel within the session.
  65. func MakeSessionId() (string, error) {
  66. randomId, err := common.MakeSecureRandomBytes(protocol.PSIPHON_API_CLIENT_SESSION_ID_LENGTH)
  67. if err != nil {
  68. return "", errors.Trace(err)
  69. }
  70. return hex.EncodeToString(randomId), nil
  71. }
  72. // NewServerContext makes the tunneled handshake request to the Psiphon server
  73. // and returns a ServerContext struct for use with subsequent Psiphon server API
  74. // requests (e.g., periodic connected and status requests).
  75. func NewServerContext(tunnel *Tunnel) (*ServerContext, error) {
  76. // For legacy servers, set up psiphonHttpsClient for
  77. // accessing the Psiphon API via the web service.
  78. var psiphonHttpsClient *http.Client
  79. if !tunnel.dialParams.ServerEntry.SupportsSSHAPIRequests() ||
  80. tunnel.config.TargetApiProtocol == protocol.PSIPHON_WEB_API_PROTOCOL {
  81. var err error
  82. psiphonHttpsClient, err = makePsiphonHttpsClient(tunnel)
  83. if err != nil {
  84. return nil, errors.Trace(err)
  85. }
  86. }
  87. serverContext := &ServerContext{
  88. tunnel: tunnel,
  89. psiphonHttpsClient: psiphonHttpsClient,
  90. paddingPRNG: prng.NewPRNGWithSeed(tunnel.dialParams.APIRequestPaddingSeed),
  91. }
  92. ignoreRegexps := tunnel.config.GetClientParameters().Get().Bool(
  93. parameters.IgnoreHandshakeStatsRegexps)
  94. err := serverContext.doHandshakeRequest(ignoreRegexps)
  95. if err != nil {
  96. return nil, errors.Trace(err)
  97. }
  98. return serverContext, nil
  99. }
  100. // doHandshakeRequest performs the "handshake" API request. The handshake
  101. // returns upgrade info, newly discovered server entries -- which are
  102. // stored -- and sponsor info (home pages, stat regexes).
  103. func (serverContext *ServerContext) doHandshakeRequest(
  104. ignoreStatsRegexps bool) error {
  105. params := serverContext.getBaseAPIParameters(baseParametersAll)
  106. // The server will return a signed copy of its own server entry when the
  107. // client specifies this 'missing_server_entry_signature' parameter.
  108. //
  109. // The purpose of this mechanism is to rapidly upgrade client local storage
  110. // from unsigned to signed server entries, and to ensure that the client has
  111. // a signed server entry for its currently connected server as required for
  112. // the client-to-client exchange feature.
  113. //
  114. // The server entry will be included in handshakeResponse.EncodedServerList,
  115. // along side discovery servers.
  116. requestedMissingSignature := false
  117. if !serverContext.tunnel.dialParams.ServerEntry.HasSignature() {
  118. requestedMissingSignature = true
  119. params["missing_server_entry_signature"] =
  120. serverContext.tunnel.dialParams.ServerEntry.Tag
  121. }
  122. doTactics := !serverContext.tunnel.config.DisableTactics
  123. networkID := ""
  124. if doTactics {
  125. // Limitation: it is assumed that the network ID obtained here is the
  126. // one that is active when the handshake request is received by the
  127. // server. However, it is remotely possible to switch networks
  128. // immediately after invoking the GetNetworkID callback and initiating
  129. // the handshake, if the tunnel protocol is meek.
  130. //
  131. // The response handling code below calls GetNetworkID again and ignores
  132. // any tactics payload if the network ID is not the same. While this
  133. // doesn't detect all cases of changing networks, it reduces the already
  134. // narrow window.
  135. networkID = serverContext.tunnel.config.GetNetworkID()
  136. err := tactics.SetTacticsAPIParameters(
  137. serverContext.tunnel.config.clientParameters,
  138. GetTacticsStorer(serverContext.tunnel.config),
  139. networkID,
  140. params)
  141. if err != nil {
  142. return errors.Trace(err)
  143. }
  144. }
  145. var response []byte
  146. if serverContext.psiphonHttpsClient == nil {
  147. params[protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS] =
  148. serverContext.tunnel.config.GetAuthorizations()
  149. request, err := serverContext.makeSSHAPIRequestPayload(params)
  150. if err != nil {
  151. return errors.Trace(err)
  152. }
  153. response, err = serverContext.tunnel.SendAPIRequest(
  154. protocol.PSIPHON_API_HANDSHAKE_REQUEST_NAME, request)
  155. if err != nil {
  156. return errors.Trace(err)
  157. }
  158. } else {
  159. // Legacy web service API request
  160. responseBody, err := serverContext.doGetRequest(
  161. makeRequestUrl(serverContext.tunnel, "", "handshake", params))
  162. if err != nil {
  163. return errors.Trace(err)
  164. }
  165. // Skip legacy format lines and just parse the JSON config line
  166. configLinePrefix := []byte("Config: ")
  167. for _, line := range bytes.Split(responseBody, []byte("\n")) {
  168. if bytes.HasPrefix(line, configLinePrefix) {
  169. response = line[len(configLinePrefix):]
  170. break
  171. }
  172. }
  173. if len(response) == 0 {
  174. return errors.TraceNew("no config line found")
  175. }
  176. }
  177. // Legacy fields:
  178. // - 'preemptive_reconnect_lifetime_milliseconds' is unused and ignored
  179. // - 'ssh_session_id' is ignored; client session ID is used instead
  180. var handshakeResponse protocol.HandshakeResponse
  181. // Initialize these fields to distinguish between psiphond omitting values in
  182. // the response and the zero value, which means unlimited rate.
  183. handshakeResponse.UpstreamBytesPerSecond = -1
  184. handshakeResponse.DownstreamBytesPerSecond = -1
  185. err := json.Unmarshal(response, &handshakeResponse)
  186. if err != nil {
  187. return errors.Trace(err)
  188. }
  189. serverContext.clientRegion = handshakeResponse.ClientRegion
  190. NoticeClientRegion(serverContext.clientRegion)
  191. var serverEntries []protocol.ServerEntryFields
  192. // Store discovered server entries
  193. // We use the server's time, as it's available here, for the server entry
  194. // timestamp since this is more reliable than the client time.
  195. for _, encodedServerEntry := range handshakeResponse.EncodedServerList {
  196. serverEntryFields, err := protocol.DecodeServerEntryFields(
  197. encodedServerEntry,
  198. common.TruncateTimestampToHour(handshakeResponse.ServerTimestamp),
  199. protocol.SERVER_ENTRY_SOURCE_DISCOVERY)
  200. if err != nil {
  201. return errors.Trace(err)
  202. }
  203. // Retain the original timestamp and source in the requestedMissingSignature
  204. // case, as this server entry was not discovered here.
  205. //
  206. // Limitation: there is a transient edge case where
  207. // requestedMissingSignature will be set for a discovery server entry that
  208. // _is_ also discovered here.
  209. if requestedMissingSignature &&
  210. serverEntryFields.GetIPAddress() == serverContext.tunnel.dialParams.ServerEntry.IpAddress {
  211. serverEntryFields.SetLocalTimestamp(serverContext.tunnel.dialParams.ServerEntry.LocalTimestamp)
  212. serverEntryFields.SetLocalSource(serverContext.tunnel.dialParams.ServerEntry.LocalSource)
  213. }
  214. err = protocol.ValidateServerEntryFields(serverEntryFields)
  215. if err != nil {
  216. // Skip this entry and continue with the next one
  217. NoticeWarning("invalid handshake server entry: %s", err)
  218. continue
  219. }
  220. serverEntries = append(serverEntries, serverEntryFields)
  221. }
  222. err = StoreServerEntries(
  223. serverContext.tunnel.config,
  224. serverEntries,
  225. true)
  226. if err != nil {
  227. return errors.Trace(err)
  228. }
  229. NoticeHomepages(handshakeResponse.Homepages)
  230. serverContext.clientUpgradeVersion = handshakeResponse.UpgradeClientVersion
  231. if handshakeResponse.UpgradeClientVersion != "" {
  232. NoticeClientUpgradeAvailable(handshakeResponse.UpgradeClientVersion)
  233. } else {
  234. NoticeClientIsLatestVersion("")
  235. }
  236. if !ignoreStatsRegexps {
  237. var regexpsNotices []string
  238. serverContext.statsRegexps, regexpsNotices = transferstats.MakeRegexps(
  239. handshakeResponse.PageViewRegexes,
  240. handshakeResponse.HttpsRequestRegexes)
  241. for _, notice := range regexpsNotices {
  242. NoticeWarning(notice)
  243. }
  244. }
  245. serverContext.serverHandshakeTimestamp = handshakeResponse.ServerTimestamp
  246. NoticeServerTimestamp(serverContext.serverHandshakeTimestamp)
  247. NoticeActiveAuthorizationIDs(handshakeResponse.ActiveAuthorizationIDs)
  248. NoticeTrafficRateLimits(
  249. handshakeResponse.UpstreamBytesPerSecond, handshakeResponse.DownstreamBytesPerSecond)
  250. if doTactics && handshakeResponse.TacticsPayload != nil &&
  251. networkID == serverContext.tunnel.config.GetNetworkID() {
  252. var payload *tactics.Payload
  253. err := json.Unmarshal(handshakeResponse.TacticsPayload, &payload)
  254. if err != nil {
  255. return errors.Trace(err)
  256. }
  257. // handshakeResponse.TacticsPayload may be "null", and payload
  258. // will successfully unmarshal as nil. As a result, the previous
  259. // handshakeResponse.TacticsPayload != nil test is insufficient.
  260. if payload != nil {
  261. tacticsRecord, err := tactics.HandleTacticsPayload(
  262. GetTacticsStorer(serverContext.tunnel.config),
  263. networkID,
  264. payload)
  265. if err != nil {
  266. return errors.Trace(err)
  267. }
  268. if tacticsRecord != nil &&
  269. prng.FlipWeightedCoin(tacticsRecord.Tactics.Probability) {
  270. err := serverContext.tunnel.config.SetClientParameters(
  271. tacticsRecord.Tag, true, tacticsRecord.Tactics.Parameters)
  272. if err != nil {
  273. NoticeInfo("apply handshake tactics failed: %s", err)
  274. }
  275. // The error will be due to invalid tactics values from
  276. // the server. When ApplyClientParameters fails, all
  277. // previous tactics values are left in place.
  278. }
  279. }
  280. }
  281. return nil
  282. }
  283. // DoConnectedRequest performs the "connected" API request. This request is
  284. // used for statistics, including unique user counting; reporting the full
  285. // tunnel establishment duration including the handshake request; and updated
  286. // fragmentor metrics.
  287. //
  288. // Users are not assigned identifiers. Instead, daily unique users are
  289. // calculated by having clients submit their last connected timestamp
  290. // (truncated to an hour, as a privacy measure). As client clocks are
  291. // unreliable, the server returns new last_connected values for the client to
  292. // store and send next time it connects.
  293. func (serverContext *ServerContext) DoConnectedRequest() error {
  294. // Limitation: as currently implemented, the last_connected exchange isn't a
  295. // distributed, atomic operation. When clients send the connected request,
  296. // the server may receive the request, count a unique user based on the
  297. // client's last_connected, and then the tunnel fails before the client
  298. // receives the response, so the client will not update its last_connected
  299. // value and submit the same one again, resulting in an inflated unique user
  300. // count.
  301. //
  302. // The SetInFlightConnectedRequest mechanism mitigates one class of connected
  303. // request interruption, a commanded shutdown in the middle of a connected
  304. // request, by allowing some time for the request to complete before
  305. // terminating the tunnel.
  306. //
  307. // TODO: consider extending the connected request protocol with additional
  308. // "acknowledgment" messages so that the server does not commit its unique
  309. // user count until after the client has acknowledged receipt and durable
  310. // storage of the new last_connected value.
  311. requestDone := make(chan struct{})
  312. defer close(requestDone)
  313. if !serverContext.tunnel.SetInFlightConnectedRequest(requestDone) {
  314. return errors.TraceNew("tunnel is closing")
  315. }
  316. defer serverContext.tunnel.SetInFlightConnectedRequest(nil)
  317. params := serverContext.getBaseAPIParameters(
  318. baseParametersOnlyUpstreamFragmentorDialParameters)
  319. lastConnected, err := getLastConnected()
  320. if err != nil {
  321. return errors.Trace(err)
  322. }
  323. params["last_connected"] = lastConnected
  324. // serverContext.tunnel.establishDuration is nanoseconds; report milliseconds
  325. params["establishment_duration"] =
  326. fmt.Sprintf("%d", serverContext.tunnel.establishDuration/time.Millisecond)
  327. var response []byte
  328. if serverContext.psiphonHttpsClient == nil {
  329. request, err := serverContext.makeSSHAPIRequestPayload(params)
  330. if err != nil {
  331. return errors.Trace(err)
  332. }
  333. response, err = serverContext.tunnel.SendAPIRequest(
  334. protocol.PSIPHON_API_CONNECTED_REQUEST_NAME, request)
  335. if err != nil {
  336. return errors.Trace(err)
  337. }
  338. } else {
  339. // Legacy web service API request
  340. response, err = serverContext.doGetRequest(
  341. makeRequestUrl(serverContext.tunnel, "", "connected", params))
  342. if err != nil {
  343. return errors.Trace(err)
  344. }
  345. }
  346. var connectedResponse protocol.ConnectedResponse
  347. err = json.Unmarshal(response, &connectedResponse)
  348. if err != nil {
  349. return errors.Trace(err)
  350. }
  351. err = SetKeyValue(
  352. datastoreLastConnectedKey, connectedResponse.ConnectedTimestamp)
  353. if err != nil {
  354. return errors.Trace(err)
  355. }
  356. return nil
  357. }
  358. func getLastConnected() (string, error) {
  359. lastConnected, err := GetKeyValue(datastoreLastConnectedKey)
  360. if err != nil {
  361. return "", errors.Trace(err)
  362. }
  363. if lastConnected == "" {
  364. lastConnected = "None"
  365. }
  366. return lastConnected, nil
  367. }
  368. // StatsRegexps gets the Regexps used for the statistics for this tunnel.
  369. func (serverContext *ServerContext) StatsRegexps() *transferstats.Regexps {
  370. return serverContext.statsRegexps
  371. }
  372. // DoStatusRequest makes a "status" API request to the server, sending session stats.
  373. func (serverContext *ServerContext) DoStatusRequest(tunnel *Tunnel) error {
  374. params := serverContext.getBaseAPIParameters(baseParametersNoDialParameters)
  375. // Note: ensure putBackStatusRequestPayload is called, to replace
  376. // payload for future attempt, in all failure cases.
  377. statusPayload, statusPayloadInfo, err := makeStatusRequestPayload(
  378. serverContext.tunnel.config,
  379. tunnel.dialParams.ServerEntry.IpAddress)
  380. if err != nil {
  381. return errors.Trace(err)
  382. }
  383. // Skip the request when there's no payload to send.
  384. if len(statusPayload) == 0 {
  385. return nil
  386. }
  387. var response []byte
  388. if serverContext.psiphonHttpsClient == nil {
  389. rawMessage := json.RawMessage(statusPayload)
  390. params["statusData"] = &rawMessage
  391. var request []byte
  392. request, err = serverContext.makeSSHAPIRequestPayload(params)
  393. if err == nil {
  394. response, err = serverContext.tunnel.SendAPIRequest(
  395. protocol.PSIPHON_API_STATUS_REQUEST_NAME, request)
  396. }
  397. } else {
  398. // Legacy web service API request
  399. response, err = serverContext.doPostRequest(
  400. makeRequestUrl(serverContext.tunnel, "", "status", params),
  401. "application/json",
  402. bytes.NewReader(statusPayload))
  403. }
  404. if err != nil {
  405. // Resend the transfer stats and tunnel stats later
  406. // Note: potential duplicate reports if the server received and processed
  407. // the request but the client failed to receive the response.
  408. putBackStatusRequestPayload(statusPayloadInfo)
  409. return errors.Trace(err)
  410. }
  411. confirmStatusRequestPayload(statusPayloadInfo)
  412. var statusResponse protocol.StatusResponse
  413. err = json.Unmarshal(response, &statusResponse)
  414. if err != nil {
  415. return errors.Trace(err)
  416. }
  417. for _, serverEntryTag := range statusResponse.InvalidServerEntryTags {
  418. PruneServerEntry(serverContext.tunnel.config, serverEntryTag)
  419. }
  420. return nil
  421. }
  422. // statusRequestPayloadInfo is a temporary structure for data used to
  423. // either "clear" or "put back" status request payload data depending
  424. // on whether or not the request succeeded.
  425. type statusRequestPayloadInfo struct {
  426. serverId string
  427. transferStats *transferstats.AccumulatedStats
  428. persistentStats map[string][][]byte
  429. }
  430. func makeStatusRequestPayload(
  431. config *Config,
  432. serverId string) ([]byte, *statusRequestPayloadInfo, error) {
  433. transferStats := transferstats.TakeOutStatsForServer(serverId)
  434. hostBytes := transferStats.GetStatsForStatusRequest()
  435. persistentStats, err := TakeOutUnreportedPersistentStats(config)
  436. if err != nil {
  437. NoticeWarning(
  438. "TakeOutUnreportedPersistentStats failed: %s", errors.Trace(err))
  439. persistentStats = nil
  440. // Proceed with transferStats only
  441. }
  442. if len(hostBytes) == 0 && len(persistentStats) == 0 {
  443. // There is no payload to send.
  444. return nil, nil, nil
  445. }
  446. payloadInfo := &statusRequestPayloadInfo{
  447. serverId, transferStats, persistentStats}
  448. payload := make(map[string]interface{})
  449. payload["host_bytes"] = hostBytes
  450. // We're not recording these fields, but legacy servers require them.
  451. payload["bytes_transferred"] = 0
  452. payload["page_views"] = make([]string, 0)
  453. payload["https_requests"] = make([]string, 0)
  454. persistentStatPayloadNames := make(map[string]string)
  455. persistentStatPayloadNames[datastorePersistentStatTypeRemoteServerList] = "remote_server_list_stats"
  456. persistentStatPayloadNames[datastorePersistentStatTypeFailedTunnel] = "failed_tunnel_stats"
  457. for statType, stats := range persistentStats {
  458. // Persistent stats records are already in JSON format
  459. jsonStats := make([]json.RawMessage, len(stats))
  460. for i, stat := range stats {
  461. jsonStats[i] = json.RawMessage(stat)
  462. }
  463. payload[persistentStatPayloadNames[statType]] = jsonStats
  464. }
  465. jsonPayload, err := json.Marshal(payload)
  466. if err != nil {
  467. // Send the transfer stats and tunnel stats later
  468. putBackStatusRequestPayload(payloadInfo)
  469. return nil, nil, errors.Trace(err)
  470. }
  471. return jsonPayload, payloadInfo, nil
  472. }
  473. func putBackStatusRequestPayload(payloadInfo *statusRequestPayloadInfo) {
  474. transferstats.PutBackStatsForServer(
  475. payloadInfo.serverId, payloadInfo.transferStats)
  476. err := PutBackUnreportedPersistentStats(payloadInfo.persistentStats)
  477. if err != nil {
  478. // These persistent stats records won't be resent until after a
  479. // datastore re-initialization.
  480. NoticeWarning(
  481. "PutBackUnreportedPersistentStats failed: %s", errors.Trace(err))
  482. }
  483. }
  484. func confirmStatusRequestPayload(payloadInfo *statusRequestPayloadInfo) {
  485. err := ClearReportedPersistentStats(payloadInfo.persistentStats)
  486. if err != nil {
  487. // These persistent stats records may be resent.
  488. NoticeWarning(
  489. "ClearReportedPersistentStats failed: %s", errors.Trace(err))
  490. }
  491. }
  492. // RecordRemoteServerListStat records a completed common or OSL remote server
  493. // list resource download.
  494. //
  495. // The RSL download event could occur when the client is unable to immediately
  496. // send a status request to a server, so these records are stored in the
  497. // persistent datastore and reported via subsequent status requests sent to
  498. // any Psiphon server.
  499. //
  500. // Note that some common event field values may change between the stat
  501. // recording and reporting, including client geolocation and host_id.
  502. //
  503. // The bytes/duration fields reflect the size and download time for the _last
  504. // chunk only_ in the case of a resumed download. The purpose of these fields
  505. // is to calculate rough data transfer rates. Both bytes and duration are
  506. // included in the log, to allow for filtering out of small transfers which
  507. // may not produce accurate rate numbers.
  508. //
  509. // Multiple "status" requests may be in flight at once (due to multi-tunnel,
  510. // asynchronous final status retry, and aggressive status requests for
  511. // pre-registered tunnels), To avoid duplicate reporting, persistent stats
  512. // records are "taken-out" by a status request and then "put back" in case the
  513. // request fails.
  514. //
  515. // Duplicate reporting may also occur when a server receives and processes a
  516. // status request but the client fails to receive the response.
  517. func RecordRemoteServerListStat(
  518. config *Config,
  519. tunneled bool,
  520. url string,
  521. etag string,
  522. bytes int64,
  523. duration time.Duration,
  524. authenticated bool) error {
  525. if !config.GetClientParameters().Get().WeightedCoinFlip(
  526. parameters.RecordRemoteServerListPersistentStatsProbability) {
  527. return nil
  528. }
  529. params := make(common.APIParameters)
  530. params["session_id"] = config.SessionID
  531. params["propagation_channel_id"] = config.PropagationChannelId
  532. params["sponsor_id"] = config.GetSponsorID()
  533. params["client_version"] = config.ClientVersion
  534. params["client_platform"] = config.ClientPlatform
  535. params["client_build_rev"] = buildinfo.GetBuildInfo().BuildRev
  536. if config.DeviceRegion != "" {
  537. params["device_region"] = config.DeviceRegion
  538. }
  539. params["client_download_timestamp"] = common.TruncateTimestampToHour(common.GetCurrentTimestamp())
  540. tunneledStr := "0"
  541. if tunneled {
  542. tunneledStr = "1"
  543. }
  544. params["tunneled"] = tunneledStr
  545. params["url"] = url
  546. params["etag"] = etag
  547. params["bytes"] = fmt.Sprintf("%d", bytes)
  548. // duration is nanoseconds; report milliseconds
  549. params["duration"] = fmt.Sprintf("%d", duration/time.Millisecond)
  550. authenticatedStr := "0"
  551. if authenticated {
  552. authenticatedStr = "1"
  553. }
  554. params["authenticated"] = authenticatedStr
  555. remoteServerListStatJson, err := json.Marshal(params)
  556. if err != nil {
  557. return errors.Trace(err)
  558. }
  559. return StorePersistentStat(
  560. config, datastorePersistentStatTypeRemoteServerList, remoteServerListStatJson)
  561. }
  562. // RecordFailedTunnelStat records metrics for a failed tunnel dial, including
  563. // dial parameters and error condition (tunnelErr).
  564. //
  565. // This uses the same reporting facility, with the same caveats, as
  566. // RecordRemoteServerListStat.
  567. func RecordFailedTunnelStat(
  568. config *Config,
  569. dialParams *DialParameters,
  570. livenessTestMetrics *livenessTestMetrics,
  571. bytesUp int64,
  572. bytesDown int64,
  573. tunnelErr error) error {
  574. if !config.GetClientParameters().Get().WeightedCoinFlip(
  575. parameters.RecordFailedTunnelPersistentStatsProbability) {
  576. return nil
  577. }
  578. lastConnected, err := getLastConnected()
  579. if err != nil {
  580. return errors.Trace(err)
  581. }
  582. params := getBaseAPIParameters(baseParametersAll, config, dialParams)
  583. delete(params, "server_secret")
  584. params["server_entry_tag"] = dialParams.ServerEntry.Tag
  585. params["last_connected"] = lastConnected
  586. params["client_failed_timestamp"] = common.TruncateTimestampToHour(common.GetCurrentTimestamp())
  587. if livenessTestMetrics != nil {
  588. params["liveness_test_upstream_bytes"] = strconv.Itoa(livenessTestMetrics.UpstreamBytes)
  589. params["liveness_test_sent_upstream_bytes"] = strconv.Itoa(livenessTestMetrics.SentUpstreamBytes)
  590. params["liveness_test_downstream_bytes"] = strconv.Itoa(livenessTestMetrics.DownstreamBytes)
  591. params["liveness_test_received_downstream_bytes"] = strconv.Itoa(livenessTestMetrics.ReceivedDownstreamBytes)
  592. }
  593. if bytesUp >= 0 {
  594. params["bytes_up"] = fmt.Sprintf("%d", bytesUp)
  595. }
  596. if bytesDown >= 0 {
  597. params["bytes_down"] = fmt.Sprintf("%d", bytesDown)
  598. }
  599. // Ensure direct server IPs are not exposed in logs. The "net" package, and
  600. // possibly other 3rd party packages, will include destination addresses in
  601. // I/O error messages.
  602. tunnelError := StripIPAddressesString(tunnelErr.Error())
  603. params["tunnel_error"] = tunnelError
  604. failedTunnelStatJson, err := json.Marshal(params)
  605. if err != nil {
  606. return errors.Trace(err)
  607. }
  608. return StorePersistentStat(
  609. config, datastorePersistentStatTypeFailedTunnel, failedTunnelStatJson)
  610. }
  611. // doGetRequest makes a tunneled HTTPS request and returns the response body.
  612. func (serverContext *ServerContext) doGetRequest(
  613. requestUrl string) (responseBody []byte, err error) {
  614. request, err := http.NewRequest("GET", requestUrl, nil)
  615. if err != nil {
  616. return nil, errors.Trace(err)
  617. }
  618. request.Header.Set("User-Agent", MakePsiphonUserAgent(serverContext.tunnel.config))
  619. response, err := serverContext.psiphonHttpsClient.Do(request)
  620. if err == nil && response.StatusCode != http.StatusOK {
  621. response.Body.Close()
  622. err = fmt.Errorf("HTTP GET request failed with response code: %d", response.StatusCode)
  623. }
  624. if err != nil {
  625. // Trim this error since it may include long URLs
  626. return nil, errors.Trace(TrimError(err))
  627. }
  628. defer response.Body.Close()
  629. body, err := ioutil.ReadAll(response.Body)
  630. if err != nil {
  631. return nil, errors.Trace(err)
  632. }
  633. return body, nil
  634. }
  635. // doPostRequest makes a tunneled HTTPS POST request.
  636. func (serverContext *ServerContext) doPostRequest(
  637. requestUrl string, bodyType string, body io.Reader) (responseBody []byte, err error) {
  638. request, err := http.NewRequest("POST", requestUrl, body)
  639. if err != nil {
  640. return nil, errors.Trace(err)
  641. }
  642. request.Header.Set("User-Agent", MakePsiphonUserAgent(serverContext.tunnel.config))
  643. request.Header.Set("Content-Type", bodyType)
  644. response, err := serverContext.psiphonHttpsClient.Do(request)
  645. if err == nil && response.StatusCode != http.StatusOK {
  646. response.Body.Close()
  647. err = fmt.Errorf("HTTP POST request failed with response code: %d", response.StatusCode)
  648. }
  649. if err != nil {
  650. // Trim this error since it may include long URLs
  651. return nil, errors.Trace(TrimError(err))
  652. }
  653. defer response.Body.Close()
  654. responseBody, err = ioutil.ReadAll(response.Body)
  655. if err != nil {
  656. return nil, errors.Trace(err)
  657. }
  658. return responseBody, nil
  659. }
  660. // makeSSHAPIRequestPayload makes a JSON payload for an SSH API request.
  661. func (serverContext *ServerContext) makeSSHAPIRequestPayload(
  662. params common.APIParameters) ([]byte, error) {
  663. jsonPayload, err := json.Marshal(params)
  664. if err != nil {
  665. return nil, errors.Trace(err)
  666. }
  667. return jsonPayload, nil
  668. }
  669. type baseParametersFilter int
  670. const (
  671. baseParametersAll baseParametersFilter = iota
  672. baseParametersOnlyUpstreamFragmentorDialParameters
  673. baseParametersNoDialParameters
  674. )
  675. func (serverContext *ServerContext) getBaseAPIParameters(
  676. filter baseParametersFilter) common.APIParameters {
  677. params := getBaseAPIParameters(
  678. filter,
  679. serverContext.tunnel.config,
  680. serverContext.tunnel.dialParams)
  681. // Add a random amount of padding to defend against API call traffic size
  682. // fingerprints. The "pad_response" field instructs the server to pad its
  683. // response accordingly.
  684. p := serverContext.tunnel.config.GetClientParameters().Get()
  685. minUpstreamPadding := p.Int(parameters.APIRequestUpstreamPaddingMinBytes)
  686. maxUpstreamPadding := p.Int(parameters.APIRequestUpstreamPaddingMaxBytes)
  687. minDownstreamPadding := p.Int(parameters.APIRequestDownstreamPaddingMinBytes)
  688. maxDownstreamPadding := p.Int(parameters.APIRequestDownstreamPaddingMaxBytes)
  689. if maxUpstreamPadding > 0 {
  690. size := serverContext.paddingPRNG.Range(minUpstreamPadding, maxUpstreamPadding)
  691. params["padding"] = strings.Repeat(" ", size)
  692. }
  693. if maxDownstreamPadding > 0 {
  694. size := serverContext.paddingPRNG.Range(minDownstreamPadding, maxDownstreamPadding)
  695. params["pad_response"] = strconv.Itoa(size)
  696. }
  697. return params
  698. }
  699. // getBaseAPIParameters returns all the common API parameters that are
  700. // included with each Psiphon API request. These common parameters are used
  701. // for metrics.
  702. func getBaseAPIParameters(
  703. filter baseParametersFilter,
  704. config *Config,
  705. dialParams *DialParameters) common.APIParameters {
  706. params := make(common.APIParameters)
  707. params["session_id"] = config.SessionID
  708. params["client_session_id"] = config.SessionID
  709. params["server_secret"] = dialParams.ServerEntry.WebServerSecret
  710. params["propagation_channel_id"] = config.PropagationChannelId
  711. params["sponsor_id"] = config.GetSponsorID()
  712. params["client_version"] = config.ClientVersion
  713. params["client_platform"] = config.ClientPlatform
  714. params["client_build_rev"] = buildinfo.GetBuildInfo().BuildRev
  715. params["tunnel_whole_device"] = strconv.Itoa(config.TunnelWholeDevice)
  716. // Blank parameters must be omitted.
  717. if config.DeviceRegion != "" {
  718. params["device_region"] = config.DeviceRegion
  719. }
  720. if filter == baseParametersAll {
  721. params["relay_protocol"] = dialParams.TunnelProtocol
  722. params["network_type"] = dialParams.GetNetworkType()
  723. if dialParams.BPFProgramName != "" {
  724. params["client_bpf"] = dialParams.BPFProgramName
  725. }
  726. if dialParams.SelectedSSHClientVersion {
  727. params["ssh_client_version"] = dialParams.SSHClientVersion
  728. }
  729. if dialParams.UpstreamProxyType != "" {
  730. params["upstream_proxy_type"] = dialParams.UpstreamProxyType
  731. }
  732. if dialParams.UpstreamProxyCustomHeaderNames != nil {
  733. params["upstream_proxy_custom_header_names"] = dialParams.UpstreamProxyCustomHeaderNames
  734. }
  735. if dialParams.FrontingProviderID != "" {
  736. params["fronting_provider_id"] = dialParams.FrontingProviderID
  737. }
  738. if dialParams.MeekDialAddress != "" {
  739. params["meek_dial_address"] = dialParams.MeekDialAddress
  740. }
  741. meekResolvedIPAddress := dialParams.MeekResolvedIPAddress.Load().(string)
  742. if meekResolvedIPAddress != "" {
  743. params["meek_resolved_ip_address"] = meekResolvedIPAddress
  744. }
  745. if dialParams.MeekSNIServerName != "" {
  746. params["meek_sni_server_name"] = dialParams.MeekSNIServerName
  747. }
  748. if dialParams.MeekHostHeader != "" {
  749. params["meek_host_header"] = dialParams.MeekHostHeader
  750. }
  751. // MeekTransformedHostName is meaningful when meek is used, which is when
  752. // MeekDialAddress != ""
  753. if dialParams.MeekDialAddress != "" {
  754. transformedHostName := "0"
  755. if dialParams.MeekTransformedHostName {
  756. transformedHostName = "1"
  757. }
  758. params["meek_transformed_host_name"] = transformedHostName
  759. }
  760. if dialParams.SelectedUserAgent {
  761. params["user_agent"] = dialParams.UserAgent
  762. }
  763. if dialParams.SelectedTLSProfile {
  764. params["tls_profile"] = dialParams.TLSProfile
  765. params["tls_version"] = dialParams.GetTLSVersionForMetrics()
  766. }
  767. if dialParams.ServerEntry.Region != "" {
  768. params["server_entry_region"] = dialParams.ServerEntry.Region
  769. }
  770. if dialParams.ServerEntry.LocalSource != "" {
  771. params["server_entry_source"] = dialParams.ServerEntry.LocalSource
  772. }
  773. // As with last_connected, this timestamp stat, which may be a precise
  774. // handshake request server timestamp, is truncated to hour granularity to
  775. // avoid introducing a reconstructable cross-session user trace into server
  776. // logs.
  777. localServerEntryTimestamp := common.TruncateTimestampToHour(
  778. dialParams.ServerEntry.LocalTimestamp)
  779. if localServerEntryTimestamp != "" {
  780. params["server_entry_timestamp"] = localServerEntryTimestamp
  781. }
  782. params[tactics.APPLIED_TACTICS_TAG_PARAMETER_NAME] =
  783. config.GetClientParameters().Get().Tag()
  784. if dialParams.DialPortNumber != "" {
  785. params["dial_port_number"] = dialParams.DialPortNumber
  786. }
  787. if dialParams.QUICVersion != "" {
  788. params["quic_version"] = dialParams.QUICVersion
  789. }
  790. if dialParams.QUICDialSNIAddress != "" {
  791. params["quic_dial_sni_address"] = dialParams.QUICDialSNIAddress
  792. }
  793. isReplay := "0"
  794. if dialParams.IsReplay {
  795. isReplay = "1"
  796. }
  797. params["is_replay"] = isReplay
  798. if config.EgressRegion != "" {
  799. params["egress_region"] = config.EgressRegion
  800. }
  801. // dialParams.DialDuration is nanoseconds; report milliseconds
  802. params["dial_duration"] = fmt.Sprintf("%d", dialParams.DialDuration/time.Millisecond)
  803. params["candidate_number"] = strconv.Itoa(dialParams.CandidateNumber)
  804. params["established_tunnels_count"] = strconv.Itoa(dialParams.EstablishedTunnelsCount)
  805. if dialParams.NetworkLatencyMultiplier != 0.0 {
  806. params["network_latency_multiplier"] =
  807. fmt.Sprintf("%f", dialParams.NetworkLatencyMultiplier)
  808. }
  809. if dialParams.DialConnMetrics != nil {
  810. metrics := dialParams.DialConnMetrics.GetMetrics()
  811. for name, value := range metrics {
  812. params[name] = fmt.Sprintf("%v", value)
  813. }
  814. }
  815. if dialParams.ObfuscatedSSHConnMetrics != nil {
  816. metrics := dialParams.ObfuscatedSSHConnMetrics.GetMetrics()
  817. for name, value := range metrics {
  818. params[name] = fmt.Sprintf("%v", value)
  819. }
  820. }
  821. } else if filter == baseParametersOnlyUpstreamFragmentorDialParameters {
  822. if dialParams.DialConnMetrics != nil {
  823. names := fragmentor.GetUpstreamMetricsNames()
  824. metrics := dialParams.DialConnMetrics.GetMetrics()
  825. for name, value := range metrics {
  826. if common.Contains(names, name) {
  827. params[name] = fmt.Sprintf("%v", value)
  828. }
  829. }
  830. }
  831. }
  832. return params
  833. }
  834. // makeRequestUrl makes a URL for a web service API request.
  835. func makeRequestUrl(tunnel *Tunnel, port, path string, params common.APIParameters) string {
  836. var requestUrl bytes.Buffer
  837. if port == "" {
  838. port = tunnel.dialParams.ServerEntry.WebServerPort
  839. }
  840. requestUrl.WriteString("https://")
  841. requestUrl.WriteString(tunnel.dialParams.ServerEntry.IpAddress)
  842. requestUrl.WriteString(":")
  843. requestUrl.WriteString(port)
  844. requestUrl.WriteString("/")
  845. requestUrl.WriteString(path)
  846. if len(params) > 0 {
  847. queryParams := url.Values{}
  848. for name, value := range params {
  849. // Note: this logic skips the tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME
  850. // parameter, which has a different type. This parameter is not recognized
  851. // by legacy servers.
  852. switch v := value.(type) {
  853. case string:
  854. queryParams.Set(name, v)
  855. case []string:
  856. // String array param encoded as JSON
  857. jsonValue, err := json.Marshal(v)
  858. if err != nil {
  859. break
  860. }
  861. queryParams.Set(name, string(jsonValue))
  862. }
  863. }
  864. requestUrl.WriteString("?")
  865. requestUrl.WriteString(queryParams.Encode())
  866. }
  867. return requestUrl.String()
  868. }
  869. // makePsiphonHttpsClient creates a Psiphon HTTPS client that tunnels web service API
  870. // requests and which validates the web server using the Psiphon server entry web server
  871. // certificate.
  872. func makePsiphonHttpsClient(tunnel *Tunnel) (httpsClient *http.Client, err error) {
  873. certificate, err := DecodeCertificate(
  874. tunnel.dialParams.ServerEntry.WebServerCertificate)
  875. if err != nil {
  876. return nil, errors.Trace(err)
  877. }
  878. tunneledDialer := func(_ context.Context, _, addr string) (conn net.Conn, err error) {
  879. return tunnel.sshClient.Dial("tcp", addr)
  880. }
  881. // Note: as with SSH API requests, there no dial context here. SSH port forward dials
  882. // cannot be interrupted directly. Closing the tunnel will interrupt both the dial and
  883. // the request. While it's possible to add a timeout here, we leave it with no explicit
  884. // timeout which is the same as SSH API requests: if the tunnel has stalled then SSH keep
  885. // alives will cause the tunnel to close.
  886. dialer := NewCustomTLSDialer(
  887. &CustomTLSConfig{
  888. ClientParameters: tunnel.config.clientParameters,
  889. Dial: tunneledDialer,
  890. VerifyLegacyCertificate: certificate,
  891. })
  892. transport := &http.Transport{
  893. DialTLS: func(network, addr string) (net.Conn, error) {
  894. return dialer(context.Background(), network, addr)
  895. },
  896. Dial: func(network, addr string) (net.Conn, error) {
  897. return nil, errors.TraceNew("HTTP not supported")
  898. },
  899. }
  900. return &http.Client{
  901. Transport: transport,
  902. }, nil
  903. }
  904. func HandleServerRequest(
  905. tunnelOwner TunnelOwner, tunnel *Tunnel, name string, payload []byte) error {
  906. switch name {
  907. case protocol.PSIPHON_API_OSL_REQUEST_NAME:
  908. return HandleOSLRequest(tunnelOwner, tunnel, payload)
  909. case protocol.PSIPHON_API_ALERT_REQUEST_NAME:
  910. return HandleAlertRequest(tunnelOwner, tunnel, payload)
  911. }
  912. return errors.Tracef("invalid request name: %s", name)
  913. }
  914. func HandleOSLRequest(
  915. tunnelOwner TunnelOwner, tunnel *Tunnel, payload []byte) error {
  916. var oslRequest protocol.OSLRequest
  917. err := json.Unmarshal(payload, &oslRequest)
  918. if err != nil {
  919. return errors.Trace(err)
  920. }
  921. if oslRequest.ClearLocalSLOKs {
  922. DeleteSLOKs()
  923. }
  924. seededNewSLOK := false
  925. for _, slok := range oslRequest.SeedPayload.SLOKs {
  926. duplicate, err := SetSLOK(slok.ID, slok.Key)
  927. if err != nil {
  928. // TODO: return error to trigger retry?
  929. NoticeWarning("SetSLOK failed: %s", errors.Trace(err))
  930. } else if !duplicate {
  931. seededNewSLOK = true
  932. }
  933. if tunnel.config.EmitSLOKs {
  934. NoticeSLOKSeeded(base64.StdEncoding.EncodeToString(slok.ID), duplicate)
  935. }
  936. }
  937. if seededNewSLOK {
  938. tunnelOwner.SignalSeededNewSLOK()
  939. }
  940. return nil
  941. }
  942. func HandleAlertRequest(
  943. tunnelOwner TunnelOwner, tunnel *Tunnel, payload []byte) error {
  944. var alertRequest protocol.AlertRequest
  945. err := json.Unmarshal(payload, &alertRequest)
  946. if err != nil {
  947. return errors.Trace(err)
  948. }
  949. if tunnel.config.EmitServerAlerts {
  950. NoticeServerAlert(alertRequest)
  951. }
  952. return nil
  953. }