diff options
author | dec05eba <dec05eba@protonmail.com> | 2021-06-11 04:51:03 +0200 |
---|---|---|
committer | dec05eba <dec05eba@protonmail.com> | 2021-06-11 04:51:03 +0200 |
commit | 89383cff1ba5d8a928262fcb4c40382a981c78c8 (patch) | |
tree | da8b6062cc6770fd37e7a6f7b8c09fb61f46f7b2 /src/plugins | |
parent | 5b3becc79461d4ecf015e33515871cc09e26e04e (diff) |
Remove dependency on youtube-dl for streaming youtube, resulting in faster video startup
Diffstat (limited to 'src/plugins')
-rw-r--r-- | src/plugins/ImageBoard.cpp | 2 | ||||
-rw-r--r-- | src/plugins/MediaGeneric.cpp | 2 | ||||
-rw-r--r-- | src/plugins/Youtube.cpp | 269 | ||||
-rw-r--r-- | src/plugins/youtube/Signature.cpp | 307 |
4 files changed, 575 insertions, 5 deletions
diff --git a/src/plugins/ImageBoard.cpp b/src/plugins/ImageBoard.cpp index 881d8ff..a2ffca7 100644 --- a/src/plugins/ImageBoard.cpp +++ b/src/plugins/ImageBoard.cpp @@ -1,7 +1,7 @@ #include "../../plugins/ImageBoard.hpp" namespace QuickMedia { - std::unique_ptr<RelatedVideosPage> ImageBoardThreadPage::create_related_videos_page(Program*, const std::string&, const std::string&) { + std::unique_ptr<RelatedVideosPage> ImageBoardThreadPage::create_related_videos_page(Program*) { return nullptr; } diff --git a/src/plugins/MediaGeneric.cpp b/src/plugins/MediaGeneric.cpp index 19ad87c..c3a8d8e 100644 --- a/src/plugins/MediaGeneric.cpp +++ b/src/plugins/MediaGeneric.cpp @@ -191,7 +191,7 @@ namespace QuickMedia { return std::make_unique<MediaGenericSearchPage>(*search_page); } - std::unique_ptr<RelatedVideosPage> MediaGenericVideoPage::create_related_videos_page(Program *program, const std::string&, const std::string&) { + std::unique_ptr<RelatedVideosPage> MediaGenericVideoPage::create_related_videos_page(Program *program) { return std::make_unique<MediaGenericRelatedPage>(program, search_page); } }
\ No newline at end of file diff --git a/src/plugins/Youtube.cpp b/src/plugins/Youtube.cpp index d9df409..9cec69c 100644 --- a/src/plugins/Youtube.cpp +++ b/src/plugins/Youtube.cpp @@ -1,10 +1,12 @@ #include "../../plugins/Youtube.hpp" +#include "../../plugins/youtube/Signature.hpp" #include "../../include/Storage.hpp" #include "../../include/NetUtils.hpp" #include "../../include/StringUtils.hpp" #include "../../include/Scale.hpp" #include "../../include/Notification.hpp" #include "../../include/Utils.hpp" +#include <json/reader.h> extern "C" { #include <HtmlParser.h> } @@ -13,6 +15,30 @@ extern "C" { #include <unistd.h> namespace QuickMedia { + bool youtube_url_extract_id(const std::string &youtube_url, std::string &youtube_video_id) { + size_t index = youtube_url.find("youtube.com/watch?v="); + if(index != std::string::npos) { + index += 20; + size_t end_index = youtube_url.find("&", index); + if(end_index == std::string::npos) + end_index = youtube_url.size(); + youtube_video_id = youtube_url.substr(index, end_index - index); + return true; + } + + index = youtube_url.find("youtu.be/"); + if(index != std::string::npos) { + index += 9; + size_t end_index = youtube_url.find("?", index); + if(end_index == std::string::npos) + end_index = youtube_url.size(); + youtube_video_id = youtube_url.substr(index, end_index - index); + return true; + } + + return false; + } + // This is a common setup of text in the youtube json static std::optional<std::string> yt_json_get_text(const Json::Value &json, const char *root_name) { if(!json.isObject()) @@ -257,7 +283,10 @@ namespace QuickMedia { body_item->set_description_color(sf::Color(179, 179, 179)); body_item->url = "https://www.youtube.com/channel/" + channel_id_json.asString(); if(thumbnail) { - body_item->thumbnail_url = std::string("https:") + thumbnail->url; + if(string_starts_with(thumbnail->url, "https:")) + body_item->thumbnail_url = thumbnail->url; + else + body_item->thumbnail_url = std::string("https:") + thumbnail->url; body_item->thumbnail_mask_type = ThumbnailMaskType::CIRCLE; body_item->thumbnail_size.x = thumbnail->width; body_item->thumbnail_size.y = thumbnail->height; @@ -386,13 +415,16 @@ namespace QuickMedia { static std::vector<CommandArg> get_cookies() { std::lock_guard<std::mutex> lock(cookies_mutex); if(cookies_filepath.empty()) { + YoutubeSignatureDecryptor::get_instance(); + Path cookies_filepath_p; if(get_cookies_filepath(cookies_filepath_p, "youtube") != 0) { show_notification("QuickMedia", "Failed to create youtube cookies file", Urgency::CRITICAL); return {}; } - // TODO: Re-enable this if the api key ever changes in the future + // TODO: Re-enable this if the api key ever changes in the future. + // Maybe also put signature decryption in the same request? since it requests the same page. #if 0 //api_key = youtube_page_find_api_key(); #else @@ -655,6 +687,11 @@ namespace QuickMedia { return PluginResult::OK; } + PluginResult YoutubeSearchPage::lazy_fetch(BodyItems&) { + get_cookies(); + return PluginResult::OK; + } + PluginResult YoutubeSearchPage::search_get_continuation(const std::string &url, const std::string ¤t_continuation_token, BodyItems &result_items) { std::string next_url = url + "&pbj=1&ctoken=" + current_continuation_token; @@ -1800,11 +1837,237 @@ namespace QuickMedia { return std::make_unique<YoutubeCommentsPage>(program, xsrf_token, comments_continuation_token); } - std::unique_ptr<RelatedVideosPage> YoutubeVideoPage::create_related_videos_page(Program *program, const std::string&, const std::string&) { + std::unique_ptr<RelatedVideosPage> YoutubeVideoPage::create_related_videos_page(Program *program) { return std::make_unique<YoutubeRelatedVideosPage>(program); } std::unique_ptr<Page> YoutubeVideoPage::create_channels_page(Program *program, const std::string &channel_url) { return std::make_unique<YoutubeChannelPage>(program, channel_url, "", "Channel videos"); } + + static std::map<std::string, std::string> http_params_parse(const std::string &http_params) { + std::map<std::string, std::string> result; + string_split(http_params, '&', [&result](const char *str, size_t size) { + const void *split_p = memchr(str, '=', size); + if(split_p == nullptr) + return true; + + std::string key(str, (const char*)split_p - str); + std::string value((const char*)split_p + 1, (str + size) - ((const char*)split_p + 1)); + key = url_param_decode(key); + value = url_param_decode(value); + result[std::move(key)] = std::move(value); + return true; + }); + return result; + } + + static std::string url_extract_param(const std::string &url, const std::string ¶m) { + std::string param_s = param + "="; + size_t index = url.find(param_s); + if(index == std::string::npos) + return ""; + + index += param_s.size(); + size_t end = url.find('&', index); + if(end == std::string::npos) + end = url.size(); + + return url.substr(index, end - index); + } + + std::string YoutubeVideoPage::get_video_url(int max_height, bool &has_embedded_audio) { + if(video_formats.empty()) { + has_embedded_audio = true; + return ""; + } + + for(const auto &video_format : video_formats) { + if(video_format.height <= max_height) { + has_embedded_audio = video_format.has_embedded_audio; + return video_format.base.url; + } + } + + has_embedded_audio = video_formats.back().has_embedded_audio; + return video_formats.back().base.url; + } + + std::string YoutubeVideoPage::get_audio_url() { + if(audio_formats.empty()) + return ""; + + return audio_formats.front().base.url; + } + + PluginResult YoutubeVideoPage::load() { + video_formats.clear(); + audio_formats.clear(); + + std::string video_id; + if(!youtube_url_extract_id(url, video_id)) { + fprintf(stderr, "Failed to extract youtube id from %s\n", url.c_str()); + return PluginResult::ERR; + } + + std::vector<CommandArg> additional_args = { + { "-H", "x-youtube-client-name: 1" }, + { "-H", "x-youtube-client-version: 2.20200626.03.00" } + }; + + std::vector<CommandArg> cookies = get_cookies(); + additional_args.insert(additional_args.end(), cookies.begin(), cookies.end()); + + std::string response; + DownloadResult download_result = download_to_string("https://www.youtube.com/get_video_info?html5=1&video_id=" + video_id + "&eurl=https://www.youtube.googleapis.com/v/" + video_id, response, std::move(additional_args), true); // TODO: true? + if(download_result != DownloadResult::OK) return download_result_to_plugin_result(download_result); + + std::string player_response_param = url_extract_param(response, "player_response"); + player_response_param = url_param_decode(player_response_param); + + Json::Value json_root; + Json::CharReaderBuilder json_builder; + std::unique_ptr<Json::CharReader> json_reader(json_builder.newCharReader()); + std::string json_errors; + if(!json_reader->parse(player_response_param.data(), player_response_param.data() + player_response_param.size(), &json_root, &json_errors)) { + fprintf(stderr, "Failed to read param as json, error: %s\n", json_errors.c_str()); + return PluginResult::ERR; + } + + if(!json_root.isObject()) + return PluginResult::ERR; + + const Json::Value &streaming_data_json = json_root["streamingData"]; + if(!streaming_data_json.isObject()) + return PluginResult::ERR; + + parse_formats(streaming_data_json); + + if(video_formats.empty() && audio_formats.empty()) + return PluginResult::ERR; + + std::sort(video_formats.begin(), video_formats.end(), [](const YoutubeVideoFormat &format1, const YoutubeVideoFormat &format2) { + return format1.base.bitrate > format2.base.bitrate; + }); + + std::sort(audio_formats.begin(), audio_formats.end(), [](const YoutubeAudioFormat &format1, const YoutubeAudioFormat &format2) { + return format1.base.bitrate > format2.base.bitrate; + }); + + return PluginResult::OK; + } + + static bool parse_cipher_format(const Json::Value &format, YoutubeFormat &youtube_format) { + std::map<std::string, std::string> cipher_params; + const Json::Value &cipher_json = format["cipher"]; + if(cipher_json.isString()) { + cipher_params = http_params_parse(cipher_json.asString()); + } else { + const Json::Value &signature_cipher_json = format["signatureCipher"]; + if(signature_cipher_json.isString()) + cipher_params = http_params_parse(signature_cipher_json.asString()); + } + + const std::string &url = cipher_params["url"]; + if(cipher_params.empty() || url.empty()) + return false; + + std::string url_decoded = url_param_decode(url); + + const std::string &s = cipher_params["s"]; + const std::string &sp = cipher_params["sp"]; + std::string sig_key; + std::string sig_value; + if(YoutubeSignatureDecryptor::get_instance().decrypt(s, sp, sig_key, sig_value)) + url_decoded += "&" + std::move(sig_key) + "=" + std::move(sig_value); + + youtube_format.url = std::move(url_decoded); + return true; + } + + void YoutubeVideoPage::parse_format(const Json::Value &format_json, bool is_adaptive) { + if(!format_json.isArray()) + return; + + for(const Json::Value &format : format_json) { + if(!format.isObject()) + continue; + + if(is_adaptive) { + // TODO: Fix. Some streams use sq/ instead of index + const Json::Value &index_range_json = format["indexRange"]; + if(index_range_json.isNull()) { + fprintf(stderr, "Ignoring adaptive stream without indexRange\n"); + continue; + } + } + + YoutubeFormat youtube_format_base; + + const Json::Value &mime_type_json = format["mimeType"]; + if(!mime_type_json.isString()) continue; + + const Json::Value &bitrate_json = format["bitrate"]; + if(!bitrate_json.isInt()) continue; + youtube_format_base.bitrate = bitrate_json.asInt(); + + if(strncmp(mime_type_json.asCString(), "video/", 6) == 0) { + bool has_embedded_audio = false; + const char *codecs_p = strstr(mime_type_json.asCString(), "codecs=\""); + if(codecs_p) { + codecs_p += 8; + const char *codecs_sep_p = strchr(codecs_p, ','); + const char *codecs_end_p = strchr(codecs_p, '"'); + has_embedded_audio = (codecs_sep_p != nullptr && (!codecs_end_p || codecs_sep_p < codecs_end_p)); + } + + YoutubeVideoFormat video_format; + video_format.base = std::move(youtube_format_base); + video_format.has_embedded_audio = has_embedded_audio; + + const Json::Value &width_json = format["width"]; + if(!width_json.isInt()) continue; + video_format.width = width_json.asInt(); + + const Json::Value &height_json = format["height"]; + if(!height_json.isInt()) continue; + video_format.height = height_json.asInt(); + + const Json::Value &fps_json = format["fps"]; + if(!fps_json.isInt()) continue; + video_format.fps = fps_json.asInt(); + + const Json::Value &url_json = format["url"]; + if(url_json.isString()) { + video_format.base.url = url_json.asString(); + } else { + if(!parse_cipher_format(format, video_format.base)) + continue; + } + + video_formats.push_back(std::move(video_format)); + } else if(strncmp(mime_type_json.asCString(), "audio/", 6) == 0) { + YoutubeAudioFormat audio_format; + audio_format.base = std::move(youtube_format_base); + + const Json::Value &url_json = format["url"]; + if(url_json.isString()) { + audio_format.base.url = url_json.asString(); + } else { + if(!parse_cipher_format(format, audio_format.base)) + continue; + } + + audio_formats.push_back(std::move(audio_format)); + } + } + } + + void YoutubeVideoPage::parse_formats(const Json::Value &streaming_data_json) { + const Json::Value &formats_json = streaming_data_json["formats"]; + parse_format(formats_json, false); + + const Json::Value &adaptive_formats_json = streaming_data_json["adaptiveFormats"]; + parse_format(adaptive_formats_json, false); + } }
\ No newline at end of file diff --git a/src/plugins/youtube/Signature.cpp b/src/plugins/youtube/Signature.cpp new file mode 100644 index 0000000..30093d5 --- /dev/null +++ b/src/plugins/youtube/Signature.cpp @@ -0,0 +1,307 @@ +#include "../../../plugins/youtube/Signature.hpp" +#include "../../../include/Storage.hpp" +#include "../../../include/Notification.hpp" +#include "../../../include/DownloadUtils.hpp" +#include "../../../include/StringUtils.hpp" +#include "../../../include/Program.hpp" +#include <regex> +#include <unistd.h> + +namespace QuickMedia { + enum UpdateDecryptFunctionError { + U_DEC_FUN_NET_ERR = 1, + U_DEC_FUN_FAILED_MATCH_ERR = 2 + }; + + static YoutubeSignatureDecryptor *instance = nullptr; + static const int timeout_default_sec = 60; + + bool YoutubeSignatureDecryptor::js_code_to_operations(const std::string &function_body_str, const std::string &var_body_str, std::vector<DecryptFuncCall> &new_func_calls, std::map<std::string, DecryptFunction> &new_func_decls) { + std::vector<std::string> function_body; + string_split(function_body_str, ';', [&function_body](const char *str, size_t size) { + function_body.push_back(std::string(str, size)); + return true; + }); + + if(function_body.empty()) + return false; + + std::vector<std::string> var_body; + string_split(var_body_str, "},", [&var_body](const char *str, size_t size) { + var_body.push_back(std::string(str, size)); + return true; + }); + + if(var_body.empty()) + return false; + + //fprintf(stderr, "function body: %s\n", function_body_str.c_str()); + for(const std::string &func_call_str : function_body) { + const size_t var_name_split_index = func_call_str.find('.'); + if(var_name_split_index == std::string::npos) return false; + + const size_t func_name_end_index = func_call_str.find('(', var_name_split_index + 1); + if(func_name_end_index == std::string::npos) return false; + + const size_t values_index = func_call_str.find(',', func_name_end_index + 1); + if(values_index == std::string::npos) return false; + + const size_t values_end_index = func_call_str.find(')', values_index + 1); + if(values_end_index == std::string::npos) return false; + + std::string func_name = func_call_str.substr(var_name_split_index + 1, func_name_end_index - (var_name_split_index + 1)); + func_name = strip(func_name); + std::string value_args = func_call_str.substr(values_index + 1, values_end_index - (values_index + 1)); + + errno = 0; + char *endptr; + const long value_int = strtol(value_args.c_str(), &endptr, 10); + if(endptr != value_args.c_str() && errno == 0) + new_func_calls.push_back({ std::move(func_name), value_int }); + else + return false; + + //fprintf(stderr, "func_call: %s, value: %ld\n", new_func_calls.back().func_name.c_str(), value_int); + } + + //fprintf(stderr, "declaration body: %s\n", var_body_str.c_str()); + for(const std::string &func_decl_str : var_body) { + const size_t func_name_split_index = func_decl_str.find(':'); + if(func_name_split_index == std::string::npos) return false; + + const size_t func_start_index = func_decl_str.find('{', func_name_split_index + 1); + if(func_start_index == std::string::npos) return false; + + std::string func_name = func_decl_str.substr(0, func_name_split_index); + func_name = strip(func_name); + std::string function_body = func_decl_str.substr(func_start_index + 1); + function_body = strip(function_body); + if(!function_body.empty() && function_body.back() == '}') + function_body.pop_back(); + + DecryptFunction decrypt_function; + if(function_body == "a.reverse()") + decrypt_function = DecryptFunction::REVERSE; + else if(function_body == "a.splice(0,b)") + decrypt_function = DecryptFunction::SPLICE; + else + decrypt_function = DecryptFunction::SWAP; + //fprintf(stderr, "declared function: %s, body: |%s|\n", func_name.c_str(), function_body.c_str()); + new_func_decls[std::move(func_name)] = decrypt_function; + } + + for(const auto &func_call : new_func_calls) { + if(new_func_decls.find(func_call.func_name) == new_func_decls.end()) { + fprintf(stderr, "YoutubeSignatureDecryptor: declaration for decryption function %s not found\n", func_call.func_name.c_str()); + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 10 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return false; + } + } + + return true; + } + + YoutubeSignatureDecryptor::YoutubeSignatureDecryptor() { + { + Path youtube_cache_dir = get_cache_dir().join("youtube"); + if(create_directory_recursive(youtube_cache_dir) != 0) { + show_notification("QuickMedia", "Failed to create youtube cache directory", Urgency::CRITICAL); + return; + } + + youtube_cache_dir.join("decryption_function"); + if(file_get_content(youtube_cache_dir, decryption_function) == 0) { + file_get_last_modified_time_seconds(youtube_cache_dir.data.c_str(), &decrypt_function_last_updated); + + size_t newline_index = decryption_function.find('\n'); + if(newline_index != std::string::npos) { + std::string function_body_str = decryption_function.substr(0, newline_index); + std::string var_body_str = decryption_function.substr(newline_index + 1); + + std::vector<DecryptFuncCall> new_func_calls; + std::map<std::string, DecryptFunction> new_func_decls; + if(js_code_to_operations(function_body_str, var_body_str, new_func_calls, new_func_decls)) { + func_calls = std::move(new_func_calls); + func_decls = std::move(new_func_decls); + + time_t time_now = time(nullptr); + if(time_now - decrypt_function_last_updated <= timeout_default_sec) + up_to_date = true; + } + } + } + } + + running = true; + poll_task = AsyncTask<void>([this]() mutable { + bool has_notified_error = false; + int decrypt_function_update_timeout_seconds = timeout_default_sec; + while(running) { + time_t time_now = time(nullptr); + if(time_now - decrypt_function_last_updated > decrypt_function_update_timeout_seconds) { + int update_res = update_decrypt_function(); + if(update_res != 0) { + if(!has_notified_error) { + if(program_is_dead_in_current_thread()) { + running = false; + return; + } else if(update_res == U_DEC_FUN_NET_ERR) { + show_notification("QuickMedia", "Failed to decrypt youtube signature. Is your internet down?", Urgency::CRITICAL); + decrypt_function_update_timeout_seconds = 10; + } else if(update_res == U_DEC_FUN_FAILED_MATCH_ERR) { + show_notification("QuickMedia", "Failed to decrypt youtube signature. Make sure you are running the latest version of QuickMedia", Urgency::CRITICAL); + running = false; + return; + } + } + has_notified_error = true; + } else { + decrypt_function_update_timeout_seconds = timeout_default_sec; + } + } + usleep(1 * 1000 * 1000); // 1 second + } + }); + } + + YoutubeSignatureDecryptor::~YoutubeSignatureDecryptor() { + running = false; + } + + // static + YoutubeSignatureDecryptor& YoutubeSignatureDecryptor::get_instance() { + if(!instance) + instance = new YoutubeSignatureDecryptor(); + return *instance; + } + + bool YoutubeSignatureDecryptor::decrypt(const std::string &s, const std::string &sp, std::string &sig_key, std::string &sig_value) { + if(s.empty() || sp.empty()) + return false; + + if(!up_to_date) { + int num_tries = 0; + const int max_tries = 30; + while(running && num_tries < max_tries && !program_is_dead_in_current_thread()) { // 6 seconds in total + std::lock_guard<std::mutex> lock(update_signature_mutex); + if(up_to_date) + break; + ++num_tries; + usleep(200 * 1000); // 200 milliseconds + } + + if(num_tries == max_tries) { + show_notification("QuickMedia", "Failed to get decryption function for youtube. Make sure your internet is up and that you are running the latest version of QuickMedia", Urgency::CRITICAL); + return false; + } + } + + std::lock_guard<std::mutex> lock(update_signature_mutex); + std::string sig = s; + for(const auto &func_call : func_calls) { + auto func_decl_it = func_decls.find(func_call.func_name); + assert(func_decl_it != func_decls.end()); + + switch(func_decl_it->second) { + case DecryptFunction::REVERSE: { + std::reverse(sig.begin(), sig.end()); + break; + } + case DecryptFunction::SPLICE: { + long int erase_index = func_call.arg; + if(erase_index > 0 && (size_t)erase_index < sig.size()) + sig.erase(0, erase_index); + break; + } + case DecryptFunction::SWAP: { + if(sig.empty()) { + fprintf(stderr, "YoutubeSignatureDecryptor: sig unexpectedly empty in swap\n"); + } else { + char c = sig[0]; + sig[0] = sig[func_call.arg % sig.size()]; + sig[func_call.arg % sig.size()] = c; + } + break; + } + } + } + + sig_key = sp; + sig_value = sig; + return true; + } + + int YoutubeSignatureDecryptor::update_decrypt_function() { + std::string response; + DownloadResult download_result = download_to_string("https://www.youtube.com/watch?v=CvFH_6DNRCY&gl=US&hl=en", response, {}, true); + if(download_result != DownloadResult::OK) { + fprintf(stderr, "YoutubeSignatureDecryptor::update_decrypt_function failed. Failed to get youtube page\n"); + return U_DEC_FUN_NET_ERR; + } + + std::smatch base_js_match; + if(!std::regex_search(response, base_js_match, std::regex(R"END((\/s\/player\/[^\/]+\/player_ias[^\/]+\/en_US\/base\.js))END", std::regex::ECMAScript)) || base_js_match.size() != 2) { + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 1 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return U_DEC_FUN_FAILED_MATCH_ERR; + } + + const std::string &url = base_js_match[1].str(); + download_result = download_to_string("https://www.youtube.com" + url, response, {}, true); + if(download_result != DownloadResult::OK) { + fprintf(stderr, "YoutubeSignatureDecryptor::update_decrypt_function failed. Failed to get https://www.youtube.com%s\n", url.c_str()); + return U_DEC_FUN_NET_ERR; + } + + std::smatch function_match; + if(!std::regex_search(response, function_match, std::regex(R"END((^|\n)\w+=function\(\w\)\{\w=\w\.split\(""\);([^\}]*)\})END", std::regex::ECMAScript)) || function_match.size() != 3) { + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 2 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return U_DEC_FUN_FAILED_MATCH_ERR; + } + + std::string function_body_str = function_match[2].str(); + size_t last_semicolon_index = function_body_str.rfind(';'); + if(last_semicolon_index == std::string::npos) { + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 3 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return U_DEC_FUN_FAILED_MATCH_ERR; + } + + function_body_str.erase(last_semicolon_index, function_body_str.size()); + string_replace_all(function_body_str, '\n', ' '); + + size_t var_dot_index = function_body_str.find('.'); + if(var_dot_index == std::string::npos) { + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 5 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return U_DEC_FUN_FAILED_MATCH_ERR; + } + + std::string var_name = function_body_str.substr(0, var_dot_index); + string_replace_all(response, '\n', ' '); + std::smatch var_body_match; + if(!std::regex_search(response, var_body_match, std::regex("var " + var_name + "=\\{(.*?)\\};", std::regex::ECMAScript)) || var_body_match.size() != 2) { + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 6 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return U_DEC_FUN_FAILED_MATCH_ERR; + } + + std::string var_body_str = var_body_match[1].str(); + string_replace_all(var_body_str, '\n', ' '); + + std::vector<DecryptFuncCall> new_func_calls; + std::map<std::string, DecryptFunction> new_func_decls; + if(!js_code_to_operations(function_body_str, var_body_str, new_func_calls, new_func_decls)) { + fprintf(stderr, "YoutubeSignatureDecryptor: Regex match 7 invalid. Youtube likely updated and QuickMedia needs to be fixed?\n"); + return U_DEC_FUN_FAILED_MATCH_ERR; + } + + { + std::lock_guard<std::mutex> lock(update_signature_mutex); + decryption_function = function_body_str + "\n" + var_body_str; + decrypt_function_last_updated = time(nullptr); + up_to_date = true; + func_calls = std::move(new_func_calls); + func_decls = std::move(new_func_decls); + } + + file_overwrite_atomic(get_cache_dir().join("youtube").join("decryption_function"), decryption_function); + return 0; + } +}
\ No newline at end of file |