serverEntry.go 26 KB

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