wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-07 07:59:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Hash contents of SSH key instead of key path (#10103)

Browse Source
This commit is contained in:
Philipp Moritz
2020-08-14 00:10:31 -07:00
committed by GitHub
parent 3a6fa7d622
commit 6b53df9599
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/autoscaler/util.py
+10 -1
View File
@@ -103,8 +103,17 @@ def with_head_node_ip(cmds):
def hash_launch_conf(node_conf, auth):
hasher = hashlib.sha1()
# For hashing, we replace the path to the key with the
# key itself. This is to make sure the hashes are the
# same even if keys live at different locations on different
# machines.
full_auth = auth.copy()
for key_type in ["ssh_private_key", "ssh_public_key"]:
if key_type in auth:
with open(auth[key_type]) as key:
full_auth[key_type] = key.read()
hasher.update(
json.dumps([node_conf, auth], sort_keys=True).encode("utf-8"))
json.dumps([node_conf, full_auth], sort_keys=True).encode("utf-8"))
return hasher.hexdigest()