Import recent fixes from mod_event_socket, work on cleanup on unload (still segfaults tho)

git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@10301 d0543943-73ff-0310-b7d9-9358b9ac24b2
This commit is contained in:
Andrew Thompson 2008-11-07 23:56:50 +00:00
parent 5bc6dc2c63
commit 08f8cb41d8
1 changed files with 56 additions and 32 deletions

View File

@ -91,7 +91,6 @@ typedef struct listener listener_t;
static struct {
int sockfd;
switch_mutex_t *mutex;
switch_mutex_t *sock_mutex;
listener_t *listeners;
uint8_t ready;
@ -124,6 +123,7 @@ static void *SWITCH_THREAD_FUNC listener_run(switch_thread_t *thread, void *obj)
static void launch_listener_thread(listener_t *listener);
static struct {
switch_mutex_t *listener_mutex;
switch_event_node_t *node;
} globals;
@ -132,7 +132,7 @@ static switch_status_t socket_logger(const switch_log_node_t *node, switch_log_l
{
listener_t *l;
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
for (l = listen_list.listeners; l; l = l->next) {
if (switch_test_flag(l, LFLAG_LOG) && l->level >= node->level) {
char *data = strdup(node->data);
@ -157,7 +157,7 @@ static switch_status_t socket_logger(const switch_log_node_t *node, switch_log_l
}
}
}
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
return SWITCH_STATUS_SUCCESS;
}
@ -223,7 +223,7 @@ static void event_handler(switch_event_t *event)
lp = listen_list.listeners;
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
while(lp) {
uint8_t send = 0;
@ -279,7 +279,19 @@ static void event_handler(switch_event_t *event)
}
}
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
}
static void close_socket(int *sock)
{
switch_mutex_lock(listen_list.sock_mutex);
if (*sock) {
shutdown(*sock, SHUT_RDWR);
close(*sock);
sock = NULL;
}
switch_mutex_unlock(listen_list.sock_mutex);
}
@ -292,23 +304,24 @@ SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_erlang_event_shutdown)
switch_log_unbind_logger(socket_logger);
close(listen_list.sockfd);
/*close_socket(&listen_list.sockfd);*/
switch_yield(500);
while (prefs.threads) {
while (prefs.threads || prefs.done == 1) {
switch_yield(10000);
if (++sanity == 1000) {
break;
}
}
switch_event_unbind(&globals.node);
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
for (l = listen_list.listeners; l; l = l->next) {
close(l->sockfd);
close_socket(&l->sockfd);
}
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
return SWITCH_STATUS_SUCCESS;
}
@ -317,10 +330,10 @@ SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_erlang_event_shutdown)
static void add_listener(listener_t *listener)
{
/* add me to the listeners so I get events */
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
listener->next = listen_list.listeners;
listen_list.listeners = listener;
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
}
@ -328,7 +341,7 @@ static void remove_listener(listener_t *listener)
{
listener_t *l, *last = NULL;
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
for (l = listen_list.listeners; l; l = l->next) {
if (l == listener) {
if (last) {
@ -339,11 +352,21 @@ static void remove_listener(listener_t *listener)
}
last = l;
}
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
}
SWITCH_MODULE_LOAD_FUNCTION(mod_erlang_event_load)
{
switch_mutex_init(&globals.listener_mutex, SWITCH_MUTEX_NESTED, pool);
if (switch_event_bind_removable(modname, SWITCH_EVENT_ALL, SWITCH_EVENT_SUBCLASS_ANY, event_handler, NULL, &globals.node) != SWITCH_STATUS_SUCCESS) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't bind!\n");
close_socket(&listen_list.sockfd);
return SWITCH_STATUS_GENERR;
}
switch_log_bind_logger(socket_logger, SWITCH_LOG_DEBUG, SWITCH_FALSE);
/* connect my internal structure to the blank pointer passed to me */
*module_interface = switch_loadable_module_create_module_interface(pool, modname);
@ -830,9 +853,9 @@ static void *SWITCH_THREAD_FUNC listener_run(switch_thread_t *thread, void *obj)
int status = 1;
void *pop;
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
prefs.threads++;
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
switch_assert(listener != NULL);
@ -984,7 +1007,7 @@ done:
switch_thread_rwlock_wrlock(listener->rwlock);
if (listener->sockfd) {
close(listener->sockfd);
close_socket(&listener->sockfd);
}
switch_thread_rwlock_unlock(listener->rwlock);
@ -1001,9 +1024,9 @@ done:
switch_core_destroy_memory_pool(&pool);
}
switch_mutex_lock(listen_list.mutex);
switch_mutex_lock(globals.listener_mutex);
prefs.threads--;
switch_mutex_unlock(listen_list.mutex);
switch_mutex_unlock(globals.listener_mutex);
return NULL;
}
@ -1092,6 +1115,7 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
struct sockaddr_in server_addr;
int on = 1;
int clientfd;
int epmdfd;
memset(&listen_list, 0, sizeof(listen_list));
config();
@ -1101,7 +1125,6 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
return SWITCH_STATUS_TERM;
}
switch_mutex_init(&listen_list.mutex, SWITCH_MUTEX_NESTED, pool);
switch_mutex_init(&listen_list.sock_mutex, SWITCH_MUTEX_NESTED, pool);
/* zero out the struct before we use it */
@ -1129,7 +1152,7 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
goto sock_fail;
}
if (setsockopt(listen_list.sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) != 0) {
if (setsockopt(listen_list.sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Failed to enable SO_REUSEADDR for socket on %s:%u : %s\n", prefs.ip, prefs.port, strerror(errno));
goto sock_fail;
}
@ -1170,12 +1193,15 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
/* init the ei stuff */
if (ei_connect_xinit(&ec, thishostname, prefs.nodename, thisnodename, (Erl_IpAddr)(&server_addr.sin_addr.s_addr), prefs.cookie, 0) < 0) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Failed to init ei connection\n");
close_socket(&listen_list.sockfd);
return SWITCH_STATUS_GENERR;
}
/* return value is -1 for error, a descriptor pointing to epmd otherwise */
if (ei_publish(&ec, prefs.port) == -1) {
if ((epmdfd = ei_publish(&ec, prefs.port)) == -1) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Failed to publish port to empd\n");
/* TODO - start epmd? */
close_socket(&listen_list.sockfd);
return SWITCH_STATUS_GENERR;
}
@ -1183,13 +1209,6 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
listen_list.ready = 1;
if (switch_event_bind_removable(modname, SWITCH_EVENT_ALL, SWITCH_EVENT_SUBCLASS_ANY, event_handler, NULL, &globals.node) != SWITCH_STATUS_SUCCESS) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't bind!\n");
return SWITCH_STATUS_GENERR;
}
switch_log_bind_logger(socket_logger, SWITCH_LOG_DEBUG, SWITCH_FALSE);
for (;;) {
/* zero out errno because ei_accept doesn't differentiate between a
* failed authentication or a socket failure, or a client version
@ -1239,7 +1258,11 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
}
close(listen_list.sockfd);
/* cleanup epmd registration */
ei_unpublish(&ec);
close(epmdfd);
close_socket(&listen_list.sockfd);
if (pool) {
switch_core_destroy_memory_pool(&pool);
@ -1254,6 +1277,7 @@ SWITCH_MODULE_RUNTIME_FUNCTION(mod_erlang_event_runtime)
switch_safe_free(prefs.acl[x]);
}
prefs.done = 2;
fail:
return SWITCH_STATUS_TERM;
}