[jsinterp] Fix escape in regex

This commit is contained in:
pukkandan 2022-09-01 13:14:04 +05:30
parent b505e8517a
commit 05deb747bb
No known key found for this signature in database
GPG key ID: 7EEE9E1E817D0A39
4 changed files with 16 additions and 6 deletions

View file

@ -352,6 +352,11 @@ class TestJSInterpreter(unittest.TestCase):
''') ''')
self.assertEqual(jsi.call_function('x').flags & re.I, re.I) self.assertEqual(jsi.call_function('x').flags & re.I, re.I)
jsi = JSInterpreter('''
function x() { let a=/,][}",],()}(\[)/; return a; }
''')
self.assertEqual(jsi.call_function('x').pattern, r',][}",],()}(\[)')
def test_char_code_at(self): def test_char_code_at(self):
jsi = JSInterpreter('function x(i){return "test".charCodeAt(i)}') jsi = JSInterpreter('function x(i){return "test".charCodeAt(i)}')
self.assertEqual(jsi.call_function('x', 0), 116) self.assertEqual(jsi.call_function('x', 0), 116)

View file

@ -122,6 +122,10 @@ _NSIG_TESTS = [
'https://www.youtube.com/s/player/113ca41c/player_ias.vflset/en_US/base.js', 'https://www.youtube.com/s/player/113ca41c/player_ias.vflset/en_US/base.js',
'cgYl-tlYkhjT7A', 'hI7BBr2zUgcmMg', 'cgYl-tlYkhjT7A', 'hI7BBr2zUgcmMg',
), ),
(
'https://www.youtube.com/s/player/c57c113c/player_ias.vflset/en_US/base.js',
'M92UUMHa8PdvPd3wyM', '3hPqLJsiNZx7yA',
),
] ]

View file

@ -2702,7 +2702,7 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
def _extract_n_function_code(self, video_id, player_url): def _extract_n_function_code(self, video_id, player_url):
player_id = self._extract_player_info(player_url) player_id = self._extract_player_info(player_url)
func_code = self.cache.load('youtube-nsig', player_id, min_ver='2022.08.19.2') func_code = self.cache.load('youtube-nsig', player_id, min_ver='2022.09.1')
jscode = func_code or self._load_player(video_id, player_url) jscode = func_code or self._load_player(video_id, player_url)
jsi = JSInterpreter(jscode) jsi = JSInterpreter(jscode)

View file

@ -245,11 +245,12 @@ class JSInterpreter:
counters[_MATCHING_PARENS[char]] += 1 counters[_MATCHING_PARENS[char]] += 1
elif not in_quote and char in counters: elif not in_quote and char in counters:
counters[char] -= 1 counters[char] -= 1
elif not escaping and char in _QUOTES and in_quote in (char, None): elif not escaping:
if in_quote or after_op or char != '/': if char in _QUOTES and in_quote in (char, None):
in_quote = None if in_quote and not in_regex_char_group else char if in_quote or after_op or char != '/':
elif in_quote == '/' and char in '[]': in_quote = None if in_quote and not in_regex_char_group else char
in_regex_char_group = char == '[' elif in_quote == '/' and char in '[]':
in_regex_char_group = char == '['
escaping = not escaping and in_quote and char == '\\' escaping = not escaping and in_quote and char == '\\'
after_op = not in_quote and char in OP_CHARS or (char.isspace() and after_op) after_op = not in_quote and char in OP_CHARS or (char.isspace() and after_op)