From 1d238b70e2655e64b1cb1a51357a9eb0c753181d Mon Sep 17 00:00:00 2001
From: llnulldisk <48621230+llnulldisk@users.noreply.github.com>
Date: Wed, 5 Oct 2022 13:57:24 +0200
Subject: Escape poll options (#128)

---
 src/Api.cpp               |  4 ++--
 src/tools/StringTools.cpp | 18 ++++++++++++++++++
 2 files changed, 20 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/Api.cpp b/src/Api.cpp
index ed1f07a..7ceb75e 100644
--- a/src/Api.cpp
+++ b/src/Api.cpp
@@ -903,7 +903,7 @@ Message::Ptr Api::sendPoll(boost::variant<std::int64_t, const std::string&> chat
     args.emplace_back("question", question);
     args.emplace_back("options", _tgTypeParser.parseArray<std::string>(
         [](const std::string& option)->std::string {
-        return "\"" + option + "\"";
+        return "\"" + StringTools::escapeJsonString(option) + "\"";
     }, options));
     if (!isAnonymous) {
         args.emplace_back("is_anonymous", isAnonymous);
@@ -1760,7 +1760,7 @@ std::vector<Sticker::Ptr> Api::getCustomEmojiStickers(const std::vector<std::str
     args.reserve(1);
 
     args.emplace_back("custom_emoji_ids", _tgTypeParser.parseArray<std::string>([] (const std::string& customEmojiId) -> std::string {
-        return "\"" + customEmojiId + "\"";
+        return "\"" + StringTools::escapeJsonString(customEmojiId) + "\"";
     }, customEmojiIds));
 
     return _tgTypeParser.parseJsonAndGetArray<Sticker>(&TgTypeParser::parseJsonAndGetSticker, sendRequest("getCustomEmojiStickers", args));
diff --git a/src/tools/StringTools.cpp b/src/tools/StringTools.cpp
index 653047c..3e5e6c4 100644
--- a/src/tools/StringTools.cpp
+++ b/src/tools/StringTools.cpp
@@ -101,4 +101,22 @@ string urlDecode(const string& value) {
     return result;
 }
 
+std::string escapeJsonString(const std::string& value) {
+    string result;
+
+    for (const char& c : value) {
+        switch (c) {
+        case '"':
+        case '\\':
+        case '/':
+            result += '\\';
+            break;
+        }
+
+        result += c;
+    }
+
+    return result;
+}
+
 }
-- 
cgit v1.2.3