From e85f88715e0ffb3d6f8a1c3ba3f0d6e7b29650af Mon Sep 17 00:00:00 2001
From: Stepan Anokhin <stepan.anokhin@gmail.com>
Date: Sat, 17 Sep 2022 23:38:01 +0700
Subject: [PATCH] Handle multiline command-line args

---
 mprof.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mprof.py b/mprof.py
index 7753d7e..752c652 100644
--- a/mprof.py
+++ b/mprof.py
@@ -1,6 +1,7 @@
 import glob
 import os
 import os.path as osp
+import shlex
 import sys
 import re
 import copy
@@ -175,8 +176,7 @@ def clean_action():
 
 def get_cmd_line(args):
     """Given a set or arguments, compute command-line."""
-    blanks = set(' \t')
-    args = [s if blanks.isdisjoint(s) else "'" + s + "'" for s in args]
+    args = [shlex.quote(s).replace('\n', '\\n') for s in args]
     return ' '.join(args)
 
 def find_first_process(name):