mirror of https://github.com/sqfmi/Watchy.git
153 lines
4.3 KiB
C++
153 lines
4.3 KiB
C++
|
#include "BLE.h"
|
||
|
|
||
|
#define SERVICE_UUID_ESPOTA "cd77498e-1ac8-48b6-aba8-4161c7342fce"
|
||
|
#define CHARACTERISTIC_UUID_ID "cd77498f-1ac8-48b6-aba8-4161c7342fce"
|
||
|
|
||
|
#define SERVICE_UUID_OTA "86b12865-4b70-4893-8ce6-9864fc00374d"
|
||
|
#define CHARACTERISTIC_UUID_FW "86b12866-4b70-4893-8ce6-9864fc00374d"
|
||
|
#define CHARACTERISTIC_UUID_HW_VERSION "86b12867-4b70-4893-8ce6-9864fc00374d"
|
||
|
#define CHARACTERISTIC_UUID_WATCHFACE_NAME "86b12868-4b70-4893-8ce6-9864fc00374d"
|
||
|
|
||
|
#define FULL_PACKET 512
|
||
|
#define CHARPOS_UPDATE_FLAG 5
|
||
|
|
||
|
#define STATUS_CONNECTED 0
|
||
|
#define STATUS_DISCONNECTED 4
|
||
|
#define STATUS_UPDATING 1
|
||
|
#define STATUS_READY 2
|
||
|
|
||
|
esp_ota_handle_t otaHandler = 0;
|
||
|
|
||
|
int status = -1;
|
||
|
int bytesReceived = 0;
|
||
|
bool updateFlag = false;
|
||
|
|
||
|
|
||
|
class BLECustomServerCallbacks: public BLEServerCallbacks {
|
||
|
void onConnect(BLEServer* pServer) {
|
||
|
status = STATUS_CONNECTED;
|
||
|
};
|
||
|
|
||
|
void onDisconnect(BLEServer* pServer) {
|
||
|
status = STATUS_DISCONNECTED;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
class otaCallback: public BLECharacteristicCallbacks {
|
||
|
public:
|
||
|
otaCallback(BLE* ble) {
|
||
|
_p_ble = ble;
|
||
|
}
|
||
|
BLE* _p_ble;
|
||
|
|
||
|
void onWrite(BLECharacteristic *pCharacteristic);
|
||
|
};
|
||
|
|
||
|
void otaCallback::onWrite(BLECharacteristic *pCharacteristic)
|
||
|
{
|
||
|
std::string rxData = pCharacteristic->getValue();
|
||
|
if (!updateFlag) { //If it's the first packet of OTA since bootup, begin OTA
|
||
|
//Serial.println("Begin FW Update");
|
||
|
esp_ota_begin(esp_ota_get_next_update_partition(NULL), OTA_SIZE_UNKNOWN, &otaHandler);
|
||
|
updateFlag = true;
|
||
|
status = STATUS_UPDATING;
|
||
|
}
|
||
|
if (_p_ble != NULL)
|
||
|
{
|
||
|
if (rxData.length() > 0)
|
||
|
{
|
||
|
esp_ota_write(otaHandler, rxData.c_str(), rxData.length());
|
||
|
bytesReceived = bytesReceived + rxData.length();
|
||
|
if (rxData.length() != FULL_PACKET)
|
||
|
{
|
||
|
esp_ota_end(otaHandler);
|
||
|
//Serial.println("End FW Update");
|
||
|
if (ESP_OK == esp_ota_set_boot_partition(esp_ota_get_next_update_partition(NULL))) {
|
||
|
status = STATUS_READY;
|
||
|
}
|
||
|
else {
|
||
|
//Serial.println("Upload Error");
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
uint8_t txData[5] = {1, 2, 3, 4, 5};
|
||
|
//delay(1000);
|
||
|
pCharacteristic->setValue((uint8_t*)txData, 5);
|
||
|
pCharacteristic->notify();
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Constructor
|
||
|
BLE::BLE(void) {
|
||
|
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Destructor
|
||
|
BLE::~BLE(void)
|
||
|
{
|
||
|
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// begin
|
||
|
bool BLE::begin(const char* localName = "Watchy BLE OTA") {
|
||
|
// Create the BLE Device
|
||
|
BLEDevice::init(localName);
|
||
|
|
||
|
// Create the BLE Server
|
||
|
pServer = BLEDevice::createServer();
|
||
|
pServer->setCallbacks(new BLECustomServerCallbacks());
|
||
|
|
||
|
// Create the BLE Service
|
||
|
pESPOTAService = pServer->createService(SERVICE_UUID_ESPOTA);
|
||
|
pService = pServer->createService(SERVICE_UUID_OTA);
|
||
|
|
||
|
// Create a BLE Characteristic
|
||
|
pESPOTAIdCharacteristic = pESPOTAService->createCharacteristic(
|
||
|
CHARACTERISTIC_UUID_ID,
|
||
|
BLECharacteristic::PROPERTY_READ
|
||
|
);
|
||
|
|
||
|
pVersionCharacteristic = pService->createCharacteristic(
|
||
|
CHARACTERISTIC_UUID_HW_VERSION,
|
||
|
BLECharacteristic::PROPERTY_READ
|
||
|
);
|
||
|
|
||
|
pWatchFaceNameCharacteristic = pService->createCharacteristic(
|
||
|
CHARACTERISTIC_UUID_WATCHFACE_NAME,
|
||
|
BLECharacteristic::PROPERTY_READ
|
||
|
);
|
||
|
|
||
|
pOtaCharacteristic = pService->createCharacteristic(
|
||
|
CHARACTERISTIC_UUID_FW,
|
||
|
BLECharacteristic::PROPERTY_NOTIFY | BLECharacteristic::PROPERTY_WRITE
|
||
|
);
|
||
|
|
||
|
pOtaCharacteristic->addDescriptor(new BLE2902());
|
||
|
pOtaCharacteristic->setCallbacks(new otaCallback(this));
|
||
|
|
||
|
// Start the service(s)
|
||
|
pESPOTAService->start();
|
||
|
pService->start();
|
||
|
|
||
|
// Start advertising
|
||
|
pServer->getAdvertising()->addServiceUUID(SERVICE_UUID_ESPOTA);
|
||
|
pServer->getAdvertising()->start();
|
||
|
|
||
|
uint8_t hardwareVersion[5] = {HARDWARE_VERSION_MAJOR, HARDWARE_VERSION_MINOR, SOFTWARE_VERSION_MAJOR, SOFTWARE_VERSION_MINOR, SOFTWARE_VERSION_PATCH};
|
||
|
pVersionCharacteristic->setValue((uint8_t*)hardwareVersion, 5);
|
||
|
pWatchFaceNameCharacteristic->setValue("Watchy 7 Segment");
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
int BLE::updateStatus(){
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
int BLE::howManyBytes(){
|
||
|
return bytesReceived;
|
||
|
}
|