blob: 44a3ff55a8d18ce6a241fbbfd7345055382f1640 [file] [log] [blame]
// Copyright 2014 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 "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/observer_list.h"
#include "components/proximity_auth/remote_device.h"
namespace proximity_auth {
class ConnectionObserver;
class WireMessage;
// Base class representing a connection with a remote device, which is a
// persistent bidirectional channel for sending and receiving wire messages.
class Connection {
enum Status {
// Constructs a connection to the given |remote_device|.
explicit Connection(const RemoteDevice& remote_device);
virtual ~Connection();
// Returns true iff the connection's status is CONNECTED.
bool IsConnected() const;
// Returns true iff the connection is currently sending a message.
bool is_sending_message() const { return is_sending_message_; }
// Sends a message to the remote device.
// |OnSendCompleted()| will be called for all observers upon completion with
// either success or failure.
void SendMessage(scoped_ptr<WireMessage> message);
void AddObserver(ConnectionObserver* observer);
void RemoveObserver(ConnectionObserver* observer);
const RemoteDevice& remote_device() const { return remote_device_; }
// Abstract methods that subclasses should implement:
// Attempts to connect to the remote device if not already connected.
virtual void Connect() = 0;
// Disconnects from the remote device.
virtual void Disconnect() = 0;
// Sets the connection's status to |status|. If this is different from the
// previous status, notifies observers of the change in status.
// Virtual for testing.
virtual void SetStatus(Status status);
Status status() const { return status_; }
// Called after attempting to send bytes over the connection, whether the
// message was successfully sent or not.
// Virtual for testing.
virtual void OnDidSendMessage(const WireMessage& message, bool success);
// Called when bytes are read from the connection. There should not be a send
// in progress when this function is called.
// Virtual for testing.
virtual void OnBytesReceived(const std::string& bytes);
// Sends bytes over the connection. The implementing class should call
// OnSendCompleted() once the send succeeds or fails. At most one send will be
// in progress.
virtual void SendMessageImpl(scoped_ptr<WireMessage> message) = 0;
// Deserializes the |recieved_bytes_| and returns the resulting WireMessage,
// or NULL if the message is malformed. Sets |is_incomplete_message| to true
// if the |serialized_message| does not have enough data to parse the header,
// or if the message length encoded in the message header exceeds the size of
// the |serialized_message|. Exposed for testing.
virtual scoped_ptr<WireMessage> DeserializeWireMessage(
bool* is_incomplete_message);
// The remote device corresponding to this connection.
const RemoteDevice remote_device_;
// The current status of the connection.
Status status_;
// The registered observers of the connection.
ObserverList<ConnectionObserver> observers_;
// A temporary buffer storing bytes received before a received message can be
// fully constructed.
std::string received_bytes_;
// Whether a message is currently in the process of being sent.
bool is_sending_message_;
} // namespace proximity_auth