aboutsummaryrefslogtreecommitdiff
path: root/unittests/Process/gdb-remote/GDBRemoteTestUtils.cpp
blob: 58cc9f50586af2e297ecede7443e0e834c0dce4a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
//===-- GDBRemoteTestUtils.cpp ----------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#if defined(_MSC_VER) && (_HAS_EXCEPTIONS == 0)
// Workaround for MSVC standard library bug, which fails to include <thread>
// when
// exceptions are disabled.
#include <eh.h>
#endif

#include "GDBRemoteTestUtils.h"

#include "lldb/Host/common/TCPSocket.h"
#include "lldb/Host/posix/ConnectionFileDescriptorPosix.h"

#include <future>

namespace lldb_private {
namespace process_gdb_remote {

void GDBRemoteTest::SetUpTestCase() {
#if defined(_MSC_VER)
  WSADATA data;
  ::WSAStartup(MAKEWORD(2, 2), &data);
#endif
}

void GDBRemoteTest::TearDownTestCase() {
#if defined(_MSC_VER)
  ::WSACleanup();
#endif
}

void Connect(GDBRemoteCommunication &client, GDBRemoteCommunication &server) {
  bool child_processes_inherit = false;
  Error error;
  TCPSocket listen_socket(child_processes_inherit, error);
  ASSERT_FALSE(error.Fail());
  error = listen_socket.Listen("127.0.0.1:0", 5);
  ASSERT_FALSE(error.Fail());

  Socket *accept_socket;
  std::future<Error> accept_error = std::async(std::launch::async, [&] {
    return listen_socket.Accept("127.0.0.1:0", child_processes_inherit,
                                accept_socket);
  });

  char connect_remote_address[64];
  snprintf(connect_remote_address, sizeof(connect_remote_address),
           "connect://localhost:%u", listen_socket.GetLocalPortNumber());

  std::unique_ptr<ConnectionFileDescriptor> conn_ap(
      new ConnectionFileDescriptor());
  ASSERT_EQ(conn_ap->Connect(connect_remote_address, nullptr),
            lldb::eConnectionStatusSuccess);

  client.SetConnection(conn_ap.release());
  ASSERT_TRUE(accept_error.get().Success());
  server.SetConnection(new ConnectionFileDescriptor(accept_socket));
}

} // namespace process_gdb_remote
} // namespace lldb_private