wing_process.c

/*
  +----------------------------------------------------------------------+
  | PHP Version 7                                                        |
  +----------------------------------------------------------------------+
  | Copyright (c) 1997-2016 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.               |
  +----------------------------------------------------------------------+
  | php windows、mac and linux daemon process support
  | Author: yuyi
  | Email: 297341015@qq.com
  | Home: http://www.itdfy.com/
  +----------------------------------------------------------------------+
*/

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "php.h"
#include "php_ini.h"
#include "ext/standard/info.h"
#include "php_wing_process.h"
#include "wing_api.h"


#define WING_ERROR_FAILED  0
#define WING_ERROR_SUCCESS 1

static int le_wing_process;
char *PHP_PATH = NULL;

zend_class_entry *wing_process_ce;

typedef struct _WING_PROCESS_INFO {
    unsigned long process_id;
    unsigned long thread_id;
    void *ext_info;
    char* command;
    char* file;
} WING_PROCESS_INFO;

/**
 * 构造函数，第一个参数为需要以守护进程执行的php文件或者是其他可执行的命令和文件
 * 如果传入的文件是一个数字，则认为根据进程id打开进程，后续可以对此进程进行控制，比如kill
 *
 * @param string $file
 */
ZEND_METHOD(wing_process, __construct)
{

    #if PHP_MAJOR_VERSION >= 7
	zend_string *_file = NULL;
	char *file         = NULL;
    int file_len       = 0;

    if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "S", &_file)) {
        return;
    }

    if (_file) {
        file = ZSTR_VAL(_file);
        efree(_file);
    }

    file_len      = strlen(file);
    #else
    char *file    = NULL;
    int file_len  = 0;

    if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &file, &file_len)) {
        return;
    }
    #endif

	WING_PROCESS_INFO *info = (WING_PROCESS_INFO*)emalloc(sizeof(WING_PROCESS_INFO)+1);
	info->process_id = 0;
	info->thread_id  = 0;
	info->ext_info   = NULL;
	info->command    = NULL;
	info->file       = (char*)emalloc(strlen(file)+1);
	strcpy(info->file, file);

	int size = file_len + 2;
	if (PHP_PATH) {
	    size += strlen(PHP_PATH);
	}
	char *command_line = NULL;

	//如果传入的文件是一个数字，则认为此数字为进程id
	if (is_numeric_string(file, strlen(file), NULL, NULL, 0)) {
	    #ifdef PHP_WIN32
		PROCESSINFO *item = new PROCESSINFO();
		DWORD process_id  = zend_atoi(file, strlen(file));

		info->process_id  = (unsigned long)process_id;

		WingQueryProcessByProcessID(item, process_id);
		if (item) {
			spprintf(&command_line, strlen(item->command_line), "%s", item->command_line);
			delete item;
		}
		#else
		info->process_id  = (unsigned long)zend_atoi(file, strlen(file));

		char *buffer = NULL;
		wing_get_cmdline(info->process_id, &buffer);
        if (buffer) {
		    spprintf(&command_line, strlen(buffer), "%s", buffer);
		} else {
		    spprintf(&command_line, strlen(""), "%s", "");
		}
		if (buffer) {
		    free(buffer);
		}

		#endif

		//这里还需要解析command_line 判断是否是php命令 还原成file 填充到 info->file
		if (command_line) {
            char *find = strstr(command_line, " ");
            if(info->file) {
                efree(info->file);
                info->file = NULL;
            }

            if (find == NULL) {
                info->file = (char*)emalloc(strlen(command_line)+1);
                strcpy(info->file, command_line);
            } else {
                char _cmd[MAX_PATH] = {0};
                strncpy(_cmd, command_line, find-command_line);
                int _dphp = strcmp(_cmd, "php");
                if (strcmp(_cmd, PHP_PATH) == 0 || _dphp == 0) {
                    //php文件
                    memset(_cmd,0,MAX_PATH);
                    strcpy(_cmd, (char*)(find+1));
                    info->file = (char*)emalloc(strlen(_cmd)+1);
                    strcpy(info->file, _cmd);

                    if (_dphp == 0) {
                        efree(command_line);
                        command_line = NULL;
                        size = strlen(PHP_PATH) + strlen(info->file)+2;
                        spprintf(&command_line, size, "%s %s\0", PHP_PATH, info->file);
                    }
                } else {
                    info->file = (char*)emalloc(strlen(command_line)+1);
                    strcpy(info->file, command_line);
                }
            }
		}
	} else {
		if (wing_file_is_php((const char*)file)){
		    if (PHP_PATH) {
			    spprintf(&command_line, size, "%s %s\0", PHP_PATH, file);
			} else {
				php_error_docref(NULL TSRMLS_CC, E_WARNING, "找不到php命令(could not find command php)");
			}
		} else {
			spprintf(&command_line, size, "%s\0", file);
		}
	}

	if (command_line) {
	    info->command  = (char*)emalloc(strlen(command_line)+1);
	    strcpy(info->command, command_line);
	    efree(command_line);
	}

	zend_update_property_long(wing_process_ce, getThis(), "process_info", strlen("process_info"), (unsigned long)info TSRMLS_CC);
}

/***
 * 析构函数
 * windows下面需要释放一些资源
 */
ZEND_METHOD(wing_process, __destruct) {

	zval *_info = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(_info);

	#ifdef PHP_WIN32
	PROCESS_INFORMATION *pi = (PROCESS_INFORMATION *)(info->ext_info);

	if (pi) {
		CloseHandle(pi->hProcess);
		CloseHandle(pi->hThread);
		delete pi;
	}
	#endif

	if (info->command) {
       efree(info->command);
    }
    efree(info->file);
    efree(info);
}

/**
 * 运行进程
 *
 * @param string $output_file 默认为null，如果不为null，则以守护进程方式运行
 * @return int 返回进程id
 */
ZEND_METHOD(wing_process, run)
{
    char *output_file = NULL;

    #if PHP_MAJOR_VERSION >= 7

    zend_string *_output_file = NULL;
	zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|S", &_output_file);

    if (_output_file) {
        output_file = ZSTR_VAL(_output_file);
        efree(_output_file);
    }

	#else
    int output_file_len = 0;
    zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|s", &output_file, &output_file_len);
	#endif

	zval *_info             = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO*)Z_LVAL_P(_info);

    #ifdef PHP_WIN32
	PROCESS_INFORMATION *pi = (PROCESS_INFORMATION *)wing_create_process(info->command, output_file);
    info->ext_info   = (void*)pi;
    info->process_id = pi->dwProcessId;
    info->thread_id  = pi->dwThreadId;
	RETURN_LONG(pi->dwProcessId);
	#else
    unsigned long childpid = wing_create_process(info->file, output_file);
    info->process_id = childpid;
	RETURN_LONG(childpid);
	#endif
}


/**
 * 等待进程退出，直接返回退出的进程id
 *
 * @param int $timout windows下面的意思为等待超时时间，linux下面的意思为是否等待，可选参数如 WNOHANG | WUNTRACED
 * @return int
 */
ZEND_METHOD(wing_process, wait)
{
	int timeout = INFINITE;
	zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|l", &timeout);

    zval *_info = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(_info);

    #ifdef PHP_WIN32
	HANDLE process = NULL;
    int process_id = 0;
	if (is_numeric_string(info->file, strlen(info->file), NULL, NULL, 0)) {
		process    = OpenProcess(PROCESS_ALL_ACCESS, FALSE, zend_atoi(info->file, strlen(info->file)));
	    process_id = zend_atoi(info->file, strlen(info->file));
	} else {
		PROCESS_INFORMATION *pi = (PROCESS_INFORMATION *)(info->ext_info);
		process                 = pi->hProcess;
		process_id              = info->process_id;
	}

	DWORD wait_result = 0;
	DWORD wait_status = WaitForSingleObject(process, timeout);

	if (wait_status != WAIT_OBJECT_0) {
		RETURN_LONG(process_id);//wait_status);
	}
	if (GetExitCodeProcess(process, &wait_result) == 0) {
		RETURN_LONG(-1);
	}

	RETURN_LONG(process_id);
	#else
	int status;
	pid_t epid = waitpid(info->process_id, &status, timeout);
	RETURN_LONG(epid);
	#endif
}

/**
 * 获取进程id
 *
 * @return int
 */
ZEND_METHOD(wing_process, getProcessId)
{
	zval *_info = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(_info);
    RETURN_LONG(info->process_id);
}


/**
 * 获取线程id
 *
 * @return int
 */
ZEND_METHOD(wing_process, getThreadId)
{
    zval *_info = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(_info);
    #ifdef PHP_WIN32
    RETURN_LONG(info->thread_id);
    #else
    RETURN_LONG(info->process_id);
    #endif
}


/**
 * 获取命令行参数
 *
 * @return string
 */
ZEND_METHOD(wing_process, getCommandLine)
{
    zval *zinfo = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(zinfo);

    #if PHP_MAJOR_VERSION >= 7
    ZVAL_STRING(return_value, info->command);
    #else
    ZVAL_STRING(return_value, info->command, 1);
    #endif
}

/**
 * 杀死进程
 *
 * @return int
 */
ZEND_METHOD(wing_process, kill)
{
    zval *_info = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(_info);

	#ifdef PHP_WIN32
	HANDLE process = NULL;

	if (is_numeric_string(info->file, strlen(info->file), NULL, NULL, 0)) {
		process   = OpenProcess(PROCESS_ALL_ACCESS, FALSE, zend_atoi(info->file, strlen(info->file)));
	} else {
		PROCESS_INFORMATION *pi = (PROCESS_INFORMATION *)(info->ext_info);
		process   = pi->hProcess;
	}

    //非安全的方式直接退出 可能造成进程数据丢失
	if (!TerminateProcess(process, 0)) {
		RETURN_FALSE;//LONG(WING_ERROR_FAILED);
		return;
	}
	//RETURN_LONG(WING_ERROR_SUCCESS);
	RETURN_TRUE;
	#else
    int process_id = 0;

    if (is_numeric_string(info->file, strlen(info->file), NULL, NULL, 0)) {
        process_id = zend_atoi(info->file, strlen(info->file));
    } else {
        process_id = info->process_id;
    }

    int status = kill(process_id, SIGKILL);

    if (status == -1) {
        RETURN_FALSE;
    }

    wait(&status);

    if (WIFSIGNALED(status)) {
        RETURN_TRUE;
    }

    RETURN_FALSE;
	#endif
}

/**
 * 返回进程占用的实际内存，单位为k
 *
 * @return int
 */
ZEND_METHOD(wing_process, getMemory)
{
	zval *_info = wing_zend_read_property(wing_process_ce, getThis(),"process_info");
    WING_PROCESS_INFO *info = (WING_PROCESS_INFO *)Z_LVAL_P(_info);

	#ifdef PHP_WIN32
	HANDLE process = NULL;

	if (is_numeric_string(info->file, strlen(info->file), NULL, NULL, 0)) {
		process = OpenProcess(PROCESS_ALL_ACCESS, FALSE, zend_atoi(info->file, strlen(info->file)));
	} else {
		PROCESS_INFORMATION *pi = (PROCESS_INFORMATION *)(info->ext_info);
		process = pi->hProcess;
	}

	if (!process) {
		RETURN_LONG(0);
	}

	PROCESS_MEMORY_COUNTERS pmc;
	GetProcessMemoryInfo(process, &pmc, sizeof(pmc));
	RETURN_LONG(pmc.WorkingSetSize/1024);
	#else
	unsigned long mem = wing_get_memory(info->process_id);
	RETURN_LONG(mem);
	#endif
}

/**
 * 静态方法，获取当前进程id
 * @call \wing\wing_process::getCurrentProcessId()
 * @return int
 */
ZEND_METHOD(wing_process, getCurrentProcessId)
{
	ZVAL_LONG(return_value, wing_get_process_id());
}
//
//ZEND_METHOD(wing_process, signal)
//{
//    int sig = -1;
//    zval* callback;
//
//    zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "lz", &sig, &callback);
//
//    #ifdef PHP_WIN32
//
//    #else
//
//    #endif
//	RETURN_TRUE
//}


static zend_function_entry wing_process_methods[] =
{
	ZEND_ME(wing_process, __construct, NULL, ZEND_ACC_PUBLIC | ZEND_ACC_CTOR)
	ZEND_ME(wing_process, __destruct, NULL, ZEND_ACC_PUBLIC | ZEND_ACC_DTOR)
	ZEND_ME(wing_process, wait, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, run, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, getProcessId, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, getThreadId, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, getCommandLine, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, kill, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, getMemory, NULL, ZEND_ACC_PUBLIC)
	ZEND_ME(wing_process, getCurrentProcessId, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
	//ZEND_ME(wing_process, signal, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
	{
	    NULL,NULL,NULL
	}
};

/**
 * 模块初始化，这里会初始化PHP_PATH，也就是php的可执行文件路径
 */
PHP_MINIT_FUNCTION(wing_process)
{

	PHP_PATH = wing_get_command_path("php");

	REGISTER_STRING_CONSTANT("WING_PROCESS_PHP",     PHP_PATH,                 CONST_CS | CONST_PERSISTENT );
	REGISTER_STRING_CONSTANT("WING_PROCESS_VERSION", PHP_WING_PROCESS_VERSION, CONST_CS | CONST_PERSISTENT );

	zend_class_entry _wing_process_ce;
	INIT_NS_CLASS_ENTRY(_wing_process_ce, "wing", "wing_process", wing_process_methods);
	wing_process_ce = zend_register_internal_class(&_wing_process_ce TSRMLS_CC);

	zend_declare_property_long(wing_process_ce, "process_info", strlen("process_info"), 0, ZEND_ACC_PRIVATE TSRMLS_CC);
	return SUCCESS;
}

/**
 * 模块释放
 */
PHP_MSHUTDOWN_FUNCTION(wing_process)
{
	if (PHP_PATH) {
		free(PHP_PATH);
	}
	return SUCCESS;
}

/**
 * 模块请求初始化
 */
PHP_RINIT_FUNCTION(wing_process)
{
#if PHP_MAJOR_VERSION >= 7
#if defined(COMPILE_DL_WING_PROCESS) && defined(ZTS)
	ZEND_TSRMLS_CACHE_UPDATE();
#endif
#endif
	return SUCCESS;
}

/**
 * 模块请求释放
 */
PHP_RSHUTDOWN_FUNCTION(wing_process)
{
	return SUCCESS;
}

/**
 * 模块相关的信息 phpinfo
 */
PHP_MINFO_FUNCTION(wing_process)
{
	php_info_print_table_start();
	php_info_print_table_header(2, "wing_process support", "enabled");
	php_info_print_table_end();
}


const zend_function_entry wing_process_functions[] = {
    //PHP_FE(wing_process_wait,NULL)
    //PHP_FE(wing_create_process_ex,NULL)
	//PHP_FE(alarm, NULL)
	PHP_FE_END	/* Must be the last line in wing_process_functions[] */
};

zend_module_entry wing_process_module_entry =
{
	STANDARD_MODULE_HEADER,
	"wing_process",
	wing_process_functions,
	PHP_MINIT(wing_process),
	PHP_MSHUTDOWN(wing_process),
	PHP_RINIT(wing_process),		/* Replace with NULL if there's nothing to do at request start */
	PHP_RSHUTDOWN(wing_process),	/* Replace with NULL if there's nothing to do at request end */
	PHP_MINFO(wing_process),
	PHP_WING_PROCESS_VERSION,
	STANDARD_MODULE_PROPERTIES
};

#ifdef COMPILE_DL_WING_PROCESS
#if PHP_MAJOR_VERSION >= 7
#ifdef ZTS
ZEND_TSRMLS_CACHE_DEFINE()
#endif
#endif
ZEND_GET_MODULE(wing_process)
#endif