serverEntry.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853
  1. /*
  2. * Copyright (c) 2015, Psiphon Inc.
  3. * All rights reserved.
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. */
  19. package protocol
  20. import (
  21. "bufio"
  22. "bytes"
  23. "crypto/ed25519"
  24. "crypto/hmac"
  25. "crypto/rand"
  26. "crypto/sha256"
  27. "encoding/base64"
  28. "encoding/hex"
  29. "encoding/json"
  30. "fmt"
  31. "io"
  32. "net"
  33. "strings"
  34. "time"
  35. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
  36. "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/errors"
  37. )
  38. // ServerEntry represents a Psiphon server. It contains information
  39. // about how to establish a tunnel connection to the server through
  40. // several protocols. Server entries are JSON records downloaded from
  41. // various sources.
  42. type ServerEntry struct {
  43. Tag string `json:"tag"`
  44. IpAddress string `json:"ipAddress"`
  45. WebServerPort string `json:"webServerPort"` // not an int
  46. WebServerSecret string `json:"webServerSecret"`
  47. WebServerCertificate string `json:"webServerCertificate"`
  48. SshPort int `json:"sshPort"`
  49. SshUsername string `json:"sshUsername"`
  50. SshPassword string `json:"sshPassword"`
  51. SshHostKey string `json:"sshHostKey"`
  52. SshObfuscatedPort int `json:"sshObfuscatedPort"`
  53. SshObfuscatedQUICPort int `json:"sshObfuscatedQUICPort"`
  54. SshObfuscatedTapDancePort int `json:"sshObfuscatedTapdancePort"`
  55. SshObfuscatedConjurePort int `json:"sshObfuscatedConjurePort"`
  56. SshObfuscatedKey string `json:"sshObfuscatedKey"`
  57. Capabilities []string `json:"capabilities"`
  58. Region string `json:"region"`
  59. FrontingProviderID string `json:"frontingProviderID"`
  60. MeekServerPort int `json:"meekServerPort"`
  61. MeekCookieEncryptionPublicKey string `json:"meekCookieEncryptionPublicKey"`
  62. MeekObfuscatedKey string `json:"meekObfuscatedKey"`
  63. MeekFrontingHost string `json:"meekFrontingHost"`
  64. MeekFrontingHosts []string `json:"meekFrontingHosts"`
  65. MeekFrontingDomain string `json:"meekFrontingDomain"`
  66. MeekFrontingAddresses []string `json:"meekFrontingAddresses"`
  67. MeekFrontingAddressesRegex string `json:"meekFrontingAddressesRegex"`
  68. MeekFrontingDisableSNI bool `json:"meekFrontingDisableSNI"`
  69. TacticsRequestPublicKey string `json:"tacticsRequestPublicKey"`
  70. TacticsRequestObfuscatedKey string `json:"tacticsRequestObfuscatedKey"`
  71. MarionetteFormat string `json:"marionetteFormat"`
  72. ConfigurationVersion int `json:"configurationVersion"`
  73. Signature string `json:"signature"`
  74. // These local fields are not expected to be present in downloaded server
  75. // entries. They are added by the client to record and report stats about
  76. // how and when server entries are obtained.
  77. // All local fields should be included the list of fields in RemoveUnsignedFields.
  78. LocalSource string `json:"localSource,omitempty"`
  79. LocalTimestamp string `json:"localTimestamp,omitempty"`
  80. IsLocalDerivedTag bool `json:"isLocalDerivedTag,omitempty"`
  81. }
  82. // ServerEntryFields is an alternate representation of ServerEntry which
  83. // enables future compatibility when unmarshaling and persisting new server
  84. // entries which may contain new, unrecognized fields not in the ServerEntry
  85. // type for a particular client version.
  86. //
  87. // When new JSON server entries with new fields are unmarshaled to ServerEntry
  88. // types, unrecognized fields are discarded. When unmarshaled to
  89. // ServerEntryFields, unrecognized fields are retained and may be persisted
  90. // and available when the client is upgraded and unmarshals to an updated
  91. // ServerEntry type.
  92. type ServerEntryFields map[string]interface{}
  93. // GetServerEntry converts a ServerEntryFields into a ServerEntry.
  94. func (fields ServerEntryFields) GetServerEntry() (*ServerEntry, error) {
  95. marshaledServerEntry, err := json.Marshal(fields)
  96. if err != nil {
  97. return nil, errors.Trace(err)
  98. }
  99. var serverEntry *ServerEntry
  100. err = json.Unmarshal(marshaledServerEntry, &serverEntry)
  101. if err != nil {
  102. return nil, errors.Trace(err)
  103. }
  104. return serverEntry, nil
  105. }
  106. func (fields ServerEntryFields) GetTag() string {
  107. tag, ok := fields["tag"]
  108. if !ok {
  109. return ""
  110. }
  111. tagStr, ok := tag.(string)
  112. if !ok {
  113. return ""
  114. }
  115. return tagStr
  116. }
  117. // SetTag sets a local, derived server entry tag. A tag is an identifier used
  118. // in server entry pruning and potentially other use cases. An explict tag,
  119. // set by the Psiphon Network, may be present in a server entry that is
  120. // imported; otherwise, the client will set a derived tag. The tag should be
  121. // generated using GenerateServerEntryTag. When SetTag finds a explicit tag,
  122. // the new, derived tag is ignored. The isLocalTag local field is set to
  123. // distinguish explict and derived tags and is used in signature verification
  124. // to determine if the tag field is part of the signature.
  125. func (fields ServerEntryFields) SetTag(tag string) {
  126. // Don't replace explicit tag
  127. if tag, ok := fields["tag"]; ok {
  128. tagStr, ok := tag.(string)
  129. if ok && tagStr != "" {
  130. isLocalDerivedTag, ok := fields["isLocalDerivedTag"]
  131. if !ok {
  132. return
  133. }
  134. isLocalDerivedTagBool, ok := isLocalDerivedTag.(bool)
  135. if ok && !isLocalDerivedTagBool {
  136. return
  137. }
  138. }
  139. }
  140. fields["tag"] = tag
  141. // Mark this tag as local
  142. fields["isLocalDerivedTag"] = true
  143. }
  144. func (fields ServerEntryFields) GetDiagnosticID() string {
  145. tag, ok := fields["tag"]
  146. if !ok {
  147. return ""
  148. }
  149. tagStr, ok := tag.(string)
  150. if !ok {
  151. return ""
  152. }
  153. return TagToDiagnosticID(tagStr)
  154. }
  155. func (fields ServerEntryFields) GetIPAddress() string {
  156. ipAddress, ok := fields["ipAddress"]
  157. if !ok {
  158. return ""
  159. }
  160. ipAddressStr, ok := ipAddress.(string)
  161. if !ok {
  162. return ""
  163. }
  164. return ipAddressStr
  165. }
  166. func (fields ServerEntryFields) GetWebServerPort() string {
  167. webServerPort, ok := fields["webServerPort"]
  168. if !ok {
  169. return ""
  170. }
  171. webServerPortStr, ok := webServerPort.(string)
  172. if !ok {
  173. return ""
  174. }
  175. return webServerPortStr
  176. }
  177. func (fields ServerEntryFields) GetWebServerSecret() string {
  178. webServerSecret, ok := fields["webServerSecret"]
  179. if !ok {
  180. return ""
  181. }
  182. webServerSecretStr, ok := webServerSecret.(string)
  183. if !ok {
  184. return ""
  185. }
  186. return webServerSecretStr
  187. }
  188. func (fields ServerEntryFields) GetWebServerCertificate() string {
  189. webServerCertificate, ok := fields["webServerCertificate"]
  190. if !ok {
  191. return ""
  192. }
  193. webServerCertificateStr, ok := webServerCertificate.(string)
  194. if !ok {
  195. return ""
  196. }
  197. return webServerCertificateStr
  198. }
  199. func (fields ServerEntryFields) GetConfigurationVersion() int {
  200. configurationVersion, ok := fields["configurationVersion"]
  201. if !ok {
  202. return 0
  203. }
  204. configurationVersionFloat, ok := configurationVersion.(float64)
  205. if !ok {
  206. return 0
  207. }
  208. return int(configurationVersionFloat)
  209. }
  210. func (fields ServerEntryFields) GetLocalSource() string {
  211. localSource, ok := fields["localSource"]
  212. if !ok {
  213. return ""
  214. }
  215. localSourceStr, ok := localSource.(string)
  216. if !ok {
  217. return ""
  218. }
  219. return localSourceStr
  220. }
  221. func (fields ServerEntryFields) SetLocalSource(source string) {
  222. fields["localSource"] = source
  223. }
  224. func (fields ServerEntryFields) GetLocalTimestamp() string {
  225. localTimestamp, ok := fields["localTimestamp"]
  226. if !ok {
  227. return ""
  228. }
  229. localTimestampStr, ok := localTimestamp.(string)
  230. if !ok {
  231. return ""
  232. }
  233. return localTimestampStr
  234. }
  235. func (fields ServerEntryFields) SetLocalTimestamp(timestamp string) {
  236. fields["localTimestamp"] = timestamp
  237. }
  238. func (fields ServerEntryFields) HasSignature() bool {
  239. signature, ok := fields["signature"]
  240. if !ok {
  241. return false
  242. }
  243. signatureStr, ok := signature.(string)
  244. if !ok {
  245. return false
  246. }
  247. return signatureStr != ""
  248. }
  249. const signaturePublicKeyDigestSize = 8
  250. // AddSignature signs a server entry and attaches a new field containing the
  251. // signature. Any existing "signature" field will be replaced.
  252. //
  253. // The signature incudes a public key ID that is derived from a digest of the
  254. // public key value. This ID is intended for future use when multiple signing
  255. // keys may be deployed.
  256. func (fields ServerEntryFields) AddSignature(publicKey, privateKey string) error {
  257. // Make a copy so that removing unsigned fields will have no side effects
  258. copyFields := make(ServerEntryFields)
  259. for k, v := range fields {
  260. copyFields[k] = v
  261. }
  262. copyFields.RemoveUnsignedFields()
  263. delete(copyFields, "signature")
  264. // Best practise would be to sign the JSON encoded server entry bytes and
  265. // append the signature to those bytes. However, due to backwards
  266. // compatibility requirements, we must retain the outer server entry encoding
  267. // as-is and insert the signature.
  268. //
  269. // Limitation: since the verifyier must remarshal its server entry before
  270. // verifying, the JSON produced there must be a byte-for-byte match to the
  271. // JSON signed here. The precise output of the JSON encoder that is used,
  272. // "encoding/json", with default formatting, as of Go 1.11.5, is therefore
  273. // part of the signature protocol.
  274. //
  275. // TODO: use a standard, canonical encoding, such as JCS:
  276. // https://tools.ietf.org/id/draft-rundgren-json-canonicalization-scheme-05.html
  277. marshaledFields, err := json.Marshal(copyFields)
  278. if err != nil {
  279. return errors.Trace(err)
  280. }
  281. decodedPublicKey, err := base64.StdEncoding.DecodeString(publicKey)
  282. if err != nil {
  283. return errors.Trace(err)
  284. }
  285. publicKeyDigest := sha256.Sum256(decodedPublicKey)
  286. publicKeyID := publicKeyDigest[:signaturePublicKeyDigestSize]
  287. decodedPrivateKey, err := base64.StdEncoding.DecodeString(privateKey)
  288. if err != nil {
  289. return errors.Trace(err)
  290. }
  291. signature := ed25519.Sign(decodedPrivateKey, marshaledFields)
  292. fields["signature"] = base64.StdEncoding.EncodeToString(
  293. append(publicKeyID, signature...))
  294. return nil
  295. }
  296. // VerifySignature verifies the signature set by AddSignature.
  297. //
  298. // VerifySignature must be called before using any server entry that is
  299. // imported from an untrusted source, such as client-to-client exchange.
  300. func (fields ServerEntryFields) VerifySignature(publicKey string) error {
  301. if publicKey == "" {
  302. return errors.TraceNew("missing public key")
  303. }
  304. // Make a copy so that removing unsigned fields will have no side effects
  305. copyFields := make(ServerEntryFields)
  306. for k, v := range fields {
  307. copyFields[k] = v
  308. }
  309. signatureField, ok := copyFields["signature"]
  310. if !ok {
  311. return errors.TraceNew("missing signature field")
  312. }
  313. signatureFieldStr, ok := signatureField.(string)
  314. if !ok {
  315. return errors.TraceNew("invalid signature field")
  316. }
  317. decodedSignatureField, err := base64.StdEncoding.DecodeString(signatureFieldStr)
  318. if err != nil {
  319. return errors.Trace(err)
  320. }
  321. if len(decodedSignatureField) < signaturePublicKeyDigestSize {
  322. return errors.TraceNew("invalid signature field length")
  323. }
  324. publicKeyID := decodedSignatureField[:signaturePublicKeyDigestSize]
  325. signature := decodedSignatureField[signaturePublicKeyDigestSize:]
  326. if len(signature) != ed25519.SignatureSize {
  327. return errors.TraceNew("invalid signature length")
  328. }
  329. decodedPublicKey, err := base64.StdEncoding.DecodeString(publicKey)
  330. if err != nil {
  331. return errors.Trace(err)
  332. }
  333. publicKeyDigest := sha256.Sum256(decodedPublicKey)
  334. expectedPublicKeyID := publicKeyDigest[:signaturePublicKeyDigestSize]
  335. if !bytes.Equal(expectedPublicKeyID, publicKeyID) {
  336. return errors.TraceNew("unexpected public key ID")
  337. }
  338. copyFields.RemoveUnsignedFields()
  339. delete(copyFields, "signature")
  340. marshaledFields, err := json.Marshal(copyFields)
  341. if err != nil {
  342. return errors.Trace(err)
  343. }
  344. if !ed25519.Verify(decodedPublicKey, marshaledFields, signature) {
  345. return errors.TraceNew("invalid signature")
  346. }
  347. return nil
  348. }
  349. // RemoveUnsignedFields prepares a server entry for signing or signature
  350. // verification by removing unsigned fields. The JSON marshalling of the
  351. // remaining fields is the data that is signed.
  352. func (fields ServerEntryFields) RemoveUnsignedFields() {
  353. delete(fields, "localSource")
  354. delete(fields, "localTimestamp")
  355. // Only non-local, explicit tags are part of the signature
  356. isLocalDerivedTag := fields["isLocalDerivedTag"]
  357. isLocalDerivedTagBool, ok := isLocalDerivedTag.(bool)
  358. if ok && isLocalDerivedTagBool {
  359. delete(fields, "tag")
  360. }
  361. delete(fields, "isLocalDerivedTag")
  362. }
  363. // NewServerEntrySignatureKeyPair creates an ed25519 key pair for use in
  364. // server entry signing and verification.
  365. func NewServerEntrySignatureKeyPair() (string, string, error) {
  366. publicKey, privateKey, err := ed25519.GenerateKey(rand.Reader)
  367. if err != nil {
  368. return "", "", errors.Trace(err)
  369. }
  370. return base64.StdEncoding.EncodeToString(publicKey),
  371. base64.StdEncoding.EncodeToString(privateKey),
  372. nil
  373. }
  374. // GetCapability returns the server capability corresponding
  375. // to the tunnel protocol.
  376. func GetCapability(protocol string) string {
  377. return strings.TrimSuffix(protocol, "-OSSH")
  378. }
  379. // GetTacticsCapability returns the server tactics capability
  380. // corresponding to the tunnel protocol.
  381. func GetTacticsCapability(protocol string) string {
  382. return GetCapability(protocol) + "-TACTICS"
  383. }
  384. // hasCapability indicates if the server entry has the specified capability.
  385. //
  386. // Any internal "PASSTHROUGH" componant in the server entry's capabilities is
  387. // ignored. The PASSTHROUGH component is used to mask protocols which are
  388. // running the passthrough mechanism from older clients which do not implement
  389. // the passthrough message. Older clients will treat these capabilities as
  390. // unknown protocols and skip them.
  391. func (serverEntry *ServerEntry) hasCapability(requiredCapability string) bool {
  392. for _, capability := range serverEntry.Capabilities {
  393. capability = strings.ReplaceAll(capability, "-PASSTHROUGH", "")
  394. if capability == requiredCapability {
  395. return true
  396. }
  397. }
  398. return false
  399. }
  400. // SupportsProtocol returns true if and only if the ServerEntry has
  401. // the necessary capability to support the specified tunnel protocol.
  402. func (serverEntry *ServerEntry) SupportsProtocol(protocol string) bool {
  403. requiredCapability := GetCapability(protocol)
  404. return serverEntry.hasCapability(requiredCapability)
  405. }
  406. // ConditionallyEnabledComponents defines an interface which can be queried to
  407. // determine which conditionally compiled protocol components are present.
  408. type ConditionallyEnabledComponents interface {
  409. QUICEnabled() bool
  410. MarionetteEnabled() bool
  411. RefractionNetworkingEnabled() bool
  412. }
  413. // GetSupportedProtocols returns a list of tunnel protocols supported
  414. // by the ServerEntry's capabilities.
  415. func (serverEntry *ServerEntry) GetSupportedProtocols(
  416. conditionallyEnabled ConditionallyEnabledComponents,
  417. useUpstreamProxy bool,
  418. limitTunnelProtocols []string,
  419. excludeIntensive bool) []string {
  420. supportedProtocols := make([]string, 0)
  421. for _, protocol := range SupportedTunnelProtocols {
  422. if useUpstreamProxy && !TunnelProtocolSupportsUpstreamProxy(protocol) {
  423. continue
  424. }
  425. if len(limitTunnelProtocols) > 0 {
  426. if !common.Contains(limitTunnelProtocols, protocol) {
  427. continue
  428. }
  429. } else {
  430. if common.Contains(DefaultDisabledTunnelProtocols, protocol) {
  431. continue
  432. }
  433. }
  434. if excludeIntensive && TunnelProtocolIsResourceIntensive(protocol) {
  435. continue
  436. }
  437. if (TunnelProtocolUsesQUIC(protocol) && !conditionallyEnabled.QUICEnabled()) ||
  438. (TunnelProtocolUsesMarionette(protocol) && !conditionallyEnabled.MarionetteEnabled()) ||
  439. (TunnelProtocolUsesRefractionNetworking(protocol) &&
  440. !conditionallyEnabled.RefractionNetworkingEnabled()) {
  441. continue
  442. }
  443. if serverEntry.SupportsProtocol(protocol) {
  444. supportedProtocols = append(supportedProtocols, protocol)
  445. }
  446. }
  447. return supportedProtocols
  448. }
  449. // GetSupportedTacticsProtocols returns a list of tunnel protocols,
  450. // supported by the ServerEntry's capabilities, that may be used
  451. // for tactics requests.
  452. func (serverEntry *ServerEntry) GetSupportedTacticsProtocols() []string {
  453. supportedProtocols := make([]string, 0)
  454. for _, protocol := range SupportedTunnelProtocols {
  455. if !TunnelProtocolUsesMeek(protocol) {
  456. continue
  457. }
  458. requiredCapability := GetTacticsCapability(protocol)
  459. if !serverEntry.hasCapability(requiredCapability) {
  460. continue
  461. }
  462. supportedProtocols = append(supportedProtocols, protocol)
  463. }
  464. return supportedProtocols
  465. }
  466. // SupportsSSHAPIRequests returns true when the server supports
  467. // SSH API requests.
  468. func (serverEntry *ServerEntry) SupportsSSHAPIRequests() bool {
  469. return serverEntry.hasCapability(CAPABILITY_SSH_API_REQUESTS)
  470. }
  471. func (serverEntry *ServerEntry) GetUntunneledWebRequestPorts() []string {
  472. ports := make([]string, 0)
  473. if serverEntry.hasCapability(CAPABILITY_UNTUNNELED_WEB_API_REQUESTS) {
  474. // Server-side configuration quirk: there's a port forward from
  475. // port 443 to the web server, which we can try, except on servers
  476. // running FRONTED_MEEK, which listens on port 443.
  477. if !serverEntry.SupportsProtocol(TUNNEL_PROTOCOL_FRONTED_MEEK) {
  478. ports = append(ports, "443")
  479. }
  480. ports = append(ports, serverEntry.WebServerPort)
  481. }
  482. return ports
  483. }
  484. func (serverEntry *ServerEntry) HasSignature() bool {
  485. return serverEntry.Signature != ""
  486. }
  487. func (serverEntry *ServerEntry) GetDiagnosticID() string {
  488. return TagToDiagnosticID(serverEntry.Tag)
  489. }
  490. // GenerateServerEntryTag creates a server entry tag value that is
  491. // cryptographically derived from the IP address and web server secret in a
  492. // way that is difficult to reverse the IP address value from the tag or
  493. // compute the tag without having the web server secret, a 256-bit random
  494. // value which is unique per server, in addition to the IP address. A database
  495. // consisting only of server entry tags should be resistent to an attack that
  496. // attempts to reverse all the server IPs, even given a small IP space (IPv4),
  497. // or some subset of the web server secrets.
  498. func GenerateServerEntryTag(ipAddress, webServerSecret string) string {
  499. h := hmac.New(sha256.New, []byte(webServerSecret))
  500. h.Write([]byte(ipAddress))
  501. return base64.StdEncoding.EncodeToString(h.Sum(nil))
  502. }
  503. // TagToDiagnosticID returns a prefix of the server entry tag that should be
  504. // sufficient to uniquely identify servers in diagnostics, while also being
  505. // more human readable than emitting the full tag. The tag is used as the base
  506. // of the diagnostic ID as it doesn't leak the server IP address in diagnostic
  507. // output.
  508. func TagToDiagnosticID(tag string) string {
  509. if len(tag) < 8 {
  510. return "<unknown>"
  511. }
  512. return tag[:8]
  513. }
  514. // EncodeServerEntry returns a string containing the encoding of
  515. // a ServerEntry following Psiphon conventions.
  516. func EncodeServerEntry(serverEntry *ServerEntry) (string, error) {
  517. return encodeServerEntry(
  518. serverEntry.IpAddress,
  519. serverEntry.WebServerPort,
  520. serverEntry.WebServerSecret,
  521. serverEntry.WebServerCertificate,
  522. serverEntry)
  523. }
  524. // EncodeServerEntryFields returns a string containing the encoding of
  525. // ServerEntryFields following Psiphon conventions.
  526. func EncodeServerEntryFields(serverEntryFields ServerEntryFields) (string, error) {
  527. return encodeServerEntry(
  528. serverEntryFields.GetIPAddress(),
  529. serverEntryFields.GetWebServerPort(),
  530. serverEntryFields.GetWebServerSecret(),
  531. serverEntryFields.GetWebServerCertificate(),
  532. serverEntryFields)
  533. }
  534. func encodeServerEntry(
  535. prefixIPAddress string,
  536. prefixWebServerPort string,
  537. prefixWebServerSecret string,
  538. prefixWebServerCertificate string,
  539. serverEntry interface{}) (string, error) {
  540. serverEntryJSON, err := json.Marshal(serverEntry)
  541. if err != nil {
  542. return "", errors.Trace(err)
  543. }
  544. // Legacy clients use a space-delimited fields prefix, and all clients expect
  545. // to at least parse the prefix in order to skip over it.
  546. //
  547. // When the server entry has no web API server certificate, the entire prefix
  548. // can be compacted down to single character placeholders. Clients that can
  549. // use the ssh API always prefer it over the web API and won't use the prefix
  550. // values.
  551. if len(prefixWebServerCertificate) == 0 {
  552. prefixIPAddress = "0"
  553. prefixWebServerPort = "0"
  554. prefixWebServerSecret = "0"
  555. prefixWebServerCertificate = "0"
  556. }
  557. return hex.EncodeToString([]byte(fmt.Sprintf(
  558. "%s %s %s %s %s",
  559. prefixIPAddress,
  560. prefixWebServerPort,
  561. prefixWebServerSecret,
  562. prefixWebServerCertificate,
  563. serverEntryJSON))), nil
  564. }
  565. // DecodeServerEntry extracts a server entry from the encoding
  566. // used by remote server lists and Psiphon server handshake requests.
  567. //
  568. // The resulting ServerEntry.LocalSource is populated with serverEntrySource,
  569. // which should be one of SERVER_ENTRY_SOURCE_EMBEDDED, SERVER_ENTRY_SOURCE_REMOTE,
  570. // SERVER_ENTRY_SOURCE_DISCOVERY, SERVER_ENTRY_SOURCE_TARGET,
  571. // SERVER_ENTRY_SOURCE_OBFUSCATED.
  572. // ServerEntry.LocalTimestamp is populated with the provided timestamp, which
  573. // should be a RFC 3339 formatted string. These local fields are stored with the
  574. // server entry and reported to the server as stats (a coarse granularity timestamp
  575. // is reported).
  576. func DecodeServerEntry(
  577. encodedServerEntry, timestamp, serverEntrySource string) (*ServerEntry, error) {
  578. serverEntry := new(ServerEntry)
  579. err := decodeServerEntry(encodedServerEntry, timestamp, serverEntrySource, serverEntry)
  580. if err != nil {
  581. return nil, errors.Trace(err)
  582. }
  583. // NOTE: if the source JSON happens to have values in these fields, they get clobbered.
  584. serverEntry.LocalSource = serverEntrySource
  585. serverEntry.LocalTimestamp = timestamp
  586. return serverEntry, nil
  587. }
  588. // DecodeServerEntryFields extracts an encoded server entry into a
  589. // ServerEntryFields type, much like DecodeServerEntry. Unrecognized fields
  590. // not in ServerEntry are retained in the ServerEntryFields.
  591. //
  592. // LocalSource/LocalTimestamp map entries are set only when the corresponding
  593. // inputs are non-blank.
  594. func DecodeServerEntryFields(
  595. encodedServerEntry, timestamp, serverEntrySource string) (ServerEntryFields, error) {
  596. serverEntryFields := make(ServerEntryFields)
  597. err := decodeServerEntry(encodedServerEntry, timestamp, serverEntrySource, &serverEntryFields)
  598. if err != nil {
  599. return nil, errors.Trace(err)
  600. }
  601. // NOTE: if the source JSON happens to have values in these fields, they get clobbered.
  602. if serverEntrySource != "" {
  603. serverEntryFields.SetLocalSource(serverEntrySource)
  604. }
  605. if timestamp != "" {
  606. serverEntryFields.SetLocalTimestamp(timestamp)
  607. }
  608. return serverEntryFields, nil
  609. }
  610. func decodeServerEntry(
  611. encodedServerEntry, timestamp, serverEntrySource string,
  612. target interface{}) error {
  613. hexDecodedServerEntry, err := hex.DecodeString(encodedServerEntry)
  614. if err != nil {
  615. return errors.Trace(err)
  616. }
  617. // Skip past legacy format (4 space delimited fields) and just parse the JSON config
  618. fields := bytes.SplitN(hexDecodedServerEntry, []byte(" "), 5)
  619. if len(fields) != 5 {
  620. return errors.TraceNew("invalid encoded server entry")
  621. }
  622. err = json.Unmarshal(fields[4], target)
  623. if err != nil {
  624. return errors.Trace(err)
  625. }
  626. return nil
  627. }
  628. // ValidateServerEntryFields checks for malformed server entries.
  629. func ValidateServerEntryFields(serverEntryFields ServerEntryFields) error {
  630. // Checks for a valid ipAddress. This is important since the IP
  631. // address is the key used to store/lookup the server entry.
  632. ipAddress := serverEntryFields.GetIPAddress()
  633. if net.ParseIP(ipAddress) == nil {
  634. return errors.Tracef("server entry has invalid ipAddress: %s", ipAddress)
  635. }
  636. // TODO: validate more fields?
  637. // Ensure locally initialized fields have been set.
  638. source := serverEntryFields.GetLocalSource()
  639. if !common.Contains(
  640. SupportedServerEntrySources, source) {
  641. return errors.Tracef("server entry has invalid source: %s", source)
  642. }
  643. timestamp := serverEntryFields.GetLocalTimestamp()
  644. _, err := time.Parse(time.RFC3339, timestamp)
  645. if err != nil {
  646. return errors.Tracef("server entry has invalid timestamp: %s", err)
  647. }
  648. return nil
  649. }
  650. // DecodeServerEntryList extracts server entries from the list encoding
  651. // used by remote server lists and Psiphon server handshake requests.
  652. // Each server entry is validated and invalid entries are skipped.
  653. // See DecodeServerEntry for note on serverEntrySource/timestamp.
  654. func DecodeServerEntryList(
  655. encodedServerEntryList, timestamp,
  656. serverEntrySource string) ([]ServerEntryFields, error) {
  657. serverEntries := make([]ServerEntryFields, 0)
  658. for _, encodedServerEntry := range strings.Split(encodedServerEntryList, "\n") {
  659. if len(encodedServerEntry) == 0 {
  660. continue
  661. }
  662. // TODO: skip this entry and continue if can't decode?
  663. serverEntryFields, err := DecodeServerEntryFields(encodedServerEntry, timestamp, serverEntrySource)
  664. if err != nil {
  665. return nil, errors.Trace(err)
  666. }
  667. if ValidateServerEntryFields(serverEntryFields) != nil {
  668. // Skip this entry and continue with the next one
  669. // TODO: invoke a logging callback
  670. continue
  671. }
  672. serverEntries = append(serverEntries, serverEntryFields)
  673. }
  674. return serverEntries, nil
  675. }
  676. // StreamingServerEntryDecoder performs the DecodeServerEntryList
  677. // operation, loading only one server entry into memory at a time.
  678. type StreamingServerEntryDecoder struct {
  679. scanner *bufio.Scanner
  680. timestamp string
  681. serverEntrySource string
  682. }
  683. // NewStreamingServerEntryDecoder creates a new StreamingServerEntryDecoder.
  684. func NewStreamingServerEntryDecoder(
  685. encodedServerEntryListReader io.Reader,
  686. timestamp, serverEntrySource string) *StreamingServerEntryDecoder {
  687. return &StreamingServerEntryDecoder{
  688. scanner: bufio.NewScanner(encodedServerEntryListReader),
  689. timestamp: timestamp,
  690. serverEntrySource: serverEntrySource,
  691. }
  692. }
  693. // Next reads and decodes, and validates the next server entry from the
  694. // input stream, returning a nil server entry when the stream is complete.
  695. //
  696. // Limitations:
  697. // - Each encoded server entry line cannot exceed bufio.MaxScanTokenSize,
  698. // the default buffer size which this decoder uses. This is 64K.
  699. // - DecodeServerEntry is called on each encoded server entry line, which
  700. // will allocate memory to hex decode and JSON deserialze the server
  701. // entry. As this is not presently reusing a fixed buffer, each call
  702. // will allocate additional memory; garbage collection is necessary to
  703. // reclaim that memory for reuse for the next server entry.
  704. //
  705. func (decoder *StreamingServerEntryDecoder) Next() (ServerEntryFields, error) {
  706. for {
  707. if !decoder.scanner.Scan() {
  708. return nil, errors.Trace(decoder.scanner.Err())
  709. }
  710. // TODO: use scanner.Bytes which doesn't allocate, instead of scanner.Text
  711. // TODO: skip this entry and continue if can't decode?
  712. serverEntryFields, err := DecodeServerEntryFields(
  713. decoder.scanner.Text(), decoder.timestamp, decoder.serverEntrySource)
  714. if err != nil {
  715. return nil, errors.Trace(err)
  716. }
  717. if ValidateServerEntryFields(serverEntryFields) != nil {
  718. // Skip this entry and continue with the next one
  719. // TODO: invoke a logging callback
  720. continue
  721. }
  722. return serverEntryFields, nil
  723. }
  724. }