debian-packages/vibe-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Make the test for #115 actually fail if a socket gets leaked.

Browse source
This commit is contained in:
Sönke Ludwig 2019-01-19 15:26:18 +01:00
parent 568cdb112b
commit 00bf61b3b7
1 changed files with 32 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
tests/issue-115-connect-fail-leak.d
View file

@ -10,13 +10,36 @@ import vibe.core.net;
void main()
{
foreach (_; 0 .. 20) {
TCPConnection conn;
try {
conn = connectTCP("127.0.0.1", 16565);
logError("Connection: %s", conn);
conn.close();
assert(false, "Didn't expect TCP connection on port 16565 to succeed");
} catch (Exception) { }
auto initial = determineSocketCount();
TCPConnection conn;
try {
conn = connectTCP("127.0.0.1", 16565);
logError("Connection: %s", conn);
conn.close();
assert(false, "Didn't expect TCP connection on port 16565 to succeed");
} catch (Exception) { }
assert(determineSocketCount() == initial, "Sockets leaked!");
}
size_t determineSocketCount()
{
import std.algorithm.searching : count;
import std.range : iota;
version (Posix) {
import core.sys.posix.sys.resource : getrlimit, rlimit, RLIMIT_NOFILE;
import core.sys.posix.fcntl : fcntl, F_GETFD;
rlimit rl;
enforce(getrlimit(RLIMIT_NOFILE, &rl) == 0);
return iota(rl.rlim_cur).count(fd => fcntl(fd, F_GETFD) != -1);
} else {
import core.sys.windows.winsock2 : getsockopt, SOL_SOCKET, SO_TYPE;
int st;
int stlen = st.sizeof;
return iota(65536).count!(s => getsockopt(s, SOL_SOCKET, SO_TYPE, cast(void*)&st, &stlen) == 0);
}
}
}