You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.5 KiB
101 lines
2.5 KiB
//
|
|
// Created by MightyPork on 2017/12/23.
|
|
//
|
|
|
|
#include "platform.h"
|
|
|
|
#include <TinyFrame.h>
|
|
|
|
#include "messages.h"
|
|
#include "utils/payload_parser.h"
|
|
#include "utils/payload_builder.h"
|
|
|
|
/** Buffer for preparing bulk chunks */
|
|
static uint8_t bulkread_buffer[BULKREAD_MAX_CHUNK];
|
|
|
|
/**
|
|
* TF listener for the bulk read transaction
|
|
*/
|
|
static TF_Result bulkread_lst(TinyFrame *tf, TF_Msg *msg)
|
|
{
|
|
struct bulk_read *bulk = msg->userdata;
|
|
|
|
// this is a final call before timeout, to clean up
|
|
if (msg->data == NULL) {
|
|
dbg("Bulk rx lst cleanup\r\n");
|
|
goto close;
|
|
}
|
|
|
|
assert_param(NULL != bulk);
|
|
|
|
if (msg->type == MSG_BULK_ABORT) {
|
|
goto close;
|
|
} else if (msg->type == MSG_BULK_READ_POLL) {
|
|
PayloadParser pp = pp_start(msg->data, msg->len, NULL);
|
|
uint32_t chunk = pp_u32(&pp);
|
|
|
|
// if past len, say we're done and close
|
|
if (bulk->offset >= bulk->len) {
|
|
TF_ClearMsg(msg);
|
|
msg->frame_id = bulk->frame_id;
|
|
msg->type = MSG_BULK_END;
|
|
TF_Respond(tf, msg);
|
|
goto close;
|
|
}
|
|
|
|
chunk = MIN(chunk, bulk->len - bulk->offset);
|
|
chunk = MIN(chunk, BULKREAD_MAX_CHUNK);
|
|
|
|
bulk->read(bulk, chunk, bulkread_buffer);
|
|
|
|
TF_ClearMsg(msg);
|
|
msg->frame_id = bulk->frame_id;
|
|
msg->type = MSG_BULK_DATA;
|
|
msg->data = bulkread_buffer;
|
|
msg->len = (TF_LEN) chunk;
|
|
TF_Respond(tf, msg);
|
|
|
|
// advance the position pointer
|
|
bulk->offset += chunk;
|
|
}
|
|
|
|
msg->userdata = bulk; // We must put it back
|
|
return TF_RENEW;
|
|
|
|
close:
|
|
if (msg->userdata) {
|
|
// Ask user to free the bulk and userdata
|
|
bulk->read(bulk, 0, NULL);
|
|
msg->userdata = NULL;
|
|
}
|
|
return TF_CLOSE;
|
|
}
|
|
|
|
/** Start the bulk read flow */
|
|
void bulkread_start(TinyFrame *tf, struct bulk_read *bulk)
|
|
{
|
|
assert_param(bulk);
|
|
assert_param(bulk->len);
|
|
assert_param(bulk->read);
|
|
|
|
bulk->offset = 0;
|
|
|
|
{
|
|
uint8_t buf[8];
|
|
PayloadBuilder pb = pb_start(buf, 4, NULL);
|
|
pb_u32(&pb, bulk->len);
|
|
pb_u32(&pb, BULKREAD_MAX_CHUNK);
|
|
|
|
// We use userdata1 to hold a reference to the bulk transfer
|
|
TF_Msg msg = {
|
|
.type = MSG_BULK_READ_OFFER,
|
|
.frame_id = bulk->frame_id,
|
|
.is_response = true, // this ensures the frame_id is not re-generated
|
|
.data = buf,
|
|
.len = (TF_LEN) pb_length(&pb),
|
|
.userdata = bulk,
|
|
};
|
|
|
|
TF_Query(tf, &msg, bulkread_lst, BULK_LST_TIMEOUT_MS);
|
|
}
|
|
}
|
|
|