From 307da10caa951c5e8d6d6f256ee3d81a66ed6bb3 Mon Sep 17 00:00:00 2001 From: Paul Cristian Sarbu Date: Thu, 16 Feb 2023 16:20:34 +0100 Subject: tests: Add proper quoting when system calling the shell --- test/utils/shell_quoting.hpp | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 test/utils/shell_quoting.hpp (limited to 'test/utils/shell_quoting.hpp') diff --git a/test/utils/shell_quoting.hpp b/test/utils/shell_quoting.hpp new file mode 100644 index 00000000..16866b8a --- /dev/null +++ b/test/utils/shell_quoting.hpp @@ -0,0 +1,34 @@ +// Copyright 2023 Huawei Cloud Computing Technology Co., Ltd. +// +// 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. + +#include + +[[nodiscard]] static inline auto QuoteForShell(std::string const& input) + -> std::string { + // To correctly quote a string for shell, replace ever ~'~ char with ~'\''~ + // and then put the resulting inside a pair of ~'~s. + auto output = std::string(R"(')"); + auto start = input.begin(); + auto pos = input.find('\''); + while (pos != std::string::npos) { + output.append( + start, start + static_cast(pos + 1)); + output += R"(\'')"; + start += static_cast(pos + 1); + pos = input.find('\'', pos + 1); + } + output.append(start, input.end()); + output += R"(')"; + return output; +} -- cgit v1.2.3