Use mutex-protected session_exists? in handle_regular_request
## Motivation and Context `handle_regular_request` was checking `@sessions.key?(session_id)` directly without holding `@mutex`, while concurrent threads could modify `@sessions` via `cleanup_session` or `handle_delete`. This created a TOCTOU race where the check could pass but the session could be deleted before subsequent use. The class already provides a mutex-protected `session_exists?` helper, and `handle_get` already uses it. This change makes `handle_regular_request` consistent with `handle_get`. ## How Has This Been Tested? Added a test that verifies `handle_regular_request` delegates to the mutex-protected `session_exists?` helper instead of accessing `@sessions` directly. All existing tests pass. ## Breaking Change None.
This commit is contained in:
parent
8823bd34ef
commit
be1bbc25c1
1 changed files with 26 additions and 2 deletions
|
|
@ -259,8 +259,7 @@ module MCP
|
|||
|
||||
def handle_regular_request(body_string, session_id)
|
||||
unless @stateless
|
||||
# If session ID is provided, but not in the sessions hash, return an error
|
||||
if session_id && !@sessions.key?(session_id)
|
||||
if session_id && !session_exists?(session_id)
|
||||
return session_not_found_response
|
||||
end
|
||||
end
|
||||
|
|
|
|||
Loading…
Reference in a new issue