/*
- * Copyright 2014 Facebook, Inc.
+ * Copyright 2015 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
-#include <features.h>
#include <netdb.h>
#include <cstddef>
#include <iostream>
#include <string>
#include <folly/IPAddress.h>
+#include <folly/Portability.h>
namespace folly {
class SocketAddress {
public:
- SocketAddress() {}
+ SocketAddress() = default;
/**
* Construct a SocketAddress from a hostname and port.
}
}
+ SocketAddress(const IPAddress& ipAddr, uint16_t port) {
+ setFromIpAddrPort(ipAddr, port);
+ }
+
SocketAddress(const SocketAddress& addr) {
port_ = addr.port_;
if (addr.getFamily() == AF_UNIX) {
setFromIpPort(ip.c_str(), port);
}
+ /**
+ * Initialize this SocketAddress from an IPAddress struct and port.
+ *
+ * @param ip The IP address in IPAddress format
+ * @param port The port (in host byte order)
+ */
+ void setFromIpAddrPort(const IPAddress& ip, uint16_t port);
+
/**
* Initialize this SocketAddress from a local port number.
*
void setFromPath(const char* path, size_t length);
+ // a typedef that allow us to compile against both winsock & POSIX sockets:
+ using SocketDesc = decltype(socket(0,0,0)); // POSIX: int, winsock: unsigned
+
/**
* Initialize this SocketAddress from a socket's peer address.
*
* Raises std::system_error on error.
*/
- void setFromPeerAddress(int socket);
+ void setFromPeerAddress(SocketDesc socket);
/**
* Initialize this SocketAddress from a socket's local address.
*
* Raises std::system_error on error.
*/
- void setFromLocalAddress(int socket);
+ void setFromLocalAddress(SocketDesc socket);
/**
* Initialize this TSocketAddress from a struct sockaddr.
struct sockaddr_un *addr;
socklen_t len;
+ /* For debugging only, will be removed */
+ uint64_t magic;
+ static constexpr uint64_t kMagic = 0x1234faceb00c;
+
socklen_t pathLength() const {
return len - offsetof(struct sockaddr_un, sun_path);
}
void init() {
addr = new sockaddr_un;
+ magic = kMagic;
addr->sun_family = AF_UNIX;
len = 0;
}
void init(const ExternalUnixAddr &other) {
addr = new sockaddr_un;
+ magic = kMagic;
len = other.len;
memcpy(addr, other.addr, len);
// Fill the rest with 0s, just for safety
sizeof(struct sockaddr_un) - len);
}
void copy(const ExternalUnixAddr &other) {
+ CHECK(magic == kMagic);
len = other.len;
memcpy(addr, other.addr, len);
}
void free() {
+ CHECK(magic == kMagic);
delete addr;
+ magic = 0;
}
};
+ // a typedef that allow us to compile against both winsock & POSIX sockets:
+ // (both arg types and calling conventions differ for both)
+ // POSIX: void setFromSocket(int socket,
+ // int(*fn)(int, struct sockaddr*, socklen_t*));
+ // mingw: void setFromSocket(unsigned socket,
+ // int(*fn)(unsigned, struct sockaddr*, socklen_t*));
+ using GetPeerNameFunc = decltype(getpeername);
+
struct addrinfo* getAddrInfo(const char* host, uint16_t port, int flags);
struct addrinfo* getAddrInfo(const char* host, const char* port, int flags);
void setFromAddrInfo(const struct addrinfo* results);
void setFromLocalAddr(const struct addrinfo* results);
- void setFromSocket(int socket, int (*fn)(int, struct sockaddr*, socklen_t*));
+ void setFromSocket(SocketDesc socket, GetPeerNameFunc fn);
std::string getIpString(int flags) const;
void getIpString(char *buf, size_t buflen, int flags) const;