1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef NET_TOOLS_FLIP_SERVER_ACCEPTOR_THREAD_H_ 6 #define NET_TOOLS_FLIP_SERVER_ACCEPTOR_THREAD_H_ 7 8 #include <list> 9 #include <string> 10 #include <vector> 11 12 #include "base/threading/simple_thread.h" 13 #include "net/tools/flip_server/epoll_server.h" 14 #include "net/tools/flip_server/sm_interface.h" 15 #include "openssl/ssl.h" 16 17 struct sockaddr_in; 18 19 namespace net { 20 21 class FlipAcceptor; 22 class MemoryCache; 23 class SMConnection; 24 struct SSLState; 25 26 // TODO(mbelshe): Get rid of this class; we don't need a lock just to set 27 // a bool cross threads - especially one which only is set once... 28 class Notification { 29 public: Notification(bool value)30 explicit Notification(bool value) : value_(value) {} 31 Notify()32 void Notify() { 33 base::AutoLock al(lock_); 34 value_ = true; 35 } HasBeenNotified()36 bool HasBeenNotified() { 37 base::AutoLock al(lock_); 38 return value_; 39 } 40 bool value_; 41 base::Lock lock_; 42 }; 43 44 class SMAcceptorThread : public base::SimpleThread, 45 public EpollCallbackInterface, 46 public SMConnectionPoolInterface { 47 public: 48 SMAcceptorThread(FlipAcceptor *acceptor, MemoryCache* memory_cache); 49 ~SMAcceptorThread(); 50 51 // EpollCallbackInteface interface OnRegistration(EpollServer * eps,int fd,int event_mask)52 virtual void OnRegistration(EpollServer* eps, int fd, int event_mask) {} OnModification(int fd,int event_mask)53 virtual void OnModification(int fd, int event_mask) {} 54 virtual void OnEvent(int fd, EpollEvent* event); OnUnregistration(int fd,bool replaced)55 virtual void OnUnregistration(int fd, bool replaced) {} OnShutdown(EpollServer * eps,int fd)56 virtual void OnShutdown(EpollServer* eps, int fd) {} 57 58 // SMConnectionPool interface 59 virtual void SMConnectionDone(SMConnection* sc); 60 61 // TODO(mbelshe): figure out if we can move these to private functions. 62 SMConnection* NewConnection(); 63 SMConnection* FindOrMakeNewSMConnection(); 64 void InitWorker(); 65 void HandleConnection(int server_fd, struct sockaddr_in *remote_addr); 66 void AcceptFromListenFD(); 67 68 // Notify the Accept thread that it is time to terminate. Quit()69 void Quit() { quitting_.Notify(); } 70 71 // Iterates through a list of active connections expiring any that have been 72 // idle longer than the configured timeout. 73 void HandleConnectionIdleTimeout(); 74 75 virtual void Run(); 76 77 private: 78 EpollServer epoll_server_; 79 FlipAcceptor* acceptor_; 80 SSLState* ssl_state_; 81 bool use_ssl_; 82 int idle_socket_timeout_s_; 83 84 std::vector<SMConnection*> unused_server_connections_; 85 std::vector<SMConnection*> tmp_unused_server_connections_; 86 std::vector<SMConnection*> allocated_server_connections_; 87 std::list<SMConnection*> active_server_connections_; 88 Notification quitting_; 89 MemoryCache* memory_cache_; 90 }; 91 92 } // namespace net 93 94 #endif // NET_TOOLS_FLIP_SERVER_ACCEPTOR_THREAD_H_ 95 96