content/child/websocket_dispatcher.cc

// Copyright 2013 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.

#include "content/child/websocket_dispatcher.h"

#include <stdint.h>
#include <map>

#include "base/logging.h"
#include "content/child/websocket_bridge.h"
#include "content/common/websocket_messages.h"
#include "ipc/ipc_message.h"
#include "url/gurl.h"

namespace content {

WebSocketDispatcher::WebSocketDispatcher()
  : channel_id_max_(0),
    weak_ptr_factory_(this) {}

WebSocketDispatcher::~WebSocketDispatcher() {}

bool WebSocketDispatcher::CanHandleMessage(const IPC::Message& msg) {
  switch (msg.type()) {
    case WebSocketMsg_AddChannelResponse::ID:
    case WebSocketMsg_NotifyStartOpeningHandshake::ID:
    case WebSocketMsg_NotifyFinishOpeningHandshake::ID:
    case WebSocketMsg_NotifyFailure::ID:
    case WebSocketMsg_SendFrame::ID:
    case WebSocketMsg_FlowControl::ID:
    case WebSocketMsg_DropChannel::ID:
    case WebSocketMsg_NotifyClosing::ID:
      return true;
    default:
      return false;
  }
}

int WebSocketDispatcher::AddBridge(WebSocketBridge* bridge) {
  ++channel_id_max_;
  bridges_.insert(std::make_pair(channel_id_max_, bridge));
  return channel_id_max_;
}

void WebSocketDispatcher::RemoveBridge(int channel_id) {
  std::map<int, WebSocketBridge*>::iterator iter = bridges_.find(channel_id);
  if (iter == bridges_.end()) {
    DVLOG(1) << "Remove a non-existent bridge(" << channel_id << ")";
    return;
  }
  bridges_.erase(iter);
}

bool WebSocketDispatcher::OnMessageReceived(const IPC::Message& msg) {
  if (!CanHandleMessage(msg))
    return false;
  WebSocketBridge* bridge = GetBridge(msg.routing_id(), msg.type());
  if (!bridge)
    return true;
  return bridge->OnMessageReceived(msg);
}

WebSocketBridge* WebSocketDispatcher::GetBridge(int channel_id, uint32_t type) {
  std::map<int, WebSocketBridge*>::iterator iter = bridges_.find(channel_id);
  if (iter == bridges_.end()) {
    DVLOG(1) << "No bridge for channel_id=" << channel_id
             << ", type=" << type;
    return NULL;
  }
  return iter->second;
}

}  // namespace content