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