forked from mirrors/gecko-dev
		
	 df3233b549
			
		
	
	
		df3233b549
		
	
	
	
	
		
			
			MozReview-Commit-ID: 5y0Aj6Bgk9u --HG-- extra : rebase_source : 015352a4b0a87734afce856fcd7b424e96c6ede8
		
			
				
	
	
		
			143 lines
		
	
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			143 lines
		
	
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 | |
| /* vim: set sw=2 ts=8 et tw=80 : */
 | |
| /* This Source Code Form is subject to the terms of the Mozilla Public
 | |
|  * License, v. 2.0. If a copy of the MPL was not distributed with this
 | |
|  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 | |
| 
 | |
| // HttpLog.h should generally be included first
 | |
| #include "HttpLog.h"
 | |
| 
 | |
| /*
 | |
|   Currently supported is h2
 | |
| */
 | |
| 
 | |
| #include "nsHttp.h"
 | |
| #include "nsHttpHandler.h"
 | |
| 
 | |
| #include "ASpdySession.h"
 | |
| #include "PSpdyPush.h"
 | |
| #include "Http2Push.h"
 | |
| #include "Http2Session.h"
 | |
| 
 | |
| #include "mozilla/Telemetry.h"
 | |
| 
 | |
| namespace mozilla {
 | |
| namespace net {
 | |
| 
 | |
| ASpdySession::ASpdySession()
 | |
| {
 | |
| }
 | |
| 
 | |
| ASpdySession::~ASpdySession() = default;
 | |
| 
 | |
| ASpdySession *
 | |
| ASpdySession::NewSpdySession(uint32_t version,
 | |
|                              nsISocketTransport *aTransport,
 | |
|                              bool attemptingEarlyData)
 | |
| {
 | |
|   // This is a necko only interface, so we can enforce version
 | |
|   // requests as a precondition
 | |
|   MOZ_ASSERT(version == HTTP_VERSION_2,
 | |
|              "Unsupported spdy version");
 | |
| 
 | |
|   // Don't do a runtime check of IsSpdyV?Enabled() here because pref value
 | |
|   // may have changed since starting negotiation. The selected protocol comes
 | |
|   // from a list provided in the SERVER HELLO filtered by our acceptable
 | |
|   // versions, so there is no risk of the server ignoring our prefs.
 | |
| 
 | |
|   Telemetry::Accumulate(Telemetry::SPDY_VERSION2, version);
 | |
| 
 | |
|   return new Http2Session(aTransport, version, attemptingEarlyData);
 | |
| }
 | |
| 
 | |
| SpdyInformation::SpdyInformation()
 | |
| {
 | |
|   // highest index of enabled protocols is the
 | |
|   // most preferred for ALPN negotiaton
 | |
|   Version[0] = HTTP_VERSION_2;
 | |
|   VersionString[0] = NS_LITERAL_CSTRING("h2");
 | |
|   ALPNCallbacks[0] = Http2Session::ALPNCallback;
 | |
| }
 | |
| 
 | |
| bool
 | |
| SpdyInformation::ProtocolEnabled(uint32_t index) const
 | |
| {
 | |
|   MOZ_ASSERT(index < kCount, "index out of range");
 | |
| 
 | |
|   return gHttpHandler->IsHttp2Enabled();
 | |
| }
 | |
| 
 | |
| nsresult
 | |
| SpdyInformation::GetNPNIndex(const nsACString &npnString,
 | |
|                              uint32_t *result) const
 | |
| {
 | |
|   if (npnString.IsEmpty())
 | |
|     return NS_ERROR_FAILURE;
 | |
| 
 | |
|   for (uint32_t index = 0; index < kCount; ++index) {
 | |
|     if (npnString.Equals(VersionString[index])) {
 | |
|       *result = index;
 | |
|       return NS_OK;
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   return NS_ERROR_FAILURE;
 | |
| }
 | |
| 
 | |
| //////////////////////////////////////////
 | |
| // SpdyPushCache
 | |
| //////////////////////////////////////////
 | |
| 
 | |
| SpdyPushCache::SpdyPushCache()
 | |
| {
 | |
| }
 | |
| 
 | |
| SpdyPushCache::~SpdyPushCache()
 | |
| {
 | |
|   mHashHttp2.Clear();
 | |
| }
 | |
| 
 | |
| bool
 | |
| SpdyPushCache::RegisterPushedStreamHttp2(const nsCString& key,
 | |
|                                          Http2PushedStream *stream)
 | |
| {
 | |
|   LOG3(("SpdyPushCache::RegisterPushedStreamHttp2 %s 0x%X\n",
 | |
|         key.get(), stream->StreamID()));
 | |
|   if(mHashHttp2.Get(key)) {
 | |
|     LOG3(("SpdyPushCache::RegisterPushedStreamHttp2 %s 0x%X duplicate key\n",
 | |
|           key.get(), stream->StreamID()));
 | |
|     return false;
 | |
|   }
 | |
|   mHashHttp2.Put(key, stream);
 | |
|   return true;
 | |
| }
 | |
| 
 | |
| Http2PushedStream *
 | |
| SpdyPushCache::RemovePushedStreamHttp2(const nsCString& key)
 | |
| {
 | |
|   Http2PushedStream *rv = mHashHttp2.Get(key);
 | |
|   LOG3(("SpdyPushCache::RemovePushedStreamHttp2 %s 0x%X\n",
 | |
|         key.get(), rv ? rv->StreamID() : 0));
 | |
|   if (rv)
 | |
|     mHashHttp2.Remove(key);
 | |
|   return rv;
 | |
| }
 | |
| 
 | |
| Http2PushedStream *
 | |
| SpdyPushCache::RemovePushedStreamHttp2ByID(const nsCString& key, const uint32_t& streamID)
 | |
| {
 | |
|   Http2PushedStream *rv = mHashHttp2.Get(key);
 | |
|   LOG3(("SpdyPushCache::RemovePushedStreamHttp2ByID %s 0x%X 0x%X",
 | |
|         key.get(), rv ? rv->StreamID() : 0, streamID));
 | |
|   if (rv && streamID == rv->StreamID()) {
 | |
|     mHashHttp2.Remove(key);
 | |
|   } else {
 | |
|     // Ensure we overwrite our rv with null in case the stream IDs don't match
 | |
|     rv = nullptr;
 | |
|   }
 | |
|   return rv;
 | |
| }
 | |
| 
 | |
| } // namespace net
 | |
| } // namespace mozilla
 | |
| 
 |