mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-28 16:26:10 +03:00
91 lines
2.8 KiB
C
91 lines
2.8 KiB
C
|
// Copyright 2013 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_QUIC_TEST_TOOLS_SERVER_THREAD_H_
|
||
|
|
#define NET_TOOLS_QUIC_TEST_TOOLS_SERVER_THREAD_H_
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include "base/macros.h"
|
||
|
|
#include "base/synchronization/waitable_event.h"
|
||
|
|
#include "base/threading/simple_thread.h"
|
||
|
|
#include "net/quic/core/quic_config.h"
|
||
|
|
#include "net/quic/platform/api/quic_containers.h"
|
||
|
|
#include "net/quic/platform/api/quic_mutex.h"
|
||
|
|
#include "net/quic/platform/api/quic_socket_address.h"
|
||
|
|
#include "net/tools/quic/quic_server.h"
|
||
|
|
|
||
|
|
namespace net {
|
||
|
|
namespace test {
|
||
|
|
|
||
|
|
// Simple wrapper class to run QuicServer in a dedicated thread.
|
||
|
|
class ServerThread : public base::SimpleThread {
|
||
|
|
public:
|
||
|
|
ServerThread(QuicServer* server, const QuicSocketAddress& address);
|
||
|
|
|
||
|
|
~ServerThread() override;
|
||
|
|
|
||
|
|
// Prepares the server, but does not start accepting connections. Useful for
|
||
|
|
// injecting mocks.
|
||
|
|
void Initialize();
|
||
|
|
|
||
|
|
// Runs the event loop. Will initialize if necessary.
|
||
|
|
void Run() override;
|
||
|
|
|
||
|
|
// Schedules the given action for execution in the event loop.
|
||
|
|
void Schedule(std::function<void()> action);
|
||
|
|
|
||
|
|
// Waits for the handshake to be confirmed for the first session created.
|
||
|
|
void WaitForCryptoHandshakeConfirmed();
|
||
|
|
|
||
|
|
// Pauses execution of the server until Resume() is called. May only be
|
||
|
|
// called once.
|
||
|
|
void Pause();
|
||
|
|
|
||
|
|
// Resumes execution of the server after Pause() has been called. May only
|
||
|
|
// be called once.
|
||
|
|
void Resume();
|
||
|
|
|
||
|
|
// Stops the server from executing and shuts it down, destroying all
|
||
|
|
// server objects.
|
||
|
|
void Quit();
|
||
|
|
|
||
|
|
// Returns the underlying server. Care must be taken to avoid data races
|
||
|
|
// when accessing the server. It is always safe to access the server
|
||
|
|
// after calling Pause() and before calling Resume().
|
||
|
|
QuicServer* server() { return server_.get(); }
|
||
|
|
|
||
|
|
// Returns the port that the server is listening on.
|
||
|
|
int GetPort();
|
||
|
|
|
||
|
|
private:
|
||
|
|
void MaybeNotifyOfHandshakeConfirmation();
|
||
|
|
void ExecuteScheduledActions();
|
||
|
|
|
||
|
|
base::WaitableEvent
|
||
|
|
confirmed_; // Notified when the first handshake is confirmed.
|
||
|
|
base::WaitableEvent pause_; // Notified when the server should pause.
|
||
|
|
base::WaitableEvent paused_; // Notitied when the server has paused
|
||
|
|
base::WaitableEvent resume_; // Notified when the server should resume.
|
||
|
|
base::WaitableEvent quit_; // Notified when the server should quit.
|
||
|
|
|
||
|
|
std::unique_ptr<QuicServer> server_;
|
||
|
|
QuicSocketAddress address_;
|
||
|
|
mutable QuicMutex port_lock_;
|
||
|
|
int port_ GUARDED_BY(port_lock_);
|
||
|
|
|
||
|
|
bool initialized_;
|
||
|
|
|
||
|
|
QuicMutex scheduled_actions_lock_;
|
||
|
|
QuicDeque<std::function<void()>> scheduled_actions_
|
||
|
|
GUARDED_BY(scheduled_actions_lock_);
|
||
|
|
|
||
|
|
DISALLOW_COPY_AND_ASSIGN(ServerThread);
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace test
|
||
|
|
} // namespace net
|
||
|
|
|
||
|
|
#endif // NET_TOOLS_QUIC_TEST_TOOLS_SERVER_THREAD_H_
|