tg_cpphost: flash cmd: Don't allow newlines in zipname

Royna2544 · Nov 21, 2023 · ebf11ac · ebf11ac
1 parent 1f3c4a6
commit ebf11ac
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -173,10 +173,14 @@ int main(void) {
             }
             parseExtArgs(message, msg);
         }
-
+        for (const auto c : msg) {
+            if (c == '\n') {
+                bot_sendReplyMessage(bot, message, "Zip names shouldn't have newlines");
+                return;
+            }
+        }
         if (!std::regex_match(StringTools::split(msg, '\n').front(), kFlashTextRegEX)) {
             std::replace(msg.begin(), msg.end(), ' ', '_');
-            std::replace(msg.begin(), msg.end(), '\n', ' ');
             if (!StringTools::endsWith(msg, kZipExtentionSuffix)) {
                 msg += kZipExtentionSuffix;
             }