// Ivy.cpp: implementation of the Ivy class. // ////////////////////////////////////////////////////////////////////// #include "IvyStdAfx.h" #include "Ivy.h" #include "IvyWatcher.h" #include "IvyApplication.h" #include "IvySynchroWnd.h" #define DEFAULT_ADDR "127.255.255.255" #define SEPARATOR ":" #define DEFAULT_PORT "2010" #define DEFAULT_DOMAIN DEFAULT_ADDR/**/SEPARATOR/**/DEFAULT_PORT ////////////////////////////////////////////////////////////////////// // Construction/Destruction ////////////////////////////////////////////////////////////////////// Ivy::~Ivy() { // remove all app and stop watcher stop(); regexp_out.clear(); if ( synchronous ) { for ( unsigned int i = 0; i < callbacks.size(); i ++) { delete callbacks[i]; } } callbacks.clear(); // if ( direct_callback ) delete direct_callback; // if ( application_callback ) delete application_callback; delete watcher; // server->Close(); delete server; if ( synchronous ) { delete IvySynchronousCallback::m_synchro; delete application_callback; } } Ivy::Ivy(const char* name, const char * ready, IvyApplicationCallback *callback, bool Synchronous) { InitializeCriticalSection( &m_application_cs ); regexp_id = 0; synchronous = Synchronous; if ( synchronous ) IvySynchronousCallback::m_synchro = new IvySynchroWnd(); ready_message = ready; appname = name; messages_classes_count = 0; messages_classes = NULL; application_callback = synchronous ? new IvySynchronousApplicationCallback(callback) : callback; direct_callback = NULL; die_callback = NULL; server = new IvyApplication(this); applicationPort = server->Create(); if ( !applicationPort ) { TRACE( " Can't Create server %d\n", server->GetLastError( ) ); return; } watcher = new IvyWatcher(this); } const char * Ivy::GetDomain(const char *domainlist) { // determine domain to use // the syntax of domain is "IpBroadcastAddr1,IpBroadcastAddr2,IpBroadcastAddr2:port" if ( domainlist ) { domain = domainlist; } if ( domain.empty() ) { #ifndef UNDER_CE size_t requiredSize; getenv_s( &requiredSize, NULL, 0, "IVYBUS"); if ( requiredSize ) { char *env = (char*)malloc( requiredSize * sizeof(char)); getenv_s( &requiredSize, env, requiredSize, "IVYBUS"); domain = env; free( env ); } #endif if ( domain.empty() ) domain = DEFAULT_DOMAIN; } // first find our UDP port size_t sep_index = domain.rfind( ':' ); if ( sep_index == -1 ) { domain = DEFAULT_DOMAIN; TRACE(" Missing ':' in domain list using default domain %s\n", domain ); } if ( sep_index == 0 ) { /* missing addr using localhost */ domain.insert(0,DEFAULT_ADDR); } return domain.c_str(); } void Ivy::start(const char *domain) { watcher->start(domain); } void Ivy::stop() { watcher->stop(); IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ++iter ) { IvyApplication *app = *iter; app->Close(); delete app; } applications.clear(); } int Ivy::BindMsg(const char *regexp, IvyMessageCallback *cb) { regexp_out.push_back( regexp ); callbacks.push_back( synchronous ? new IvySynchronousMessageCallback(cb) : cb ); /* send to already connected */ IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ++iter ) { IvyApplication *app = *iter; app->SendMsg(IvyApplication::AddRegexp, regexp_id, regexp ); } return regexp_id++; } int Ivy::BindMsg( IvyMessageCallback *cb, const char *regexp, ... ) { char buffer[4096]; va_list args; va_start( args, regexp ); /* Initialize variable arguments. */ _vsnprintf_s( buffer, sizeof(buffer), sizeof(buffer)-1, regexp, args ); va_end( args); regexp_out.push_back( regexp ); callbacks.push_back( synchronous ? new IvySynchronousMessageCallback(cb) : cb ); /* send to already connected */ IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ++iter ) { IvyApplication *app = *iter; app->SendMsg(IvyApplication::AddRegexp, regexp_id, buffer ); } return regexp_id++; } void Ivy::UnbindMsg(int id) { regexp_out[ id ] = ""; callbacks[ id ] = NULL; /* send to already connected */ IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ++iter ) { IvyApplication *app = *iter; app->SendMsg(IvyApplication::DelRegexp, id, "" ); } } void Ivy::BindDirectMsg(IvyDirectMessageCallback *callback) { direct_callback = callback; } unsigned int Ivy::GetApplicationPort() { return applicationPort; } void Ivy::AddApplication(IvyApplication *app) { EnterCriticalSection( &m_application_cs ); // Check for disconnected Application IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ) { IvyApplication *disc_app = *iter++; if ( disc_app->m_hSocket == INVALID_SOCKET ) { applications.remove( disc_app ); delete disc_app; } } applications.push_back( app ); LeaveCriticalSection( &m_application_cs ); SendSubscriptions( app ); } void Ivy::RemoveApplication(IvyApplication * app) { /// OLD NOT called because of deallocation PB // the real remove is done at arrival of a new Application // or at the bus Stop ASSERT( TRUE ); if ( app ) { EnterCriticalSection( &m_application_cs ); applications.remove( app ); LeaveCriticalSection( &m_application_cs ); delete app; } } void Ivy::SendSubscriptions(IvyApplication *app) { app->SendMsg( IvyApplication::StartRegexp, GetApplicationPort(), appname.c_str()); for ( unsigned int id = 0 ; id < regexp_out.size(); id++ ) { const ivy::string& regexp = regexp_out[id]; if ( !regexp.empty() ) app->SendMsg( IvyApplication::AddRegexp, id, regexp.c_str()); } app->SendMsg( IvyApplication::EndRegexp, 0); } int Ivy::SendMsg(const char * message, ... ) { int count = 0; char buffer[4096]; va_list args; va_start( args, message ); /* Initialize variable arguments. */ _vsnprintf_s( buffer, sizeof(buffer), sizeof(buffer)-1, message, args ); va_end( args); /* send to already connected */ IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ++iter ) { IvyApplication *app = *iter; count += app->SendMsg( buffer ); } return count; } void Ivy::SendDieMsg( IvyApplication *app ) { app->SendMsg( IvyApplication::Die, 0 ); } IvyApplication * Ivy::GetApplication(const char *name) { IvyApplication *app = NULL; EnterCriticalSection( &m_application_cs ); IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ) { IvyApplication *ap = *iter++; if ( (ap->m_hSocket != INVALID_SOCKET) && ap->appname == name ) { app = ap; break; // dont return because of LeaveCriticalSection !!! } } LeaveCriticalSection( &m_application_cs ); return app; } void Ivy::CallMessageCallback(IvyApplication *app, int id, int argc, const char ** argv) { IvyMessageCallback *callback; callback = callbacks[ id ]; if ( callback ) { callback->OnMessage( app, argc, argv ); } } void Ivy::CallDirectMessageCallback(IvyApplication *app, int id, const char *arg) { if ( direct_callback ) { direct_callback->OnDirectMessage( app, id, arg ); } } bool Ivy::CallDieCallback(IvyApplication *app, int id, const char *arg) { if ( die_callback ) { return die_callback->OnDie( app, id, arg ); } return TRUE; } void Ivy::CallApplicationConnectedCallback(IvyApplication * app) { if ( application_callback ) { application_callback->OnApplicationConnected( app ); } } void Ivy::CallApplicationDisconnectedCallback(IvyApplication * app) { if ( application_callback ) { application_callback->OnApplicationDisconnected( app ); } } void Ivy::SendDirectMsg(IvyApplication * app, int id, const char *message) { app->SendMsg( IvyApplication::DirectMsg, id, message ); } bool Ivy::CheckRegexp(const char * exp) { /* accepte tout par default */ int i; bool regexp_ok = true; if ( *exp =='^' && messages_classes_count !=0 ) { regexp_ok = false; for ( i = 0 ; i < messages_classes_count; i++ ) { if (strncmp( messages_classes[i], exp+1, strlen( messages_classes[i] )) == 0) return 1; } } return regexp_ok; } void Ivy::Classes(int argc, const char **argv ) { messages_classes_count = argc; messages_classes = argv; } bool Ivy::CheckConnected(IvyApplication * app) { if (app->remoteService == 0) /* old application dont check */ return false; /* check to see if app already connected */ IvyApplicationList::iterator iter; for ( iter = applications.begin() ; iter != applications.end() ; ++iter ) { IvyApplication *application = *iter; if ( application != app && application->SameApplication(app)) return true; } return false; }