api.go 53 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579
  1. /*
  2. * Copyright (c) 2016, 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 server
  20. import (
  21. "crypto/hmac"
  22. "crypto/sha256"
  23. "crypto/subtle"
  24. "encoding/base64"
  25. "encoding/json"
  26. std_errors "errors"
  27. "net"
  28. "regexp"
  29. "strconv"
  30. "strings"
  31. "time"
  32. "unicode"
  33. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
  34. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/errors"
  35. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/fragmentor"
  36. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/protocol"
  37. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/tactics"
  38. )
  39. const (
  40. MAX_API_PARAMS_SIZE = 256 * 1024 // 256KB
  41. PADDING_MAX_BYTES = 16 * 1024
  42. CLIENT_PLATFORM_ANDROID = "Android"
  43. CLIENT_PLATFORM_WINDOWS = "Windows"
  44. CLIENT_PLATFORM_IOS = "iOS"
  45. )
  46. // sshAPIRequestHandler routes Psiphon API requests transported as
  47. // JSON objects via the SSH request mechanism.
  48. //
  49. // The API request handlers, handshakeAPIRequestHandler, etc., are
  50. // reused by webServer which offers the Psiphon API via web transport.
  51. //
  52. // The API request parameters and event log values follow the legacy
  53. // psi_web protocol and naming conventions. The API is compatible with
  54. // all tunnel-core clients but are not backwards compatible with all
  55. // legacy clients.
  56. func sshAPIRequestHandler(
  57. support *SupportServices,
  58. clientAddr string,
  59. geoIPData GeoIPData,
  60. authorizedAccessTypes []string,
  61. name string,
  62. requestPayload []byte) ([]byte, error) {
  63. // Notes:
  64. //
  65. // - For SSH requests, MAX_API_PARAMS_SIZE is implicitly enforced
  66. // by max SSH request packet size.
  67. //
  68. // - The param protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS is an
  69. // array of base64-encoded strings; the base64 representation should
  70. // not be decoded to []byte values. The default behavior of
  71. // https://golang.org/pkg/encoding/json/#Unmarshal for a target of
  72. // type map[string]interface{} will unmarshal a base64-encoded string
  73. // to a string, not a decoded []byte, as required.
  74. var params common.APIParameters
  75. err := json.Unmarshal(requestPayload, &params)
  76. if err != nil {
  77. return nil, errors.Tracef(
  78. "invalid payload for request name: %s: %s", name, err)
  79. }
  80. return dispatchAPIRequestHandler(
  81. support,
  82. protocol.PSIPHON_SSH_API_PROTOCOL,
  83. clientAddr,
  84. geoIPData,
  85. authorizedAccessTypes,
  86. name,
  87. params)
  88. }
  89. // dispatchAPIRequestHandler is the common dispatch point for both
  90. // web and SSH API requests.
  91. func dispatchAPIRequestHandler(
  92. support *SupportServices,
  93. apiProtocol string,
  94. clientAddr string,
  95. geoIPData GeoIPData,
  96. authorizedAccessTypes []string,
  97. name string,
  98. params common.APIParameters) (response []byte, reterr error) {
  99. // Before invoking the handlers, enforce some preconditions:
  100. //
  101. // - A handshake request must precede any other requests.
  102. // - When the handshake results in a traffic rules state where
  103. // the client is immediately exhausted, no requests
  104. // may succeed. This case ensures that blocked clients do
  105. // not log "connected", etc.
  106. //
  107. // Only one handshake request may be made. There is no check here
  108. // to enforce that handshakeAPIRequestHandler will be called at
  109. // most once. The SetHandshakeState call in handshakeAPIRequestHandler
  110. // enforces that only a single handshake is made; enforcing that there
  111. // ensures no race condition even if concurrent requests are
  112. // in flight.
  113. if name != protocol.PSIPHON_API_HANDSHAKE_REQUEST_NAME {
  114. // TODO: same session-ID-lookup TODO in handshakeAPIRequestHandler
  115. // applies here.
  116. sessionID, err := getStringRequestParam(params, "client_session_id")
  117. if err == nil {
  118. // Note: follows/duplicates baseParams validation
  119. if !isHexDigits(support.Config, sessionID) {
  120. err = std_errors.New("invalid param: client_session_id")
  121. }
  122. }
  123. if err != nil {
  124. return nil, errors.Trace(err)
  125. }
  126. completed, exhausted, err := support.TunnelServer.GetClientHandshaked(sessionID)
  127. if err != nil {
  128. return nil, errors.Trace(err)
  129. }
  130. if !completed {
  131. return nil, errors.TraceNew("handshake not completed")
  132. }
  133. if exhausted {
  134. return nil, errors.TraceNew("exhausted after handshake")
  135. }
  136. }
  137. switch name {
  138. case protocol.PSIPHON_API_HANDSHAKE_REQUEST_NAME:
  139. return handshakeAPIRequestHandler(
  140. support, apiProtocol, clientAddr, geoIPData, params)
  141. case protocol.PSIPHON_API_CONNECTED_REQUEST_NAME:
  142. return connectedAPIRequestHandler(
  143. support, clientAddr, geoIPData, authorizedAccessTypes, params)
  144. case protocol.PSIPHON_API_STATUS_REQUEST_NAME:
  145. return statusAPIRequestHandler(
  146. support, clientAddr, geoIPData, authorizedAccessTypes, params)
  147. case protocol.PSIPHON_API_CLIENT_VERIFICATION_REQUEST_NAME:
  148. return clientVerificationAPIRequestHandler(
  149. support, clientAddr, geoIPData, authorizedAccessTypes, params)
  150. }
  151. return nil, errors.Tracef("invalid request name: %s", name)
  152. }
  153. var handshakeRequestParams = append(
  154. append(
  155. append(
  156. []requestParamSpec{
  157. // Legacy clients may not send "session_id" in handshake
  158. {"session_id", isHexDigits, requestParamOptional},
  159. {"missing_server_entry_signature", isBase64String, requestParamOptional}},
  160. baseParams...),
  161. baseDialParams...),
  162. tacticsParams...)
  163. // handshakeAPIRequestHandler implements the "handshake" API request.
  164. // Clients make the handshake immediately after establishing a tunnel
  165. // connection; the response tells the client what homepage to open, what
  166. // stats to record, etc.
  167. func handshakeAPIRequestHandler(
  168. support *SupportServices,
  169. apiProtocol string,
  170. clientAddr string,
  171. geoIPData GeoIPData,
  172. params common.APIParameters) ([]byte, error) {
  173. // Note: ignoring legacy "known_servers" params
  174. err := validateRequestParams(support.Config, params, handshakeRequestParams)
  175. if err != nil {
  176. return nil, errors.Trace(err)
  177. }
  178. sessionID, _ := getStringRequestParam(params, "client_session_id")
  179. sponsorID, _ := getStringRequestParam(params, "sponsor_id")
  180. clientVersion, _ := getStringRequestParam(params, "client_version")
  181. clientPlatform, _ := getStringRequestParam(params, "client_platform")
  182. isMobile := isMobileClientPlatform(clientPlatform)
  183. normalizedPlatform := normalizeClientPlatform(clientPlatform)
  184. // establishedTunnelsCount is used in traffic rule selection. When omitted by
  185. // the client, a value of 0 will be used.
  186. establishedTunnelsCount, _ := getIntStringRequestParam(params, "established_tunnels_count")
  187. // splitTunnelOwnRegion indicates if the client is requesting split tunnel
  188. // mode to be applied to the client's own country. When omitted by the
  189. // client, the value will be false.
  190. //
  191. // When split_tunnel_regions is non-empty, split tunnel mode will be
  192. // applied for the specified country codes. When omitted by the client,
  193. // the value will be an empty slice.
  194. splitTunnelOwnRegion, _ := getBoolStringRequestParam(params, "split_tunnel")
  195. splitTunnelOtherRegions, _ := getStringArrayRequestParam(params, "split_tunnel_regions")
  196. ownRegion := ""
  197. if splitTunnelOwnRegion {
  198. ownRegion = geoIPData.Country
  199. }
  200. var splitTunnelLookup *splitTunnelLookup
  201. if ownRegion != "" || len(splitTunnelOtherRegions) > 0 {
  202. splitTunnelLookup, err = newSplitTunnelLookup(ownRegion, splitTunnelOtherRegions)
  203. if err != nil {
  204. return nil, errors.Trace(err)
  205. }
  206. }
  207. var authorizations []string
  208. if params[protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS] != nil {
  209. authorizations, err = getStringArrayRequestParam(params, protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS)
  210. if err != nil {
  211. return nil, errors.Trace(err)
  212. }
  213. }
  214. deviceRegion, ok := getOptionalStringRequestParam(params, "device_region")
  215. if !ok {
  216. deviceRegion = GEOIP_UNKNOWN_VALUE
  217. }
  218. // Note: no guarantee that PsinetDatabase won't reload between database calls
  219. db := support.PsinetDatabase
  220. httpsRequestRegexes, domainBytesChecksum := db.GetHttpsRequestRegexes(sponsorID)
  221. // Flag the SSH client as having completed its handshake. This
  222. // may reselect traffic rules and starts allowing port forwards.
  223. // TODO: in the case of SSH API requests, the actual sshClient could
  224. // be passed in and used here. The session ID lookup is only strictly
  225. // necessary to support web API requests.
  226. handshakeStateInfo, err := support.TunnelServer.SetClientHandshakeState(
  227. sessionID,
  228. handshakeState{
  229. completed: true,
  230. apiProtocol: apiProtocol,
  231. apiParams: copyBaseSessionAndDialParams(params),
  232. domainBytesChecksum: domainBytesChecksum,
  233. establishedTunnelsCount: establishedTunnelsCount,
  234. splitTunnelLookup: splitTunnelLookup,
  235. deviceRegion: deviceRegion,
  236. },
  237. authorizations)
  238. if err != nil {
  239. return nil, errors.Trace(err)
  240. }
  241. tacticsPayload, err := support.TacticsServer.GetTacticsPayload(
  242. common.GeoIPData(geoIPData), params)
  243. if err != nil {
  244. return nil, errors.Trace(err)
  245. }
  246. var marshaledTacticsPayload []byte
  247. if tacticsPayload != nil {
  248. marshaledTacticsPayload, err = json.Marshal(tacticsPayload)
  249. if err != nil {
  250. return nil, errors.Trace(err)
  251. }
  252. // Log a metric when new tactics are issued. Logging here indicates that
  253. // the handshake tactics mechanism is active; but logging for every
  254. // handshake creates unneccesary log data.
  255. if len(tacticsPayload.Tactics) > 0 {
  256. logFields := getRequestLogFields(
  257. tactics.TACTICS_METRIC_EVENT_NAME,
  258. geoIPData,
  259. handshakeStateInfo.authorizedAccessTypes,
  260. params,
  261. handshakeRequestParams)
  262. logFields[tactics.NEW_TACTICS_TAG_LOG_FIELD_NAME] = tacticsPayload.Tag
  263. logFields[tactics.IS_TACTICS_REQUEST_LOG_FIELD_NAME] = false
  264. log.LogRawFieldsWithTimestamp(logFields)
  265. }
  266. }
  267. // The log comes _after_ SetClientHandshakeState, in case that call rejects
  268. // the state change (for example, if a second handshake is performed)
  269. //
  270. // The handshake event is no longer shipped to log consumers, so this is
  271. // simply a diagnostic log. Since the "server_tunnel" event includes all
  272. // common API parameters and "handshake_completed" flag, this handshake
  273. // log is mostly redundant and set to debug level.
  274. log.WithTraceFields(
  275. getRequestLogFields(
  276. "",
  277. geoIPData,
  278. handshakeStateInfo.authorizedAccessTypes,
  279. params,
  280. handshakeRequestParams)).Debug("handshake")
  281. pad_response, _ := getPaddingSizeRequestParam(params, "pad_response")
  282. // Discover new servers
  283. disableDiscovery, err := support.TunnelServer.GetClientDisableDiscovery(sessionID)
  284. if err != nil {
  285. return nil, errors.Trace(err)
  286. }
  287. var encodedServerList []string
  288. if !disableDiscovery {
  289. host, _, err := net.SplitHostPort(clientAddr)
  290. if err != nil {
  291. return nil, errors.Trace(err)
  292. }
  293. clientIP := net.ParseIP(host)
  294. if clientIP == nil {
  295. return nil, errors.TraceNew("missing client IP")
  296. }
  297. encodedServerList = db.DiscoverServers(
  298. calculateDiscoveryValue(support.Config.DiscoveryValueHMACKey, clientIP))
  299. }
  300. // When the client indicates that it used an unsigned server entry for this
  301. // connection, return a signed copy of the server entry for the client to
  302. // upgrade to. See also: comment in psiphon.doHandshakeRequest.
  303. //
  304. // The missing_server_entry_signature parameter value is a server entry tag,
  305. // which is used to select the correct server entry for servers with multiple
  306. // entries. Identifying the server entries tags instead of server IPs prevents
  307. // an enumeration attack, where a malicious client can abuse this facilty to
  308. // check if an arbitrary IP address is a Psiphon server.
  309. serverEntryTag, ok := getOptionalStringRequestParam(
  310. params, "missing_server_entry_signature")
  311. if ok {
  312. ownServerEntry, ok := support.Config.GetOwnEncodedServerEntry(serverEntryTag)
  313. if ok {
  314. encodedServerList = append(encodedServerList, ownServerEntry)
  315. }
  316. }
  317. // PageViewRegexes is obsolete and not used by any tunnel-core clients. In
  318. // the JSON response, return an empty array instead of null for legacy
  319. // clients.
  320. homepages := db.GetRandomizedHomepages(
  321. sponsorID, geoIPData.Country, geoIPData.ASN, deviceRegion, isMobile)
  322. handshakeResponse := protocol.HandshakeResponse{
  323. SSHSessionID: sessionID,
  324. Homepages: homepages,
  325. UpgradeClientVersion: db.GetUpgradeClientVersion(clientVersion, normalizedPlatform),
  326. PageViewRegexes: make([]map[string]string, 0),
  327. HttpsRequestRegexes: httpsRequestRegexes,
  328. EncodedServerList: encodedServerList,
  329. ClientRegion: geoIPData.Country,
  330. ClientAddress: clientAddr,
  331. ServerTimestamp: common.GetCurrentTimestamp(),
  332. ActiveAuthorizationIDs: handshakeStateInfo.activeAuthorizationIDs,
  333. TacticsPayload: marshaledTacticsPayload,
  334. UpstreamBytesPerSecond: handshakeStateInfo.upstreamBytesPerSecond,
  335. DownstreamBytesPerSecond: handshakeStateInfo.downstreamBytesPerSecond,
  336. Padding: strings.Repeat(" ", pad_response),
  337. }
  338. responsePayload, err := json.Marshal(handshakeResponse)
  339. if err != nil {
  340. return nil, errors.Trace(err)
  341. }
  342. return responsePayload, nil
  343. }
  344. // calculateDiscoveryValue derives a value from the client IP address to be
  345. // used as input in the server discovery algorithm.
  346. // See https://github.com/Psiphon-Inc/psiphon-automation/tree/master/Automation/psi_ops_discovery.py
  347. // for full details.
  348. func calculateDiscoveryValue(discoveryValueHMACKey string, ipAddress net.IP) int {
  349. // From: psi_ops_discovery.calculate_ip_address_strategy_value:
  350. // # Mix bits from all octets of the client IP address to determine the
  351. // # bucket. An HMAC is used to prevent pre-calculation of buckets for IPs.
  352. // return ord(hmac.new(HMAC_KEY, ip_address, hashlib.sha256).digest()[0])
  353. // TODO: use 3-octet algorithm?
  354. hash := hmac.New(sha256.New, []byte(discoveryValueHMACKey))
  355. hash.Write([]byte(ipAddress.String()))
  356. return int(hash.Sum(nil)[0])
  357. }
  358. // uniqueUserParams are the connected request parameters which are logged for
  359. // unique_user events.
  360. var uniqueUserParams = append(
  361. []requestParamSpec{
  362. {"last_connected", isLastConnected, 0}},
  363. baseSessionParams...)
  364. var connectedRequestParams = append(
  365. []requestParamSpec{
  366. {"establishment_duration", isIntString, requestParamOptional | requestParamLogStringAsInt}},
  367. uniqueUserParams...)
  368. // updateOnConnectedParamNames are connected request parameters which are
  369. // copied to update data logged with server_tunnel: these fields either only
  370. // ship with or ship newer data with connected requests.
  371. var updateOnConnectedParamNames = append(
  372. []string{
  373. "last_connected",
  374. "establishment_duration",
  375. },
  376. fragmentor.GetUpstreamMetricsNames()...)
  377. // connectedAPIRequestHandler implements the "connected" API request. Clients
  378. // make the connected request once a tunnel connection has been established
  379. // and at least once per 24h for long-running tunnels. The last_connected
  380. // input value, which should be a connected_timestamp output from a previous
  381. // connected response, is used to calculate unique user stats.
  382. // connected_timestamp is truncated as a privacy measure.
  383. func connectedAPIRequestHandler(
  384. support *SupportServices,
  385. clientAddr string,
  386. geoIPData GeoIPData,
  387. authorizedAccessTypes []string,
  388. params common.APIParameters) ([]byte, error) {
  389. err := validateRequestParams(support.Config, params, connectedRequestParams)
  390. if err != nil {
  391. return nil, errors.Trace(err)
  392. }
  393. sessionID, _ := getStringRequestParam(params, "client_session_id")
  394. lastConnected, _ := getStringRequestParam(params, "last_connected")
  395. // Update, for server_tunnel logging, upstream fragmentor metrics, as the
  396. // client may have performed more upstream fragmentation since the previous
  397. // metrics reported by the handshake request. Also, additional fields that
  398. // are reported only in the connected request are added to server_tunnel
  399. // here.
  400. // TODO: same session-ID-lookup TODO in handshakeAPIRequestHandler
  401. // applies here.
  402. err = support.TunnelServer.UpdateClientAPIParameters(
  403. sessionID, copyUpdateOnConnectedParams(params))
  404. if err != nil {
  405. return nil, errors.Trace(err)
  406. }
  407. connectedTimestamp := common.TruncateTimestampToHour(common.GetCurrentTimestamp())
  408. // The finest required granularity for unique users is daily. To save space,
  409. // only record a "unique_user" log event when the client's last_connected is
  410. // in the previous day relative to the new connected_timestamp.
  411. logUniqueUser := false
  412. if lastConnected == "None" {
  413. logUniqueUser = true
  414. } else {
  415. t1, _ := time.Parse(time.RFC3339, lastConnected)
  416. year, month, day := t1.Date()
  417. d1 := time.Date(year, month, day, 0, 0, 0, 0, time.UTC)
  418. t2, _ := time.Parse(time.RFC3339, connectedTimestamp)
  419. year, month, day = t2.Date()
  420. d2 := time.Date(year, month, day, 0, 0, 0, 0, time.UTC)
  421. if t1.Before(t2) && d1 != d2 {
  422. logUniqueUser = true
  423. }
  424. }
  425. if logUniqueUser {
  426. log.LogRawFieldsWithTimestamp(
  427. getRequestLogFields(
  428. "unique_user",
  429. geoIPData,
  430. authorizedAccessTypes,
  431. params,
  432. uniqueUserParams))
  433. }
  434. pad_response, _ := getPaddingSizeRequestParam(params, "pad_response")
  435. connectedResponse := protocol.ConnectedResponse{
  436. ConnectedTimestamp: connectedTimestamp,
  437. Padding: strings.Repeat(" ", pad_response),
  438. }
  439. responsePayload, err := json.Marshal(connectedResponse)
  440. if err != nil {
  441. return nil, errors.Trace(err)
  442. }
  443. return responsePayload, nil
  444. }
  445. var statusRequestParams = baseSessionParams
  446. var remoteServerListStatParams = append(
  447. []requestParamSpec{
  448. {"client_download_timestamp", isISO8601Date, 0},
  449. {"tunneled", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  450. {"url", isAnyString, 0},
  451. {"etag", isAnyString, 0},
  452. {"bytes", isIntString, requestParamOptional | requestParamLogStringAsInt},
  453. {"duration", isIntString, requestParamOptional | requestParamLogStringAsInt},
  454. {"authenticated", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  455. {"fronting_provider_id", isAnyString, requestParamOptional},
  456. {"meek_dial_address", isDialAddress, requestParamOptional | requestParamLogOnlyForFrontedMeekOrConjure},
  457. {"meek_resolved_ip_address", isIPAddress, requestParamOptional | requestParamLogOnlyForFrontedMeekOrConjure},
  458. {"meek_sni_server_name", isDomain, requestParamOptional},
  459. {"meek_host_header", isHostHeader, requestParamOptional | requestParamNotLoggedForUnfrontedMeekNonTransformedHeader},
  460. {"meek_transformed_host_name", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  461. {"user_agent", isAnyString, requestParamOptional},
  462. {"tls_profile", isAnyString, requestParamOptional},
  463. {"tls_version", isAnyString, requestParamOptional},
  464. {"tls_fragmented", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  465. },
  466. baseSessionParams...)
  467. // Backwards compatibility case: legacy clients do not include these fields in
  468. // the remote_server_list_stats entries. Use the values from the outer status
  469. // request as an approximation (these values reflect the client at persistent
  470. // stat shipping time, which may differ from the client at persistent stat
  471. // recording time). Note that all but client_build_rev, device_region, and
  472. // device_location are required fields.
  473. var remoteServerListStatBackwardsCompatibilityParamNames = []string{
  474. "session_id",
  475. "propagation_channel_id",
  476. "sponsor_id",
  477. "client_version",
  478. "client_platform",
  479. "client_build_rev",
  480. "device_region",
  481. "device_location",
  482. }
  483. var failedTunnelStatParams = append(
  484. []requestParamSpec{
  485. {"server_entry_tag", isAnyString, requestParamOptional},
  486. {"session_id", isHexDigits, 0},
  487. {"last_connected", isLastConnected, 0},
  488. {"client_failed_timestamp", isISO8601Date, 0},
  489. {"record_probability", isFloatString, requestParamOptional | requestParamLogStringAsFloat},
  490. {"liveness_test_upstream_bytes", isIntString, requestParamOptional | requestParamLogStringAsInt},
  491. {"liveness_test_sent_upstream_bytes", isIntString, requestParamOptional | requestParamLogStringAsInt},
  492. {"liveness_test_downstream_bytes", isIntString, requestParamOptional | requestParamLogStringAsInt},
  493. {"liveness_test_received_downstream_bytes", isIntString, requestParamOptional | requestParamLogStringAsInt},
  494. {"bytes_up", isIntString, requestParamOptional | requestParamLogStringAsInt},
  495. {"bytes_down", isIntString, requestParamOptional | requestParamLogStringAsInt},
  496. {"tunnel_error", isAnyString, 0}},
  497. baseSessionAndDialParams...)
  498. // statusAPIRequestHandler implements the "status" API request.
  499. // Clients make periodic status requests which deliver client-side
  500. // recorded data transfer and tunnel duration stats.
  501. // Note from psi_web implementation: no input validation on domains;
  502. // any string is accepted (regex transform may result in arbitrary
  503. // string). Stats processor must handle this input with care.
  504. func statusAPIRequestHandler(
  505. support *SupportServices,
  506. clientAddr string,
  507. geoIPData GeoIPData,
  508. authorizedAccessTypes []string,
  509. params common.APIParameters) ([]byte, error) {
  510. err := validateRequestParams(support.Config, params, statusRequestParams)
  511. if err != nil {
  512. return nil, errors.Trace(err)
  513. }
  514. sessionID, _ := getStringRequestParam(params, "client_session_id")
  515. statusData, err := getJSONObjectRequestParam(params, "statusData")
  516. if err != nil {
  517. return nil, errors.Trace(err)
  518. }
  519. // Logs are queued until the input is fully validated. Otherwise, stats
  520. // could be double counted if the client has a bug in its request
  521. // formatting: partial stats would be logged (counted), the request would
  522. // fail, and clients would then resend all the same stats again.
  523. logQueue := make([]LogFields, 0)
  524. // Domain bytes transferred stats
  525. // Older clients may not submit this data
  526. // Clients are expected to send host_bytes/domain_bytes stats only when
  527. // configured to do so in the handshake reponse. Legacy clients may still
  528. // report "(OTHER)" host_bytes when no regexes are set. Drop those stats.
  529. acceptDomainBytes, err := support.TunnelServer.AcceptClientDomainBytes(sessionID)
  530. if err != nil {
  531. return nil, errors.Trace(err)
  532. }
  533. if acceptDomainBytes && statusData["host_bytes"] != nil {
  534. hostBytes, err := getMapStringInt64RequestParam(statusData, "host_bytes")
  535. if err != nil {
  536. return nil, errors.Trace(err)
  537. }
  538. for domain, bytes := range hostBytes {
  539. domainBytesFields := getRequestLogFields(
  540. "domain_bytes",
  541. geoIPData,
  542. authorizedAccessTypes,
  543. params,
  544. statusRequestParams)
  545. domainBytesFields["domain"] = domain
  546. domainBytesFields["bytes"] = bytes
  547. logQueue = append(logQueue, domainBytesFields)
  548. }
  549. }
  550. // Limitation: for "persistent" stats, host_id and geolocation is time-of-sending
  551. // not time-of-recording.
  552. // Remote server list download persistent stats.
  553. // Older clients may not submit this data.
  554. if statusData["remote_server_list_stats"] != nil {
  555. remoteServerListStats, err := getJSONObjectArrayRequestParam(statusData, "remote_server_list_stats")
  556. if err != nil {
  557. return nil, errors.Trace(err)
  558. }
  559. for _, remoteServerListStat := range remoteServerListStats {
  560. for _, name := range remoteServerListStatBackwardsCompatibilityParamNames {
  561. if _, ok := remoteServerListStat[name]; !ok {
  562. if field, ok := params[name]; ok {
  563. remoteServerListStat[name] = field
  564. }
  565. }
  566. }
  567. // For validation, copy expected fields from the outer
  568. // statusRequestParams.
  569. remoteServerListStat["server_secret"] = params["server_secret"]
  570. remoteServerListStat["client_session_id"] = params["client_session_id"]
  571. err := validateRequestParams(support.Config, remoteServerListStat, remoteServerListStatParams)
  572. if err != nil {
  573. // Occasionally, clients may send corrupt persistent stat data. Do not
  574. // fail the status request, as this will lead to endless retries.
  575. log.WithTraceFields(LogFields{"error": err}).Warning("remote_server_list_stats entry dropped")
  576. continue
  577. }
  578. remoteServerListFields := getRequestLogFields(
  579. "remote_server_list",
  580. geoIPData,
  581. authorizedAccessTypes,
  582. remoteServerListStat,
  583. remoteServerListStatParams)
  584. logQueue = append(logQueue, remoteServerListFields)
  585. }
  586. }
  587. // Failed tunnel persistent stats.
  588. // Older clients may not submit this data.
  589. var invalidServerEntryTags map[string]bool
  590. if statusData["failed_tunnel_stats"] != nil {
  591. // Note: no guarantee that PsinetDatabase won't reload between database calls
  592. db := support.PsinetDatabase
  593. invalidServerEntryTags = make(map[string]bool)
  594. failedTunnelStats, err := getJSONObjectArrayRequestParam(statusData, "failed_tunnel_stats")
  595. if err != nil {
  596. return nil, errors.Trace(err)
  597. }
  598. for _, failedTunnelStat := range failedTunnelStats {
  599. // failed_tunnel supplies a full set of base params, but the server secret
  600. // must use the correct value from the outer statusRequestParams.
  601. failedTunnelStat["server_secret"] = params["server_secret"]
  602. err := validateRequestParams(support.Config, failedTunnelStat, failedTunnelStatParams)
  603. if err != nil {
  604. // Occasionally, clients may send corrupt persistent stat data. Do not
  605. // fail the status request, as this will lead to endless retries.
  606. //
  607. // TODO: trigger pruning if the data corruption indicates corrupt server
  608. // entry storage?
  609. log.WithTraceFields(LogFields{"error": err}).Warning("failed_tunnel_stats entry dropped")
  610. continue
  611. }
  612. failedTunnelFields := getRequestLogFields(
  613. "failed_tunnel",
  614. geoIPData,
  615. authorizedAccessTypes,
  616. failedTunnelStat,
  617. failedTunnelStatParams)
  618. // Return a list of servers, identified by server entry tag, that are
  619. // invalid and presumed to be deleted. This information is used by clients
  620. // to prune deleted servers from their local datastores and stop attempting
  621. // connections to servers that no longer exist.
  622. //
  623. // This mechanism uses tags instead of server IPs: (a) to prevent an
  624. // enumeration attack, where a malicious client can query the entire IPv4
  625. // range and build a map of the Psiphon network; (b) to deal with recyling
  626. // cases where a server deleted and its IP is reused for a new server with
  627. // a distinct server entry.
  628. //
  629. // IsValidServerEntryTag ensures that the local copy of psinet is not stale
  630. // before returning a negative result, to mitigate accidental pruning.
  631. //
  632. // In addition, when the reported dial port number is 0, flag the server
  633. // entry as invalid to trigger client pruning. This covers a class of
  634. // invalid/semi-functional server entries, found in practice to be stored
  635. // by clients, where some protocol port number has been omitted -- due to
  636. // historical bugs in various server entry handling implementations. When
  637. // missing from a server entry loaded by a client, the port number
  638. // evaluates to 0, the zero value, which is not a valid port number even if
  639. // were not missing.
  640. serverEntryTag, ok := getOptionalStringRequestParam(failedTunnelStat, "server_entry_tag")
  641. if ok {
  642. serverEntryValid := db.IsValidServerEntryTag(serverEntryTag)
  643. if serverEntryValid {
  644. dialPortNumber, err := getIntStringRequestParam(failedTunnelStat, "dial_port_number")
  645. if err == nil && dialPortNumber == 0 {
  646. serverEntryValid = false
  647. }
  648. }
  649. if !serverEntryValid {
  650. invalidServerEntryTags[serverEntryTag] = true
  651. }
  652. // Add a field to the failed_tunnel log indicating if the server entry is
  653. // valid.
  654. failedTunnelFields["server_entry_valid"] = serverEntryValid
  655. }
  656. // Log failed_tunnel.
  657. logQueue = append(logQueue, failedTunnelFields)
  658. }
  659. }
  660. for _, logItem := range logQueue {
  661. log.LogRawFieldsWithTimestamp(logItem)
  662. }
  663. pad_response, _ := getPaddingSizeRequestParam(params, "pad_response")
  664. statusResponse := protocol.StatusResponse{
  665. Padding: strings.Repeat(" ", pad_response),
  666. }
  667. if len(invalidServerEntryTags) > 0 {
  668. statusResponse.InvalidServerEntryTags = make([]string, len(invalidServerEntryTags))
  669. i := 0
  670. for tag := range invalidServerEntryTags {
  671. statusResponse.InvalidServerEntryTags[i] = tag
  672. i++
  673. }
  674. }
  675. responsePayload, err := json.Marshal(statusResponse)
  676. if err != nil {
  677. return nil, errors.Trace(err)
  678. }
  679. return responsePayload, nil
  680. }
  681. // clientVerificationAPIRequestHandler is just a compliance stub
  682. // for older Android clients that still send verification requests
  683. func clientVerificationAPIRequestHandler(
  684. support *SupportServices,
  685. clientAddr string,
  686. geoIPData GeoIPData,
  687. authorizedAccessTypes []string,
  688. params common.APIParameters) ([]byte, error) {
  689. return make([]byte, 0), nil
  690. }
  691. var tacticsParams = []requestParamSpec{
  692. {tactics.STORED_TACTICS_TAG_PARAMETER_NAME, isAnyString, requestParamOptional},
  693. {tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME, nil, requestParamOptional | requestParamJSON},
  694. }
  695. var tacticsRequestParams = append(
  696. append([]requestParamSpec(nil), tacticsParams...),
  697. baseSessionAndDialParams...)
  698. func getTacticsAPIParameterValidator(config *Config) common.APIParameterValidator {
  699. return func(params common.APIParameters) error {
  700. return validateRequestParams(config, params, tacticsRequestParams)
  701. }
  702. }
  703. func getTacticsAPIParameterLogFieldFormatter() common.APIParameterLogFieldFormatter {
  704. return func(geoIPData common.GeoIPData, params common.APIParameters) common.LogFields {
  705. logFields := getRequestLogFields(
  706. tactics.TACTICS_METRIC_EVENT_NAME,
  707. GeoIPData(geoIPData),
  708. nil, // authorizedAccessTypes are not known yet
  709. params,
  710. tacticsRequestParams)
  711. return common.LogFields(logFields)
  712. }
  713. }
  714. // requestParamSpec defines a request parameter. Each param is expected to be
  715. // a string, unless requestParamArray is specified, in which case an array of
  716. // strings is expected.
  717. type requestParamSpec struct {
  718. name string
  719. validator func(*Config, string) bool
  720. flags uint32
  721. }
  722. const (
  723. requestParamOptional = 1
  724. requestParamNotLogged = 1 << 1
  725. requestParamArray = 1 << 2
  726. requestParamJSON = 1 << 3
  727. requestParamLogStringAsInt = 1 << 4
  728. requestParamLogStringAsFloat = 1 << 5
  729. requestParamLogStringLengthAsInt = 1 << 6
  730. requestParamLogFlagAsBool = 1 << 7
  731. requestParamLogOnlyForFrontedMeekOrConjure = 1 << 8
  732. requestParamNotLoggedForUnfrontedMeekNonTransformedHeader = 1 << 9
  733. )
  734. // baseParams are the basic request parameters that are expected for all API
  735. // requests and log events.
  736. var baseParams = []requestParamSpec{
  737. {"server_secret", isServerSecret, requestParamNotLogged},
  738. {"client_session_id", isHexDigits, requestParamNotLogged},
  739. {"propagation_channel_id", isHexDigits, 0},
  740. {"sponsor_id", isHexDigits, 0},
  741. {"client_version", isIntString, requestParamLogStringAsInt},
  742. {"client_platform", isClientPlatform, 0},
  743. {"client_features", isAnyString, requestParamOptional | requestParamArray},
  744. {"client_build_rev", isHexDigits, requestParamOptional},
  745. {"device_region", isAnyString, requestParamOptional},
  746. {"device_location", isGeoHashString, requestParamOptional},
  747. }
  748. // baseSessionParams adds to baseParams the required session_id parameter. For
  749. // all requests except handshake, all existing clients are expected to send
  750. // session_id. Legacy clients may not send "session_id" in handshake.
  751. var baseSessionParams = append(
  752. []requestParamSpec{
  753. {"session_id", isHexDigits, 0}},
  754. baseParams...)
  755. // baseDialParams are the dial parameters, per-tunnel network protocol and
  756. // obfuscation metrics which are logged with server_tunnel, failed_tunnel, and
  757. // tactics.
  758. var baseDialParams = []requestParamSpec{
  759. {"relay_protocol", isRelayProtocol, 0},
  760. {"ssh_client_version", isAnyString, requestParamOptional},
  761. {"upstream_proxy_type", isUpstreamProxyType, requestParamOptional},
  762. {"upstream_proxy_custom_header_names", isAnyString, requestParamOptional | requestParamArray},
  763. {"fronting_provider_id", isAnyString, requestParamOptional},
  764. {"meek_dial_address", isDialAddress, requestParamOptional | requestParamLogOnlyForFrontedMeekOrConjure},
  765. {"meek_resolved_ip_address", isIPAddress, requestParamOptional | requestParamLogOnlyForFrontedMeekOrConjure},
  766. {"meek_sni_server_name", isDomain, requestParamOptional},
  767. {"meek_host_header", isHostHeader, requestParamOptional | requestParamNotLoggedForUnfrontedMeekNonTransformedHeader},
  768. {"meek_transformed_host_name", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  769. {"user_agent", isAnyString, requestParamOptional},
  770. {"tls_profile", isAnyString, requestParamOptional},
  771. {"tls_version", isAnyString, requestParamOptional},
  772. {"server_entry_region", isRegionCode, requestParamOptional},
  773. {"server_entry_source", isServerEntrySource, requestParamOptional},
  774. {"server_entry_timestamp", isISO8601Date, requestParamOptional},
  775. {tactics.APPLIED_TACTICS_TAG_PARAMETER_NAME, isAnyString, requestParamOptional},
  776. {"dial_port_number", isIntString, requestParamOptional | requestParamLogStringAsInt},
  777. {"quic_version", isAnyString, requestParamOptional},
  778. {"quic_dial_sni_address", isAnyString, requestParamOptional},
  779. {"quic_disable_client_path_mtu_discovery", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  780. {"upstream_bytes_fragmented", isIntString, requestParamOptional | requestParamLogStringAsInt},
  781. {"upstream_min_bytes_written", isIntString, requestParamOptional | requestParamLogStringAsInt},
  782. {"upstream_max_bytes_written", isIntString, requestParamOptional | requestParamLogStringAsInt},
  783. {"upstream_min_delayed", isIntString, requestParamOptional | requestParamLogStringAsInt},
  784. {"upstream_max_delayed", isIntString, requestParamOptional | requestParamLogStringAsInt},
  785. {"padding", isAnyString, requestParamOptional | requestParamLogStringLengthAsInt},
  786. {"pad_response", isIntString, requestParamOptional | requestParamLogStringAsInt},
  787. {"is_replay", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  788. {"egress_region", isRegionCode, requestParamOptional},
  789. {"dial_duration", isIntString, requestParamOptional | requestParamLogStringAsInt},
  790. {"candidate_number", isIntString, requestParamOptional | requestParamLogStringAsInt},
  791. {"established_tunnels_count", isIntString, requestParamOptional | requestParamLogStringAsInt},
  792. {"upstream_ossh_padding", isIntString, requestParamOptional | requestParamLogStringAsInt},
  793. {"meek_cookie_size", isIntString, requestParamOptional | requestParamLogStringAsInt},
  794. {"meek_limit_request", isIntString, requestParamOptional | requestParamLogStringAsInt},
  795. {"meek_redial_probability", isFloatString, requestParamOptional | requestParamLogStringAsFloat},
  796. {"meek_tls_padding", isIntString, requestParamOptional | requestParamLogStringAsInt},
  797. {"network_latency_multiplier", isFloatString, requestParamOptional | requestParamLogStringAsFloat},
  798. {"client_bpf", isAnyString, requestParamOptional},
  799. {"network_type", isAnyString, requestParamOptional},
  800. {"conjure_cached", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  801. {"conjure_delay", isIntString, requestParamOptional | requestParamLogStringAsInt},
  802. {"conjure_transport", isAnyString, requestParamOptional},
  803. {"split_tunnel", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  804. {"split_tunnel_regions", isRegionCode, requestParamOptional | requestParamArray},
  805. {"dns_preresolved", isAnyString, requestParamOptional},
  806. {"dns_preferred", isAnyString, requestParamOptional},
  807. {"dns_transform", isAnyString, requestParamOptional},
  808. {"dns_attempt", isIntString, requestParamOptional | requestParamLogStringAsInt},
  809. {"http_transform", isAnyString, requestParamOptional},
  810. {"seed_transform", isAnyString, requestParamOptional},
  811. {"ossh_prefix", isAnyString, requestParamOptional},
  812. {"tls_fragmented", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  813. }
  814. // baseSessionAndDialParams adds baseDialParams to baseSessionParams.
  815. var baseSessionAndDialParams = append(
  816. append(
  817. []requestParamSpec{},
  818. baseSessionParams...),
  819. baseDialParams...)
  820. func validateRequestParams(
  821. config *Config,
  822. params common.APIParameters,
  823. expectedParams []requestParamSpec) error {
  824. for _, expectedParam := range expectedParams {
  825. value := params[expectedParam.name]
  826. if value == nil {
  827. if expectedParam.flags&requestParamOptional != 0 {
  828. continue
  829. }
  830. return errors.Tracef("missing param: %s", expectedParam.name)
  831. }
  832. var err error
  833. switch {
  834. case expectedParam.flags&requestParamArray != 0:
  835. err = validateStringArrayRequestParam(config, expectedParam, value)
  836. case expectedParam.flags&requestParamJSON != 0:
  837. // No validation: the JSON already unmarshalled; the parameter
  838. // user will validate that the JSON contains the expected
  839. // objects/data.
  840. // TODO: without validation, any valid JSON will be logged
  841. // by getRequestLogFields, even if the parameter user validates
  842. // and rejects the parameter.
  843. default:
  844. err = validateStringRequestParam(config, expectedParam, value)
  845. }
  846. if err != nil {
  847. return errors.Trace(err)
  848. }
  849. }
  850. return nil
  851. }
  852. // copyBaseSessionAndDialParams makes a copy of the params which includes only
  853. // the baseSessionAndDialParams.
  854. func copyBaseSessionAndDialParams(params common.APIParameters) common.APIParameters {
  855. // Note: not a deep copy; assumes baseSessionAndDialParams values are all
  856. // scalar types (int, string, etc.)
  857. paramsCopy := make(common.APIParameters)
  858. for _, baseParam := range baseSessionAndDialParams {
  859. value := params[baseParam.name]
  860. if value == nil {
  861. continue
  862. }
  863. paramsCopy[baseParam.name] = value
  864. }
  865. return paramsCopy
  866. }
  867. func copyUpdateOnConnectedParams(params common.APIParameters) common.APIParameters {
  868. // Note: not a deep copy
  869. paramsCopy := make(common.APIParameters)
  870. for _, name := range updateOnConnectedParamNames {
  871. value := params[name]
  872. if value == nil {
  873. continue
  874. }
  875. paramsCopy[name] = value
  876. }
  877. return paramsCopy
  878. }
  879. func validateStringRequestParam(
  880. config *Config,
  881. expectedParam requestParamSpec,
  882. value interface{}) error {
  883. strValue, ok := value.(string)
  884. if !ok {
  885. return errors.Tracef("unexpected string param type: %s", expectedParam.name)
  886. }
  887. if !expectedParam.validator(config, strValue) {
  888. return errors.Tracef("invalid param: %s: %s", expectedParam.name, strValue)
  889. }
  890. return nil
  891. }
  892. func validateStringArrayRequestParam(
  893. config *Config,
  894. expectedParam requestParamSpec,
  895. value interface{}) error {
  896. arrayValue, ok := value.([]interface{})
  897. if !ok {
  898. return errors.Tracef("unexpected array param type: %s", expectedParam.name)
  899. }
  900. for _, value := range arrayValue {
  901. err := validateStringRequestParam(config, expectedParam, value)
  902. if err != nil {
  903. return errors.Trace(err)
  904. }
  905. }
  906. return nil
  907. }
  908. // getRequestLogFields makes LogFields to log the API event following
  909. // the legacy psi_web and current ELK naming conventions.
  910. func getRequestLogFields(
  911. eventName string,
  912. geoIPData GeoIPData,
  913. authorizedAccessTypes []string,
  914. params common.APIParameters,
  915. expectedParams []requestParamSpec) LogFields {
  916. logFields := make(LogFields)
  917. if eventName != "" {
  918. logFields["event_name"] = eventName
  919. }
  920. geoIPData.SetLogFields(logFields)
  921. if len(authorizedAccessTypes) > 0 {
  922. logFields["authorized_access_types"] = authorizedAccessTypes
  923. }
  924. if params == nil {
  925. return logFields
  926. }
  927. for _, expectedParam := range expectedParams {
  928. if expectedParam.flags&requestParamNotLogged != 0 {
  929. continue
  930. }
  931. var tunnelProtocol string
  932. if value, ok := params["relay_protocol"]; ok {
  933. tunnelProtocol, _ = value.(string)
  934. }
  935. if expectedParam.flags&requestParamLogOnlyForFrontedMeekOrConjure != 0 &&
  936. !protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol) &&
  937. !protocol.TunnelProtocolUsesConjure(tunnelProtocol) {
  938. continue
  939. }
  940. if expectedParam.flags&requestParamNotLoggedForUnfrontedMeekNonTransformedHeader != 0 &&
  941. protocol.TunnelProtocolUsesMeek(tunnelProtocol) &&
  942. !protocol.TunnelProtocolUsesFrontedMeek(tunnelProtocol) {
  943. // Non-HTTP unfronted meek protocols never tranform the host header.
  944. if protocol.TunnelProtocolUsesMeekHTTPS(tunnelProtocol) {
  945. continue
  946. }
  947. var transformedHostName string
  948. if value, ok := params["meek_transformed_host_name"]; ok {
  949. transformedHostName, _ = value.(string)
  950. }
  951. if transformedHostName != "1" {
  952. continue
  953. }
  954. }
  955. value := params[expectedParam.name]
  956. if value == nil {
  957. // Special case: older clients don't send this value,
  958. // so log a default.
  959. if expectedParam.name == "tunnel_whole_device" {
  960. value = "0"
  961. } else {
  962. // Skip omitted, optional params
  963. continue
  964. }
  965. }
  966. switch v := value.(type) {
  967. case string:
  968. strValue := v
  969. // Special cases:
  970. // - Number fields are encoded as integer types.
  971. // - For ELK performance we record certain domain-or-IP
  972. // fields as one of two different values based on type;
  973. // we also omit port from these host:port fields for now.
  974. // - Boolean fields that come into the api as "1"/"0"
  975. // must be logged as actual boolean values
  976. switch expectedParam.name {
  977. case "meek_dial_address":
  978. host, _, _ := net.SplitHostPort(strValue)
  979. if isIPAddress(nil, host) {
  980. logFields["meek_dial_ip_address"] = host
  981. } else {
  982. logFields["meek_dial_domain"] = host
  983. }
  984. case "upstream_proxy_type":
  985. // Submitted value could be e.g., "SOCKS5" or "socks5"; log lowercase
  986. logFields[expectedParam.name] = strings.ToLower(strValue)
  987. case tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME:
  988. // Due to a client bug, clients may deliever an incorrect ""
  989. // value for speed_test_samples via the web API protocol. Omit
  990. // the field in this case.
  991. case "tunnel_error":
  992. // net/url.Error, returned from net/url.Parse, contains the original input
  993. // URL, which may contain PII. New clients strip this out by using
  994. // common.SafeParseURL. Legacy clients will still send the full error
  995. // message, so strip it out here. The target substring should be unique to
  996. // legacy clients.
  997. target := "upstreamproxy error: proxyURI url.Parse: parse "
  998. index := strings.Index(strValue, target)
  999. if index != -1 {
  1000. strValue = strValue[:index+len(target)] + "<redacted>"
  1001. }
  1002. logFields[expectedParam.name] = strValue
  1003. default:
  1004. if expectedParam.flags&requestParamLogStringAsInt != 0 {
  1005. intValue, _ := strconv.Atoi(strValue)
  1006. logFields[expectedParam.name] = intValue
  1007. } else if expectedParam.flags&requestParamLogStringAsFloat != 0 {
  1008. floatValue, _ := strconv.ParseFloat(strValue, 64)
  1009. logFields[expectedParam.name] = floatValue
  1010. } else if expectedParam.flags&requestParamLogStringLengthAsInt != 0 {
  1011. logFields[expectedParam.name] = len(strValue)
  1012. } else if expectedParam.flags&requestParamLogFlagAsBool != 0 {
  1013. // Submitted value could be "0" or "1"
  1014. // "0" and non "0"/"1" values should be transformed to false
  1015. // "1" should be transformed to true
  1016. if strValue == "1" {
  1017. logFields[expectedParam.name] = true
  1018. } else {
  1019. logFields[expectedParam.name] = false
  1020. }
  1021. } else {
  1022. logFields[expectedParam.name] = strValue
  1023. }
  1024. }
  1025. case []interface{}:
  1026. if expectedParam.name == tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME {
  1027. logFields[expectedParam.name] = makeSpeedTestSamplesLogField(v)
  1028. } else {
  1029. logFields[expectedParam.name] = v
  1030. }
  1031. default:
  1032. logFields[expectedParam.name] = v
  1033. }
  1034. }
  1035. return logFields
  1036. }
  1037. // makeSpeedTestSamplesLogField renames the tactics.SpeedTestSample json tag
  1038. // fields to more verbose names for metrics.
  1039. func makeSpeedTestSamplesLogField(samples []interface{}) []interface{} {
  1040. // TODO: use reflection and add additional tags, e.g.,
  1041. // `json:"s" log:"timestamp"` to remove hard-coded
  1042. // tag value dependency?
  1043. logSamples := make([]interface{}, len(samples))
  1044. for i, sample := range samples {
  1045. logSample := make(map[string]interface{})
  1046. if m, ok := sample.(map[string]interface{}); ok {
  1047. for k, v := range m {
  1048. logK := k
  1049. switch k {
  1050. case "s":
  1051. logK = "timestamp"
  1052. case "r":
  1053. logK = "server_region"
  1054. case "p":
  1055. logK = "relay_protocol"
  1056. case "t":
  1057. logK = "round_trip_time_ms"
  1058. case "u":
  1059. logK = "bytes_up"
  1060. case "d":
  1061. logK = "bytes_down"
  1062. }
  1063. logSample[logK] = v
  1064. }
  1065. }
  1066. logSamples[i] = logSample
  1067. }
  1068. return logSamples
  1069. }
  1070. func getOptionalStringRequestParam(params common.APIParameters, name string) (string, bool) {
  1071. if params[name] == nil {
  1072. return "", false
  1073. }
  1074. value, ok := params[name].(string)
  1075. if !ok {
  1076. return "", false
  1077. }
  1078. return value, true
  1079. }
  1080. func getStringRequestParam(params common.APIParameters, name string) (string, error) {
  1081. if params[name] == nil {
  1082. return "", errors.Tracef("missing param: %s", name)
  1083. }
  1084. value, ok := params[name].(string)
  1085. if !ok {
  1086. return "", errors.Tracef("invalid param: %s", name)
  1087. }
  1088. return value, nil
  1089. }
  1090. func getIntStringRequestParam(params common.APIParameters, name string) (int, error) {
  1091. if params[name] == nil {
  1092. return 0, errors.Tracef("missing param: %s", name)
  1093. }
  1094. valueStr, ok := params[name].(string)
  1095. if !ok {
  1096. return 0, errors.Tracef("invalid param: %s", name)
  1097. }
  1098. value, err := strconv.Atoi(valueStr)
  1099. if !ok {
  1100. return 0, errors.Trace(err)
  1101. }
  1102. return value, nil
  1103. }
  1104. func getBoolStringRequestParam(params common.APIParameters, name string) (bool, error) {
  1105. if params[name] == nil {
  1106. return false, errors.Tracef("missing param: %s", name)
  1107. }
  1108. valueStr, ok := params[name].(string)
  1109. if !ok {
  1110. return false, errors.Tracef("invalid param: %s", name)
  1111. }
  1112. if valueStr == "1" {
  1113. return true, nil
  1114. }
  1115. return false, nil
  1116. }
  1117. func getPaddingSizeRequestParam(params common.APIParameters, name string) (int, error) {
  1118. value, err := getIntStringRequestParam(params, name)
  1119. if err != nil {
  1120. return 0, errors.Trace(err)
  1121. }
  1122. if value < 0 {
  1123. value = 0
  1124. }
  1125. if value > PADDING_MAX_BYTES {
  1126. value = PADDING_MAX_BYTES
  1127. }
  1128. return int(value), nil
  1129. }
  1130. func getJSONObjectRequestParam(params common.APIParameters, name string) (common.APIParameters, error) {
  1131. if params[name] == nil {
  1132. return nil, errors.Tracef("missing param: %s", name)
  1133. }
  1134. // Note: generic unmarshal of JSON produces map[string]interface{}, not common.APIParameters
  1135. value, ok := params[name].(map[string]interface{})
  1136. if !ok {
  1137. return nil, errors.Tracef("invalid param: %s", name)
  1138. }
  1139. return common.APIParameters(value), nil
  1140. }
  1141. func getJSONObjectArrayRequestParam(params common.APIParameters, name string) ([]common.APIParameters, error) {
  1142. if params[name] == nil {
  1143. return nil, errors.Tracef("missing param: %s", name)
  1144. }
  1145. value, ok := params[name].([]interface{})
  1146. if !ok {
  1147. return nil, errors.Tracef("invalid param: %s", name)
  1148. }
  1149. result := make([]common.APIParameters, len(value))
  1150. for i, item := range value {
  1151. // Note: generic unmarshal of JSON produces map[string]interface{}, not common.APIParameters
  1152. resultItem, ok := item.(map[string]interface{})
  1153. if !ok {
  1154. return nil, errors.Tracef("invalid param: %s", name)
  1155. }
  1156. result[i] = common.APIParameters(resultItem)
  1157. }
  1158. return result, nil
  1159. }
  1160. func getMapStringInt64RequestParam(params common.APIParameters, name string) (map[string]int64, error) {
  1161. if params[name] == nil {
  1162. return nil, errors.Tracef("missing param: %s", name)
  1163. }
  1164. // TODO: can't use common.APIParameters type?
  1165. value, ok := params[name].(map[string]interface{})
  1166. if !ok {
  1167. return nil, errors.Tracef("invalid param: %s", name)
  1168. }
  1169. result := make(map[string]int64)
  1170. for k, v := range value {
  1171. numValue, ok := v.(float64)
  1172. if !ok {
  1173. return nil, errors.Tracef("invalid param: %s", name)
  1174. }
  1175. result[k] = int64(numValue)
  1176. }
  1177. return result, nil
  1178. }
  1179. func getStringArrayRequestParam(params common.APIParameters, name string) ([]string, error) {
  1180. if params[name] == nil {
  1181. return nil, errors.Tracef("missing param: %s", name)
  1182. }
  1183. value, ok := params[name].([]interface{})
  1184. if !ok {
  1185. return nil, errors.Tracef("invalid param: %s", name)
  1186. }
  1187. result := make([]string, len(value))
  1188. for i, v := range value {
  1189. strValue, ok := v.(string)
  1190. if !ok {
  1191. return nil, errors.Tracef("invalid param: %s", name)
  1192. }
  1193. result[i] = strValue
  1194. }
  1195. return result, nil
  1196. }
  1197. // Normalize reported client platform. Android clients, for example, report
  1198. // OS version, rooted status, and Google Play build status in the clientPlatform
  1199. // string along with "Android".
  1200. func normalizeClientPlatform(clientPlatform string) string {
  1201. if strings.Contains(strings.ToLower(clientPlatform), strings.ToLower(CLIENT_PLATFORM_ANDROID)) {
  1202. return CLIENT_PLATFORM_ANDROID
  1203. } else if strings.HasPrefix(clientPlatform, CLIENT_PLATFORM_IOS) {
  1204. return CLIENT_PLATFORM_IOS
  1205. }
  1206. return CLIENT_PLATFORM_WINDOWS
  1207. }
  1208. func isAnyString(config *Config, value string) bool {
  1209. return true
  1210. }
  1211. func isMobileClientPlatform(clientPlatform string) bool {
  1212. normalizedClientPlatform := normalizeClientPlatform(clientPlatform)
  1213. return normalizedClientPlatform == CLIENT_PLATFORM_ANDROID ||
  1214. normalizedClientPlatform == CLIENT_PLATFORM_IOS
  1215. }
  1216. // Input validators follow the legacy validations rules in psi_web.
  1217. func isServerSecret(config *Config, value string) bool {
  1218. return subtle.ConstantTimeCompare(
  1219. []byte(value),
  1220. []byte(config.WebServerSecret)) == 1
  1221. }
  1222. func isHexDigits(_ *Config, value string) bool {
  1223. // Allows both uppercase in addition to lowercase, for legacy support.
  1224. return -1 == strings.IndexFunc(value, func(c rune) bool {
  1225. return !unicode.Is(unicode.ASCII_Hex_Digit, c)
  1226. })
  1227. }
  1228. func isBase64String(_ *Config, value string) bool {
  1229. _, err := base64.StdEncoding.DecodeString(value)
  1230. return err == nil
  1231. }
  1232. func isDigits(_ *Config, value string) bool {
  1233. return -1 == strings.IndexFunc(value, func(c rune) bool {
  1234. return c < '0' || c > '9'
  1235. })
  1236. }
  1237. func isIntString(_ *Config, value string) bool {
  1238. _, err := strconv.Atoi(value)
  1239. return err == nil
  1240. }
  1241. func isFloatString(_ *Config, value string) bool {
  1242. _, err := strconv.ParseFloat(value, 64)
  1243. return err == nil
  1244. }
  1245. func isClientPlatform(_ *Config, value string) bool {
  1246. return -1 == strings.IndexFunc(value, func(c rune) bool {
  1247. // Note: stricter than psi_web's Python string.whitespace
  1248. return unicode.Is(unicode.White_Space, c)
  1249. })
  1250. }
  1251. func isRelayProtocol(_ *Config, value string) bool {
  1252. return common.Contains(protocol.SupportedTunnelProtocols, value)
  1253. }
  1254. func isBooleanFlag(_ *Config, value string) bool {
  1255. return value == "0" || value == "1"
  1256. }
  1257. func isUpstreamProxyType(_ *Config, value string) bool {
  1258. value = strings.ToLower(value)
  1259. return value == "http" || value == "socks5" || value == "socks4a"
  1260. }
  1261. func isRegionCode(_ *Config, value string) bool {
  1262. if len(value) != 2 {
  1263. return false
  1264. }
  1265. return -1 == strings.IndexFunc(value, func(c rune) bool {
  1266. return c < 'A' || c > 'Z'
  1267. })
  1268. }
  1269. func isDialAddress(_ *Config, value string) bool {
  1270. // "<host>:<port>", where <host> is a domain or IP address
  1271. parts := strings.Split(value, ":")
  1272. if len(parts) != 2 {
  1273. return false
  1274. }
  1275. if !isIPAddress(nil, parts[0]) && !isDomain(nil, parts[0]) {
  1276. return false
  1277. }
  1278. if !isDigits(nil, parts[1]) {
  1279. return false
  1280. }
  1281. _, err := strconv.Atoi(parts[1])
  1282. if err != nil {
  1283. return false
  1284. }
  1285. // Allow port numbers outside [0,65535] to accommodate failed_tunnel cases.
  1286. return true
  1287. }
  1288. func isIPAddress(_ *Config, value string) bool {
  1289. return net.ParseIP(value) != nil
  1290. }
  1291. var isDomainRegex = regexp.MustCompile(`[a-zA-Z\d-]{1,63}$`)
  1292. func isDomain(_ *Config, value string) bool {
  1293. // From: http://stackoverflow.com/questions/2532053/validate-a-hostname-string
  1294. //
  1295. // "ensures that each segment
  1296. // * contains at least one character and a maximum of 63 characters
  1297. // * consists only of allowed characters
  1298. // * doesn't begin or end with a hyphen"
  1299. //
  1300. if len(value) > 255 {
  1301. return false
  1302. }
  1303. value = strings.TrimSuffix(value, ".")
  1304. for _, part := range strings.Split(value, ".") {
  1305. // Note: regexp doesn't support the following Perl expression which
  1306. // would check for '-' prefix/suffix: "(?!-)[a-zA-Z\\d-]{1,63}(?<!-)$"
  1307. if strings.HasPrefix(part, "-") || strings.HasSuffix(part, "-") {
  1308. return false
  1309. }
  1310. if !isDomainRegex.Match([]byte(part)) {
  1311. return false
  1312. }
  1313. }
  1314. return true
  1315. }
  1316. func isHostHeader(_ *Config, value string) bool {
  1317. // "<host>:<port>", where <host> is a domain or IP address and ":<port>" is optional
  1318. if strings.Contains(value, ":") {
  1319. return isDialAddress(nil, value)
  1320. }
  1321. return isIPAddress(nil, value) || isDomain(nil, value)
  1322. }
  1323. func isServerEntrySource(_ *Config, value string) bool {
  1324. return common.Contains(protocol.SupportedServerEntrySources, value)
  1325. }
  1326. var isISO8601DateRegex = regexp.MustCompile(
  1327. `(?P<year>[0-9]{4})-(?P<month>[0-9]{1,2})-(?P<day>[0-9]{1,2})T(?P<hour>[0-9]{2}):(?P<minute>[0-9]{2}):(?P<second>[0-9]{2})(\.(?P<fraction>[0-9]+))?(?P<timezone>Z|(([-+])([0-9]{2}):([0-9]{2})))`)
  1328. func isISO8601Date(_ *Config, value string) bool {
  1329. return isISO8601DateRegex.Match([]byte(value))
  1330. }
  1331. func isLastConnected(_ *Config, value string) bool {
  1332. return value == "None" || isISO8601Date(nil, value)
  1333. }
  1334. const geohashAlphabet = "0123456789bcdefghjkmnpqrstuvwxyz"
  1335. func isGeoHashString(_ *Config, value string) bool {
  1336. // Verify that the string is between 1 and 12 characters long
  1337. // and contains only characters from the geohash alphabet.
  1338. if len(value) < 1 || len(value) > 12 {
  1339. return false
  1340. }
  1341. for _, c := range value {
  1342. if strings.Index(geohashAlphabet, string(c)) == -1 {
  1343. return false
  1344. }
  1345. }
  1346. return true
  1347. }