fix(server): return stdio parse errors

Author: he-yufeng

src/mcp/server/stdio.py

@@ -17,9 +17,12 @@ async def run_server():
     ```
 """
 
+import json
+import re
 import sys
 from contextlib import asynccontextmanager
 from io import TextIOWrapper
+from typing import Any, cast
 
 import anyio
 import anyio.lowlevel
@@ -28,6 +31,50 @@ async def run_server():
 from mcp.shared._context_streams import create_context_streams
 from mcp.shared.message import SessionMessage
 
+_JSONRPC_ID_PATTERN = re.compile(r'"id"\s*:\s*(-?\d+|"[^"\\]*")')
+
+
+def _request_id_from_raw_message(line: str) -> types.RequestId | None:
+    try:
+        raw_message: Any = json.loads(line)
+    except Exception:
+        raw_message = None
+
+    if not isinstance(raw_message, dict):
+        match = _JSONRPC_ID_PATTERN.search(line)
+        if not match:
+            return None
+
+        raw_request_id = match.group(1)
+        if raw_request_id.startswith('"'):
+            return json.loads(raw_request_id)
+        return int(raw_request_id)
+
+    raw_message_dict = cast(dict[str, Any], raw_message)
+    request_id = raw_message_dict.get("id")
+    if isinstance(request_id, str) or type(request_id) is int:
+        return request_id
+    return None
+
+
+def _error_response_from_parse_failure(line: str, exc: Exception) -> SessionMessage:
+    request_id = _request_id_from_raw_message(line)
+    message = str(exc)
+    if "Invalid JSON" in message:
+        code = types.PARSE_ERROR
+        prefix = "Parse error"
+    else:
+        code = types.INVALID_REQUEST
+        prefix = "Invalid request"
+
+    return SessionMessage(
+        types.JSONRPCError(
+            jsonrpc="2.0",
+            id=request_id,
+            error=types.ErrorData(code=code, message=f"{prefix}: {message}"),
+        )
+    )
+
 
 @asynccontextmanager
 async def stdio_server(stdin: anyio.AsyncFile[str] | None = None, stdout: anyio.AsyncFile[str] | None = None):
@@ -53,7 +100,8 @@ async def stdin_reader():
                     try:
                         message = types.jsonrpc_message_adapter.validate_json(line, by_name=False)
                     except Exception as exc:
-                        await read_stream_writer.send(exc)
+                        error_response = _error_response_from_parse_failure(line, exc)
+                        await write_stream.send(error_response)
                         continue
 
                     session_message = SessionMessage(message)

tests/server/test_stdio.py

@@ -1,4 +1,5 @@
 import io
+import json
 import sys
 from io import TextIOWrapper
 
@@ -7,7 +8,14 @@
 
 from mcp.server.stdio import stdio_server
 from mcp.shared.message import SessionMessage
-from mcp.types import JSONRPCMessage, JSONRPCRequest, JSONRPCResponse, jsonrpc_message_adapter
+from mcp.types import (
+    PARSE_ERROR,
+    JSONRPCError,
+    JSONRPCMessage,
+    JSONRPCRequest,
+    JSONRPCResponse,
+    jsonrpc_message_adapter,
+)
 
 
 @pytest.mark.anyio
@@ -68,8 +76,8 @@ async def test_stdio_server_invalid_utf8(monkeypatch: pytest.MonkeyPatch):
     """Non-UTF-8 bytes on stdin must not crash the server.
 
     Invalid bytes are replaced with U+FFFD, which then fails JSON parsing and
-    is delivered as an in-stream exception. Subsequent valid messages must
-    still be processed.
+    is returned as a JSON-RPC parse error. Subsequent valid messages must still
+    be processed.
     """
     # \xff\xfe are invalid UTF-8 start bytes.
     valid = JSONRPCRequest(jsonrpc="2.0", id=1, method="ping")
@@ -78,17 +86,54 @@ async def test_stdio_server_invalid_utf8(monkeypatch: pytest.MonkeyPatch):
     # Replace sys.stdin with a wrapper whose .buffer is our raw bytes, so that
     # stdio_server()'s default path wraps it with errors='replace'.
     monkeypatch.setattr(sys, "stdin", TextIOWrapper(raw_stdin, encoding="utf-8"))
-    monkeypatch.setattr(sys, "stdout", TextIOWrapper(io.BytesIO(), encoding="utf-8"))
+    stdout = io.StringIO()
 
     with anyio.fail_after(5):
-        async with stdio_server() as (read_stream, write_stream):
-            await write_stream.aclose()
+        async with stdio_server(stdout=anyio.AsyncFile(stdout)) as (read_stream, write_stream):
             async with read_stream:  # pragma: no branch
-                # First line: \xff\xfe -> U+FFFD U+FFFD -> JSON parse fails -> exception in stream
+                # First line: \xff\xfe -> U+FFFD U+FFFD -> JSON parse fails -> error response on stdout
                 first = await read_stream.receive()
-                assert isinstance(first, Exception)
 
                 # Second line: valid message still comes through
-                second = await read_stream.receive()
-                assert isinstance(second, SessionMessage)
-                assert second.message == valid
+                assert isinstance(first, SessionMessage)
+                assert first.message == valid
+
+            await write_stream.aclose()
+
+    stdout.seek(0)
+    output = stdout.read()
+    error = jsonrpc_message_adapter.validate_json(output.strip())
+    assert isinstance(error, JSONRPCError)
+    assert error.id is None
+    assert error.error.code == PARSE_ERROR
+
+
+@pytest.mark.anyio
+async def test_stdio_server_parse_error_completes_id_bearing_request():
+    params: object = {"leaf": True}
+    for index in reversed(range(256)):
+        params = {f"p{index}": params}
+    line = json.dumps({"jsonrpc": "2.0", "id": 900256, "method": "ping", "params": params}) + "\n"
+
+    stdin = io.StringIO(line)
+    stdout = io.StringIO()
+
+    with anyio.fail_after(5):
+        async with stdio_server(stdin=anyio.AsyncFile(stdin), stdout=anyio.AsyncFile(stdout)) as (
+            read_stream,
+            write_stream,
+        ):
+            async with read_stream:
+                with pytest.raises(anyio.EndOfStream):
+                    await read_stream.receive()
+            await write_stream.aclose()
+
+    stdout.seek(0)
+    output_lines = stdout.readlines()
+    assert len(output_lines) == 1
+
+    response = jsonrpc_message_adapter.validate_json(output_lines[0].strip())
+    assert isinstance(response, JSONRPCError)
+    assert response.id == 900256
+    assert response.error.code == PARSE_ERROR
+    assert "Parse error" in response.error.message