| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "tools/gn/escape.h" | |
| 6 | |
| 7 #include "base/containers/stack_container.h" | |
| 8 | |
| 9 namespace { | |
| 10 | |
| 11 template<typename DestString> | |
| 12 void EscapeStringToString(const base::StringPiece& str, | |
| 13 const EscapeOptions& options, | |
| 14 DestString* dest) { | |
| 15 bool used_quotes = false; | |
| 16 | |
| 17 for (size_t i = 0; i < str.size(); i++) { | |
| 18 if (str[i] == '$' && options.mode == ESCAPE_NINJA) { | |
| 19 // Escape dollars signs since ninja treats these specially. | |
| 20 dest->push_back('$'); | |
| 21 dest->push_back('$'); | |
| 22 } else if (str[i] == '"' && options.mode == ESCAPE_SHELL) { | |
| 23 // Escape quotes with backslashes for the command-line (Ninja doesn't | |
| 24 // care). | |
| 25 dest->push_back('\\'); | |
| 26 dest->push_back('"'); | |
| 27 } else if (str[i] == ' ') { | |
| 28 if (options.mode == ESCAPE_NINJA) { | |
| 29 // For ninja just escape spaces with $. | |
| 30 dest->push_back('$'); | |
| 31 } else if (options.mode == ESCAPE_SHELL && !options.inhibit_quoting) { | |
| 32 // For the shell, quote the whole string. | |
| 33 if (!used_quotes) { | |
| 34 used_quotes = true; | |
| 35 dest->insert(dest->begin(), '"'); | |
| 36 } | |
| 37 } | |
| 38 dest->push_back(' '); | |
| 39 #if defined(OS_WIN) | |
| 40 } else if (str[i] == '/' && options.convert_slashes) { | |
| 41 // Convert slashes on Windows if requested. | |
| 42 dest->push_back('\\'); | |
| 43 #else | |
| 44 } else if (str[i] == '\\' && options.mode == ESCAPE_SHELL) { | |
| 45 // For non-Windows shell, escape backslashes. | |
| 46 dest->push_back('\\'); | |
| 47 dest->push_back('\\'); | |
| 48 #endif | |
| 49 } else { | |
| 50 dest->push_back(str[i]); | |
| 51 } | |
| 52 } | |
| 53 | |
| 54 if (used_quotes) | |
| 55 dest->push_back('"'); | |
| 56 } | |
| 57 | |
| 58 } // namespace | |
| 59 | |
| 60 std::string EscapeString(const base::StringPiece& str, | |
| 61 const EscapeOptions& options) { | |
| 62 std::string result; | |
| 63 result.reserve(str.size() + 4); // Guess we'll add a couple of extra chars. | |
| 64 EscapeStringToString(str, options, &result); | |
| 65 return result; | |
| 66 } | |
| 67 | |
| 68 void EscapeStringToStream(std::ostream& out, | |
| 69 const base::StringPiece& str, | |
| 70 const EscapeOptions& options) { | |
| 71 // Escape to a stack buffer and then write out to the stream. | |
| 72 base::StackVector<char, 256> result; | |
| 73 result->reserve(str.size() + 4); // Guess we'll add a couple of extra chars. | |
| 74 EscapeStringToString(str, options, &result.container()); | |
| 75 if (!result->empty()) | |
| 76 out.write(result->data(), result->size()); | |
| 77 } | |
| OLD | NEW |