[Freeswitch-trunk] [commit] r13901 - freeswitch/trunk/src/mod/event_handlers/mod_event_multicast

[FreeSWITCH SVN]

Author: andrew
Date: Mon Jun 22 19:59:03 2009
New Revision: 13901

Log:
Add initial support for multicast peer up/down state tracking


Modified:
   freeswitch/trunk/src/mod/event_handlers/mod_event_multicast/mod_event_multicast.c

Modified: freeswitch/trunk/src/mod/event_handlers/mod_event_multicast/mod_event_multicast.c
==============================================================================
--- freeswitch/trunk/src/mod/event_handlers/mod_event_multicast/mod_event_multicast.c	(original)
+++ freeswitch/trunk/src/mod/event_handlers/mod_event_multicast/mod_event_multicast.c	Mon Jun 22 19:59:03 2009
@@ -63,14 +63,22 @@
 	uint8_t ttl;
 	char *psk;
 	switch_mutex_t *mutex;
+	switch_hash_t *peer_hash;
 } globals;
 
+struct peer_status {
+	switch_bool_t active;
+	int lastseen;
+};
+
 SWITCH_DECLARE_GLOBAL_STRING_FUNC(set_global_address, globals.address);
 SWITCH_DECLARE_GLOBAL_STRING_FUNC(set_global_bindings, globals.bindings);
 #ifdef HAVE_OPENSSL
 SWITCH_DECLARE_GLOBAL_STRING_FUNC(set_global_psk, globals.psk);
 #endif
 #define MULTICAST_EVENT "multicast::event"
+#define MULTICAST_PEERUP "multicast::peerup"
+#define MULTICAST_PEERDOWN "multicast::peerdown"
 static switch_status_t load_config(void)
 {
 	switch_status_t status = SWITCH_STATUS_SUCCESS;
@@ -169,6 +177,42 @@
 	}
 
 	if (event->subclass_name && !strcmp(event->subclass_name, MULTICAST_EVENT)) {
+		char * event_name;
+		if ((event_name = switch_event_get_header(event, "orig-event-name")) && !strcasecmp(event_name, "HEARTBEAT")) {
+			char *sender = switch_event_get_header(event, "orig-multicast-sender");
+			struct peer_status *p;
+			int now = switch_epoch_time_now(NULL);
+			
+			if (!(p = switch_core_hash_find(globals.peer_hash, sender))) {
+				switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Host %s not already in hash\n", sender);
+				p = switch_core_alloc(module_pool, sizeof(struct peer_status));
+				p->active = SWITCH_FALSE;
+				p->lastseen = 0;
+			/*} else {*/
+				/*switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "Host %s last seen %d seconds ago\n", sender, now - p->lastseen);*/
+			}
+
+			if (!p->active) {
+				switch_event_t *local_event;
+				if (switch_event_create_subclass(&local_event, SWITCH_EVENT_CUSTOM, MULTICAST_PEERUP) == SWITCH_STATUS_SUCCESS) {
+					char lastseen[21];
+					switch_event_add_header_string(local_event, SWITCH_STACK_BOTTOM, "Peer", sender);
+					if (p->lastseen) {
+						switch_snprintf(lastseen, sizeof(lastseen), "%d", p->lastseen);
+					} else {
+						switch_snprintf(lastseen, sizeof(lastseen), "%s", "Never");
+					}
+					switch_event_add_header_string(local_event, SWITCH_STACK_BOTTOM, "Lastseen", lastseen);
+					switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Peer %s has come up; last seen: %s\n", sender, lastseen);
+
+					switch_event_fire(&local_event);
+				}
+			}
+			p->active = SWITCH_TRUE;
+			p->lastseen = now;
+
+			switch_core_hash_insert(globals.peer_hash, sender, p);
+		}
 		/* ignore our own events to avoid ping pong */
 		return;
 	}
@@ -187,6 +231,36 @@
 		switch_mutex_unlock(globals.mutex);
 	}
 
+	if (event->event_id == SWITCH_EVENT_HEARTBEAT) {
+		switch_hash_index_t *cur;
+		switch_ssize_t keylen;
+		const void *key;
+		void *value;
+		int now = switch_epoch_time_now(NULL);
+		struct peer_status *last;
+		char *host;
+		
+		for (cur = switch_hash_first(NULL, globals.peer_hash); cur; cur = switch_hash_next(cur)) {
+			switch_hash_this(cur, &key, &keylen, &value);
+			host = (char*) key;
+			last = (struct peer_status*) value;
+			if (last->active && (now - (last->lastseen)) > 60) {
+				switch_event_t *local_event;
+
+				last->active = SWITCH_FALSE;
+				if (switch_event_create_subclass(&local_event, SWITCH_EVENT_CUSTOM, MULTICAST_PEERDOWN) == SWITCH_STATUS_SUCCESS) {
+					char lastseen[21];
+					switch_event_add_header_string(local_event, SWITCH_STACK_BOTTOM, "Peer", host);
+					switch_snprintf(lastseen, sizeof(lastseen), "%d", last->lastseen);
+					switch_event_add_header_string(local_event, SWITCH_STACK_BOTTOM, "Lastseen", lastseen);
+					switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Peer %s has gone down; last seen: %s\n", host, lastseen);
+
+					switch_event_fire(&local_event);
+				}
+			}
+		}
+	}
+
 	switch_mutex_lock(globals.mutex);
 	if (globals.event_list[(uint8_t) SWITCH_EVENT_ALL]) {
 		send = 1;
@@ -271,6 +345,7 @@
 	module_pool = pool;
 
 	switch_core_hash_init(&globals.event_hash, module_pool);
+	switch_core_hash_init(&globals.peer_hash, module_pool);
 
 	gethostname(globals.hostname, sizeof(globals.hostname));
 	globals.host_hash = switch_hashfunc_default(globals.hostname, &hlen);
@@ -319,7 +394,17 @@
 	*module_interface = switch_loadable_module_create_module_interface(pool, modname);
 
 	if (switch_event_reserve_subclass(MULTICAST_EVENT) != SWITCH_STATUS_SUCCESS) {
-		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't register subclass!\n");
+		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't register subclass %s!\n", MULTICAST_EVENT);
+		return SWITCH_STATUS_GENERR;
+	}
+
+	if (switch_event_reserve_subclass(MULTICAST_PEERUP) != SWITCH_STATUS_SUCCESS) {
+		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't register subclass %s!\n", MULTICAST_PEERUP);
+		return SWITCH_STATUS_GENERR;
+	}
+
+	if (switch_event_reserve_subclass(MULTICAST_PEERDOWN) != SWITCH_STATUS_SUCCESS) {
+		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldn't register subclass %s!\n", MULTICAST_PEERDOWN);
 		return SWITCH_STATUS_GENERR;
 	}