pinebuds/services/tota/tota_buffer_manager.cpp
Ben V. Brown 75381150fd Formatting
Formatting Pass 1

Lots of fixups to adding stdint and stdbool all over the place

Formatting Pass 2
Formatting Pass 3
Formatting Pass 4

Update app_bt_stream.cpp
2023-02-02 07:56:49 +11:00

127 lines
No EOL
4.3 KiB
C++

/***************************************************************************
*
* Copyright 2015-2019 BES.
* All rights reserved. All unpublished rights reserved.
*
* No part of this work may be used or reproduced in any form or by any
* means, or stored in a database or retrieval system, without prior written
* permission of BES.
*
* Use of this work is governed by a license granted by BES.
* This work contains confidential and proprietary information of
* BES. which is protected by copyright, trade secret,
* trademark and other intellectual property rights.
*
****************************************************************************/
#include "tota_buffer_manager.h"
#include "app_tota_cmd_code.h"
#include "cmsis_os.h"
#include "string.h"
osMutexDef(stream_buf_mutex);
osMutexId stream_buf_mutex_id;
static stream_buf_t stream_buf;
static osThreadId father_thread_tid;
void tota_stream_buffer_init(osThreadId tid) {
stream_buf.dataSize = 0;
stream_buf.availableSpace = TOTA_STREAM_BUF_SIZE;
stream_buf.readPos = 0;
stream_buf.writePos = 0;
stream_buf.flushBytes = 0;
stream_buf_mutex_id = osMutexCreate(osMutex(stream_buf_mutex));
father_thread_tid = tid;
}
bool tota_stream_buffer_write(uint8_t *buf, uint32_t bufLen) {
osMutexWait(stream_buf_mutex_id, osWaitForever);
if (stream_buf.availableSpace > bufLen) {
// write to stream buf
if (stream_buf.writePos + bufLen <= TOTA_STREAM_BUF_SIZE) {
memcpy(stream_buf.buf + stream_buf.writePos, buf, bufLen);
stream_buf.writePos += bufLen;
if (stream_buf.writePos == TOTA_STREAM_BUF_SIZE)
stream_buf.writePos = 0;
} else {
uint32_t part1 = TOTA_STREAM_BUF_SIZE - stream_buf.writePos;
memcpy(stream_buf.buf + stream_buf.writePos, buf, part1);
memcpy(stream_buf.buf, buf + part1, bufLen - part1);
stream_buf.writePos = bufLen - part1;
}
stream_buf.dataSize += bufLen;
stream_buf.availableSpace -= bufLen;
TOTA_LOG_DBG(1, "buffer> after write: %5u bytes", bufLen);
TOTA_LOG_DBG(4, "buffer> size:%5u available:%5u writepos:%5u readpos:%5u:",
stream_buf.dataSize, stream_buf.availableSpace,
stream_buf.writePos, stream_buf.readPos);
osMutexRelease(stream_buf_mutex_id);
/* set signal to father thread, has data */
osSignalSet(father_thread_tid, 0x0001);
return true;
} else {
osMutexRelease(stream_buf_mutex_id);
return false;
}
}
bool tota_stream_buffer_read(uint8_t *rbuf, uint32_t readSize,
uint32_t *flushbytes) {
osMutexWait(stream_buf_mutex_id, osWaitForever);
if (readSize < stream_buf.dataSize) {
if (stream_buf.readPos + readSize <= TOTA_STREAM_BUF_SIZE) {
memcpy(rbuf, stream_buf.buf + stream_buf.readPos, readSize);
stream_buf.readPos += readSize;
if (stream_buf.readPos == TOTA_STREAM_BUF_SIZE)
stream_buf.readPos = 0;
} else {
uint32_t part1 = TOTA_STREAM_BUF_SIZE - stream_buf.readPos;
memcpy(rbuf, stream_buf.buf + stream_buf.readPos, part1);
memcpy(rbuf + part1, stream_buf.buf, readSize - part1);
stream_buf.readPos = readSize - part1;
}
stream_buf.dataSize -= readSize;
stream_buf.availableSpace += readSize;
/* debug */
TOTA_LOG_DBG(1, "buffer> after read: %5u bytes", readSize);
TOTA_LOG_DBG(4, "buffer> size:%5u available:%5u writepos:%5u readpos:%5u:",
stream_buf.dataSize, stream_buf.availableSpace,
stream_buf.writePos, stream_buf.readPos);
if (flushbytes != NULL) {
*flushbytes = stream_buf.flushBytes;
stream_buf.flushBytes = 0;
}
osMutexRelease(stream_buf_mutex_id);
return true;
} else {
osMutexRelease(stream_buf_mutex_id);
return false;
}
}
void tota_stream_buffer_flush(void) {
osMutexWait(stream_buf_mutex_id, osWaitForever);
stream_buf.flushBytes = stream_buf.dataSize;
stream_buf.availableSpace = TOTA_STREAM_BUF_SIZE;
stream_buf.dataSize = 0;
stream_buf.readPos = 0;
stream_buf.writePos = 0;
osMutexRelease(stream_buf_mutex_id);
}
void tota_stream_buffer_clean(void) {
osMutexWait(stream_buf_mutex_id, osWaitForever);
stream_buf.flushBytes = 0;
stream_buf.availableSpace = TOTA_STREAM_BUF_SIZE;
stream_buf.dataSize = 0;
stream_buf.readPos = 0;
stream_buf.writePos = 0;
osMutexRelease(stream_buf_mutex_id);
}