123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287 |
- <?php
- namespace PhpParser;
- class Lexer
- {
- protected $code;
- protected $tokens;
- protected $pos;
- protected $line;
- protected $filePos;
- protected $tokenMap;
- protected $dropTokens;
- protected $usedAttributes;
- /**
- * Creates a Lexer.
- *
- * @param array $options Options array. Currently only the 'usedAttributes' option is supported,
- * which is an array of attributes to add to the AST nodes. Possible attributes
- * are: 'comments', 'startLine', 'endLine', 'startTokenPos', 'endTokenPos',
- * 'startFilePos', 'endFilePos'. The option defaults to the first three.
- * For more info see getNextToken() docs.
- */
- public function __construct(array $options = array()) {
- // map from internal tokens to PhpParser tokens
- $this->tokenMap = $this->createTokenMap();
- // map of tokens to drop while lexing (the map is only used for isset lookup,
- // that's why the value is simply set to 1; the value is never actually used.)
- $this->dropTokens = array_fill_keys(array(T_WHITESPACE, T_OPEN_TAG), 1);
- // the usedAttributes member is a map of the used attribute names to a dummy
- // value (here "true")
- $options += array(
- 'usedAttributes' => array('comments', 'startLine', 'endLine'),
- );
- $this->usedAttributes = array_fill_keys($options['usedAttributes'], true);
- }
- /**
- * Initializes the lexer for lexing the provided source code.
- *
- * @param string $code The source code to lex
- *
- * @throws Error on lexing errors (unterminated comment or unexpected character)
- */
- public function startLexing($code) {
- $scream = ini_set('xdebug.scream', '0');
- $this->resetErrors();
- $this->tokens = @token_get_all($code);
- $this->handleErrors();
- if (false !== $scream) {
- ini_set('xdebug.scream', $scream);
- }
- $this->code = $code; // keep the code around for __halt_compiler() handling
- $this->pos = -1;
- $this->line = 1;
- $this->filePos = 0;
- }
- protected function resetErrors() {
- // set error_get_last() to defined state by forcing an undefined variable error
- set_error_handler(function() { return false; }, 0);
- @$undefinedVariable;
- restore_error_handler();
- }
- protected function handleErrors() {
- $error = error_get_last();
- if (preg_match(
- '~^Unterminated comment starting line ([0-9]+)$~',
- $error['message'], $matches
- )) {
- throw new Error('Unterminated comment', (int) $matches[1]);
- }
- if (preg_match(
- '~^Unexpected character in input: \'(.)\' \(ASCII=([0-9]+)\)~s',
- $error['message'], $matches
- )) {
- throw new Error(sprintf(
- 'Unexpected character "%s" (ASCII %d)',
- $matches[1], $matches[2]
- ));
- }
- // PHP cuts error message after null byte, so need special case
- if (preg_match('~^Unexpected character in input: \'$~', $error['message'])) {
- throw new Error('Unexpected null byte');
- }
- }
- /**
- * Fetches the next token.
- *
- * The available attributes are determined by the 'usedAttributes' option, which can
- * be specified in the constructor. The following attributes are supported:
- *
- * * 'comments' => Array of PhpParser\Comment or PhpParser\Comment\Doc instances,
- * representing all comments that occurred between the previous
- * non-discarded token and the current one.
- * * 'startLine' => Line in which the node starts.
- * * 'endLine' => Line in which the node ends.
- * * 'startTokenPos' => Offset into the token array of the first token in the node.
- * * 'endTokenPos' => Offset into the token array of the last token in the node.
- * * 'startFilePos' => Offset into the code string of the first character that is part of the node.
- * * 'endFilePos' => Offset into the code string of the last character that is part of the node
- *
- * @param mixed $value Variable to store token content in
- * @param mixed $startAttributes Variable to store start attributes in
- * @param mixed $endAttributes Variable to store end attributes in
- *
- * @return int Token id
- */
- public function getNextToken(&$value = null, &$startAttributes = null, &$endAttributes = null) {
- $startAttributes = array();
- $endAttributes = array();
- while (1) {
- if (isset($this->tokens[++$this->pos])) {
- $token = $this->tokens[$this->pos];
- } else {
- // EOF token with ID 0
- $token = "\0";
- }
- if (isset($this->usedAttributes['startTokenPos'])) {
- $startAttributes['startTokenPos'] = $this->pos;
- }
- if (isset($this->usedAttributes['startFilePos'])) {
- $startAttributes['startFilePos'] = $this->filePos;
- }
- if (is_string($token)) {
- // bug in token_get_all
- if ('b"' === $token) {
- $value = 'b"';
- $this->filePos += 2;
- $id = ord('"');
- } else {
- $value = $token;
- $this->filePos += 1;
- $id = ord($token);
- }
- if (isset($this->usedAttributes['startLine'])) {
- $startAttributes['startLine'] = $this->line;
- }
- if (isset($this->usedAttributes['endLine'])) {
- $endAttributes['endLine'] = $this->line;
- }
- if (isset($this->usedAttributes['endTokenPos'])) {
- $endAttributes['endTokenPos'] = $this->pos;
- }
- if (isset($this->usedAttributes['endFilePos'])) {
- $endAttributes['endFilePos'] = $this->filePos - 1;
- }
- return $id;
- } else {
- $this->line += substr_count($token[1], "\n");
- $this->filePos += strlen($token[1]);
- if (T_COMMENT === $token[0]) {
- if (isset($this->usedAttributes['comments'])) {
- $startAttributes['comments'][] = new Comment($token[1], $token[2]);
- }
- } elseif (T_DOC_COMMENT === $token[0]) {
- if (isset($this->usedAttributes['comments'])) {
- $startAttributes['comments'][] = new Comment\Doc($token[1], $token[2]);
- }
- } elseif (!isset($this->dropTokens[$token[0]])) {
- $value = $token[1];
- if (isset($this->usedAttributes['startLine'])) {
- $startAttributes['startLine'] = $token[2];
- }
- if (isset($this->usedAttributes['endLine'])) {
- $endAttributes['endLine'] = $this->line;
- }
- if (isset($this->usedAttributes['endTokenPos'])) {
- $endAttributes['endTokenPos'] = $this->pos;
- }
- if (isset($this->usedAttributes['endFilePos'])) {
- $endAttributes['endFilePos'] = $this->filePos - 1;
- }
- return $this->tokenMap[$token[0]];
- }
- }
- }
- throw new \RuntimeException('Reached end of lexer loop');
- }
- /**
- * Returns the token array for current code.
- *
- * The token array is in the same format as provided by the
- * token_get_all() function and does not discard tokens (i.e.
- * whitespace and comments are included). The token position
- * attributes are against this token array.
- *
- * @return array Array of tokens in token_get_all() format
- */
- public function getTokens() {
- return $this->tokens;
- }
- /**
- * Handles __halt_compiler() by returning the text after it.
- *
- * @return string Remaining text
- */
- public function handleHaltCompiler() {
- // text after T_HALT_COMPILER, still including ();
- $textAfter = substr($this->code, $this->filePos);
- // ensure that it is followed by ();
- // this simplifies the situation, by not allowing any comments
- // in between of the tokens.
- if (!preg_match('~^\s*\(\s*\)\s*(?:;|\?>\r?\n?)~', $textAfter, $matches)) {
- throw new Error('__HALT_COMPILER must be followed by "();"');
- }
- // prevent the lexer from returning any further tokens
- $this->pos = count($this->tokens);
- // return with (); removed
- return (string) substr($textAfter, strlen($matches[0])); // (string) converts false to ''
- }
- /**
- * Creates the token map.
- *
- * The token map maps the PHP internal token identifiers
- * to the identifiers used by the Parser. Additionally it
- * maps T_OPEN_TAG_WITH_ECHO to T_ECHO and T_CLOSE_TAG to ';'.
- *
- * @return array The token map
- */
- protected function createTokenMap() {
- $tokenMap = array();
- // 256 is the minimum possible token number, as everything below
- // it is an ASCII value
- for ($i = 256; $i < 1000; ++$i) {
- if (T_DOUBLE_COLON === $i) {
- // T_DOUBLE_COLON is equivalent to T_PAAMAYIM_NEKUDOTAYIM
- $tokenMap[$i] = Parser::T_PAAMAYIM_NEKUDOTAYIM;
- } elseif(T_OPEN_TAG_WITH_ECHO === $i) {
- // T_OPEN_TAG_WITH_ECHO with dropped T_OPEN_TAG results in T_ECHO
- $tokenMap[$i] = Parser::T_ECHO;
- } elseif(T_CLOSE_TAG === $i) {
- // T_CLOSE_TAG is equivalent to ';'
- $tokenMap[$i] = ord(';');
- } elseif ('UNKNOWN' !== $name = token_name($i)) {
- if ('T_HASHBANG' === $name) {
- // HHVM uses a special token for #! hashbang lines
- $tokenMap[$i] = Parser::T_INLINE_HTML;
- } else if (defined($name = 'PhpParser\Parser::' . $name)) {
- // Other tokens can be mapped directly
- $tokenMap[$i] = constant($name);
- }
- }
- }
- // HHVM uses a special token for numbers that overflow to double
- if (defined('T_ONUMBER')) {
- $tokenMap[T_ONUMBER] = Parser::T_DNUMBER;
- }
- // HHVM also has a separate token for the __COMPILER_HALT_OFFSET__ constant
- if (defined('T_COMPILER_HALT_OFFSET')) {
- $tokenMap[T_COMPILER_HALT_OFFSET] = Parser::T_STRING;
- }
- return $tokenMap;
- }
- }
|