Skip to content

Commit

Permalink
Use extra char in notification message array for null terminator Infi…
Browse files Browse the repository at this point in the history
…niTimeOrg#1695

commit d7ca78b
Author: Finlay Davidson <finlay.davidson@coderclass.nl>
Date:   Fri Mar 17 13:48:26 2023 +0100

    alertnotificationservice: Make use of extra char in message array for \0

    Also move category reading down for clarity

commit d1b16a7
Author: Finlay Davidson <finlay.davidson@coderclass.nl>
Date:   Fri Mar 17 13:38:52 2023 +0100

    alertnotificationclient: Make use of extra char in message array for \0
  • Loading branch information
mark9064 committed May 21, 2023
1 parent c8155fd commit 1dc2847
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 24 deletions.
20 changes: 9 additions & 11 deletions src/components/ble/AlertNotificationClient.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -143,24 +143,22 @@ int AlertNotificationClient::OnDescriptorDiscoveryEventCallback(uint16_t connect

void AlertNotificationClient::OnNotification(ble_gap_event* event) {
if (event->notify_rx.attr_handle == newAlertHandle) {
constexpr size_t stringTerminatorSize = 1; // end of string '\0'
constexpr size_t headerSize = 3;
const auto maxMessageSize {NotificationManager::MaximumMessageSize()};
const auto maxBufferSize {maxMessageSize + headerSize};
constexpr uint32_t headerSize = 3;
constexpr uint32_t maxMessageSize = NotificationManager::MaximumMessageSize();

// Ignore notifications with empty message
const auto packetLen = OS_MBUF_PKTLEN(event->notify_rx.om);
if (packetLen <= headerSize)
const uint32_t packetLen = OS_MBUF_PKTLEN(event->notify_rx.om);
if (packetLen <= headerSize) {
return;
}

size_t bufferSize = std::min(packetLen + stringTerminatorSize, maxBufferSize);
auto messageSize = std::min(maxMessageSize, (bufferSize - headerSize));
const uint32_t messageSize = std::min(maxMessageSize, packetLen - headerSize);

NotificationManager::Notification notif;
os_mbuf_copydata(event->notify_rx.om, headerSize, messageSize - 1, notif.message.data());
notif.message[messageSize - 1] = '\0';
os_mbuf_copydata(event->notify_rx.om, headerSize, messageSize, notif.message.data());
notif.message[messageSize] = '\0';
notif.size = messageSize;
notif.category = Pinetime::Controllers::NotificationManager::Categories::SimpleAlert;
notif.category = NotificationManager::Categories::SimpleAlert;
notificationManager.Push(std::move(notif));

systemTask.PushMessage(Pinetime::System::Messages::OnNewNotification);
Expand Down
22 changes: 9 additions & 13 deletions src/components/ble/AlertNotificationService.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -46,27 +46,24 @@ AlertNotificationService::AlertNotificationService(System::SystemTask& systemTas

int AlertNotificationService::OnAlert(struct ble_gatt_access_ctxt* ctxt) {
if (ctxt->op == BLE_GATT_ACCESS_OP_WRITE_CHR) {
constexpr size_t stringTerminatorSize = 1; // end of string '\0'
constexpr size_t headerSize = 3;
const auto maxMessageSize {NotificationManager::MaximumMessageSize()};
const auto maxBufferSize {maxMessageSize + headerSize};
constexpr uint32_t headerSize = 3;
constexpr uint32_t maxMessageSize = NotificationManager::MaximumMessageSize();

// Ignore notifications with empty message
const auto packetLen = OS_MBUF_PKTLEN(ctxt->om);
const uint32_t packetLen = OS_MBUF_PKTLEN(ctxt->om);
if (packetLen <= headerSize) {
return 0;
}

size_t bufferSize = std::min(packetLen + stringTerminatorSize, maxBufferSize);
auto messageSize = std::min(maxMessageSize, (bufferSize - headerSize));
Categories category;
const uint32_t messageSize = std::min(maxMessageSize, packetLen - headerSize);

NotificationManager::Notification notif;
os_mbuf_copydata(ctxt->om, headerSize, messageSize - 1, notif.message.data());
os_mbuf_copydata(ctxt->om, 0, 1, &category);
notif.message[messageSize - 1] = '\0';
os_mbuf_copydata(ctxt->om, headerSize, messageSize, notif.message.data());
notif.message[messageSize] = '\0';
notif.size = messageSize;

Categories category;
os_mbuf_copydata(ctxt->om, 0, 1, &category);
// TODO convert all ANS categories to NotificationController categories
switch (category) {
case Categories::Call:
Expand All @@ -77,9 +74,8 @@ int AlertNotificationService::OnAlert(struct ble_gatt_access_ctxt* ctxt) {
break;
}

auto event = Pinetime::System::Messages::OnNewNotification;
notificationManager.Push(std::move(notif));
systemTask.PushMessage(event);
systemTask.PushMessage(Pinetime::System::Messages::OnNewNotification);
}
return 0;
}
Expand Down

0 comments on commit 1dc2847

Please sign in to comment.