original change

net/socket_stream/socket_stream_job.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_SOCKET_STREAM_SOCKET_STREAM_JOB_H_
 #define NET_SOCKET_STREAM_SOCKET_STREAM_JOB_H_
 
 #include <string>
 
 #include "base/memory/ref_counted.h"
 #include "net/base/net_export.h"
 #include "net/socket_stream/socket_stream.h"
 
 class GURL;
 
 namespace net {
 
+class CookieStore;
 class SSLConfigService;
 class SSLInfo;
 class TransportSecurityState;
 
 // SocketStreamJob represents full-duplex communication over SocketStream.
 // If a protocol (e.g. WebSocket protocol) needs to inspect/modify data
 // over SocketStream, you can implement protocol specific job (e.g.
 // WebSocketJob) to do some work on data over SocketStream.
 // Registers the protocol specific SocketStreamJob by RegisterProtocolFactory
 // and call CreateSocketStreamJob to create SocketStreamJob for the URL.
 class NET_EXPORT SocketStreamJob
     : public base::RefCountedThreadSafe<SocketStreamJob> {
  public:
   // Callback function implemented by protocol handlers to create new jobs.
   typedef SocketStreamJob* (ProtocolFactory)(const GURL& url,
-                                             SocketStream::Delegate* delegate);
+                                             SocketStream::Delegate* delegate,
+                                             URLRequestContext* context,
+                                             CookieStore* cookie_store);
 
   static ProtocolFactory* RegisterProtocolFactory(const std::string& scheme,
                                                   ProtocolFactory* factory);
 
   static SocketStreamJob* CreateSocketStreamJob(
       const GURL& url,
       SocketStream::Delegate* delegate,
       TransportSecurityState* sts,
-      SSLConfigService* ssl);
+      SSLConfigService* ssl,
+      URLRequestContext* context,
+      CookieStore* cookie_store);
 
   SocketStreamJob();
   void InitSocketStream(SocketStream* socket) {
     socket_ = socket;
   }
 
   virtual SocketStream::UserData* GetUserData(const void* key) const;
   virtual void SetUserData(const void* key, SocketStream::UserData* data);
 
   URLRequestContext* context() const {
     return socket_.get() ? socket_->context() : 0;
   }
-  void set_context(URLRequestContext* context) {
-    if (socket_.get())
-      socket_->set_context(context);
+  CookieStore* cookie_store() const {
+    return socket_.get() ? socket_->cookie_store() : 0;
   }
 
   virtual void Connect();
 
   virtual bool SendData(const char* data, int len);
 
   virtual void Close();
 
   virtual void RestartWithAuth(const AuthCredentials& credentials);
 
   virtual void CancelWithError(int error);
 
   virtual void CancelWithSSLError(const net::SSLInfo& ssl_info);
 
   virtual void ContinueDespiteError();
 
   virtual void DetachDelegate();
 
+  virtual void DetachContext();
+
  protected:
   friend class WebSocketJobTest;
   friend class base::RefCountedThreadSafe<SocketStreamJob>;
   virtual ~SocketStreamJob();
 
   scoped_refptr<SocketStream> socket_;
 
   DISALLOW_COPY_AND_ASSIGN(SocketStreamJob);
 };
 
 }  // namespace net
 
 #endif  // NET_SOCKET_STREAM_SOCKET_STREAM_JOB_H_