net/socket_stream/socket_stream_job.h

// Copyright (c) 2010 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/ref_counted.h"
#include "net/socket_stream/socket_stream.h"

class GURL;

namespace net {

// 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 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);

  static ProtocolFactory* RegisterProtocolFactory(const std::string& scheme,
                                                  ProtocolFactory* factory);

  static SocketStreamJob* CreateSocketStreamJob(
      const GURL& url, SocketStream::Delegate* delegate);

  SocketStreamJob() {}
  void InitSocketStream(SocketStream* socket) {
    socket_ = socket;
  }

  virtual SocketStream::UserData *GetUserData(const void* key) const {
    return socket_->GetUserData(key);
  }
  virtual void SetUserData(const void* key, SocketStream::UserData* data) {
    socket_->SetUserData(key, data);
  }

  URLRequestContext* context() const {
    return socket_->context();
  }
  void set_context(URLRequestContext* context) {
    socket_->set_context(context);
  }

  virtual void Connect() {
    socket_->Connect();
  }

  virtual bool SendData(const char* data, int len) {
    return socket_->SendData(data, len);
  }

  virtual void Close() {
    socket_->Close();
  }

  virtual void RestartWithAuth(
      const std::wstring& username,
      const std::wstring& password) {
    socket_->RestartWithAuth(username, password);
  }

  virtual void DetachDelegate() {
    socket_->DetachDelegate();
  }

 protected:
  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_