| /****************************************************************************** |
| * |
| * Copyright (C) 2014 Google, Inc. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at: |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| ******************************************************************************/ |
| |
| #define LOG_TAG "bt_core_module" |
| |
| #include <base/logging.h> |
| #include <dlfcn.h> |
| #include <string.h> |
| |
| #include <mutex> |
| #include <unordered_map> |
| |
| #include "btcore/include/module.h" |
| #include "osi/include/allocator.h" |
| #include "osi/include/log.h" |
| #include "osi/include/osi.h" |
| |
| typedef enum { |
| MODULE_STATE_NONE = 0, |
| MODULE_STATE_INITIALIZED = 1, |
| MODULE_STATE_STARTED = 2 |
| } module_state_t; |
| |
| static std::unordered_map<const module_t*, module_state_t> metadata; |
| |
| // TODO(jamuraa): remove this lock after the startup sequence is clean |
| static std::mutex metadata_mutex; |
| |
| static bool call_lifecycle_function(module_lifecycle_fn function); |
| static module_state_t get_module_state(const module_t* module); |
| static void set_module_state(const module_t* module, module_state_t state); |
| |
| void module_management_start(void) {} |
| |
| void module_management_stop(void) { |
| metadata.clear(); |
| } |
| |
| const module_t* get_module(const char* name) { |
| module_t* module = (module_t*)dlsym(RTLD_DEFAULT, name); |
| CHECK(module); |
| return module; |
| } |
| |
| bool module_init(const module_t* module) { |
| CHECK(module != NULL); |
| CHECK(get_module_state(module) == MODULE_STATE_NONE); |
| |
| LOG_INFO(LOG_TAG, "%s Initializing module \"%s\"", __func__, module->name); |
| if (!call_lifecycle_function(module->init)) { |
| LOG_ERROR(LOG_TAG, "%s Failed to initialize module \"%s\"", __func__, |
| module->name); |
| return false; |
| } |
| LOG_INFO(LOG_TAG, "%s Initialized module \"%s\"", __func__, module->name); |
| |
| set_module_state(module, MODULE_STATE_INITIALIZED); |
| return true; |
| } |
| |
| bool module_start_up(const module_t* module) { |
| CHECK(module != NULL); |
| // TODO(zachoverflow): remove module->init check once automagic order/call is |
| // in place. |
| // This hack is here so modules which don't require init don't have to have |
| // useless calls |
| // as we're converting the startup sequence. |
| CHECK(get_module_state(module) == MODULE_STATE_INITIALIZED || |
| module->init == NULL); |
| |
| LOG_INFO(LOG_TAG, "%s Starting module \"%s\"", __func__, module->name); |
| if (!call_lifecycle_function(module->start_up)) { |
| LOG_ERROR(LOG_TAG, "%s Failed to start up module \"%s\"", __func__, |
| module->name); |
| return false; |
| } |
| LOG_INFO(LOG_TAG, "%s Started module \"%s\"", __func__, module->name); |
| |
| set_module_state(module, MODULE_STATE_STARTED); |
| return true; |
| } |
| |
| void module_shut_down(const module_t* module) { |
| CHECK(module != NULL); |
| module_state_t state = get_module_state(module); |
| CHECK(state <= MODULE_STATE_STARTED); |
| |
| // Only something to do if the module was actually started |
| if (state < MODULE_STATE_STARTED) return; |
| |
| LOG_INFO(LOG_TAG, "%s Shutting down module \"%s\"", __func__, module->name); |
| if (!call_lifecycle_function(module->shut_down)) { |
| LOG_ERROR(LOG_TAG, |
| "%s Failed to shutdown module \"%s\". Continuing anyway.", |
| __func__, module->name); |
| } |
| LOG_INFO(LOG_TAG, "%s Shutdown of module \"%s\" completed", __func__, |
| module->name); |
| |
| set_module_state(module, MODULE_STATE_INITIALIZED); |
| } |
| |
| void module_clean_up(const module_t* module) { |
| CHECK(module != NULL); |
| module_state_t state = get_module_state(module); |
| CHECK(state <= MODULE_STATE_INITIALIZED); |
| |
| // Only something to do if the module was actually initialized |
| if (state < MODULE_STATE_INITIALIZED) return; |
| |
| LOG_INFO(LOG_TAG, "%s Cleaning up module \"%s\"", __func__, module->name); |
| if (!call_lifecycle_function(module->clean_up)) { |
| LOG_ERROR(LOG_TAG, "%s Failed to cleanup module \"%s\". Continuing anyway.", |
| __func__, module->name); |
| } |
| LOG_INFO(LOG_TAG, "%s Cleanup of module \"%s\" completed", __func__, |
| module->name); |
| |
| set_module_state(module, MODULE_STATE_NONE); |
| } |
| |
| static bool call_lifecycle_function(module_lifecycle_fn function) { |
| // A NULL lifecycle function means it isn't needed, so assume success |
| if (!function) return true; |
| |
| future_t* future = function(); |
| |
| // A NULL future means synchronous success |
| if (!future) return true; |
| |
| // Otherwise fall back to the future |
| return future_await(future); |
| } |
| |
| static module_state_t get_module_state(const module_t* module) { |
| std::lock_guard<std::mutex> lock(metadata_mutex); |
| auto map_ptr = metadata.find(module); |
| |
| return (map_ptr != metadata.end()) ? map_ptr->second : MODULE_STATE_NONE; |
| } |
| |
| static void set_module_state(const module_t* module, module_state_t state) { |
| std::lock_guard<std::mutex> lock(metadata_mutex); |
| metadata[module] = state; |
| } |
| |
| // TODO(zachoverflow): remove when everything modulized |
| // Temporary callback-wrapper-related code |
| |
| typedef struct { |
| const module_t* module; |
| thread_t* lifecycle_thread; |
| thread_t* callback_thread; // we don't own this thread |
| thread_fn callback; |
| bool success; |
| } callbacked_wrapper_t; |
| |
| static void run_wrapped_start_up(void* context); |
| static void post_result_to_callback(void* context); |
| |
| void module_start_up_callbacked_wrapper(const module_t* module, |
| thread_t* callback_thread, |
| thread_fn callback) { |
| callbacked_wrapper_t* wrapper = |
| (callbacked_wrapper_t*)osi_calloc(sizeof(callbacked_wrapper_t)); |
| |
| wrapper->module = module; |
| wrapper->lifecycle_thread = thread_new("module_wrapper"); |
| wrapper->callback_thread = callback_thread; |
| wrapper->callback = callback; |
| |
| // Run the actual module start up |
| thread_post(wrapper->lifecycle_thread, run_wrapped_start_up, wrapper); |
| } |
| |
| static void run_wrapped_start_up(void* context) { |
| CHECK(context); |
| |
| callbacked_wrapper_t* wrapper = (callbacked_wrapper_t*)context; |
| wrapper->success = module_start_up(wrapper->module); |
| |
| // Post the result back to the callback |
| thread_post(wrapper->callback_thread, post_result_to_callback, wrapper); |
| } |
| |
| static void post_result_to_callback(void* context) { |
| CHECK(context); |
| |
| callbacked_wrapper_t* wrapper = (callbacked_wrapper_t*)context; |
| |
| // Save the values we need for callback |
| void* result = wrapper->success ? FUTURE_SUCCESS : FUTURE_FAIL; |
| thread_fn callback = wrapper->callback; |
| |
| // Clean up the resources we used |
| thread_free(wrapper->lifecycle_thread); |
| osi_free(wrapper); |
| |
| callback(result); |
| } |