1 files changed, 95 insertions, 0 deletions
diff --git a/net/tools/flip_server/acceptor_thread.h b/net/tools/flip_server/acceptor_thread.h
new file mode 100644
index 0000000..df161d7
--- /dev/null
+++ b/net/tools/flip_server/acceptor_thread.h
@@ -0,0 +1,95 @@
+// Copyright (c) 2009 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_TOOLS_FLIP_SERVER_ACCEPTOR_THREAD_H_
+#define NET_TOOLS_FLIP_SERVER_ACCEPTOR_THREAD_H_
+
+#include <list>
+#include <string>
+#include <vector>
+
+#include "base/threading/simple_thread.h"
+#include "net/tools/flip_server/epoll_server.h"
+#include "net/tools/flip_server/sm_interface.h"
+#include "openssl/ssl.h"
+
+struct sockaddr_in;
+
+namespace net {
+
+class FlipAcceptor;
+class MemoryCache;
+class SMConnection;
+class SSLState;
+
+// TODO(mbelshe):  Get rid of this class; we don't need a lock just to set
+//    a bool cross threads - especially one which only is set once...
+class Notification {
+ public:
+   explicit Notification(bool value) : value_(value) {}
+
+   void Notify() {
+     base::AutoLock al(lock_);
+     value_ = true;
+   }
+   bool HasBeenNotified() {
+     base::AutoLock al(lock_);
+     return value_;
+   }
+   bool value_;
+   base::Lock lock_;
+};
+
+class SMAcceptorThread : public base::SimpleThread,
+                         public EpollCallbackInterface,
+                         public SMConnectionPoolInterface {
+ public:
+  SMAcceptorThread(FlipAcceptor *acceptor, MemoryCache* memory_cache);
+  ~SMAcceptorThread();
+
+  // EpollCallbackInteface interface
+  virtual void OnRegistration(EpollServer* eps, int fd, int event_mask) {}
+  virtual void OnModification(int fd, int event_mask) {}
+  virtual void OnEvent(int fd, EpollEvent* event);
+  virtual void OnUnregistration(int fd, bool replaced) {}
+  virtual void OnShutdown(EpollServer* eps, int fd) {}
+
+  // SMConnectionPool interface
+  virtual void SMConnectionDone(SMConnection* sc);
+
+  // TODO(mbelshe): figure out if we can move these to private functions.
+  SMConnection* NewConnection();
+  SMConnection* FindOrMakeNewSMConnection();
+  void InitWorker();
+  void HandleConnection(int server_fd, struct sockaddr_in *remote_addr);
+  void AcceptFromListenFD();
+
+  // Notify the Accept thread that it is time to terminate.
+  void Quit() { quitting_.Notify(); }
+
+  // Iterates through a list of active connections expiring any that have been
+  // idle longer than the configured timeout.
+  void HandleConnectionIdleTimeout();
+
+  void Run();
+
+ private:
+  EpollServer epoll_server_;
+  FlipAcceptor* acceptor_;
+  SSLState* ssl_state_;
+  bool use_ssl_;
+  int idle_socket_timeout_s_;
+
+  std::vector<SMConnection*> unused_server_connections_;
+  std::vector<SMConnection*> tmp_unused_server_connections_;
+  std::vector<SMConnection*> allocated_server_connections_;
+  std::list<SMConnection*> active_server_connections_;
+  Notification quitting_;
+  MemoryCache* memory_cache_;
+};
+
+}  // namespace net
+
+#endif  // NET_TOOLS_FLIP_SERVER_ACCEPTOR_THREAD_H_
+