Unreviewed, rolling out r245418.
[WebKit-https.git] / Tools / TestWebKitAPI / TCPServer.h
1 /*
2  * Copyright (C) 2019 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #include <thread>
29 #include <wtf/Function.h>
30 #include <wtf/Vector.h>
31
32 extern "C" {
33 struct SSL;
34 int SSL_read(SSL*, void*, int);
35 int SSL_write(SSL*, const void*, int);
36 }
37
38 namespace TestWebKitAPI {
39
40 class TCPServer {
41 public:
42     using Socket = int;
43     using Port = uint16_t;
44     static constexpr Port InvalidPort = 0;
45     
46     TCPServer(Function<void(Socket)>&&, size_t connections = 1);
47     enum class Protocol : bool {
48         HTTPS, HTTPSProxy
49     };
50     TCPServer(Protocol, Function<void(SSL*)>&&);
51     ~TCPServer();
52     
53     Port port() const { return m_port; }
54     
55 private:
56     Optional<Socket> socketBindListen(size_t connections);
57     void listenForConnections(size_t connections);
58
59     Port m_port { InvalidPort };
60     std::thread m_listeningThread;
61     Vector<std::thread> m_connectionThreads;
62     Function<void(Socket)> m_connectionHandler;
63 };
64
65 } // namespace TestWebKitAPI