Home | History | Annotate | Download | only in test_tools 
      1 // Copyright 2013 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_QUIC_SERVER_THREAD_H_
      6 #define NET_TOOLS_QUIC_SERVER_THREAD_H_
      7 
      8 #include "base/threading/simple_thread.h"
      9 #include "net/base/ip_endpoint.h"
     10 #include "net/quic/quic_config.h"
     11 #include "net/tools/quic/quic_server.h"
     12 
     13 namespace net {
     14 namespace tools {
     15 namespace test {
     16 
     17 // Simple wrapper class to run server in a thread.
     18 class ServerThread : public base::SimpleThread {
     19  public:
     20   ServerThread(IPEndPoint address,
     21                const QuicConfig& config,
     22                const QuicVersionVector& supported_versions,
     23                bool strike_register_no_startup_period);
     24 
     25   virtual ~ServerThread();
     26 
     27   // SimpleThread implementation.
     28   virtual void Run() OVERRIDE;
     29 
     30   // Waits until the server has started and is listening for requests.
     31   void WaitForServerStartup();
     32 
     33   // Waits for the handshake to be confirmed for the first session created.
     34   void WaitForCryptoHandshakeConfirmed();
     35 
     36   // Pauses execution of the server until Resume() is called.  May only be
     37   // called once.
     38   void Pause();
     39 
     40   // Resumes execution of the server after Pause() has been called.  May only
     41   // be called once.
     42   void Resume();
     43 
     44   // Stops the server from executing and shuts it down, destroying all
     45   // server objects.
     46   void Quit();
     47 
     48   // Returns the underlying server.  Care must be taken to avoid data races
     49   // when accessing the server.  It is always safe to access the server
     50   // after calling Pause() and before calling Resume().
     51   QuicServer* server() { return &server_; }
     52 
     53   // Returns the port that the server is listening on.
     54   int GetPort();
     55 
     56  private:
     57   void MaybeNotifyOfHandshakeConfirmation();
     58 
     59   base::WaitableEvent listening_;  // Notified when the server is listening.
     60   base::WaitableEvent confirmed_;  // Notified when the first handshake is
     61                                    // confirmed.
     62   base::WaitableEvent pause_;      // Notified when the server should pause.
     63   base::WaitableEvent paused_;     // Notitied when the server has paused
     64   base::WaitableEvent resume_;     // Notified when the server should resume.
     65   base::WaitableEvent quit_;       // Notified when the server should quit.
     66 
     67   tools::QuicServer server_;
     68   IPEndPoint address_;
     69   base::Lock port_lock_;
     70   int port_;
     71 
     72   DISALLOW_COPY_AND_ASSIGN(ServerThread);
     73 };
     74 
     75 }  // namespace test
     76 }  // namespace tools
     77 }  // namespace net
     78 
     79 #endif  // NET_TOOLS_QUIC_SERVER_THREAD_H_
     80