From 89cf34dc331d60bff14bce554493846ec4aaf46a Mon Sep 17 00:00:00 2001 From: Aleksandr Dremov Date: Wed, 25 Feb 2026 18:01:36 +0100 Subject: [PATCH 1/2] Add base64 encoding for shell command execution Encode shell command in base64 and update command execution. --- csub.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/csub.py b/csub.py index db3625d..d04e46b 100644 --- a/csub.py +++ b/csub.py @@ -12,6 +12,7 @@ from __future__ import annotations import argparse +import base64 import subprocess import sys from pathlib import Path @@ -191,12 +192,17 @@ def build_runai_command( env.get("EXTRA_SECRET_KEYS", "").split(","), ) + b64_payload = base64.b64encode(shell_command.encode("utf-8")).decode("utf-8") cmd.extend( [ "--", "/bin/zsh", "-c", - shell_command, + ( + f"echo {b64_payload} | base64 -d > /tmp/job_payload.sh && " + f"cat /tmp/job_payload.sh && " + f"/bin/zsh /tmp/job_payload.sh" + ) ] ) return cmd, job_name From bef0fdd5b365234198a3db6c9f88b1df022c921c Mon Sep 17 00:00:00 2001 From: Aleksandr Dremov Date: Wed, 25 Feb 2026 18:04:02 +0100 Subject: [PATCH 2/2] Fix formatting in command execution for job payload --- csub.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/csub.py b/csub.py index d04e46b..d8f8d7a 100644 --- a/csub.py +++ b/csub.py @@ -200,8 +200,8 @@ def build_runai_command( "-c", ( f"echo {b64_payload} | base64 -d > /tmp/job_payload.sh && " - f"cat /tmp/job_payload.sh && " - f"/bin/zsh /tmp/job_payload.sh" + "cat /tmp/job_payload.sh && " + "/bin/zsh /tmp/job_payload.sh" ) ] )