fix(agent): change to a secondary chroot_exec so it happens in a diff…

.gitignore

@@ -1,4 +1,5 @@
 .cursorignore
+.cursor
 .pytest_cache
 .idea
 **/err.txt

agent/skyhook-agent/src/skyhook_agent/chroot_exec.py

@@ -0,0 +1,50 @@
+# 
+# LICENSE START
+#
+#    Copyright (c) NVIDIA CORPORATION.  All rights reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License");
+#    you may not use this file except in compliance with the License.
+#    You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS,
+#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#    See the License for the specific language governing permissions and
+#    limitations under the License.
+#
+# LICENSE END
+# 
+
+import os
+import stat
+import json
+import sys
+import subprocess
+import shutil
+
+
+def chroot_exec(config: dict, chroot_dir: str):
+    cmds = config["cmd"]
+    no_chmod = config["no_chmod"]
+    if chroot_dir != "local":
+        os.chroot(chroot_dir)
+    try:
+        if not no_chmod:
+            # chmod +x the step
+            os.chmod(cmds[0], os.stat(cmds[0]).st_mode | stat.S_IXGRP | stat.S_IXUSR | stat.S_IXOTH)
+        subprocess.run(cmds, check=True)
+    except:
+        raise
+
+
+if __name__ == "__main__":
+    control_file = sys.argv[1]
+    chroot_dir = sys.argv[2]
+
+    with open(control_file, "r") as f:
+        config = json.load(f)
+
+    chroot_exec(config, chroot_dir)