gecko-dev/media/mtransport/transportlayerloopback.h

141 строка
3.0 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=2 et sw=2 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/. */
// Original author: ekr@rtfm.com
#ifndef transportlayerloopback_h__
#define transportlayerloopback_h__
#include "nspr.h"
#include "prio.h"
#include "prlock.h"
#include <memory>
#include <queue>
#include "nsAutoPtr.h"
#include "nsCOMPtr.h"
#include "nsITimer.h"
#include "m_cpp_utils.h"
#include "transportflow.h"
#include "transportlayer.h"
// A simple loopback transport layer that is used for testing.
namespace mozilla {
class TransportLayerLoopback : public TransportLayer {
public:
TransportLayerLoopback() :
peer_(nullptr),
timer_(nullptr),
target_(nullptr),
packets_(),
packets_lock_(nullptr),
deliverer_(nullptr) {}
~TransportLayerLoopback() {
while (!packets_.empty()) {
QueuedPacket *packet = packets_.front();
packets_.pop();
delete packet;
}
if (packets_lock_) {
PR_DestroyLock(packets_lock_);
}
timer_->Cancel();
deliverer_->Detach();
}
// Init
nsresult Init();
// Connect to the other side
void Connect(TransportLayerLoopback* peer);
// Disconnect
void Disconnect() {
TransportLayerLoopback *peer = peer_;
peer_ = nullptr;
if (peer) {
peer->Disconnect();
}
}
// Overrides for TransportLayer
virtual TransportResult SendPacket(const unsigned char *data, size_t len);
// Deliver queued packets
void DeliverPackets();
TRANSPORT_LAYER_ID("loopback")
private:
DISALLOW_COPY_ASSIGN(TransportLayerLoopback);
// A queued packet
class QueuedPacket {
public:
QueuedPacket() : data_(nullptr), len_(0) {}
~QueuedPacket() {
delete [] data_;
}
void Assign(const unsigned char *data, size_t len) {
data_ = new unsigned char[len];
memcpy(static_cast<void *>(data_),
static_cast<const void *>(data), len);
len_ = len;
}
const unsigned char *data() const { return data_; }
size_t len() const { return len_; }
private:
DISALLOW_COPY_ASSIGN(QueuedPacket);
unsigned char *data_;
size_t len_;
};
// A timer to deliver packets if some are available
// Fires every 100 ms
class Deliverer : public nsITimerCallback {
public:
Deliverer(TransportLayerLoopback *layer) :
layer_(layer) {}
virtual ~Deliverer() {
}
void Detach() {
layer_ = nullptr;
}
NS_DECL_ISUPPORTS
NS_DECL_NSITIMERCALLBACK
private:
DISALLOW_COPY_ASSIGN(Deliverer);
TransportLayerLoopback *layer_;
};
// Queue a packet for delivery
nsresult QueuePacket(const unsigned char *data, size_t len);
TransportLayerLoopback* peer_;
nsCOMPtr<nsITimer> timer_;
nsCOMPtr<nsIEventTarget> target_;
std::queue<QueuedPacket *> packets_;
PRLock *packets_lock_;
nsRefPtr<Deliverer> deliverer_;
};
} // close namespace
#endif