php扩展开发---kqueue 实例

来源:互联网 发布:win7删除u盘linux分区 编辑:程序博客网 时间:2024/05/29 03:09
/*  +----------------------------------------------------------------------+  | PHP Version 5                                                        |  +----------------------------------------------------------------------+  | 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.               |  +----------------------------------------------------------------------+  | Author: xing2233@qq.com                                                             |  +----------------------------------------------------------------------+*//* $Id$ */#ifdef HAVE_CONFIG_H#include "config.h"#endif//#include <Zend/zend_interfaces.h>#include "php_xing2233.h"/* If you declare any globals in php_xing2233.h uncomment this:ZEND_DECLARE_MODULE_GLOBALS(xing2233)*//* True global resources - no need for thread safety here */static int le_xing2233;/* {{{ PHP_INI *//* Remove comments and fill if you need to have entries in php.iniPHP_INI_BEGIN()    STD_PHP_INI_ENTRY("xing2233.global_value",      "42", PHP_INI_ALL, OnUpdateLong, global_value, zend_xing2233_globals, xing2233_globals)    STD_PHP_INI_ENTRY("xing2233.global_string", "foobar", PHP_INI_ALL, OnUpdateString, global_string, zend_xing2233_globals, xing2233_globals)PHP_INI_END()*//* }}} *//* Remove the following function when you have successfully modified config.m4   so that your module can be compiled into PHP, it exists only for testing   purposes. *//* Every user-visible function in PHP should document itself in the source *//* {{{ proto string confirm_xing2233_compiled(string arg)   Return a string to confirm that the module is compiled in */PHP_FUNCTION(confirm_xing2233_compiled){//    zval *response;//    MAKE_STD_ZVAL(response);//    array_init(response);//    add_assoc_string(response, "end", "111", 1);//    zend_hash_add(EG(active_symbol_table),"response", sizeof("response"), &response, sizeof(zval*), NULL);//    php_printf("111");}/* }}} *//* The previous line is meant for vim and emacs, so it can correctly fold and   unfold functions in source code. See the corresponding marks just before   function definition, where the functions purpose is also documented. Please   follow this convention for the convenience of others editing your code.*/int xSocket_bind(int sock, int type, char *host, int *port, TSRMLS_D){    int ret;    struct sockaddr_in addr_in4;    struct sockaddr_in6 addr_in6;    struct sockaddr_un addr_un;    socklen_t len;    //SO_REUSEADDR option    int option = 1;    if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &option, sizeof(int)) < 0)    {        php_error_docref(NULL TSRMLS_CC, E_ERROR, "setsockopt failed");    }    //reuse port    //unix socket    if (type == SW_SOCK_UNIX_DGRAM || type == SW_SOCK_UNIX_STREAM)    {        bzero(&addr_un, sizeof(addr_un));        unlink(host);        addr_un.sun_family = AF_UNIX;        strncpy(addr_un.sun_path, host, sizeof(addr_un.sun_path) - 1);        ret = bind(sock, (struct sockaddr*) &addr_un, sizeof(addr_un));    }        //IPv6    else if (type > SW_SOCK_UDP)    {        bzero(&addr_in6, sizeof(addr_in6));        inet_pton(AF_INET6, host, &(addr_in6.sin6_addr));        addr_in6.sin6_port = htons(*port);        addr_in6.sin6_family = AF_INET6;        ret = bind(sock, (struct sockaddr *) &addr_in6, sizeof(addr_in6));        if (ret == 0 && *port == 0)        {            len = sizeof(addr_in6);            if (getsockname(sock, (struct sockaddr *) &addr_in6, &len) != -1)            {                *port = ntohs(addr_in6.sin6_port);            }        }    }        //IPv4    else    {        bzero(&addr_in4, sizeof(addr_in4));        inet_pton(AF_INET, host, &(addr_in4.sin_addr));        addr_in4.sin_port = htons(*port);        addr_in4.sin_family = AF_INET;        ret = bind(sock, (struct sockaddr *) &addr_in4, sizeof(addr_in4));        if (ret == 0 && *port == 0)        {            len = sizeof(addr_in4);            if (getsockname(sock, (struct sockaddr *) &addr_in4, &len) != -1)            {                *port = ntohs(addr_in4.sin_port);            }        }    }    //bind failed    if (ret < 0)    {//        swoole_error_log(SW_LOG_WARNING, SW_ERROR_SYSTEM_CALL_FAIL, "bind(%s:%d) failed. Error: %s [%d]", host, *port, strerror(errno), errno);//        return SW_ERR;        return FAILURE;    }    return ret;}int xSocket_create(int type, TSRMLS_D) {    int _domain;    int _type;    switch (type) {        case SW_SOCK_TCP:            _domain = PF_INET;            _type = SOCK_STREAM;            break;        case SW_SOCK_TCP6:            _domain = PF_INET6;            _type = SOCK_STREAM;            break;        case SW_SOCK_UDP:            _domain = PF_INET;            _type = SOCK_DGRAM;            break;        case SW_SOCK_UDP6:            _domain = PF_INET6;            _type = SOCK_DGRAM;            break;        case SW_SOCK_UNIX_DGRAM:            _domain = PF_UNIX;            _type = SOCK_DGRAM;            break;        case SW_SOCK_UNIX_STREAM:            _domain = PF_UNIX;            _type = SOCK_STREAM;            break;        default:            php_error_docref(NULL TSRMLS_CC, E_ERROR, "unknown socket type [%d]", type);            return FAILURE;    }    return socket(_domain, _type, 0);}zend_class_entry xing2233_http_server_ce;zend_class_entry *xing2233_http_server_class_entry_ptr;//responsezend_class_entry xing2233_http_server_ce_response;zend_class_entry *xing2233_http_server_class_entry_ptr_response;PHP_METHOD(xing_http_server, __construct){    char *serv_host;    int host_len;    long serv_port, serv_mode, sock_type; //port 类型必须是long ,int不起作用    if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s|lll", &serv_host, &host_len, &serv_port, &serv_mode, &sock_type) == FAILURE)    {        php_printf("failure");        return ;    }    zend_update_property_stringl(            xing2233_http_server_class_entry_ptr,            getThis(),            "host",            sizeof("host") - 1,            serv_host,            host_len TSRMLS_CC    );    zend_update_property_long(            xing2233_http_server_class_entry_ptr,            getThis(),            "port",            sizeof("port") - 1,            serv_port TSRMLS_CC    );    zend_update_property_long(            xing2233_http_server_class_entry_ptr,            getThis(),            "mode",            sizeof("mode") - 1,            serv_mode TSRMLS_CC    );    zend_update_property_long(            xing2233_http_server_class_entry_ptr,            getThis(),            "type",            sizeof("type") - 1,            sock_type TSRMLS_CC    );}int make_sock_non_blocking(int sock, TSRMLS_D){    int flags;    flags = fcntl(sock, F_GETFL, 0);    if (flags < 0)    {        php_error_docref(NULL TSRMLS_CC, E_ERROR, "fcntl sock failed.");    }    flags |= O_NONBLOCK;    return fcntl(sock, F_SETFL, flags);}PHP_METHOD(xing_http_server, start){    char *serv_host;    int host_len;    long serv_port, serv_mode, sock_type;    zval *host = zend_read_property(            xing2233_http_server_class_entry_ptr,            getThis(),            "host",            sizeof("host") - 1 , 0 TSRMLS_CC    );    serv_host = Z_STRVAL_P(host);    zval *port = zend_read_property(            xing2233_http_server_class_entry_ptr,            getThis(),            "port",            sizeof("port") - 1 , 0 TSRMLS_CC    );    int ser_port = Z_LVAL_P(port);    zval *mode = zend_read_property(            xing2233_http_server_class_entry_ptr,            getThis(),            "mode",            sizeof("mode") - 1 , 0 TSRMLS_CC    );    serv_mode = Z_LVAL_P(mode);    zval *type = zend_read_property(            xing2233_http_server_class_entry_ptr,            getThis(),            "type",            sizeof("type") - 1 , 0 TSRMLS_CC    );    int soc_type = Z_LVAL_P(type);    int sock = xSocket_create(soc_type ,TSRMLS_C);    if (sock < 0)    {        php_error_docref(NULL TSRMLS_CC, E_ERROR, "create socket failed.");        return ;    }    if (xSocket_bind(sock, soc_type, serv_host, &ser_port, TSRMLS_C) < 0)    {        close(sock);        php_error_docref(NULL TSRMLS_CC, E_ERROR, "bind sock failed.");        return ;    }    make_sock_non_blocking(sock, TSRMLS_C);    if (listen(sock, SW_BACKLOG) < 0)    {        php_error_docref(NULL TSRMLS_CC, E_ERROR, "listen failed");        return ;    }    zval *onResponse = zend_read_property(            xing2233_http_server_class_entry_ptr,            getThis(),            "onResponse",            sizeof("onResponse") - 1 , 0 TSRMLS_CC    );    //回调方法参数赋值    zval *args[2];    MAKE_STD_ZVAL(args[0]);    MAKE_STD_ZVAL(args[1]);    ZVAL_STRING(args[0], "request", 1);    object_init_ex(args[1], xing2233_http_server_class_entry_ptr_response);//    zval *resonseValue;    //    if (call_user_function(            EG(function_table),            NULL,            onResponse,            return_value, 2, args TSRMLS_CC) != SUCCESS) {        php_printf("callback faild");        RETURN_FALSE;    }    zval *response = zend_read_property(            xing2233_http_server_class_entry_ptr_response,            args[1],            "response",            sizeof("response") - 1 , 0 TSRMLS_CC    );    zval **end;    if (zend_hash_find(Z_ARRVAL_P(response), "end", strlen("end") + 1 , (void**)&end) != SUCCESS)    {    }    //直接赋值 char *endVal = Z_STRVAL_PP(end) 不起作用    char endVal[Z_STRLEN_PP(end)];    snprintf(endVal, Z_STRLEN_PP(end) + 1, Z_STRVAL_PP(end));    zval_ptr_dtor(&args[0]);    zval_ptr_dtor(&args[1]);//    in_port_t    struct sockaddr_in remote_addr;    socklen_t remote_addr_size = sizeof(struct sockaddr);    int kq = kqueue();    struct kevent changes[1];    EV_SET(&changes[0], sock, EVFILT_READ, EV_ADD,0, 0, NULL);    if (kevent(kq, changes, 1, NULL, 0, NULL) < 0)    {        php_error_docref(NULL TSRMLS_CC, E_ERROR, "registe sock failed");    }    struct kevent events[2];    while (1)    {        int ret = kevent(kq, NULL, 0, events, 2, NULL);        for (int i = 0; i < ret; ++i) {            if (events[i].ident == sock)            {                    int client = accept(sock, (struct sockaddr *) &remote_addr, &remote_addr_size);                    if (client < 0)                    {                        php_printf("%s", "accept failed");                        break;                    }                    php_printf("Accepted connection on descriptor=%d host=%s port=%d\n", client, inet_ntoa(remote_addr.sin_addr), remote_addr.sin_port);                    make_sock_non_blocking(client, TSRMLS_C);                    EV_SET(&changes[0], client, EVFILT_READ, EV_ADD, 0, 0, NULL);                    if (kevent(kq, changes, 1, NULL, 0, NULL) < 0) {                        php_error_docref(NULL TSRMLS_CC, E_ERROR, "registe client failed");                    }//                    char word[BUFSIZ];//                    sprintf(word,//                            "HTTP/1.1 200 OK\r\nContent-Type:text/html;charset=utf8\r\nContent-Length: %lu\r\n\r\n%s",//                            strlen(endVal), endVal);//                    if (send(client, word, strlen(word), 0) < 0) {//                        php_error_docref(NULL TSRMLS_CC, E_ERROR, "send failed");//                        close(client);//                        return;//                    }//                    close(client);                //}            }            else            {                int done = 0;                while (1)                {                    ssize_t count;                    char buf[512];                    count = read(events[i].ident, buf, sizeof(buf));                    if (count < 0)                    {                        if (errno != EAGAIN)                        {                            done = 1;                        }                        break;                    }                    else if (count == 0)                    {                        done = 1;                        break;                    }                    write(1, buf, count);                }                if (done)                {                    php_printf("Closed connection on descriptor=%lu\n", events[i].ident);                    close(events[i].ident);                }            }        }    }    free(events);    close(sock);}PHP_METHOD(xing_http_server, on){//    php_printf("this is on");    zval *callback;    zval *event_name;    if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "zz", &event_name, &callback) == FAILURE)    {        php_printf("failure");    }    if (strncasecmp("request", Z_STRVAL_P(event_name), Z_STRLEN_P(event_name)) == 0)    {        zend_update_property(                xing2233_http_server_class_entry_ptr,                getThis(),                "onResponse",                sizeof("onResponse") - 1,                callback TSRMLS_CC        );    }    else    {        //判断callback是不是回调方法        char *func_name = NULL;        if (!zend_is_callable(callback, 0, &func_name TSRMLS_CC)) {            php_error_docref(NULL TSRMLS_CC, E_ERROR, "function '%s' is not callable", func_name);            efree(func_name);            RETURN_FALSE;        }        efree(func_name);        //回调方法参数赋值        zval *args[1];        MAKE_STD_ZVAL(args[0]);        ZVAL_LONG(args[0], 1);        if (call_user_function(EG(function_table), NULL, callback, return_value, 1, args TSRMLS_CC) != SUCCESS) {            php_printf("callback faild");            RETURN_FALSE;        }        zval_ptr_dtor(&args[0]);    }//    efree(func_name);}PHP_METHOD(xing_http_server_response, header){    char *header_one;    char *header_two;    int header_one_len, header_two_len;    if (zend_parse_parameters(            ZEND_NUM_ARGS() TSRMLS_CC,            "ss",            &header_one,            &header_one_len,            &header_two,            &header_two_len) == FAILURE)    {        php_printf("failure");    }}PHP_METHOD(xing_http_server_response, end){    char *end_one;    int end_one_len;    if (zend_parse_parameters(            ZEND_NUM_ARGS() TSRMLS_CC,            "s",            &end_one,            &end_one_len) == FAILURE)    {        php_printf("failure");    }    zval *response;    MAKE_STD_ZVAL(response);    array_init(response);    zend_update_property(            xing2233_http_server_class_entry_ptr_response,            getThis(),            "response",            sizeof("response") - 1,            response TSRMLS_CC    );    add_assoc_stringl_ex(response, "end", sizeof("end"), end_one, end_one_len, 1);    zval_ptr_dtor(&response);}static zend_function_entry xing2233_http_server_methods[] = {        PHP_ME(xing_http_server, __construct, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)        PHP_ME(xing_http_server, on, NULL, ZEND_ACC_PUBLIC)        PHP_ME(xing_http_server, start, NULL, ZEND_ACC_PUBLIC)        PHP_FE_END};//reponsestatic zend_function_entry xing2233_http_server_methods_response[] = {        PHP_ME(xing_http_server_response, header, NULL, ZEND_ACC_PUBLIC)        PHP_ME(xing_http_server_response, end, NULL, ZEND_ACC_PUBLIC)        PHP_FE_END};PHP_MINIT_FUNCTION(xing2233){    /* If you have INI entries, uncomment these lines    REGISTER_INI_ENTRIES();    */    INIT_CLASS_ENTRY(xing2233_http_server_ce, "xing_http_server", xing2233_http_server_methods);    xing2233_http_server_class_entry_ptr = zend_register_internal_class(&xing2233_http_server_ce TSRMLS_CC);    INIT_CLASS_ENTRY(xing2233_http_server_ce_response, "xing_http_server_response", xing2233_http_server_methods_response);    xing2233_http_server_class_entry_ptr_response = zend_register_internal_class(&xing2233_http_server_ce_response TSRMLS_CC);    //constant    REGISTER_LONG_CONSTANT("XING_SOCK_TCP", SW_SOCK_TCP, CONST_CS|CONST_PERSISTENT);    //mode    REGISTER_LONG_CONSTANT("XING_BASE", SW_MODE_SINGLE, CONST_CS | CONST_PERSISTENT);    REGISTER_LONG_CONSTANT("XING_THREAD", SW_MODE_THREAD, CONST_CS | CONST_PERSISTENT);    REGISTER_LONG_CONSTANT("XING_PROCESS", SW_MODE_PROCESS, CONST_CS | CONST_PERSISTENT);    return SUCCESS;}/* }}} *//* {{{ PHP_MSHUTDOWN_FUNCTION */PHP_MSHUTDOWN_FUNCTION(xing2233){    /* uncomment this line if you have INI entries    UNREGISTER_INI_ENTRIES();    *///    FILE *fp=fopen("/Users/albert/Documents/php/php-5.6.22/ext/xing2233/m.txt","a+");//    fprintf(fp, "%d\n", time_of_minit);//    fclose(fp);//  free(module_number);    return SUCCESS;}/* }}} *//* Remove if there's nothing to do at request start *//* {{{ PHP_RINIT_FUNCTION *///int time_of_rinit;PHP_RINIT_FUNCTION(xing2233){//    time_of_rinit = time(NULL);    return SUCCESS;}/* }}} *//* Remove if there's nothing to do at request end *//* {{{ PHP_RSHUTDOWN_FUNCTION */PHP_RSHUTDOWN_FUNCTION(xing2233){//    FILE *fp=fopen("/Users/albert/Documents/php/php-5.6.22/ext/xing2233/r.txt","a+");//    fprintf(fp, "%d\n", time_of_rinit);//    fclose(fp);    return SUCCESS;}/* }}} *//* {{{ PHP_MINFO_FUNCTION */PHP_MINFO_FUNCTION(xing2233){    php_info_print_table_start();    php_info_print_table_header(2, "xing2233 support", "enabled");    php_info_print_table_end();    /* Remove comments if you have entries in php.ini    DISPLAY_INI_ENTRIES();    */}/* }}} *//* {{{ xing2233_functions[] * * Every user visible function must have an entry in xing2233_functions[]. */const zend_function_entry xing2233_functions[] = {    PHP_FE(confirm_xing2233_compiled,   NULL)       /* For testing, remove later. */    PHP_FE_END  /* Must be the last line in xing2233_functions[] */};/* }}} *//* {{{ xing2233_module_entry */zend_module_entry xing2233_module_entry = {    STANDARD_MODULE_HEADER,    "xing2233",    xing2233_functions,    PHP_MINIT(xing2233),    PHP_MSHUTDOWN(xing2233),    PHP_RINIT(xing2233),        /* Replace with NULL if there's nothing to do at request start */    PHP_RSHUTDOWN(xing2233),    /* Replace with NULL if there's nothing to do at request end */    PHP_MINFO(xing2233),    PHP_XING2233_VERSION,    STANDARD_MODULE_PROPERTIES};/* }}} */#ifdef COMPILE_DL_XING2233ZEND_GET_MODULE(xing2233)#endif/* * Local variables: * tab-width: 4 * c-basic-offset: 4 * End: * vim600: noet sw=4 ts=4 fdm=marker * vim<600: noet sw=4 ts=4 */
原创粉丝点击