]> jfr.im git - yt-dlp.git/blame - yt_dlp/cookies.py
[ie/youtube] Suppress "Unavailable videos are hidden" warning (#10159)
[yt-dlp.git] / yt_dlp / cookies.py
CommitLineData
14f25df2 1import base64
b87e01c1 2import collections
f59f5ef8 3import contextlib
c305a25c 4import datetime as dt
cbed249a 5import glob
54007a45 6import http.cookiejar
8817a80d 7import http.cookies
b87e01c1 8import io
982ee69a
MB
9import json
10import os
9bd13fe5 11import re
982ee69a 12import shutil
982ee69a
MB
13import struct
14import subprocess
15import sys
16import tempfile
2e4585da 17import time
b87e01c1 18import urllib.request
f59f5ef8 19from enum import Enum, auto
982ee69a
MB
20from hashlib import pbkdf2_hmac
21
1d3586d0 22from .aes import (
23 aes_cbc_decrypt_bytes,
24 aes_gcm_decrypt_and_verify_bytes,
25 unpad_pkcs7,
26)
2792092a
SS
27from .compat import functools # isort: split
28from .compat import compat_os_name
9b8ee23b 29from .dependencies import (
30 _SECRETSTORAGE_UNAVAILABLE_REASON,
31 secretstorage,
32 sqlite3,
33)
97ec5bc5 34from .minicurses import MultilinePrinter, QuietMultilinePrinter
d2c8aadf 35from .utils import (
2792092a 36 DownloadError,
d2c8aadf 37 Popen,
d2c8aadf 38 error_to_str,
39 expand_path,
22df97f9 40 is_path_like,
b87e01c1 41 sanitize_url,
42 str_or_none,
d2c8aadf 43 try_call,
b87e01c1 44 write_string,
d2c8aadf 45)
1b392f90 46from .utils._utils import _YDLLogger
4bf91228 47from .utils.networking import normalize_url
982ee69a 48
dd9ad97b 49CHROMIUM_BASED_BROWSERS = {'brave', 'chrome', 'chromium', 'edge', 'opera', 'vivaldi', 'whale'}
982ee69a
MB
50SUPPORTED_BROWSERS = CHROMIUM_BASED_BROWSERS | {'firefox', 'safari'}
51
52
1b392f90 53class YDLLogger(_YDLLogger):
54 def warning(self, message, only_once=False): # compat
55 return super().warning(message, once=only_once)
982ee69a 56
2e4585da 57 class ProgressBar(MultilinePrinter):
58 _DELAY, _timer = 0.1, 0
59
60 def print(self, message):
61 if time.time() - self._timer > self._DELAY:
62 self.print_at_line(f'[Cookies] {message}', 0)
63 self._timer = time.time()
64
97ec5bc5 65 def progress_bar(self):
66 """Return a context manager with a print method. (Optional)"""
67 # Do not print to files/pipes, loggers, or when --no-progress is used
68 if not self._ydl or self._ydl.params.get('noprogress') or self._ydl.params.get('logger'):
69 return
591bb9d3 70 file = self._ydl._out_files.error
97ec5bc5 71 try:
72 if not file.isatty():
73 return
74 except BaseException:
75 return
2e4585da 76 return self.ProgressBar(file, preserve_output=False)
97ec5bc5 77
78
79def _create_progress_bar(logger):
80 if hasattr(logger, 'progress_bar'):
81 printer = logger.progress_bar()
82 if printer:
83 return printer
84 printer = QuietMultilinePrinter()
85 printer.print = lambda _: None
86 return printer
87
982ee69a
MB
88
89def load_cookies(cookie_file, browser_specification, ydl):
90 cookie_jars = []
91 if browser_specification is not None:
9bd13fe5 92 browser_name, profile, keyring, container = _parse_browser_specification(*browser_specification)
93 cookie_jars.append(
94 extract_cookies_from_browser(browser_name, profile, YDLLogger(ydl), keyring=keyring, container=container))
982ee69a
MB
95
96 if cookie_file is not None:
22df97f9 97 is_filename = is_path_like(cookie_file)
d76fa1f3 98 if is_filename:
99 cookie_file = expand_path(cookie_file)
100
982ee69a 101 jar = YoutubeDLCookieJar(cookie_file)
d76fa1f3 102 if not is_filename or os.access(cookie_file, os.R_OK):
62b5c94c 103 jar.load()
982ee69a
MB
104 cookie_jars.append(jar)
105
106 return _merge_cookie_jars(cookie_jars)
107
108
9bd13fe5 109def extract_cookies_from_browser(browser_name, profile=None, logger=YDLLogger(), *, keyring=None, container=None):
982ee69a 110 if browser_name == 'firefox':
9bd13fe5 111 return _extract_firefox_cookies(profile, container, logger)
982ee69a
MB
112 elif browser_name == 'safari':
113 return _extract_safari_cookies(profile, logger)
114 elif browser_name in CHROMIUM_BASED_BROWSERS:
f59f5ef8 115 return _extract_chrome_cookies(browser_name, profile, keyring, logger)
982ee69a 116 else:
86e5f3ed 117 raise ValueError(f'unknown browser: {browser_name}')
982ee69a
MB
118
119
9bd13fe5 120def _extract_firefox_cookies(profile, container, logger):
982ee69a 121 logger.info('Extracting cookies from firefox')
9b8ee23b 122 if not sqlite3:
767b02a9 123 logger.warning('Cannot extract cookies from firefox without sqlite3 support. '
47ab66db 124 'Please use a Python interpreter compiled with sqlite3 support')
767b02a9 125 return YoutubeDLCookieJar()
982ee69a
MB
126
127 if profile is None:
cbed249a 128 search_roots = list(_firefox_browser_dirs())
982ee69a 129 elif _is_path(profile):
cbed249a 130 search_roots = [profile]
982ee69a 131 else:
cbed249a
SS
132 search_roots = [os.path.join(path, profile) for path in _firefox_browser_dirs()]
133 search_root = ', '.join(map(repr, search_roots))
982ee69a 134
cbed249a 135 cookie_database_path = _newest(_firefox_cookie_dbs(search_roots))
825d3ce3 136 if cookie_database_path is None:
137 raise FileNotFoundError(f'could not find firefox cookies database in {search_root}')
138 logger.debug(f'Extracting cookies from: "{cookie_database_path}"')
139
9bd13fe5 140 container_id = None
825d3ce3 141 if container not in (None, 'none'):
142 containers_path = os.path.join(os.path.dirname(cookie_database_path), 'containers.json')
9bd13fe5 143 if not os.path.isfile(containers_path) or not os.access(containers_path, os.R_OK):
144 raise FileNotFoundError(f'could not read containers.json in {search_root}')
dab87ca2 145 with open(containers_path, encoding='utf8') as containers:
9bd13fe5 146 identities = json.load(containers).get('identities', [])
147 container_id = next((context.get('userContextId') for context in identities if container in (
148 context.get('name'),
add96eb9 149 try_call(lambda: re.fullmatch(r'userContext([^\.]+)\.label', context['l10nID']).group()),
9bd13fe5 150 )), None)
151 if not isinstance(container_id, int):
152 raise ValueError(f'could not find firefox container "{container}" in containers.json')
153
0930b11f 154 with tempfile.TemporaryDirectory(prefix='yt_dlp') as tmpdir:
982ee69a
MB
155 cursor = None
156 try:
157 cursor = _open_database_copy(cookie_database_path, tmpdir)
9bd13fe5 158 if isinstance(container_id, int):
9bd13fe5 159 logger.debug(
160 f'Only loading cookies from firefox container "{container}", ID {container_id}')
9bd13fe5 161 cursor.execute(
825d3ce3 162 'SELECT host, name, value, path, expiry, isSecure FROM moz_cookies WHERE originAttributes LIKE ? OR originAttributes LIKE ?',
163 (f'%userContextId={container_id}', f'%userContextId={container_id}&%'))
164 elif container == 'none':
165 logger.debug('Only loading cookies not belonging to any container')
166 cursor.execute(
167 'SELECT host, name, value, path, expiry, isSecure FROM moz_cookies WHERE NOT INSTR(originAttributes,"userContextId=")')
168 else:
9bd13fe5 169 cursor.execute('SELECT host, name, value, path, expiry, isSecure FROM moz_cookies')
982ee69a 170 jar = YoutubeDLCookieJar()
97ec5bc5 171 with _create_progress_bar(logger) as progress_bar:
172 table = cursor.fetchall()
173 total_cookie_count = len(table)
174 for i, (host, name, value, path, expiry, is_secure) in enumerate(table):
175 progress_bar.print(f'Loading cookie {i: 6d}/{total_cookie_count: 6d}')
ac668111 176 cookie = http.cookiejar.Cookie(
97ec5bc5 177 version=0, name=name, value=value, port=None, port_specified=False,
178 domain=host, domain_specified=bool(host), domain_initial_dot=host.startswith('.'),
179 path=path, path_specified=bool(path), secure=is_secure, expires=expiry, discard=False,
180 comment=None, comment_url=None, rest={})
181 jar.set_cookie(cookie)
86e5f3ed 182 logger.info(f'Extracted {len(jar)} cookies from firefox')
982ee69a
MB
183 return jar
184 finally:
185 if cursor is not None:
186 cursor.connection.close()
187
188
cbed249a 189def _firefox_browser_dirs():
dec30912 190 if sys.platform in ('cygwin', 'win32'):
cbed249a
SS
191 yield os.path.expandvars(R'%APPDATA%\Mozilla\Firefox\Profiles')
192
982ee69a 193 elif sys.platform == 'darwin':
cbed249a
SS
194 yield os.path.expanduser('~/Library/Application Support/Firefox/Profiles')
195
196 else:
2ab2651a
DM
197 yield from map(os.path.expanduser, (
198 '~/.mozilla/firefox',
199 '~/snap/firefox/common/.mozilla/firefox',
200 '~/.var/app/org.mozilla.firefox/.mozilla/firefox',
201 ))
cbed249a
SS
202
203
204def _firefox_cookie_dbs(roots):
205 for root in map(os.path.abspath, roots):
206 for pattern in ('', '*/', 'Profiles/*/'):
207 yield from glob.iglob(os.path.join(root, pattern, 'cookies.sqlite'))
982ee69a
MB
208
209
210def _get_chromium_based_browser_settings(browser_name):
211 # https://chromium.googlesource.com/chromium/src/+/HEAD/docs/user_data_dir.md
dec30912 212 if sys.platform in ('cygwin', 'win32'):
982ee69a
MB
213 appdata_local = os.path.expandvars('%LOCALAPPDATA%')
214 appdata_roaming = os.path.expandvars('%APPDATA%')
215 browser_dir = {
19a03940 216 'brave': os.path.join(appdata_local, R'BraveSoftware\Brave-Browser\User Data'),
217 'chrome': os.path.join(appdata_local, R'Google\Chrome\User Data'),
218 'chromium': os.path.join(appdata_local, R'Chromium\User Data'),
219 'edge': os.path.join(appdata_local, R'Microsoft\Edge\User Data'),
220 'opera': os.path.join(appdata_roaming, R'Opera Software\Opera Stable'),
221 'vivaldi': os.path.join(appdata_local, R'Vivaldi\User Data'),
dd9ad97b 222 'whale': os.path.join(appdata_local, R'Naver\Naver Whale\User Data'),
982ee69a
MB
223 }[browser_name]
224
225 elif sys.platform == 'darwin':
226 appdata = os.path.expanduser('~/Library/Application Support')
227 browser_dir = {
228 'brave': os.path.join(appdata, 'BraveSoftware/Brave-Browser'),
229 'chrome': os.path.join(appdata, 'Google/Chrome'),
230 'chromium': os.path.join(appdata, 'Chromium'),
231 'edge': os.path.join(appdata, 'Microsoft Edge'),
232 'opera': os.path.join(appdata, 'com.operasoftware.Opera'),
233 'vivaldi': os.path.join(appdata, 'Vivaldi'),
dd9ad97b 234 'whale': os.path.join(appdata, 'Naver/Whale'),
982ee69a
MB
235 }[browser_name]
236
237 else:
dec30912
CMT
238 config = _config_home()
239 browser_dir = {
240 'brave': os.path.join(config, 'BraveSoftware/Brave-Browser'),
241 'chrome': os.path.join(config, 'google-chrome'),
242 'chromium': os.path.join(config, 'chromium'),
243 'edge': os.path.join(config, 'microsoft-edge'),
244 'opera': os.path.join(config, 'opera'),
245 'vivaldi': os.path.join(config, 'vivaldi'),
dd9ad97b 246 'whale': os.path.join(config, 'naver-whale'),
dec30912 247 }[browser_name]
982ee69a
MB
248
249 # Linux keyring names can be determined by snooping on dbus while opening the browser in KDE:
250 # dbus-monitor "interface='org.kde.KWallet'" "type=method_return"
251 keyring_name = {
252 'brave': 'Brave',
253 'chrome': 'Chrome',
254 'chromium': 'Chromium',
29b208f6 255 'edge': 'Microsoft Edge' if sys.platform == 'darwin' else 'Chromium',
982ee69a
MB
256 'opera': 'Opera' if sys.platform == 'darwin' else 'Chromium',
257 'vivaldi': 'Vivaldi' if sys.platform == 'darwin' else 'Chrome',
dd9ad97b 258 'whale': 'Whale',
982ee69a
MB
259 }[browser_name]
260
261 browsers_without_profiles = {'opera'}
262
263 return {
264 'browser_dir': browser_dir,
265 'keyring_name': keyring_name,
add96eb9 266 'supports_profiles': browser_name not in browsers_without_profiles,
982ee69a
MB
267 }
268
269
f59f5ef8 270def _extract_chrome_cookies(browser_name, profile, keyring, logger):
86e5f3ed 271 logger.info(f'Extracting cookies from {browser_name}')
767b02a9 272
9b8ee23b 273 if not sqlite3:
19a03940 274 logger.warning(f'Cannot extract cookies from {browser_name} without sqlite3 support. '
47ab66db 275 'Please use a Python interpreter compiled with sqlite3 support')
767b02a9
MB
276 return YoutubeDLCookieJar()
277
982ee69a
MB
278 config = _get_chromium_based_browser_settings(browser_name)
279
280 if profile is None:
281 search_root = config['browser_dir']
282 elif _is_path(profile):
283 search_root = profile
284 config['browser_dir'] = os.path.dirname(profile) if config['supports_profiles'] else profile
285 else:
286 if config['supports_profiles']:
287 search_root = os.path.join(config['browser_dir'], profile)
288 else:
86e5f3ed 289 logger.error(f'{browser_name} does not support profiles')
982ee69a
MB
290 search_root = config['browser_dir']
291
cbed249a 292 cookie_database_path = _newest(_find_files(search_root, 'Cookies', logger))
982ee69a 293 if cookie_database_path is None:
86e5f3ed 294 raise FileNotFoundError(f'could not find {browser_name} cookies database in "{search_root}"')
295 logger.debug(f'Extracting cookies from: "{cookie_database_path}"')
982ee69a 296
f59f5ef8 297 decryptor = get_cookie_decryptor(config['browser_dir'], config['keyring_name'], logger, keyring=keyring)
982ee69a 298
0930b11f 299 with tempfile.TemporaryDirectory(prefix='yt_dlp') as tmpdir:
982ee69a
MB
300 cursor = None
301 try:
302 cursor = _open_database_copy(cookie_database_path, tmpdir)
303 cursor.connection.text_factory = bytes
304 column_names = _get_column_names(cursor, 'cookies')
305 secure_column = 'is_secure' if 'is_secure' in column_names else 'secure'
19a03940 306 cursor.execute(f'SELECT host_key, name, value, encrypted_value, path, expires_utc, {secure_column} FROM cookies')
982ee69a
MB
307 jar = YoutubeDLCookieJar()
308 failed_cookies = 0
f59f5ef8 309 unencrypted_cookies = 0
97ec5bc5 310 with _create_progress_bar(logger) as progress_bar:
311 table = cursor.fetchall()
312 total_cookie_count = len(table)
313 for i, line in enumerate(table):
314 progress_bar.print(f'Loading cookie {i: 6d}/{total_cookie_count: 6d}')
315 is_encrypted, cookie = _process_chrome_cookie(decryptor, *line)
316 if not cookie:
982ee69a
MB
317 failed_cookies += 1
318 continue
97ec5bc5 319 elif not is_encrypted:
320 unencrypted_cookies += 1
321 jar.set_cookie(cookie)
982ee69a 322 if failed_cookies > 0:
86e5f3ed 323 failed_message = f' ({failed_cookies} could not be decrypted)'
982ee69a
MB
324 else:
325 failed_message = ''
86e5f3ed 326 logger.info(f'Extracted {len(jar)} cookies from {browser_name}{failed_message}')
24146491 327 counts = decryptor._cookie_counts.copy()
f59f5ef8 328 counts['unencrypted'] = unencrypted_cookies
86e5f3ed 329 logger.debug(f'cookie version breakdown: {counts}')
982ee69a 330 return jar
2792092a
SS
331 except PermissionError as error:
332 if compat_os_name == 'nt' and error.errno == 13:
333 message = 'Could not copy Chrome cookie database. See https://github.com/yt-dlp/yt-dlp/issues/7271 for more info'
334 logger.error(message)
335 raise DownloadError(message) # force exit
336 raise
982ee69a
MB
337 finally:
338 if cursor is not None:
339 cursor.connection.close()
340
341
97ec5bc5 342def _process_chrome_cookie(decryptor, host_key, name, value, encrypted_value, path, expires_utc, is_secure):
0f06bcd7 343 host_key = host_key.decode()
344 name = name.decode()
345 value = value.decode()
346 path = path.decode()
97ec5bc5 347 is_encrypted = not value and encrypted_value
348
349 if is_encrypted:
350 value = decryptor.decrypt(encrypted_value)
351 if value is None:
352 return is_encrypted, None
353
f1f15897
SL
354 # In chrome, session cookies have expires_utc set to 0
355 # In our cookie-store, cookies that do not expire should have expires set to None
356 if not expires_utc:
357 expires_utc = None
358
ac668111 359 return is_encrypted, http.cookiejar.Cookie(
97ec5bc5 360 version=0, name=name, value=value, port=None, port_specified=False,
361 domain=host_key, domain_specified=bool(host_key), domain_initial_dot=host_key.startswith('.'),
362 path=path, path_specified=bool(path), secure=is_secure, expires=expires_utc, discard=False,
363 comment=None, comment_url=None, rest={})
364
365
982ee69a
MB
366class ChromeCookieDecryptor:
367 """
368 Overview:
369
370 Linux:
371 - cookies are either v10 or v11
372 - v10: AES-CBC encrypted with a fixed key
b38d4c94 373 - also attempts empty password if decryption fails
982ee69a 374 - v11: AES-CBC encrypted with an OS protected key (keyring)
b38d4c94 375 - also attempts empty password if decryption fails
982ee69a
MB
376 - v11 keys can be stored in various places depending on the activate desktop environment [2]
377
378 Mac:
379 - cookies are either v10 or not v10
380 - v10: AES-CBC encrypted with an OS protected key (keyring) and more key derivation iterations than linux
381 - not v10: 'old data' stored as plaintext
382
383 Windows:
384 - cookies are either v10 or not v10
385 - v10: AES-GCM encrypted with a key which is encrypted with DPAPI
386 - not v10: encrypted with DPAPI
387
388 Sources:
389 - [1] https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/
b38d4c94 390 - [2] https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/key_storage_linux.cc
982ee69a
MB
391 - KeyStorageLinux::CreateService
392 """
393
0fa7d2c8 394 _cookie_counts = {}
982ee69a 395
0fa7d2c8 396 def decrypt(self, encrypted_value):
19a03940 397 raise NotImplementedError('Must be implemented by sub classes')
f59f5ef8 398
982ee69a 399
f59f5ef8 400def get_cookie_decryptor(browser_root, browser_keyring_name, logger, *, keyring=None):
56ba69e4 401 if sys.platform == 'darwin':
982ee69a 402 return MacChromeCookieDecryptor(browser_keyring_name, logger)
56ba69e4 403 elif sys.platform in ('win32', 'cygwin'):
982ee69a 404 return WindowsChromeCookieDecryptor(browser_root, logger)
56ba69e4 405 return LinuxChromeCookieDecryptor(browser_keyring_name, logger, keyring=keyring)
982ee69a
MB
406
407
408class LinuxChromeCookieDecryptor(ChromeCookieDecryptor):
f59f5ef8 409 def __init__(self, browser_keyring_name, logger, *, keyring=None):
982ee69a
MB
410 self._logger = logger
411 self._v10_key = self.derive_key(b'peanuts')
b38d4c94 412 self._empty_key = self.derive_key(b'')
f59f5ef8 413 self._cookie_counts = {'v10': 0, 'v11': 0, 'other': 0}
9b7a48ab
SS
414 self._browser_keyring_name = browser_keyring_name
415 self._keyring = keyring
416
417 @functools.cached_property
418 def _v11_key(self):
419 password = _get_linux_keyring_password(self._browser_keyring_name, self._keyring, self._logger)
420 return None if password is None else self.derive_key(password)
982ee69a
MB
421
422 @staticmethod
423 def derive_key(password):
424 # values from
b38d4c94 425 # https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/os_crypt_linux.cc
982ee69a
MB
426 return pbkdf2_sha1(password, salt=b'saltysalt', iterations=1, key_length=16)
427
428 def decrypt(self, encrypted_value):
b38d4c94
MB
429 """
430
431 following the same approach as the fix in [1]: if cookies fail to decrypt then attempt to decrypt
432 with an empty password. The failure detection is not the same as what chromium uses so the
433 results won't be perfect
434
435 References:
436 - [1] https://chromium.googlesource.com/chromium/src/+/bbd54702284caca1f92d656fdcadf2ccca6f4165%5E%21/
437 - a bugfix to try an empty password as a fallback
438 """
982ee69a
MB
439 version = encrypted_value[:3]
440 ciphertext = encrypted_value[3:]
441
442 if version == b'v10':
f59f5ef8 443 self._cookie_counts['v10'] += 1
b38d4c94 444 return _decrypt_aes_cbc_multi(ciphertext, (self._v10_key, self._empty_key), self._logger)
982ee69a
MB
445
446 elif version == b'v11':
f59f5ef8 447 self._cookie_counts['v11'] += 1
982ee69a 448 if self._v11_key is None:
f59f5ef8 449 self._logger.warning('cannot decrypt v11 cookies: no key found', only_once=True)
982ee69a 450 return None
b38d4c94 451 return _decrypt_aes_cbc_multi(ciphertext, (self._v11_key, self._empty_key), self._logger)
982ee69a
MB
452
453 else:
b38d4c94 454 self._logger.warning(f'unknown cookie version: "{version}"', only_once=True)
f59f5ef8 455 self._cookie_counts['other'] += 1
982ee69a
MB
456 return None
457
458
459class MacChromeCookieDecryptor(ChromeCookieDecryptor):
460 def __init__(self, browser_keyring_name, logger):
461 self._logger = logger
f440b14f 462 password = _get_mac_keyring_password(browser_keyring_name, logger)
982ee69a 463 self._v10_key = None if password is None else self.derive_key(password)
f59f5ef8 464 self._cookie_counts = {'v10': 0, 'other': 0}
982ee69a
MB
465
466 @staticmethod
467 def derive_key(password):
468 # values from
b38d4c94 469 # https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/os_crypt_mac.mm
982ee69a
MB
470 return pbkdf2_sha1(password, salt=b'saltysalt', iterations=1003, key_length=16)
471
472 def decrypt(self, encrypted_value):
473 version = encrypted_value[:3]
474 ciphertext = encrypted_value[3:]
475
476 if version == b'v10':
f59f5ef8 477 self._cookie_counts['v10'] += 1
982ee69a
MB
478 if self._v10_key is None:
479 self._logger.warning('cannot decrypt v10 cookies: no key found', only_once=True)
480 return None
481
b38d4c94 482 return _decrypt_aes_cbc_multi(ciphertext, (self._v10_key,), self._logger)
982ee69a
MB
483
484 else:
f59f5ef8 485 self._cookie_counts['other'] += 1
982ee69a 486 # other prefixes are considered 'old data' which were stored as plaintext
b38d4c94 487 # https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/os_crypt_mac.mm
982ee69a
MB
488 return encrypted_value
489
490
491class WindowsChromeCookieDecryptor(ChromeCookieDecryptor):
492 def __init__(self, browser_root, logger):
493 self._logger = logger
494 self._v10_key = _get_windows_v10_key(browser_root, logger)
f59f5ef8
MB
495 self._cookie_counts = {'v10': 0, 'other': 0}
496
982ee69a
MB
497 def decrypt(self, encrypted_value):
498 version = encrypted_value[:3]
499 ciphertext = encrypted_value[3:]
500
501 if version == b'v10':
f59f5ef8 502 self._cookie_counts['v10'] += 1
982ee69a
MB
503 if self._v10_key is None:
504 self._logger.warning('cannot decrypt v10 cookies: no key found', only_once=True)
505 return None
982ee69a 506
b38d4c94 507 # https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/os_crypt_win.cc
982ee69a
MB
508 # kNonceLength
509 nonce_length = 96 // 8
510 # boringssl
511 # EVP_AEAD_AES_GCM_TAG_LEN
512 authentication_tag_length = 16
513
514 raw_ciphertext = ciphertext
515 nonce = raw_ciphertext[:nonce_length]
516 ciphertext = raw_ciphertext[nonce_length:-authentication_tag_length]
517 authentication_tag = raw_ciphertext[-authentication_tag_length:]
518
519 return _decrypt_aes_gcm(ciphertext, self._v10_key, nonce, authentication_tag, self._logger)
520
521 else:
f59f5ef8 522 self._cookie_counts['other'] += 1
982ee69a 523 # any other prefix means the data is DPAPI encrypted
b38d4c94 524 # https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/os_crypt_win.cc
0f06bcd7 525 return _decrypt_windows_dpapi(encrypted_value, self._logger).decode()
982ee69a
MB
526
527
528def _extract_safari_cookies(profile, logger):
982ee69a 529 if sys.platform != 'darwin':
86e5f3ed 530 raise ValueError(f'unsupported platform: {sys.platform}')
982ee69a 531
a58182b7
NV
532 if profile:
533 cookies_path = os.path.expanduser(profile)
534 if not os.path.isfile(cookies_path):
535 raise FileNotFoundError('custom safari cookies database not found')
536
537 else:
538 cookies_path = os.path.expanduser('~/Library/Cookies/Cookies.binarycookies')
982ee69a 539
1f7db853 540 if not os.path.isfile(cookies_path):
a58182b7
NV
541 logger.debug('Trying secondary cookie location')
542 cookies_path = os.path.expanduser('~/Library/Containers/com.apple.Safari/Data/Library/Cookies/Cookies.binarycookies')
543 if not os.path.isfile(cookies_path):
544 raise FileNotFoundError('could not find safari cookies database')
982ee69a
MB
545
546 with open(cookies_path, 'rb') as f:
547 cookies_data = f.read()
548
549 jar = parse_safari_cookies(cookies_data, logger=logger)
86e5f3ed 550 logger.info(f'Extracted {len(jar)} cookies from safari')
982ee69a
MB
551 return jar
552
553
554class ParserError(Exception):
555 pass
556
557
558class DataParser:
559 def __init__(self, data, logger):
560 self._data = data
561 self.cursor = 0
562 self._logger = logger
563
564 def read_bytes(self, num_bytes):
565 if num_bytes < 0:
86e5f3ed 566 raise ParserError(f'invalid read of {num_bytes} bytes')
982ee69a
MB
567 end = self.cursor + num_bytes
568 if end > len(self._data):
569 raise ParserError('reached end of input')
570 data = self._data[self.cursor:end]
571 self.cursor = end
572 return data
573
574 def expect_bytes(self, expected_value, message):
575 value = self.read_bytes(len(expected_value))
576 if value != expected_value:
86e5f3ed 577 raise ParserError(f'unexpected value: {value} != {expected_value} ({message})')
982ee69a
MB
578
579 def read_uint(self, big_endian=False):
580 data_format = '>I' if big_endian else '<I'
581 return struct.unpack(data_format, self.read_bytes(4))[0]
582
583 def read_double(self, big_endian=False):
584 data_format = '>d' if big_endian else '<d'
585 return struct.unpack(data_format, self.read_bytes(8))[0]
586
587 def read_cstring(self):
588 buffer = []
589 while True:
590 c = self.read_bytes(1)
591 if c == b'\x00':
0f06bcd7 592 return b''.join(buffer).decode()
982ee69a
MB
593 else:
594 buffer.append(c)
595
596 def skip(self, num_bytes, description='unknown'):
597 if num_bytes > 0:
19a03940 598 self._logger.debug(f'skipping {num_bytes} bytes ({description}): {self.read_bytes(num_bytes)!r}')
982ee69a 599 elif num_bytes < 0:
86e5f3ed 600 raise ParserError(f'invalid skip of {num_bytes} bytes')
982ee69a
MB
601
602 def skip_to(self, offset, description='unknown'):
603 self.skip(offset - self.cursor, description)
604
605 def skip_to_end(self, description='unknown'):
606 self.skip_to(len(self._data), description)
607
608
609def _mac_absolute_time_to_posix(timestamp):
c305a25c 610 return int((dt.datetime(2001, 1, 1, 0, 0, tzinfo=dt.timezone.utc) + dt.timedelta(seconds=timestamp)).timestamp())
982ee69a
MB
611
612
613def _parse_safari_cookies_header(data, logger):
614 p = DataParser(data, logger)
615 p.expect_bytes(b'cook', 'database signature')
616 number_of_pages = p.read_uint(big_endian=True)
617 page_sizes = [p.read_uint(big_endian=True) for _ in range(number_of_pages)]
618 return page_sizes, p.cursor
619
620
621def _parse_safari_cookies_page(data, jar, logger):
622 p = DataParser(data, logger)
623 p.expect_bytes(b'\x00\x00\x01\x00', 'page signature')
624 number_of_cookies = p.read_uint()
625 record_offsets = [p.read_uint() for _ in range(number_of_cookies)]
626 if number_of_cookies == 0:
86e5f3ed 627 logger.debug(f'a cookies page of size {len(data)} has no cookies')
982ee69a
MB
628 return
629
630 p.skip_to(record_offsets[0], 'unknown page header field')
631
97ec5bc5 632 with _create_progress_bar(logger) as progress_bar:
633 for i, record_offset in enumerate(record_offsets):
634 progress_bar.print(f'Loading cookie {i: 6d}/{number_of_cookies: 6d}')
635 p.skip_to(record_offset, 'space between records')
636 record_length = _parse_safari_cookies_record(data[record_offset:], jar, logger)
637 p.read_bytes(record_length)
982ee69a
MB
638 p.skip_to_end('space in between pages')
639
640
641def _parse_safari_cookies_record(data, jar, logger):
642 p = DataParser(data, logger)
643 record_size = p.read_uint()
644 p.skip(4, 'unknown record field 1')
645 flags = p.read_uint()
646 is_secure = bool(flags & 0x0001)
647 p.skip(4, 'unknown record field 2')
648 domain_offset = p.read_uint()
649 name_offset = p.read_uint()
650 path_offset = p.read_uint()
651 value_offset = p.read_uint()
652 p.skip(8, 'unknown record field 3')
653 expiration_date = _mac_absolute_time_to_posix(p.read_double())
654 _creation_date = _mac_absolute_time_to_posix(p.read_double()) # noqa: F841
655
656 try:
657 p.skip_to(domain_offset)
658 domain = p.read_cstring()
659
660 p.skip_to(name_offset)
661 name = p.read_cstring()
662
663 p.skip_to(path_offset)
664 path = p.read_cstring()
665
666 p.skip_to(value_offset)
667 value = p.read_cstring()
668 except UnicodeDecodeError:
f440b14f 669 logger.warning('failed to parse Safari cookie because UTF-8 decoding failed', only_once=True)
982ee69a
MB
670 return record_size
671
672 p.skip_to(record_size, 'space at the end of the record')
673
ac668111 674 cookie = http.cookiejar.Cookie(
982ee69a
MB
675 version=0, name=name, value=value, port=None, port_specified=False,
676 domain=domain, domain_specified=bool(domain), domain_initial_dot=domain.startswith('.'),
677 path=path, path_specified=bool(path), secure=is_secure, expires=expiration_date, discard=False,
678 comment=None, comment_url=None, rest={})
679 jar.set_cookie(cookie)
680 return record_size
681
682
683def parse_safari_cookies(data, jar=None, logger=YDLLogger()):
684 """
685 References:
686 - https://github.com/libyal/dtformats/blob/main/documentation/Safari%20Cookies.asciidoc
687 - this data appears to be out of date but the important parts of the database structure is the same
688 - there are a few bytes here and there which are skipped during parsing
689 """
690 if jar is None:
691 jar = YoutubeDLCookieJar()
692 page_sizes, body_start = _parse_safari_cookies_header(data, logger)
693 p = DataParser(data[body_start:], logger)
694 for page_size in page_sizes:
695 _parse_safari_cookies_page(p.read_bytes(page_size), jar, logger)
696 p.skip_to_end('footer')
697 return jar
698
699
f59f5ef8
MB
700class _LinuxDesktopEnvironment(Enum):
701 """
702 https://chromium.googlesource.com/chromium/src/+/refs/heads/main/base/nix/xdg_util.h
703 DesktopEnvironment
704 """
705 OTHER = auto()
706 CINNAMON = auto()
b38d4c94 707 DEEPIN = auto()
f59f5ef8 708 GNOME = auto()
b38d4c94
MB
709 KDE3 = auto()
710 KDE4 = auto()
711 KDE5 = auto()
712 KDE6 = auto()
f59f5ef8 713 PANTHEON = auto()
b38d4c94 714 UKUI = auto()
f59f5ef8
MB
715 UNITY = auto()
716 XFCE = auto()
b38d4c94 717 LXQT = auto()
982ee69a
MB
718
719
f59f5ef8
MB
720class _LinuxKeyring(Enum):
721 """
b38d4c94 722 https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/key_storage_util_linux.h
f59f5ef8
MB
723 SelectedLinuxBackend
724 """
2e023649 725 KWALLET = auto() # KDE4
b38d4c94
MB
726 KWALLET5 = auto()
727 KWALLET6 = auto()
2e023649 728 GNOMEKEYRING = auto()
729 BASICTEXT = auto()
f59f5ef8
MB
730
731
732SUPPORTED_KEYRINGS = _LinuxKeyring.__members__.keys()
733
734
b38d4c94 735def _get_linux_desktop_environment(env, logger):
f59f5ef8
MB
736 """
737 https://chromium.googlesource.com/chromium/src/+/refs/heads/main/base/nix/xdg_util.cc
738 GetDesktopEnvironment
739 """
740 xdg_current_desktop = env.get('XDG_CURRENT_DESKTOP', None)
741 desktop_session = env.get('DESKTOP_SESSION', None)
742 if xdg_current_desktop is not None:
a8520244
PW
743 for part in map(str.strip, xdg_current_desktop.split(':')):
744 if part == 'Unity':
745 if desktop_session is not None and 'gnome-fallback' in desktop_session:
746 return _LinuxDesktopEnvironment.GNOME
747 else:
748 return _LinuxDesktopEnvironment.UNITY
749 elif part == 'Deepin':
750 return _LinuxDesktopEnvironment.DEEPIN
751 elif part == 'GNOME':
f59f5ef8 752 return _LinuxDesktopEnvironment.GNOME
a8520244
PW
753 elif part == 'X-Cinnamon':
754 return _LinuxDesktopEnvironment.CINNAMON
755 elif part == 'KDE':
756 kde_version = env.get('KDE_SESSION_VERSION', None)
757 if kde_version == '5':
758 return _LinuxDesktopEnvironment.KDE5
759 elif kde_version == '6':
760 return _LinuxDesktopEnvironment.KDE6
761 elif kde_version == '4':
762 return _LinuxDesktopEnvironment.KDE4
763 else:
764 logger.info(f'unknown KDE version: "{kde_version}". Assuming KDE4')
765 return _LinuxDesktopEnvironment.KDE4
766 elif part == 'Pantheon':
767 return _LinuxDesktopEnvironment.PANTHEON
768 elif part == 'XFCE':
769 return _LinuxDesktopEnvironment.XFCE
770 elif part == 'UKUI':
771 return _LinuxDesktopEnvironment.UKUI
772 elif part == 'LXQt':
773 return _LinuxDesktopEnvironment.LXQT
774 logger.info(f'XDG_CURRENT_DESKTOP is set to an unknown value: "{xdg_current_desktop}"')
b38d4c94 775
f59f5ef8 776 elif desktop_session is not None:
b38d4c94
MB
777 if desktop_session == 'deepin':
778 return _LinuxDesktopEnvironment.DEEPIN
779 elif desktop_session in ('mate', 'gnome'):
f59f5ef8 780 return _LinuxDesktopEnvironment.GNOME
b38d4c94
MB
781 elif desktop_session in ('kde4', 'kde-plasma'):
782 return _LinuxDesktopEnvironment.KDE4
783 elif desktop_session == 'kde':
784 if 'KDE_SESSION_VERSION' in env:
785 return _LinuxDesktopEnvironment.KDE4
786 else:
787 return _LinuxDesktopEnvironment.KDE3
788 elif 'xfce' in desktop_session or desktop_session == 'xubuntu':
f59f5ef8 789 return _LinuxDesktopEnvironment.XFCE
b38d4c94
MB
790 elif desktop_session == 'ukui':
791 return _LinuxDesktopEnvironment.UKUI
792 else:
793 logger.info(f'DESKTOP_SESSION is set to an unknown value: "{desktop_session}"')
794
f59f5ef8
MB
795 else:
796 if 'GNOME_DESKTOP_SESSION_ID' in env:
797 return _LinuxDesktopEnvironment.GNOME
798 elif 'KDE_FULL_SESSION' in env:
b38d4c94
MB
799 if 'KDE_SESSION_VERSION' in env:
800 return _LinuxDesktopEnvironment.KDE4
801 else:
802 return _LinuxDesktopEnvironment.KDE3
fa8fd951 803 return _LinuxDesktopEnvironment.OTHER
f59f5ef8
MB
804
805
806def _choose_linux_keyring(logger):
807 """
b38d4c94
MB
808 SelectBackend in [1]
809
810 There is currently support for forcing chromium to use BASIC_TEXT by creating a file called
811 `Disable Local Encryption` [1] in the user data dir. The function to write this file (`WriteBackendUse()` [1])
812 does not appear to be called anywhere other than in tests, so the user would have to create this file manually
813 and so would be aware enough to tell yt-dlp to use the BASIC_TEXT keyring.
814
815 References:
816 - [1] https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/key_storage_util_linux.cc
f59f5ef8 817 """
b38d4c94 818 desktop_environment = _get_linux_desktop_environment(os.environ, logger)
86e5f3ed 819 logger.debug(f'detected desktop environment: {desktop_environment.name}')
b38d4c94 820 if desktop_environment == _LinuxDesktopEnvironment.KDE4:
2e023649 821 linux_keyring = _LinuxKeyring.KWALLET
b38d4c94
MB
822 elif desktop_environment == _LinuxDesktopEnvironment.KDE5:
823 linux_keyring = _LinuxKeyring.KWALLET5
824 elif desktop_environment == _LinuxDesktopEnvironment.KDE6:
825 linux_keyring = _LinuxKeyring.KWALLET6
826 elif desktop_environment in (
add96eb9 827 _LinuxDesktopEnvironment.KDE3, _LinuxDesktopEnvironment.LXQT, _LinuxDesktopEnvironment.OTHER,
b38d4c94 828 ):
2e023649 829 linux_keyring = _LinuxKeyring.BASICTEXT
982ee69a 830 else:
2e023649 831 linux_keyring = _LinuxKeyring.GNOMEKEYRING
f59f5ef8
MB
832 return linux_keyring
833
834
b38d4c94 835def _get_kwallet_network_wallet(keyring, logger):
f59f5ef8
MB
836 """ The name of the wallet used to store network passwords.
837
b38d4c94 838 https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/kwallet_dbus.cc
f59f5ef8
MB
839 KWalletDBus::NetworkWallet
840 which does a dbus call to the following function:
841 https://api.kde.org/frameworks/kwallet/html/classKWallet_1_1Wallet.html
842 Wallet::NetworkWallet
843 """
844 default_wallet = 'kdewallet'
845 try:
2e023649 846 if keyring == _LinuxKeyring.KWALLET:
b38d4c94
MB
847 service_name = 'org.kde.kwalletd'
848 wallet_path = '/modules/kwalletd'
849 elif keyring == _LinuxKeyring.KWALLET5:
850 service_name = 'org.kde.kwalletd5'
851 wallet_path = '/modules/kwalletd5'
852 elif keyring == _LinuxKeyring.KWALLET6:
853 service_name = 'org.kde.kwalletd6'
854 wallet_path = '/modules/kwalletd6'
855 else:
856 raise ValueError(keyring)
857
f0c9fb96 858 stdout, _, returncode = Popen.run([
f59f5ef8 859 'dbus-send', '--session', '--print-reply=literal',
b38d4c94
MB
860 f'--dest={service_name}',
861 wallet_path,
add96eb9 862 'org.kde.KWallet.networkWallet',
f0c9fb96 863 ], text=True, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
f59f5ef8 864
f0c9fb96 865 if returncode:
f59f5ef8
MB
866 logger.warning('failed to read NetworkWallet')
867 return default_wallet
868 else:
f0c9fb96 869 logger.debug(f'NetworkWallet = "{stdout.strip()}"')
870 return stdout.strip()
a44ca5a4 871 except Exception as e:
86e5f3ed 872 logger.warning(f'exception while obtaining NetworkWallet: {e}')
f59f5ef8
MB
873 return default_wallet
874
875
b38d4c94
MB
876def _get_kwallet_password(browser_keyring_name, keyring, logger):
877 logger.debug(f'using kwallet-query to obtain password from {keyring.name}')
f59f5ef8
MB
878
879 if shutil.which('kwallet-query') is None:
880 logger.error('kwallet-query command not found. KWallet and kwallet-query '
881 'must be installed to read from KWallet. kwallet-query should be'
882 'included in the kwallet package for your distribution')
883 return b''
884
b38d4c94 885 network_wallet = _get_kwallet_network_wallet(keyring, logger)
f59f5ef8
MB
886
887 try:
f0c9fb96 888 stdout, _, returncode = Popen.run([
f59f5ef8 889 'kwallet-query',
86e5f3ed 890 '--read-password', f'{browser_keyring_name} Safe Storage',
891 '--folder', f'{browser_keyring_name} Keys',
add96eb9 892 network_wallet,
f59f5ef8
MB
893 ], stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
894
f0c9fb96 895 if returncode:
896 logger.error(f'kwallet-query failed with return code {returncode}. '
897 'Please consult the kwallet-query man page for details')
f59f5ef8
MB
898 return b''
899 else:
900 if stdout.lower().startswith(b'failed to read'):
901 logger.debug('failed to read password from kwallet. Using empty string instead')
902 # this sometimes occurs in KDE because chrome does not check hasEntry and instead
903 # just tries to read the value (which kwallet returns "") whereas kwallet-query
904 # checks hasEntry. To verify this:
905 # dbus-monitor "interface='org.kde.KWallet'" "type=method_return"
906 # while starting chrome.
b38d4c94
MB
907 # this was identified as a bug later and fixed in
908 # https://chromium.googlesource.com/chromium/src/+/bbd54702284caca1f92d656fdcadf2ccca6f4165%5E%21/#F0
909 # https://chromium.googlesource.com/chromium/src/+/5463af3c39d7f5b6d11db7fbd51e38cc1974d764
f59f5ef8
MB
910 return b''
911 else:
912 logger.debug('password found')
f0c9fb96 913 return stdout.rstrip(b'\n')
a44ca5a4 914 except Exception as e:
915 logger.warning(f'exception running kwallet-query: {error_to_str(e)}')
f59f5ef8
MB
916 return b''
917
918
919def _get_gnome_keyring_password(browser_keyring_name, logger):
9b8ee23b 920 if not secretstorage:
921 logger.error(f'secretstorage not available {_SECRETSTORAGE_UNAVAILABLE_REASON}')
f59f5ef8
MB
922 return b''
923 # the Gnome keyring does not seem to organise keys in the same way as KWallet,
924 # using `dbus-monitor` during startup, it can be observed that chromium lists all keys
925 # and presumably searches for its key in the list. It appears that we must do the same.
926 # https://github.com/jaraco/keyring/issues/556
927 with contextlib.closing(secretstorage.dbus_init()) as con:
928 col = secretstorage.get_default_collection(con)
929 for item in col.get_all_items():
86e5f3ed 930 if item.get_label() == f'{browser_keyring_name} Safe Storage':
f59f5ef8 931 return item.get_secret()
add96eb9 932 logger.error('failed to read from keyring')
933 return b''
f59f5ef8
MB
934
935
936def _get_linux_keyring_password(browser_keyring_name, keyring, logger):
937 # note: chrome/chromium can be run with the following flags to determine which keyring backend
938 # it has chosen to use
939 # chromium --enable-logging=stderr --v=1 2>&1 | grep key_storage_
940 # Chromium supports a flag: --password-store=<basic|gnome|kwallet> so the automatic detection
941 # will not be sufficient in all cases.
942
2c539d49 943 keyring = _LinuxKeyring[keyring] if keyring else _choose_linux_keyring(logger)
f59f5ef8
MB
944 logger.debug(f'Chosen keyring: {keyring.name}')
945
2e023649 946 if keyring in (_LinuxKeyring.KWALLET, _LinuxKeyring.KWALLET5, _LinuxKeyring.KWALLET6):
b38d4c94 947 return _get_kwallet_password(browser_keyring_name, keyring, logger)
2e023649 948 elif keyring == _LinuxKeyring.GNOMEKEYRING:
f59f5ef8 949 return _get_gnome_keyring_password(browser_keyring_name, logger)
2e023649 950 elif keyring == _LinuxKeyring.BASICTEXT:
f59f5ef8
MB
951 # when basic text is chosen, all cookies are stored as v10 (so no keyring password is required)
952 return None
953 assert False, f'Unknown keyring {keyring}'
954
955
956def _get_mac_keyring_password(browser_keyring_name, logger):
957 logger.debug('using find-generic-password to obtain password from OSX keychain')
958 try:
46a5b335 959 stdout, _, returncode = Popen.run(
d3c93ec2 960 ['security', 'find-generic-password',
961 '-w', # write password to stdout
962 '-a', browser_keyring_name, # match 'account'
86e5f3ed 963 '-s', f'{browser_keyring_name} Safe Storage'], # match 'service'
d3c93ec2 964 stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
46a5b335 965 if returncode:
966 logger.warning('find-generic-password failed')
967 return None
f0c9fb96 968 return stdout.rstrip(b'\n')
a44ca5a4 969 except Exception as e:
970 logger.warning(f'exception running find-generic-password: {error_to_str(e)}')
f59f5ef8 971 return None
982ee69a
MB
972
973
974def _get_windows_v10_key(browser_root, logger):
b38d4c94
MB
975 """
976 References:
977 - [1] https://chromium.googlesource.com/chromium/src/+/refs/heads/main/components/os_crypt/sync/os_crypt_win.cc
978 """
cbed249a 979 path = _newest(_find_files(browser_root, 'Local State', logger))
982ee69a
MB
980 if path is None:
981 logger.error('could not find local state file')
982 return None
97ec5bc5 983 logger.debug(f'Found local state file at "{path}"')
86e5f3ed 984 with open(path, encoding='utf8') as f:
982ee69a
MB
985 data = json.load(f)
986 try:
b38d4c94 987 # kOsCryptEncryptedKeyPrefName in [1]
982ee69a
MB
988 base64_key = data['os_crypt']['encrypted_key']
989 except KeyError:
990 logger.error('no encrypted key in Local State')
991 return None
14f25df2 992 encrypted_key = base64.b64decode(base64_key)
b38d4c94 993 # kDPAPIKeyPrefix in [1]
982ee69a
MB
994 prefix = b'DPAPI'
995 if not encrypted_key.startswith(prefix):
996 logger.error('invalid key')
997 return None
998 return _decrypt_windows_dpapi(encrypted_key[len(prefix):], logger)
999
1000
1001def pbkdf2_sha1(password, salt, iterations, key_length):
1002 return pbkdf2_hmac('sha1', password, salt, iterations, key_length)
1003
1004
b38d4c94
MB
1005def _decrypt_aes_cbc_multi(ciphertext, keys, logger, initialization_vector=b' ' * 16):
1006 for key in keys:
1007 plaintext = unpad_pkcs7(aes_cbc_decrypt_bytes(ciphertext, key, initialization_vector))
1008 try:
1009 return plaintext.decode()
1010 except UnicodeDecodeError:
1011 pass
1012 logger.warning('failed to decrypt cookie (AES-CBC) because UTF-8 decoding failed. Possibly the key is wrong?', only_once=True)
1013 return None
982ee69a
MB
1014
1015
1016def _decrypt_aes_gcm(ciphertext, key, nonce, authentication_tag, logger):
982ee69a 1017 try:
09906f55 1018 plaintext = aes_gcm_decrypt_and_verify_bytes(ciphertext, key, authentication_tag, nonce)
982ee69a 1019 except ValueError:
f440b14f 1020 logger.warning('failed to decrypt cookie (AES-GCM) because the MAC check failed. Possibly the key is wrong?', only_once=True)
982ee69a
MB
1021 return None
1022
1023 try:
0f06bcd7 1024 return plaintext.decode()
982ee69a 1025 except UnicodeDecodeError:
f440b14f 1026 logger.warning('failed to decrypt cookie (AES-GCM) because UTF-8 decoding failed. Possibly the key is wrong?', only_once=True)
982ee69a
MB
1027 return None
1028
1029
1030def _decrypt_windows_dpapi(ciphertext, logger):
1031 """
1032 References:
1033 - https://docs.microsoft.com/en-us/windows/win32/api/dpapi/nf-dpapi-cryptunprotectdata
1034 """
fe0918bb 1035
1036 import ctypes
1037 import ctypes.wintypes
982ee69a
MB
1038
1039 class DATA_BLOB(ctypes.Structure):
fe0918bb 1040 _fields_ = [('cbData', ctypes.wintypes.DWORD),
982ee69a
MB
1041 ('pbData', ctypes.POINTER(ctypes.c_char))]
1042
1043 buffer = ctypes.create_string_buffer(ciphertext)
1044 blob_in = DATA_BLOB(ctypes.sizeof(buffer), buffer)
1045 blob_out = DATA_BLOB()
1046 ret = ctypes.windll.crypt32.CryptUnprotectData(
1047 ctypes.byref(blob_in), # pDataIn
1048 None, # ppszDataDescr: human readable description of pDataIn
1049 None, # pOptionalEntropy: salt?
1050 None, # pvReserved: must be NULL
1051 None, # pPromptStruct: information about prompts to display
1052 0, # dwFlags
add96eb9 1053 ctypes.byref(blob_out), # pDataOut
982ee69a
MB
1054 )
1055 if not ret:
f9be9cb9 1056 logger.warning('failed to decrypt with DPAPI', only_once=True)
982ee69a
MB
1057 return None
1058
1059 result = ctypes.string_at(blob_out.pbData, blob_out.cbData)
1060 ctypes.windll.kernel32.LocalFree(blob_out.pbData)
1061 return result
1062
1063
1064def _config_home():
1065 return os.environ.get('XDG_CONFIG_HOME', os.path.expanduser('~/.config'))
1066
1067
1068def _open_database_copy(database_path, tmpdir):
1069 # cannot open sqlite databases if they are already in use (e.g. by the browser)
1070 database_copy_path = os.path.join(tmpdir, 'temporary.sqlite')
1071 shutil.copy(database_path, database_copy_path)
1072 conn = sqlite3.connect(database_copy_path)
1073 return conn.cursor()
1074
1075
1076def _get_column_names(cursor, table_name):
86e5f3ed 1077 table_info = cursor.execute(f'PRAGMA table_info({table_name})').fetchall()
0f06bcd7 1078 return [row[1].decode() for row in table_info]
982ee69a
MB
1079
1080
cbed249a
SS
1081def _newest(files):
1082 return max(files, key=lambda path: os.lstat(path).st_mtime, default=None)
1083
1084
1085def _find_files(root, filename, logger):
982ee69a 1086 # if there are multiple browser profiles, take the most recently used one
cbed249a 1087 i = 0
97ec5bc5 1088 with _create_progress_bar(logger) as progress_bar:
cbed249a 1089 for curr_root, _, files in os.walk(root):
97ec5bc5 1090 for file in files:
1091 i += 1
1092 progress_bar.print(f'Searching for "{filename}": {i: 6d} files searched')
1093 if file == filename:
cbed249a 1094 yield os.path.join(curr_root, file)
982ee69a
MB
1095
1096
1097def _merge_cookie_jars(jars):
1098 output_jar = YoutubeDLCookieJar()
1099 for jar in jars:
1100 for cookie in jar:
1101 output_jar.set_cookie(cookie)
1102 if jar.filename is not None:
1103 output_jar.filename = jar.filename
1104 return output_jar
1105
1106
1107def _is_path(value):
cbed249a 1108 return any(sep in value for sep in (os.path.sep, os.path.altsep) if sep)
982ee69a
MB
1109
1110
9bd13fe5 1111def _parse_browser_specification(browser_name, profile=None, keyring=None, container=None):
982ee69a
MB
1112 if browser_name not in SUPPORTED_BROWSERS:
1113 raise ValueError(f'unsupported browser: "{browser_name}"')
f59f5ef8
MB
1114 if keyring not in (None, *SUPPORTED_KEYRINGS):
1115 raise ValueError(f'unsupported keyring: "{keyring}"')
2b24afa6 1116 if profile is not None and _is_path(expand_path(profile)):
1117 profile = expand_path(profile)
9bd13fe5 1118 return browser_name, profile, keyring, container
8817a80d
SS
1119
1120
1121class LenientSimpleCookie(http.cookies.SimpleCookie):
1122 """More lenient version of http.cookies.SimpleCookie"""
1123 # From https://github.com/python/cpython/blob/v3.10.7/Lib/http/cookies.py
36069409
SS
1124 # We use Morsel's legal key chars to avoid errors on setting values
1125 _LEGAL_KEY_CHARS = r'\w\d' + re.escape('!#$%&\'*+-.:^_`|~')
1126 _LEGAL_VALUE_CHARS = _LEGAL_KEY_CHARS + re.escape('(),/<=>?@[]{}')
8817a80d
SS
1127
1128 _RESERVED = {
add96eb9 1129 'expires',
1130 'path',
1131 'comment',
1132 'domain',
1133 'max-age',
1134 'secure',
1135 'httponly',
1136 'version',
1137 'samesite',
8817a80d
SS
1138 }
1139
add96eb9 1140 _FLAGS = {'secure', 'httponly'}
8817a80d
SS
1141
1142 # Added 'bad' group to catch the remaining value
add96eb9 1143 _COOKIE_PATTERN = re.compile(r'''
8817a80d
SS
1144 \s* # Optional whitespace at start of cookie
1145 (?P<key> # Start of group 'key'
add96eb9 1146 [''' + _LEGAL_KEY_CHARS + r''']+?# Any word of at least one letter
8817a80d
SS
1147 ) # End of group 'key'
1148 ( # Optional group: there may not be a value.
1149 \s*=\s* # Equal Sign
1150 ( # Start of potential value
1151 (?P<val> # Start of group 'val'
1152 "(?:[^\\"]|\\.)*" # Any doublequoted string
1153 | # or
1154 \w{3},\s[\w\d\s-]{9,11}\s[\d:]{8}\sGMT # Special case for "expires" attr
1155 | # or
add96eb9 1156 [''' + _LEGAL_VALUE_CHARS + r''']* # Any word or empty string
8817a80d
SS
1157 ) # End of group 'val'
1158 | # or
1159 (?P<bad>(?:\\;|[^;])*?) # 'bad' group fallback for invalid values
1160 ) # End of potential value
1161 )? # End of optional value group
1162 \s* # Any number of spaces.
1163 (\s+|;|$) # Ending either at space, semicolon, or EOS.
add96eb9 1164 ''', re.ASCII | re.VERBOSE)
8817a80d
SS
1165
1166 def load(self, data):
1167 # Workaround for https://github.com/yt-dlp/yt-dlp/issues/4776
1168 if not isinstance(data, str):
1169 return super().load(data)
1170
1171 morsel = None
36069409
SS
1172 for match in self._COOKIE_PATTERN.finditer(data):
1173 if match.group('bad'):
8817a80d
SS
1174 morsel = None
1175 continue
1176
36069409 1177 key, value = match.group('key', 'val')
8817a80d 1178
36069409
SS
1179 is_attribute = False
1180 if key.startswith('$'):
1181 key = key[1:]
1182 is_attribute = True
8817a80d
SS
1183
1184 lower_key = key.lower()
1185 if lower_key in self._RESERVED:
1186 if morsel is None:
1187 continue
1188
1189 if value is None:
1190 if lower_key not in self._FLAGS:
1191 morsel = None
1192 continue
1193 value = True
1194 else:
1195 value, _ = self.value_decode(value)
1196
1197 morsel[key] = value
1198
36069409
SS
1199 elif is_attribute:
1200 morsel = None
1201
8817a80d
SS
1202 elif value is not None:
1203 morsel = self.get(key, http.cookies.Morsel())
1204 real_value, coded_value = self.value_decode(value)
1205 morsel.set(key, real_value, coded_value)
1206 self[key] = morsel
1207
1208 else:
1209 morsel = None
b87e01c1 1210
1211
1212class YoutubeDLCookieJar(http.cookiejar.MozillaCookieJar):
1213 """
1214 See [1] for cookie file format.
1215
1216 1. https://curl.haxx.se/docs/http-cookies.html
1217 """
1218 _HTTPONLY_PREFIX = '#HttpOnly_'
1219 _ENTRY_LEN = 7
1220 _HEADER = '''# Netscape HTTP Cookie File
1221# This file is generated by yt-dlp. Do not edit.
1222
1223'''
1224 _CookieFileEntry = collections.namedtuple(
1225 'CookieFileEntry',
1226 ('domain_name', 'include_subdomains', 'path', 'https_only', 'expires_at', 'name', 'value'))
1227
1228 def __init__(self, filename=None, *args, **kwargs):
1229 super().__init__(None, *args, **kwargs)
1230 if is_path_like(filename):
1231 filename = os.fspath(filename)
1232 self.filename = filename
1233
1234 @staticmethod
1235 def _true_or_false(cndn):
1236 return 'TRUE' if cndn else 'FALSE'
1237
1238 @contextlib.contextmanager
1239 def open(self, file, *, write=False):
1240 if is_path_like(file):
1241 with open(file, 'w' if write else 'r', encoding='utf-8') as f:
1242 yield f
1243 else:
1244 if write:
1245 file.truncate(0)
1246 yield file
1247
62b5c94c 1248 def _really_save(self, f, ignore_discard, ignore_expires):
b87e01c1 1249 now = time.time()
1250 for cookie in self:
1251 if (not ignore_discard and cookie.discard
1252 or not ignore_expires and cookie.is_expired(now)):
1253 continue
1254 name, value = cookie.name, cookie.value
1255 if value is None:
1256 # cookies.txt regards 'Set-Cookie: foo' as a cookie
1257 # with no name, whereas http.cookiejar regards it as a
1258 # cookie with no value.
1259 name, value = '', name
add96eb9 1260 f.write('{}\n'.format('\t'.join((
b87e01c1 1261 cookie.domain,
1262 self._true_or_false(cookie.domain.startswith('.')),
1263 cookie.path,
1264 self._true_or_false(cookie.secure),
1265 str_or_none(cookie.expires, default=''),
add96eb9 1266 name, value,
1267 ))))
b87e01c1 1268
62b5c94c 1269 def save(self, filename=None, ignore_discard=True, ignore_expires=True):
b87e01c1 1270 """
1271 Save cookies to a file.
1272 Code is taken from CPython 3.6
1273 https://github.com/python/cpython/blob/8d999cbf4adea053be6dbb612b9844635c4dfb8e/Lib/http/cookiejar.py#L2091-L2117 """
1274
1275 if filename is None:
1276 if self.filename is not None:
1277 filename = self.filename
1278 else:
1279 raise ValueError(http.cookiejar.MISSING_FILENAME_TEXT)
1280
1281 # Store session cookies with `expires` set to 0 instead of an empty string
1282 for cookie in self:
1283 if cookie.expires is None:
1284 cookie.expires = 0
1285
1286 with self.open(filename, write=True) as f:
1287 f.write(self._HEADER)
62b5c94c 1288 self._really_save(f, ignore_discard, ignore_expires)
b87e01c1 1289
62b5c94c 1290 def load(self, filename=None, ignore_discard=True, ignore_expires=True):
b87e01c1 1291 """Load cookies from a file."""
1292 if filename is None:
1293 if self.filename is not None:
1294 filename = self.filename
1295 else:
1296 raise ValueError(http.cookiejar.MISSING_FILENAME_TEXT)
1297
1298 def prepare_line(line):
1299 if line.startswith(self._HTTPONLY_PREFIX):
1300 line = line[len(self._HTTPONLY_PREFIX):]
1301 # comments and empty lines are fine
1302 if line.startswith('#') or not line.strip():
1303 return line
1304 cookie_list = line.split('\t')
1305 if len(cookie_list) != self._ENTRY_LEN:
add96eb9 1306 raise http.cookiejar.LoadError(f'invalid length {len(cookie_list)}')
b87e01c1 1307 cookie = self._CookieFileEntry(*cookie_list)
1308 if cookie.expires_at and not cookie.expires_at.isdigit():
add96eb9 1309 raise http.cookiejar.LoadError(f'invalid expires at {cookie.expires_at}')
b87e01c1 1310 return line
1311
1312 cf = io.StringIO()
1313 with self.open(filename) as f:
1314 for line in f:
1315 try:
1316 cf.write(prepare_line(line))
1317 except http.cookiejar.LoadError as e:
1318 if f'{line.strip()} '[0] in '[{"':
1319 raise http.cookiejar.LoadError(
1320 'Cookies file must be Netscape formatted, not JSON. See '
1321 'https://github.com/yt-dlp/yt-dlp/wiki/FAQ#how-do-i-pass-cookies-to-yt-dlp')
1322 write_string(f'WARNING: skipping cookie file entry due to {e}: {line!r}\n')
1323 continue
1324 cf.seek(0)
1325 self._really_load(cf, filename, ignore_discard, ignore_expires)
1326 # Session cookies are denoted by either `expires` field set to
1327 # an empty string or 0. MozillaCookieJar only recognizes the former
1328 # (see [1]). So we need force the latter to be recognized as session
1329 # cookies on our own.
1330 # Session cookies may be important for cookies-based authentication,
1331 # e.g. usually, when user does not check 'Remember me' check box while
1332 # logging in on a site, some important cookies are stored as session
1333 # cookies so that not recognizing them will result in failed login.
1334 # 1. https://bugs.python.org/issue17164
1335 for cookie in self:
1336 # Treat `expires=0` cookies as session cookies
1337 if cookie.expires == 0:
1338 cookie.expires = None
1339 cookie.discard = True
1340
1341 def get_cookie_header(self, url):
1342 """Generate a Cookie HTTP header for a given url"""
4bf91228 1343 cookie_req = urllib.request.Request(normalize_url(sanitize_url(url)))
b87e01c1 1344 self.add_cookie_header(cookie_req)
1345 return cookie_req.get_header('Cookie')
ad54c913 1346
1ceb657b 1347 def get_cookies_for_url(self, url):
1348 """Generate a list of Cookie objects for a given url"""
1349 # Policy `_now` attribute must be set before calling `_cookies_for_request`
1350 # Ref: https://github.com/python/cpython/blob/3.7/Lib/http/cookiejar.py#L1360
1351 self._policy._now = self._now = int(time.time())
4bf91228 1352 return self._cookies_for_request(urllib.request.Request(normalize_url(sanitize_url(url))))
1ceb657b 1353
ad54c913 1354 def clear(self, *args, **kwargs):
1355 with contextlib.suppress(KeyError):
1356 return super().clear(*args, **kwargs)