R-yan/l1ving_youtube-dl
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[jsinterp] Fixing TokenStrem pop, label statement, function body

Browse Source
This commit is contained in:
sulyi 2016-12-11 23:30:03 +01:00
parent f24cafea89
commit a8a445f04c
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
youtube_dl/jsinterp/jsinterp.py
View File

@ -292,14 +292,11 @@ class JSInterpreter(object):
raise ExtractorError('Unexpected sequence %s at %d' % (peek_value, peek_pos))
else:
token_stream.pop()
# label
else:
token_stream.chk_id()
token_id, label_name, token_pos = token_stream.pop()
token_id, token_value, token_pos = token_stream.pop()
if token_id is not Token.COLON:
raise ExtractorError('''Label statement missing ':' at %d''' % token_pos)
token_id, token_value, token_pos = token_stream.peek(2)
if token_id is Token.COLON:
token_id, label_name, token_pos = token_stream.pop(2)
token_stream.chk_id(last=True)
statement = (Token.LABEL, label_name, self._next_statement(token_stream, stack_top - 1))
# expr
@ -520,11 +517,12 @@ class JSInterpreter(object):
elif not is_expr:
raise ExtractorError('Function declaration at %d is missing identifier' % token_pos)
if token_id is Token.POPEN:
open_pos = token_pos
else:
if token_id is not Token.POPEN:
raise ExtractorError('Expected argument list at %d' % token_pos)
token_stream.pop()
open_pos = token_pos
args = []
while True:
token_id, token_value, token_pos = token_stream.peek()

10
youtube_dl/jsinterp/tstream.py
View File

@ -159,10 +159,12 @@ class TokenStream(object):
return self.peeked[count - 1]
def pop(self, count=1):
if not self.peeked:
self.peek()
for _ in range(count):
self._last = self.peeked.pop()
if count > len(self.peeked):
self.peek(count)
self.flush()
else:
self._last = self.peeked[count - 1]
self.peeked = self.peeked[count:]
return self._last
def flush(self):