Files
obs-studio/UI/window-basic-main-browser.cpp
T
Ryan Foster 50dc5e5345 frontend: Ensure cookie ID is always 16 hexadecimal characters
At a glance, the GenId function looks like it can only return a
16-character hexadecimal string with all characters being [0-9A-F].
However, it seems that it can rarely return a 16-character string that
has one or two space characters at the beginning due to the value of id,
from which the final string is derived, being too low (lower than
1152921504606846976 or 0x1000000000000000) and the printf format
specifier having a width of 16. This results in a string of less than 16
characters that is padded with blank spaces. The end result is a cookie
directory that has leading spaces in its name, which can cause various
issues, such as breaking syncing on OneDrive.

If we set the format specifier to pad with zeroes instead of spaces, the
resulting hexadecimal value is always 16 characters long without spaces.

(cherry picked from commit 9bd4514b45)
2025-06-27 16:37:10 -04:00

159 lines
3.8 KiB
C++

/******************************************************************************
Copyright (C) 2023 by Lain Bailey <lain@obsproject.com>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/
#include <QDir>
#include <QThread>
#include <QMessageBox>
#include <qt-wrappers.hpp>
#include "window-basic-main.hpp"
#include <random>
#ifdef BROWSER_AVAILABLE
#include <browser-panel.hpp>
#endif
struct QCef;
struct QCefCookieManager;
extern QCef *cef;
extern QCefCookieManager *panel_cookies;
static std::string GenId()
{
std::random_device rd;
std::mt19937_64 e2(rd());
std::uniform_int_distribution<uint64_t> dist(0, 0xFFFFFFFFFFFFFFFF);
uint64_t id = dist(e2);
char id_str[20];
snprintf(id_str, sizeof(id_str), "%016llX", (unsigned long long)id);
return std::string(id_str);
}
void CheckExistingCookieId()
{
OBSBasic *main = OBSBasic::Get();
if (config_has_user_value(main->Config(), "Panels", "CookieId"))
return;
config_set_string(main->Config(), "Panels", "CookieId", GenId().c_str());
}
#ifdef BROWSER_AVAILABLE
static void InitPanelCookieManager()
{
if (!cef)
return;
if (panel_cookies)
return;
CheckExistingCookieId();
OBSBasic *main = OBSBasic::Get();
const char *cookie_id = config_get_string(main->Config(), "Panels", "CookieId");
std::string sub_path;
sub_path += "obs_profile_cookies/";
sub_path += cookie_id;
panel_cookies = cef->create_cookie_manager(sub_path);
}
#endif
void DestroyPanelCookieManager()
{
#ifdef BROWSER_AVAILABLE
if (panel_cookies) {
panel_cookies->FlushStore();
delete panel_cookies;
panel_cookies = nullptr;
}
#endif
}
void DeleteCookies()
{
#ifdef BROWSER_AVAILABLE
if (panel_cookies) {
panel_cookies->DeleteCookies("", "");
}
#endif
}
void DuplicateCurrentCookieProfile(ConfigFile &config)
{
#ifdef BROWSER_AVAILABLE
if (cef) {
OBSBasic *main = OBSBasic::Get();
std::string cookie_id = config_get_string(main->Config(), "Panels", "CookieId");
std::string src_path;
src_path += "obs_profile_cookies/";
src_path += cookie_id;
std::string new_id = GenId();
std::string dst_path;
dst_path += "obs_profile_cookies/";
dst_path += new_id;
BPtr<char> src_path_full = cef->get_cookie_path(src_path);
BPtr<char> dst_path_full = cef->get_cookie_path(dst_path);
QDir srcDir(src_path_full.Get());
QDir dstDir(dst_path_full.Get());
if (srcDir.exists()) {
if (!dstDir.exists())
dstDir.mkdir(dst_path_full.Get());
QStringList files = srcDir.entryList(QDir::Files);
for (const QString &file : files) {
QString src = QString(src_path_full);
QString dst = QString(dst_path_full);
src += QDir::separator() + file;
dst += QDir::separator() + file;
QFile::copy(src, dst);
}
}
config_set_string(config, "Panels", "CookieId", cookie_id.c_str());
config_set_string(main->Config(), "Panels", "CookieId", new_id.c_str());
}
#else
UNUSED_PARAMETER(config);
#endif
}
void OBSBasic::InitBrowserPanelSafeBlock()
{
#ifdef BROWSER_AVAILABLE
if (!cef)
return;
if (cef->init_browser()) {
InitPanelCookieManager();
return;
}
ExecThreadedWithoutBlocking([] { cef->wait_for_browser_init(); }, QTStr("BrowserPanelInit.Title"),
QTStr("BrowserPanelInit.Text"));
InitPanelCookieManager();
#endif
}