pebble/src/fw/console/pulse_bulkio_external_flash.c
2025-01-27 11:38:16 -08:00

113 lines
3.6 KiB
C

/*
* Copyright 2024 Google LLC
*
* 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 "pulse_bulkio_domain_handler.h"
#include "console/pulse_protocol_impl.h"
#include "drivers/flash.h"
#include "flash_region/flash_region.h"
#include "kernel/pbl_malloc.h"
#include "system/status_codes.h"
#include "util/attributes.h"
#include <stdint.h>
typedef struct PACKED ExternalFlashEraseOptions {
uint32_t address;
uint32_t length;
} ExternalFlashEraseOptions;
typedef struct ExternalFlashEraseState {
uint32_t address;
uint32_t length;
unsigned int next_sector;
uint8_t cookie;
} ExternalFlashEraseState;
static int external_flash_domain_read(uint8_t *buf, uint32_t address, uint32_t length,
void *context) {
flash_read_bytes(buf, address, length);
return length;
}
static int external_flash_domain_write(uint8_t *buf, uint32_t address, uint32_t length,
void *context) {
flash_write_bytes(buf, address, length);
return length;
}
static int external_flash_domain_stat(uint8_t *resp, size_t resp_max_len, void *context) {
return E_INVALID_OPERATION;
}
static void prv_erase_sector(void *context, status_t result) {
ExternalFlashEraseState *state = context;
const unsigned int sectors_to_erase = (
state->length + SECTOR_SIZE_BYTES - 1) / SECTOR_SIZE_BYTES;
if (FAILED(result)) {
pulse_bulkio_erase_message_send(PulseBulkIODomainType_ExternalFlash, result, state->cookie);
kernel_free(state);
} else if (state->next_sector < sectors_to_erase) {
unsigned int sector_addr = state->address + state->next_sector * SECTOR_SIZE_BYTES;
state->next_sector += 1;
pulse_bulkio_erase_message_send(PulseBulkIODomainType_ExternalFlash, S_TRUE, state->cookie);
flash_erase_sector(sector_addr, prv_erase_sector, state);
} else {
pulse_bulkio_erase_message_send(PulseBulkIODomainType_ExternalFlash, S_SUCCESS, state->cookie);
kernel_free(state);
}
}
static status_t external_flash_domain_erase(uint8_t *packet_data, size_t length, uint8_t cookie) {
if (length != sizeof(ExternalFlashEraseOptions)) {
return E_INVALID_ARGUMENT;
}
ExternalFlashEraseOptions *options = (ExternalFlashEraseOptions*)packet_data;
ExternalFlashEraseState *state = kernel_malloc(sizeof(ExternalFlashEraseState));
*state = (ExternalFlashEraseState) {
.address = options->address,
.length = options->length,
.next_sector = 0,
.cookie = cookie
};
prv_erase_sector(state, 0);
// Return a non-zero code to indicate the erase is still in process
return S_TRUE;
}
static status_t external_flash_domain_open(uint8_t *packet_data, size_t length, void **resp) {
return S_SUCCESS;
}
static status_t external_flash_domain_close(void *context) {
return S_SUCCESS;
}
PulseBulkIODomainHandler pulse_bulkio_domain_external_flash = {
.id = PulseBulkIODomainType_ExternalFlash,
.open_proc = external_flash_domain_open,
.close_proc = external_flash_domain_close,
.read_proc = external_flash_domain_read,
.write_proc = external_flash_domain_write,
.stat_proc = external_flash_domain_stat,
.erase_proc = external_flash_domain_erase
};