Socket.h

Go to the documentation of this file.

//===-- Socket.h ------------------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#ifndef LLDB_HOST_SOCKET_H
#define LLDB_HOST_SOCKET_H
 
#include <memory>
#include <string>
 
#include "lldb/lldb-private.h"
 
#include "lldb/Host/SocketAddress.h"
#include "lldb/Utility/IOObject.h"
#include "lldb/Utility/Status.h"
 
#ifdef _WIN32
#include "lldb/Host/windows/windows.h"
#include <winsock2.h>
#include <ws2tcpip.h>
#endif
 
namespace llvm {
class StringRef;
}
 
namespace lldb_private {
 
#if defined(_WIN32)
typedef SOCKET NativeSocket;
#else
typedef int NativeSocket;
#endif
class TCPSocket;
class UDPSocket;
 
class Socket : public IOObject {
public:
  enum SocketProtocol {
    ProtocolTcp,
    ProtocolUdp,
    ProtocolUnixDomain,
    ProtocolUnixAbstract
  };
 
  struct HostAndPort {
    std::string hostname;
    uint16_t port;
 
    bool operator==(const HostAndPort &R) const {
      return port == R.port && hostname == R.hostname;
    }
  };
 
  static const NativeSocket kInvalidSocketValue;
 
  ~Socket() override;
 
  static const char *FindSchemeByProtocol(const SocketProtocol protocol);
  static bool FindProtocolByScheme(const char *scheme,
                                   SocketProtocol &protocol);
 
  static llvm::Error Initialize();
  static void Terminate();
 
  static std::unique_ptr<Socket> Create(const SocketProtocol protocol,
                                        bool child_processes_inherit,
                                        Status &error);
 
  virtual Status Connect(llvm::StringRef name) = 0;
  virtual Status Listen(llvm::StringRef name, int backlog) = 0;
  virtual Status Accept(Socket *&socket) = 0;
 
  // Initialize a Tcp Socket object in listening mode.  listen and accept are
  // implemented separately because the caller may wish to manipulate or query
  // the socket after it is initialized, but before entering a blocking accept.
  static llvm::Expected<std::unique_ptr<TCPSocket>>
  TcpListen(llvm::StringRef host_and_port, bool child_processes_inherit,
            int backlog = 5);
 
  static llvm::Expected<std::unique_ptr<Socket>>
  TcpConnect(llvm::StringRef host_and_port, bool child_processes_inherit);
 
  static llvm::Expected<std::unique_ptr<UDPSocket>>
  UdpConnect(llvm::StringRef host_and_port, bool child_processes_inherit);
 
  int GetOption(int level, int option_name, int &option_value);
  int SetOption(int level, int option_name, int option_value);
 
  NativeSocket GetNativeSocket() const { return m_socket; }
  SocketProtocol GetSocketProtocol() const { return m_protocol; }
 
  Status Read(void *buf, size_t &num_bytes) override;
  Status Write(const void *buf, size_t &num_bytes) override;
 
  Status Close() override;
 
  bool IsValid() const override { return m_socket != kInvalidSocketValue; }
  WaitableHandle GetWaitableHandle() override;
 
  static llvm::Expected<HostAndPort>
  DecodeHostAndPort(llvm::StringRef host_and_port);
 
  // If this Socket is connected then return the URI used to connect.
  virtual std::string GetRemoteConnectionURI() const { return ""; };
 
protected:
  Socket(SocketProtocol protocol, bool should_close,
         bool m_child_process_inherit);
 
  virtual size_t Send(const void *buf, const size_t num_bytes);
 
  static void SetLastError(Status &error);
  static NativeSocket CreateSocket(const int domain, const int type,
                                   const int protocol,
                                   bool child_processes_inherit, Status &error);
  static NativeSocket AcceptSocket(NativeSocket sockfd, struct sockaddr *addr,
                                   socklen_t *addrlen,
                                   bool child_processes_inherit, Status &error);
 
  SocketProtocol m_protocol;
  NativeSocket m_socket;
  bool m_child_processes_inherit;
  bool m_should_close_fd;
};
 
llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
                              const Socket::HostAndPort &HP);
 
} // namespace lldb_private
 
#endif // LLDB_HOST_SOCKET_H