api.go 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120
  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/subtle"
  22. "encoding/json"
  23. "errors"
  24. "fmt"
  25. "net"
  26. "regexp"
  27. "runtime/debug"
  28. "strconv"
  29. "strings"
  30. "unicode"
  31. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
  32. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/protocol"
  33. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/tactics"
  34. )
  35. const (
  36. MAX_API_PARAMS_SIZE = 256 * 1024 // 256KB
  37. PADDING_MAX_BYTES = 16 * 1024
  38. CLIENT_PLATFORM_ANDROID = "Android"
  39. CLIENT_PLATFORM_WINDOWS = "Windows"
  40. CLIENT_PLATFORM_IOS = "iOS"
  41. )
  42. // sshAPIRequestHandler routes Psiphon API requests transported as
  43. // JSON objects via the SSH request mechanism.
  44. //
  45. // The API request handlers, handshakeAPIRequestHandler, etc., are
  46. // reused by webServer which offers the Psiphon API via web transport.
  47. //
  48. // The API request parameters and event log values follow the legacy
  49. // psi_web protocol and naming conventions. The API is compatible with
  50. // all tunnel-core clients but are not backwards compatible with all
  51. // legacy clients.
  52. //
  53. func sshAPIRequestHandler(
  54. support *SupportServices,
  55. geoIPData GeoIPData,
  56. authorizedAccessTypes []string,
  57. name string,
  58. requestPayload []byte) ([]byte, error) {
  59. // Notes:
  60. //
  61. // - For SSH requests, MAX_API_PARAMS_SIZE is implicitly enforced
  62. // by max SSH request packet size.
  63. //
  64. // - The param protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS is an
  65. // array of base64-encoded strings; the base64 representation should
  66. // not be decoded to []byte values. The default behavior of
  67. // https://golang.org/pkg/encoding/json/#Unmarshal for a target of
  68. // type map[string]interface{} will unmarshal a base64-encoded string
  69. // to a string, not a decoded []byte, as required.
  70. var params common.APIParameters
  71. err := json.Unmarshal(requestPayload, &params)
  72. if err != nil {
  73. return nil, common.ContextError(
  74. fmt.Errorf("invalid payload for request name: %s: %s", name, err))
  75. }
  76. return dispatchAPIRequestHandler(
  77. support,
  78. protocol.PSIPHON_SSH_API_PROTOCOL,
  79. geoIPData,
  80. authorizedAccessTypes,
  81. name,
  82. params)
  83. }
  84. // dispatchAPIRequestHandler is the common dispatch point for both
  85. // web and SSH API requests.
  86. func dispatchAPIRequestHandler(
  87. support *SupportServices,
  88. apiProtocol string,
  89. geoIPData GeoIPData,
  90. authorizedAccessTypes []string,
  91. name string,
  92. params common.APIParameters) (response []byte, reterr error) {
  93. // Recover from and log any unexpected panics caused by user input
  94. // handling bugs. User inputs should be properly validated; this
  95. // mechanism is only a last resort to prevent the process from
  96. // terminating in the case of a bug.
  97. defer func() {
  98. if e := recover(); e != nil {
  99. if intentionalPanic, ok := e.(IntentionalPanicError); ok {
  100. panic(intentionalPanic)
  101. } else {
  102. log.LogPanicRecover(e, debug.Stack())
  103. reterr = common.ContextError(errors.New("request handler panic"))
  104. }
  105. }
  106. }()
  107. // Before invoking the handlers, enforce some preconditions:
  108. //
  109. // - A handshake request must precede any other requests.
  110. // - When the handshake results in a traffic rules state where
  111. // the client is immediately exhausted, no requests
  112. // may succeed. This case ensures that blocked clients do
  113. // not log "connected", etc.
  114. //
  115. // Only one handshake request may be made. There is no check here
  116. // to enforce that handshakeAPIRequestHandler will be called at
  117. // most once. The SetHandshakeState call in handshakeAPIRequestHandler
  118. // enforces that only a single handshake is made; enforcing that there
  119. // ensures no race condition even if concurrent requests are
  120. // in flight.
  121. if name != protocol.PSIPHON_API_HANDSHAKE_REQUEST_NAME {
  122. // TODO: same session-ID-lookup TODO in handshakeAPIRequestHandler
  123. // applies here.
  124. sessionID, err := getStringRequestParam(params, "client_session_id")
  125. if err == nil {
  126. // Note: follows/duplicates baseRequestParams validation
  127. if !isHexDigits(support.Config, sessionID) {
  128. err = errors.New("invalid param: client_session_id")
  129. }
  130. }
  131. if err != nil {
  132. return nil, common.ContextError(err)
  133. }
  134. completed, exhausted, err := support.TunnelServer.GetClientHandshaked(sessionID)
  135. if err != nil {
  136. return nil, common.ContextError(err)
  137. }
  138. if !completed {
  139. return nil, common.ContextError(errors.New("handshake not completed"))
  140. }
  141. if exhausted {
  142. return nil, common.ContextError(errors.New("exhausted after handshake"))
  143. }
  144. }
  145. switch name {
  146. case protocol.PSIPHON_API_HANDSHAKE_REQUEST_NAME:
  147. return handshakeAPIRequestHandler(support, apiProtocol, geoIPData, params)
  148. case protocol.PSIPHON_API_CONNECTED_REQUEST_NAME:
  149. return connectedAPIRequestHandler(support, geoIPData, authorizedAccessTypes, params)
  150. case protocol.PSIPHON_API_STATUS_REQUEST_NAME:
  151. return statusAPIRequestHandler(support, geoIPData, authorizedAccessTypes, params)
  152. case protocol.PSIPHON_API_CLIENT_VERIFICATION_REQUEST_NAME:
  153. return clientVerificationAPIRequestHandler(support, geoIPData, authorizedAccessTypes, params)
  154. }
  155. return nil, common.ContextError(fmt.Errorf("invalid request name: %s", name))
  156. }
  157. var handshakeRequestParams = append(
  158. append(
  159. // Note: legacy clients may not send "session_id" in handshake
  160. []requestParamSpec{{"session_id", isHexDigits, requestParamOptional}},
  161. tacticsParams...),
  162. baseRequestParams...)
  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. geoIPData GeoIPData,
  171. params common.APIParameters) ([]byte, error) {
  172. // Note: ignoring "known_servers" params
  173. err := validateRequestParams(support.Config, params, baseRequestParams)
  174. if err != nil {
  175. return nil, common.ContextError(err)
  176. }
  177. sessionID, _ := getStringRequestParam(params, "client_session_id")
  178. sponsorID, _ := getStringRequestParam(params, "sponsor_id")
  179. clientVersion, _ := getStringRequestParam(params, "client_version")
  180. clientPlatform, _ := getStringRequestParam(params, "client_platform")
  181. isMobile := isMobileClientPlatform(clientPlatform)
  182. normalizedPlatform := normalizeClientPlatform(clientPlatform)
  183. var authorizations []string
  184. if params[protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS] != nil {
  185. authorizations, err = getStringArrayRequestParam(params, protocol.PSIPHON_API_HANDSHAKE_AUTHORIZATIONS)
  186. if err != nil {
  187. return nil, common.ContextError(err)
  188. }
  189. }
  190. // Note: no guarantee that PsinetDatabase won't reload between database calls
  191. db := support.PsinetDatabase
  192. httpsRequestRegexes := db.GetHttpsRequestRegexes(sponsorID)
  193. // Flag the SSH client as having completed its handshake. This
  194. // may reselect traffic rules and starts allowing port forwards.
  195. // TODO: in the case of SSH API requests, the actual sshClient could
  196. // be passed in and used here. The session ID lookup is only strictly
  197. // necessary to support web API requests.
  198. activeAuthorizationIDs, authorizedAccessTypes, err := support.TunnelServer.SetClientHandshakeState(
  199. sessionID,
  200. handshakeState{
  201. completed: true,
  202. apiProtocol: apiProtocol,
  203. apiParams: copyBaseRequestParams(params),
  204. expectDomainBytes: len(httpsRequestRegexes) > 0,
  205. },
  206. authorizations)
  207. if err != nil {
  208. return nil, common.ContextError(err)
  209. }
  210. tacticsPayload, err := support.TacticsServer.GetTacticsPayload(
  211. common.GeoIPData(geoIPData), params)
  212. if err != nil {
  213. return nil, common.ContextError(err)
  214. }
  215. var marshaledTacticsPayload []byte
  216. if tacticsPayload != nil {
  217. marshaledTacticsPayload, err = json.Marshal(tacticsPayload)
  218. if err != nil {
  219. return nil, common.ContextError(err)
  220. }
  221. // Log a metric when new tactics are issued. Logging here indicates that
  222. // the handshake tactics mechanism is active; but logging for every
  223. // handshake creates unneccesary log data.
  224. if len(tacticsPayload.Tactics) > 0 {
  225. logFields := getRequestLogFields(
  226. tactics.TACTICS_METRIC_EVENT_NAME,
  227. geoIPData,
  228. authorizedAccessTypes,
  229. params,
  230. handshakeRequestParams)
  231. logFields[tactics.NEW_TACTICS_TAG_LOG_FIELD_NAME] = tacticsPayload.Tag
  232. logFields[tactics.IS_TACTICS_REQUEST_LOG_FIELD_NAME] = false
  233. log.LogRawFieldsWithTimestamp(logFields)
  234. }
  235. }
  236. // The log comes _after_ SetClientHandshakeState, in case that call rejects
  237. // the state change (for example, if a second handshake is performed)
  238. //
  239. // The handshake event is no longer shipped to log consumers, so this is
  240. // simply a diagnostic log. Since the "server_tunnel" event includes all
  241. // common API parameters and "handshake_completed" flag, this handshake
  242. // log is mostly redundant and set to debug level.
  243. log.WithContextFields(
  244. getRequestLogFields(
  245. "",
  246. geoIPData,
  247. authorizedAccessTypes,
  248. params,
  249. baseRequestParams)).Debug("handshake")
  250. pad_response, _ := getPaddingSizeRequestParam(params, "pad_response")
  251. handshakeResponse := protocol.HandshakeResponse{
  252. SSHSessionID: sessionID,
  253. Homepages: db.GetRandomizedHomepages(sponsorID, geoIPData.Country, isMobile),
  254. UpgradeClientVersion: db.GetUpgradeClientVersion(clientVersion, normalizedPlatform),
  255. PageViewRegexes: make([]map[string]string, 0),
  256. HttpsRequestRegexes: httpsRequestRegexes,
  257. EncodedServerList: db.DiscoverServers(geoIPData.DiscoveryValue),
  258. ClientRegion: geoIPData.Country,
  259. ServerTimestamp: common.GetCurrentTimestamp(),
  260. ActiveAuthorizationIDs: activeAuthorizationIDs,
  261. TacticsPayload: marshaledTacticsPayload,
  262. Padding: strings.Repeat(" ", pad_response),
  263. }
  264. responsePayload, err := json.Marshal(handshakeResponse)
  265. if err != nil {
  266. return nil, common.ContextError(err)
  267. }
  268. return responsePayload, nil
  269. }
  270. var connectedRequestParams = append(
  271. []requestParamSpec{
  272. {"session_id", isHexDigits, 0},
  273. {"last_connected", isLastConnected, 0},
  274. {"establishment_duration", isIntString, requestParamOptional | requestParamLogStringAsInt}},
  275. baseRequestParams...)
  276. // connectedAPIRequestHandler implements the "connected" API request.
  277. // Clients make the connected request once a tunnel connection has been
  278. // established and at least once per day. The last_connected input value,
  279. // which should be a connected_timestamp output from a previous connected
  280. // response, is used to calculate unique user stats.
  281. // connected_timestamp is truncated as a privacy measure.
  282. func connectedAPIRequestHandler(
  283. support *SupportServices,
  284. geoIPData GeoIPData,
  285. authorizedAccessTypes []string,
  286. params common.APIParameters) ([]byte, error) {
  287. err := validateRequestParams(support.Config, params, connectedRequestParams)
  288. if err != nil {
  289. return nil, common.ContextError(err)
  290. }
  291. // Update upstream fragmentor metrics, as the client may have performed
  292. // more upstream fragmentation since the previous metrics reported by the
  293. // handshake request.
  294. // TODO: same session-ID-lookup TODO in handshakeAPIRequestHandler
  295. // applies here.
  296. sessionID, _ := getStringRequestParam(params, "client_session_id")
  297. err = support.TunnelServer.UpdateClientAPIParameters(
  298. sessionID, copyUpstreamFragmentorParams(params))
  299. if err != nil {
  300. return nil, common.ContextError(err)
  301. }
  302. log.LogRawFieldsWithTimestamp(
  303. getRequestLogFields(
  304. "connected",
  305. geoIPData,
  306. authorizedAccessTypes,
  307. params,
  308. connectedRequestParams))
  309. pad_response, _ := getPaddingSizeRequestParam(params, "pad_response")
  310. connectedResponse := protocol.ConnectedResponse{
  311. ConnectedTimestamp: common.TruncateTimestampToHour(common.GetCurrentTimestamp()),
  312. Padding: strings.Repeat(" ", pad_response),
  313. }
  314. responsePayload, err := json.Marshal(connectedResponse)
  315. if err != nil {
  316. return nil, common.ContextError(err)
  317. }
  318. return responsePayload, nil
  319. }
  320. var statusRequestParams = append(
  321. []requestParamSpec{
  322. {"session_id", isHexDigits, 0},
  323. {"connected", isBooleanFlag, requestParamLogFlagAsBool}},
  324. baseRequestParams...)
  325. // statusAPIRequestHandler implements the "status" API request.
  326. // Clients make periodic status requests which deliver client-side
  327. // recorded data transfer and tunnel duration stats.
  328. // Note from psi_web implementation: no input validation on domains;
  329. // any string is accepted (regex transform may result in arbitrary
  330. // string). Stats processor must handle this input with care.
  331. func statusAPIRequestHandler(
  332. support *SupportServices,
  333. geoIPData GeoIPData,
  334. authorizedAccessTypes []string,
  335. params common.APIParameters) ([]byte, error) {
  336. err := validateRequestParams(support.Config, params, statusRequestParams)
  337. if err != nil {
  338. return nil, common.ContextError(err)
  339. }
  340. sessionID, _ := getStringRequestParam(params, "client_session_id")
  341. statusData, err := getJSONObjectRequestParam(params, "statusData")
  342. if err != nil {
  343. return nil, common.ContextError(err)
  344. }
  345. // Logs are queued until the input is fully validated. Otherwise, stats
  346. // could be double counted if the client has a bug in its request
  347. // formatting: partial stats would be logged (counted), the request would
  348. // fail, and clients would then resend all the same stats again.
  349. logQueue := make([]LogFields, 0)
  350. // Domain bytes transferred stats
  351. // Older clients may not submit this data
  352. // Clients are expected to send host_bytes/domain_bytes stats only when
  353. // configured to do so in the handshake reponse. Legacy clients may still
  354. // report "(OTHER)" host_bytes when no regexes are set. Drop those stats.
  355. domainBytesExpected, err := support.TunnelServer.ExpectClientDomainBytes(sessionID)
  356. if err != nil {
  357. return nil, common.ContextError(err)
  358. }
  359. if domainBytesExpected && statusData["host_bytes"] != nil {
  360. hostBytes, err := getMapStringInt64RequestParam(statusData, "host_bytes")
  361. if err != nil {
  362. return nil, common.ContextError(err)
  363. }
  364. for domain, bytes := range hostBytes {
  365. domainBytesFields := getRequestLogFields(
  366. "domain_bytes",
  367. geoIPData,
  368. authorizedAccessTypes,
  369. params,
  370. statusRequestParams)
  371. domainBytesFields["domain"] = domain
  372. domainBytesFields["bytes"] = bytes
  373. logQueue = append(logQueue, domainBytesFields)
  374. }
  375. }
  376. // Remote server list download stats
  377. // Older clients may not submit this data
  378. if statusData["remote_server_list_stats"] != nil {
  379. remoteServerListStats, err := getJSONObjectArrayRequestParam(statusData, "remote_server_list_stats")
  380. if err != nil {
  381. return nil, common.ContextError(err)
  382. }
  383. for _, remoteServerListStat := range remoteServerListStats {
  384. remoteServerListFields := getRequestLogFields(
  385. "remote_server_list",
  386. geoIPData,
  387. authorizedAccessTypes,
  388. params,
  389. statusRequestParams)
  390. clientDownloadTimestamp, err := getStringRequestParam(remoteServerListStat, "client_download_timestamp")
  391. if err != nil {
  392. return nil, common.ContextError(err)
  393. }
  394. remoteServerListFields["client_download_timestamp"] = clientDownloadTimestamp
  395. url, err := getStringRequestParam(remoteServerListStat, "url")
  396. if err != nil {
  397. return nil, common.ContextError(err)
  398. }
  399. remoteServerListFields["url"] = url
  400. etag, err := getStringRequestParam(remoteServerListStat, "etag")
  401. if err != nil {
  402. return nil, common.ContextError(err)
  403. }
  404. remoteServerListFields["etag"] = etag
  405. logQueue = append(logQueue, remoteServerListFields)
  406. }
  407. }
  408. for _, logItem := range logQueue {
  409. log.LogRawFieldsWithTimestamp(logItem)
  410. }
  411. pad_response, _ := getPaddingSizeRequestParam(params, "pad_response")
  412. return make([]byte, pad_response), nil
  413. }
  414. // clientVerificationAPIRequestHandler is just a compliance stub
  415. // for older Android clients that still send verification requests
  416. func clientVerificationAPIRequestHandler(
  417. support *SupportServices,
  418. geoIPData GeoIPData,
  419. authorizedAccessTypes []string,
  420. params common.APIParameters) ([]byte, error) {
  421. return make([]byte, 0), nil
  422. }
  423. var tacticsParams = []requestParamSpec{
  424. {tactics.STORED_TACTICS_TAG_PARAMETER_NAME, isAnyString, requestParamOptional},
  425. {tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME, nil, requestParamOptional | requestParamJSON},
  426. }
  427. var tacticsRequestParams = append(
  428. append(
  429. []requestParamSpec{{"session_id", isHexDigits, 0}},
  430. tacticsParams...),
  431. baseRequestParams...)
  432. func getTacticsAPIParameterValidator(config *Config) common.APIParameterValidator {
  433. return func(params common.APIParameters) error {
  434. return validateRequestParams(config, params, tacticsRequestParams)
  435. }
  436. }
  437. func getTacticsAPIParameterLogFieldFormatter() common.APIParameterLogFieldFormatter {
  438. return func(geoIPData common.GeoIPData, params common.APIParameters) common.LogFields {
  439. logFields := getRequestLogFields(
  440. tactics.TACTICS_METRIC_EVENT_NAME,
  441. GeoIPData(geoIPData),
  442. nil, // authorizedAccessTypes are not known yet
  443. params,
  444. tacticsRequestParams)
  445. return common.LogFields(logFields)
  446. }
  447. }
  448. type requestParamSpec struct {
  449. name string
  450. validator func(*Config, string) bool
  451. flags uint32
  452. }
  453. const (
  454. requestParamOptional = 1
  455. requestParamNotLogged = 2
  456. requestParamArray = 4
  457. requestParamJSON = 8
  458. requestParamLogStringAsInt = 16
  459. requestParamLogStringLengthAsInt = 32
  460. requestParamLogFlagAsBool = 64
  461. )
  462. var upstreamFragmentorParams = []requestParamSpec{
  463. {"upstream_bytes_fragmented", isIntString, requestParamOptional | requestParamLogStringAsInt},
  464. {"upstream_min_bytes_written", isIntString, requestParamOptional | requestParamLogStringAsInt},
  465. {"upstream_max_bytes_written", isIntString, requestParamOptional | requestParamLogStringAsInt},
  466. {"upstream_min_delayed", isIntString, requestParamOptional | requestParamLogStringAsInt},
  467. {"upstream_max_delayed", isIntString, requestParamOptional | requestParamLogStringAsInt},
  468. }
  469. // baseRequestParams is the list of required and optional
  470. // request parameters; derived from COMMON_INPUTS and
  471. // OPTIONAL_COMMON_INPUTS in psi_web.
  472. // Each param is expected to be a string, unless requestParamArray
  473. // is specified, in which case an array of string is expected.
  474. var baseRequestParams = append(
  475. []requestParamSpec{
  476. {"server_secret", isServerSecret, requestParamNotLogged},
  477. {"client_session_id", isHexDigits, requestParamNotLogged},
  478. {"propagation_channel_id", isHexDigits, 0},
  479. {"sponsor_id", isHexDigits, 0},
  480. {"client_version", isIntString, requestParamLogStringAsInt},
  481. {"client_platform", isClientPlatform, 0},
  482. {"client_build_rev", isHexDigits, requestParamOptional},
  483. {"relay_protocol", isRelayProtocol, 0},
  484. {"tunnel_whole_device", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  485. {"device_region", isAnyString, requestParamOptional},
  486. {"ssh_client_version", isAnyString, requestParamOptional},
  487. {"upstream_proxy_type", isUpstreamProxyType, requestParamOptional},
  488. {"upstream_proxy_custom_header_names", isAnyString, requestParamOptional | requestParamArray},
  489. {"meek_dial_address", isDialAddress, requestParamOptional},
  490. {"meek_resolved_ip_address", isIPAddress, requestParamOptional},
  491. {"meek_sni_server_name", isDomain, requestParamOptional},
  492. {"meek_host_header", isHostHeader, requestParamOptional},
  493. {"meek_transformed_host_name", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  494. {"user_agent", isAnyString, requestParamOptional},
  495. {"tls_profile", isAnyString, requestParamOptional},
  496. {"server_entry_region", isRegionCode, requestParamOptional},
  497. {"server_entry_source", isServerEntrySource, requestParamOptional},
  498. {"server_entry_timestamp", isISO8601Date, requestParamOptional},
  499. {tactics.APPLIED_TACTICS_TAG_PARAMETER_NAME, isAnyString, requestParamOptional},
  500. {"dial_port_number", isIntString, requestParamOptional | requestParamLogStringAsInt},
  501. {"quic_version", isAnyString, requestParamOptional},
  502. {"quic_dial_sni_address", isAnyString, requestParamOptional},
  503. {"padding", isAnyString, requestParamOptional | requestParamLogStringLengthAsInt},
  504. {"pad_response", isIntString, requestParamOptional | requestParamLogStringAsInt},
  505. {"is_replay", isBooleanFlag, requestParamOptional | requestParamLogFlagAsBool},
  506. {"egress_region", isRegionCode, requestParamOptional},
  507. },
  508. upstreamFragmentorParams...)
  509. func validateRequestParams(
  510. config *Config,
  511. params common.APIParameters,
  512. expectedParams []requestParamSpec) error {
  513. for _, expectedParam := range expectedParams {
  514. value := params[expectedParam.name]
  515. if value == nil {
  516. if expectedParam.flags&requestParamOptional != 0 {
  517. continue
  518. }
  519. return common.ContextError(
  520. fmt.Errorf("missing param: %s", expectedParam.name))
  521. }
  522. var err error
  523. switch {
  524. case expectedParam.flags&requestParamArray != 0:
  525. err = validateStringArrayRequestParam(config, expectedParam, value)
  526. case expectedParam.flags&requestParamJSON != 0:
  527. // No validation: the JSON already unmarshalled; the parameter
  528. // user will validate that the JSON contains the expected
  529. // objects/data.
  530. // TODO: without validation, any valid JSON will be logged
  531. // by getRequestLogFields, even if the parameter user validates
  532. // and rejects the parameter.
  533. default:
  534. err = validateStringRequestParam(config, expectedParam, value)
  535. }
  536. if err != nil {
  537. return common.ContextError(err)
  538. }
  539. }
  540. return nil
  541. }
  542. // copyBaseRequestParams makes a copy of the params which
  543. // includes only the baseRequestParams.
  544. func copyBaseRequestParams(params common.APIParameters) common.APIParameters {
  545. // Note: not a deep copy; assumes baseRequestParams values
  546. // are all scalar types (int, string, etc.)
  547. paramsCopy := make(common.APIParameters)
  548. for _, baseParam := range baseRequestParams {
  549. value := params[baseParam.name]
  550. if value == nil {
  551. continue
  552. }
  553. paramsCopy[baseParam.name] = value
  554. }
  555. return paramsCopy
  556. }
  557. func copyUpstreamFragmentorParams(params common.APIParameters) common.APIParameters {
  558. // Note: not a deep copy
  559. paramsCopy := make(common.APIParameters)
  560. for _, baseParam := range upstreamFragmentorParams {
  561. value := params[baseParam.name]
  562. if value == nil {
  563. continue
  564. }
  565. paramsCopy[baseParam.name] = value
  566. }
  567. return paramsCopy
  568. }
  569. func validateStringRequestParam(
  570. config *Config,
  571. expectedParam requestParamSpec,
  572. value interface{}) error {
  573. strValue, ok := value.(string)
  574. if !ok {
  575. return common.ContextError(
  576. fmt.Errorf("unexpected string param type: %s", expectedParam.name))
  577. }
  578. if !expectedParam.validator(config, strValue) {
  579. return common.ContextError(
  580. fmt.Errorf("invalid param: %s: %s", expectedParam.name, strValue))
  581. }
  582. return nil
  583. }
  584. func validateStringArrayRequestParam(
  585. config *Config,
  586. expectedParam requestParamSpec,
  587. value interface{}) error {
  588. arrayValue, ok := value.([]interface{})
  589. if !ok {
  590. return common.ContextError(
  591. fmt.Errorf("unexpected string param type: %s", expectedParam.name))
  592. }
  593. for _, value := range arrayValue {
  594. err := validateStringRequestParam(config, expectedParam, value)
  595. if err != nil {
  596. return common.ContextError(err)
  597. }
  598. }
  599. return nil
  600. }
  601. // getRequestLogFields makes LogFields to log the API event following
  602. // the legacy psi_web and current ELK naming conventions.
  603. func getRequestLogFields(
  604. eventName string,
  605. geoIPData GeoIPData,
  606. authorizedAccessTypes []string,
  607. params common.APIParameters,
  608. expectedParams []requestParamSpec) LogFields {
  609. logFields := make(LogFields)
  610. if eventName != "" {
  611. logFields["event_name"] = eventName
  612. }
  613. // In psi_web, the space replacement was done to accommodate space
  614. // delimited logging, which is no longer required; we retain the
  615. // transformation so that stats aggregation isn't impacted.
  616. logFields["client_region"] = strings.Replace(geoIPData.Country, " ", "_", -1)
  617. logFields["client_city"] = strings.Replace(geoIPData.City, " ", "_", -1)
  618. logFields["client_isp"] = strings.Replace(geoIPData.ISP, " ", "_", -1)
  619. if len(authorizedAccessTypes) > 0 {
  620. logFields["authorized_access_types"] = authorizedAccessTypes
  621. }
  622. if params == nil {
  623. return logFields
  624. }
  625. for _, expectedParam := range expectedParams {
  626. if expectedParam.flags&requestParamNotLogged != 0 {
  627. continue
  628. }
  629. value := params[expectedParam.name]
  630. if value == nil {
  631. // Special case: older clients don't send this value,
  632. // so log a default.
  633. if expectedParam.name == "tunnel_whole_device" {
  634. value = "0"
  635. } else {
  636. // Skip omitted, optional params
  637. continue
  638. }
  639. }
  640. switch v := value.(type) {
  641. case string:
  642. strValue := v
  643. // Special cases:
  644. // - Number fields are encoded as integer types.
  645. // - For ELK performance we record certain domain-or-IP
  646. // fields as one of two different values based on type;
  647. // we also omit port from these host:port fields for now.
  648. // - Boolean fields that come into the api as "1"/"0"
  649. // must be logged as actual boolean values
  650. switch expectedParam.name {
  651. case "meek_dial_address":
  652. host, _, _ := net.SplitHostPort(strValue)
  653. if isIPAddress(nil, host) {
  654. logFields["meek_dial_ip_address"] = host
  655. } else {
  656. logFields["meek_dial_domain"] = host
  657. }
  658. case "upstream_proxy_type":
  659. // Submitted value could be e.g., "SOCKS5" or "socks5"; log lowercase
  660. logFields[expectedParam.name] = strings.ToLower(strValue)
  661. case tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME:
  662. // Due to a client bug, clients may deliever an incorrect ""
  663. // value for speed_test_samples via the web API protocol. Omit
  664. // the field in this case.
  665. default:
  666. if expectedParam.flags&requestParamLogStringAsInt != 0 {
  667. intValue, _ := strconv.Atoi(strValue)
  668. logFields[expectedParam.name] = intValue
  669. } else if expectedParam.flags&requestParamLogStringLengthAsInt != 0 {
  670. logFields[expectedParam.name] = len(strValue)
  671. } else if expectedParam.flags&requestParamLogFlagAsBool != 0 {
  672. // Submitted value could be "0" or "1"
  673. // "0" and non "0"/"1" values should be transformed to false
  674. // "1" should be transformed to true
  675. if strValue == "1" {
  676. logFields[expectedParam.name] = true
  677. } else {
  678. logFields[expectedParam.name] = false
  679. }
  680. } else {
  681. logFields[expectedParam.name] = strValue
  682. }
  683. }
  684. case []interface{}:
  685. if expectedParam.name == tactics.SPEED_TEST_SAMPLES_PARAMETER_NAME {
  686. logFields[expectedParam.name] = makeSpeedTestSamplesLogField(v)
  687. } else {
  688. logFields[expectedParam.name] = v
  689. }
  690. default:
  691. logFields[expectedParam.name] = v
  692. }
  693. }
  694. return logFields
  695. }
  696. // makeSpeedTestSamplesLogField renames the tactics.SpeedTestSample json tag
  697. // fields to more verbose names for metrics.
  698. func makeSpeedTestSamplesLogField(samples []interface{}) []interface{} {
  699. // TODO: use reflection and add additional tags, e.g.,
  700. // `json:"s" log:"timestamp"` to remove hard-coded
  701. // tag value dependency?
  702. logSamples := make([]interface{}, len(samples))
  703. for i, sample := range samples {
  704. logSample := make(map[string]interface{})
  705. if m, ok := sample.(map[string]interface{}); ok {
  706. for k, v := range m {
  707. logK := k
  708. switch k {
  709. case "s":
  710. logK = "timestamp"
  711. case "r":
  712. logK = "server_region"
  713. case "p":
  714. logK = "relay_protocol"
  715. case "t":
  716. logK = "round_trip_time_ms"
  717. case "u":
  718. logK = "bytes_up"
  719. case "d":
  720. logK = "bytes_down"
  721. }
  722. logSample[logK] = v
  723. }
  724. }
  725. logSamples[i] = logSample
  726. }
  727. return logSamples
  728. }
  729. func getStringRequestParam(params common.APIParameters, name string) (string, error) {
  730. if params[name] == nil {
  731. return "", common.ContextError(fmt.Errorf("missing param: %s", name))
  732. }
  733. value, ok := params[name].(string)
  734. if !ok {
  735. return "", common.ContextError(fmt.Errorf("invalid param: %s", name))
  736. }
  737. return value, nil
  738. }
  739. func getInt64RequestParam(params common.APIParameters, name string) (int64, error) {
  740. if params[name] == nil {
  741. return 0, common.ContextError(fmt.Errorf("missing param: %s", name))
  742. }
  743. value, ok := params[name].(float64)
  744. if !ok {
  745. return 0, common.ContextError(fmt.Errorf("invalid param: %s", name))
  746. }
  747. return int64(value), nil
  748. }
  749. func getPaddingSizeRequestParam(params common.APIParameters, name string) (int, error) {
  750. value, err := getInt64RequestParam(params, name)
  751. if err != nil {
  752. return 0, common.ContextError(err)
  753. }
  754. if value < 0 {
  755. value = 0
  756. }
  757. if value > PADDING_MAX_BYTES {
  758. value = PADDING_MAX_BYTES
  759. }
  760. return int(value), nil
  761. }
  762. func getJSONObjectRequestParam(params common.APIParameters, name string) (common.APIParameters, error) {
  763. if params[name] == nil {
  764. return nil, common.ContextError(fmt.Errorf("missing param: %s", name))
  765. }
  766. // Note: generic unmarshal of JSON produces map[string]interface{}, not common.APIParameters
  767. value, ok := params[name].(map[string]interface{})
  768. if !ok {
  769. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  770. }
  771. return common.APIParameters(value), nil
  772. }
  773. func getJSONObjectArrayRequestParam(params common.APIParameters, name string) ([]common.APIParameters, error) {
  774. if params[name] == nil {
  775. return nil, common.ContextError(fmt.Errorf("missing param: %s", name))
  776. }
  777. value, ok := params[name].([]interface{})
  778. if !ok {
  779. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  780. }
  781. result := make([]common.APIParameters, len(value))
  782. for i, item := range value {
  783. // Note: generic unmarshal of JSON produces map[string]interface{}, not common.APIParameters
  784. resultItem, ok := item.(map[string]interface{})
  785. if !ok {
  786. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  787. }
  788. result[i] = common.APIParameters(resultItem)
  789. }
  790. return result, nil
  791. }
  792. func getMapStringInt64RequestParam(params common.APIParameters, name string) (map[string]int64, error) {
  793. if params[name] == nil {
  794. return nil, common.ContextError(fmt.Errorf("missing param: %s", name))
  795. }
  796. // TODO: can't use common.APIParameters type?
  797. value, ok := params[name].(map[string]interface{})
  798. if !ok {
  799. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  800. }
  801. result := make(map[string]int64)
  802. for k, v := range value {
  803. numValue, ok := v.(float64)
  804. if !ok {
  805. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  806. }
  807. result[k] = int64(numValue)
  808. }
  809. return result, nil
  810. }
  811. func getStringArrayRequestParam(params common.APIParameters, name string) ([]string, error) {
  812. if params[name] == nil {
  813. return nil, common.ContextError(fmt.Errorf("missing param: %s", name))
  814. }
  815. value, ok := params[name].([]interface{})
  816. if !ok {
  817. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  818. }
  819. result := make([]string, len(value))
  820. for i, v := range value {
  821. strValue, ok := v.(string)
  822. if !ok {
  823. return nil, common.ContextError(fmt.Errorf("invalid param: %s", name))
  824. }
  825. result[i] = strValue
  826. }
  827. return result, nil
  828. }
  829. // Normalize reported client platform. Android clients, for example, report
  830. // OS version, rooted status, and Google Play build status in the clientPlatform
  831. // string along with "Android".
  832. func normalizeClientPlatform(clientPlatform string) string {
  833. if strings.Contains(strings.ToLower(clientPlatform), strings.ToLower(CLIENT_PLATFORM_ANDROID)) {
  834. return CLIENT_PLATFORM_ANDROID
  835. } else if strings.HasPrefix(clientPlatform, CLIENT_PLATFORM_IOS) {
  836. return CLIENT_PLATFORM_IOS
  837. }
  838. return CLIENT_PLATFORM_WINDOWS
  839. }
  840. func isAnyString(config *Config, value string) bool {
  841. return true
  842. }
  843. func isMobileClientPlatform(clientPlatform string) bool {
  844. normalizedClientPlatform := normalizeClientPlatform(clientPlatform)
  845. return normalizedClientPlatform == CLIENT_PLATFORM_ANDROID ||
  846. normalizedClientPlatform == CLIENT_PLATFORM_IOS
  847. }
  848. // Input validators follow the legacy validations rules in psi_web.
  849. func isServerSecret(config *Config, value string) bool {
  850. return subtle.ConstantTimeCompare(
  851. []byte(value),
  852. []byte(config.WebServerSecret)) == 1
  853. }
  854. func isHexDigits(_ *Config, value string) bool {
  855. // Allows both uppercase in addition to lowercase, for legacy support.
  856. return -1 == strings.IndexFunc(value, func(c rune) bool {
  857. return !unicode.Is(unicode.ASCII_Hex_Digit, c)
  858. })
  859. }
  860. func isDigits(_ *Config, value string) bool {
  861. return -1 == strings.IndexFunc(value, func(c rune) bool {
  862. return c < '0' || c > '9'
  863. })
  864. }
  865. func isIntString(_ *Config, value string) bool {
  866. _, err := strconv.Atoi(value)
  867. return err == nil
  868. }
  869. func isClientPlatform(_ *Config, value string) bool {
  870. return -1 == strings.IndexFunc(value, func(c rune) bool {
  871. // Note: stricter than psi_web's Python string.whitespace
  872. return unicode.Is(unicode.White_Space, c)
  873. })
  874. }
  875. func isRelayProtocol(_ *Config, value string) bool {
  876. return common.Contains(protocol.SupportedTunnelProtocols, value)
  877. }
  878. func isBooleanFlag(_ *Config, value string) bool {
  879. return value == "0" || value == "1"
  880. }
  881. func isUpstreamProxyType(_ *Config, value string) bool {
  882. value = strings.ToLower(value)
  883. return value == "http" || value == "socks5" || value == "socks4a"
  884. }
  885. func isRegionCode(_ *Config, value string) bool {
  886. if len(value) != 2 {
  887. return false
  888. }
  889. return -1 == strings.IndexFunc(value, func(c rune) bool {
  890. return c < 'A' || c > 'Z'
  891. })
  892. }
  893. func isDialAddress(_ *Config, value string) bool {
  894. // "<host>:<port>", where <host> is a domain or IP address
  895. parts := strings.Split(value, ":")
  896. if len(parts) != 2 {
  897. return false
  898. }
  899. if !isIPAddress(nil, parts[0]) && !isDomain(nil, parts[0]) {
  900. return false
  901. }
  902. if !isDigits(nil, parts[1]) {
  903. return false
  904. }
  905. port, err := strconv.Atoi(parts[1])
  906. if err != nil {
  907. return false
  908. }
  909. return port > 0 && port < 65536
  910. }
  911. func isIPAddress(_ *Config, value string) bool {
  912. return net.ParseIP(value) != nil
  913. }
  914. var isDomainRegex = regexp.MustCompile("[a-zA-Z\\d-]{1,63}$")
  915. func isDomain(_ *Config, value string) bool {
  916. // From: http://stackoverflow.com/questions/2532053/validate-a-hostname-string
  917. //
  918. // "ensures that each segment
  919. // * contains at least one character and a maximum of 63 characters
  920. // * consists only of allowed characters
  921. // * doesn't begin or end with a hyphen"
  922. //
  923. if len(value) > 255 {
  924. return false
  925. }
  926. value = strings.TrimSuffix(value, ".")
  927. for _, part := range strings.Split(value, ".") {
  928. // Note: regexp doesn't support the following Perl expression which
  929. // would check for '-' prefix/suffix: "(?!-)[a-zA-Z\\d-]{1,63}(?<!-)$"
  930. if strings.HasPrefix(part, "-") || strings.HasSuffix(part, "-") {
  931. return false
  932. }
  933. if !isDomainRegex.Match([]byte(part)) {
  934. return false
  935. }
  936. }
  937. return true
  938. }
  939. func isHostHeader(_ *Config, value string) bool {
  940. // "<host>:<port>", where <host> is a domain or IP address and ":<port>" is optional
  941. if strings.Contains(value, ":") {
  942. return isDialAddress(nil, value)
  943. }
  944. return isIPAddress(nil, value) || isDomain(nil, value)
  945. }
  946. func isServerEntrySource(_ *Config, value string) bool {
  947. return common.Contains(protocol.SupportedServerEntrySources, value)
  948. }
  949. var isISO8601DateRegex = regexp.MustCompile(
  950. "(?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})))")
  951. func isISO8601Date(_ *Config, value string) bool {
  952. return isISO8601DateRegex.Match([]byte(value))
  953. }
  954. func isLastConnected(_ *Config, value string) bool {
  955. return value == "None" || value == "Unknown" || isISO8601Date(nil, value)
  956. }