tcp_server_thread_for_testing.h

Go to the documentation of this file.

// Copyright 2016 Google Inc.

#ifndef PAGESPEED_KERNEL_BASE_TCP_SERVER_THREAD_FOR_TESTING_H_
#define PAGESPEED_KERNEL_BASE_TCP_SERVER_THREAD_FOR_TESTING_H_

#include "apr_pools.h"
#include "apr_network_io.h"
#include "pagespeed/kernel/base/condvar.h"
#include "pagespeed/kernel/base/scoped_ptr.h"
#include "pagespeed/kernel/base/string_util.h"
#include "pagespeed/kernel/base/thread.h"
#include "pagespeed/kernel/base/thread_annotations.h"
#include "pagespeed/kernel/base/thread_system.h"

namespace net_instaweb {


class TcpServerThreadForTesting : public ThreadSystem::Thread {
 public:
  TcpServerThreadForTesting(apr_port_t listen_port, StringPiece thread_name,
                            ThreadSystem* thread_system);

  void ShutDown();

  virtual ~TcpServerThreadForTesting();

  apr_port_t GetListeningPort();

  static void PickListenPortOnce(apr_port_t* port_number);

 protected:
  apr_pool_t* pool() { return pool_; }

 private:
  virtual void HandleClientConnection(apr_socket_t* sock) = 0;

  apr_socket_t* CreateAndBindSocket() EXCLUSIVE_LOCKS_REQUIRED(mutex_);

  static void CreateAndBindSocket(apr_pool_t* apr_pool, apr_socket_t** socket,
                                  apr_port_t* port);

  void Run() override;

  scoped_ptr<ThreadSystem::CondvarCapableMutex> mutex_;
  scoped_ptr<ThreadSystem::Condvar> ready_notify_ GUARDED_BY(mutex_);
  apr_pool_t* pool_;
  const apr_port_t requested_listen_port_;
  apr_port_t actual_listening_port_ GUARDED_BY(mutex_);
  apr_socket_t* listen_sock_ GUARDED_BY(mutex_);
  bool terminating_ GUARDED_BY(mutex_);
  bool is_shut_down_;
};

}  

#endif