mirror of
https://github.com/microsoft/debugpy.git
synced 2025-12-23 08:48:12 +00:00
4f6638b0a6
Fix #357: "argsExpansion" does not do what it says in VSCode Treat non-array "args" in debug config as a request to prevent shell argument escaping and allow shell expansion. Remove "argsExpansion".
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License. See LICENSE in the project root
|
|
# for license information.
|
|
|
|
import pytest
|
|
|
|
from debugpy.common import log
|
|
from tests import debug
|
|
from tests.debug import runners, targets
|
|
from tests.patterns import some
|
|
|
|
|
|
@pytest.mark.parametrize("target", targets.all)
|
|
@pytest.mark.parametrize("run", runners.all)
|
|
def test_args(pyfile, target, run):
|
|
@pyfile
|
|
def code_to_debug():
|
|
import sys
|
|
import debuggee
|
|
from debuggee import backchannel
|
|
|
|
debuggee.setup()
|
|
backchannel.send(sys.argv)
|
|
|
|
args = ["--arg1", "arg2", "-arg3", "--", "arg4", "-a"]
|
|
|
|
with debug.Session() as session:
|
|
backchannel = session.open_backchannel()
|
|
with run(session, target(code_to_debug, args=args)):
|
|
pass
|
|
argv = backchannel.receive()
|
|
assert argv == [some.str] + args
|
|
|
|
|
|
@pytest.mark.parametrize("target", targets.all)
|
|
@pytest.mark.parametrize("run", runners.all_launch)
|
|
@pytest.mark.parametrize("expansion", ["preserve", "expand"])
|
|
def test_shell_expansion(pyfile, target, run, expansion):
|
|
if expansion == "expand" and run.console == "internalConsole":
|
|
pytest.skip('Shell expansion is not supported for "internalConsole"')
|
|
|
|
@pyfile
|
|
def code_to_debug():
|
|
import sys
|
|
import debuggee
|
|
from debuggee import backchannel
|
|
|
|
debuggee.setup()
|
|
backchannel.send(sys.argv)
|
|
|
|
def expand(args):
|
|
if expansion != "expand":
|
|
return
|
|
log.info("Before expansion: {0}", args)
|
|
for i, arg in enumerate(args):
|
|
if arg.startswith("$"):
|
|
args[i] = arg[1:]
|
|
log.info("After expansion: {0}", args)
|
|
|
|
class Session(debug.Session):
|
|
def run_in_terminal(self, args, cwd, env):
|
|
expand(args)
|
|
return super().run_in_terminal(args, cwd, env)
|
|
|
|
argslist = ["0", "$1", "2"]
|
|
args = argslist if expansion == "preserve" else " ".join(argslist)
|
|
with Session() as session:
|
|
backchannel = session.open_backchannel()
|
|
with run(session, target(code_to_debug, args=args)):
|
|
pass
|
|
|
|
argv = backchannel.receive()
|
|
|
|
expand(argslist)
|
|
assert argv == [some.str] + argslist
|