mirror of
https://github.com/dlang/phobos.git
synced 2025-04-26 21:22:20 +03:00
3848 lines
112 KiB
D
3848 lines
112 KiB
D
// Written in the D programming language
|
|
|
|
// NOTE: When working on this module, be sure to run tests with -debug=std_socket
|
|
// E.g.: dmd -version=StdUnittest -debug=std_socket -unittest -main -run socket
|
|
// This will enable some tests which are too slow or flaky to run as part of CI.
|
|
|
|
/*
|
|
Copyright (C) 2004-2011 Christopher E. Miller
|
|
|
|
socket.d 1.4
|
|
Jan 2011
|
|
|
|
Thanks to Benjamin Herr for his assistance.
|
|
*/
|
|
|
|
/**
|
|
* Socket primitives.
|
|
* Example: See [listener.d](https://github.com/dlang/undeaD/blob/master/dmdsamples/listener.d) and [htmlget.d](https://github.com/dlang/undeaD/blob/master/dmdsamples/htmlget.d)
|
|
* License: $(HTTP www.boost.org/LICENSE_1_0.txt, Boost License 1.0).
|
|
* Authors: Christopher E. Miller, $(HTTP klickverbot.at, David Nadlinger),
|
|
* $(HTTP thecybershadow.net, Vladimir Panteleev)
|
|
* Source: $(PHOBOSSRC std/socket.d)
|
|
*/
|
|
|
|
module std.socket;
|
|
|
|
import core.stdc.stdint, core.stdc.stdlib, core.stdc.string, std.conv, std.string;
|
|
|
|
import core.stdc.config;
|
|
import core.time : dur, Duration;
|
|
import std.exception;
|
|
|
|
import std.internal.cstring;
|
|
|
|
version (iOS)
|
|
version = iOSDerived;
|
|
else version (TVOS)
|
|
version = iOSDerived;
|
|
else version (WatchOS)
|
|
version = iOSDerived;
|
|
|
|
@safe:
|
|
|
|
version (Windows)
|
|
{
|
|
pragma (lib, "ws2_32.lib");
|
|
pragma (lib, "wsock32.lib");
|
|
|
|
import core.sys.windows.winbase, std.windows.syserror;
|
|
public import core.sys.windows.winsock2;
|
|
private alias _ctimeval = core.sys.windows.winsock2.timeval;
|
|
private alias _clinger = core.sys.windows.winsock2.linger;
|
|
|
|
enum socket_t : SOCKET { INVALID_SOCKET }
|
|
private const int _SOCKET_ERROR = SOCKET_ERROR;
|
|
|
|
/**
|
|
* On Windows, there is no `SO_REUSEPORT`.
|
|
* However, `SO_REUSEADDR` is equivalent to `SO_REUSEPORT` there.
|
|
* $(LINK https://learn.microsoft.com/en-us/windows/win32/winsock/using-so-reuseaddr-and-so-exclusiveaddruse)
|
|
*/
|
|
private enum SO_REUSEPORT = SO_REUSEADDR;
|
|
|
|
private int _lasterr() nothrow @nogc
|
|
{
|
|
return WSAGetLastError();
|
|
}
|
|
}
|
|
else version (Posix)
|
|
{
|
|
version (linux)
|
|
{
|
|
enum : int
|
|
{
|
|
TCP_KEEPIDLE = 4,
|
|
TCP_KEEPINTVL = 5
|
|
}
|
|
}
|
|
|
|
public import core.sys.posix.netinet.in_;
|
|
import core.sys.posix.arpa.inet;
|
|
import core.sys.posix.fcntl;
|
|
import core.sys.posix.netdb;
|
|
import core.sys.posix.netinet.tcp;
|
|
import core.sys.posix.sys.select;
|
|
import core.sys.posix.sys.socket;
|
|
import core.sys.posix.sys.time;
|
|
import core.sys.posix.sys.un : sockaddr_un;
|
|
import core.sys.posix.unistd;
|
|
private alias _ctimeval = core.sys.posix.sys.time.timeval;
|
|
private alias _clinger = core.sys.posix.sys.socket.linger;
|
|
|
|
import core.stdc.errno;
|
|
|
|
enum socket_t : int32_t { _init = -1 }
|
|
private const int _SOCKET_ERROR = -1;
|
|
|
|
private enum : int
|
|
{
|
|
SD_RECEIVE = SHUT_RD,
|
|
SD_SEND = SHUT_WR,
|
|
SD_BOTH = SHUT_RDWR
|
|
}
|
|
|
|
private int _lasterr() nothrow @nogc
|
|
{
|
|
return errno;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
static assert(0, "No socket support for this platform yet.");
|
|
}
|
|
|
|
version (StdUnittest)
|
|
{
|
|
// Print a message on exception instead of failing the unittest.
|
|
private void softUnittest(void delegate() @safe test, int line = __LINE__) @trusted
|
|
{
|
|
debug (std_socket)
|
|
test();
|
|
else
|
|
{
|
|
import std.stdio : writefln;
|
|
try
|
|
test();
|
|
catch (Throwable e)
|
|
writefln("Ignoring std.socket(%d) test failure (likely caused by flaky environment): %s", line, e.msg);
|
|
}
|
|
}
|
|
|
|
// Without debug=std_socket, still compile the slow tests, just don't run them.
|
|
debug (std_socket)
|
|
private enum runSlowTests = true;
|
|
else
|
|
private enum runSlowTests = false;
|
|
}
|
|
|
|
/// Base exception thrown by `std.socket`.
|
|
class SocketException: Exception
|
|
{
|
|
mixin basicExceptionCtors;
|
|
}
|
|
|
|
version (CRuntime_Glibc) version = GNU_STRERROR;
|
|
version (CRuntime_UClibc) version = GNU_STRERROR;
|
|
|
|
/*
|
|
* Needs to be public so that SocketOSException can be thrown outside of
|
|
* std.socket (since it uses it as a default argument), but it probably doesn't
|
|
* need to actually show up in the docs, since there's not really any public
|
|
* need for it outside of being a default argument.
|
|
*/
|
|
string formatSocketError(int err) @trusted
|
|
{
|
|
version (Posix)
|
|
{
|
|
char[80] buf;
|
|
const(char)* cs;
|
|
version (GNU_STRERROR)
|
|
{
|
|
cs = strerror_r(err, buf.ptr, buf.length);
|
|
}
|
|
else
|
|
{
|
|
auto errs = strerror_r(err, buf.ptr, buf.length);
|
|
if (errs == 0)
|
|
cs = buf.ptr;
|
|
else
|
|
return "Socket error " ~ to!string(err);
|
|
}
|
|
|
|
auto len = strlen(cs);
|
|
|
|
if (cs[len - 1] == '\n')
|
|
len--;
|
|
if (cs[len - 1] == '\r')
|
|
len--;
|
|
return cs[0 .. len].idup;
|
|
}
|
|
else
|
|
version (Windows)
|
|
{
|
|
return generateSysErrorMsg(err);
|
|
}
|
|
else
|
|
return "Socket error " ~ to!string(err);
|
|
}
|
|
|
|
/// Returns the error message of the most recently encountered network error.
|
|
@property string lastSocketError()
|
|
{
|
|
return formatSocketError(_lasterr());
|
|
}
|
|
|
|
/// Socket exception representing network errors reported by the operating system.
|
|
class SocketOSException: SocketException
|
|
{
|
|
int errorCode; /// Platform-specific error code.
|
|
|
|
///
|
|
this(string msg,
|
|
string file = __FILE__,
|
|
size_t line = __LINE__,
|
|
Throwable next = null,
|
|
int err = _lasterr(),
|
|
string function(int) @trusted errorFormatter = &formatSocketError)
|
|
{
|
|
errorCode = err;
|
|
|
|
if (msg.length)
|
|
super(msg ~ ": " ~ errorFormatter(err), file, line, next);
|
|
else
|
|
super(errorFormatter(err), file, line, next);
|
|
}
|
|
|
|
///
|
|
this(string msg,
|
|
Throwable next,
|
|
string file = __FILE__,
|
|
size_t line = __LINE__,
|
|
int err = _lasterr(),
|
|
string function(int) @trusted errorFormatter = &formatSocketError)
|
|
{
|
|
this(msg, file, line, next, err, errorFormatter);
|
|
}
|
|
|
|
///
|
|
this(string msg,
|
|
int err,
|
|
string function(int) @trusted errorFormatter = &formatSocketError,
|
|
string file = __FILE__,
|
|
size_t line = __LINE__,
|
|
Throwable next = null)
|
|
{
|
|
this(msg, file, line, next, err, errorFormatter);
|
|
}
|
|
}
|
|
|
|
/// Socket exception representing invalid parameters specified by user code.
|
|
class SocketParameterException: SocketException
|
|
{
|
|
mixin basicExceptionCtors;
|
|
}
|
|
|
|
/**
|
|
* Socket exception representing attempts to use network capabilities not
|
|
* available on the current system.
|
|
*/
|
|
class SocketFeatureException: SocketException
|
|
{
|
|
mixin basicExceptionCtors;
|
|
}
|
|
|
|
|
|
/**
|
|
* Returns:
|
|
* `true` if the last socket operation failed because the socket
|
|
* was in non-blocking mode and the operation would have blocked,
|
|
* or if the socket is in blocking mode and set a `SNDTIMEO` or `RCVTIMEO`,
|
|
* and the operation timed out.
|
|
*/
|
|
bool wouldHaveBlocked() nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
return _lasterr() == WSAEWOULDBLOCK || _lasterr() == WSAETIMEDOUT;
|
|
else version (Posix)
|
|
return _lasterr() == EAGAIN;
|
|
else
|
|
static assert(0, "No socket support for this platform yet.");
|
|
}
|
|
|
|
@safe unittest
|
|
{
|
|
auto sockets = socketPair();
|
|
auto s = sockets[0];
|
|
s.setOption(SocketOptionLevel.SOCKET, SocketOption.RCVTIMEO, dur!"msecs"(10));
|
|
ubyte[] buffer = new ubyte[](16);
|
|
auto rec = s.receive(buffer);
|
|
assert(rec == -1 && wouldHaveBlocked());
|
|
}
|
|
|
|
|
|
private immutable
|
|
{
|
|
typeof(&getnameinfo) getnameinfoPointer;
|
|
typeof(&getaddrinfo) getaddrinfoPointer;
|
|
typeof(&freeaddrinfo) freeaddrinfoPointer;
|
|
}
|
|
|
|
shared static this() @system
|
|
{
|
|
version (Windows)
|
|
{
|
|
WSADATA wd;
|
|
|
|
// Winsock will still load if an older version is present.
|
|
// The version is just a request.
|
|
int val;
|
|
val = WSAStartup(0x2020, &wd);
|
|
if (val) // Request Winsock 2.2 for IPv6.
|
|
throw new SocketOSException("Unable to initialize socket library", val);
|
|
|
|
// These functions may not be present on older Windows versions.
|
|
// See the comment in InternetAddress.toHostNameString() for details.
|
|
auto ws2Lib = GetModuleHandleA("ws2_32.dll");
|
|
if (ws2Lib)
|
|
{
|
|
getnameinfoPointer = cast(typeof(getnameinfoPointer))
|
|
GetProcAddress(ws2Lib, "getnameinfo");
|
|
getaddrinfoPointer = cast(typeof(getaddrinfoPointer))
|
|
GetProcAddress(ws2Lib, "getaddrinfo");
|
|
freeaddrinfoPointer = cast(typeof(freeaddrinfoPointer))
|
|
GetProcAddress(ws2Lib, "freeaddrinfo");
|
|
}
|
|
}
|
|
else version (Posix)
|
|
{
|
|
getnameinfoPointer = &getnameinfo;
|
|
getaddrinfoPointer = &getaddrinfo;
|
|
freeaddrinfoPointer = &freeaddrinfo;
|
|
}
|
|
}
|
|
|
|
|
|
shared static ~this() @system nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
{
|
|
WSACleanup();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* The communication domain used to resolve an address.
|
|
*/
|
|
enum AddressFamily: ushort
|
|
{
|
|
UNSPEC = AF_UNSPEC, /// Unspecified address family
|
|
UNIX = AF_UNIX, /// Local communication (Unix socket)
|
|
INET = AF_INET, /// Internet Protocol version 4
|
|
IPX = AF_IPX, /// Novell IPX
|
|
APPLETALK = AF_APPLETALK, /// AppleTalk
|
|
INET6 = AF_INET6, /// Internet Protocol version 6
|
|
}
|
|
|
|
|
|
/**
|
|
* Communication semantics
|
|
*/
|
|
enum SocketType: int
|
|
{
|
|
STREAM = SOCK_STREAM, /// Sequenced, reliable, two-way communication-based byte streams
|
|
DGRAM = SOCK_DGRAM, /// Connectionless, unreliable datagrams with a fixed maximum length; data may be lost or arrive out of order
|
|
RAW = SOCK_RAW, /// Raw protocol access
|
|
RDM = SOCK_RDM, /// Reliably-delivered message datagrams
|
|
SEQPACKET = SOCK_SEQPACKET, /// Sequenced, reliable, two-way connection-based datagrams with a fixed maximum length
|
|
}
|
|
|
|
|
|
/**
|
|
* Protocol
|
|
*/
|
|
enum ProtocolType: int
|
|
{
|
|
IP = IPPROTO_IP, /// Internet Protocol version 4
|
|
ICMP = IPPROTO_ICMP, /// Internet Control Message Protocol
|
|
IGMP = IPPROTO_IGMP, /// Internet Group Management Protocol
|
|
GGP = IPPROTO_GGP, /// Gateway to Gateway Protocol
|
|
TCP = IPPROTO_TCP, /// Transmission Control Protocol
|
|
PUP = IPPROTO_PUP, /// PARC Universal Packet Protocol
|
|
UDP = IPPROTO_UDP, /// User Datagram Protocol
|
|
IDP = IPPROTO_IDP, /// Xerox NS protocol
|
|
RAW = IPPROTO_RAW, /// Raw IP packets
|
|
IPV6 = IPPROTO_IPV6, /// Internet Protocol version 6
|
|
}
|
|
|
|
|
|
/**
|
|
* Class for retrieving protocol information.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* auto proto = new Protocol;
|
|
* writeln("About protocol TCP:");
|
|
* if (proto.getProtocolByType(ProtocolType.TCP))
|
|
* {
|
|
* writefln(" Name: %s", proto.name);
|
|
* foreach (string s; proto.aliases)
|
|
* writefln(" Alias: %s", s);
|
|
* }
|
|
* else
|
|
* writeln(" No information found");
|
|
* ---
|
|
*/
|
|
class Protocol
|
|
{
|
|
/// These members are populated when one of the following functions are called successfully:
|
|
ProtocolType type;
|
|
string name; /// ditto
|
|
string[] aliases; /// ditto
|
|
|
|
|
|
void populate(protoent* proto) @system pure nothrow
|
|
{
|
|
type = cast(ProtocolType) proto.p_proto;
|
|
name = to!string(proto.p_name);
|
|
|
|
int i;
|
|
for (i = 0;; i++)
|
|
{
|
|
if (!proto.p_aliases[i])
|
|
break;
|
|
}
|
|
|
|
if (i)
|
|
{
|
|
aliases = new string[i];
|
|
for (i = 0; i != aliases.length; i++)
|
|
{
|
|
aliases[i] =
|
|
to!string(proto.p_aliases[i]);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
aliases = null;
|
|
}
|
|
}
|
|
|
|
/** Returns: false on failure */
|
|
bool getProtocolByName(scope const(char)[] name) @trusted nothrow
|
|
{
|
|
protoent* proto;
|
|
proto = getprotobyname(name.tempCString());
|
|
if (!proto)
|
|
return false;
|
|
populate(proto);
|
|
return true;
|
|
}
|
|
|
|
|
|
/** Returns: false on failure */
|
|
// Same as getprotobynumber().
|
|
bool getProtocolByType(ProtocolType type) @trusted nothrow
|
|
{
|
|
protoent* proto;
|
|
proto = getprotobynumber(type);
|
|
if (!proto)
|
|
return false;
|
|
populate(proto);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
|
|
// Skip this test on Android because getprotobyname/number are
|
|
// unimplemented in bionic.
|
|
version (CRuntime_Bionic) {} else
|
|
@safe unittest
|
|
{
|
|
// import std.stdio : writefln;
|
|
softUnittest({
|
|
Protocol proto = new Protocol;
|
|
assert(proto.getProtocolByType(ProtocolType.TCP));
|
|
//writeln("About protocol TCP:");
|
|
//writefln("\tName: %s", proto.name);
|
|
// foreach (string s; proto.aliases)
|
|
// {
|
|
// writefln("\tAlias: %s", s);
|
|
// }
|
|
assert(proto.name == "tcp");
|
|
assert(proto.aliases.length == 1 && proto.aliases[0] == "TCP");
|
|
});
|
|
}
|
|
|
|
|
|
/**
|
|
* Class for retrieving service information.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* auto serv = new Service;
|
|
* writeln("About service epmap:");
|
|
* if (serv.getServiceByName("epmap", "tcp"))
|
|
* {
|
|
* writefln(" Service: %s", serv.name);
|
|
* writefln(" Port: %d", serv.port);
|
|
* writefln(" Protocol: %s", serv.protocolName);
|
|
* foreach (string s; serv.aliases)
|
|
* writefln(" Alias: %s", s);
|
|
* }
|
|
* else
|
|
* writefln(" No service for epmap.");
|
|
* ---
|
|
*/
|
|
class Service
|
|
{
|
|
/// These members are populated when one of the following functions are called successfully:
|
|
string name;
|
|
string[] aliases; /// ditto
|
|
ushort port; /// ditto
|
|
string protocolName; /// ditto
|
|
|
|
|
|
void populate(servent* serv) @system pure nothrow
|
|
{
|
|
name = to!string(serv.s_name);
|
|
port = ntohs(cast(ushort) serv.s_port);
|
|
protocolName = to!string(serv.s_proto);
|
|
|
|
int i;
|
|
for (i = 0;; i++)
|
|
{
|
|
if (!serv.s_aliases[i])
|
|
break;
|
|
}
|
|
|
|
if (i)
|
|
{
|
|
aliases = new string[i];
|
|
for (i = 0; i != aliases.length; i++)
|
|
{
|
|
aliases[i] =
|
|
to!string(serv.s_aliases[i]);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
aliases = null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* If a protocol name is omitted, any protocol will be matched.
|
|
* Returns: false on failure.
|
|
*/
|
|
bool getServiceByName(scope const(char)[] name, scope const(char)[] protocolName = null) @trusted nothrow
|
|
{
|
|
servent* serv;
|
|
serv = getservbyname(name.tempCString(), protocolName.tempCString());
|
|
if (!serv)
|
|
return false;
|
|
populate(serv);
|
|
return true;
|
|
}
|
|
|
|
|
|
/// ditto
|
|
bool getServiceByPort(ushort port, scope const(char)[] protocolName = null) @trusted nothrow
|
|
{
|
|
servent* serv;
|
|
serv = getservbyport(port, protocolName.tempCString());
|
|
if (!serv)
|
|
return false;
|
|
populate(serv);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
|
|
@safe unittest
|
|
{
|
|
import std.stdio : writefln;
|
|
softUnittest({
|
|
Service serv = new Service;
|
|
if (serv.getServiceByName("epmap", "tcp"))
|
|
{
|
|
// writefln("About service epmap:");
|
|
// writefln("\tService: %s", serv.name);
|
|
// writefln("\tPort: %d", serv.port);
|
|
// writefln("\tProtocol: %s", serv.protocolName);
|
|
// foreach (string s; serv.aliases)
|
|
// {
|
|
// writefln("\tAlias: %s", s);
|
|
// }
|
|
// For reasons unknown this is loc-srv on Wine and epmap on Windows
|
|
assert(serv.name == "loc-srv" || serv.name == "epmap", serv.name);
|
|
assert(serv.port == 135);
|
|
assert(serv.protocolName == "tcp");
|
|
}
|
|
else
|
|
{
|
|
writefln("No service for epmap.");
|
|
}
|
|
});
|
|
}
|
|
|
|
|
|
private mixin template socketOSExceptionCtors()
|
|
{
|
|
///
|
|
this(string msg, string file = __FILE__, size_t line = __LINE__,
|
|
Throwable next = null, int err = _lasterr())
|
|
{
|
|
super(msg, file, line, next, err);
|
|
}
|
|
|
|
///
|
|
this(string msg, Throwable next, string file = __FILE__,
|
|
size_t line = __LINE__, int err = _lasterr())
|
|
{
|
|
super(msg, next, file, line, err);
|
|
}
|
|
|
|
///
|
|
this(string msg, int err, string file = __FILE__, size_t line = __LINE__,
|
|
Throwable next = null)
|
|
{
|
|
super(msg, next, file, line, err);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Class for exceptions thrown from an `InternetHost`.
|
|
*/
|
|
class HostException: SocketOSException
|
|
{
|
|
mixin socketOSExceptionCtors;
|
|
}
|
|
|
|
/**
|
|
* Class for resolving IPv4 addresses.
|
|
*
|
|
* Consider using `getAddress`, `parseAddress` and `Address` methods
|
|
* instead of using this class directly.
|
|
*/
|
|
class InternetHost
|
|
{
|
|
/// These members are populated when one of the following functions are called successfully:
|
|
string name;
|
|
string[] aliases; /// ditto
|
|
uint[] addrList; /// ditto
|
|
|
|
|
|
void validHostent(in hostent* he)
|
|
{
|
|
if (he.h_addrtype != cast(int) AddressFamily.INET || he.h_length != 4)
|
|
throw new HostException("Address family mismatch");
|
|
}
|
|
|
|
|
|
void populate(hostent* he) @system pure nothrow
|
|
{
|
|
int i;
|
|
char* p;
|
|
|
|
name = to!string(he.h_name);
|
|
|
|
for (i = 0;; i++)
|
|
{
|
|
p = he.h_aliases[i];
|
|
if (!p)
|
|
break;
|
|
}
|
|
|
|
if (i)
|
|
{
|
|
aliases = new string[i];
|
|
for (i = 0; i != aliases.length; i++)
|
|
{
|
|
aliases[i] =
|
|
to!string(he.h_aliases[i]);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
aliases = null;
|
|
}
|
|
|
|
for (i = 0;; i++)
|
|
{
|
|
p = he.h_addr_list[i];
|
|
if (!p)
|
|
break;
|
|
}
|
|
|
|
if (i)
|
|
{
|
|
addrList = new uint[i];
|
|
for (i = 0; i != addrList.length; i++)
|
|
{
|
|
addrList[i] = ntohl(*(cast(uint*) he.h_addr_list[i]));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
addrList = null;
|
|
}
|
|
}
|
|
|
|
private bool getHostNoSync(string opMixin, T)(T param) @system
|
|
{
|
|
mixin(opMixin);
|
|
if (!he)
|
|
return false;
|
|
validHostent(he);
|
|
populate(he);
|
|
return true;
|
|
}
|
|
|
|
version (Windows)
|
|
alias getHost = getHostNoSync;
|
|
else
|
|
{
|
|
// posix systems use global state for return value, so we
|
|
// must synchronize across all threads
|
|
private bool getHost(string opMixin, T)(T param) @system
|
|
{
|
|
synchronized(typeid(this))
|
|
return getHostNoSync!(opMixin, T)(param);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Resolve host name.
|
|
* Returns: false if unable to resolve.
|
|
*/
|
|
bool getHostByName(scope const(char)[] name) @trusted
|
|
{
|
|
static if (is(typeof(gethostbyname_r)))
|
|
{
|
|
return getHostNoSync!q{
|
|
hostent he_v;
|
|
hostent* he;
|
|
ubyte[256] buffer_v = void;
|
|
auto buffer = buffer_v[];
|
|
auto param_zTmp = param.tempCString();
|
|
while (true)
|
|
{
|
|
he = &he_v;
|
|
int errno;
|
|
if (gethostbyname_r(param_zTmp, he, buffer.ptr, buffer.length, &he, &errno) == ERANGE)
|
|
buffer.length = buffer.length * 2;
|
|
else
|
|
break;
|
|
}
|
|
}(name);
|
|
}
|
|
else
|
|
{
|
|
return getHost!q{
|
|
auto he = gethostbyname(param.tempCString());
|
|
}(name);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Resolve IPv4 address number.
|
|
*
|
|
* Params:
|
|
* addr = The IPv4 address to resolve, in host byte order.
|
|
* Returns:
|
|
* false if unable to resolve.
|
|
*/
|
|
bool getHostByAddr(uint addr) @trusted
|
|
{
|
|
return getHost!q{
|
|
auto x = htonl(param);
|
|
auto he = gethostbyaddr(&x, 4, cast(int) AddressFamily.INET);
|
|
}(addr);
|
|
}
|
|
|
|
/**
|
|
* Same as previous, but addr is an IPv4 address string in the
|
|
* dotted-decimal form $(I a.b.c.d).
|
|
* Returns: false if unable to resolve.
|
|
*/
|
|
bool getHostByAddr(scope const(char)[] addr) @trusted
|
|
{
|
|
return getHost!q{
|
|
auto x = inet_addr(param.tempCString());
|
|
enforce(x != INADDR_NONE,
|
|
new SocketParameterException("Invalid IPv4 address"));
|
|
auto he = gethostbyaddr(&x, 4, cast(int) AddressFamily.INET);
|
|
}(addr);
|
|
}
|
|
}
|
|
|
|
///
|
|
@safe unittest
|
|
{
|
|
InternetHost ih = new InternetHost;
|
|
|
|
ih.getHostByAddr(0x7F_00_00_01);
|
|
assert(ih.addrList[0] == 0x7F_00_00_01);
|
|
ih.getHostByAddr("127.0.0.1");
|
|
assert(ih.addrList[0] == 0x7F_00_00_01);
|
|
|
|
if (!ih.getHostByName("www.digitalmars.com"))
|
|
return; // don't fail if not connected to internet
|
|
|
|
assert(ih.addrList.length);
|
|
InternetAddress ia = new InternetAddress(ih.addrList[0], InternetAddress.PORT_ANY);
|
|
assert(ih.name == "www.digitalmars.com" || ih.name == "digitalmars.com",
|
|
ih.name);
|
|
|
|
/* The following assert randomly fails in the test suite.
|
|
* https://issues.dlang.org/show_bug.cgi?id=22791
|
|
* So just ignore it when it fails.
|
|
*/
|
|
//assert(ih.getHostByAddr(ih.addrList[0]));
|
|
if (ih.getHostByAddr(ih.addrList[0]))
|
|
{
|
|
string getHostNameFromInt = ih.name.dup;
|
|
|
|
// This randomly fails in the compiler test suite
|
|
//assert(ih.getHostByAddr(ia.toAddrString()));
|
|
|
|
if (ih.getHostByAddr(ia.toAddrString()))
|
|
{
|
|
string getHostNameFromStr = ih.name.dup;
|
|
assert(getHostNameFromInt == getHostNameFromStr);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/// Holds information about a socket _address retrieved by `getAddressInfo`.
|
|
struct AddressInfo
|
|
{
|
|
AddressFamily family; /// Address _family
|
|
SocketType type; /// Socket _type
|
|
ProtocolType protocol; /// Protocol
|
|
Address address; /// Socket _address
|
|
string canonicalName; /// Canonical name, when `AddressInfoFlags.CANONNAME` is used.
|
|
}
|
|
|
|
/**
|
|
* A subset of flags supported on all platforms with getaddrinfo.
|
|
* Specifies option flags for `getAddressInfo`.
|
|
*/
|
|
enum AddressInfoFlags: int
|
|
{
|
|
/// The resulting addresses will be used in a call to `Socket.bind`.
|
|
PASSIVE = AI_PASSIVE,
|
|
|
|
/// The canonical name is returned in `canonicalName` member in the first `AddressInfo`.
|
|
CANONNAME = AI_CANONNAME,
|
|
|
|
/**
|
|
* The `node` parameter passed to `getAddressInfo` must be a numeric string.
|
|
* This will suppress any potentially lengthy network host address lookups.
|
|
*/
|
|
NUMERICHOST = AI_NUMERICHOST,
|
|
}
|
|
|
|
|
|
/**
|
|
* On POSIX, getaddrinfo uses its own error codes, and thus has its own
|
|
* formatting function.
|
|
*/
|
|
private string formatGaiError(int err) @trusted
|
|
{
|
|
version (Windows)
|
|
{
|
|
return generateSysErrorMsg(err);
|
|
}
|
|
else
|
|
{
|
|
synchronized
|
|
return to!string(gai_strerror(err));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Provides _protocol-independent translation from host names to socket
|
|
* addresses. If advanced functionality is not required, consider using
|
|
* `getAddress` for compatibility with older systems.
|
|
*
|
|
* Returns: Array with one `AddressInfo` per socket address.
|
|
*
|
|
* Throws: `SocketOSException` on failure, or `SocketFeatureException`
|
|
* if this functionality is not available on the current system.
|
|
*
|
|
* Params:
|
|
* node = string containing host name or numeric address
|
|
* options = optional additional parameters, identified by type:
|
|
* $(UL $(LI `string` - service name or port number)
|
|
* $(LI `AddressInfoFlags` - option flags)
|
|
* $(LI `AddressFamily` - address family to filter by)
|
|
* $(LI `SocketType` - socket type to filter by)
|
|
* $(LI `ProtocolType` - protocol to filter by))
|
|
*
|
|
* Example:
|
|
* ---
|
|
* // Roundtrip DNS resolution
|
|
* auto results = getAddressInfo("www.digitalmars.com");
|
|
* assert(results[0].address.toHostNameString() ==
|
|
* "digitalmars.com");
|
|
*
|
|
* // Canonical name
|
|
* results = getAddressInfo("www.digitalmars.com",
|
|
* AddressInfoFlags.CANONNAME);
|
|
* assert(results[0].canonicalName == "digitalmars.com");
|
|
*
|
|
* // IPv6 resolution
|
|
* results = getAddressInfo("ipv6.google.com");
|
|
* assert(results[0].family == AddressFamily.INET6);
|
|
*
|
|
* // Multihomed resolution
|
|
* results = getAddressInfo("google.com");
|
|
* assert(results.length > 1);
|
|
*
|
|
* // Parsing IPv4
|
|
* results = getAddressInfo("127.0.0.1",
|
|
* AddressInfoFlags.NUMERICHOST);
|
|
* assert(results.length && results[0].family ==
|
|
* AddressFamily.INET);
|
|
*
|
|
* // Parsing IPv6
|
|
* results = getAddressInfo("::1",
|
|
* AddressInfoFlags.NUMERICHOST);
|
|
* assert(results.length && results[0].family ==
|
|
* AddressFamily.INET6);
|
|
* ---
|
|
*/
|
|
AddressInfo[] getAddressInfo(T...)(scope const(char)[] node, scope T options)
|
|
{
|
|
const(char)[] service = null;
|
|
addrinfo hints;
|
|
hints.ai_family = AF_UNSPEC;
|
|
|
|
foreach (i, option; options)
|
|
{
|
|
static if (is(typeof(option) : const(char)[]))
|
|
service = options[i];
|
|
else
|
|
static if (is(typeof(option) == AddressInfoFlags))
|
|
hints.ai_flags |= option;
|
|
else
|
|
static if (is(typeof(option) == AddressFamily))
|
|
hints.ai_family = option;
|
|
else
|
|
static if (is(typeof(option) == SocketType))
|
|
hints.ai_socktype = option;
|
|
else
|
|
static if (is(typeof(option) == ProtocolType))
|
|
hints.ai_protocol = option;
|
|
else
|
|
static assert(0, "Unknown getAddressInfo option type: " ~ typeof(option).stringof);
|
|
}
|
|
|
|
return () @trusted { return getAddressInfoImpl(node, service, &hints); }();
|
|
}
|
|
|
|
@system unittest
|
|
{
|
|
struct Oops
|
|
{
|
|
const(char[]) breakSafety()
|
|
{
|
|
*cast(int*) 0xcafebabe = 0xdeadbeef;
|
|
return null;
|
|
}
|
|
alias breakSafety this;
|
|
}
|
|
assert(!__traits(compiles, () {
|
|
getAddressInfo("", Oops.init);
|
|
}), "getAddressInfo breaks @safe");
|
|
}
|
|
|
|
private AddressInfo[] getAddressInfoImpl(scope const(char)[] node, scope const(char)[] service, addrinfo* hints) @system
|
|
{
|
|
import std.array : appender;
|
|
|
|
if (getaddrinfoPointer && freeaddrinfoPointer)
|
|
{
|
|
addrinfo* ai_res;
|
|
|
|
int ret = getaddrinfoPointer(
|
|
node.tempCString(),
|
|
service.tempCString(),
|
|
hints, &ai_res);
|
|
enforce(ret == 0, new SocketOSException("getaddrinfo error", ret, &formatGaiError));
|
|
scope(exit) freeaddrinfoPointer(ai_res);
|
|
|
|
auto result = appender!(AddressInfo[])();
|
|
|
|
// Use const to force UnknownAddressReference to copy the sockaddr.
|
|
for (const(addrinfo)* ai = ai_res; ai; ai = ai.ai_next)
|
|
result ~= AddressInfo(
|
|
cast(AddressFamily) ai.ai_family,
|
|
cast(SocketType ) ai.ai_socktype,
|
|
cast(ProtocolType ) ai.ai_protocol,
|
|
new UnknownAddressReference(ai.ai_addr, cast(socklen_t) ai.ai_addrlen),
|
|
ai.ai_canonname ? to!string(ai.ai_canonname) : null);
|
|
|
|
assert(result.data.length > 0);
|
|
return result.data;
|
|
}
|
|
|
|
throw new SocketFeatureException("Address info lookup is not available " ~
|
|
"on this system.");
|
|
}
|
|
|
|
|
|
@safe unittest
|
|
{
|
|
softUnittest({
|
|
if (getaddrinfoPointer)
|
|
{
|
|
// Roundtrip DNS resolution
|
|
auto results = getAddressInfo("www.digitalmars.com");
|
|
assert(results[0].address.toHostNameString() == "digitalmars.com");
|
|
|
|
// Canonical name
|
|
results = getAddressInfo("www.digitalmars.com",
|
|
AddressInfoFlags.CANONNAME);
|
|
assert(results[0].canonicalName == "digitalmars.com");
|
|
|
|
// IPv6 resolution
|
|
//results = getAddressInfo("ipv6.google.com");
|
|
//assert(results[0].family == AddressFamily.INET6);
|
|
|
|
// Multihomed resolution
|
|
//results = getAddressInfo("google.com");
|
|
//assert(results.length > 1);
|
|
|
|
// Parsing IPv4
|
|
results = getAddressInfo("127.0.0.1", AddressInfoFlags.NUMERICHOST);
|
|
assert(results.length && results[0].family == AddressFamily.INET);
|
|
|
|
// Parsing IPv6
|
|
results = getAddressInfo("::1", AddressInfoFlags.NUMERICHOST);
|
|
assert(results.length && results[0].family == AddressFamily.INET6);
|
|
}
|
|
});
|
|
|
|
if (getaddrinfoPointer)
|
|
{
|
|
auto results = getAddressInfo(null, "1234", AddressInfoFlags.PASSIVE,
|
|
SocketType.STREAM, ProtocolType.TCP, AddressFamily.INET);
|
|
assert(results.length == 1 && results[0].address.toString() == "0.0.0.0:1234");
|
|
}
|
|
}
|
|
|
|
|
|
private ushort serviceToPort(scope const(char)[] service)
|
|
{
|
|
if (service == "")
|
|
return InternetAddress.PORT_ANY;
|
|
else
|
|
if (isNumeric(service))
|
|
return to!ushort(service);
|
|
else
|
|
{
|
|
auto s = new Service();
|
|
s.getServiceByName(service);
|
|
return s.port;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Provides _protocol-independent translation from host names to socket
|
|
* addresses. Uses `getAddressInfo` if the current system supports it,
|
|
* and `InternetHost` otherwise.
|
|
*
|
|
* Returns: Array with one `Address` instance per socket address.
|
|
*
|
|
* Throws: `SocketOSException` on failure.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* writeln("Resolving www.digitalmars.com:");
|
|
* try
|
|
* {
|
|
* auto addresses = getAddress("www.digitalmars.com");
|
|
* foreach (address; addresses)
|
|
* writefln(" IP: %s", address.toAddrString());
|
|
* }
|
|
* catch (SocketException e)
|
|
* writefln(" Lookup failed: %s", e.msg);
|
|
* ---
|
|
*/
|
|
Address[] getAddress(scope const(char)[] hostname, scope const(char)[] service = null)
|
|
{
|
|
if (getaddrinfoPointer && freeaddrinfoPointer)
|
|
{
|
|
// use getAddressInfo
|
|
auto infos = getAddressInfo(hostname, service);
|
|
Address[] results;
|
|
results.length = infos.length;
|
|
foreach (i, ref result; results)
|
|
result = infos[i].address;
|
|
return results;
|
|
}
|
|
else
|
|
return getAddress(hostname, serviceToPort(service));
|
|
}
|
|
|
|
/// ditto
|
|
Address[] getAddress(scope const(char)[] hostname, ushort port)
|
|
{
|
|
if (getaddrinfoPointer && freeaddrinfoPointer)
|
|
return getAddress(hostname, to!string(port));
|
|
else
|
|
{
|
|
// use getHostByName
|
|
auto ih = new InternetHost;
|
|
if (!ih.getHostByName(hostname))
|
|
throw new AddressException(
|
|
text("Unable to resolve host '", hostname, "'"));
|
|
|
|
Address[] results;
|
|
foreach (uint addr; ih.addrList)
|
|
results ~= new InternetAddress(addr, port);
|
|
return results;
|
|
}
|
|
}
|
|
|
|
|
|
@safe unittest
|
|
{
|
|
softUnittest({
|
|
auto addresses = getAddress("63.105.9.61");
|
|
assert(addresses.length && addresses[0].toAddrString() == "63.105.9.61");
|
|
|
|
if (getaddrinfoPointer)
|
|
{
|
|
// test via gethostbyname
|
|
auto getaddrinfoPointerBackup = getaddrinfoPointer;
|
|
cast() getaddrinfoPointer = null;
|
|
scope(exit) () @trusted { cast() getaddrinfoPointer = getaddrinfoPointerBackup; }();
|
|
|
|
addresses = getAddress("63.105.9.61");
|
|
assert(addresses.length && addresses[0].toAddrString() == "63.105.9.61");
|
|
}
|
|
});
|
|
}
|
|
|
|
|
|
/**
|
|
* Provides _protocol-independent parsing of network addresses. Does not
|
|
* attempt name resolution. Uses `getAddressInfo` with
|
|
* `AddressInfoFlags.NUMERICHOST` if the current system supports it, and
|
|
* `InternetAddress` otherwise.
|
|
*
|
|
* Returns: An `Address` instance representing specified address.
|
|
*
|
|
* Throws: `SocketException` on failure.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* writeln("Enter IP address:");
|
|
* string ip = readln().chomp();
|
|
* try
|
|
* {
|
|
* Address address = parseAddress(ip);
|
|
* writefln("Looking up reverse of %s:",
|
|
* address.toAddrString());
|
|
* try
|
|
* {
|
|
* string reverse = address.toHostNameString();
|
|
* if (reverse)
|
|
* writefln(" Reverse name: %s", reverse);
|
|
* else
|
|
* writeln(" Reverse hostname not found.");
|
|
* }
|
|
* catch (SocketException e)
|
|
* writefln(" Lookup error: %s", e.msg);
|
|
* }
|
|
* catch (SocketException e)
|
|
* {
|
|
* writefln(" %s is not a valid IP address: %s",
|
|
* ip, e.msg);
|
|
* }
|
|
* ---
|
|
*/
|
|
Address parseAddress(scope const(char)[] hostaddr, scope const(char)[] service = null)
|
|
{
|
|
if (getaddrinfoPointer && freeaddrinfoPointer)
|
|
return getAddressInfo(hostaddr, service, AddressInfoFlags.NUMERICHOST)[0].address;
|
|
else
|
|
return parseAddress(hostaddr, serviceToPort(service));
|
|
}
|
|
|
|
/// ditto
|
|
Address parseAddress(scope const(char)[] hostaddr, ushort port)
|
|
{
|
|
if (getaddrinfoPointer && freeaddrinfoPointer)
|
|
return parseAddress(hostaddr, to!string(port));
|
|
else
|
|
{
|
|
auto in4_addr = InternetAddress.parse(hostaddr);
|
|
enforce(in4_addr != InternetAddress.ADDR_NONE,
|
|
new SocketParameterException("Invalid IP address"));
|
|
return new InternetAddress(in4_addr, port);
|
|
}
|
|
}
|
|
|
|
|
|
@safe unittest
|
|
{
|
|
softUnittest({
|
|
auto address = parseAddress("63.105.9.61");
|
|
assert(address.toAddrString() == "63.105.9.61");
|
|
|
|
if (getaddrinfoPointer)
|
|
{
|
|
// test via inet_addr
|
|
auto getaddrinfoPointerBackup = getaddrinfoPointer;
|
|
cast() getaddrinfoPointer = null;
|
|
scope(exit) () @trusted { cast() getaddrinfoPointer = getaddrinfoPointerBackup; }();
|
|
|
|
address = parseAddress("63.105.9.61");
|
|
assert(address.toAddrString() == "63.105.9.61");
|
|
}
|
|
|
|
assert(collectException!SocketException(parseAddress("Invalid IP address")));
|
|
});
|
|
}
|
|
|
|
|
|
/**
|
|
* Class for exceptions thrown from an `Address`.
|
|
*/
|
|
class AddressException: SocketOSException
|
|
{
|
|
mixin socketOSExceptionCtors;
|
|
}
|
|
|
|
|
|
/**
|
|
* Abstract class for representing a socket address.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* writeln("About www.google.com port 80:");
|
|
* try
|
|
* {
|
|
* Address[] addresses = getAddress("www.google.com", 80);
|
|
* writefln(" %d addresses found.", addresses.length);
|
|
* foreach (int i, Address a; addresses)
|
|
* {
|
|
* writefln(" Address %d:", i+1);
|
|
* writefln(" IP address: %s", a.toAddrString());
|
|
* writefln(" Hostname: %s", a.toHostNameString());
|
|
* writefln(" Port: %s", a.toPortString());
|
|
* writefln(" Service name: %s",
|
|
* a.toServiceNameString());
|
|
* }
|
|
* }
|
|
* catch (SocketException e)
|
|
* writefln(" Lookup error: %s", e.msg);
|
|
* ---
|
|
*/
|
|
abstract class Address
|
|
{
|
|
/// Returns pointer to underlying `sockaddr` structure.
|
|
abstract @property sockaddr* name() pure nothrow @nogc;
|
|
abstract @property const(sockaddr)* name() const pure nothrow @nogc; /// ditto
|
|
|
|
/// Returns actual size of underlying `sockaddr` structure.
|
|
abstract @property socklen_t nameLen() const pure nothrow @nogc;
|
|
|
|
// Socket.remoteAddress, Socket.localAddress, and Socket.receiveFrom
|
|
// use setNameLen to set the actual size of the address as returned by
|
|
// getsockname, getpeername, and recvfrom, respectively.
|
|
// The following implementation is sufficient for fixed-length addresses,
|
|
// and ensures that the length is not changed.
|
|
// Must be overridden for variable-length addresses.
|
|
protected void setNameLen(socklen_t len)
|
|
{
|
|
if (len != this.nameLen)
|
|
throw new AddressException(
|
|
format("%s expects address of length %d, not %d", typeid(this),
|
|
this.nameLen, len), 0);
|
|
}
|
|
|
|
/// Family of this address.
|
|
@property AddressFamily addressFamily() const pure nothrow @nogc
|
|
{
|
|
return cast(AddressFamily) name.sa_family;
|
|
}
|
|
|
|
// Common code for toAddrString and toHostNameString
|
|
private string toHostString(bool numeric) @trusted const
|
|
{
|
|
// getnameinfo() is the recommended way to perform a reverse (name)
|
|
// lookup on both Posix and Windows. However, it is only available
|
|
// on Windows XP and above, and not included with the WinSock import
|
|
// libraries shipped with DMD. Thus, we check for getnameinfo at
|
|
// runtime in the shared module constructor, and use it if it's
|
|
// available in the base class method. Classes for specific network
|
|
// families (e.g. InternetHost) override this method and use a
|
|
// deprecated, albeit commonly-available method when getnameinfo()
|
|
// is not available.
|
|
// http://technet.microsoft.com/en-us/library/aa450403.aspx
|
|
if (getnameinfoPointer)
|
|
{
|
|
auto buf = new char[NI_MAXHOST];
|
|
auto ret = getnameinfoPointer(
|
|
name, nameLen,
|
|
buf.ptr, cast(uint) buf.length,
|
|
null, 0,
|
|
numeric ? NI_NUMERICHOST : NI_NAMEREQD);
|
|
|
|
if (!numeric)
|
|
{
|
|
if (ret == EAI_NONAME)
|
|
return null;
|
|
version (Windows)
|
|
if (ret == WSANO_DATA)
|
|
return null;
|
|
}
|
|
|
|
enforce(ret == 0, new AddressException("Could not get " ~
|
|
(numeric ? "host address" : "host name")));
|
|
return assumeUnique(buf[0 .. strlen(buf.ptr)]);
|
|
}
|
|
|
|
throw new SocketFeatureException((numeric ? "Host address" : "Host name") ~
|
|
" lookup for this address family is not available on this system.");
|
|
}
|
|
|
|
// Common code for toPortString and toServiceNameString
|
|
private string toServiceString(bool numeric) @trusted const
|
|
{
|
|
// See toHostNameString() for details about getnameinfo().
|
|
if (getnameinfoPointer)
|
|
{
|
|
auto buf = new char[NI_MAXSERV];
|
|
enforce(getnameinfoPointer(
|
|
name, nameLen,
|
|
null, 0,
|
|
buf.ptr, cast(uint) buf.length,
|
|
numeric ? NI_NUMERICSERV : NI_NAMEREQD
|
|
) == 0, new AddressException("Could not get " ~
|
|
(numeric ? "port number" : "service name")));
|
|
return assumeUnique(buf[0 .. strlen(buf.ptr)]);
|
|
}
|
|
|
|
throw new SocketFeatureException((numeric ? "Port number" : "Service name") ~
|
|
" lookup for this address family is not available on this system.");
|
|
}
|
|
|
|
/**
|
|
* Attempts to retrieve the host address as a human-readable string.
|
|
*
|
|
* Throws: `AddressException` on failure, or `SocketFeatureException`
|
|
* if address retrieval for this address family is not available on the
|
|
* current system.
|
|
*/
|
|
string toAddrString() const
|
|
{
|
|
return toHostString(true);
|
|
}
|
|
|
|
/**
|
|
* Attempts to retrieve the host name as a fully qualified domain name.
|
|
*
|
|
* Returns: The FQDN corresponding to this `Address`, or `null` if
|
|
* the host name did not resolve.
|
|
*
|
|
* Throws: `AddressException` on error, or `SocketFeatureException`
|
|
* if host name lookup for this address family is not available on the
|
|
* current system.
|
|
*/
|
|
string toHostNameString() const
|
|
{
|
|
return toHostString(false);
|
|
}
|
|
|
|
/**
|
|
* Attempts to retrieve the numeric port number as a string.
|
|
*
|
|
* Throws: `AddressException` on failure, or `SocketFeatureException`
|
|
* if port number retrieval for this address family is not available on the
|
|
* current system.
|
|
*/
|
|
string toPortString() const
|
|
{
|
|
return toServiceString(true);
|
|
}
|
|
|
|
/**
|
|
* Attempts to retrieve the service name as a string.
|
|
*
|
|
* Throws: `AddressException` on failure, or `SocketFeatureException`
|
|
* if service name lookup for this address family is not available on the
|
|
* current system.
|
|
*/
|
|
string toServiceNameString() const
|
|
{
|
|
return toServiceString(false);
|
|
}
|
|
|
|
/// Human readable string representing this address.
|
|
override string toString() const
|
|
{
|
|
try
|
|
{
|
|
string host = toAddrString();
|
|
string port = toPortString();
|
|
if (host.indexOf(':') >= 0)
|
|
return "[" ~ host ~ "]:" ~ port;
|
|
else
|
|
return host ~ ":" ~ port;
|
|
}
|
|
catch (SocketException)
|
|
return "Unknown";
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Encapsulates an unknown socket address.
|
|
*/
|
|
class UnknownAddress: Address
|
|
{
|
|
protected:
|
|
sockaddr sa;
|
|
|
|
|
|
public:
|
|
override @property sockaddr* name() return
|
|
{
|
|
return &sa;
|
|
}
|
|
|
|
override @property const(sockaddr)* name() const return
|
|
{
|
|
return &sa;
|
|
}
|
|
|
|
|
|
override @property socklen_t nameLen() const
|
|
{
|
|
return cast(socklen_t) sa.sizeof;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
* Encapsulates a reference to an arbitrary
|
|
* socket address.
|
|
*/
|
|
class UnknownAddressReference: Address
|
|
{
|
|
protected:
|
|
sockaddr* sa;
|
|
socklen_t len;
|
|
|
|
public:
|
|
/// Constructs an `Address` with a reference to the specified `sockaddr`.
|
|
this(sockaddr* sa, socklen_t len) pure nothrow @nogc
|
|
{
|
|
this.sa = sa;
|
|
this.len = len;
|
|
}
|
|
|
|
/// Constructs an `Address` with a copy of the specified `sockaddr`.
|
|
this(const(sockaddr)* sa, socklen_t len) @system pure nothrow
|
|
{
|
|
this.sa = cast(sockaddr*) (cast(ubyte*) sa)[0 .. len].dup.ptr;
|
|
this.len = len;
|
|
}
|
|
|
|
override @property sockaddr* name()
|
|
{
|
|
return sa;
|
|
}
|
|
|
|
override @property const(sockaddr)* name() const
|
|
{
|
|
return sa;
|
|
}
|
|
|
|
|
|
override @property socklen_t nameLen() const
|
|
{
|
|
return cast(socklen_t) len;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Encapsulates an IPv4 (Internet Protocol version 4) socket address.
|
|
*
|
|
* Consider using `getAddress`, `parseAddress` and `Address` methods
|
|
* instead of using this class directly.
|
|
*/
|
|
class InternetAddress: Address
|
|
{
|
|
protected:
|
|
sockaddr_in sin;
|
|
|
|
|
|
this() pure nothrow @nogc
|
|
{
|
|
}
|
|
|
|
|
|
public:
|
|
override @property sockaddr* name() return
|
|
{
|
|
return cast(sockaddr*)&sin;
|
|
}
|
|
|
|
override @property const(sockaddr)* name() const return
|
|
{
|
|
return cast(const(sockaddr)*)&sin;
|
|
}
|
|
|
|
|
|
override @property socklen_t nameLen() const
|
|
{
|
|
return cast(socklen_t) sin.sizeof;
|
|
}
|
|
|
|
|
|
enum uint ADDR_ANY = INADDR_ANY; /// Any IPv4 host address.
|
|
enum uint ADDR_NONE = INADDR_NONE; /// An invalid IPv4 host address.
|
|
enum ushort PORT_ANY = 0; /// Any IPv4 port number.
|
|
|
|
/// Returns the IPv4 _port number (in host byte order).
|
|
@property ushort port() const pure nothrow @nogc
|
|
{
|
|
return ntohs(sin.sin_port);
|
|
}
|
|
|
|
/// Returns the IPv4 address number (in host byte order).
|
|
@property uint addr() const pure nothrow @nogc
|
|
{
|
|
return ntohl(sin.sin_addr.s_addr);
|
|
}
|
|
|
|
/**
|
|
* Construct a new `InternetAddress`.
|
|
* Params:
|
|
* addr = an IPv4 address string in the dotted-decimal form a.b.c.d,
|
|
* or a host name which will be resolved using an `InternetHost`
|
|
* object.
|
|
* port = port number, may be `PORT_ANY`.
|
|
*/
|
|
this(scope const(char)[] addr, ushort port)
|
|
{
|
|
uint uiaddr = parse(addr);
|
|
if (ADDR_NONE == uiaddr)
|
|
{
|
|
InternetHost ih = new InternetHost;
|
|
if (!ih.getHostByName(addr))
|
|
//throw new AddressException("Invalid internet address");
|
|
throw new AddressException(
|
|
text("Unable to resolve host '", addr, "'"));
|
|
uiaddr = ih.addrList[0];
|
|
}
|
|
sin.sin_family = AddressFamily.INET;
|
|
sin.sin_addr.s_addr = htonl(uiaddr);
|
|
sin.sin_port = htons(port);
|
|
}
|
|
|
|
/**
|
|
* Construct a new `InternetAddress`.
|
|
* Params:
|
|
* addr = (optional) an IPv4 address in host byte order, may be `ADDR_ANY`.
|
|
* port = port number, may be `PORT_ANY`.
|
|
*/
|
|
this(uint addr, ushort port) pure nothrow @nogc
|
|
{
|
|
sin.sin_family = AddressFamily.INET;
|
|
sin.sin_addr.s_addr = htonl(addr);
|
|
sin.sin_port = htons(port);
|
|
}
|
|
|
|
/// ditto
|
|
this(ushort port) pure nothrow @nogc
|
|
{
|
|
sin.sin_family = AddressFamily.INET;
|
|
sin.sin_addr.s_addr = ADDR_ANY;
|
|
sin.sin_port = htons(port);
|
|
}
|
|
|
|
/**
|
|
* Construct a new `InternetAddress`.
|
|
* Params:
|
|
* addr = A sockaddr_in as obtained from lower-level API calls such as getifaddrs.
|
|
*/
|
|
this(sockaddr_in addr) pure nothrow @nogc
|
|
{
|
|
assert(addr.sin_family == AddressFamily.INET, "Socket address is not of INET family.");
|
|
sin = addr;
|
|
}
|
|
|
|
/// Human readable string representing the IPv4 address in dotted-decimal form.
|
|
override string toAddrString() @trusted const
|
|
{
|
|
return to!string(inet_ntoa(sin.sin_addr));
|
|
}
|
|
|
|
/// Human readable string representing the IPv4 port.
|
|
override string toPortString() const
|
|
{
|
|
return std.conv.to!string(port);
|
|
}
|
|
|
|
/**
|
|
* Attempts to retrieve the host name as a fully qualified domain name.
|
|
*
|
|
* Returns: The FQDN corresponding to this `InternetAddress`, or
|
|
* `null` if the host name did not resolve.
|
|
*
|
|
* Throws: `AddressException` on error.
|
|
*/
|
|
override string toHostNameString() const
|
|
{
|
|
// getnameinfo() is the recommended way to perform a reverse (name)
|
|
// lookup on both Posix and Windows. However, it is only available
|
|
// on Windows XP and above, and not included with the WinSock import
|
|
// libraries shipped with DMD. Thus, we check for getnameinfo at
|
|
// runtime in the shared module constructor, and fall back to the
|
|
// deprecated getHostByAddr() if it could not be found. See also:
|
|
// http://technet.microsoft.com/en-us/library/aa450403.aspx
|
|
|
|
if (getnameinfoPointer)
|
|
return super.toHostNameString();
|
|
else
|
|
{
|
|
auto host = new InternetHost();
|
|
if (!host.getHostByAddr(ntohl(sin.sin_addr.s_addr)))
|
|
return null;
|
|
return host.name;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Provides support for comparing equality with another
|
|
* InternetAddress of the same type.
|
|
* Returns: true if the InternetAddresses share the same address and
|
|
* port number.
|
|
*/
|
|
override bool opEquals(Object o) const
|
|
{
|
|
auto other = cast(InternetAddress) o;
|
|
return other && this.sin.sin_addr.s_addr == other.sin.sin_addr.s_addr &&
|
|
this.sin.sin_port == other.sin.sin_port;
|
|
}
|
|
|
|
///
|
|
@system unittest
|
|
{
|
|
auto addr1 = new InternetAddress("127.0.0.1", 80);
|
|
auto addr2 = new InternetAddress("127.0.0.2", 80);
|
|
|
|
assert(addr1 == addr1);
|
|
assert(addr1 != addr2);
|
|
}
|
|
|
|
/**
|
|
* Parse an IPv4 address string in the dotted-decimal form $(I a.b.c.d)
|
|
* and return the number.
|
|
* Returns: If the string is not a legitimate IPv4 address,
|
|
* `ADDR_NONE` is returned.
|
|
*/
|
|
static uint parse(scope const(char)[] addr) @trusted nothrow
|
|
{
|
|
return ntohl(inet_addr(addr.tempCString()));
|
|
}
|
|
|
|
/**
|
|
* Convert an IPv4 address number in host byte order to a human readable
|
|
* string representing the IPv4 address in dotted-decimal form.
|
|
*/
|
|
static string addrToString(uint addr) @trusted nothrow
|
|
{
|
|
in_addr sin_addr;
|
|
sin_addr.s_addr = htonl(addr);
|
|
return to!string(inet_ntoa(sin_addr));
|
|
}
|
|
}
|
|
|
|
|
|
@safe unittest
|
|
{
|
|
softUnittest({
|
|
const InternetAddress ia = new InternetAddress("63.105.9.61", 80);
|
|
assert(ia.toString() == "63.105.9.61:80");
|
|
});
|
|
|
|
softUnittest({
|
|
// test construction from a sockaddr_in
|
|
sockaddr_in sin;
|
|
|
|
sin.sin_addr.s_addr = htonl(0x7F_00_00_01); // 127.0.0.1
|
|
sin.sin_family = AddressFamily.INET;
|
|
sin.sin_port = htons(80);
|
|
|
|
const InternetAddress ia = new InternetAddress(sin);
|
|
assert(ia.toString() == "127.0.0.1:80");
|
|
});
|
|
|
|
softUnittest({
|
|
// test reverse lookup
|
|
auto ih = new InternetHost;
|
|
if (ih.getHostByName("digitalmars.com"))
|
|
{
|
|
const ia = new InternetAddress(ih.addrList[0], 80);
|
|
assert(ia.toHostNameString() == "digitalmars.com");
|
|
|
|
if (getnameinfoPointer)
|
|
{
|
|
// test reverse lookup, via gethostbyaddr
|
|
auto getnameinfoPointerBackup = getnameinfoPointer;
|
|
cast() getnameinfoPointer = null;
|
|
scope(exit) () @trusted { cast() getnameinfoPointer = getnameinfoPointerBackup; }();
|
|
|
|
assert(ia.toHostNameString() == "digitalmars.com");
|
|
}
|
|
}
|
|
});
|
|
|
|
if (runSlowTests)
|
|
softUnittest({
|
|
// test failing reverse lookup
|
|
const InternetAddress ia = new InternetAddress("255.255.255.255", 80);
|
|
assert(ia.toHostNameString() is null);
|
|
|
|
if (getnameinfoPointer)
|
|
{
|
|
// test failing reverse lookup, via gethostbyaddr
|
|
auto getnameinfoPointerBackup = getnameinfoPointer;
|
|
cast() getnameinfoPointer = null;
|
|
scope(exit) () @trusted { cast() getnameinfoPointer = getnameinfoPointerBackup; }();
|
|
|
|
assert(ia.toHostNameString() is null);
|
|
}
|
|
});
|
|
}
|
|
|
|
|
|
/**
|
|
* Encapsulates an IPv6 (Internet Protocol version 6) socket address.
|
|
*
|
|
* Consider using `getAddress`, `parseAddress` and `Address` methods
|
|
* instead of using this class directly.
|
|
*/
|
|
class Internet6Address: Address
|
|
{
|
|
protected:
|
|
sockaddr_in6 sin6;
|
|
|
|
|
|
this() pure nothrow @nogc
|
|
{
|
|
}
|
|
|
|
|
|
public:
|
|
override @property sockaddr* name() return
|
|
{
|
|
return cast(sockaddr*)&sin6;
|
|
}
|
|
|
|
override @property const(sockaddr)* name() const return
|
|
{
|
|
return cast(const(sockaddr)*)&sin6;
|
|
}
|
|
|
|
|
|
override @property socklen_t nameLen() const
|
|
{
|
|
return cast(socklen_t) sin6.sizeof;
|
|
}
|
|
|
|
|
|
/// Any IPv6 host address.
|
|
static @property ref const(ubyte)[16] ADDR_ANY() pure nothrow @nogc
|
|
{
|
|
static if (is(typeof(IN6ADDR_ANY)))
|
|
{
|
|
version (Windows)
|
|
{
|
|
static immutable addr = IN6ADDR_ANY.s6_addr;
|
|
return addr;
|
|
}
|
|
else
|
|
return IN6ADDR_ANY.s6_addr;
|
|
}
|
|
else static if (is(typeof(in6addr_any)))
|
|
{
|
|
return in6addr_any.s6_addr;
|
|
}
|
|
else
|
|
static assert(0);
|
|
}
|
|
|
|
/// Any IPv6 port number.
|
|
enum ushort PORT_ANY = 0;
|
|
|
|
/// Returns the IPv6 port number.
|
|
@property ushort port() const pure nothrow @nogc
|
|
{
|
|
return ntohs(sin6.sin6_port);
|
|
}
|
|
|
|
/// Returns the IPv6 address.
|
|
@property ubyte[16] addr() const pure nothrow @nogc
|
|
{
|
|
return sin6.sin6_addr.s6_addr;
|
|
}
|
|
|
|
/**
|
|
* Construct a new `Internet6Address`.
|
|
* Params:
|
|
* addr = an IPv6 host address string in the form described in RFC 2373,
|
|
* or a host name which will be resolved using `getAddressInfo`.
|
|
* service = (optional) service name.
|
|
*/
|
|
this(scope const(char)[] addr, scope const(char)[] service = null) @trusted
|
|
{
|
|
auto results = getAddressInfo(addr, service, AddressFamily.INET6);
|
|
assert(results.length && results[0].family == AddressFamily.INET6);
|
|
sin6 = *cast(sockaddr_in6*) results[0].address.name;
|
|
}
|
|
|
|
/**
|
|
* Construct a new `Internet6Address`.
|
|
* Params:
|
|
* addr = an IPv6 host address string in the form described in RFC 2373,
|
|
* or a host name which will be resolved using `getAddressInfo`.
|
|
* port = port number, may be `PORT_ANY`.
|
|
*/
|
|
this(scope const(char)[] addr, ushort port)
|
|
{
|
|
if (port == PORT_ANY)
|
|
this(addr);
|
|
else
|
|
this(addr, to!string(port));
|
|
}
|
|
|
|
/**
|
|
* Construct a new `Internet6Address`.
|
|
* Params:
|
|
* addr = (optional) an IPv6 host address in host byte order, or
|
|
* `ADDR_ANY`.
|
|
* port = port number, may be `PORT_ANY`.
|
|
*/
|
|
this(ubyte[16] addr, ushort port) pure nothrow @nogc
|
|
{
|
|
sin6.sin6_family = AddressFamily.INET6;
|
|
sin6.sin6_addr.s6_addr = addr;
|
|
sin6.sin6_port = htons(port);
|
|
}
|
|
|
|
/// ditto
|
|
this(ushort port) pure nothrow @nogc
|
|
{
|
|
sin6.sin6_family = AddressFamily.INET6;
|
|
sin6.sin6_addr.s6_addr = ADDR_ANY;
|
|
sin6.sin6_port = htons(port);
|
|
}
|
|
|
|
/**
|
|
* Construct a new `Internet6Address`.
|
|
* Params:
|
|
* addr = A sockaddr_in6 as obtained from lower-level API calls such as getifaddrs.
|
|
*/
|
|
this(sockaddr_in6 addr) pure nothrow @nogc
|
|
{
|
|
assert(addr.sin6_family == AddressFamily.INET6);
|
|
sin6 = addr;
|
|
}
|
|
|
|
/**
|
|
* Parse an IPv6 host address string as described in RFC 2373, and return the
|
|
* address.
|
|
* Throws: `SocketException` on error.
|
|
*/
|
|
static ubyte[16] parse(scope const(char)[] addr) @trusted
|
|
{
|
|
// Although we could use inet_pton here, it's only available on Windows
|
|
// versions starting with Vista, so use getAddressInfo with NUMERICHOST
|
|
// instead.
|
|
auto results = getAddressInfo(addr, AddressInfoFlags.NUMERICHOST);
|
|
if (results.length && results[0].family == AddressFamily.INET6)
|
|
return (cast(sockaddr_in6*) results[0].address.name).sin6_addr.s6_addr;
|
|
throw new AddressException("Not an IPv6 address", 0);
|
|
}
|
|
}
|
|
|
|
|
|
@safe unittest
|
|
{
|
|
softUnittest({
|
|
const Internet6Address ia = new Internet6Address("::1", 80);
|
|
assert(ia.toString() == "[::1]:80");
|
|
});
|
|
|
|
softUnittest({
|
|
// test construction from a sockaddr_in6
|
|
sockaddr_in6 sin;
|
|
|
|
sin.sin6_addr.s6_addr = [ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1]; // [::1]
|
|
sin.sin6_family = AddressFamily.INET6;
|
|
sin.sin6_port = htons(80);
|
|
|
|
const Internet6Address ia = new Internet6Address(sin);
|
|
assert(ia.toString() == "[::1]:80");
|
|
});
|
|
}
|
|
|
|
|
|
version (StdDdoc)
|
|
{
|
|
static if (!is(sockaddr_un))
|
|
{
|
|
// This exists only to allow the constructor taking
|
|
// a sockaddr_un to be compilable for documentation
|
|
// on platforms that don't supply a sockaddr_un.
|
|
struct sockaddr_un
|
|
{
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Encapsulates an address for a Unix domain socket (`AF_UNIX`),
|
|
* i.e. a socket bound to a path name in the file system.
|
|
* Available only on supported systems.
|
|
*
|
|
* Linux also supports an abstract address namespace, in which addresses
|
|
* are independent of the file system. A socket address is abstract
|
|
* iff `path` starts with a _null byte (`'\0'`). Null bytes in other
|
|
* positions of an abstract address are allowed and have no special
|
|
* meaning.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* auto addr = new UnixAddress("/var/run/dbus/system_bus_socket");
|
|
* auto abstractAddr = new UnixAddress("\0/tmp/dbus-OtHLWmCLPR");
|
|
* ---
|
|
*
|
|
* See_Also: $(HTTP man7.org/linux/man-pages/man7/unix.7.html, UNIX(7))
|
|
*/
|
|
class UnixAddress: Address
|
|
{
|
|
private this() pure nothrow @nogc {}
|
|
|
|
/// Construct a new `UnixAddress` from the specified path.
|
|
this(scope const(char)[] path) { }
|
|
|
|
/**
|
|
* Construct a new `UnixAddress`.
|
|
* Params:
|
|
* addr = A sockaddr_un as obtained from lower-level API calls.
|
|
*/
|
|
this(sockaddr_un addr) pure nothrow @nogc { }
|
|
|
|
/// Get the underlying _path.
|
|
@property string path() const { return null; }
|
|
|
|
/// ditto
|
|
override string toString() const { return null; }
|
|
|
|
override @property sockaddr* name() { return null; }
|
|
override @property const(sockaddr)* name() const { return null; }
|
|
override @property socklen_t nameLen() const { return 0; }
|
|
}
|
|
}
|
|
else
|
|
static if (is(sockaddr_un))
|
|
{
|
|
class UnixAddress: Address
|
|
{
|
|
protected:
|
|
socklen_t _nameLen;
|
|
|
|
struct
|
|
{
|
|
align (1):
|
|
sockaddr_un sun;
|
|
char unused = '\0'; // placeholder for a terminating '\0'
|
|
}
|
|
|
|
this() pure nothrow @nogc
|
|
{
|
|
sun.sun_family = AddressFamily.UNIX;
|
|
sun.sun_path = '?';
|
|
_nameLen = sun.sizeof;
|
|
}
|
|
|
|
override void setNameLen(socklen_t len) @trusted
|
|
{
|
|
if (len > sun.sizeof)
|
|
throw new SocketParameterException("Not enough socket address storage");
|
|
_nameLen = len;
|
|
}
|
|
|
|
public:
|
|
override @property sockaddr* name() return
|
|
{
|
|
return cast(sockaddr*)&sun;
|
|
}
|
|
|
|
override @property const(sockaddr)* name() const return
|
|
{
|
|
return cast(const(sockaddr)*)&sun;
|
|
}
|
|
|
|
override @property socklen_t nameLen() @trusted const
|
|
{
|
|
return _nameLen;
|
|
}
|
|
|
|
this(scope const(char)[] path) @trusted pure
|
|
{
|
|
enforce(path.length <= sun.sun_path.sizeof, new SocketParameterException("Path too long"));
|
|
sun.sun_family = AddressFamily.UNIX;
|
|
sun.sun_path.ptr[0 .. path.length] = (cast(byte[]) path)[];
|
|
_nameLen = cast(socklen_t)
|
|
{
|
|
auto len = sockaddr_un.init.sun_path.offsetof + path.length;
|
|
// Pathname socket address must be terminated with '\0'
|
|
// which must be included in the address length.
|
|
if (sun.sun_path.ptr[0])
|
|
{
|
|
sun.sun_path.ptr[path.length] = 0;
|
|
++len;
|
|
}
|
|
return len;
|
|
}();
|
|
}
|
|
|
|
this(sockaddr_un addr) pure nothrow @nogc
|
|
{
|
|
assert(addr.sun_family == AddressFamily.UNIX);
|
|
sun = addr;
|
|
}
|
|
|
|
@property string path() @trusted const pure
|
|
{
|
|
auto len = _nameLen - sockaddr_un.init.sun_path.offsetof;
|
|
if (len == 0)
|
|
return null; // An empty path may be returned from getpeername
|
|
// For pathname socket address we need to strip off the terminating '\0'
|
|
if (sun.sun_path.ptr[0])
|
|
--len;
|
|
return (cast(const(char)*) sun.sun_path.ptr)[0 .. len].idup;
|
|
}
|
|
|
|
override string toString() const pure
|
|
{
|
|
return path;
|
|
}
|
|
}
|
|
|
|
@safe unittest
|
|
{
|
|
import core.stdc.stdio : remove;
|
|
|
|
version (iOSDerived)
|
|
{
|
|
// Slightly different version of `std.file.deleteme` to reduce the path
|
|
// length on iOS derived platforms. Due to the sandbox, the length
|
|
// of paths can quickly become too long.
|
|
static string deleteme()
|
|
{
|
|
import std.conv : text;
|
|
import std.process : thisProcessID;
|
|
import std.file : tempDir;
|
|
|
|
return text(tempDir, thisProcessID);
|
|
}
|
|
}
|
|
|
|
else
|
|
import std.file : deleteme;
|
|
|
|
immutable ubyte[] data = [1, 2, 3, 4];
|
|
Socket[2] pair;
|
|
|
|
const basePath = deleteme;
|
|
auto names = [ basePath ~ "-socket" ];
|
|
version (linux)
|
|
names ~= "\0" ~ basePath ~ "-abstract\0unix\0socket";
|
|
|
|
foreach (name; names)
|
|
{
|
|
auto address = new UnixAddress(name);
|
|
|
|
auto listener = new Socket(AddressFamily.UNIX, SocketType.STREAM);
|
|
scope(exit) listener.close();
|
|
listener.bind(address);
|
|
scope(exit) () @trusted { if (name[0]) remove(name.tempCString()); } ();
|
|
assert(listener.localAddress.toString == name);
|
|
|
|
listener.listen(1);
|
|
|
|
pair[0] = new Socket(AddressFamily.UNIX, SocketType.STREAM);
|
|
scope(exit) listener.close();
|
|
|
|
pair[0].connect(address);
|
|
scope(exit) pair[0].close();
|
|
|
|
pair[1] = listener.accept();
|
|
scope(exit) pair[1].close();
|
|
|
|
pair[0].send(data);
|
|
|
|
auto buf = new ubyte[data.length];
|
|
pair[1].receive(buf);
|
|
assert(buf == data);
|
|
|
|
// getpeername is free to return an empty name for a unix
|
|
// domain socket pair or unbound socket. Let's confirm it
|
|
// returns successfully and doesn't throw anything.
|
|
// See https://issues.dlang.org/show_bug.cgi?id=20544
|
|
assertNotThrown(pair[1].remoteAddress().toString());
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Exception thrown by `Socket.accept`.
|
|
*/
|
|
class SocketAcceptException: SocketOSException
|
|
{
|
|
mixin socketOSExceptionCtors;
|
|
}
|
|
|
|
/// How a socket is shutdown:
|
|
enum SocketShutdown: int
|
|
{
|
|
RECEIVE = SD_RECEIVE, /// socket receives are disallowed
|
|
SEND = SD_SEND, /// socket sends are disallowed
|
|
BOTH = SD_BOTH, /// both RECEIVE and SEND
|
|
}
|
|
|
|
|
|
/// Socket flags that may be OR'ed together:
|
|
enum SocketFlags: int
|
|
{
|
|
NONE = 0, /// no flags specified
|
|
|
|
OOB = MSG_OOB, /// out-of-band stream data
|
|
PEEK = MSG_PEEK, /// peek at incoming data without removing it from the queue, only for receiving
|
|
DONTROUTE = MSG_DONTROUTE, /// data should not be subject to routing; this flag may be ignored. Only for sending
|
|
}
|
|
|
|
|
|
/// Duration timeout value.
|
|
struct TimeVal
|
|
{
|
|
_ctimeval ctimeval;
|
|
alias tv_sec_t = typeof(ctimeval.tv_sec);
|
|
alias tv_usec_t = typeof(ctimeval.tv_usec);
|
|
|
|
/// Number of _seconds.
|
|
pure nothrow @nogc @property
|
|
ref inout(tv_sec_t) seconds() inout return
|
|
{
|
|
return ctimeval.tv_sec;
|
|
}
|
|
|
|
/// Number of additional _microseconds.
|
|
pure nothrow @nogc @property
|
|
ref inout(tv_usec_t) microseconds() inout return
|
|
{
|
|
return ctimeval.tv_usec;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* A collection of sockets for use with `Socket.select`.
|
|
*
|
|
* `SocketSet` wraps the platform `fd_set` type. However, unlike
|
|
* `fd_set`, `SocketSet` is not statically limited to `FD_SETSIZE`
|
|
* or any other limit, and grows as needed.
|
|
*/
|
|
class SocketSet
|
|
{
|
|
private:
|
|
version (Windows)
|
|
{
|
|
// On Windows, fd_set is an array of socket handles,
|
|
// following a word containing the fd_set instance size.
|
|
// We use one dynamic array for everything, and use its first
|
|
// element(s) for the count.
|
|
|
|
alias fd_set_count_type = typeof(fd_set.init.fd_count);
|
|
alias fd_set_type = typeof(fd_set.init.fd_array[0]);
|
|
static assert(fd_set_type.sizeof == socket_t.sizeof);
|
|
|
|
// Number of fd_set_type elements at the start of our array that are
|
|
// used for the socket count and alignment
|
|
|
|
enum FD_SET_OFFSET = fd_set.fd_array.offsetof / fd_set_type.sizeof;
|
|
static assert(FD_SET_OFFSET);
|
|
static assert(fd_set.fd_count.offsetof % fd_set_type.sizeof == 0);
|
|
|
|
fd_set_type[] set;
|
|
|
|
void resize(size_t size) pure nothrow
|
|
{
|
|
set.length = FD_SET_OFFSET + size;
|
|
}
|
|
|
|
ref inout(fd_set_count_type) count() @trusted @property inout pure nothrow @nogc
|
|
{
|
|
assert(set.length);
|
|
return *cast(inout(fd_set_count_type)*)set.ptr;
|
|
}
|
|
|
|
size_t capacity() @property const pure nothrow @nogc
|
|
{
|
|
return set.length - FD_SET_OFFSET;
|
|
}
|
|
|
|
inout(socket_t)[] fds() @trusted inout @property pure nothrow @nogc
|
|
{
|
|
return cast(inout(socket_t)[])set[FD_SET_OFFSET .. FD_SET_OFFSET+count];
|
|
}
|
|
}
|
|
else
|
|
version (Posix)
|
|
{
|
|
// On Posix, fd_set is a bit array. We assume that the fd_set
|
|
// type (declared in core.sys.posix.sys.select) is a structure
|
|
// containing a single field, a static array.
|
|
|
|
static assert(fd_set.tupleof.length == 1);
|
|
|
|
// This is the type used in the fd_set array.
|
|
// Using the type of the correct size is important for big-endian
|
|
// architectures.
|
|
|
|
alias fd_set_type = typeof(fd_set.init.tupleof[0][0]);
|
|
|
|
// Number of file descriptors represented by one fd_set_type
|
|
|
|
enum FD_NFDBITS = 8 * fd_set_type.sizeof;
|
|
|
|
static fd_set_type mask(uint n) pure nothrow @nogc
|
|
{
|
|
return (cast(fd_set_type) 1) << (n % FD_NFDBITS);
|
|
}
|
|
|
|
// Array size to fit that many sockets
|
|
|
|
static size_t lengthFor(size_t size) pure nothrow @nogc
|
|
{
|
|
return (size + (FD_NFDBITS-1)) / FD_NFDBITS;
|
|
}
|
|
|
|
fd_set_type[] set;
|
|
|
|
void resize(size_t size) pure nothrow
|
|
{
|
|
set.length = lengthFor(size);
|
|
}
|
|
|
|
// Make sure we can fit that many sockets
|
|
|
|
void setMinCapacity(size_t size) pure nothrow
|
|
{
|
|
auto length = lengthFor(size);
|
|
if (set.length < length)
|
|
set.length = length;
|
|
}
|
|
|
|
size_t capacity() @property const pure nothrow @nogc
|
|
{
|
|
return set.length * FD_NFDBITS;
|
|
}
|
|
|
|
int maxfd;
|
|
}
|
|
else
|
|
static assert(false, "Unknown platform");
|
|
|
|
public:
|
|
|
|
/**
|
|
* Create a SocketSet with a specific initial capacity (defaults to
|
|
* `FD_SETSIZE`, the system's default capacity).
|
|
*/
|
|
this(size_t size = FD_SETSIZE) pure nothrow
|
|
{
|
|
resize(size);
|
|
reset();
|
|
}
|
|
|
|
/// Reset the `SocketSet` so that there are 0 `Socket`s in the collection.
|
|
void reset() pure nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
count = 0;
|
|
else
|
|
{
|
|
set[] = 0;
|
|
maxfd = -1;
|
|
}
|
|
}
|
|
|
|
|
|
void add(socket_t s) @trusted pure nothrow
|
|
{
|
|
version (Windows)
|
|
{
|
|
if (count == capacity)
|
|
{
|
|
set.length *= 2;
|
|
set.length = set.capacity;
|
|
}
|
|
++count;
|
|
fds[$-1] = s;
|
|
}
|
|
else
|
|
{
|
|
auto index = s / FD_NFDBITS;
|
|
auto length = set.length;
|
|
if (index >= length)
|
|
{
|
|
while (index >= length)
|
|
length *= 2;
|
|
set.length = length;
|
|
set.length = set.capacity;
|
|
}
|
|
set[index] |= mask(s);
|
|
if (maxfd < s)
|
|
maxfd = s;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Add a `Socket` to the collection.
|
|
* The socket must not already be in the collection.
|
|
*/
|
|
void add(Socket s) pure nothrow
|
|
{
|
|
add(s.sock);
|
|
}
|
|
|
|
void remove(socket_t s) pure nothrow
|
|
{
|
|
version (Windows)
|
|
{
|
|
import std.algorithm.searching : countUntil;
|
|
auto fds = fds;
|
|
auto p = fds.countUntil(s);
|
|
if (p >= 0)
|
|
fds[p] = fds[--count];
|
|
}
|
|
else
|
|
{
|
|
auto index = s / FD_NFDBITS;
|
|
if (index >= set.length)
|
|
return;
|
|
set[index] &= ~mask(s);
|
|
// note: adjusting maxfd would require scanning the set, not worth it
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Remove this `Socket` from the collection.
|
|
* Does nothing if the socket is not in the collection already.
|
|
*/
|
|
void remove(Socket s) pure nothrow
|
|
{
|
|
remove(s.sock);
|
|
}
|
|
|
|
int isSet(socket_t s) const pure nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
{
|
|
import std.algorithm.searching : canFind;
|
|
return fds.canFind(s) ? 1 : 0;
|
|
}
|
|
else
|
|
{
|
|
if (s > maxfd)
|
|
return 0;
|
|
auto index = s / FD_NFDBITS;
|
|
return (set[index] & mask(s)) ? 1 : 0;
|
|
}
|
|
}
|
|
|
|
|
|
/// Return nonzero if this `Socket` is in the collection.
|
|
int isSet(Socket s) const pure nothrow @nogc
|
|
{
|
|
return isSet(s.sock);
|
|
}
|
|
|
|
|
|
/**
|
|
* Returns:
|
|
* The current capacity of this `SocketSet`. The exact
|
|
* meaning of the return value varies from platform to platform.
|
|
*
|
|
* Note:
|
|
* Since D 2.065, this value does not indicate a
|
|
* restriction, and `SocketSet` will grow its capacity as
|
|
* needed automatically.
|
|
*/
|
|
@property uint max() const pure nothrow @nogc
|
|
{
|
|
return cast(uint) capacity;
|
|
}
|
|
|
|
|
|
fd_set* toFd_set() @trusted pure nothrow @nogc
|
|
{
|
|
return cast(fd_set*) set.ptr;
|
|
}
|
|
|
|
|
|
int selectn() const pure nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
{
|
|
return count;
|
|
}
|
|
else version (Posix)
|
|
{
|
|
return maxfd + 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
@safe unittest
|
|
{
|
|
auto fds = cast(socket_t[])
|
|
[cast(socket_t) 1, 2, 0, 1024, 17, 42, 1234, 77, 77+32, 77+64];
|
|
auto set = new SocketSet();
|
|
foreach (fd; fds) assert(!set.isSet(fd));
|
|
foreach (fd; fds) set.add(fd);
|
|
foreach (fd; fds) assert(set.isSet(fd));
|
|
|
|
// Make sure SocketSet reimplements fd_set correctly
|
|
auto fdset = set.toFd_set();
|
|
foreach (fd; fds[0]..cast(socket_t)(fds[$-1]+1))
|
|
assert(cast(bool) set.isSet(fd) == cast(bool)(() @trusted => FD_ISSET(fd, fdset))());
|
|
|
|
foreach (fd; fds)
|
|
{
|
|
assert(set.isSet(fd));
|
|
set.remove(fd);
|
|
assert(!set.isSet(fd));
|
|
}
|
|
}
|
|
|
|
@safe unittest
|
|
{
|
|
version (iOSDerived)
|
|
{
|
|
enum PAIRS = 256;
|
|
enum LIMIT = 1024;
|
|
}
|
|
else
|
|
{
|
|
enum PAIRS = 768;
|
|
enum LIMIT = 2048;
|
|
}
|
|
|
|
softUnittest({
|
|
version (Posix)
|
|
() @trusted
|
|
{
|
|
static assert(LIMIT > PAIRS*2);
|
|
import core.sys.posix.sys.resource;
|
|
rlimit fileLimit;
|
|
getrlimit(RLIMIT_NOFILE, &fileLimit);
|
|
assert(fileLimit.rlim_max > LIMIT, "Open file hard limit too low");
|
|
fileLimit.rlim_cur = LIMIT;
|
|
setrlimit(RLIMIT_NOFILE, &fileLimit);
|
|
} ();
|
|
|
|
Socket[2][PAIRS] pairs;
|
|
foreach (ref pair; pairs)
|
|
pair = socketPair();
|
|
scope(exit)
|
|
{
|
|
foreach (pair; pairs)
|
|
{
|
|
pair[0].close();
|
|
pair[1].close();
|
|
}
|
|
}
|
|
|
|
import std.random;
|
|
auto rng = Xorshift(42);
|
|
pairs[].randomShuffle(rng);
|
|
|
|
auto readSet = new SocketSet();
|
|
auto writeSet = new SocketSet();
|
|
auto errorSet = new SocketSet();
|
|
|
|
foreach (testPair; pairs)
|
|
{
|
|
void fillSets()
|
|
{
|
|
readSet.reset();
|
|
writeSet.reset();
|
|
errorSet.reset();
|
|
foreach (ref pair; pairs)
|
|
foreach (s; pair[])
|
|
{
|
|
readSet.add(s);
|
|
writeSet.add(s);
|
|
errorSet.add(s);
|
|
}
|
|
}
|
|
|
|
fillSets();
|
|
auto n = Socket.select(readSet, writeSet, errorSet);
|
|
assert(n == PAIRS*2); // All in writeSet
|
|
assert(writeSet.isSet(testPair[0]));
|
|
assert(writeSet.isSet(testPair[1]));
|
|
assert(!readSet.isSet(testPair[0]));
|
|
assert(!readSet.isSet(testPair[1]));
|
|
assert(!errorSet.isSet(testPair[0]));
|
|
assert(!errorSet.isSet(testPair[1]));
|
|
|
|
ubyte[1] b;
|
|
// Socket.send can't be marked with `scope`
|
|
// -> @safe DIP1000 code can't use it - see https://github.com/dlang/phobos/pull/6204
|
|
() @trusted {
|
|
testPair[0].send(b[]);
|
|
}();
|
|
fillSets();
|
|
n = Socket.select(readSet, null, null);
|
|
assert(n == 1); // testPair[1]
|
|
assert(readSet.isSet(testPair[1]));
|
|
assert(!readSet.isSet(testPair[0]));
|
|
// Socket.receive can't be marked with `scope`
|
|
// -> @safe DIP1000 code can't use it - see https://github.com/dlang/phobos/pull/6204
|
|
() @trusted {
|
|
testPair[1].receive(b[]);
|
|
}();
|
|
}
|
|
});
|
|
}
|
|
|
|
// https://issues.dlang.org/show_bug.cgi?id=14012
|
|
// https://issues.dlang.org/show_bug.cgi?id=14013
|
|
@safe unittest
|
|
{
|
|
auto set = new SocketSet(1);
|
|
assert(set.max >= 0);
|
|
|
|
enum LIMIT = 4096;
|
|
foreach (n; 0 .. LIMIT)
|
|
set.add(cast(socket_t) n);
|
|
assert(set.max >= LIMIT);
|
|
}
|
|
|
|
/// The level at which a socket option is defined:
|
|
enum SocketOptionLevel: int
|
|
{
|
|
SOCKET = SOL_SOCKET, /// Socket level
|
|
IP = ProtocolType.IP, /// Internet Protocol version 4 level
|
|
ICMP = ProtocolType.ICMP, /// Internet Control Message Protocol level
|
|
IGMP = ProtocolType.IGMP, /// Internet Group Management Protocol level
|
|
GGP = ProtocolType.GGP, /// Gateway to Gateway Protocol level
|
|
TCP = ProtocolType.TCP, /// Transmission Control Protocol level
|
|
PUP = ProtocolType.PUP, /// PARC Universal Packet Protocol level
|
|
UDP = ProtocolType.UDP, /// User Datagram Protocol level
|
|
IDP = ProtocolType.IDP, /// Xerox NS protocol level
|
|
RAW = ProtocolType.RAW, /// Raw IP packet level
|
|
IPV6 = ProtocolType.IPV6, /// Internet Protocol version 6 level
|
|
}
|
|
|
|
/// _Linger information for use with SocketOption.LINGER.
|
|
struct Linger
|
|
{
|
|
_clinger clinger;
|
|
|
|
private alias l_onoff_t = typeof(_clinger.init.l_onoff );
|
|
private alias l_linger_t = typeof(_clinger.init.l_linger);
|
|
|
|
/// Nonzero for _on.
|
|
pure nothrow @nogc @property
|
|
ref inout(l_onoff_t) on() inout return
|
|
{
|
|
return clinger.l_onoff;
|
|
}
|
|
|
|
/// Linger _time.
|
|
pure nothrow @nogc @property
|
|
ref inout(l_linger_t) time() inout return
|
|
{
|
|
return clinger.l_linger;
|
|
}
|
|
}
|
|
|
|
/// Specifies a socket option:
|
|
enum SocketOption: int
|
|
{
|
|
DEBUG = SO_DEBUG, /// Record debugging information
|
|
BROADCAST = SO_BROADCAST, /// Allow transmission of broadcast messages
|
|
REUSEADDR = SO_REUSEADDR, /// Allow local reuse of address
|
|
/**
|
|
* Allow local reuse of port
|
|
*
|
|
* On Windows, this is equivalent to `SocketOption.REUSEADDR`.
|
|
* There is in fact no option named `REUSEPORT`.
|
|
* However, `SocketOption.REUSEADDR` matches the behavior of
|
|
* `SocketOption.REUSEPORT` on other platforms. Further details on this
|
|
* topic can be found here:
|
|
* $(LINK https://learn.microsoft.com/en-us/windows/win32/winsock/using-so-reuseaddr-and-so-exclusiveaddruse)
|
|
*
|
|
* On Linux, this ensures fair distribution of incoming connections accross threads.
|
|
*
|
|
* See_Also:
|
|
* https://lwn.net/Articles/542629/
|
|
*/
|
|
REUSEPORT = SO_REUSEPORT,
|
|
LINGER = SO_LINGER, /// Linger on close if unsent data is present
|
|
OOBINLINE = SO_OOBINLINE, /// Receive out-of-band data in band
|
|
SNDBUF = SO_SNDBUF, /// Send buffer size
|
|
RCVBUF = SO_RCVBUF, /// Receive buffer size
|
|
DONTROUTE = SO_DONTROUTE, /// Do not route
|
|
SNDTIMEO = SO_SNDTIMEO, /// Send timeout
|
|
RCVTIMEO = SO_RCVTIMEO, /// Receive timeout
|
|
ERROR = SO_ERROR, /// Retrieve and clear error status
|
|
KEEPALIVE = SO_KEEPALIVE, /// Enable keep-alive packets
|
|
ACCEPTCONN = SO_ACCEPTCONN, /// Listen
|
|
RCVLOWAT = SO_RCVLOWAT, /// Minimum number of input bytes to process
|
|
SNDLOWAT = SO_SNDLOWAT, /// Minimum number of output bytes to process
|
|
TYPE = SO_TYPE, /// Socket type
|
|
|
|
// SocketOptionLevel.TCP:
|
|
TCP_NODELAY = .TCP_NODELAY, /// Disable the Nagle algorithm for send coalescing
|
|
|
|
// SocketOptionLevel.IPV6:
|
|
IPV6_UNICAST_HOPS = .IPV6_UNICAST_HOPS, /// IP unicast hop limit
|
|
IPV6_MULTICAST_IF = .IPV6_MULTICAST_IF, /// IP multicast interface
|
|
IPV6_MULTICAST_LOOP = .IPV6_MULTICAST_LOOP, /// IP multicast loopback
|
|
IPV6_MULTICAST_HOPS = .IPV6_MULTICAST_HOPS, /// IP multicast hops
|
|
IPV6_JOIN_GROUP = .IPV6_JOIN_GROUP, /// Add an IP group membership
|
|
IPV6_LEAVE_GROUP = .IPV6_LEAVE_GROUP, /// Drop an IP group membership
|
|
IPV6_V6ONLY = .IPV6_V6ONLY, /// Treat wildcard bind as AF_INET6-only
|
|
}
|
|
|
|
|
|
/**
|
|
* Class that creates a network communication endpoint using
|
|
* the Berkeley sockets interface.
|
|
*/
|
|
class Socket
|
|
{
|
|
private:
|
|
socket_t sock;
|
|
AddressFamily _family;
|
|
|
|
version (Windows)
|
|
bool _blocking = true; /// Property to get or set whether the socket is blocking or nonblocking.
|
|
|
|
// The WinSock timeouts seem to be effectively skewed by a constant
|
|
// offset of about half a second (value in milliseconds). This has
|
|
// been confirmed on updated (as of Jun 2011) Windows XP, Windows 7
|
|
// and Windows Server 2008 R2 boxes. The unittest below tests this
|
|
// behavior.
|
|
enum WINSOCK_TIMEOUT_SKEW = 500;
|
|
|
|
@safe unittest
|
|
{
|
|
if (runSlowTests)
|
|
softUnittest({
|
|
import std.datetime.stopwatch : StopWatch;
|
|
import std.typecons : Yes;
|
|
|
|
enum msecs = 1000;
|
|
auto pair = socketPair();
|
|
auto testSock = pair[0];
|
|
testSock.setOption(SocketOptionLevel.SOCKET,
|
|
SocketOption.RCVTIMEO, dur!"msecs"(msecs));
|
|
|
|
auto sw = StopWatch(Yes.autoStart);
|
|
ubyte[1] buf;
|
|
testSock.receive(buf);
|
|
sw.stop();
|
|
|
|
Duration readBack = void;
|
|
testSock.getOption(SocketOptionLevel.SOCKET, SocketOption.RCVTIMEO, readBack);
|
|
|
|
assert(readBack.total!"msecs" == msecs);
|
|
assert(sw.peek().total!"msecs" > msecs - 100 && sw.peek().total!"msecs" < msecs + 100);
|
|
});
|
|
}
|
|
|
|
void setSock(socket_t handle)
|
|
{
|
|
assert(handle != socket_t.init);
|
|
sock = handle;
|
|
|
|
// Set the option to disable SIGPIPE on send() if the platform
|
|
// has it (e.g. on OS X).
|
|
static if (is(typeof(SO_NOSIGPIPE)))
|
|
{
|
|
setOption(SocketOptionLevel.SOCKET, cast(SocketOption) SO_NOSIGPIPE, true);
|
|
}
|
|
}
|
|
|
|
|
|
// For use with accepting().
|
|
protected this() pure nothrow @nogc
|
|
{
|
|
}
|
|
|
|
|
|
public:
|
|
|
|
/**
|
|
* Create a blocking socket. If a single protocol type exists to support
|
|
* this socket type within the address family, the `ProtocolType` may be
|
|
* omitted.
|
|
*/
|
|
this(AddressFamily af, SocketType type, ProtocolType protocol) @trusted
|
|
{
|
|
_family = af;
|
|
auto handle = cast(socket_t) socket(af, type, protocol);
|
|
if (handle == socket_t.init)
|
|
throw new SocketOSException("Unable to create socket");
|
|
setSock(handle);
|
|
}
|
|
|
|
/// ditto
|
|
this(AddressFamily af, SocketType type)
|
|
{
|
|
/* A single protocol exists to support this socket type within the
|
|
* protocol family, so the ProtocolType is assumed.
|
|
*/
|
|
this(af, type, cast(ProtocolType) 0); // Pseudo protocol number.
|
|
}
|
|
|
|
|
|
/// ditto
|
|
this(AddressFamily af, SocketType type, scope const(char)[] protocolName) @trusted
|
|
{
|
|
protoent* proto;
|
|
proto = getprotobyname(protocolName.tempCString());
|
|
if (!proto)
|
|
throw new SocketOSException("Unable to find the protocol");
|
|
this(af, type, cast(ProtocolType) proto.p_proto);
|
|
}
|
|
|
|
|
|
/**
|
|
* Create a blocking socket using the parameters from the specified
|
|
* `AddressInfo` structure.
|
|
*/
|
|
this(const scope AddressInfo info)
|
|
{
|
|
this(info.family, info.type, info.protocol);
|
|
}
|
|
|
|
/// Use an existing socket handle.
|
|
this(socket_t sock, AddressFamily af) pure nothrow @nogc
|
|
{
|
|
assert(sock != socket_t.init);
|
|
this.sock = sock;
|
|
this._family = af;
|
|
}
|
|
|
|
|
|
~this() nothrow @nogc
|
|
{
|
|
close();
|
|
}
|
|
|
|
|
|
/// Get underlying socket handle.
|
|
@property socket_t handle() const pure nothrow @nogc
|
|
{
|
|
return sock;
|
|
}
|
|
|
|
/**
|
|
* Releases the underlying socket handle from the Socket object. Once it
|
|
* is released, you cannot use the Socket object's methods anymore. This
|
|
* also means the Socket destructor will no longer close the socket - it
|
|
* becomes your responsibility.
|
|
*
|
|
* To get the handle without releasing it, use the `handle` property.
|
|
*/
|
|
@property socket_t release() pure nothrow @nogc
|
|
{
|
|
auto h = sock;
|
|
this.sock = socket_t.init;
|
|
return h;
|
|
}
|
|
|
|
/**
|
|
* Get/set socket's blocking flag.
|
|
*
|
|
* When a socket is blocking, calls to receive(), accept(), and send()
|
|
* will block and wait for data/action.
|
|
* A non-blocking socket will immediately return instead of blocking.
|
|
*/
|
|
@property bool blocking() @trusted const nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
{
|
|
return _blocking;
|
|
}
|
|
else version (Posix)
|
|
{
|
|
return !(fcntl(handle, F_GETFL, 0) & O_NONBLOCK);
|
|
}
|
|
}
|
|
|
|
/// ditto
|
|
@property void blocking(bool byes) @trusted
|
|
{
|
|
version (Windows)
|
|
{
|
|
uint num = !byes;
|
|
if (_SOCKET_ERROR == ioctlsocket(sock, FIONBIO, &num))
|
|
goto err;
|
|
_blocking = byes;
|
|
}
|
|
else version (Posix)
|
|
{
|
|
int x = fcntl(sock, F_GETFL, 0);
|
|
if (-1 == x)
|
|
goto err;
|
|
if (byes)
|
|
x &= ~O_NONBLOCK;
|
|
else
|
|
x |= O_NONBLOCK;
|
|
if (-1 == fcntl(sock, F_SETFL, x))
|
|
goto err;
|
|
}
|
|
return; // Success.
|
|
|
|
err:
|
|
throw new SocketOSException("Unable to set socket blocking");
|
|
}
|
|
|
|
|
|
/// Get the socket's address family.
|
|
@property AddressFamily addressFamily()
|
|
{
|
|
return _family;
|
|
}
|
|
|
|
/// Property that indicates if this is a valid, alive socket.
|
|
@property bool isAlive() @trusted const
|
|
{
|
|
int type;
|
|
socklen_t typesize = cast(socklen_t) type.sizeof;
|
|
return !getsockopt(sock, SOL_SOCKET, SO_TYPE, cast(char*)&type, &typesize);
|
|
}
|
|
|
|
/**
|
|
* Associate a local address with this socket.
|
|
*
|
|
* Params:
|
|
* addr = The $(LREF Address) to associate this socket with.
|
|
*
|
|
* Throws: $(LREF SocketOSException) when unable to bind the socket.
|
|
*/
|
|
void bind(Address addr) @trusted
|
|
{
|
|
if (_SOCKET_ERROR == .bind(sock, addr.name, addr.nameLen))
|
|
throw new SocketOSException("Unable to bind socket");
|
|
}
|
|
|
|
/**
|
|
* Establish a connection. If the socket is blocking, connect waits for
|
|
* the connection to be made. If the socket is nonblocking, connect
|
|
* returns immediately and the connection attempt is still in progress.
|
|
*/
|
|
void connect(Address to) @trusted
|
|
{
|
|
if (_SOCKET_ERROR == .connect(sock, to.name, to.nameLen))
|
|
{
|
|
int err;
|
|
err = _lasterr();
|
|
|
|
if (!blocking)
|
|
{
|
|
version (Windows)
|
|
{
|
|
if (WSAEWOULDBLOCK == err)
|
|
return;
|
|
}
|
|
else version (Posix)
|
|
{
|
|
if (EINPROGRESS == err)
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
static assert(0);
|
|
}
|
|
}
|
|
throw new SocketOSException("Unable to connect socket", err);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Listen for an incoming connection. `bind` must be called before you
|
|
* can `listen`. The `backlog` is a request of how many pending
|
|
* incoming connections are queued until `accept`ed.
|
|
*/
|
|
void listen(int backlog) @trusted
|
|
{
|
|
if (_SOCKET_ERROR == .listen(sock, backlog))
|
|
throw new SocketOSException("Unable to listen on socket");
|
|
}
|
|
|
|
/**
|
|
* Called by `accept` when a new `Socket` must be created for a new
|
|
* connection. To use a derived class, override this method and return an
|
|
* instance of your class. The returned `Socket`'s handle must not be
|
|
* set; `Socket` has a protected constructor `this()` to use in this
|
|
* situation.
|
|
*
|
|
* Override to use a derived class.
|
|
* The returned socket's handle must not be set.
|
|
*/
|
|
protected Socket accepting() pure nothrow
|
|
{
|
|
return new Socket;
|
|
}
|
|
|
|
/**
|
|
* Accept an incoming connection. If the socket is blocking, `accept`
|
|
* waits for a connection request. Throws `SocketAcceptException` if
|
|
* unable to _accept. See `accepting` for use with derived classes.
|
|
*/
|
|
Socket accept() @trusted
|
|
{
|
|
auto newsock = cast(socket_t).accept(sock, null, null);
|
|
if (socket_t.init == newsock)
|
|
throw new SocketAcceptException("Unable to accept socket connection");
|
|
|
|
Socket newSocket;
|
|
try
|
|
{
|
|
newSocket = accepting();
|
|
assert(newSocket.sock == socket_t.init);
|
|
|
|
newSocket.setSock(newsock);
|
|
version (Windows)
|
|
newSocket._blocking = _blocking; //inherits blocking mode
|
|
newSocket._family = _family; //same family
|
|
}
|
|
catch (Throwable o)
|
|
{
|
|
_close(newsock);
|
|
throw o;
|
|
}
|
|
|
|
return newSocket;
|
|
}
|
|
|
|
/// Disables sends and/or receives.
|
|
void shutdown(SocketShutdown how) @trusted nothrow @nogc
|
|
{
|
|
.shutdown(sock, cast(int) how);
|
|
}
|
|
|
|
|
|
private static void _close(socket_t sock) @system nothrow @nogc
|
|
{
|
|
version (Windows)
|
|
{
|
|
.closesocket(sock);
|
|
}
|
|
else version (Posix)
|
|
{
|
|
.close(sock);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Immediately drop any connections and release socket resources.
|
|
* The `Socket` object is no longer usable after `close`.
|
|
* Calling `shutdown` before `close` is recommended
|
|
* for connection-oriented sockets.
|
|
*/
|
|
void close() scope @trusted nothrow @nogc
|
|
{
|
|
_close(sock);
|
|
sock = socket_t.init;
|
|
}
|
|
|
|
|
|
/**
|
|
* Returns: The local machine's host name
|
|
*/
|
|
static @property string hostName() @trusted // getter
|
|
{
|
|
char[256] result; // Host names are limited to 255 chars.
|
|
if (_SOCKET_ERROR == .gethostname(result.ptr, result.length))
|
|
throw new SocketOSException("Unable to obtain host name");
|
|
return to!string(result.ptr);
|
|
}
|
|
|
|
/// Remote endpoint `Address`.
|
|
@property Address remoteAddress() @trusted
|
|
{
|
|
Address addr = createAddress();
|
|
socklen_t nameLen = addr.nameLen;
|
|
if (_SOCKET_ERROR == .getpeername(sock, addr.name, &nameLen))
|
|
throw new SocketOSException("Unable to obtain remote socket address");
|
|
addr.setNameLen(nameLen);
|
|
assert(addr.addressFamily == _family);
|
|
return addr;
|
|
}
|
|
|
|
/// Local endpoint `Address`.
|
|
@property Address localAddress() @trusted
|
|
{
|
|
Address addr = createAddress();
|
|
socklen_t nameLen = addr.nameLen;
|
|
if (_SOCKET_ERROR == .getsockname(sock, addr.name, &nameLen))
|
|
throw new SocketOSException("Unable to obtain local socket address");
|
|
addr.setNameLen(nameLen);
|
|
assert(addr.addressFamily == _family);
|
|
return addr;
|
|
}
|
|
|
|
/**
|
|
* Send or receive error code. See `wouldHaveBlocked`,
|
|
* `lastSocketError` and `Socket.getErrorText` for obtaining more
|
|
* information about the error.
|
|
*/
|
|
enum int ERROR = _SOCKET_ERROR;
|
|
|
|
private static int capToInt(size_t size) nothrow @nogc
|
|
{
|
|
// Windows uses int instead of size_t for length arguments.
|
|
// Luckily, the send/recv functions make no guarantee that
|
|
// all the data is sent, so we use that to send at most
|
|
// int.max bytes.
|
|
return size > size_t(int.max) ? int.max : cast(int) size;
|
|
}
|
|
|
|
/**
|
|
* Send data on the connection. If the socket is blocking and there is no
|
|
* buffer space left, `send` waits.
|
|
* Returns: The number of bytes actually sent, or `Socket.ERROR` on
|
|
* failure.
|
|
*/
|
|
ptrdiff_t send(scope const(void)[] buf, SocketFlags flags) @trusted
|
|
{
|
|
static if (is(typeof(MSG_NOSIGNAL)))
|
|
{
|
|
flags = cast(SocketFlags)(flags | MSG_NOSIGNAL);
|
|
}
|
|
version (Windows)
|
|
auto sent = .send(sock, buf.ptr, capToInt(buf.length), cast(int) flags);
|
|
else
|
|
auto sent = .send(sock, buf.ptr, buf.length, cast(int) flags);
|
|
return sent;
|
|
}
|
|
|
|
/// ditto
|
|
ptrdiff_t send(scope const(void)[] buf)
|
|
{
|
|
return send(buf, SocketFlags.NONE);
|
|
}
|
|
|
|
/**
|
|
* Send data to a specific destination Address. If the destination address is
|
|
* not specified, a connection must have been made and that address is used.
|
|
* If the socket is blocking and there is no buffer space left, `sendTo` waits.
|
|
* Returns: The number of bytes actually sent, or `Socket.ERROR` on
|
|
* failure.
|
|
*/
|
|
ptrdiff_t sendTo(scope const(void)[] buf, SocketFlags flags, Address to) @trusted
|
|
{
|
|
static if (is(typeof(MSG_NOSIGNAL)))
|
|
{
|
|
flags = cast(SocketFlags)(flags | MSG_NOSIGNAL);
|
|
}
|
|
version (Windows)
|
|
return .sendto(
|
|
sock, buf.ptr, capToInt(buf.length),
|
|
cast(int) flags, to.name, to.nameLen
|
|
);
|
|
else
|
|
return .sendto(sock, buf.ptr, buf.length, cast(int) flags, to.name, to.nameLen);
|
|
}
|
|
|
|
/// ditto
|
|
ptrdiff_t sendTo(scope const(void)[] buf, Address to)
|
|
{
|
|
return sendTo(buf, SocketFlags.NONE, to);
|
|
}
|
|
|
|
|
|
//assumes you connect()ed
|
|
/// ditto
|
|
ptrdiff_t sendTo(scope const(void)[] buf, SocketFlags flags) @trusted
|
|
{
|
|
static if (is(typeof(MSG_NOSIGNAL)))
|
|
{
|
|
flags = cast(SocketFlags)(flags | MSG_NOSIGNAL);
|
|
}
|
|
version (Windows)
|
|
return .sendto(sock, buf.ptr, capToInt(buf.length), cast(int) flags, null, 0);
|
|
else
|
|
return .sendto(sock, buf.ptr, buf.length, cast(int) flags, null, 0);
|
|
}
|
|
|
|
|
|
//assumes you connect()ed
|
|
/// ditto
|
|
ptrdiff_t sendTo(scope const(void)[] buf)
|
|
{
|
|
return sendTo(buf, SocketFlags.NONE);
|
|
}
|
|
|
|
|
|
/**
|
|
* Receive data on the connection. If the socket is blocking, `receive`
|
|
* waits until there is data to be received.
|
|
* Returns: The number of bytes actually received, `0` if the remote side
|
|
* has closed the connection, or `Socket.ERROR` on failure.
|
|
*/
|
|
ptrdiff_t receive(scope void[] buf, SocketFlags flags) @trusted
|
|
{
|
|
version (Windows) // Does not use size_t
|
|
{
|
|
return buf.length
|
|
? .recv(sock, buf.ptr, capToInt(buf.length), cast(int) flags)
|
|
: 0;
|
|
}
|
|
else
|
|
{
|
|
return buf.length
|
|
? .recv(sock, buf.ptr, buf.length, cast(int) flags)
|
|
: 0;
|
|
}
|
|
}
|
|
|
|
/// ditto
|
|
ptrdiff_t receive(scope void[] buf)
|
|
{
|
|
return receive(buf, SocketFlags.NONE);
|
|
}
|
|
|
|
/**
|
|
* Receive data and get the remote endpoint `Address`.
|
|
* If the socket is blocking, `receiveFrom` waits until there is data to
|
|
* be received.
|
|
* Returns: The number of bytes actually received, `0` if the remote side
|
|
* has closed the connection, or `Socket.ERROR` on failure.
|
|
*/
|
|
ptrdiff_t receiveFrom(scope void[] buf, SocketFlags flags, ref Address from) @trusted
|
|
{
|
|
if (!buf.length) //return 0 and don't think the connection closed
|
|
return 0;
|
|
if (from is null || from.addressFamily != _family)
|
|
from = createAddress();
|
|
socklen_t nameLen = from.nameLen;
|
|
version (Windows)
|
|
auto read = .recvfrom(sock, buf.ptr, capToInt(buf.length), cast(int) flags, from.name, &nameLen);
|
|
|
|
else
|
|
auto read = .recvfrom(sock, buf.ptr, buf.length, cast(int) flags, from.name, &nameLen);
|
|
|
|
if (read >= 0)
|
|
{
|
|
from.setNameLen(nameLen);
|
|
assert(from.addressFamily == _family);
|
|
}
|
|
return read;
|
|
}
|
|
|
|
|
|
/// ditto
|
|
ptrdiff_t receiveFrom(scope void[] buf, ref Address from)
|
|
{
|
|
return receiveFrom(buf, SocketFlags.NONE, from);
|
|
}
|
|
|
|
|
|
//assumes you connect()ed
|
|
/// ditto
|
|
ptrdiff_t receiveFrom(scope void[] buf, SocketFlags flags) @trusted
|
|
{
|
|
if (!buf.length) //return 0 and don't think the connection closed
|
|
return 0;
|
|
version (Windows)
|
|
{
|
|
auto read = .recvfrom(sock, buf.ptr, capToInt(buf.length), cast(int) flags, null, null);
|
|
// if (!read) //connection closed
|
|
return read;
|
|
}
|
|
else
|
|
{
|
|
auto read = .recvfrom(sock, buf.ptr, buf.length, cast(int) flags, null, null);
|
|
// if (!read) //connection closed
|
|
return read;
|
|
}
|
|
}
|
|
|
|
|
|
//assumes you connect()ed
|
|
/// ditto
|
|
ptrdiff_t receiveFrom(scope void[] buf)
|
|
{
|
|
return receiveFrom(buf, SocketFlags.NONE);
|
|
}
|
|
|
|
|
|
/**
|
|
* Get a socket option.
|
|
* Returns: The number of bytes written to `result`.
|
|
* The length, in bytes, of the actual result - very different from getsockopt()
|
|
*/
|
|
int getOption(SocketOptionLevel level, SocketOption option, scope void[] result) @trusted
|
|
{
|
|
socklen_t len = cast(socklen_t) result.length;
|
|
if (_SOCKET_ERROR == .getsockopt(sock, cast(int) level, cast(int) option, result.ptr, &len))
|
|
throw new SocketOSException("Unable to get socket option");
|
|
return len;
|
|
}
|
|
|
|
|
|
/// Common case of getting integer and boolean options.
|
|
int getOption(SocketOptionLevel level, SocketOption option, out int32_t result) @trusted
|
|
{
|
|
return getOption(level, option, (&result)[0 .. 1]);
|
|
}
|
|
|
|
|
|
/// Get the linger option.
|
|
int getOption(SocketOptionLevel level, SocketOption option, out Linger result) @trusted
|
|
{
|
|
//return getOption(cast(SocketOptionLevel) SocketOptionLevel.SOCKET, SocketOption.LINGER, (&result)[0 .. 1]);
|
|
return getOption(level, option, (&result.clinger)[0 .. 1]);
|
|
}
|
|
|
|
/// Get a timeout (duration) option.
|
|
void getOption(SocketOptionLevel level, SocketOption option, out Duration result) @trusted
|
|
{
|
|
enforce(option == SocketOption.SNDTIMEO || option == SocketOption.RCVTIMEO,
|
|
new SocketParameterException("Not a valid timeout option: " ~ to!string(option)));
|
|
// WinSock returns the timeout values as a milliseconds DWORD,
|
|
// while Linux and BSD return a timeval struct.
|
|
version (Windows)
|
|
{
|
|
int msecs;
|
|
getOption(level, option, (&msecs)[0 .. 1]);
|
|
if (option == SocketOption.RCVTIMEO)
|
|
msecs += WINSOCK_TIMEOUT_SKEW;
|
|
result = dur!"msecs"(msecs);
|
|
}
|
|
else version (Posix)
|
|
{
|
|
TimeVal tv;
|
|
getOption(level, option, (&tv.ctimeval)[0 .. 1]);
|
|
result = dur!"seconds"(tv.seconds) + dur!"usecs"(tv.microseconds);
|
|
}
|
|
else static assert(false);
|
|
}
|
|
|
|
/// Set a socket option.
|
|
void setOption(SocketOptionLevel level, SocketOption option, scope void[] value) @trusted
|
|
{
|
|
if (_SOCKET_ERROR == .setsockopt(sock, cast(int) level,
|
|
cast(int) option, value.ptr, cast(uint) value.length))
|
|
throw new SocketOSException("Unable to set socket option");
|
|
}
|
|
|
|
|
|
/// Common case for setting integer and boolean options.
|
|
void setOption(SocketOptionLevel level, SocketOption option, int32_t value) @trusted
|
|
{
|
|
setOption(level, option, (&value)[0 .. 1]);
|
|
}
|
|
|
|
|
|
/// Set the linger option.
|
|
void setOption(SocketOptionLevel level, SocketOption option, Linger value) @trusted
|
|
{
|
|
//setOption(cast(SocketOptionLevel) SocketOptionLevel.SOCKET, SocketOption.LINGER, (&value)[0 .. 1]);
|
|
setOption(level, option, (&value.clinger)[0 .. 1]);
|
|
}
|
|
|
|
/**
|
|
* Sets a timeout (duration) option, i.e. `SocketOption.SNDTIMEO` or
|
|
* `RCVTIMEO`. Zero indicates no timeout.
|
|
*
|
|
* In a typical application, you might also want to consider using
|
|
* a non-blocking socket instead of setting a timeout on a blocking one.
|
|
*
|
|
* Note: While the receive timeout setting is generally quite accurate
|
|
* on *nix systems even for smaller durations, there are two issues to
|
|
* be aware of on Windows: First, although undocumented, the effective
|
|
* timeout duration seems to be the one set on the socket plus half
|
|
* a second. `setOption()` tries to compensate for that, but still,
|
|
* timeouts under 500ms are not possible on Windows. Second, be aware
|
|
* that the actual amount of time spent until a blocking call returns
|
|
* randomly varies on the order of 10ms.
|
|
*
|
|
* Params:
|
|
* level = The level at which a socket option is defined.
|
|
* option = Either `SocketOption.SNDTIMEO` or `SocketOption.RCVTIMEO`.
|
|
* value = The timeout duration to set. Must not be negative.
|
|
*
|
|
* Throws: `SocketException` if setting the options fails.
|
|
*
|
|
* Example:
|
|
* ---
|
|
* import std.datetime;
|
|
* import std.typecons;
|
|
* auto pair = socketPair();
|
|
* scope(exit) foreach (s; pair) s.close();
|
|
*
|
|
* // Set a receive timeout, and then wait at one end of
|
|
* // the socket pair, knowing that no data will arrive.
|
|
* pair[0].setOption(SocketOptionLevel.SOCKET,
|
|
* SocketOption.RCVTIMEO, dur!"seconds"(1));
|
|
*
|
|
* auto sw = StopWatch(Yes.autoStart);
|
|
* ubyte[1] buffer;
|
|
* pair[0].receive(buffer);
|
|
* writefln("Waited %s ms until the socket timed out.",
|
|
* sw.peek.msecs);
|
|
* ---
|
|
*/
|
|
void setOption(SocketOptionLevel level, SocketOption option, Duration value) @trusted
|
|
{
|
|
enforce(option == SocketOption.SNDTIMEO || option == SocketOption.RCVTIMEO,
|
|
new SocketParameterException("Not a valid timeout option: " ~ to!string(option)));
|
|
|
|
enforce(value >= dur!"hnsecs"(0), new SocketParameterException(
|
|
"Timeout duration must not be negative."));
|
|
|
|
version (Windows)
|
|
{
|
|
import std.algorithm.comparison : max;
|
|
|
|
auto msecs = to!int(value.total!"msecs");
|
|
if (msecs != 0 && option == SocketOption.RCVTIMEO)
|
|
msecs = max(1, msecs - WINSOCK_TIMEOUT_SKEW);
|
|
setOption(level, option, msecs);
|
|
}
|
|
else version (Posix)
|
|
{
|
|
_ctimeval tv;
|
|
value.split!("seconds", "usecs")(tv.tv_sec, tv.tv_usec);
|
|
setOption(level, option, (&tv)[0 .. 1]);
|
|
}
|
|
else static assert(false);
|
|
}
|
|
|
|
/**
|
|
* Get a text description of this socket's error status, and clear the
|
|
* socket's error status.
|
|
*/
|
|
string getErrorText()
|
|
{
|
|
int32_t error;
|
|
getOption(SocketOptionLevel.SOCKET, SocketOption.ERROR, error);
|
|
return formatSocketError(error);
|
|
}
|
|
|
|
/**
|
|
* Enables TCP keep-alive with the specified parameters.
|
|
*
|
|
* Params:
|
|
* time = Number of seconds with no activity until the first
|
|
* keep-alive packet is sent.
|
|
* interval = Number of seconds between when successive keep-alive
|
|
* packets are sent if no acknowledgement is received.
|
|
*
|
|
* Throws: `SocketOSException` if setting the options fails, or
|
|
* `SocketFeatureException` if setting keep-alive parameters is
|
|
* unsupported on the current platform.
|
|
*/
|
|
void setKeepAlive(int time, int interval) @trusted
|
|
{
|
|
version (Windows)
|
|
{
|
|
tcp_keepalive options;
|
|
options.onoff = 1;
|
|
options.keepalivetime = time * 1000;
|
|
options.keepaliveinterval = interval * 1000;
|
|
uint cbBytesReturned;
|
|
enforce(WSAIoctl(sock, SIO_KEEPALIVE_VALS,
|
|
&options, options.sizeof,
|
|
null, 0,
|
|
&cbBytesReturned, null, null) == 0,
|
|
new SocketOSException("Error setting keep-alive"));
|
|
}
|
|
else
|
|
static if (is(typeof(TCP_KEEPIDLE)) && is(typeof(TCP_KEEPINTVL)))
|
|
{
|
|
setOption(SocketOptionLevel.TCP, cast(SocketOption) TCP_KEEPIDLE, time);
|
|
setOption(SocketOptionLevel.TCP, cast(SocketOption) TCP_KEEPINTVL, interval);
|
|
setOption(SocketOptionLevel.SOCKET, SocketOption.KEEPALIVE, true);
|
|
}
|
|
else
|
|
throw new SocketFeatureException("Setting keep-alive options " ~
|
|
"is not supported on this platform");
|
|
}
|
|
|
|
/**
|
|
* Wait for a socket to change status. A wait timeout of $(REF Duration, core, time) or
|
|
* `TimeVal`, may be specified; if a timeout is not specified or the
|
|
* `TimeVal` is `null`, the maximum timeout is used. The `TimeVal`
|
|
* timeout has an unspecified value when `select` returns.
|
|
* Returns: The number of sockets with status changes, `0` on timeout,
|
|
* or `-1` on interruption. If the return value is greater than `0`,
|
|
* the `SocketSets` are updated to only contain the sockets having status
|
|
* changes. For a connecting socket, a write status change means the
|
|
* connection is established and it's able to send. For a listening socket,
|
|
* a read status change means there is an incoming connection request and
|
|
* it's able to accept.
|
|
*
|
|
* `SocketSet`'s updated to include only those sockets which an event occured.
|
|
* For a `connect()`ing socket, writeability means connected.
|
|
* For a `listen()`ing socket, readability means listening
|
|
* `Winsock`; possibly internally limited to 64 sockets per set.
|
|
*
|
|
* Returns:
|
|
* the number of events, 0 on timeout, or -1 on interruption
|
|
*/
|
|
static int select(SocketSet checkRead, SocketSet checkWrite, SocketSet checkError, Duration timeout) @trusted
|
|
{
|
|
auto vals = timeout.split!("seconds", "usecs")();
|
|
TimeVal tv;
|
|
tv.seconds = cast(tv.tv_sec_t ) vals.seconds;
|
|
tv.microseconds = cast(tv.tv_usec_t) vals.usecs;
|
|
return select(checkRead, checkWrite, checkError, &tv);
|
|
}
|
|
|
|
/// ditto
|
|
//maximum timeout
|
|
static int select(SocketSet checkRead, SocketSet checkWrite, SocketSet checkError)
|
|
{
|
|
return select(checkRead, checkWrite, checkError, null);
|
|
}
|
|
|
|
/// Ditto
|
|
static int select(SocketSet checkRead, SocketSet checkWrite, SocketSet checkError, TimeVal* timeout) @trusted
|
|
in
|
|
{
|
|
//make sure none of the SocketSet's are the same object
|
|
if (checkRead)
|
|
{
|
|
assert(checkRead !is checkWrite);
|
|
assert(checkRead !is checkError);
|
|
}
|
|
if (checkWrite)
|
|
{
|
|
assert(checkWrite !is checkError);
|
|
}
|
|
}
|
|
do
|
|
{
|
|
fd_set* fr, fw, fe;
|
|
int n = 0;
|
|
|
|
version (Windows)
|
|
{
|
|
// Windows has a problem with empty fd_set`s that aren't null.
|
|
fr = checkRead && checkRead.count ? checkRead.toFd_set() : null;
|
|
fw = checkWrite && checkWrite.count ? checkWrite.toFd_set() : null;
|
|
fe = checkError && checkError.count ? checkError.toFd_set() : null;
|
|
}
|
|
else
|
|
{
|
|
if (checkRead)
|
|
{
|
|
fr = checkRead.toFd_set();
|
|
n = checkRead.selectn();
|
|
}
|
|
else
|
|
{
|
|
fr = null;
|
|
}
|
|
|
|
if (checkWrite)
|
|
{
|
|
fw = checkWrite.toFd_set();
|
|
int _n;
|
|
_n = checkWrite.selectn();
|
|
if (_n > n)
|
|
n = _n;
|
|
}
|
|
else
|
|
{
|
|
fw = null;
|
|
}
|
|
|
|
if (checkError)
|
|
{
|
|
fe = checkError.toFd_set();
|
|
int _n;
|
|
_n = checkError.selectn();
|
|
if (_n > n)
|
|
n = _n;
|
|
}
|
|
else
|
|
{
|
|
fe = null;
|
|
}
|
|
|
|
// Make sure the sets' capacity matches, to avoid select reading
|
|
// out of bounds just because one set was bigger than another
|
|
if (checkRead ) checkRead .setMinCapacity(n);
|
|
if (checkWrite) checkWrite.setMinCapacity(n);
|
|
if (checkError) checkError.setMinCapacity(n);
|
|
}
|
|
|
|
int result = .select(n, fr, fw, fe, &timeout.ctimeval);
|
|
|
|
version (Windows)
|
|
{
|
|
if (_SOCKET_ERROR == result && WSAGetLastError() == WSAEINTR)
|
|
return -1;
|
|
}
|
|
else version (Posix)
|
|
{
|
|
if (_SOCKET_ERROR == result && errno == EINTR)
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
static assert(0);
|
|
}
|
|
|
|
if (_SOCKET_ERROR == result)
|
|
throw new SocketOSException("Socket select error");
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
/**
|
|
* Can be overridden to support other addresses.
|
|
* Returns: A new `Address` object for the current address family.
|
|
*/
|
|
protected Address createAddress() pure nothrow
|
|
{
|
|
Address result;
|
|
switch (_family)
|
|
{
|
|
static if (is(sockaddr_un))
|
|
{
|
|
case AddressFamily.UNIX:
|
|
result = new UnixAddress;
|
|
break;
|
|
}
|
|
|
|
case AddressFamily.INET:
|
|
result = new InternetAddress;
|
|
break;
|
|
|
|
case AddressFamily.INET6:
|
|
result = new Internet6Address;
|
|
break;
|
|
|
|
default:
|
|
result = new UnknownAddress;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/// Shortcut class for a TCP Socket.
|
|
class TcpSocket: Socket
|
|
{
|
|
/// Constructs a blocking TCP Socket.
|
|
this(AddressFamily family)
|
|
{
|
|
super(family, SocketType.STREAM, ProtocolType.TCP);
|
|
}
|
|
|
|
/// Constructs a blocking IPv4 TCP Socket.
|
|
this()
|
|
{
|
|
this(AddressFamily.INET);
|
|
}
|
|
|
|
|
|
//shortcut
|
|
/// Constructs a blocking TCP Socket and connects to the given `Address`.
|
|
this(Address connectTo)
|
|
{
|
|
this(connectTo.addressFamily);
|
|
connect(connectTo);
|
|
}
|
|
}
|
|
|
|
|
|
/// Shortcut class for a UDP Socket.
|
|
class UdpSocket: Socket
|
|
{
|
|
/// Constructs a blocking UDP Socket.
|
|
this(AddressFamily family)
|
|
{
|
|
super(family, SocketType.DGRAM, ProtocolType.UDP);
|
|
}
|
|
|
|
|
|
/// Constructs a blocking IPv4 UDP Socket.
|
|
this()
|
|
{
|
|
this(AddressFamily.INET);
|
|
}
|
|
}
|
|
|
|
@safe unittest
|
|
{
|
|
byte[] buf;
|
|
buf.length = 1;
|
|
Address addr;
|
|
auto s = new UdpSocket;
|
|
s.blocking = false;
|
|
s.bind(new InternetAddress(InternetAddress.PORT_ANY));
|
|
s.receiveFrom(buf, addr);
|
|
}
|
|
|
|
// https://issues.dlang.org/show_bug.cgi?id=16514
|
|
@safe unittest
|
|
{
|
|
void checkAttributes(string attributes)()
|
|
{
|
|
mixin(attributes ~ q{ void function() fun = {};});
|
|
fun();
|
|
}
|
|
|
|
class TestSocket : Socket
|
|
{
|
|
override
|
|
{
|
|
@property pure nothrow @nogc @safe socket_t handle() const
|
|
{
|
|
checkAttributes!q{pure nothrow @nogc @safe}; assert(0);
|
|
}
|
|
@property nothrow @nogc @trusted bool blocking() const
|
|
{
|
|
checkAttributes!q{nothrow @nogc @trusted}; assert(0);
|
|
}
|
|
@property @trusted void blocking(bool byes)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@property @safe AddressFamily addressFamily()
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@property @trusted bool isAlive() const
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@trusted void bind(Address addr)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@trusted void connect(Address to)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@trusted void listen(int backlog)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
protected pure nothrow @safe Socket accepting()
|
|
{
|
|
checkAttributes!q{pure nothrow @safe}; assert(0);
|
|
}
|
|
@trusted Socket accept()
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
nothrow @nogc @trusted void shutdown(SocketShutdown how)
|
|
{
|
|
checkAttributes!q{nothrow @nogc @trusted};
|
|
}
|
|
nothrow @nogc @trusted scope void close()
|
|
{
|
|
checkAttributes!q{nothrow @nogc @trusted};
|
|
}
|
|
@property @trusted Address remoteAddress()
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@property @trusted Address localAddress()
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@trusted ptrdiff_t send(scope const(void)[] buf, SocketFlags flags)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@safe ptrdiff_t send(scope const(void)[] buf)
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted ptrdiff_t sendTo(scope const(void)[] buf, SocketFlags flags, Address to)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@safe ptrdiff_t sendTo(scope const(void)[] buf, Address to)
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted ptrdiff_t sendTo(scope const(void)[] buf, SocketFlags flags)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@safe ptrdiff_t sendTo(scope const(void)[] buf)
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted ptrdiff_t receive(scope void[] buf, SocketFlags flags)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@safe ptrdiff_t receive(scope void[] buf)
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted ptrdiff_t receiveFrom(scope void[] buf, SocketFlags flags, ref Address from)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@safe ptrdiff_t receiveFrom(scope void[] buf, ref Address from)
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted ptrdiff_t receiveFrom(scope void[] buf, SocketFlags flags)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@safe ptrdiff_t receiveFrom(scope void[] buf)
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted int getOption(SocketOptionLevel level, SocketOption option, scope void[] result)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@trusted int getOption(SocketOptionLevel level, SocketOption option, out int32_t result)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@trusted int getOption(SocketOptionLevel level, SocketOption option, out Linger result)
|
|
{
|
|
checkAttributes!q{@trusted}; assert(0);
|
|
}
|
|
@trusted void getOption(SocketOptionLevel level, SocketOption option, out Duration result)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@trusted void setOption(SocketOptionLevel level, SocketOption option, scope void[] value)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@trusted void setOption(SocketOptionLevel level, SocketOption option, int32_t value)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@trusted void setOption(SocketOptionLevel level, SocketOption option, Linger value)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@trusted void setOption(SocketOptionLevel level, SocketOption option, Duration value)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
@safe string getErrorText()
|
|
{
|
|
checkAttributes!q{@safe}; assert(0);
|
|
}
|
|
@trusted void setKeepAlive(int time, int interval)
|
|
{
|
|
checkAttributes!q{@trusted};
|
|
}
|
|
protected pure nothrow @safe Address createAddress()
|
|
{
|
|
checkAttributes!q{pure nothrow @safe}; assert(0);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Creates a pair of connected sockets.
|
|
*
|
|
* The two sockets are indistinguishable.
|
|
*
|
|
* Throws: `SocketException` if creation of the sockets fails.
|
|
*/
|
|
Socket[2] socketPair() @trusted
|
|
{
|
|
version (Posix)
|
|
{
|
|
int[2] socks;
|
|
if (socketpair(AF_UNIX, SOCK_STREAM, 0, socks) == -1)
|
|
throw new SocketOSException("Unable to create socket pair");
|
|
|
|
Socket toSocket(size_t id)
|
|
{
|
|
auto s = new Socket;
|
|
s.setSock(cast(socket_t) socks[id]);
|
|
s._family = AddressFamily.UNIX;
|
|
return s;
|
|
}
|
|
|
|
return [toSocket(0), toSocket(1)];
|
|
}
|
|
else version (Windows)
|
|
{
|
|
// We do not have socketpair() on Windows, just manually create a
|
|
// pair of sockets connected over some localhost port.
|
|
Socket[2] result;
|
|
|
|
auto listener = new TcpSocket();
|
|
listener.setOption(SocketOptionLevel.SOCKET, SocketOption.REUSEADDR, true);
|
|
listener.bind(new InternetAddress(INADDR_LOOPBACK, InternetAddress.PORT_ANY));
|
|
auto addr = listener.localAddress;
|
|
listener.listen(1);
|
|
|
|
result[0] = new TcpSocket(addr);
|
|
result[1] = listener.accept();
|
|
|
|
listener.close();
|
|
return result;
|
|
}
|
|
else
|
|
static assert(false);
|
|
}
|
|
|
|
///
|
|
@safe unittest
|
|
{
|
|
immutable ubyte[4] data = [1, 2, 3, 4];
|
|
auto pair = socketPair();
|
|
scope(exit) foreach (s; pair) s.close();
|
|
|
|
pair[0].send(data[]);
|
|
|
|
auto buf = new ubyte[data.length];
|
|
pair[1].receive(buf);
|
|
assert(buf == data);
|
|
}
|