Change shutdown behavior to allow graceful handle leaks.
This avoids crashing in case of any handle references left-over in GC allocated memory that gets finalized after module destructors have already been run.
This commit is contained in:
parent
38a4e4ca5f
commit
e2e8bf30aa
|
@ -22,7 +22,7 @@ else alias NativeEventDriver = EventDriver;
|
||||||
assert(s_driver !is null, "setupEventDriver() was not called for this thread.");
|
assert(s_driver !is null, "setupEventDriver() was not called for this thread.");
|
||||||
else {
|
else {
|
||||||
if (!s_driver) {
|
if (!s_driver) {
|
||||||
s_driver = mallocT!NativeEventDriver();
|
s_driver = mallocT!NativeEventDriver;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return s_driver;
|
return s_driver;
|
||||||
|
@ -43,8 +43,8 @@ static if (!is(NativeEventDriver == EventDriver)) {
|
||||||
if (!s_isMainThread) {
|
if (!s_isMainThread) {
|
||||||
if (!--s_initCount) {
|
if (!--s_initCount) {
|
||||||
if (s_driver) {
|
if (s_driver) {
|
||||||
s_driver.dispose();
|
if (s_driver.dispose())
|
||||||
freeT(s_driver);
|
freeT(s_driver);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -61,8 +61,8 @@ static if (!is(NativeEventDriver == EventDriver)) {
|
||||||
{
|
{
|
||||||
if (!--s_initCount) {
|
if (!--s_initCount) {
|
||||||
if (s_driver) {
|
if (s_driver) {
|
||||||
s_driver.dispose();
|
if (s_driver.dispose())
|
||||||
freeT(s_driver);
|
freeT(s_driver);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -52,8 +52,13 @@ interface EventDriver {
|
||||||
/// Directory change watching
|
/// Directory change watching
|
||||||
@property inout(EventDriverWatchers) watchers() inout;
|
@property inout(EventDriverWatchers) watchers() inout;
|
||||||
|
|
||||||
/// Releases all resources associated with the driver
|
/** Releases all resources associated with the driver.
|
||||||
void dispose();
|
|
||||||
|
In case of any left-over referenced handles, this function returns
|
||||||
|
`false` and does not free any resources. It may choose to free the
|
||||||
|
resources once the last handle gets dereferenced.
|
||||||
|
*/
|
||||||
|
bool dispose();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -88,9 +88,29 @@ final class PosixEventDriver(Loop : PosixEventLoop) : EventDriver {
|
||||||
final override @property inout(FileDriver) files() inout { return m_files; }
|
final override @property inout(FileDriver) files() inout { return m_files; }
|
||||||
final override @property inout(WatcherDriver) watchers() inout { return m_watchers; }
|
final override @property inout(WatcherDriver) watchers() inout { return m_watchers; }
|
||||||
|
|
||||||
final override void dispose()
|
final override bool dispose()
|
||||||
{
|
{
|
||||||
if (!m_loop) return;
|
import core.thread : Thread;
|
||||||
|
import taggedalgebraic : hasType;
|
||||||
|
|
||||||
|
if (!m_loop) return true;
|
||||||
|
|
||||||
|
static string getThreadName()
|
||||||
|
{
|
||||||
|
string thname;
|
||||||
|
try thname = Thread.getThis().name;
|
||||||
|
catch (Exception e) assert(false, e.msg);
|
||||||
|
return thname.length ? thname : "unknown";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (m_loop.m_handleCount > 0) {
|
||||||
|
print("Warning (thread: %s): leaking eventcore driver because there are still active handles", getThreadName());
|
||||||
|
foreach (id, ref s; m_loop.m_fds)
|
||||||
|
if (!s.specific.hasType!(typeof(null)) && !(s.common.flags & FDFlags.internal))
|
||||||
|
print(" FD %s (%s)", id, s.specific.kind);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
m_files.dispose();
|
m_files.dispose();
|
||||||
m_dns.dispose();
|
m_dns.dispose();
|
||||||
m_core.dispose();
|
m_core.dispose();
|
||||||
|
@ -108,6 +128,8 @@ final class PosixEventDriver(Loop : PosixEventLoop) : EventDriver {
|
||||||
freeT(m_loop);
|
freeT(m_loop);
|
||||||
} ();
|
} ();
|
||||||
catch (Exception e) assert(false, e.msg);
|
catch (Exception e) assert(false, e.msg);
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -279,6 +301,7 @@ package class PosixEventLoop {
|
||||||
|
|
||||||
package {
|
package {
|
||||||
AlgebraicChoppedVector!(FDSlot, StreamSocketSlot, StreamListenSocketSlot, DgramSocketSlot, DNSSlot, WatcherSlot, EventSlot, SignalSlot) m_fds;
|
AlgebraicChoppedVector!(FDSlot, StreamSocketSlot, StreamListenSocketSlot, DgramSocketSlot, DNSSlot, WatcherSlot, EventSlot, SignalSlot) m_fds;
|
||||||
|
size_t m_handleCount = 0;
|
||||||
size_t m_waiterCount = 0;
|
size_t m_waiterCount = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -338,6 +361,8 @@ package class PosixEventLoop {
|
||||||
common.flags = flags;
|
common.flags = flags;
|
||||||
specific = slot_init;
|
specific = slot_init;
|
||||||
}
|
}
|
||||||
|
if (!(flags & FDFlags.internal))
|
||||||
|
m_handleCount++;
|
||||||
}
|
}
|
||||||
|
|
||||||
package void clearFD(T)(FD fd)
|
package void clearFD(T)(FD fd)
|
||||||
|
@ -347,6 +372,10 @@ package class PosixEventLoop {
|
||||||
auto slot = () @trusted { return &m_fds[fd.value]; } ();
|
auto slot = () @trusted { return &m_fds[fd.value]; } ();
|
||||||
assert(slot.common.refCount == 0, "Clearing referenced file descriptor slot.");
|
assert(slot.common.refCount == 0, "Clearing referenced file descriptor slot.");
|
||||||
assert(slot.specific.hasType!T, "Clearing file descriptor slot with unmatched type.");
|
assert(slot.specific.hasType!T, "Clearing file descriptor slot with unmatched type.");
|
||||||
|
|
||||||
|
if (!(slot.common.flags & FDFlags.internal))
|
||||||
|
m_handleCount--;
|
||||||
|
|
||||||
if (slot.common.userDataDestructor)
|
if (slot.common.userDataDestructor)
|
||||||
() @trusted { slot.common.userDataDestructor(slot.common.userData.ptr); } ();
|
() @trusted { slot.common.userDataDestructor(slot.common.userData.ptr); } ();
|
||||||
if (!(slot.common.flags & FDFlags.internal))
|
if (!(slot.common.flags & FDFlags.internal))
|
||||||
|
|
|
@ -5,7 +5,7 @@ version (Windows):
|
||||||
import eventcore.driver;
|
import eventcore.driver;
|
||||||
import eventcore.drivers.timer;
|
import eventcore.drivers.timer;
|
||||||
import eventcore.internal.consumablequeue;
|
import eventcore.internal.consumablequeue;
|
||||||
import eventcore.internal.utils : mallocT, freeT, nogc_assert;
|
import eventcore.internal.utils : mallocT, freeT, nogc_assert, print;
|
||||||
import eventcore.internal.win32;
|
import eventcore.internal.win32;
|
||||||
import core.sync.mutex : Mutex;
|
import core.sync.mutex : Mutex;
|
||||||
import core.time : Duration;
|
import core.time : Duration;
|
||||||
|
@ -61,6 +61,29 @@ final class WinAPIEventDriverCore : EventDriverCore {
|
||||||
} catch (Exception e) assert(false, e.msg);
|
} catch (Exception e) assert(false, e.msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
package bool checkForLeakedHandles()
|
||||||
|
@trusted {
|
||||||
|
import core.thread : Thread;
|
||||||
|
|
||||||
|
static string getThreadName()
|
||||||
|
{
|
||||||
|
string thname;
|
||||||
|
try thname = Thread.getThis().name;
|
||||||
|
catch (Exception e) assert(false, e.msg);
|
||||||
|
return thname.length ? thname : "unknown";
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (k; m_handles.byKey) {
|
||||||
|
print("Warning (thread: %s): Leaked handles detected at driver shutdown", getThreadName());
|
||||||
|
foreach (ks; m_handles.byKeyValue)
|
||||||
|
if (!ks.value.specific.hasType!(typeof(null)))
|
||||||
|
print(" FD %04X (%s)", ks.key, ks.value.specific.kind);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
override size_t waiterCount() { return m_waiterCount + m_timers.pendingCount; }
|
override size_t waiterCount() { return m_waiterCount + m_timers.pendingCount; }
|
||||||
|
|
||||||
package void addWaiter() @nogc { m_waiterCount++; }
|
package void addWaiter() @nogc { m_waiterCount++; }
|
||||||
|
|
|
@ -74,9 +74,14 @@ final class WinAPIEventDriver : EventDriver {
|
||||||
override @property inout(WinAPIEventDriverSignals) signals() inout { return m_signals; }
|
override @property inout(WinAPIEventDriverSignals) signals() inout { return m_signals; }
|
||||||
override @property inout(WinAPIEventDriverWatchers) watchers() inout { return m_watchers; }
|
override @property inout(WinAPIEventDriverWatchers) watchers() inout { return m_watchers; }
|
||||||
|
|
||||||
override void dispose()
|
override bool dispose()
|
||||||
{
|
{
|
||||||
if (!m_events) return;
|
if (!m_events) return true;
|
||||||
|
|
||||||
|
if (m_core.checkForLeakedHandles()) return false;
|
||||||
|
if (m_events.checkForLeakedHandles()) return false;
|
||||||
|
if (m_sockets.checkForLeakedHandles()) return false;
|
||||||
|
|
||||||
m_events.dispose();
|
m_events.dispose();
|
||||||
m_core.dispose();
|
m_core.dispose();
|
||||||
assert(threadInstance !is null);
|
assert(threadInstance !is null);
|
||||||
|
@ -93,5 +98,7 @@ final class WinAPIEventDriver : EventDriver {
|
||||||
freeT(m_signals);
|
freeT(m_signals);
|
||||||
} ();
|
} ();
|
||||||
catch (Exception e) assert(false, e.msg);
|
catch (Exception e) assert(false, e.msg);
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,7 +6,7 @@ import eventcore.driver;
|
||||||
import eventcore.drivers.winapi.core;
|
import eventcore.drivers.winapi.core;
|
||||||
import eventcore.internal.win32;
|
import eventcore.internal.win32;
|
||||||
import eventcore.internal.consumablequeue;
|
import eventcore.internal.consumablequeue;
|
||||||
import eventcore.internal.utils : mallocT, freeT, nogc_assert;
|
import eventcore.internal.utils : mallocT, freeT, nogc_assert, print;
|
||||||
|
|
||||||
|
|
||||||
final class WinAPIEventDriverEvents : EventDriverEvents {
|
final class WinAPIEventDriverEvents : EventDriverEvents {
|
||||||
|
@ -45,6 +45,15 @@ final class WinAPIEventDriverEvents : EventDriverEvents {
|
||||||
freeT(m_pending);
|
freeT(m_pending);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
package bool checkForLeakedHandles()
|
||||||
|
@trusted {
|
||||||
|
foreach (evt; m_events.byKey) {
|
||||||
|
print("Warning: Event handles leaked at driver shutdown.");
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
override EventID create()
|
override EventID create()
|
||||||
{
|
{
|
||||||
auto id = EventID(m_idCounter++);
|
auto id = EventID(m_idCounter++);
|
||||||
|
|
|
@ -18,6 +18,7 @@ final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
private {
|
private {
|
||||||
alias SocketVector = AlgebraicChoppedVector!(SocketSlot, StreamSocketSlot, StreamListenSocketSlot, DatagramSocketSlot);
|
alias SocketVector = AlgebraicChoppedVector!(SocketSlot, StreamSocketSlot, StreamListenSocketSlot, DatagramSocketSlot);
|
||||||
SocketVector m_sockets;
|
SocketVector m_sockets;
|
||||||
|
size_t m_socketCount = 0;
|
||||||
WinAPIEventDriverCore m_core;
|
WinAPIEventDriverCore m_core;
|
||||||
DWORD m_tid;
|
DWORD m_tid;
|
||||||
HWND m_hwnd;
|
HWND m_hwnd;
|
||||||
|
@ -38,6 +39,13 @@ final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
|
|
||||||
package @property size_t waiterCount() const { return m_waiters; }
|
package @property size_t waiterCount() const { return m_waiters; }
|
||||||
|
|
||||||
|
package bool checkForLeakedHandles()
|
||||||
|
{
|
||||||
|
if (m_socketCount == 0) return false;
|
||||||
|
print("Warning: Socket handles leaked at driver shutdown.");
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
override StreamSocketFD connectStream(scope Address peer_address, scope Address bind_address, ConnectCallback on_connect)
|
override StreamSocketFD connectStream(scope Address peer_address, scope Address bind_address, ConnectCallback on_connect)
|
||||||
@trusted {
|
@trusted {
|
||||||
assert(m_tid == GetCurrentThreadId());
|
assert(m_tid == GetCurrentThreadId());
|
||||||
|
@ -802,6 +810,7 @@ final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
|
|
||||||
private void initSocketSlot(SocketFD fd)
|
private void initSocketSlot(SocketFD fd)
|
||||||
{
|
{
|
||||||
|
m_socketCount++;
|
||||||
m_sockets[fd.value].common.refCount = 1;
|
m_sockets[fd.value].common.refCount = 1;
|
||||||
m_sockets[fd.value].common.fd = fd;
|
m_sockets[fd.value].common.fd = fd;
|
||||||
m_sockets[fd.value].common.driver = this;
|
m_sockets[fd.value].common.driver = this;
|
||||||
|
@ -809,6 +818,7 @@ final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
|
|
||||||
package void clearSocketSlot(FD fd)
|
package void clearSocketSlot(FD fd)
|
||||||
@nogc {
|
@nogc {
|
||||||
|
m_socketCount--;
|
||||||
auto slot = () @trusted { return &m_sockets[fd]; } ();
|
auto slot = () @trusted { return &m_sockets[fd]; } ();
|
||||||
if (slot.common.userDataDestructor)
|
if (slot.common.userDataDestructor)
|
||||||
() @trusted { slot.common.userDataDestructor(slot.common.userData.ptr); } ();
|
() @trusted { slot.common.userDataDestructor(slot.common.userData.ptr); } ();
|
||||||
|
|
Loading…
Reference in a new issue