diff options
Diffstat (limited to 'Src/replicant/Wasabi/ServiceManager.cpp')
| -rw-r--r-- | Src/replicant/Wasabi/ServiceManager.cpp | 201 |
1 files changed, 201 insertions, 0 deletions
diff --git a/Src/replicant/Wasabi/ServiceManager.cpp b/Src/replicant/Wasabi/ServiceManager.cpp new file mode 100644 index 00000000..2d6887aa --- /dev/null +++ b/Src/replicant/Wasabi/ServiceManager.cpp @@ -0,0 +1,201 @@ +#include "ServiceManager.h" +#include "api__wasabi-replicant.h" +#include "service/ifc_servicefactory.h" +#include "service/svccb.h" + +using namespace nu; + +ServiceManager::ServiceManager() +{ +#ifdef _WIN32 + component_wait = CreateSemaphoreW(NULL, 0, LONG_MAX, NULL); +#else + sem_init(&component_wait, 0, 0); +#endif + +} + +ServiceManager::~ServiceManager() +{ + #ifdef _WIN32 + if (component_wait) + CloseHandle(component_wait); +#else + sem_destroy(&component_wait); +#endif +} + +int ServiceManager::Dispatchable_QueryInterface(GUID interface_guid, void **object) +{ + if (interface_guid == ifc_component_sync::GetInterfaceGUID()) + { + *object = (ifc_component_sync *)this; + } + return NErr_Unknown; +} +//------------------------------------------- +int ServiceManager::GetServiceIndex(GUID key) +{ + for(int idx = 0; idx < services_indexer.size(); idx++) + { + if (memcmp(&key, &services_indexer[idx], sizeof(GUID)) == 0) + { + return idx; + } + } + + return -1; +} + +int ServiceManager::Service_Register(ifc_serviceFactory *svc) +{ + AutoLock lock(serviceGuard LOCKNAME("ServiceManager::service_register")); + GUID service_type = svc->GetServiceType(); + GUID service_id = svc->GetGUID(); + + // add the service to the master list + ifc_serviceFactory* new_factory = services[service_id]; + if (new_factory) // if someone already has this GUID, we need to replace + { + // replace factory in services_by_type map + ServiceList* type_list = services_by_type[service_type]; + + if (type_list) + { + for (ServiceList::iterator itr=type_list->begin();itr!=type_list->end();itr++) + { + ifc_serviceFactory *f = *itr; + if (f->GetGUID() == service_id) + { + *itr = svc; + } + } + } + // tell the old factory we're kicking its ass to the curb. + new_factory->ServiceNotify(ifc_serviceFactory::ONUNREGISTERED); + // HAKAN: + // Should we delete old factory? + // new_factory = svc; + } + else // not used yet, just assign + { + //new_factory = svc; + services_indexer.push_back(service_id); + + // add it to the by-type lookup + ServiceList *&type_list = services_by_type[service_type]; + if (!type_list) + type_list = new ServiceList; + + type_list->push_back(svc); + } + + services[service_id]=svc; + + // send notifications + svc->ServiceNotify(ifc_serviceFactory::ONREGISTERED); + + WASABI2_API_SYSCB->IssueCallback(Service::event_type, + Service::on_register, + (intptr_t)&service_type, reinterpret_cast<intptr_t>(svc)); + return NErr_Success; +} + +int ServiceManager::Service_Unregister(ifc_serviceFactory *svc) +{ + AutoLock lock(serviceGuard LOCKNAME("ServiceManager::Service_Unregister")); + GUID service_type = svc->GetServiceType(); + GUID service_id = svc->GetGUID(); + + // remove it from the master list + ServiceMap::iterator itr = services.find(service_id); + if (itr != services.end()) + services.erase(itr); + + // and from the type lookup map + ServiceList *type_list = services_by_type[service_type]; + if (type_list) + { + //type_list->eraseObject(svc); + for (auto it = type_list->begin(); it != type_list->end(); it++) + { + if (*it == svc) + { + it = type_list->erase(it); + break; + } + } + } + WASABI2_API_SYSCB->IssueCallback(Service::event_type, Service::on_deregister, (intptr_t)&service_type, reinterpret_cast<intptr_t>(svc)); + svc->ServiceNotify(ifc_serviceFactory::ONUNREGISTERED); + + return NErr_Success; +} + +size_t ServiceManager::Service_GetServiceCount(GUID svc_type) +{ + AutoLock lock(serviceGuard LOCKNAME("ServiceManager::Service_GetServiceCount")); + ServiceList *type_list = services_by_type[svc_type]; + if (type_list) + return type_list->size(); + else + return 0; +} + +ifc_serviceFactory *ServiceManager::Service_EnumService(GUID svc_type, size_t n) +{ + AutoLock lock(serviceGuard LOCKNAME("ServiceManager::Service_EnumService")); + ServiceList *type_list = services_by_type[svc_type]; + if (type_list && (size_t)n < type_list->size()) + return type_list->at(n); + else + return 0; +} + +ifc_serviceFactory *ServiceManager::Service_EnumService(size_t n) +{ + AutoLock lock(serviceGuard LOCKNAME("ServiceManager::Service_EnumService")); + if ((size_t)n < services.size()) + { + //return services.at(n).second; + if (n < services_indexer.size()) + { + GUID g = services_indexer[n]; + return services[g]; + } + } + + return 0; +} + +ifc_serviceFactory *ServiceManager::Service_GetServiceByGUID(GUID guid) +{ + AutoLock lock(serviceGuard LOCKNAME("ServiceManager::service_getServiceByGuid")); + ServiceMap::iterator itr = services.find(guid); + if (itr != services.end()) + return itr->second; + else + return 0; +} + +void ServiceManager::Service_ComponentDone() +{ +#ifdef _WIN32 + ReleaseSemaphore(component_wait, 1, NULL); +#else + sem_post(&component_wait); +#endif +} + +int ServiceManager::ComponentSync_Wait(size_t count) +{ + while (count--) + { +#ifdef _WIN32 + WaitForSingleObject(component_wait, INFINITE); +#else + sem_wait(&component_wait); +#endif + } + return NErr_Success; +} |