1
0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2025-02-13 09:02:43 +01:00

[JSInterp] Fix bit-shift coercion for player 9c6dfc4a

This commit is contained in:
dirkf 2025-02-06 21:09:00 +00:00
parent 26b6f15d14
commit 711e72c292
4 changed files with 34 additions and 6 deletions

View File

@ -459,6 +459,10 @@ class TestJSInterpreter(unittest.TestCase):
self._test('function f(){return undefined >> 5}', 0)
self._test('function f(){return 42 << NaN}', 42)
self._test('function f(){return 42 << Infinity}', 42)
self._test('function f(){return 0.0 << null}', 0)
self._test('function f(){return NaN << 42}', 0)
self._test('function f(){return "21.9" << 1}', 42)
self._test('function f(){return 21 << 4294967297}', 42)
def test_negative(self):
self._test('function f(){return 2 * -2.0 ;}', -4)

View File

@ -219,6 +219,10 @@ _NSIG_TESTS = [
'https://www.youtube.com/s/player/2f1832d2/player_ias.vflset/en_US/base.js',
'YWt1qdbe8SAfkoPHW5d', 'RrRjWQOJmBiP',
),
(
'https://www.youtube.com/s/player/9c6dfc4a/player_ias.vflset/en_US/base.js',
'jbu7ylIosQHyJyJV', 'uwI0ESiynAmhNg',
),
]

View File

@ -3116,17 +3116,21 @@ else:
compat_kwargs = lambda kwargs: kwargs
# compat_numeric_types
try:
compat_numeric_types = (int, float, long, complex)
except NameError: # Python 3
compat_numeric_types = (int, float, complex)
# compat_integer_types
try:
compat_integer_types = (int, long)
except NameError: # Python 3
compat_integer_types = (int, )
# compat_int
compat_int = compat_integer_types[-1]
if sys.version_info < (2, 7):
def compat_socket_create_connection(address, timeout, source_address=None):
@ -3532,6 +3536,7 @@ __all__ = [
'compat_http_client',
'compat_http_server',
'compat_input',
'compat_int',
'compat_integer_types',
'compat_itertools_count',
'compat_itertools_zip_longest',

View File

@ -24,6 +24,8 @@ from .compat import (
compat_collections_chain_map as ChainMap,
compat_contextlib_suppress,
compat_filter as filter,
compat_int,
compat_integer_types,
compat_itertools_zip_longest as zip_longest,
compat_map as map,
compat_numeric_types,
@ -70,14 +72,27 @@ class JS_Undefined(object):
pass
def _js_bit_op(op):
def _js_bit_op(op, is_shift=False):
def zeroise(x):
return 0 if x in (None, JS_Undefined, _NaN, _Infinity) else x
def zeroise(x, is_shift_arg=False):
if isinstance(x, compat_integer_types):
return (x % 32) if is_shift_arg else (x & 0xffffffff)
try:
x = float(x)
if is_shift_arg:
x = int(x % 32)
elif x < 0:
x = -compat_int(-x % 0xffffffff)
else:
x = compat_int(x % 0xffffffff)
except (ValueError, TypeError):
# also here for int(NaN), including float('inf') % 32
x = 0
return x
@wraps_op(op)
def wrapped(a, b):
return op(zeroise(a), zeroise(b)) & 0xffffffff
return op(zeroise(a), zeroise(b, is_shift)) & 0xffffffff
return wrapped
@ -253,8 +268,8 @@ def _js_typeof(expr):
# avoid dict to maintain order
# definition None => Defined in JSInterpreter._operator
_OPERATORS = (
('>>', _js_bit_op(operator.rshift)),
('<<', _js_bit_op(operator.lshift)),
('>>', _js_bit_op(operator.rshift, True)),
('<<', _js_bit_op(operator.lshift, True)),
('+', _js_add),
('-', _js_arith_op(operator.sub)),
('*', _js_arith_op(operator.mul)),