Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

@jerome3o-anthropic@davidfiala@jspahrsummers@Nedomas
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n 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;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

@jerome3o-anthropic@davidfiala@jspahrsummers@Nedomas
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

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

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

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

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

@jerome3o-anthropic@davidfiala@jspahrsummers@Nedomas
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

@jerome3o-anthropic@davidfiala@jspahrsummers@Nedomas
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

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

Added clean shutdown + tests - #115

Merged
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown
Jan 8, 2025
Merged

Added clean shutdown + tests#115
jerome3o-anthropic merged 4 commits into
mainfrom
jerome/fix/clean-stdio-shutdown

Conversation

@jerome3o-anthropic

Copy link
Copy Markdown
Member

Motivation and Context

#113

When closing a StdioServerTransport, the process would hang indefinitely due to unclosed stdin/stdout streams and lingering event listeners. This was particularly noticeable in server implementations that needed to shut down cleanly. The fix ensures proper cleanup of all resources, preventing process hangs.

How Has This Been Tested?

  • Added unit test in src/server/stdio.test.ts verifying proper cleanup of stream resources and event listeners
  • Added integration test in src/integration-tests/process-cleanup.test.ts that confirms the process exits cleanly after transport closure
  • Manually verified fix using a minimal reproduction case that previously demonstrated the hang

Breaking Changes

None. This is a bug fix that maintains the existing API contract.

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update

Checklist

  • I have read the MCP Documentation
  • My code follows the repository's style guidelines
  • New and existing tests pass locally
  • I have added appropriate error handling
  • I have added or updated documentation as needed

Additional context

The fix involves three key changes in StdioServerTransport.close():

  1. Removing all event listeners, including 'drain' listeners that were previously overlooked
  2. Explicitly destroying both stdin and stdout streams
  3. Maintaining existing cleanup of the read buffer and onclose callback

The integration test was added to prevent regression, as unit tests alone didn't catch this issue due to the asynchronous nature of Node.js stream cleanup.

@jspahrsummersjspahrsummers left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for working on this! Some blocking discussion, but this is definitely important

Comment threadsrc/server/stdio.ts Outdated
Comment threadsrc/server/stdio.ts Outdated
@jspahrsummersjspahrsummers linked an issue Jan 3, 2025 that may be closed by this pull request
Comment threadsrc/server/stdio.ts
if (remainingDataListeners === 0) {
// Only pause stdin if we were the only listener
// This prevents interfering with other parts of the application that might be using stdin
this._stdin.pause();

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm a bit surprised that this has any effect on termination, but then reading about Node.js readable streams, I'm like 😵

It feels a bit janky to do things this way, but I don't really have a better idea…

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Per top-level PR comment, I don't see how both MCP and anything else can co-exist on the same stdin stream.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Certainly not while messages are streaming in. The only question is whether there'd ever be a reason for a server implementation to want to read from stdin after terminating the normal MCP stuff. Probably not? But it's always hard to anticipate how people will use a library.

@davidfiala

Copy link
Copy Markdown

I believe that since you are communicating a specific protocol over stdin/stdout, then it does assume you have permanently taken over both STDIO FDs.

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Here was an approximate workaround that I put at the top of servers to route the most common occurences to stderr instead:

// make console.log and console.info go to STDERR because MCP is going to take over STDOUTconstonlyStderr=newConsole({stdout: process.stderr,stderr: process.stderr});console.info=onlyStderr.info;console.log=onlyStderr.log;

In this case, I'd argue that yes it's fine/expected for you to end the stdin/stdout streams. It should be clearly documented that this behavior occurs IMHO. Likewise, perhaps you should hotpatch the console logs in your server.connect() on StdioServerTransport. The thing is, if the user happens to console.log BEFORE getting to your constructor, then there'd already be corruption-worthy output that the MCP client would be exposed to when it begins to read stdout. There's worse workarounds for that as well (ie, a shared sentinel to broadcast the true beginning of the MCP stream), but that is a breaking protocol change.

@jspahrsummers

Copy link
Copy Markdown
Member

This has interesting implications that I've had to work around. Specifically, wouldn't a console.log or console.info effectively corrupt your communication stream, if it occurs during any part of the server execution?

Yes, this is discussed here: https://modelcontextprotocol.io/docs/tools/debugging#server-side-logging

@Nedomas

Copy link
Copy Markdown

Even though the fix here is for stdio specifically, I’m wondering isn’t the same leak happening in the SSE transport too?

I can’t seem to be getting rid of the active connection even with doing this:

constclient=newClient({name: 'superinterface-mcp-client',version: '1.0.0',},{capabilities: {}})consttransport=ewSSEClientTransport(newURL('some-url'))awaitclient.connect(transport)awaitclient.close()awaittransport.close()

But might be an unrelated bug?

@jerome3o-anthropic

Copy link
Copy Markdown
MemberAuthor

seem to be getting rid of the active connection even with doing

That's odd - likely a separate issue. Going to merge this now and follow up on the SSE bug later

@jerome3o-anthropic
jerome3o-anthropic merged commit e72ff8d into mainJan 8, 2025
@jerome3o-anthropic
jerome3o-anthropic deleted the jerome/fix/clean-stdio-shutdown branch January 8, 2025 18:07
Pizzaface pushed a commit to RewstApp/mcp-inspector that referenced this pull request May 2, 2025
…issue-114
fix(server) Avoid modifying path resolution for executables on MacOS
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.

Unable to cleanly shutdown/close MCP Server

4 participants

@jerome3o-anthropic@davidfiala@jspahrsummers@Nedomas