resolve potential attack in linker connection building

microsoft · Dec 13, 2024 · 916c887 · 916c887
1 parent 53e0ddf
commit 916c887
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/network/linkers_socket.cpp b/src/network/linkers_socket.cpp
@@ -157,6 +157,9 @@ void Linkers::ListenThread(int incoming_cnt) {
     }
     int* ptr_in_rank = reinterpret_cast<int*>(buffer);
     int in_rank = *ptr_in_rank;
+    if (in_rank < 0 && in_rank >= num_machines_) {
+      Log::Fatal("Invalid rank %d found during initialization of linkers. The world size is %d", in_rank, num_machines_);
+    }
     // add new socket
     SetLinker(in_rank, handler);
     ++connected_cnt;