| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775 |
- /*
- * Copyright (c) 2016, Psiphon Inc.
- * All rights reserved.
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- *
- */
- package server
- import (
- "bytes"
- "context"
- "crypto/rand"
- "crypto/tls"
- "encoding/base64"
- "encoding/hex"
- "encoding/json"
- std_errors "errors"
- "hash/crc64"
- "io"
- "net"
- "net/http"
- "runtime"
- "strconv"
- "strings"
- "sync"
- "sync/atomic"
- "time"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/errors"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/monotime"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/obfuscator"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/parameters"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/prng"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/protocol"
- "github.com/Psiphon-Labs/psiphon-tunnel-core/psiphon/common/values"
- tris "github.com/Psiphon-Labs/tls-tris"
- lrucache "github.com/cognusion/go-cache-lru"
- "github.com/juju/ratelimit"
- "golang.org/x/crypto/nacl/box"
- )
- // MeekServer is based on meek-server.go from Tor and Psiphon:
- //
- // https://gitweb.torproject.org/pluggable-transports/meek.git/blob/HEAD:/meek-client/meek-client.go
- // CC0 1.0 Universal
- //
- // https://bitbucket.org/psiphon/psiphon-circumvention-system/src/default/go/meek-client/meek-client.go
- const (
- // Protocol version 1 clients can handle arbitrary length response bodies. Older clients
- // report no version number and expect at most 64K response bodies.
- MEEK_PROTOCOL_VERSION_1 = 1
- // Protocol version 2 clients initiate a session by sending an encrypted and obfuscated meek
- // cookie with their initial HTTP request. Connection information is contained within the
- // encrypted cookie payload. The server inspects the cookie and establishes a new session and
- // returns a new random session ID back to client via Set-Cookie header. The client uses this
- // session ID on all subsequent requests for the remainder of the session.
- MEEK_PROTOCOL_VERSION_2 = 2
- // Protocol version 3 clients include resiliency enhancements and will add a Range header
- // when retrying a request for a partially downloaded response payload.
- MEEK_PROTOCOL_VERSION_3 = 3
- MEEK_MAX_REQUEST_PAYLOAD_LENGTH = 65536
- MEEK_MIN_SESSION_ID_LENGTH = 8
- MEEK_MAX_SESSION_ID_LENGTH = 20
- MEEK_DEFAULT_TURN_AROUND_TIMEOUT = 10 * time.Millisecond
- MEEK_DEFAULT_EXTENDED_TURN_AROUND_TIMEOUT = 100 * time.Millisecond
- MEEK_DEFAULT_SKIP_EXTENDED_TURN_AROUND_THRESHOLD = 8192
- MEEK_DEFAULT_MAX_SESSION_STALENESS = 45 * time.Second
- MEEK_DEFAULT_HTTP_CLIENT_IO_TIMEOUT = 45 * time.Second
- MEEK_DEFAULT_RESPONSE_BUFFER_LENGTH = 65536
- MEEK_DEFAULT_POOL_BUFFER_LENGTH = 65536
- MEEK_DEFAULT_POOL_BUFFER_COUNT = 2048
- )
- // MeekServer implements the meek protocol, which tunnels TCP traffic (in the case of Psiphon,
- // Obfuscated SSH traffic) over HTTP. Meek may be fronted (through a CDN) or direct and may be
- // HTTP or HTTPS.
- //
- // Upstream traffic arrives in HTTP request bodies and downstream traffic is sent in response
- // bodies. The sequence of traffic for a given flow is associated using a session ID that's
- // set as a HTTP cookie for the client to submit with each request.
- //
- // MeekServer hooks into TunnelServer via the net.Conn interface by transforming the
- // HTTP payload traffic for a given session into net.Conn conforming Read()s and Write()s via
- // the meekConn struct.
- type MeekServer struct {
- support *SupportServices
- listener net.Listener
- listenerTunnelProtocol string
- listenerPort int
- passthroughAddress string
- turnAroundTimeout time.Duration
- extendedTurnAroundTimeout time.Duration
- skipExtendedTurnAroundThreshold int
- maxSessionStaleness time.Duration
- httpClientIOTimeout time.Duration
- tlsConfig *tris.Config
- obfuscatorSeedHistory *obfuscator.SeedHistory
- clientHandler func(clientTunnelProtocol string, clientConn net.Conn)
- openConns *common.Conns
- stopBroadcast <-chan struct{}
- sessionsLock sync.RWMutex
- sessions map[string]*meekSession
- checksumTable *crc64.Table
- bufferPool *CachedResponseBufferPool
- rateLimitLock sync.Mutex
- rateLimitHistory *lrucache.Cache
- rateLimitCount int
- rateLimitSignalGC chan struct{}
- }
- // NewMeekServer initializes a new meek server.
- func NewMeekServer(
- support *SupportServices,
- listener net.Listener,
- listenerTunnelProtocol string,
- listenerPort int,
- useTLS, isFronted, useObfuscatedSessionTickets bool,
- clientHandler func(clientTunnelProtocol string, clientConn net.Conn),
- stopBroadcast <-chan struct{}) (*MeekServer, error) {
- passthroughAddress := support.Config.TunnelProtocolPassthroughAddresses[listenerTunnelProtocol]
- turnAroundTimeout := MEEK_DEFAULT_TURN_AROUND_TIMEOUT
- if support.Config.MeekTurnAroundTimeoutMilliseconds != nil {
- turnAroundTimeout = time.Duration(
- *support.Config.MeekTurnAroundTimeoutMilliseconds) * time.Millisecond
- }
- extendedTurnAroundTimeout := MEEK_DEFAULT_EXTENDED_TURN_AROUND_TIMEOUT
- if support.Config.MeekExtendedTurnAroundTimeoutMilliseconds != nil {
- extendedTurnAroundTimeout = time.Duration(
- *support.Config.MeekExtendedTurnAroundTimeoutMilliseconds) * time.Millisecond
- }
- skipExtendedTurnAroundThreshold := MEEK_DEFAULT_SKIP_EXTENDED_TURN_AROUND_THRESHOLD
- if support.Config.MeekSkipExtendedTurnAroundThresholdBytes != nil {
- skipExtendedTurnAroundThreshold = *support.Config.MeekSkipExtendedTurnAroundThresholdBytes
- }
- maxSessionStaleness := MEEK_DEFAULT_MAX_SESSION_STALENESS
- if support.Config.MeekMaxSessionStalenessMilliseconds != nil {
- maxSessionStaleness = time.Duration(
- *support.Config.MeekMaxSessionStalenessMilliseconds) * time.Millisecond
- }
- httpClientIOTimeout := MEEK_DEFAULT_HTTP_CLIENT_IO_TIMEOUT
- if support.Config.MeekHTTPClientIOTimeoutMilliseconds != nil {
- httpClientIOTimeout = time.Duration(
- *support.Config.MeekHTTPClientIOTimeoutMilliseconds) * time.Millisecond
- }
- checksumTable := crc64.MakeTable(crc64.ECMA)
- bufferLength := MEEK_DEFAULT_POOL_BUFFER_LENGTH
- if support.Config.MeekCachedResponsePoolBufferSize != 0 {
- bufferLength = support.Config.MeekCachedResponsePoolBufferSize
- }
- bufferCount := MEEK_DEFAULT_POOL_BUFFER_COUNT
- if support.Config.MeekCachedResponsePoolBufferCount != 0 {
- bufferCount = support.Config.MeekCachedResponsePoolBufferCount
- }
- _, thresholdSeconds, _, _, _, _, _, _, reapFrequencySeconds, maxEntries :=
- support.TrafficRulesSet.GetMeekRateLimiterConfig()
- rateLimitHistory := lrucache.NewWithLRU(
- time.Duration(thresholdSeconds)*time.Second,
- time.Duration(reapFrequencySeconds)*time.Second,
- maxEntries)
- bufferPool := NewCachedResponseBufferPool(bufferLength, bufferCount)
- meekServer := &MeekServer{
- support: support,
- listener: listener,
- listenerTunnelProtocol: listenerTunnelProtocol,
- listenerPort: listenerPort,
- passthroughAddress: passthroughAddress,
- turnAroundTimeout: turnAroundTimeout,
- extendedTurnAroundTimeout: extendedTurnAroundTimeout,
- skipExtendedTurnAroundThreshold: skipExtendedTurnAroundThreshold,
- maxSessionStaleness: maxSessionStaleness,
- httpClientIOTimeout: httpClientIOTimeout,
- obfuscatorSeedHistory: obfuscator.NewSeedHistory(nil),
- clientHandler: clientHandler,
- openConns: common.NewConns(),
- stopBroadcast: stopBroadcast,
- sessions: make(map[string]*meekSession),
- checksumTable: checksumTable,
- bufferPool: bufferPool,
- rateLimitHistory: rateLimitHistory,
- rateLimitSignalGC: make(chan struct{}, 1),
- }
- if useTLS {
- tlsConfig, err := meekServer.makeMeekTLSConfig(
- isFronted, useObfuscatedSessionTickets)
- if err != nil {
- return nil, errors.Trace(err)
- }
- meekServer.tlsConfig = tlsConfig
- }
- return meekServer, nil
- }
- type meekContextKey struct {
- key string
- }
- var meekNetConnContextKey = &meekContextKey{"net.Conn"}
- // Run runs the meek server; this function blocks while serving HTTP or
- // HTTPS connections on the specified listener. This function also runs
- // a goroutine which cleans up expired meek client sessions.
- //
- // To stop the meek server, both Close() the listener and set the stopBroadcast
- // signal specified in NewMeekServer.
- func (server *MeekServer) Run() error {
- waitGroup := new(sync.WaitGroup)
- waitGroup.Add(1)
- go func() {
- defer waitGroup.Done()
- ticker := time.NewTicker(server.maxSessionStaleness / 2)
- defer ticker.Stop()
- for {
- select {
- case <-ticker.C:
- server.deleteExpiredSessions()
- case <-server.stopBroadcast:
- return
- }
- }
- }()
- waitGroup.Add(1)
- go func() {
- defer waitGroup.Done()
- server.rateLimitWorker()
- }()
- // Serve HTTP or HTTPS
- //
- // - WriteTimeout may include time awaiting request, as per:
- // https://blog.cloudflare.com/the-complete-guide-to-golang-net-http-timeouts
- // - Legacy meek-server wrapped each client HTTP connection with an explicit idle
- // timeout net.Conn and didn't use http.Server timeouts. We could do the same
- // here (use ActivityMonitoredConn) but the stock http.Server timeouts should
- // now be sufficient.
- httpServer := &http.Server{
- ReadTimeout: server.httpClientIOTimeout,
- WriteTimeout: server.httpClientIOTimeout,
- Handler: server,
- ConnState: server.httpConnStateCallback,
- ConnContext: func(ctx context.Context, conn net.Conn) context.Context {
- return context.WithValue(ctx, meekNetConnContextKey, conn)
- },
- // Disable auto HTTP/2 (https://golang.org/doc/go1.6)
- TLSNextProto: make(map[string]func(*http.Server, *tls.Conn, http.Handler)),
- }
- // Note: Serve() will be interrupted by listener.Close() call
- var err error
- if server.tlsConfig != nil {
- httpsServer := HTTPSServer{Server: httpServer}
- err = httpsServer.ServeTLS(server.listener, server.tlsConfig)
- } else {
- err = httpServer.Serve(server.listener)
- }
- // Can't check for the exact error that Close() will cause in Accept(),
- // (see: https://code.google.com/p/go/issues/detail?id=4373). So using an
- // explicit stop signal to stop gracefully.
- select {
- case <-server.stopBroadcast:
- err = nil
- default:
- }
- // deleteExpiredSessions calls deleteSession which may block waiting
- // for active request handlers to complete; timely shutdown requires
- // stopping the listener and closing all existing connections before
- // awaiting the reaperWaitGroup.
- server.listener.Close()
- server.openConns.CloseAll()
- waitGroup.Wait()
- return err
- }
- // ServeHTTP handles meek client HTTP requests, where the request body
- // contains upstream traffic and the response will contain downstream
- // traffic.
- func (server *MeekServer) ServeHTTP(responseWriter http.ResponseWriter, request *http.Request) {
- // Note: no longer requiring that the request method is POST
- // Check for the expected meek/session ID cookie.
- // Also check for prohibited HTTP headers.
- var meekCookie *http.Cookie
- for _, c := range request.Cookies() {
- meekCookie = c
- break
- }
- if meekCookie == nil || len(meekCookie.Value) == 0 {
- log.WithTrace().Warning("missing meek cookie")
- common.TerminateHTTPConnection(responseWriter, request)
- return
- }
- if len(server.support.Config.MeekProhibitedHeaders) > 0 {
- for _, header := range server.support.Config.MeekProhibitedHeaders {
- value := request.Header.Get(header)
- if header != "" {
- log.WithTraceFields(LogFields{
- "header": header,
- "value": value,
- }).Warning("prohibited meek header")
- common.TerminateHTTPConnection(responseWriter, request)
- return
- }
- }
- }
- // A valid meek cookie indicates which class of request this is:
- //
- // 1. A new meek session. Create a new session ID and proceed with
- // relaying tunnel traffic.
- //
- // 2. An existing meek session. Resume relaying tunnel traffic.
- //
- // 3. A request to an endpoint. This meek connection is not for relaying
- // tunnel traffic. Instead, the request is handed off to a custom handler.
- sessionID,
- session,
- underlyingConn,
- endPoint,
- endPointGeoIPData,
- err := server.getSessionOrEndpoint(request, meekCookie)
- if err != nil {
- // Debug since session cookie errors commonly occur during
- // normal operation.
- log.WithTraceFields(LogFields{"error": err}).Debug("session lookup failed")
- common.TerminateHTTPConnection(responseWriter, request)
- return
- }
- if endPoint != "" {
- // Endpoint mode. Currently, this means it's handled by the tactics
- // request handler.
- handled := server.support.TacticsServer.HandleEndPoint(
- endPoint, common.GeoIPData(*endPointGeoIPData), responseWriter, request)
- if !handled {
- log.WithTraceFields(LogFields{"endPoint": endPoint}).Info("unhandled endpoint")
- common.TerminateHTTPConnection(responseWriter, request)
- }
- return
- }
- // Tunnel relay mode.
- // Ensure that there's only one concurrent request handler per client
- // session. Depending on the nature of a network disruption, it can
- // happen that a client detects a failure and retries while the server
- // is still streaming response in the handler for the _previous_ client
- // request.
- //
- // Even if the session.cachedResponse were safe for concurrent
- // use (it is not), concurrent handling could lead to loss of session
- // since upstream data read by the first request may not reach the
- // cached response before the second request reads the cached data.
- //
- // The existing handler will stream response data, holding the lock,
- // for no more than MEEK_EXTENDED_TURN_AROUND_TIMEOUT.
- //
- // TODO: interrupt an existing handler? The existing handler will be
- // sending data to the cached response, but if that buffer fills, the
- // session will be lost.
- requestNumber := atomic.AddInt64(&session.requestCount, 1)
- // Wait for the existing request to complete.
- session.lock.Lock()
- defer session.lock.Unlock()
- // Count this metric once the lock is acquired, to avoid concurrent and
- // potentially incorrect session.underlyingConn updates.
- //
- // It should never be the case that a new underlyingConn has the same
- // value as the previous session.underlyingConn, as each is a net.Conn
- // interface which includes a pointer, and the previous value cannot
- // be garbage collected until session.underlyingConn is updated.
- if session.underlyingConn != underlyingConn {
- atomic.AddInt64(&session.metricUnderlyingConnCount, 1)
- session.underlyingConn = underlyingConn
- }
- // If a newer request has arrived while waiting, discard this one.
- // Do not delay processing the newest request.
- //
- // If the session expired and was deleted while this request was waiting,
- // discard this request. The session is no longer valid, and the final call
- // to session.cachedResponse.Reset may have already occured, so any further
- // session.cachedResponse access may deplete resources (fail to refill the pool).
- if atomic.LoadInt64(&session.requestCount) > requestNumber || session.deleted {
- common.TerminateHTTPConnection(responseWriter, request)
- return
- }
- // pumpReads causes a TunnelServer/SSH goroutine blocking on a Read to
- // read the request body as upstream traffic.
- // TODO: run pumpReads and pumpWrites concurrently?
- // pumpReads checksums the request payload and skips relaying it when
- // it matches the immediately previous request payload. This allows
- // clients to resend request payloads, when retrying due to connection
- // interruption, without knowing whether the server has received or
- // relayed the data.
- requestSize, err := session.clientConn.pumpReads(request.Body)
- if err != nil {
- if err != io.EOF {
- // Debug since errors such as "i/o timeout" occur during normal operation;
- // also, golang network error messages may contain client IP.
- log.WithTraceFields(LogFields{"error": err}).Debug("read request failed")
- }
- common.TerminateHTTPConnection(responseWriter, request)
- // Note: keep session open to allow client to retry
- return
- }
- // The extended turn around mechanism optimizes for downstream flows by
- // sending more data in the response as long as it's available. As a
- // heuristic, when the request size meets a threshold, optimize instead
- // of upstream flows by skipping the extended turn around.
- skipExtendedTurnAround := requestSize >= int64(server.skipExtendedTurnAroundThreshold)
- // Set cookie before writing the response.
- if session.meekProtocolVersion >= MEEK_PROTOCOL_VERSION_2 && !session.sessionIDSent {
- // Replace the meek cookie with the session ID.
- // SetCookie for the the session ID cookie is only set once, to reduce overhead. This
- // session ID value replaces the original meek cookie value.
- http.SetCookie(responseWriter, &http.Cookie{Name: meekCookie.Name, Value: sessionID})
- session.sessionIDSent = true
- }
- // When streaming data into the response body, a copy is
- // retained in the cachedResponse buffer. This allows the
- // client to retry and request that the response be resent
- // when the HTTP connection is interrupted.
- //
- // If a Range header is present, the client is retrying,
- // possibly after having received a partial response. In
- // this case, use any cached response to attempt to resend
- // the response, starting from the resend position the client
- // indicates.
- //
- // When the resend position is not available -- because the
- // cachedResponse buffer could not hold it -- the client session
- // is closed, as there's no way to resume streaming the payload
- // uninterrupted.
- //
- // The client may retry before a cached response is prepared,
- // so a cached response is not always used when a Range header
- // is present.
- //
- // TODO: invalid Range header is ignored; should it be otherwise?
- position, isRetry := checkRangeHeader(request)
- if isRetry {
- atomic.AddInt64(&session.metricClientRetries, 1)
- }
- hasCompleteCachedResponse := session.cachedResponse.HasPosition(0)
- // The client is not expected to send position > 0 when there is
- // no cached response; let that case fall through to the next
- // HasPosition check which will fail and close the session.
- var responseSize int
- var responseError error
- if isRetry && (hasCompleteCachedResponse || position > 0) {
- if !session.cachedResponse.HasPosition(position) {
- greaterThanSwapInt64(&session.metricCachedResponseMissPosition, int64(position))
- common.TerminateHTTPConnection(responseWriter, request)
- session.delete(true)
- return
- }
- responseWriter.WriteHeader(http.StatusPartialContent)
- // TODO:
- // - enforce a max extended buffer count per client, for
- // fairness? Throttling may make this unnecessary.
- // - cachedResponse can now start releasing extended buffers,
- // as response bytes before "position" will never be requested
- // again?
- responseSize, responseError = session.cachedResponse.CopyFromPosition(position, responseWriter)
- greaterThanSwapInt64(&session.metricPeakCachedResponseHitSize, int64(responseSize))
- // The client may again fail to receive the payload and may again
- // retry, so not yet releasing cachedResponse buffers.
- } else {
- // _Now_ we release buffers holding data from the previous
- // response. And then immediately stream the new response into
- // newly acquired buffers.
- session.cachedResponse.Reset()
- // Note: this code depends on an implementation detail of
- // io.MultiWriter: a Write() to the MultiWriter writes first
- // to the cache, and then to the response writer. So if the
- // write to the response writer fails, the payload is cached.
- multiWriter := io.MultiWriter(session.cachedResponse, responseWriter)
- // The client expects 206, not 200, whenever it sets a Range header,
- // which it may do even when no cached response is prepared.
- if isRetry {
- responseWriter.WriteHeader(http.StatusPartialContent)
- }
- // pumpWrites causes a TunnelServer/SSH goroutine blocking on a Write to
- // write its downstream traffic through to the response body.
- responseSize, responseError = session.clientConn.pumpWrites(multiWriter, skipExtendedTurnAround)
- greaterThanSwapInt64(&session.metricPeakResponseSize, int64(responseSize))
- greaterThanSwapInt64(&session.metricPeakCachedResponseSize, int64(session.cachedResponse.Available()))
- }
- // responseError is the result of writing the body either from CopyFromPosition or pumpWrites
- if responseError != nil {
- if responseError != io.EOF {
- // Debug since errors such as "i/o timeout" occur during normal operation;
- // also, golang network error messages may contain client IP.
- log.WithTraceFields(LogFields{"error": responseError}).Debug("write response failed")
- }
- common.TerminateHTTPConnection(responseWriter, request)
- // Note: keep session open to allow client to retry
- return
- }
- }
- func checkRangeHeader(request *http.Request) (int, bool) {
- rangeHeader := request.Header.Get("Range")
- if rangeHeader == "" {
- return 0, false
- }
- prefix := "bytes="
- suffix := "-"
- if !strings.HasPrefix(rangeHeader, prefix) ||
- !strings.HasSuffix(rangeHeader, suffix) {
- return 0, false
- }
- rangeHeader = strings.TrimPrefix(rangeHeader, prefix)
- rangeHeader = strings.TrimSuffix(rangeHeader, suffix)
- position, err := strconv.Atoi(rangeHeader)
- if err != nil {
- return 0, false
- }
- return position, true
- }
- // getSessionOrEndpoint checks if the cookie corresponds to an existing tunnel
- // relay session ID. If no session is found, the cookie must be an obfuscated
- // meek cookie. A new session is created when the meek cookie indicates relay
- // mode; or the endpoint is returned when the meek cookie indicates endpoint
- // mode.
- func (server *MeekServer) getSessionOrEndpoint(
- request *http.Request, meekCookie *http.Cookie) (string, *meekSession, net.Conn, string, *GeoIPData, error) {
- underlyingConn := request.Context().Value(meekNetConnContextKey).(net.Conn)
- // Check for an existing session.
- server.sessionsLock.RLock()
- existingSessionID := meekCookie.Value
- session, ok := server.sessions[existingSessionID]
- server.sessionsLock.RUnlock()
- if ok {
- // TODO: can multiple http client connections using same session cookie
- // cause race conditions on session struct?
- session.touch()
- return existingSessionID, session, underlyingConn, "", nil, nil
- }
- // Determine the client remote address, which is used for geolocation
- // stats, rate limiting, anti-probing, discovery, and tactics selection
- // logic.
- //
- // When an intermediate proxy or CDN is in use, we may be
- // able to determine the original client address by inspecting HTTP
- // headers such as X-Forwarded-For.
- //
- // We trust only headers provided by CDNs. Fronted Psiphon server hosts
- // should be configured to accept tunnel connections only from CDN edges.
- // When the CDN passes along a chain of IPs, as in X-Forwarded-For, we
- // trust only the right-most IP, which is provided by the CDN.
- clientIP, _, err := net.SplitHostPort(request.RemoteAddr)
- if err != nil {
- return "", nil, nil, "", nil, errors.Trace(err)
- }
- if net.ParseIP(clientIP) == nil {
- return "", nil, nil, "", nil, errors.TraceNew("invalid IP address")
- }
- if protocol.TunnelProtocolUsesFrontedMeek(server.listenerTunnelProtocol) &&
- len(server.support.Config.MeekProxyForwardedForHeaders) > 0 {
- // When there are multiple header names in MeekProxyForwardedForHeaders,
- // the first valid match is preferred. MeekProxyForwardedForHeaders should be
- // configured to use header names that are always provided by the CDN(s) and
- // not header names that may be passed through from clients.
- for _, header := range server.support.Config.MeekProxyForwardedForHeaders {
- // In the case where there are multiple headers,
- // request.Header.Get returns the first header, but we want the
- // last header; so use request.Header.Values and select the last
- // value. As per RFC 2616 section 4.2, a proxy must not change
- // the order of field values, which implies that it should append
- // values to the last header.
- values := request.Header.Values(header)
- if len(values) > 0 {
- value := values[len(values)-1]
- // Some headers, such as X-Forwarded-For, are a comma-separated
- // list of IPs (each proxy in a chain). Select the last IP.
- IPs := strings.Split(value, ",")
- IP := IPs[len(IPs)-1]
- // Remove optional whitespace surrounding the commas.
- IP = strings.TrimSpace(IP)
- if net.ParseIP(IP) != nil {
- clientIP = IP
- break
- }
- }
- }
- }
- geoIPData := server.support.GeoIPService.Lookup(clientIP)
- // The session is new (or expired). Treat the cookie value as a new meek
- // cookie, extract the payload, and create a new session.
- // Limitation: when the cookie is a session ID for an expired session, we
- // still attempt to treat it as a meek cookie. As it stands, that yields
- // either base64 decoding errors (RawStdEncoding vs. StdEncoding) or
- // length errors. We could log cleaner errors ("session is expired") by
- // checking that the cookie is a well-formed (base64.RawStdEncoding) value
- // between MEEK_MIN_SESSION_ID_LENGTH and MEEK_MAX_SESSION_ID_LENGTH
- // bytes -- assuming that MEEK_MAX_SESSION_ID_LENGTH is too short to be a
- // valid meek cookie.
- payloadJSON, err := server.getMeekCookiePayload(clientIP, meekCookie.Value)
- if err != nil {
- return "", nil, nil, "", nil, errors.Trace(err)
- }
- // Note: this meek server ignores legacy values PsiphonClientSessionId
- // and PsiphonServerAddress.
- var clientSessionData protocol.MeekCookieData
- err = json.Unmarshal(payloadJSON, &clientSessionData)
- if err != nil {
- return "", nil, nil, "", nil, errors.Trace(err)
- }
- tunnelProtocol := server.listenerTunnelProtocol
- if clientSessionData.ClientTunnelProtocol != "" {
- if !protocol.IsValidClientTunnelProtocol(
- clientSessionData.ClientTunnelProtocol,
- server.listenerTunnelProtocol,
- server.support.Config.GetRunningProtocols()) {
- return "", nil, nil, "", nil, errors.Tracef(
- "invalid client tunnel protocol: %s", clientSessionData.ClientTunnelProtocol)
- }
- tunnelProtocol = clientSessionData.ClientTunnelProtocol
- }
- // Any rate limit is enforced after the meek cookie is validated, so a prober
- // without the obfuscation secret will be unable to fingerprint the server
- // based on response time combined with the rate limit configuration. The
- // rate limit is primarily intended to limit memory resource consumption and
- // not the overhead incurred by cookie validation.
- if server.rateLimit(clientIP, geoIPData, tunnelProtocol) {
- return "", nil, nil, "", nil, errors.TraceNew("rate limit exceeded")
- }
- // Handle endpoints before enforcing CheckEstablishTunnels.
- // Currently, endpoints are tactics requests, and we allow these to be
- // handled by servers which would otherwise reject new tunnels.
- if clientSessionData.EndPoint != "" {
- return "", nil, nil, clientSessionData.EndPoint, &geoIPData, nil
- }
- // Don't create new sessions when not establishing. A subsequent SSH handshake
- // will not succeed, so creating a meek session just wastes resources.
- if server.support.TunnelServer != nil &&
- !server.support.TunnelServer.CheckEstablishTunnels() {
- return "", nil, nil, "", nil, errors.TraceNew("not establishing tunnels")
- }
- // Disconnect immediately if the tactics for the client restricts usage of
- // the fronting provider ID. The probability may be used to influence
- // usage of a given fronting provider; but when only that provider works
- // for a given client, and the probability is less than 1.0, the client
- // can retry until it gets a successful coin flip.
- //
- // Clients will also skip candidates with restricted fronting provider IDs.
- // The client-side probability, RestrictFrontingProviderIDsClientProbability,
- // is applied independently of the server-side coin flip here.
- //
- // At this stage, GeoIP tactics filters are active, but handshake API
- // parameters are not.
- //
- // See the comment in server.LoadConfig regarding fronting provider ID
- // limitations.
- if protocol.TunnelProtocolUsesFrontedMeek(server.listenerTunnelProtocol) &&
- server.support.ServerTacticsParametersCache != nil {
- p, err := server.support.ServerTacticsParametersCache.Get(geoIPData)
- if err != nil {
- return "", nil, nil, "", nil, errors.Trace(err)
- }
- if !p.IsNil() &&
- common.Contains(
- p.Strings(parameters.RestrictFrontingProviderIDs),
- server.support.Config.GetFrontingProviderID()) {
- if p.WeightedCoinFlip(
- parameters.RestrictFrontingProviderIDsServerProbability) {
- return "", nil, nil, "", nil, errors.TraceNew("restricted fronting provider")
- }
- }
- }
- // Create a new session
- bufferLength := MEEK_DEFAULT_RESPONSE_BUFFER_LENGTH
- if server.support.Config.MeekCachedResponseBufferSize != 0 {
- bufferLength = server.support.Config.MeekCachedResponseBufferSize
- }
- cachedResponse := NewCachedResponse(bufferLength, server.bufferPool)
- session = &meekSession{
- meekProtocolVersion: clientSessionData.MeekProtocolVersion,
- sessionIDSent: false,
- cachedResponse: cachedResponse,
- cookieName: meekCookie.Name,
- contentType: request.Header.Get("Content-Type"),
- }
- session.touch()
- // Create a new meek conn that will relay the payload
- // between meek request/responses and the tunnel server client
- // handler. The client IP is also used to initialize the
- // meek conn with a useful value to return when the tunnel
- // server calls conn.RemoteAddr() to get the client's IP address.
- // Assumes clientIP is a valid IP address; the port value is a stub
- // and is expected to be ignored.
- clientConn := newMeekConn(
- server,
- session,
- underlyingConn,
- &net.TCPAddr{
- IP: net.ParseIP(clientIP),
- Port: 0,
- },
- clientSessionData.MeekProtocolVersion)
- session.clientConn = clientConn
- // Note: MEEK_PROTOCOL_VERSION_1 doesn't support changing the
- // meek cookie to a session ID; v1 clients always send the
- // original meek cookie value with each request. The issue with
- // v1 is that clients which wake after a device sleep will attempt
- // to resume a meek session and the server can't differentiate
- // between resuming a session and creating a new session. This
- // causes the v1 client connection to hang/timeout.
- sessionID := meekCookie.Value
- if clientSessionData.MeekProtocolVersion >= MEEK_PROTOCOL_VERSION_2 {
- sessionID, err = makeMeekSessionID()
- if err != nil {
- return "", nil, nil, "", nil, errors.Trace(err)
- }
- }
- server.sessionsLock.Lock()
- server.sessions[sessionID] = session
- server.sessionsLock.Unlock()
- // Note: from the tunnel server's perspective, this client connection
- // will close when session.delete calls Close() on the meekConn.
- server.clientHandler(clientSessionData.ClientTunnelProtocol, session.clientConn)
- return sessionID, session, underlyingConn, "", nil, nil
- }
- func (server *MeekServer) rateLimit(
- clientIP string, geoIPData GeoIPData, tunnelProtocol string) bool {
- historySize,
- thresholdSeconds,
- tunnelProtocols,
- regions,
- ISPs,
- ASNs,
- cities,
- GCTriggerCount, _, _ :=
- server.support.TrafficRulesSet.GetMeekRateLimiterConfig()
- if historySize == 0 {
- return false
- }
- if len(tunnelProtocols) > 0 {
- if !common.Contains(tunnelProtocols, tunnelProtocol) {
- return false
- }
- }
- if len(regions) > 0 || len(ISPs) > 0 || len(ASNs) > 0 || len(cities) > 0 {
- if len(regions) > 0 {
- if !common.Contains(regions, geoIPData.Country) {
- return false
- }
- }
- if len(ISPs) > 0 {
- if !common.Contains(ISPs, geoIPData.ISP) {
- return false
- }
- }
- if len(ASNs) > 0 {
- if !common.Contains(ASNs, geoIPData.ASN) {
- return false
- }
- }
- if len(cities) > 0 {
- if !common.Contains(cities, geoIPData.City) {
- return false
- }
- }
- }
- // With IPv6, individual users or sites are users commonly allocated a /64
- // or /56, so rate limit by /56.
- rateLimitIP := clientIP
- IP := net.ParseIP(clientIP)
- if IP != nil && IP.To4() == nil {
- rateLimitIP = IP.Mask(net.CIDRMask(56, 128)).String()
- }
- // go-cache-lru is safe for concurrent access, but lacks an atomic
- // compare-and-set type operations to check if an entry exists before
- // adding a new one. This mutex ensures the Get and Add are atomic
- // (as well as synchronizing access to rateLimitCount).
- server.rateLimitLock.Lock()
- var rateLimiter *ratelimit.Bucket
- entry, ok := server.rateLimitHistory.Get(rateLimitIP)
- if ok {
- rateLimiter = entry.(*ratelimit.Bucket)
- } else {
- rateLimiter = ratelimit.NewBucketWithQuantum(
- time.Duration(thresholdSeconds)*time.Second,
- int64(historySize),
- int64(historySize))
- server.rateLimitHistory.Set(
- rateLimitIP,
- rateLimiter,
- time.Duration(thresholdSeconds)*time.Second)
- }
- limit := rateLimiter.TakeAvailable(1) < 1
- triggerGC := false
- if limit {
- server.rateLimitCount += 1
- if server.rateLimitCount >= GCTriggerCount {
- triggerGC = true
- server.rateLimitCount = 0
- }
- }
- server.rateLimitLock.Unlock()
- if triggerGC {
- select {
- case server.rateLimitSignalGC <- struct{}{}:
- default:
- }
- }
- return limit
- }
- func (server *MeekServer) rateLimitWorker() {
- for {
- select {
- case <-server.rateLimitSignalGC:
- runtime.GC()
- case <-server.stopBroadcast:
- return
- }
- }
- }
- func (server *MeekServer) deleteSession(sessionID string) {
- // Don't obtain the server.sessionsLock write lock until modifying
- // server.sessions, as the session.delete can block for up to
- // MEEK_HTTP_CLIENT_IO_TIMEOUT. Allow new sessions to be added
- // concurrently.
- //
- // Since a lock isn't held for the duration, concurrent calls to
- // deleteSession with the same sessionID could happen; this is
- // not expected since only the reaper goroutine calls deleteExpiredSessions
- // (and in any case concurrent execution of the ok block is not an issue).
- server.sessionsLock.RLock()
- session, ok := server.sessions[sessionID]
- server.sessionsLock.RUnlock()
- if ok {
- session.delete(false)
- server.sessionsLock.Lock()
- delete(server.sessions, sessionID)
- server.sessionsLock.Unlock()
- }
- }
- func (server *MeekServer) deleteExpiredSessions() {
- // A deleteSession call may block for up to MEEK_HTTP_CLIENT_IO_TIMEOUT,
- // so grab a snapshot list of expired sessions and do not hold a lock for
- // the duration of deleteExpiredSessions. This allows new sessions to be
- // added concurrently.
- //
- // New sessions added after the snapshot is taken will be checked for
- // expiry on subsequent periodic calls to deleteExpiredSessions.
- //
- // To avoid long delays in releasing resources, individual deletes are
- // performed concurrently.
- server.sessionsLock.Lock()
- expiredSessionIDs := make([]string, 0)
- for sessionID, session := range server.sessions {
- if session.expired() {
- expiredSessionIDs = append(expiredSessionIDs, sessionID)
- }
- }
- server.sessionsLock.Unlock()
- start := time.Now()
- deleteWaitGroup := new(sync.WaitGroup)
- for _, sessionID := range expiredSessionIDs {
- deleteWaitGroup.Add(1)
- go func(sessionID string) {
- defer deleteWaitGroup.Done()
- server.deleteSession(sessionID)
- }(sessionID)
- }
- deleteWaitGroup.Wait()
- log.WithTraceFields(
- LogFields{"elapsed time": time.Since(start)}).Debug("deleted expired sessions")
- }
- // httpConnStateCallback tracks open persistent HTTP/HTTPS connections to the
- // meek server.
- func (server *MeekServer) httpConnStateCallback(conn net.Conn, connState http.ConnState) {
- switch connState {
- case http.StateNew:
- server.openConns.Add(conn)
- case http.StateHijacked, http.StateClosed:
- server.openConns.Remove(conn)
- }
- }
- // getMeekCookiePayload extracts the payload from a meek cookie. The cookie
- // payload is base64 encoded, obfuscated, and NaCl encrypted.
- func (server *MeekServer) getMeekCookiePayload(
- clientIP string, cookieValue string) ([]byte, error) {
- decodedValue, err := base64.StdEncoding.DecodeString(cookieValue)
- if err != nil {
- return nil, errors.Trace(err)
- }
- // The data consists of an obfuscated seed message prepended
- // to the obfuscated, encrypted payload. The server obfuscator
- // will read the seed message, leaving the remaining encrypted
- // data in the reader.
- reader := bytes.NewReader(decodedValue[:])
- obfuscator, err := obfuscator.NewServerObfuscator(
- &obfuscator.ObfuscatorConfig{
- Keyword: server.support.Config.MeekObfuscatedKey,
- SeedHistory: server.obfuscatorSeedHistory,
- IrregularLogger: func(clientIP string, err error, logFields common.LogFields) {
- logIrregularTunnel(
- server.support,
- server.listenerTunnelProtocol,
- server.listenerPort,
- clientIP,
- errors.Trace(err),
- LogFields(logFields))
- },
- },
- clientIP,
- reader)
- if err != nil {
- return nil, errors.Trace(err)
- }
- offset, err := reader.Seek(0, 1)
- if err != nil {
- return nil, errors.Trace(err)
- }
- encryptedPayload := decodedValue[offset:]
- obfuscator.ObfuscateClientToServer(encryptedPayload)
- var nonce [24]byte
- var privateKey, ephemeralPublicKey [32]byte
- decodedPrivateKey, err := base64.StdEncoding.DecodeString(
- server.support.Config.MeekCookieEncryptionPrivateKey)
- if err != nil {
- return nil, errors.Trace(err)
- }
- copy(privateKey[:], decodedPrivateKey)
- if len(encryptedPayload) < 32 {
- return nil, errors.TraceNew("unexpected encrypted payload size")
- }
- copy(ephemeralPublicKey[0:32], encryptedPayload[0:32])
- payload, ok := box.Open(nil, encryptedPayload[32:], &nonce, &ephemeralPublicKey, &privateKey)
- if !ok {
- return nil, errors.TraceNew("open box failed")
- }
- return payload, nil
- }
- // makeMeekTLSConfig creates a TLS config for a meek HTTPS listener.
- // Currently, this config is optimized for fronted meek where the nature
- // of the connection is non-circumvention; it's optimized for performance
- // assuming the peer is an uncensored CDN.
- func (server *MeekServer) makeMeekTLSConfig(
- isFronted bool, useObfuscatedSessionTickets bool) (*tris.Config, error) {
- certificate, privateKey, err := common.GenerateWebServerCertificate(values.GetHostName())
- if err != nil {
- return nil, errors.Trace(err)
- }
- tlsCertificate, err := tris.X509KeyPair(
- []byte(certificate), []byte(privateKey))
- if err != nil {
- return nil, errors.Trace(err)
- }
- // Vary the minimum version to frustrate scanning/fingerprinting of unfronted servers.
- // Limitation: like the certificate, this value changes on restart.
- minVersionCandidates := []uint16{tris.VersionTLS10, tris.VersionTLS11, tris.VersionTLS12}
- minVersion := minVersionCandidates[prng.Intn(len(minVersionCandidates))]
- config := &tris.Config{
- Certificates: []tris.Certificate{tlsCertificate},
- NextProtos: []string{"http/1.1"},
- MinVersion: minVersion,
- UseExtendedMasterSecret: true,
- }
- if isFronted {
- // This is a reordering of the supported CipherSuites in golang 1.6[*]. Non-ephemeral key
- // CipherSuites greatly reduce server load, and we try to select these since the meek
- // protocol is providing obfuscation, not privacy/integrity (this is provided by the
- // tunneled SSH), so we don't benefit from the perfect forward secrecy property provided
- // by ephemeral key CipherSuites.
- // https://github.com/golang/go/blob/1cb3044c9fcd88e1557eca1bf35845a4108bc1db/src/crypto/tls/cipher_suites.go#L75
- //
- // This optimization is applied only when there's a CDN in front of the meek server; in
- // unfronted cases we prefer a more natural TLS handshake.
- //
- // [*] the list has since been updated, removing CipherSuites using RC4 and 3DES.
- config.CipherSuites = []uint16{
- tris.TLS_RSA_WITH_AES_128_GCM_SHA256,
- tris.TLS_RSA_WITH_AES_256_GCM_SHA384,
- tris.TLS_RSA_WITH_AES_128_CBC_SHA,
- tris.TLS_RSA_WITH_AES_256_CBC_SHA,
- tris.TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256,
- tris.TLS_ECDHE_ECDSA_WITH_AES_128_GCM_SHA256,
- tris.TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384,
- tris.TLS_ECDHE_ECDSA_WITH_AES_256_GCM_SHA384,
- tris.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
- tris.TLS_ECDHE_ECDSA_WITH_AES_128_CBC_SHA,
- tris.TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA,
- tris.TLS_ECDHE_ECDSA_WITH_AES_256_CBC_SHA,
- }
- config.PreferServerCipherSuites = true
- }
- if useObfuscatedSessionTickets {
- // See obfuscated session ticket overview
- // in NewObfuscatedClientSessionCache.
- config.UseObfuscatedSessionTickets = true
- var obfuscatedSessionTicketKey [32]byte
- key, err := hex.DecodeString(server.support.Config.MeekObfuscatedKey)
- if err == nil && len(key) != 32 {
- err = std_errors.New("invalid obfuscated session key length")
- }
- if err != nil {
- return nil, errors.Trace(err)
- }
- copy(obfuscatedSessionTicketKey[:], key)
- var standardSessionTicketKey [32]byte
- _, err = rand.Read(standardSessionTicketKey[:])
- if err != nil {
- return nil, errors.Trace(err)
- }
- // Note: SessionTicketKey needs to be set, or else, it appears,
- // tris.Config.serverInit() will clobber the value set by
- // SetSessionTicketKeys.
- config.SessionTicketKey = obfuscatedSessionTicketKey
- config.SetSessionTicketKeys([][32]byte{
- standardSessionTicketKey,
- obfuscatedSessionTicketKey})
- }
- // When configured, initialize passthrough mode, an anti-probing defense.
- // Clients must prove knowledge of the obfuscated key via a message sent in
- // the TLS ClientHello random field.
- //
- // When clients fail to provide a valid message, the client connection is
- // relayed to the designated passthrough address, typically another web site.
- // The entire flow is relayed, including the original ClientHello, so the
- // client will perform a TLS handshake with the passthrough target.
- //
- // Irregular events are logged for invalid client activity.
- if server.passthroughAddress != "" {
- config.PassthroughAddress = server.passthroughAddress
- config.PassthroughVerifyMessage = func(
- message []byte) bool {
- return obfuscator.VerifyTLSPassthroughMessage(
- !server.support.Config.LegacyPassthrough,
- server.support.Config.MeekObfuscatedKey,
- message)
- }
- config.PassthroughLogInvalidMessage = func(
- clientIP string) {
- logIrregularTunnel(
- server.support,
- server.listenerTunnelProtocol,
- server.listenerPort,
- clientIP,
- errors.TraceNew("invalid passthrough message"),
- nil)
- }
- config.PassthroughHistoryAddNew = func(
- clientIP string,
- clientRandom []byte) bool {
- // Use a custom, shorter TTL based on the validity period of the
- // passthrough message.
- TTL := obfuscator.TLS_PASSTHROUGH_TIME_PERIOD
- if server.support.Config.LegacyPassthrough {
- TTL = obfuscator.HISTORY_SEED_TTL
- }
- // strictMode is true as, unlike with meek cookies, legitimate meek clients
- // never retry TLS connections using a previous random value.
- ok, logFields := server.obfuscatorSeedHistory.AddNewWithTTL(
- true,
- clientIP,
- "client-random",
- clientRandom,
- TTL)
- if logFields != nil {
- logIrregularTunnel(
- server.support,
- server.listenerTunnelProtocol,
- server.listenerPort,
- clientIP,
- errors.TraceNew("duplicate passthrough message"),
- LogFields(*logFields))
- }
- return ok
- }
- }
- return config, nil
- }
- type meekSession struct {
- // Note: 64-bit ints used with atomic operations are placed
- // at the start of struct to ensure 64-bit alignment.
- // (https://golang.org/pkg/sync/atomic/#pkg-note-BUG)
- lastActivity int64
- requestCount int64
- metricClientRetries int64
- metricPeakResponseSize int64
- metricPeakCachedResponseSize int64
- metricPeakCachedResponseHitSize int64
- metricCachedResponseMissPosition int64
- metricUnderlyingConnCount int64
- lock sync.Mutex
- deleted bool
- underlyingConn net.Conn
- clientConn *meekConn
- meekProtocolVersion int
- sessionIDSent bool
- cachedResponse *CachedResponse
- cookieName string
- contentType string
- }
- func (session *meekSession) touch() {
- atomic.StoreInt64(&session.lastActivity, int64(monotime.Now()))
- }
- func (session *meekSession) expired() bool {
- if session.clientConn == nil {
- // Not fully initialized. meekSession.clientConn will be set before adding
- // the session to MeekServer.sessions.
- return false
- }
- lastActivity := monotime.Time(atomic.LoadInt64(&session.lastActivity))
- return monotime.Since(lastActivity) >
- session.clientConn.meekServer.maxSessionStaleness
- }
- // delete releases all resources allocated by a session.
- func (session *meekSession) delete(haveLock bool) {
- // TODO: close the persistent HTTP client connection, if one exists?
- // This final call session.cachedResponse.Reset releases shared resources.
- //
- // This call requires exclusive access. session.lock is be obtained before
- // calling session.cachedResponse.Reset. Once the lock is obtained, no
- // request for this session is being processed concurrently, and pending
- // requests will block at session.lock.
- //
- // This logic assumes that no further session.cachedResponse access occurs,
- // or else resources may deplete (buffers won't be returned to the pool).
- // These requirements are achieved by obtaining the lock, setting
- // session.deleted, and any subsequent request handlers checking
- // session.deleted immediately after obtaining the lock.
- //
- // session.lock.Lock may block for up to MEEK_HTTP_CLIENT_IO_TIMEOUT,
- // the timeout for any active request handler processing a session
- // request.
- //
- // When the lock must be acquired, clientConn.Close is called first, to
- // interrupt any existing request handler blocking on pumpReads or pumpWrites.
- session.clientConn.Close()
- if !haveLock {
- session.lock.Lock()
- }
- // Release all extended buffers back to the pool.
- // session.cachedResponse.Reset is not safe for concurrent calls.
- session.cachedResponse.Reset()
- session.deleted = true
- if !haveLock {
- session.lock.Unlock()
- }
- }
- // GetMetrics implements the common.MetricsSource interface.
- func (session *meekSession) GetMetrics() common.LogFields {
- logFields := make(common.LogFields)
- logFields["meek_client_retries"] = atomic.LoadInt64(&session.metricClientRetries)
- logFields["meek_peak_response_size"] = atomic.LoadInt64(&session.metricPeakResponseSize)
- logFields["meek_peak_cached_response_size"] = atomic.LoadInt64(&session.metricPeakCachedResponseSize)
- logFields["meek_peak_cached_response_hit_size"] = atomic.LoadInt64(&session.metricPeakCachedResponseHitSize)
- logFields["meek_cached_response_miss_position"] = atomic.LoadInt64(&session.metricCachedResponseMissPosition)
- logFields["meek_underlying_connection_count"] = atomic.LoadInt64(&session.metricUnderlyingConnCount)
- logFields["meek_cookie_name"] = session.cookieName
- logFields["meek_content_type"] = session.contentType
- return logFields
- }
- // makeMeekSessionID creates a new session ID. The variable size is intended to
- // frustrate traffic analysis of both plaintext and TLS meek traffic.
- func makeMeekSessionID() (string, error) {
- size := MEEK_MIN_SESSION_ID_LENGTH +
- prng.Intn(MEEK_MAX_SESSION_ID_LENGTH-MEEK_MIN_SESSION_ID_LENGTH)
- sessionID, err := common.MakeSecureRandomBytes(size)
- if err != nil {
- return "", errors.Trace(err)
- }
- // Omit padding to maximize variable size space. To the client, the session
- // ID is an opaque string cookie value.
- return base64.RawStdEncoding.EncodeToString(sessionID), nil
- }
- // meekConn implements the net.Conn interface and is to be used as a client
- // connection by the tunnel server (being passed to sshServer.handleClient).
- // meekConn bridges net/http request/response payload readers and writers
- // and goroutines calling Read()s and Write()s.
- type meekConn struct {
- meekServer *MeekServer
- meekSession *meekSession
- firstUnderlyingConn net.Conn
- remoteAddr net.Addr
- protocolVersion int
- closeBroadcast chan struct{}
- closed int32
- lastReadChecksum *uint64
- readLock sync.Mutex
- emptyReadBuffer chan *bytes.Buffer
- partialReadBuffer chan *bytes.Buffer
- fullReadBuffer chan *bytes.Buffer
- writeLock sync.Mutex
- nextWriteBuffer chan []byte
- writeResult chan error
- }
- func newMeekConn(
- meekServer *MeekServer,
- meekSession *meekSession,
- underlyingConn net.Conn,
- remoteAddr net.Addr,
- protocolVersion int) *meekConn {
- // In order to inspect its properties, meekConn will hold a reference to
- // firstUnderlyingConn, the _first_ underlying TCP conn, for the full
- // lifetime of meekConn, which may exceed the lifetime of firstUnderlyingConn
- // and include subsequent underlying TCP conns. In this case, it is expected
- // that firstUnderlyingConn will be closed by "net/http", so no OS resources
- // (e.g., a socket) are retained longer than necessary.
- conn := &meekConn{
- meekServer: meekServer,
- meekSession: meekSession,
- firstUnderlyingConn: underlyingConn,
- remoteAddr: remoteAddr,
- protocolVersion: protocolVersion,
- closeBroadcast: make(chan struct{}),
- closed: 0,
- emptyReadBuffer: make(chan *bytes.Buffer, 1),
- partialReadBuffer: make(chan *bytes.Buffer, 1),
- fullReadBuffer: make(chan *bytes.Buffer, 1),
- nextWriteBuffer: make(chan []byte, 1),
- writeResult: make(chan error, 1),
- }
- // Read() calls and pumpReads() are synchronized by exchanging control
- // of a single readBuffer. This is the same scheme used in and described
- // in psiphon.MeekConn.
- conn.emptyReadBuffer <- new(bytes.Buffer)
- return conn
- }
- // GetMetrics implements the common.MetricsSource interface. The metrics are
- // maintained in the meek session type; but logTunnel, which calls
- // MetricsSource.GetMetrics, has a pointer only to this conn, so it calls
- // through to the session.
- func (conn *meekConn) GetMetrics() common.LogFields {
- logFields := conn.meekSession.GetMetrics()
- if conn.meekServer.passthroughAddress != "" {
- logFields["passthrough_address"] = conn.meekServer.passthroughAddress
- }
- // Include metrics, such as fragmentor metrics, from the _first_ underlying
- // TCP conn. Properties of subsequent underlying TCP conns are not reflected
- // in these metrics; we assume that the first TCP conn, which most likely
- // transits the various protocol handshakes, is most significant.
- underlyingMetrics, ok := conn.firstUnderlyingConn.(common.MetricsSource)
- if ok {
- logFields.Add(underlyingMetrics.GetMetrics())
- }
- return logFields
- }
- // GetUnderlyingTCPAddrs implements the common.UnderlyingTCPAddrSource
- // interface, returning the TCP addresses for the _first_ underlying TCP
- // connection in the meek tunnel.
- func (conn *meekConn) GetUnderlyingTCPAddrs() (*net.TCPAddr, *net.TCPAddr, bool) {
- localAddr, ok := conn.firstUnderlyingConn.LocalAddr().(*net.TCPAddr)
- if !ok {
- return nil, nil, false
- }
- remoteAddr, ok := conn.firstUnderlyingConn.RemoteAddr().(*net.TCPAddr)
- if !ok {
- return nil, nil, false
- }
- return localAddr, remoteAddr, true
- }
- // SetReplay implements the common.FragmentorReplayAccessor interface, applying
- // the inputs to the _first_ underlying TCP connection in the meek tunnel. If
- // the underlying connection is closed, the SetSeed call will have no effect.
- func (conn *meekConn) SetReplay(PRNG *prng.PRNG) {
- fragmentor, ok := conn.firstUnderlyingConn.(common.FragmentorReplayAccessor)
- if ok {
- fragmentor.SetReplay(PRNG)
- }
- }
- // GetReplay implements the FragmentorReplayAccessor interface, getting the
- // outputs from the _first_ underlying TCP connection in the meek tunnel.
- //
- // We assume that the first TCP conn is most significant: the initial TCP
- // connection most likely fragments protocol handshakes; and, in the case the
- // packet manipulation, any selected packet manipulation spec would have been
- // successful.
- func (conn *meekConn) GetReplay() (*prng.Seed, bool) {
- fragmentor, ok := conn.firstUnderlyingConn.(common.FragmentorReplayAccessor)
- if ok {
- return fragmentor.GetReplay()
- }
- return nil, false
- }
- // pumpReads causes goroutines blocking on meekConn.Read() to read
- // from the specified reader. This function blocks until the reader
- // is fully consumed or the meekConn is closed. A read buffer allows
- // up to MEEK_MAX_REQUEST_PAYLOAD_LENGTH bytes to be read and buffered
- // without a Read() immediately consuming the bytes, but there's still
- // a possibility of a stall if no Read() calls are made after this
- // read buffer is full.
- // Returns the number of request bytes read.
- // Note: assumes only one concurrent call to pumpReads
- func (conn *meekConn) pumpReads(reader io.Reader) (int64, error) {
- // Use either an empty or partial buffer. By using a partial
- // buffer, pumpReads will not block if the Read() caller has
- // not fully drained the read buffer.
- var readBuffer *bytes.Buffer
- select {
- case readBuffer = <-conn.emptyReadBuffer:
- case readBuffer = <-conn.partialReadBuffer:
- case <-conn.closeBroadcast:
- return 0, io.EOF
- }
- newDataOffset := readBuffer.Len()
- // Since we need to read the full request payload in order to
- // take its checksum before relaying it, the read buffer can
- // grow to up to 2 x MEEK_MAX_REQUEST_PAYLOAD_LENGTH + 1.
- // +1 allows for an explicit check for request payloads that
- // exceed the maximum permitted length.
- limitReader := io.LimitReader(reader, MEEK_MAX_REQUEST_PAYLOAD_LENGTH+1)
- n, err := readBuffer.ReadFrom(limitReader)
- if err == nil && n == MEEK_MAX_REQUEST_PAYLOAD_LENGTH+1 {
- err = std_errors.New("invalid request payload length")
- }
- // If the request read fails, don't relay the new data. This allows
- // the client to retry and resend its request payload without
- // interrupting/duplicating the payload flow.
- if err != nil {
- readBuffer.Truncate(newDataOffset)
- conn.replaceReadBuffer(readBuffer)
- return 0, errors.Trace(err)
- }
- // Check if request payload checksum matches immediately
- // previous payload. On match, assume this is a client retry
- // sending payload that was already relayed and skip this
- // payload. Payload is OSSH ciphertext and almost surely
- // will not repeat. In the highly unlikely case that it does,
- // the underlying SSH connection will fail and the client
- // must reconnect.
- checksum := crc64.Checksum(
- readBuffer.Bytes()[newDataOffset:], conn.meekServer.checksumTable)
- if conn.lastReadChecksum == nil {
- conn.lastReadChecksum = new(uint64)
- } else if *conn.lastReadChecksum == checksum {
- readBuffer.Truncate(newDataOffset)
- }
- *conn.lastReadChecksum = checksum
- conn.replaceReadBuffer(readBuffer)
- return n, nil
- }
- var errMeekConnectionHasClosed = std_errors.New("meek connection has closed")
- // Read reads from the meekConn into buffer. Read blocks until
- // some data is read or the meekConn closes. Under the hood, it
- // waits for pumpReads to submit a reader to read from.
- // Note: lock is to conform with net.Conn concurrency semantics
- func (conn *meekConn) Read(buffer []byte) (int, error) {
- conn.readLock.Lock()
- defer conn.readLock.Unlock()
- var readBuffer *bytes.Buffer
- select {
- case readBuffer = <-conn.partialReadBuffer:
- case readBuffer = <-conn.fullReadBuffer:
- case <-conn.closeBroadcast:
- return 0, errors.Trace(errMeekConnectionHasClosed)
- }
- n, err := readBuffer.Read(buffer)
- conn.replaceReadBuffer(readBuffer)
- return n, err
- }
- func (conn *meekConn) replaceReadBuffer(readBuffer *bytes.Buffer) {
- length := readBuffer.Len()
- if length >= MEEK_MAX_REQUEST_PAYLOAD_LENGTH {
- conn.fullReadBuffer <- readBuffer
- } else if length == 0 {
- conn.emptyReadBuffer <- readBuffer
- } else {
- conn.partialReadBuffer <- readBuffer
- }
- }
- // pumpWrites causes goroutines blocking on meekConn.Write() to write
- // to the specified writer. This function blocks until the meek response
- // body limits (size for protocol v1, turn around time for protocol v2+)
- // are met, or the meekConn is closed.
- //
- // Note: channel scheme assumes only one concurrent call to pumpWrites
- func (conn *meekConn) pumpWrites(
- writer io.Writer, skipExtendedTurnAround bool) (int, error) {
- startTime := time.Now()
- timeout := time.NewTimer(conn.meekServer.turnAroundTimeout)
- defer timeout.Stop()
- n := 0
- for {
- select {
- case buffer := <-conn.nextWriteBuffer:
- written, err := writer.Write(buffer)
- n += written
- // Assumes that writeResult won't block.
- // Note: always send the err to writeResult,
- // as the Write() caller is blocking on this.
- conn.writeResult <- err
- if err != nil {
- return n, err
- }
- if conn.protocolVersion < MEEK_PROTOCOL_VERSION_1 {
- // Pre-protocol version 1 clients expect at most
- // MEEK_MAX_REQUEST_PAYLOAD_LENGTH response bodies
- return n, nil
- }
- if skipExtendedTurnAround {
- // When fast turn around is indicated, skip the extended turn
- // around timeout. This optimizes for upstream flows.
- return n, nil
- }
- totalElapsedTime := time.Since(startTime) / time.Millisecond
- if totalElapsedTime >= conn.meekServer.extendedTurnAroundTimeout {
- return n, nil
- }
- timeout.Reset(conn.meekServer.turnAroundTimeout)
- case <-timeout.C:
- return n, nil
- case <-conn.closeBroadcast:
- return n, errors.Trace(errMeekConnectionHasClosed)
- }
- }
- }
- // Write writes the buffer to the meekConn. It blocks until the
- // entire buffer is written to or the meekConn closes. Under the
- // hood, it waits for sufficient pumpWrites calls to consume the
- // write buffer.
- // Note: lock is to conform with net.Conn concurrency semantics
- func (conn *meekConn) Write(buffer []byte) (int, error) {
- conn.writeLock.Lock()
- defer conn.writeLock.Unlock()
- // TODO: may be more efficient to send whole buffer
- // and have pumpWrites stash partial buffer when can't
- // send it all.
- n := 0
- for n < len(buffer) {
- end := n + MEEK_MAX_REQUEST_PAYLOAD_LENGTH
- if end > len(buffer) {
- end = len(buffer)
- }
- // Only write MEEK_MAX_REQUEST_PAYLOAD_LENGTH at a time,
- // to ensure compatibility with v1 protocol.
- chunk := buffer[n:end]
- select {
- case conn.nextWriteBuffer <- chunk:
- case <-conn.closeBroadcast:
- return n, errors.Trace(errMeekConnectionHasClosed)
- }
- // Wait for the buffer to be processed.
- select {
- case <-conn.writeResult:
- // The err from conn.writeResult comes from the
- // io.MultiWriter used in pumpWrites, which writes
- // to both the cached response and the HTTP response.
- //
- // Don't stop on error here, since only writing
- // to the HTTP response will fail, and the client
- // may retry and use the cached response.
- //
- // It's possible that the cached response buffer
- // is too small for the client to successfully
- // retry, but that cannot be determined. In this
- // case, the meek connection will eventually fail.
- //
- // err is already logged in ServeHTTP.
- case <-conn.closeBroadcast:
- return n, errors.Trace(errMeekConnectionHasClosed)
- }
- n += len(chunk)
- }
- return n, nil
- }
- // Close closes the meekConn. This will interrupt any blocked
- // Read, Write, pumpReads, and pumpWrites.
- func (conn *meekConn) Close() error {
- if atomic.CompareAndSwapInt32(&conn.closed, 0, 1) {
- close(conn.closeBroadcast)
- // In general, we reply on "net/http" to close underlying TCP conns. In this
- // case, we can directly close the first once, if it's still open.
- conn.firstUnderlyingConn.Close()
- }
- return nil
- }
- // Stub implementation of net.Conn.LocalAddr
- func (conn *meekConn) LocalAddr() net.Addr {
- return nil
- }
- // RemoteAddr returns the remoteAddr specified in newMeekConn. This
- // acts as a proxy for the actual remote address, which is either a
- // direct HTTP/HTTPS connection remote address, or in the case of
- // downstream proxy of CDN fronts, some other value determined via
- // HTTP headers.
- func (conn *meekConn) RemoteAddr() net.Addr {
- return conn.remoteAddr
- }
- // SetDeadline is not a true implementation of net.Conn.SetDeadline. It
- // merely checks that the requested timeout exceeds the MEEK_MAX_SESSION_STALENESS
- // period. When it does, and the session is idle, the meekConn Read/Write will
- // be interrupted and return an error (not a timeout error) before the deadline.
- // In other words, this conn will approximate the desired functionality of
- // timing out on idle on or before the requested deadline.
- func (conn *meekConn) SetDeadline(t time.Time) error {
- // Overhead: nanoseconds (https://blog.cloudflare.com/its-go-time-on-linux/)
- if time.Now().Add(conn.meekServer.maxSessionStaleness).Before(t) {
- return nil
- }
- return errors.TraceNew("not supported")
- }
- // Stub implementation of net.Conn.SetReadDeadline
- func (conn *meekConn) SetReadDeadline(t time.Time) error {
- return errors.TraceNew("not supported")
- }
- // Stub implementation of net.Conn.SetWriteDeadline
- func (conn *meekConn) SetWriteDeadline(t time.Time) error {
- return errors.TraceNew("not supported")
- }
|