root/readline/readline_cli.c

/* [<][>][^][v][top][bottom][index][help] */

DEFINITIONS

This source file includes following definitions.
  1. readline_shell_write
  2. readline_shell_ub_write
  3. cli_readline_init_globals
  4. PHP_INI_BEGIN
  5. cli_get_prompt
  6. cli_is_valid_code
  7. cli_completion_generator_ht
  8. cli_completion_generator_var
  9. cli_completion_generator_ini
  10. cli_completion_generator_func
  11. cli_completion_generator_class
  12. cli_completion_generator_define
  13. cli_completion_generator
  14. cli_code_completion
  15. readline_shell_run
  16. PHP_MINIT_FUNCTION
  17. PHP_MSHUTDOWN_FUNCTION
  18. PHP_MINFO_FUNCTION

/*
   +----------------------------------------------------------------------+
   | PHP Version 5                                                        |
   +----------------------------------------------------------------------+
   | Copyright (c) 1997-2013 The PHP Group                                |
   +----------------------------------------------------------------------+
   | This source file is subject to version 3.01 of the PHP license,      |
   | that is bundled with this package in the file LICENSE, and is        |
   | available through the world-wide-web at the following url:           |
   | http://www.php.net/license/3_01.txt                                  |
   | If you did not receive a copy of the PHP license and are unable to   |
   | obtain it through the world-wide-web, please send a note to          |
   | license@php.net so we can mail you a copy immediately.               |
   +----------------------------------------------------------------------+
   | Author: Marcus Boerger <helly@php.net>                               |
   |         Johannes Schlueter <johannes@php.net>                        |
   +----------------------------------------------------------------------+
*/

/* $Id$ */

#include "php.h"

#ifndef HAVE_RL_COMPLETION_MATCHES
#define rl_completion_matches completion_matches
#endif

#include "php_globals.h"
#include "php_variables.h"
#include "zend_hash.h"
#include "zend_modules.h"

#include "SAPI.h"

#if HAVE_SETLOCALE
#include <locale.h>
#endif
#include "zend.h"
#include "zend_extensions.h"
#include "php_ini.h"
#include "php_globals.h"
#include "php_main.h"
#include "fopen_wrappers.h"
#include "ext/standard/php_standard.h"
#include "ext/standard/php_smart_str.h"

#ifdef __riscos__
#include <unixlib/local.h>
#endif

#if HAVE_LIBEDIT
#include <editline/readline.h>
#else
#include <readline/readline.h>
#include <readline/history.h>
#endif

#include "zend_compile.h"
#include "zend_execute.h"
#include "zend_highlight.h"
#include "zend_indent.h"
#include "zend_exceptions.h"

#include "sapi/cli/cli.h"
#include "readline_cli.h"

#ifdef COMPILE_DL_READLINE
#include <dlfcn.h>
#endif

#ifndef RTLD_DEFAULT
#define RTLD_DEFAULT NULL
#endif

#define DEFAULT_PROMPT "\\b \\> "

ZEND_DECLARE_MODULE_GLOBALS(cli_readline);

static char php_last_char = '\0';
static FILE *pager_pipe = NULL;

static size_t readline_shell_write(const char *str, uint str_length TSRMLS_DC) /* {{{ */
{
        if (CLIR_G(prompt_str)) {
                smart_str_appendl(CLIR_G(prompt_str), str, str_length);
                return str_length;
        }

        if (CLIR_G(pager) && *CLIR_G(pager) && !pager_pipe) {
                pager_pipe = VCWD_POPEN(CLIR_G(pager), "w");
        }
        if (pager_pipe) {
                return fwrite(str, 1, MIN(str_length, 16384), pager_pipe);
        }

        return -1;
}
/* }}} */

static int readline_shell_ub_write(const char *str, uint str_length TSRMLS_DC) /* {{{ */
{
        php_last_char = str[str_length-1];
        return -1;
}
/* }}} */

static void cli_readline_init_globals(zend_cli_readline_globals *rg TSRMLS_DC)
{
        rg->pager = NULL;
        rg->prompt = NULL;
        rg->prompt_str = NULL;
}

PHP_INI_BEGIN()
        STD_PHP_INI_ENTRY("cli.pager", "", PHP_INI_ALL, OnUpdateString, pager, zend_cli_readline_globals, cli_readline_globals)
        STD_PHP_INI_ENTRY("cli.prompt", DEFAULT_PROMPT, PHP_INI_ALL, OnUpdateString, prompt, zend_cli_readline_globals, cli_readline_globals)
PHP_INI_END()



typedef enum {
        body,
        sstring,
        dstring,
        sstring_esc,
        dstring_esc,
        comment_line,
        comment_block,
        heredoc_start,
        heredoc,
        outside,
} php_code_type;

static char *cli_get_prompt(char *block, char prompt TSRMLS_DC) /* {{{ */
{
        smart_str retval = {0};
        char *prompt_spec = CLIR_G(prompt) ? CLIR_G(prompt) : DEFAULT_PROMPT;

        do {
                if (*prompt_spec == '\\') {
                        switch (prompt_spec[1]) {
                        case '\\':
                                smart_str_appendc(&retval, '\\');
                                prompt_spec++;
                                break;
                        case 'n':
                                smart_str_appendc(&retval, '\n');
                                prompt_spec++;
                                break;
                        case 't':
                                smart_str_appendc(&retval, '\t');
                                prompt_spec++;
                                break;
                        case 'e':
                                smart_str_appendc(&retval, '\033');
                                prompt_spec++;
                                break;


                        case 'v':
                                smart_str_appends(&retval, PHP_VERSION);
                                prompt_spec++;
                                break;
                        case 'b':
                                smart_str_appends(&retval, block);
                                prompt_spec++;
                                break;
                        case '>':
                                smart_str_appendc(&retval, prompt);
                                prompt_spec++;
                                break;
                        case '`':
                                smart_str_appendc(&retval, '`');
                                prompt_spec++;
                                break;
                        default:
                                smart_str_appendc(&retval, '\\');
                                break;
                        }
                } else if (*prompt_spec == '`') {
                        char *prompt_end = strstr(prompt_spec + 1, "`");
                        char *code;

                        if (prompt_end) {
                                code = estrndup(prompt_spec + 1, prompt_end - prompt_spec - 1);

                                CLIR_G(prompt_str) = &retval;
                                zend_try {
                                        zend_eval_stringl(code, prompt_end - prompt_spec - 1, NULL, "php prompt code" TSRMLS_CC);
                                } zend_end_try();
                                CLIR_G(prompt_str) = NULL;
                                efree(code);
                                prompt_spec = prompt_end;
                        }
                } else {
                        smart_str_appendc(&retval, *prompt_spec);
                }
        } while (++prompt_spec && *prompt_spec);
        smart_str_0(&retval);   
        return retval.c;
}
/* }}} */

static int cli_is_valid_code(char *code, int len, char **prompt TSRMLS_DC) /* {{{ */
{
        int valid_end = 1, last_valid_end;
        int brackets_count = 0;
        int brace_count = 0;
        int i;
        php_code_type code_type = body;
        char *heredoc_tag;
        int heredoc_len;

        for (i = 0; i < len; ++i) {
                switch(code_type) {
                        default:
                                switch(code[i]) {
                                        case '{':
                                                brackets_count++;
                                                valid_end = 0;
                                                break;
                                        case '}':
                                                if (brackets_count > 0) {
                                                        brackets_count--;
                                                }
                                                valid_end = brackets_count ? 0 : 1;
                                                break;
                                        case '(':
                                                brace_count++;
                                                valid_end = 0;
                                                break;
                                        case ')':
                                                if (brace_count > 0) {
                                                        brace_count--;
                                                }
                                                valid_end = 0;
                                                break;
                                        case ';':
                                                valid_end = brace_count == 0 && brackets_count == 0;
                                                break;
                                        case ' ':
                                        case '\r':
                                        case '\n':
                                        case '\t':
                                                break;
                                        case '\'':
                                                code_type = sstring;
                                                break;
                                        case '"':
                                                code_type = dstring;
                                                break;
                                        case '#':
                                                code_type = comment_line;
                                                break;
                                        case '/':
                                                if (code[i+1] == '/') {
                                                        i++;
                                                        code_type = comment_line;
                                                        break;
                                                }
                                                if (code[i+1] == '*') {
                                                        last_valid_end = valid_end;
                                                        valid_end = 0;
                                                        code_type = comment_block;
                                                        i++;
                                                        break;
                                                }
                                                valid_end = 0;
                                                break;
                                        case '%':
                                                if (!CG(asp_tags)) {
                                                        valid_end = 0;
                                                        break;
                                                }
                                                /* no break */
                                        case '?':
                                                if (code[i+1] == '>') {
                                                        i++;
                                                        code_type = outside;
                                                        break;
                                                }
                                                valid_end = 0;
                                                break;
                                        case '<':
                                                valid_end = 0;
                                                if (i + 2 < len && code[i+1] == '<' && code[i+2] == '<') {
                                                        i += 2;
                                                        code_type = heredoc_start;
                                                        heredoc_len = 0;
                                                }
                                                break;
                                        default:
                                                valid_end = 0;
                                                break;
                                }
                                break;
                        case sstring:
                                if (code[i] == '\\') {
                                        code_type = sstring_esc;
                                } else {
                                        if (code[i] == '\'') {
                                                code_type = body;
                                        }
                                }
                                break;
                        case sstring_esc:
                                code_type = sstring;
                                break;
                        case dstring:
                                if (code[i] == '\\') {
                                        code_type = dstring_esc;
                                } else {
                                        if (code[i] == '"') {
                                                code_type = body;
                                        }
                                }
                                break;
                        case dstring_esc:
                                code_type = dstring;
                                break;
                        case comment_line:
                                if (code[i] == '\n') {
                                        code_type = body;
                                }
                                break;
                        case comment_block:
                                if (code[i-1] == '*' && code[i] == '/') {
                                        code_type = body;
                                        valid_end = last_valid_end;
                                }
                                break;
                        case heredoc_start:
                                switch(code[i]) {
                                        case ' ':
                                        case '\t':
                                        case '\'':
                                                break;
                                        case '\r':
                                        case '\n':
                                                code_type = heredoc;
                                                break;
                                        default:
                                                if (!heredoc_len) {
                                                        heredoc_tag = code+i;
                                                }
                                                heredoc_len++;
                                                break;
                                }
                                break;
                        case heredoc:
                                if (code[i - (heredoc_len + 1)] == '\n' && !strncmp(code + i - heredoc_len, heredoc_tag, heredoc_len) && code[i] == '\n') {
                                        code_type = body;
                                } else if (code[i - (heredoc_len + 2)] == '\n' && !strncmp(code + i - heredoc_len - 1, heredoc_tag, heredoc_len) && code[i-1] == ';' && code[i] == '\n') {
                                        code_type = body;
                                        valid_end = 1;
                                }
                                break;
                        case outside:
                                if ((CG(short_tags) && !strncmp(code+i-1, "<?", 2))
                                ||  (CG(asp_tags) && !strncmp(code+i-1, "<%", 2))
                                ||  (i > 3 && !strncmp(code+i-4, "<?php", 5))
                                ) {
                                        code_type = body;
                                }
                                break;
                }
        }

        switch (code_type) {
                default:
                        if (brace_count) {
                                *prompt = cli_get_prompt("php", '(' TSRMLS_CC);
                        } else if (brackets_count) {
                                *prompt = cli_get_prompt("php", '{' TSRMLS_CC);
                        } else {
                                *prompt = cli_get_prompt("php", '>' TSRMLS_CC);
                        }
                        break;
                case sstring:
                case sstring_esc:
                        *prompt = cli_get_prompt("php", '\'' TSRMLS_CC);
                        break;
                case dstring:
                case dstring_esc:
                        *prompt = cli_get_prompt("php", '"' TSRMLS_CC);
                        break;
                case comment_block:
                        *prompt = cli_get_prompt("/* ", '>' TSRMLS_CC);
                        break;
                case heredoc:
                        *prompt = cli_get_prompt("<<<", '>' TSRMLS_CC);
                        break;
                case outside:
                        *prompt = cli_get_prompt("   ", '>' TSRMLS_CC);
                        break;
        }

        if (!valid_end || brackets_count) {
                return 0;
        } else {
                return 1;
        }
}
/* }}} */

static char *cli_completion_generator_ht(const char *text, int textlen, int *state, HashTable *ht, void **pData TSRMLS_DC) /* {{{ */
{
        char *name;
        ulong number;

        if (!(*state % 2)) {
                zend_hash_internal_pointer_reset(ht);
                (*state)++;
        }
        while(zend_hash_has_more_elements(ht) == SUCCESS) {
                zend_hash_get_current_key(ht, &name, &number, 0);
                if (!textlen || !strncmp(name, text, textlen)) {
                        if (pData) {
                                zend_hash_get_current_data(ht, pData);
                        }
                        zend_hash_move_forward(ht);
                        return name;
                }
                if (zend_hash_move_forward(ht) == FAILURE) {
                        break;
                }
        }
        (*state)++;
        return NULL;
} /* }}} */

static char *cli_completion_generator_var(const char *text, int textlen, int *state TSRMLS_DC) /* {{{ */
{
        char *retval, *tmp;

        tmp = retval = cli_completion_generator_ht(text + 1, textlen - 1, state, EG(active_symbol_table), NULL TSRMLS_CC);
        if (retval) {
                retval = malloc(strlen(tmp) + 2);
                retval[0] = '$';
                strcpy(&retval[1], tmp);
                rl_completion_append_character = '\0';
        }
        return retval;
} /* }}} */

static char *cli_completion_generator_ini(const char *text, int textlen, int *state TSRMLS_DC) /* {{{ */
{
        char *retval, *tmp;

        tmp = retval = cli_completion_generator_ht(text + 1, textlen - 1, state, EG(ini_directives), NULL TSRMLS_CC);
        if (retval) {
                retval = malloc(strlen(tmp) + 2);
                retval[0] = '#';
                strcpy(&retval[1], tmp);
                rl_completion_append_character = '=';
        }
        return retval;
} /* }}} */

static char *cli_completion_generator_func(const char *text, int textlen, int *state, HashTable *ht TSRMLS_DC) /* {{{ */
{
        zend_function *func;
        char *retval = cli_completion_generator_ht(text, textlen, state, ht, (void**)&func TSRMLS_CC);
        if (retval) {
                rl_completion_append_character = '(';
                retval = strdup(func->common.function_name);
        }
        
        return retval;
} /* }}} */

static char *cli_completion_generator_class(const char *text, int textlen, int *state TSRMLS_DC) /* {{{ */
{
        zend_class_entry **pce;
        char *retval = cli_completion_generator_ht(text, textlen, state, EG(class_table), (void**)&pce TSRMLS_CC);
        if (retval) {
                rl_completion_append_character = '\0';
                retval = strdup((*pce)->name);
        }
        
        return retval;
} /* }}} */

static char *cli_completion_generator_define(const char *text, int textlen, int *state, HashTable *ht TSRMLS_DC) /* {{{ */
{
        zend_class_entry **pce;
        char *retval = cli_completion_generator_ht(text, textlen, state, ht, (void**)&pce TSRMLS_CC);
        if (retval) {
                rl_completion_append_character = '\0';
                retval = strdup(retval);
        }
        
        return retval;
} /* }}} */

static int cli_completion_state;

static char *cli_completion_generator(const char *text, int index) /* {{{ */
{
/*
TODO:
- constants
- maybe array keys
- language constructs and other things outside a hashtable (echo, try, function, class, ...)
- object/class members

- future: respect scope ("php > function foo() { $[tab]" should only expand to local variables...)
*/
        char *retval = NULL;
        int textlen = strlen(text);
        TSRMLS_FETCH();

        if (!index) {
                cli_completion_state = 0;
        }
        if (text[0] == '$') {
                retval = cli_completion_generator_var(text, textlen, &cli_completion_state TSRMLS_CC);
        } else if (text[0] == '#') {
                retval = cli_completion_generator_ini(text, textlen, &cli_completion_state TSRMLS_CC);
        } else {
                char *lc_text, *class_name, *class_name_end;
                int class_name_len;
                zend_class_entry **pce = NULL;
                
                class_name_end = strstr(text, "::");
                if (class_name_end) {
                        class_name_len = class_name_end - text;
                        class_name = zend_str_tolower_dup(text, class_name_len);
                        class_name[class_name_len] = '\0'; /* not done automatically */
                        if (zend_lookup_class(class_name, class_name_len, &pce TSRMLS_CC)==FAILURE) {
                                efree(class_name);
                                return NULL;
                        }
                        lc_text = zend_str_tolower_dup(class_name_end + 2, textlen - 2 - class_name_len);
                        textlen -= (class_name_len + 2);
                } else {
                        lc_text = zend_str_tolower_dup(text, textlen);
                }

                switch (cli_completion_state) {
                        case 0:
                        case 1:
                                retval = cli_completion_generator_func(lc_text, textlen, &cli_completion_state, pce ? &(*pce)->function_table : EG(function_table) TSRMLS_CC);
                                if (retval) {
                                        break;
                                }
                        case 2:
                        case 3:
                                retval = cli_completion_generator_define(text, textlen, &cli_completion_state, pce ? &(*pce)->constants_table : EG(zend_constants) TSRMLS_CC);
                                if (retval || pce) {
                                        break;
                                }
                        case 4:
                        case 5:
                                retval = cli_completion_generator_class(lc_text, textlen, &cli_completion_state TSRMLS_CC);
                                break;
                        default:
                                break;
                }
                efree(lc_text);
                if (class_name_end) {
                        efree(class_name);
                }
                if (pce && retval) {
                        int len = class_name_len + 2 + strlen(retval) + 1;
                        char *tmp = malloc(len);
                        
                        snprintf(tmp, len, "%s::%s", (*pce)->name, retval);
                        free(retval);
                        retval = tmp;
                }
        }
        
        return retval;
} /* }}} */

static char **cli_code_completion(const char *text, int start, int end) /* {{{ */
{
        return rl_completion_matches(text, cli_completion_generator);
}
/* }}} */

static int readline_shell_run(TSRMLS_D) /* {{{ */
{
        char *line;
        size_t size = 4096, pos = 0, len;
        char *code = emalloc(size);
        char *prompt = cli_get_prompt("php", '>' TSRMLS_CC);
        char *history_file;

        if (PG(auto_prepend_file) && PG(auto_prepend_file)[0]) {
                zend_file_handle *prepend_file_p;
                zend_file_handle prepend_file = {0};

                prepend_file.filename = PG(auto_prepend_file);
                prepend_file.opened_path = NULL;
                prepend_file.free_filename = 0;
                prepend_file.type = ZEND_HANDLE_FILENAME;
                prepend_file_p = &prepend_file;

                zend_execute_scripts(ZEND_REQUIRE TSRMLS_CC, NULL, 1, prepend_file_p);
        }

        history_file = tilde_expand("~/.php_history");
        rl_attempted_completion_function = cli_code_completion;
        rl_special_prefixes = "$";
        read_history(history_file);

        EG(exit_status) = 0;
        while ((line = readline(prompt)) != NULL) {
                if (strcmp(line, "exit") == 0 || strcmp(line, "quit") == 0) {
                        free(line);
                        break;
                }

                if (!pos && !*line) {
                        free(line);
                        continue;
                }

                len = strlen(line);

                if (line[0] == '#') {
                        char *param = strstr(&line[1], "=");
                        if (param) {
                                char *cmd;
                                uint cmd_len;
                                param++;
                                cmd_len = param - &line[1] - 1;
                                cmd = estrndup(&line[1], cmd_len);

                                zend_alter_ini_entry_ex(cmd, cmd_len + 1, param, strlen(param), PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0 TSRMLS_CC);
                                efree(cmd);
                                add_history(line);

                                efree(prompt);
                                /* TODO: This might be wrong! */
                                prompt = cli_get_prompt("php", '>' TSRMLS_CC);
                                continue;
                        }
                }

                if (pos + len + 2 > size) {
                        size = pos + len + 2;
                        code = erealloc(code, size);
                }
                memcpy(&code[pos], line, len);
                pos += len;
                code[pos] = '\n';
                code[++pos] = '\0';

                if (*line) {
                        add_history(line);
                }

                free(line);
                efree(prompt);

                if (!cli_is_valid_code(code, pos, &prompt TSRMLS_CC)) {
                        continue;
                }

                zend_try {
                        zend_eval_stringl(code, pos, NULL, "php shell code" TSRMLS_CC);
                } zend_end_try();

                pos = 0;
                                        
                if (!pager_pipe && php_last_char != '\0' && php_last_char != '\n') {
                        readline_shell_write("\n", 1 TSRMLS_CC);
                }

                if (EG(exception)) {
                        zend_exception_error(EG(exception), E_WARNING TSRMLS_CC);
                }

                if (pager_pipe) {
                        fclose(pager_pipe);
                        pager_pipe = NULL;
                }

                php_last_char = '\0';
        }
        write_history(history_file);
        free(history_file);
        efree(code);
        efree(prompt);
        return EG(exit_status);
}
/* }}} */

/*
#ifdef COMPILE_DL_READLINE
This dlsym() is always used as even the CGI SAPI is linked against "CLI"-only
extensions. If that is being changed dlsym() should only be used when building
this extension sharedto offer compatibility.
*/
#define GET_SHELL_CB(cb) \
        do { \
                (cb) = NULL; \
                cli_shell_callbacks_t *(*get_callbacks)(); \
                get_callbacks = dlsym(RTLD_DEFAULT, "php_cli_get_shell_callbacks"); \
                if (get_callbacks) { \
                        (cb) = get_callbacks(); \
                } \
        } while(0)
/*#else
#define GET_SHELL_CB(cb) (cb) = php_cli_get_shell_callbacks()
#endif*/

PHP_MINIT_FUNCTION(cli_readline)
{
        cli_shell_callbacks_t *cb;

        ZEND_INIT_MODULE_GLOBALS(cli_readline, cli_readline_init_globals, NULL);
        REGISTER_INI_ENTRIES();

#if HAVE_LIBEDIT
        REGISTER_STRING_CONSTANT("READLINE_LIB", "libedit", CONST_CS|CONST_PERSISTENT);
#else
        REGISTER_STRING_CONSTANT("READLINE_LIB", "readline", CONST_CS|CONST_PERSISTENT);
#endif

        GET_SHELL_CB(cb);
        if (cb) {
                cb->cli_shell_write = readline_shell_write;
                cb->cli_shell_ub_write = readline_shell_ub_write;
                cb->cli_shell_run = readline_shell_run;
        }

        return SUCCESS;
}

PHP_MSHUTDOWN_FUNCTION(cli_readline)
{
        cli_shell_callbacks_t *cb;

        UNREGISTER_INI_ENTRIES();

        GET_SHELL_CB(cb);
        if (cb) {
                cb->cli_shell_write = NULL;
                cb->cli_shell_ub_write = NULL;
                cb->cli_shell_run = NULL;
        }

        return SUCCESS;
}

PHP_MINFO_FUNCTION(cli_readline)
{
        php_info_print_table_start();
        php_info_print_table_header(2, "Readline Support", "enabled");
        php_info_print_table_row(2, "Readline library", (rl_library_version ? rl_library_version : "Unknown"));
        php_info_print_table_end();

        DISPLAY_INI_ENTRIES();
}

/*
 * Local variables:
 * tab-width: 4
 * c-basic-offset: 4
 * End:
 * vim600: sw=4 ts=4 fdm=marker
 * vim<600: sw=4 ts=4
 */

/* [<][>][^][v][top][bottom][index][help] */