tree 23b3bbbfe42b8b6d078aa8c9e172a28a3a340ec1
parent 12a6f5aa6bf3e11668293c36b436eead2f3b8614
author Juncheng Gu <juncgu@gmail.com> 1698963165 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1698963165 +0000

[Fix] add validation logics to TCPStore queries (#107607)

This PR fixes #106294.

Due to the lack of request validation mechanism, TCPStore in torch mistakenly treats nmap scan messages as valid query messages, which leads to DDP OOM. The simple solution enforces the very first query from a client is a validation query with a predefined magic number. If the validation fails, the server will terminate the connection.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/107607
Approved by: https://github.com/cbalioglu, https://github.com/XilunWu