1 Star 0 Fork 345

十年砍材 / swoole-src

forked from swoole / swoole-src 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
swoole_event.cc 25.39 KB
一键复制 编辑 原始数据 按行查看 历史
韩天峰 提交于 2019-09-21 21:47 . use API
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849
/*
+----------------------------------------------------------------------+
| Swoole |
+----------------------------------------------------------------------+
| Copyright (c) 2012-2015 The Swoole Group |
+----------------------------------------------------------------------+
| This source file is subject to version 2.0 of the Apache 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.apache.org/licenses/LICENSE-2.0.html |
| If you did not receive a copy of the Apache2.0 license and are unable|
| to obtain it through the world-wide-web, please send a note to |
| license@swoole.com so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Author: Tianfeng Han <mikan.tenny@gmail.com> |
+----------------------------------------------------------------------+
*/
#include "php_swoole_cxx.h"
using namespace swoole;
zend_class_entry *swoole_event_ce;
static zend_object_handlers swoole_event_handlers;
typedef struct
{
zval zsocket;
zend_fcall_info_cache fci_cache_read;
zend_fcall_info_cache fci_cache_write;
} php_event_object;
static int php_swoole_event_onRead(swReactor *reactor, swEvent *event);
static int php_swoole_event_onWrite(swReactor *reactor, swEvent *event);
static int php_swoole_event_onError(swReactor *reactor, swEvent *event);
static void php_swoole_event_onDefer(void *data);
static void php_swoole_event_onEndCallback(void *data);
static PHP_FUNCTION(swoole_event_add);
static PHP_FUNCTION(swoole_event_set);
static PHP_FUNCTION(swoole_event_del);
static PHP_FUNCTION(swoole_event_write);
static PHP_FUNCTION(swoole_event_wait);
static PHP_FUNCTION(swoole_event_rshutdown);
static PHP_FUNCTION(swoole_event_exit);
static PHP_FUNCTION(swoole_event_defer);
static PHP_FUNCTION(swoole_event_cycle);
static PHP_FUNCTION(swoole_event_dispatch);
static PHP_FUNCTION(swoole_event_isset);
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_void, 0, 0, 0)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_add, 0, 0, 2)
ZEND_ARG_INFO(0, fd)
ZEND_ARG_CALLABLE_INFO(0, read_callback, 1)
ZEND_ARG_CALLABLE_INFO(0, write_callback, 1)
ZEND_ARG_INFO(0, events)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_set, 0, 0, 1)
ZEND_ARG_INFO(0, fd)
ZEND_ARG_CALLABLE_INFO(0, read_callback, 1)
ZEND_ARG_CALLABLE_INFO(0, write_callback, 1)
ZEND_ARG_INFO(0, events)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_write, 0, 0, 2)
ZEND_ARG_INFO(0, fd)
ZEND_ARG_INFO(0, data)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_defer, 0, 0, 1)
ZEND_ARG_CALLABLE_INFO(0, callback, 0)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_cycle, 0, 0, 1)
ZEND_ARG_CALLABLE_INFO(0, callback, 1)
ZEND_ARG_INFO(0, before)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_del, 0, 0, 1)
ZEND_ARG_INFO(0, fd)
ZEND_END_ARG_INFO()
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_event_isset, 0, 0, 1)
ZEND_ARG_INFO(0, fd)
ZEND_ARG_INFO(0, events)
ZEND_END_ARG_INFO()
static const zend_function_entry swoole_event_methods[] =
{
ZEND_FENTRY(add, ZEND_FN(swoole_event_add), arginfo_swoole_event_add, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(del, ZEND_FN(swoole_event_del), arginfo_swoole_event_del, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(set, ZEND_FN(swoole_event_set), arginfo_swoole_event_set, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(isset, ZEND_FN(swoole_event_isset), arginfo_swoole_event_isset, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(dispatch, ZEND_FN(swoole_event_dispatch), arginfo_swoole_void, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(defer, ZEND_FN(swoole_event_defer), arginfo_swoole_event_defer, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(cycle, ZEND_FN(swoole_event_cycle), arginfo_swoole_event_cycle, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(write, ZEND_FN(swoole_event_write), arginfo_swoole_event_write, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(wait, ZEND_FN(swoole_event_wait), arginfo_swoole_void, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(rshutdown, ZEND_FN(swoole_event_rshutdown), arginfo_swoole_void, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
ZEND_FENTRY(exit, ZEND_FN(swoole_event_exit), arginfo_swoole_void, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
PHP_FE_END
};
void php_swoole_event_minit(int module_number)
{
SW_INIT_CLASS_ENTRY(swoole_event, "Swoole\\Event", "swoole_event", NULL, swoole_event_methods);
SW_SET_CLASS_CREATE(swoole_event, sw_zend_create_object_deny);
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "add", CG(function_table), "swoole_event_add");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "del", CG(function_table), "swoole_event_del");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "set", CG(function_table), "swoole_event_set");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "isset", CG(function_table), "swoole_event_isset");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "dispatch", CG(function_table), "swoole_event_dispatch");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "defer", CG(function_table), "swoole_event_defer");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "cycle", CG(function_table), "swoole_event_cycle");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "write", CG(function_table), "swoole_event_write");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "wait", CG(function_table), "swoole_event_wait");
SW_FUNCTION_ALIAS(&swoole_event_ce->function_table, "exit", CG(function_table), "swoole_event_exit");
}
static void php_event_object_free(void* data)
{
php_event_object *peo = (php_event_object*) data;
if (peo->fci_cache_read.function_handler)
{
sw_zend_fci_cache_discard(&peo->fci_cache_read);
}
if (peo->fci_cache_write.function_handler)
{
sw_zend_fci_cache_discard(&peo->fci_cache_write);
}
zval_ptr_dtor((&peo->zsocket));
efree(peo);
}
static int php_swoole_event_onRead(swReactor *reactor, swEvent *event)
{
php_event_object *peo = (php_event_object *) event->socket->object;
if (UNEXPECTED(sw_zend_call_function_ex2(NULL, &peo->fci_cache_read, 1, &peo->zsocket, NULL) != SUCCESS))
{
php_swoole_fatal_error(E_WARNING, "%s: onRead callback handler error, fd [%d] will be removed from reactor", ZSTR_VAL(swoole_event_ce->name), swoole_convert_to_fd(&peo->zsocket));
event->socket->object = NULL;
swoole_event_defer(php_event_object_free, peo);
swoole_event_del(event->fd);
return SW_ERR;
}
return SW_OK;
}
static int php_swoole_event_onWrite(swReactor *reactor, swEvent *event)
{
php_event_object *peo = (php_event_object *) event->socket->object;
if (UNEXPECTED(sw_zend_call_function_ex2(NULL, &peo->fci_cache_write, 1, &peo->zsocket, NULL) != SUCCESS))
{
php_swoole_fatal_error(E_WARNING, "%s: onWrite callback handler error, fd [%d] will be removed from reactor", ZSTR_VAL(swoole_event_ce->name), swoole_convert_to_fd(&peo->zsocket));
event->socket->object = NULL;
swoole_event_defer(php_event_object_free, peo);
swoole_event_del(event->fd);
return SW_ERR;
}
return SW_OK;
}
static int php_swoole_event_onError(swReactor *reactor, swEvent *event)
{
if (!(event->socket->events & SW_EVENT_ERROR))
{
if (event->socket->events & SW_EVENT_READ)
{
return swReactor_get_handler(reactor, SW_EVENT_READ, event->socket->fdtype)(reactor, event);
}
else
{
return swReactor_get_handler(reactor, SW_EVENT_WRITE, event->socket->fdtype)(reactor, event);
}
}
int error;
socklen_t len = sizeof(error);
if (getsockopt(event->fd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)
{
php_swoole_sys_error(E_WARNING, "swoole_event->onError[1]: getsockopt[sock=%d] failed", event->fd);
}
if (error != 0)
{
php_swoole_fatal_error(E_WARNING, "swoole_event->onError[1]: socket error. Error: %s [%d]", strerror(error), error);
}
php_event_object_free(event->socket->object);
swoole_event_del(event->fd);
return SW_OK;
}
static void php_swoole_event_onDefer(void *data)
{
zend_fcall_info_cache *fci_cache = (zend_fcall_info_cache *) data;
if (UNEXPECTED(sw_zend_call_function_ex2(NULL, fci_cache, 0, NULL, NULL) != SUCCESS))
{
php_swoole_error(E_WARNING, "%s::defer callback handler error", ZSTR_VAL(swoole_event_ce->name));
}
sw_zend_fci_cache_discard(fci_cache);
efree(fci_cache);
}
static void php_swoole_event_onEndCallback(void *data)
{
zend_fcall_info_cache *fci_cache = (zend_fcall_info_cache *) data;
if (UNEXPECTED(sw_zend_call_function_ex2(NULL, (zend_fcall_info_cache *) fci_cache, 0, NULL, NULL) != SUCCESS))
{
php_swoole_error(E_WARNING, "%s::defer callback handler error", ZSTR_VAL(swoole_event_ce->name));
}
}
int php_swoole_reactor_init()
{
if (!SWOOLE_G(cli))
{
php_swoole_fatal_error(E_ERROR, "async-io must be used in PHP CLI mode");
return SW_ERR;
}
if (SwooleG.serv)
{
if (swIsTaskWorker() && !SwooleG.serv->task_enable_coroutine)
{
php_swoole_fatal_error(E_ERROR, "Unable to use async-io in task processes, please set `task_enable_coroutine` to true");
return SW_ERR;
}
if (swIsManager())
{
php_swoole_fatal_error(E_ERROR, "Unable to use async-io in manager process");
return SW_ERR;
}
}
if (!SwooleTG.reactor)
{
swTraceLog(SW_TRACE_PHP, "init reactor");
swoole_event_init();
SwooleTG.reactor->wait_exit = 1;
php_swoole_register_shutdown_function("Swoole\\Event::rshutdown");
}
return SW_OK;
}
void php_swoole_event_wait()
{
if (PG(last_error_message))
{
switch (PG(last_error_type))
{
case E_ERROR:
case E_CORE_ERROR:
case E_USER_ERROR:
case E_COMPILE_ERROR:
return;
default:
break;
}
}
if (!SwooleTG.reactor)
{
return;
}
#ifdef HAVE_SIGNALFD
if (SwooleTG.reactor->check_signalfd)
{
swSignalfd_setup(SwooleTG.reactor);
}
#endif
if (!swReactor_empty(SwooleTG.reactor))
{
// Don't disable object slot reuse while running shutdown functions:
// https://github.com/php/php-src/commit/bd6eabd6591ae5a7c9ad75dfbe7cc575fa907eac
#if defined(EG_FLAGS_IN_SHUTDOWN) && !defined(EG_FLAGS_OBJECT_STORE_NO_REUSE)
zend_bool in_shutdown = EG(flags) & EG_FLAGS_IN_SHUTDOWN;
EG(flags) &= ~EG_FLAGS_IN_SHUTDOWN;
#endif
int ret = SwooleTG.reactor->wait(SwooleTG.reactor, NULL);
if (ret < 0)
{
php_swoole_sys_error(E_ERROR, "reactor wait failed");
}
#if defined(EG_FLAGS_IN_SHUTDOWN) && !defined(EG_FLAGS_OBJECT_STORE_NO_REUSE)
if (in_shutdown)
{
EG(flags) |= EG_FLAGS_IN_SHUTDOWN;
}
#endif
}
swoole_event_free();
}
void php_swoole_event_exit()
{
if (SwooleTG.reactor)
{
SwooleTG.reactor->running = 0;
}
}
int swoole_convert_to_fd(zval *zsocket)
{
int fd = -1;
switch (Z_TYPE_P(zsocket))
{
case IS_RESOURCE:
{
php_stream *stream;
if ((php_stream_from_zval_no_verify(stream, zsocket)))
{
if (php_stream_cast(stream, PHP_STREAM_AS_FD_FOR_SELECT | PHP_STREAM_CAST_INTERNAL, (void **) &fd, 1) == SUCCESS && fd >= 0)
{
return fd;
}
}
#ifdef SWOOLE_SOCKETS_SUPPORT
else
{
php_socket *php_sock;
if ((php_sock = (php_socket *) zend_fetch_resource_ex(zsocket, NULL, php_sockets_le_socket())))
{
fd = php_sock->bsd_socket;
return fd;
}
}
#endif
php_swoole_fatal_error(E_WARNING, "fd argument must be either valid PHP stream or valid PHP socket resource");
return SW_ERR;
}
case IS_LONG:
{
fd = Z_LVAL_P(zsocket);
if (fd < 0)
{
php_swoole_fatal_error(E_WARNING, "invalid file descriptor#%d passed", fd);
return SW_ERR;
}
return fd;
}
case IS_OBJECT:
{
zval *zfd = NULL;
if (instanceof_function(Z_OBJCE_P(zsocket), swoole_socket_coro_ce))
{
zfd = sw_zend_read_property(Z_OBJCE_P(zsocket), zsocket, ZEND_STRL("fd"), 0);
}
else if (instanceof_function(Z_OBJCE_P(zsocket), swoole_client_ce))
{
zfd = sw_zend_read_property(Z_OBJCE_P(zsocket), zsocket, ZEND_STRL("sock"), 0);
}
else if (instanceof_function(Z_OBJCE_P(zsocket), swoole_process_ce))
{
zfd = sw_zend_read_property(Z_OBJCE_P(zsocket), zsocket, ZEND_STRL("pipe"), 0);
}
if (zfd == NULL || Z_TYPE_P(zfd) != IS_LONG)
{
return SW_ERR;
}
return Z_LVAL_P(zfd);
}
default:
php_swoole_fatal_error(E_WARNING, "invalid file descriptor passed");
return SW_ERR;
}
}
int swoole_convert_to_fd_ex(zval *zsocket, int *async)
{
int fd;
*async = 0;
if (Z_TYPE_P(zsocket) == IS_RESOURCE)
{
php_stream *stream;
if ((php_stream_from_zval_no_verify(stream, zsocket)))
{
if (php_stream_cast(stream, PHP_STREAM_AS_FD_FOR_SELECT | PHP_STREAM_CAST_INTERNAL, (void **)&fd, 1) == SUCCESS && fd >= 0)
{
*async = (stream->wrapper && (stream->wrapper->wops == php_plain_files_wrapper.wops)) ? 0 : 1;
return fd;
}
}
#ifdef SWOOLE_SOCKETS_SUPPORT
else
{
php_socket *php_sock;
if ((php_sock = (php_socket *) zend_fetch_resource_ex(zsocket, NULL, php_sockets_le_socket())))
{
fd = php_sock->bsd_socket;
*async = 1;
return fd;
}
}
#endif
}
php_swoole_fatal_error(E_WARNING, "fd argument must be either valid PHP stream or valid PHP socket resource");
return SW_ERR;
}
#ifdef SWOOLE_SOCKETS_SUPPORT
php_socket* swoole_convert_to_socket(int sock)
{
php_socket *socket_object = (php_socket *) emalloc(sizeof *socket_object);
bzero(socket_object, sizeof(php_socket));
socket_object->bsd_socket = sock;
socket_object->blocking = 1;
struct sockaddr_storage addr;
socklen_t addr_len = sizeof(addr);
if (getsockname(sock, (struct sockaddr*) &addr, &addr_len) == 0)
{
socket_object->type = addr.ss_family;
}
else
{
php_swoole_sys_error(E_WARNING, "unable to obtain socket family");
_error:
efree(socket_object);
return NULL;
}
int t = fcntl(sock, F_GETFL);
if (t == -1)
{
php_swoole_sys_error(E_WARNING, "unable to obtain blocking state");
goto _error;
}
else
{
socket_object->blocking = !(t & O_NONBLOCK);
}
return socket_object;
}
void swoole_php_socket_free(zval *zsocket)
{
php_socket *php_sock;
php_sock = (php_socket *) zend_fetch_resource_ex(zsocket, NULL, php_sockets_le_socket());
php_sock->bsd_socket = -1;
sw_zval_free(zsocket);
}
#endif
static void check_reactor()
{
php_swoole_check_reactor();
if (!swReactor_isset_handler(SwooleTG.reactor, SW_FD_USER))
{
swReactor_set_handler(SwooleTG.reactor, SW_FD_USER | SW_EVENT_READ, php_swoole_event_onRead);
swReactor_set_handler(SwooleTG.reactor, SW_FD_USER | SW_EVENT_WRITE, php_swoole_event_onWrite);
swReactor_set_handler(SwooleTG.reactor, SW_FD_USER | SW_EVENT_ERROR, php_swoole_event_onError);
}
}
static PHP_FUNCTION(swoole_event_add)
{
zval *zfd;
zend_fcall_info fci_read = empty_fcall_info;
zend_fcall_info_cache fci_cache_read = empty_fcall_info_cache;
zend_fcall_info fci_write = empty_fcall_info;
zend_fcall_info_cache fci_cache_write = empty_fcall_info_cache;
zend_long events = 0;
ZEND_PARSE_PARAMETERS_START(1, 4)
Z_PARAM_ZVAL(zfd)
Z_PARAM_OPTIONAL
Z_PARAM_FUNC_EX(fci_read, fci_cache_read, 1, 0)
Z_PARAM_FUNC_EX(fci_write, fci_cache_write, 1, 0)
Z_PARAM_LONG(events)
ZEND_PARSE_PARAMETERS_END_EX(RETURN_FALSE);
if (fci_read.size == 0 && fci_write.size == 0)
{
php_swoole_fatal_error(E_WARNING, "both read and write callbacks are emoty");
RETURN_FALSE;
}
int socket_fd = swoole_convert_to_fd(zfd);
if (socket_fd < 0)
{
php_swoole_fatal_error(E_WARNING, "unknow fd type");
RETURN_FALSE;
}
if (socket_fd == 0 && (events & SW_EVENT_WRITE))
{
php_swoole_fatal_error(E_WARNING, "invalid socket fd [%d]", socket_fd);
RETURN_FALSE;
}
php_event_object *peo = (php_event_object *) ecalloc(1, sizeof(php_event_object));
Z_TRY_ADDREF_P(zfd);
peo->zsocket = *zfd;
if (fci_read.size != 0)
{
sw_zend_fci_cache_persist(&fci_cache_read);
peo->fci_cache_read = fci_cache_read;
}
if (fci_write.size != 0)
{
sw_zend_fci_cache_persist(&fci_cache_write);
peo->fci_cache_write = fci_cache_write;
}
check_reactor();
swSocket_set_nonblock(socket_fd); // must be nonblock
if (swoole_event_add(socket_fd, events, SW_FD_USER) < 0)
{
php_swoole_fatal_error(E_WARNING, "swoole_event_add failed");
RETURN_FALSE;
}
swSocket *socket = swReactor_get(SwooleTG.reactor, socket_fd);
socket->object = peo;
socket->nonblock = 1;
socket->fdtype = SW_FD_USER;
RETURN_LONG(socket_fd);
}
static PHP_FUNCTION(swoole_event_write)
{
zval *zfd;
char *data;
size_t len;
if (zend_parse_parameters(ZEND_NUM_ARGS(), "zs", &zfd, &data, &len) == FAILURE)
{
RETURN_FALSE;
}
if (len == 0)
{
php_swoole_fatal_error(E_WARNING, "data empty");
RETURN_FALSE;
}
int socket_fd = swoole_convert_to_fd(zfd);
if (socket_fd < 0)
{
php_swoole_fatal_error(E_WARNING, "unknow type");
RETURN_FALSE;
}
check_reactor();
if (swoole_event_write(socket_fd, data, len) < 0)
{
RETURN_FALSE;
}
else
{
RETURN_TRUE;
}
}
static PHP_FUNCTION(swoole_event_set)
{
if (!SwooleTG.reactor)
{
php_swoole_fatal_error(E_WARNING, "reactor is not ready, cannot call swoole_event_set");
RETURN_FALSE;
}
zval *zfd;
zend_fcall_info fci_read = empty_fcall_info;
zend_fcall_info_cache fci_cache_read = empty_fcall_info_cache;
zend_fcall_info fci_write = empty_fcall_info;
zend_fcall_info_cache fci_cache_write = empty_fcall_info_cache;
zend_long events = 0;
ZEND_PARSE_PARAMETERS_START(1, 4)
Z_PARAM_ZVAL(zfd)
Z_PARAM_OPTIONAL
Z_PARAM_FUNC_EX(fci_read, fci_cache_read, 1, 0)
Z_PARAM_FUNC_EX(fci_write, fci_cache_write, 1, 0)
Z_PARAM_LONG(events)
ZEND_PARSE_PARAMETERS_END_EX(RETURN_FALSE);
int socket_fd = swoole_convert_to_fd(zfd);
if (socket_fd < 0)
{
php_swoole_fatal_error(E_WARNING, "unknow type");
RETURN_FALSE;
}
swSocket *socket = swReactor_get(SwooleTG.reactor, socket_fd);
if (socket->fdtype != SW_FD_USER || socket->removed)
{
php_swoole_fatal_error(E_WARNING, "socket[%d] is not found in the reactor", socket_fd);
RETURN_FALSE;
}
php_event_object *reactor_fd = (php_event_object *) socket->object;
if (fci_read.size != 0)
{
if (reactor_fd->fci_cache_read.function_handler)
{
sw_zend_fci_cache_discard(&reactor_fd->fci_cache_read);
}
reactor_fd->fci_cache_read = fci_cache_read;
}
if (fci_write.size != 0)
{
if (reactor_fd->fci_cache_write.function_handler)
{
sw_zend_fci_cache_discard(&reactor_fd->fci_cache_write);
}
reactor_fd->fci_cache_write = fci_cache_write;
}
if ((events & SW_EVENT_READ) && reactor_fd->fci_cache_read.function_handler == NULL)
{
php_swoole_fatal_error(E_WARNING, "%s: unable to find read callback of fd [%d]", ZSTR_VAL(swoole_event_ce->name), socket_fd);
RETURN_FALSE;
}
if ((events & SW_EVENT_WRITE) && reactor_fd->fci_cache_write.function_handler == NULL)
{
php_swoole_fatal_error(E_WARNING, "%s: unable to find write callback of fd [%d]", ZSTR_VAL(swoole_event_ce->name), socket_fd);
RETURN_FALSE;
}
if (swoole_event_set(socket_fd, events, SW_FD_USER) < 0)
{
php_swoole_fatal_error(E_WARNING, "%s::set failed", ZSTR_VAL(swoole_event_ce->name));
RETURN_FALSE;
}
RETURN_TRUE;
}
static PHP_FUNCTION(swoole_event_del)
{
zval *zfd;
if (!SwooleTG.reactor)
{
php_swoole_fatal_error(E_WARNING, "reactor is not ready, cannot call swoole_event_del");
RETURN_FALSE;
}
if (zend_parse_parameters(ZEND_NUM_ARGS(), "z", &zfd) == FAILURE)
{
RETURN_FALSE;
}
int socket_fd = swoole_convert_to_fd(zfd);
if (socket_fd < 0)
{
php_swoole_fatal_error(E_WARNING, "unknow type");
RETURN_FALSE;
}
swSocket *socket = swReactor_get(SwooleTG.reactor, socket_fd);
if (socket->object)
{
swoole_event_defer(php_event_object_free, socket->object);
socket->object = NULL;
}
SW_CHECK_RETURN(swoole_event_del(socket_fd));
}
static PHP_FUNCTION(swoole_event_defer)
{
zend_fcall_info fci = empty_fcall_info;
zend_fcall_info_cache *fci_cache = (zend_fcall_info_cache *) ecalloc(1, sizeof(zend_fcall_info_cache));
ZEND_PARSE_PARAMETERS_START(1, 1)
Z_PARAM_FUNC(fci, *fci_cache)
ZEND_PARSE_PARAMETERS_END_EX(efree(fci_cache); RETURN_FALSE);
php_swoole_check_reactor();
sw_zend_fci_cache_persist(fci_cache);
swoole_event_defer(php_swoole_event_onDefer, fci_cache);
RETURN_TRUE;
}
static PHP_FUNCTION(swoole_event_cycle)
{
if (!SwooleTG.reactor)
{
php_swoole_fatal_error(E_WARNING, "reactor is not ready, cannot call %s", ZSTR_VAL(swoole_event_ce->name));
RETURN_FALSE;
}
zend_fcall_info _fci = empty_fcall_info;
zend_fcall_info_cache _fci_cache = empty_fcall_info_cache;
zend_bool before = 0;
ZEND_PARSE_PARAMETERS_START(1, 2)
Z_PARAM_FUNC_EX(_fci, _fci_cache, 1, 0)
Z_PARAM_OPTIONAL
Z_PARAM_BOOL(before)
ZEND_PARSE_PARAMETERS_END_EX(RETURN_FALSE);
if (_fci.size == 0)
{
if (SwooleTG.reactor->idle_task.callback == NULL)
{
RETURN_FALSE;
}
else
{
swoole_event_defer(sw_zend_fci_cache_free, SwooleTG.reactor->idle_task.data);
SwooleTG.reactor->idle_task.callback = NULL;
SwooleTG.reactor->idle_task.data = NULL;
RETURN_TRUE;
}
}
zend_fcall_info_cache *fci_cache = (zend_fcall_info_cache *) emalloc(sizeof(zend_fcall_info_cache));
*fci_cache = _fci_cache;
sw_zend_fci_cache_persist(fci_cache);
if (!before)
{
if (SwooleTG.reactor->idle_task.data != NULL)
{
swoole_event_defer(sw_zend_fci_cache_free, SwooleTG.reactor->idle_task.data);
}
SwooleTG.reactor->idle_task.callback = php_swoole_event_onEndCallback;
SwooleTG.reactor->idle_task.data = fci_cache;
}
else
{
if (SwooleTG.reactor->future_task.data != NULL)
{
swoole_event_defer(sw_zend_fci_cache_free, SwooleTG.reactor->future_task.data);
}
SwooleTG.reactor->future_task.callback = php_swoole_event_onEndCallback;
SwooleTG.reactor->future_task.data = fci_cache;
//Registration onBegin callback function
swReactor_activate_future_task(SwooleTG.reactor);
}
RETURN_TRUE;
}
static PHP_FUNCTION(swoole_event_exit)
{
php_swoole_event_exit();
}
static PHP_FUNCTION(swoole_event_wait)
{
if (!SwooleTG.reactor)
{
return;
}
php_swoole_event_wait();
}
static PHP_FUNCTION(swoole_event_rshutdown)
{
/* prevent the program from jumping out of the rshutdown */
zend_try
{
PHP_FN(swoole_event_wait)(INTERNAL_FUNCTION_PARAM_PASSTHRU);
}
zend_end_try();
}
static PHP_FUNCTION(swoole_event_dispatch)
{
if (!SwooleTG.reactor)
{
RETURN_FALSE;
}
SwooleTG.reactor->once = 1;
#ifdef HAVE_SIGNALFD
if (SwooleTG.reactor->check_signalfd)
{
swSignalfd_setup(SwooleTG.reactor);
}
#endif
int ret = SwooleTG.reactor->wait(SwooleTG.reactor, NULL);
if (ret < 0)
{
php_swoole_sys_error(E_ERROR, "reactor wait failed");
}
SwooleTG.reactor->once = 0;
RETURN_TRUE;
}
static PHP_FUNCTION(swoole_event_isset)
{
if (!SwooleTG.reactor)
{
RETURN_FALSE;
}
zval *zfd;
zend_long events = SW_EVENT_READ | SW_EVENT_WRITE;
if (zend_parse_parameters(ZEND_NUM_ARGS(), "z|l", &zfd, &events) == FAILURE)
{
RETURN_FALSE;
}
int socket_fd = swoole_convert_to_fd(zfd);
if (socket_fd < 0)
{
php_swoole_fatal_error(E_WARNING, "unknow type");
RETURN_FALSE;
}
swSocket *_socket = swReactor_get(SwooleTG.reactor, socket_fd);
if (_socket == NULL || _socket->removed)
{
RETURN_FALSE;
}
if (_socket->events & events)
{
RETURN_TRUE;
}
else
{
RETURN_FALSE;
}
}
C
1
https://gitee.com/JackZhang_2016/swoole.git
git@gitee.com:JackZhang_2016/swoole.git
JackZhang_2016
swoole
swoole-src
master

搜索帮助