From 5f8de423f190bbb79a62f804151bc24824fa32d8 Mon Sep 17 00:00:00 2001
From: "Matt A. Tobin" <mattatobin@localhost.localdomain>
Date: Fri, 2 Feb 2018 04:16:08 -0500
Subject: Add m-esr52 at 52.6.0

---
 security/nss/gtests/ssl_gtest/test_io.h | 152 ++++++++++++++++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 security/nss/gtests/ssl_gtest/test_io.h

(limited to 'security/nss/gtests/ssl_gtest/test_io.h')

diff --git a/security/nss/gtests/ssl_gtest/test_io.h b/security/nss/gtests/ssl_gtest/test_io.h
new file mode 100644
index 000000000..b78db0dc6
--- /dev/null
+++ b/security/nss/gtests/ssl_gtest/test_io.h
@@ -0,0 +1,152 @@
+/* -*- 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/. */
+
+#ifndef test_io_h_
+#define test_io_h_
+
+#include <string.h>
+#include <map>
+#include <memory>
+#include <ostream>
+#include <queue>
+#include <string>
+
+#include "prio.h"
+
+namespace nss_test {
+
+class DataBuffer;
+class Packet;
+class DummyPrSocket;  // Fwd decl.
+
+// Allow us to inspect a packet before it is written.
+class PacketFilter {
+ public:
+  enum Action {
+    KEEP,    // keep the original packet unmodified
+    CHANGE,  // change the packet to a different value
+    DROP     // drop the packet
+  };
+
+  virtual ~PacketFilter() {}
+
+  // The packet filter takes input and has the option of mutating it.
+  //
+  // A filter that modifies the data places the modified data in *output and
+  // returns CHANGE.  A filter that does not modify data returns LEAVE, in which
+  // case the value in *output is ignored.  A Filter can return DROP, in which
+  // case the packet is dropped (and *output is ignored).
+  virtual Action Filter(const DataBuffer& input, DataBuffer* output) = 0;
+};
+
+enum Mode { STREAM, DGRAM };
+
+inline std::ostream& operator<<(std::ostream& os, Mode m) {
+  return os << ((m == STREAM) ? "TLS" : "DTLS");
+}
+
+class DummyPrSocket {
+ public:
+  ~DummyPrSocket();
+
+  static PRFileDesc* CreateFD(const std::string& name,
+                              Mode mode);  // Returns an FD.
+  static DummyPrSocket* GetAdapter(PRFileDesc* fd);
+
+  DummyPrSocket* peer() const { return peer_; }
+  void SetPeer(DummyPrSocket* peer) { peer_ = peer; }
+  void SetPacketFilter(PacketFilter* filter);
+  // Drops peer, packet filter and any outstanding packets.
+  void Reset();
+
+  void PacketReceived(const DataBuffer& data);
+  int32_t Read(void* data, int32_t len);
+  int32_t Recv(void* buf, int32_t buflen);
+  int32_t Write(const void* buf, int32_t length);
+  void CloseWrites() { writeable_ = false; }
+
+  Mode mode() const { return mode_; }
+  bool readable() const { return !input_.empty(); }
+
+ private:
+  DummyPrSocket(const std::string& name, Mode mode)
+      : name_(name),
+        mode_(mode),
+        peer_(nullptr),
+        input_(),
+        filter_(nullptr),
+        writeable_(true) {}
+
+  const std::string name_;
+  Mode mode_;
+  DummyPrSocket* peer_;
+  std::queue<Packet*> input_;
+  PacketFilter* filter_;
+  bool writeable_;
+};
+
+// Marker interface.
+class PollTarget {};
+
+enum Event { READABLE_EVENT, TIMER_EVENT /* Must be last */ };
+
+typedef void (*PollCallback)(PollTarget*, Event);
+
+class Poller {
+ public:
+  static Poller* Instance();  // Get a singleton.
+  static void Shutdown();     // Shut it down.
+
+  class Timer {
+   public:
+    Timer(PRTime deadline, PollTarget* target, PollCallback callback)
+        : deadline_(deadline), target_(target), callback_(callback) {}
+    void Cancel() { callback_ = nullptr; }
+
+    PRTime deadline_;
+    PollTarget* target_;
+    PollCallback callback_;
+  };
+
+  void Wait(Event event, DummyPrSocket* adapter, PollTarget* target,
+            PollCallback cb);
+  void Cancel(Event event, DummyPrSocket* adapter);
+  void SetTimer(uint32_t timer_ms, PollTarget* target, PollCallback cb,
+                Timer** handle);
+  bool Poll();
+
+ private:
+  Poller() : waiters_(), timers_() {}
+  ~Poller();
+
+  class Waiter {
+   public:
+    Waiter(DummyPrSocket* io) : io_(io) {
+      memset(&callbacks_[0], 0, sizeof(callbacks_));
+    }
+
+    void WaitFor(Event event, PollCallback callback);
+
+    DummyPrSocket* io_;
+    PollTarget* targets_[TIMER_EVENT];
+    PollCallback callbacks_[TIMER_EVENT];
+  };
+
+  class TimerComparator {
+   public:
+    bool operator()(const Timer* lhs, const Timer* rhs) {
+      return lhs->deadline_ > rhs->deadline_;
+    }
+  };
+
+  static Poller* instance;
+  std::map<DummyPrSocket*, Waiter*> waiters_;
+  std::priority_queue<Timer*, std::vector<Timer*>, TimerComparator> timers_;
+};
+
+}  // end of namespace
+
+#endif
-- 
cgit v1.2.3