Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Add copy buttons to all
 blocks
(function() {
function addCopyButtons() {
document.querySelectorAll('pre code').forEach(function(codeBlock) {
if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;
codeBlock.parentElement.setAttribute('data-copy-added', 'true');
var btn = document.createElement('button');
btn.textContent = 'Copy';
btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';
btn.onmouseover = function() { this.style.opacity = '1'; };
btn.onmouseout = function() { this.style.opacity = '0.7'; };
btn.onclick = function() {
navigator.clipboard.writeText(codeBlock.textContent).then(function() {
btn.textContent = 'Copied!';
setTimeout(function() { btn.textContent = 'Copy'; }, 1500);
});
};
codeBlock.parentElement.style.position = 'relative';
codeBlock.parentElement.appendChild(btn);
});
}
addCopyButtons();
// Re-run on dynamic content
var observer = new MutationObserver(addCopyButtons);
observer.observe(document.body, { childList: true, subtree: true });
})();
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Force GitHub README to respect dark mode (function() { var style = document.createElement('style'); style.textContent = ' .markdown-body { color-scheme: dark light; } .markdown-body pre { background: #161b22 !important; } .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; } .markdown-body table th, .markdown-body table td { border-color: #30363d !important; } .markdown-body img { background: #0d1117; } .markdown-body blockquote { border-left-color: #8b949e; } .markdown-body hr { border-color: #30363d; } '; document.head.appendChild(style); })(); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + ' Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Highlight search terms from Google/DuckDuckGo/Bing referrer (function() { var ref = document.referrer; var terms = []; if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) { var url = new URL(ref); var q = url.searchParams.get('q') || url.searchParams.get('p'); if (q) { terms = q.split(/\s+/).filter(function(t) { return t.length > 2; }); } } if (terms.length === 0) return; var style = document.createElement('style'); style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }'; document.head.appendChild(style); function highlight(node) { if (node.nodeType === 3) { // text node var text = node.textContent; var found = false; terms.forEach(function(term) { var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\]\\]/g, '\\') + ')', 'gi'); if (regex.test(text)) { found = true; var frag = document.createDocumentFragment(); var parts = text.split(regex); parts.forEach(function(part, i) { if (i % 2 === 0) { frag.appendChild(document.createTextNode(part)); } else { var span = document.createElement('span'); span.className = 'userscript-highlight'; span.textContent = part; frag.appendChild(span); } }); node.parentNode.replaceChild(frag, node); } }); } else if (node.nodeType === 1 && node.childNodes) { // element var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT']; if (!skipTags.includes(node.tagName)) { Array.from(node.childNodes).forEach(highlight); } } } highlight(document.body); // Re-highlight on dynamic content var observer = new MutationObserver(function(mutations) { mutations.forEach(function(m) { m.addedNodes.forEach(function(node) { if (node.nodeType === 1 || node.nodeType === 3) highlight(node); }); }); }); observer.observe(document.body, { childList: true, subtree: true }); })(); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + ' Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Strip utm_, fbclid, gclid, etc. from all links on page (function() { var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content', 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid', 'ref', 'ref_src', 'source', 'medium', 'campaign']; function cleanUrl(url) { try { var u = new URL(url, window.location.origin); var changed = false; trackingParams.forEach(function(p) { if (u.searchParams.has(p)) { u.searchParams.delete(p); changed = true; } }); return changed ? u.toString() : url; } catch (e) { return url; } } function cleanLinks() { document.querySelectorAll('a[href]').forEach(function(a) { var clean = cleanUrl(a.href); if (clean !== a.href) a.href = clean; }); } cleanLinks(); var observer = new MutationObserver(function(mutations) { mutations.forEach(function(m) { m.addedNodes.forEach(function(node) { if (node.nodeType === 1) { if (node.tagName === 'A') cleanLinks(); node.querySelectorAll('a[href]').forEach(function(a) { var clean = cleanUrl(a.href); if (clean !== a.href) a.href = clean; }); } }); }); }); observer.observe(document.body, { childList: true, subtree: true }); })(); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + ' Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Auto-enable theater mode on YouTube (function() { function tryTheater() { var btn = document.querySelector('button[aria-label="Theater mode"], ytd-player #player button[title="Theater mode"]'); if (btn && !btn.classList.contains('activated')) { btn.click(); } } // Try immediately tryTheater(); // Try after navigation (SPA) var lastUrl = location.href; setInterval(function() { if (location.href !== lastUrl) { lastUrl = location.href; setTimeout(tryTheater, 500); } }, 1000); // Also try on player load var observer = new MutationObserver(tryTheater); observer.observe(document.body, { childList: true, subtree: true }); })(); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + ' Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Remove or un-stick sticky/fixed headers that block content (function() { function unstick() { document.querySelectorAll('header, nav, [role="banner"], .header, .navbar, .sticky, .fixed-top, [style*="position: fixed"], [style*="position:sticky"]').forEach(function(el) { if (el.style.position === 'fixed' || el.style.position === 'sticky' || getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') { el.style.position = 'static'; el.style.top = 'auto'; el.style.zIndex = 'auto'; } }); } unstick(); var observer = new MutationObserver(unstick); observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] }); })(); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + ' Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5
, 'i'); if (__m === '*' || __re.test(location.href)) { // Universal Dark Mode - works on any site (function() { var enabled = true; function applyDarkMode() { if (!enabled) return; // Create style element if it doesn't exist var style = document.getElementById('universal-dark-mode-style'); if (!style) { style = document.createElement('style'); style.id = 'universal-dark-mode-style'; document.head.appendChild(style); } // Dark mode CSS - inverts colors but preserves images/video style.textContent = ' /* Invert everything except media */ html { filter: invert(1) hue-rotate(180deg) !important; background: #1a1a2e !important; } /* Restore images, videos, iframes, canvas */ img, video, iframe, canvas, svg, picture, [style*="background-image"] { filter: invert(1) hue-rotate(180deg) !important; } /* Preserve specific elements that should not be inverted */ .no-dark-mode, .no-dark-mode *, [data-theme="light"], [data-theme="light"], .ace_editor, .ace_editor *, .CodeMirror, .CodeMirror *, .monaco-editor, .monaco-editor *, .markdown-body pre, .markdown-body pre *, .highlight, .highlight *, pre code, pre code * { filter: none !important; } /* Fix common UI elements */ .modal, .popup, .dropdown-menu, .tooltip, .popover { filter: invert(1) hue-rotate(180deg) !important; background: #2d2d44 !important; border-color: #444 !important; } /* Scrollbars */ ::-webkit-scrollbar { background: #1a1a2e !important; } ::-webkit-scrollbar-thumb { background: #444 !important; } ::-webkit-scrollbar-thumb:hover { background: #555 !important; } /* Selection */ ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; } ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; } '; } function removeDarkMode() { var style = document.getElementById('universal-dark-mode-style'); if (style) style.remove(); } // Toggle with Alt+Shift+D document.addEventListener('keydown', function(e) { if (e.altKey && e.shiftKey && e.key === 'D') { e.preventDefault(); enabled = !enabled; if (enabled) { applyDarkMode(); console.log('[Universal Dark Mode] Enabled'); } else { removeDarkMode(); console.log('[Universal Dark Mode] Disabled'); } } }); // Apply on load applyDarkMode(); // Re-apply on dynamic content var observer = new MutationObserver(function(mutations) { if (enabled && !document.getElementById('universal-dark-mode-style')) { applyDarkMode(); } }); observer.observe(document.head, { childList: true }); console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle'); })(); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })(); Close the QUIC datagram transports on shutdown by CoryLowe5 · Pull Request #373 · pgjones/hypercorn · GitHub
Skip to content

Close the QUIC datagram transports on shutdown - #373

Open
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports
Open

Close the QUIC datagram transports on shutdown#373
CoryLowe5 wants to merge 1 commit into
pgjones:mainfrom
CoryLowe5:close-datagram-transports

Conversation

@CoryLowe5

Copy link
Copy Markdown

worker_serve keeps its TCP servers in servers and closes them in
its finally, but the transport returned by create_datagram_endpoint
is discarded, so shutdown cannot close what it never kept. Each
QUIC-serving worker leaks its datagram transport, visible as a
GC-time unclosed transport ResourceWarning under python -W always
(we found it via tracemalloc pointing into hypercorn/asyncio/run.py
from a test suite that serves QUIC in-process many times per run).

The fix keeps the datagram transports and closes them after the
graceful drain, alongside the existing server close. After the drain
rather than before it, deliberately: unlike a TCP Server — where
close() only stops new connections — the datagram transport is both
the listener and the data path for established QUIC connections, so
closing it earlier would break in-flight traffic during
graceful_timeout.

Includes a test (tests/asyncio/test_run.py) that serves one QUIC
socket through the real worker_serve and asserts every transport
created by create_datagram_endpoint is closing after it returns; it
skips when aioquic is not installed, matching the h3 extra. A
self-contained reproduction script is attached below in a comment for
convenience.

(The trio worker was inspected and left alone: it hands its socket to
trio.socket.from_stdlib_socket inside the UDPServer under the
nursery — different ownership shape, not measured, not touched.)

worker_serve keeps its TCP servers in `servers` and closes them in its
finally, but the transport returned by create_datagram_endpoint was
discarded, so shutdown could not close what it never kept. Each
QUIC-serving worker leaked its datagram transport, visible as a GC-time
"unclosed transport" ResourceWarning (python -W always).
Keep the datagram transports and close them after the graceful drain,
alongside the existing server close. Closing after the drain rather
than before it preserves in-flight QUIC traffic during
graceful_timeout: unlike a TCP Server, the datagram transport is both
the listener and the data path for established connections.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@CoryLowe5

Copy link
Copy Markdown
Author

Reproduction (self-contained — generates a throwaway self-signed certificate; run with python -W always repro_datagram_leak.py). On 0.18.0 it prints LEAK: 1 'unclosed transport' ResourceWarning(s); with this branch it prints CLEAN.

"""Reproduce: hypercorn's asyncio worker leaks its QUIC datagram transport.`worker_serve` keeps its TCP servers in `servers` and closes them in itsfinally, but the transport returned by `create_datagram_endpoint` isdiscarded into `_`, so shutdown cannot close what it never kept. Theleaked transport surfaces as a GC-time ResourceWarning.Run against hypercorn 0.18.0 with the h3 extra: python -W always repro_datagram_leak.pyExpected output on 0.18.0: LEAK: 1 'unclosed transport' ResourceWarning(s)With the fix: CLEAN: no unclosed-transport ResourceWarningSelf-contained: generates a throwaway self-signed certificate."""importasyncioimportdatetimeimportgcimporttempfileimportwarningsfrompathlibimportPathfromcryptographyimportx509fromcryptography.hazmat.primitivesimporthashes, serializationfromcryptography.hazmat.primitives.asymmetricimportecfromcryptography.x509.oidimportNameOIDfromhypercorn.app_wrappersimportASGIWrapperfromhypercorn.asyncio.runimportworker_servefromhypercorn.configimportConfigasyncdefapp(scope, receive, send):
ifscope["type"] =="lifespan":
whileTrue:
message=awaitreceive()
ifmessage["type"] =="lifespan.startup":
awaitsend({"type": "lifespan.startup.complete"})
elifmessage["type"] =="lifespan.shutdown":
awaitsend({"type": "lifespan.shutdown.complete"})
returndefwrite_self_signed(directory: Path) ->tuple[Path, Path]:
key=ec.generate_private_key(ec.SECP256R1())
name=x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, "localhost")])
now=datetime.datetime.now(datetime.timezone.utc)
cert= (
x509.CertificateBuilder()
.subject_name(name)
.issuer_name(name)
.public_key(key.public_key())
.serial_number(x509.random_serial_number())
.not_valid_before(now)
.not_valid_after(now+datetime.timedelta(days=1))
.sign(key, hashes.SHA256())
)
cert_path=directory/"cert.pem"key_path=directory/"key.pem"cert_path.write_bytes(cert.public_bytes(serialization.Encoding.PEM))
key_path.write_bytes(
key.private_bytes(
serialization.Encoding.PEM,
serialization.PrivateFormat.TraditionalOpenSSL,
serialization.NoEncryption(),
)
)
returncert_path, key_pathasyncdefmain() ->None:
withtempfile.TemporaryDirectory() astmp:
cert, key=write_self_signed(Path(tmp))
config=Config()
config.bind= []
config.quic_bind= ["127.0.0.1:0"]
config.certfile=str(cert)
config.keyfile=str(key)
config.graceful_timeout=0.1asyncdefshutdown_now() ->None:
returnNoneawaitworker_serve(
ASGIWrapper(app), config, shutdown_trigger=shutdown_now
)
withwarnings.catch_warnings(record=True) ascaught:
warnings.simplefilter("always")
asyncio.run(main())
gc.collect()
leaks= [wforwincaughtifissubclass(w.category, ResourceWarning)
and"unclosed transport"instr(w.message)]
ifleaks:
print(f"LEAK: {len(leaks)} 'unclosed transport' ResourceWarning(s)")
forwinleaks:
print(f" {w.message}")
else:
print("CLEAN: no unclosed-transport ResourceWarning")

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

@CoryLowe5