| /* |
| * |
| * Copyright (c) 2020 Project CHIP Authors |
| * All rights reserved. |
| * |
| * 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. |
| */ |
| |
| #include "AppTask.h" |
| #include "AppConfig.h" |
| #include "BoltLockManager.h" |
| #include "LEDWidget.h" |
| #include "ThreadUtil.h" |
| |
| #include <app-common/zap-generated/attribute-id.h> |
| #include <app-common/zap-generated/attribute-type.h> |
| #include <app-common/zap-generated/attributes/Accessors.h> |
| #include <app-common/zap-generated/cluster-id.h> |
| #include <app/server/OnboardingCodesUtil.h> |
| #include <app/server/Server.h> |
| #include <credentials/DeviceAttestationCredsProvider.h> |
| #include <credentials/examples/DeviceAttestationCredsExample.h> |
| #include <lib/support/CHIPMem.h> |
| #include <lib/support/CodeUtils.h> |
| #include <lib/support/ErrorStr.h> |
| #include <system/SystemClock.h> |
| |
| #if CONFIG_CHIP_OTA_REQUESTOR |
| #include "OTAUtil.h" |
| #include <app/clusters/ota-requestor/BDXDownloader.h> |
| #include <app/clusters/ota-requestor/DefaultOTARequestorStorage.h> |
| #include <app/clusters/ota-requestor/GenericOTARequestorDriver.h> |
| #include <app/clusters/ota-requestor/OTARequestor.h> |
| #endif |
| |
| #include <dk_buttons_and_leds.h> |
| #include <logging/log.h> |
| #include <zephyr.h> |
| |
| using namespace ::chip; |
| using namespace ::chip::app; |
| using namespace ::chip::Credentials; |
| using namespace ::chip::DeviceLayer; |
| |
| #define FACTORY_RESET_TRIGGER_TIMEOUT 3000 |
| #define FACTORY_RESET_CANCEL_WINDOW_TIMEOUT 3000 |
| #define APP_EVENT_QUEUE_SIZE 10 |
| #define BUTTON_PUSH_EVENT 1 |
| #define BUTTON_RELEASE_EVENT 0 |
| |
| namespace { |
| constexpr EndpointId kLockEndpointId = 1; |
| |
| LOG_MODULE_DECLARE(app); |
| K_MSGQ_DEFINE(sAppEventQueue, sizeof(AppEvent), APP_EVENT_QUEUE_SIZE, alignof(AppEvent)); |
| k_timer sFunctionTimer; |
| |
| LEDWidget sStatusLED; |
| LEDWidget sLockLED; |
| LEDWidget sUnusedLED; |
| LEDWidget sUnusedLED_1; |
| |
| bool sIsThreadProvisioned = false; |
| bool sIsThreadEnabled = false; |
| bool sHaveBLEConnections = false; |
| |
| #if CONFIG_CHIP_OTA_REQUESTOR |
| DefaultOTARequestorStorage sRequestorStorage; |
| GenericOTARequestorDriver sOTARequestorDriver; |
| chip::BDXDownloader sBDXDownloader; |
| chip::OTARequestor sOTARequestor; |
| #endif |
| |
| } // namespace |
| |
| AppTask AppTask::sAppTask; |
| |
| CHIP_ERROR AppTask::Init() |
| { |
| // Initialize CHIP stack |
| LOG_INF("Init CHIP stack"); |
| |
| CHIP_ERROR err = chip::Platform::MemoryInit(); |
| if (err != CHIP_NO_ERROR) |
| { |
| LOG_ERR("Platform::MemoryInit() failed"); |
| return err; |
| } |
| |
| err = PlatformMgr().InitChipStack(); |
| if (err != CHIP_NO_ERROR) |
| { |
| LOG_ERR("PlatformMgr().InitChipStack() failed"); |
| return err; |
| } |
| |
| err = ThreadStackMgr().InitThreadStack(); |
| if (err != CHIP_NO_ERROR) |
| { |
| LOG_ERR("ThreadStackMgr().InitThreadStack() failed"); |
| return err; |
| } |
| |
| #ifdef CONFIG_OPENTHREAD_MTD_SED |
| err = ConnectivityMgr().SetThreadDeviceType(ConnectivityManager::kThreadDeviceType_SleepyEndDevice); |
| #elif CONFIG_OPENTHREAD_MTD |
| err = ConnectivityMgr().SetThreadDeviceType(ConnectivityManager::kThreadDeviceType_MinimalEndDevice); |
| #else |
| err = ConnectivityMgr().SetThreadDeviceType(ConnectivityManager::kThreadDeviceType_FullEndDevice); |
| #endif |
| if (err != CHIP_NO_ERROR) |
| { |
| LOG_ERR("ConnectivityMgr().SetThreadDeviceType() failed"); |
| return err; |
| } |
| |
| // Initialize LEDs |
| LEDWidget::InitGpio(); |
| LEDWidget::SetStateUpdateCallback(LEDStateUpdateHandler); |
| |
| sStatusLED.Init(SYSTEM_STATE_LED); |
| sLockLED.Init(LOCK_STATE_LED); |
| sLockLED.Set(!BoltLockMgr().IsUnlocked()); |
| |
| sUnusedLED.Init(DK_LED3); |
| sUnusedLED_1.Init(DK_LED4); |
| |
| UpdateStatusLED(); |
| |
| BoltLockMgr().Init(); |
| BoltLockMgr().SetCallbacks(ActionInitiated, ActionCompleted); |
| |
| // Initialize buttons |
| int ret = dk_buttons_init(ButtonEventHandler); |
| if (ret) |
| { |
| LOG_ERR("dk_buttons_init() failed"); |
| return chip::System::MapErrorZephyr(ret); |
| } |
| |
| // Initialize function button timer |
| k_timer_init(&sFunctionTimer, &AppTask::TimerEventHandler, nullptr); |
| k_timer_user_data_set(&sFunctionTimer, this); |
| |
| #ifdef CONFIG_MCUMGR_SMP_BT |
| // Initialize DFU over SMP |
| GetDFUOverSMP().Init(RequestSMPAdvertisingStart); |
| GetDFUOverSMP().ConfirmNewImage(); |
| #endif |
| |
| // Initialize CHIP server |
| SetDeviceAttestationCredentialsProvider(Examples::GetExampleDACProvider()); |
| InitOTARequestor(); |
| ReturnErrorOnFailure(chip::Server::GetInstance().Init()); |
| ConfigurationMgr().LogDeviceConfig(); |
| PrintOnboardingCodes(chip::RendezvousInformationFlags(chip::RendezvousInformationFlag::kBLE)); |
| |
| // Add CHIP event handler and start CHIP thread. |
| // Note that all the initialization code should happen prior to this point to avoid data races |
| // between the main and the CHIP threads. |
| PlatformMgr().AddEventHandler(ChipEventHandler, 0); |
| |
| err = PlatformMgr().StartEventLoopTask(); |
| if (err != CHIP_NO_ERROR) |
| { |
| LOG_ERR("PlatformMgr().StartEventLoopTask() failed"); |
| } |
| |
| return err; |
| } |
| |
| void AppTask::InitOTARequestor() |
| { |
| #if CONFIG_CHIP_OTA_REQUESTOR |
| OTAImageProcessorNrf::Get().SetOTADownloader(&sBDXDownloader); |
| sBDXDownloader.SetImageProcessorDelegate(&OTAImageProcessorNrf::Get()); |
| sOTARequestorDriver.Init(&sOTARequestor, &OTAImageProcessorNrf::Get()); |
| sRequestorStorage.Init(Server::GetInstance().GetPersistentStorage()); |
| sOTARequestor.Init(Server::GetInstance(), sRequestorStorage, sOTARequestorDriver, sBDXDownloader); |
| chip::SetRequestorInstance(&sOTARequestor); |
| #endif |
| } |
| |
| CHIP_ERROR AppTask::StartApp() |
| { |
| ReturnErrorOnFailure(Init()); |
| |
| AppEvent event = {}; |
| |
| while (true) |
| { |
| k_msgq_get(&sAppEventQueue, &event, K_FOREVER); |
| DispatchEvent(&event); |
| } |
| |
| return CHIP_NO_ERROR; |
| } |
| |
| void AppTask::LockActionEventHandler(AppEvent * aEvent) |
| { |
| BoltLockManager::Action_t action = BoltLockManager::INVALID_ACTION; |
| int32_t actor = 0; |
| |
| if (aEvent->Type == AppEvent::kEventType_Lock) |
| { |
| action = static_cast<BoltLockManager::Action_t>(aEvent->LockEvent.Action); |
| actor = aEvent->LockEvent.Actor; |
| } |
| else if (aEvent->Type == AppEvent::kEventType_Button) |
| { |
| action = BoltLockMgr().IsUnlocked() ? BoltLockManager::LOCK_ACTION : BoltLockManager::UNLOCK_ACTION; |
| actor = AppEvent::kEventType_Button; |
| } |
| |
| if (action != BoltLockManager::INVALID_ACTION && !BoltLockMgr().InitiateAction(actor, action)) |
| LOG_INF("Action is already in progress or active."); |
| } |
| |
| void AppTask::ButtonEventHandler(uint32_t button_state, uint32_t has_changed) |
| { |
| AppEvent button_event; |
| button_event.Type = AppEvent::kEventType_Button; |
| |
| if (LOCK_BUTTON_MASK & button_state & has_changed) |
| { |
| button_event.ButtonEvent.PinNo = LOCK_BUTTON; |
| button_event.ButtonEvent.Action = BUTTON_PUSH_EVENT; |
| button_event.Handler = LockActionEventHandler; |
| sAppTask.PostEvent(&button_event); |
| } |
| |
| if (FUNCTION_BUTTON_MASK & has_changed) |
| { |
| button_event.ButtonEvent.PinNo = FUNCTION_BUTTON; |
| button_event.ButtonEvent.Action = (FUNCTION_BUTTON_MASK & button_state) ? BUTTON_PUSH_EVENT : BUTTON_RELEASE_EVENT; |
| button_event.Handler = FunctionHandler; |
| sAppTask.PostEvent(&button_event); |
| } |
| |
| if (THREAD_START_BUTTON_MASK & button_state & has_changed) |
| { |
| button_event.ButtonEvent.PinNo = THREAD_START_BUTTON; |
| button_event.ButtonEvent.Action = BUTTON_PUSH_EVENT; |
| button_event.Handler = StartThreadHandler; |
| sAppTask.PostEvent(&button_event); |
| } |
| |
| if (BLE_ADVERTISEMENT_START_BUTTON_MASK & button_state & has_changed) |
| { |
| button_event.ButtonEvent.PinNo = BLE_ADVERTISEMENT_START_BUTTON; |
| button_event.ButtonEvent.Action = BUTTON_PUSH_EVENT; |
| button_event.Handler = StartBLEAdvertisementHandler; |
| sAppTask.PostEvent(&button_event); |
| } |
| } |
| |
| void AppTask::TimerEventHandler(k_timer * timer) |
| { |
| AppEvent event; |
| event.Type = AppEvent::kEventType_Timer; |
| event.TimerEvent.Context = k_timer_user_data_get(timer); |
| event.Handler = FunctionTimerEventHandler; |
| sAppTask.PostEvent(&event); |
| } |
| |
| void AppTask::FunctionTimerEventHandler(AppEvent * aEvent) |
| { |
| if (aEvent->Type != AppEvent::kEventType_Timer) |
| return; |
| |
| // If we reached here, the button was held past FACTORY_RESET_TRIGGER_TIMEOUT, initiate factory reset |
| if (sAppTask.mFunctionTimerActive && sAppTask.mFunction == kFunction_SoftwareUpdate) |
| { |
| LOG_INF("Factory Reset Triggered. Release button within %ums to cancel.", FACTORY_RESET_TRIGGER_TIMEOUT); |
| |
| // Start timer for FACTORY_RESET_CANCEL_WINDOW_TIMEOUT to allow user to cancel, if required. |
| sAppTask.StartTimer(FACTORY_RESET_CANCEL_WINDOW_TIMEOUT); |
| sAppTask.mFunction = kFunction_FactoryReset; |
| |
| #ifdef CONFIG_STATE_LEDS |
| // Turn off all LEDs before starting blink to make sure blink is co-ordinated. |
| sStatusLED.Set(false); |
| sLockLED.Set(false); |
| sUnusedLED_1.Set(false); |
| sUnusedLED.Set(false); |
| |
| sStatusLED.Blink(500); |
| sLockLED.Blink(500); |
| sUnusedLED.Blink(500); |
| sUnusedLED_1.Blink(500); |
| #endif |
| } |
| else if (sAppTask.mFunctionTimerActive && sAppTask.mFunction == kFunction_FactoryReset) |
| { |
| // Actually trigger Factory Reset |
| sAppTask.mFunction = kFunction_NoneSelected; |
| |
| chip::Server::GetInstance().ScheduleFactoryReset(); |
| } |
| } |
| |
| #ifdef CONFIG_MCUMGR_SMP_BT |
| void AppTask::RequestSMPAdvertisingStart(void) |
| { |
| AppEvent event; |
| event.Type = AppEvent::kEventType_StartSMPAdvertising; |
| event.Handler = [](AppEvent *) { GetDFUOverSMP().StartBLEAdvertising(); }; |
| sAppTask.PostEvent(&event); |
| } |
| #endif |
| |
| void AppTask::FunctionHandler(AppEvent * aEvent) |
| { |
| if (aEvent->ButtonEvent.PinNo != FUNCTION_BUTTON) |
| return; |
| |
| // To trigger software update: press the FUNCTION_BUTTON button briefly (< FACTORY_RESET_TRIGGER_TIMEOUT) |
| // To initiate factory reset: press the FUNCTION_BUTTON for FACTORY_RESET_TRIGGER_TIMEOUT + FACTORY_RESET_CANCEL_WINDOW_TIMEOUT |
| // All LEDs start blinking after FACTORY_RESET_TRIGGER_TIMEOUT to signal factory reset has been initiated. |
| // To cancel factory reset: release the FUNCTION_BUTTON once all LEDs start blinking within the |
| // FACTORY_RESET_CANCEL_WINDOW_TIMEOUT |
| if (aEvent->ButtonEvent.Action == BUTTON_PUSH_EVENT) |
| { |
| if (!sAppTask.mFunctionTimerActive && sAppTask.mFunction == kFunction_NoneSelected) |
| { |
| sAppTask.StartTimer(FACTORY_RESET_TRIGGER_TIMEOUT); |
| |
| sAppTask.mFunction = kFunction_SoftwareUpdate; |
| } |
| } |
| else |
| { |
| // If the button was released before factory reset got initiated, trigger a software update. |
| if (sAppTask.mFunctionTimerActive && sAppTask.mFunction == kFunction_SoftwareUpdate) |
| { |
| sAppTask.CancelTimer(); |
| sAppTask.mFunction = kFunction_NoneSelected; |
| |
| #ifdef CONFIG_MCUMGR_SMP_BT |
| GetDFUOverSMP().StartServer(); |
| #else |
| LOG_INF("Software update is disabled"); |
| #endif |
| } |
| else if (sAppTask.mFunctionTimerActive && sAppTask.mFunction == kFunction_FactoryReset) |
| { |
| sUnusedLED.Set(false); |
| sUnusedLED_1.Set(false); |
| |
| // Set lock status LED back to show state of lock. |
| sLockLED.Set(!BoltLockMgr().IsUnlocked()); |
| |
| UpdateStatusLED(); |
| sAppTask.CancelTimer(); |
| |
| // Change the function to none selected since factory reset has been canceled. |
| sAppTask.mFunction = kFunction_NoneSelected; |
| |
| LOG_INF("Factory Reset has been Canceled"); |
| } |
| } |
| } |
| |
| void AppTask::StartThreadHandler(AppEvent * aEvent) |
| { |
| if (aEvent->ButtonEvent.PinNo != THREAD_START_BUTTON) |
| return; |
| |
| if (chip::Server::GetInstance().AddTestCommissioning() != CHIP_NO_ERROR) |
| { |
| LOG_ERR("Failed to add test pairing"); |
| } |
| |
| if (!ConnectivityMgr().IsThreadProvisioned()) |
| { |
| StartDefaultThreadNetwork(); |
| LOG_INF("Device is not commissioned to a Thread network. Starting with the default configuration."); |
| } |
| else |
| { |
| LOG_INF("Device is commissioned to a Thread network."); |
| } |
| } |
| |
| void AppTask::StartBLEAdvertisementHandler(AppEvent *) |
| { |
| if (Server::GetInstance().GetFabricTable().FabricCount() != 0) |
| { |
| LOG_INF("Matter service BLE advertising not started - device is already commissioned"); |
| return; |
| } |
| |
| if (ConnectivityMgr().IsBLEAdvertisingEnabled()) |
| { |
| LOG_INF("BLE advertising is already enabled"); |
| return; |
| } |
| |
| if (Server::GetInstance().GetCommissioningWindowManager().OpenBasicCommissioningWindow() != CHIP_NO_ERROR) |
| { |
| LOG_ERR("OpenBasicCommissioningWindow() failed"); |
| } |
| } |
| |
| void AppTask::UpdateLedStateEventHandler(AppEvent * aEvent) |
| { |
| if (aEvent->Type == AppEvent::kEventType_UpdateLedState) |
| { |
| aEvent->UpdateLedStateEvent.LedWidget->UpdateState(); |
| } |
| } |
| |
| void AppTask::LEDStateUpdateHandler(LEDWidget & ledWidget) |
| { |
| AppEvent event; |
| event.Type = AppEvent::kEventType_UpdateLedState; |
| event.Handler = UpdateLedStateEventHandler; |
| event.UpdateLedStateEvent.LedWidget = &ledWidget; |
| sAppTask.PostEvent(&event); |
| } |
| |
| void AppTask::UpdateStatusLED() |
| { |
| #ifdef CONFIG_STATE_LEDS |
| /* Update the status LED. |
| * |
| * If thread and service provisioned, keep the LED On constantly. |
| * |
| * If the system has ble connection(s) uptill the stage above, THEN blink the LED at an even |
| * rate of 100ms. |
| * |
| * Otherwise, blink the LED On for a very short time. */ |
| if (sIsThreadProvisioned && sIsThreadEnabled) |
| { |
| sStatusLED.Set(true); |
| } |
| else if (sHaveBLEConnections) |
| { |
| sStatusLED.Blink(100, 100); |
| } |
| else |
| { |
| sStatusLED.Blink(50, 950); |
| } |
| #endif |
| } |
| |
| void AppTask::ChipEventHandler(const ChipDeviceEvent * event, intptr_t /* arg */) |
| { |
| switch (event->Type) |
| { |
| case DeviceEventType::kCHIPoBLEAdvertisingChange: |
| #ifdef CONFIG_CHIP_NFC_COMMISSIONING |
| if (event->CHIPoBLEAdvertisingChange.Result == kActivity_Started) |
| { |
| if (NFCMgr().IsTagEmulationStarted()) |
| { |
| LOG_INF("NFC Tag emulation is already started"); |
| } |
| else |
| { |
| ShareQRCodeOverNFC(chip::RendezvousInformationFlags(chip::RendezvousInformationFlag::kBLE)); |
| } |
| } |
| else if (event->CHIPoBLEAdvertisingChange.Result == kActivity_Stopped) |
| { |
| NFCMgr().StopTagEmulation(); |
| } |
| #endif |
| sHaveBLEConnections = ConnectivityMgr().NumBLEConnections() != 0; |
| UpdateStatusLED(); |
| break; |
| case DeviceEventType::kThreadStateChange: |
| sIsThreadProvisioned = ConnectivityMgr().IsThreadProvisioned(); |
| sIsThreadEnabled = ConnectivityMgr().IsThreadEnabled(); |
| UpdateStatusLED(); |
| break; |
| default: |
| break; |
| } |
| } |
| |
| void AppTask::CancelTimer() |
| { |
| k_timer_stop(&sFunctionTimer); |
| mFunctionTimerActive = false; |
| } |
| |
| void AppTask::StartTimer(uint32_t aTimeoutInMs) |
| { |
| k_timer_start(&sFunctionTimer, K_MSEC(aTimeoutInMs), K_NO_WAIT); |
| mFunctionTimerActive = true; |
| } |
| |
| void AppTask::ActionInitiated(BoltLockManager::Action_t aAction, int32_t aActor) |
| { |
| // If the action has been initiated by the lock, update the bolt lock trait |
| // and start flashing the LEDs rapidly to indicate action initiation. |
| if (aAction == BoltLockManager::LOCK_ACTION) |
| { |
| LOG_INF("Lock Action has been initiated"); |
| } |
| else if (aAction == BoltLockManager::UNLOCK_ACTION) |
| { |
| LOG_INF("Unlock Action has been initiated"); |
| } |
| |
| sLockLED.Blink(50, 50); |
| } |
| |
| void AppTask::ActionCompleted(BoltLockManager::Action_t aAction, int32_t aActor) |
| { |
| // if the action has been completed by the lock, update the bolt lock trait. |
| // Turn on the lock LED if in a LOCKED state OR |
| // Turn off the lock LED if in an UNLOCKED state. |
| if (aAction == BoltLockManager::LOCK_ACTION) |
| { |
| LOG_INF("Lock Action has been completed"); |
| sLockLED.Set(true); |
| } |
| else if (aAction == BoltLockManager::UNLOCK_ACTION) |
| { |
| LOG_INF("Unlock Action has been completed"); |
| sLockLED.Set(false); |
| } |
| |
| if (aActor == AppEvent::kEventType_Button) |
| { |
| sAppTask.UpdateClusterState(); |
| } |
| } |
| |
| void AppTask::PostLockActionRequest(int32_t aActor, BoltLockManager::Action_t aAction) |
| { |
| AppEvent event; |
| event.Type = AppEvent::kEventType_Lock; |
| event.LockEvent.Actor = aActor; |
| event.LockEvent.Action = aAction; |
| event.Handler = LockActionEventHandler; |
| PostEvent(&event); |
| } |
| |
| void AppTask::PostEvent(AppEvent * aEvent) |
| { |
| if (k_msgq_put(&sAppEventQueue, aEvent, K_NO_WAIT)) |
| { |
| LOG_INF("Failed to post event to app task event queue"); |
| } |
| } |
| |
| void AppTask::DispatchEvent(AppEvent * aEvent) |
| { |
| if (aEvent->Handler) |
| { |
| aEvent->Handler(aEvent); |
| } |
| else |
| { |
| LOG_INF("Event received with no handler. Dropping event."); |
| } |
| } |
| |
| void AppTask::UpdateClusterState() |
| { |
| // write the new on/off value |
| EmberAfStatus status = Clusters::OnOff::Attributes::OnOff::Set(kLockEndpointId, !BoltLockMgr().IsUnlocked()); |
| |
| if (status != EMBER_ZCL_STATUS_SUCCESS) |
| { |
| LOG_ERR("Updating on/off %x", status); |
| } |
| } |