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

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

@ -10,7 +10,8 @@ import vibe.core.net;
void main() void main()
{ {
foreach (_; 0 .. 20) { auto initial = determineSocketCount();
TCPConnection conn; TCPConnection conn;
try { try {
conn = connectTCP("127.0.0.1", 16565); conn = connectTCP("127.0.0.1", 16565);
@ -18,5 +19,27 @@ void main()
conn.close(); conn.close();
assert(false, "Didn't expect TCP connection on port 16565 to succeed"); assert(false, "Didn't expect TCP connection on port 16565 to succeed");
} catch (Exception) { } } 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);
} }
} }