diff --git a/SpotifyKeyDumper/Utils.cpp b/SpotifyKeyDumper/Utils.cpp index 3d9ec58..d5af8cf 100644 --- a/SpotifyKeyDumper/Utils.cpp +++ b/SpotifyKeyDumper/Utils.cpp @@ -1,6 +1,7 @@ #include "pch.h" #include "Utils.h" #include "aes.h" +#include const uint8_t IV[] = { 0x72, 0xE0, 0x67, 0xFB, 0xDD, 0xCB, 0xCF, 0x77, 0xEB, 0xE8, 0xBC, 0x64, 0x3F, 0x63, 0x0D, 0x93 }; @@ -135,6 +136,7 @@ struct SongInfo static const std::string albumSearchPattern = "\x68\x65\x69\x67\x68\x74\x22\x20\x3A\x20\x36\x34\x30"; static const std::wstring songDirRoot = L"Downloads"; +static const std::string songRegex = "https?:\\/\\/(?:www\.)?([-a-zA-Z0-9@:%._\\+~#=]{1,256}\\.[a-zA-Z0-9()]{1,6})\\b([-a-zA-Z0-9()@:%_\\+.~#?&\\/\\/=]*)"; static std::wstring songDir = songDirRoot; void Utils::DownloadSong(std::string fileId, std::string trackUri, std::string key, std::string authToken) { @@ -156,15 +158,36 @@ void Utils::DownloadSong(std::string fileId, std::string trackUri, std::string k } // Parse storage resolve response to get the encrypted song data's URL - std::string songHost = (srStr.substr(srStr.find("https://") + 8)) - .erase(srStr.substr(srStr.find("https://") + 8).find("/audio/")); - std::string songPath = srStr.substr(srStr.find("/audio/")).erase(srStr.substr(srStr.find("/audio/")).find("=") + 85); + std::string songHost; + std::string songPath; + try { + std::regex re(songRegex); + std::smatch match; + if (std::regex_search(srStr, match, re) && match.size() > 1) { + songHost = match.str(1); + songPath = match.str(2); + } + else { + std::cout << "Error: Download URL not found" << std::endl; + return; + } + } + catch (std::regex_error& e) { + // Syntax error in the regular expression + std::cout << "Error: regex_error" << std::endl; + return; + } // Download encrypted song data from Spotify std::string songStr = DownloadSpotifyUrl(songHost, songPath, ""); //std::cout << "URL: " << songHost + songPath << std::endl; + if (songStr.empty()) { + std::cout << "Error: Could not download audio" << std::endl; + return; + } + if (songStr.substr(0, 6).compare("") == 0) { std::cout << "Error: " + songStr << std::endl;