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.
92 lines
2.3 KiB
92 lines
2.3 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"
|
|
|
|
/**
|
|
* TF listener for the bulk write transaction
|
|
*/
|
|
static TF_Result bulkwrite_lst(TinyFrame *tf, TF_Msg *msg)
|
|
{
|
|
BulkWrite *bulk = msg->userdata;
|
|
|
|
// this is a final call before timeout, to clean up
|
|
if (msg->data == NULL) {
|
|
if (bulk->offset != bulk->len) {
|
|
TF_Error("Bulk write ended prematurely with %d written out of %d", (int)bulk->offset, (int)bulk->len);
|
|
}
|
|
goto close;
|
|
}
|
|
|
|
assert_param(NULL != bulk);
|
|
|
|
if (msg->type == MSG_BULK_ABORT) {
|
|
goto close;
|
|
}
|
|
else if (msg->type == MSG_BULK_DATA || msg->type == MSG_BULK_END) {
|
|
// if past len, speak up
|
|
if (bulk->offset >= bulk->len) {
|
|
com_respond_error(bulk->frame_id, E_PROTOCOL_BREACH);
|
|
goto close;
|
|
}
|
|
|
|
bulk->write(bulk, msg->data, msg->len);
|
|
|
|
// Return SUCCESS
|
|
com_respond_ok(msg->frame_id);
|
|
|
|
// advance the position pointer (can be used by the write callback)
|
|
bulk->offset += msg->len;
|
|
|
|
if (msg->type == MSG_BULK_END) {
|
|
// this was the last chunk
|
|
goto close;
|
|
}
|
|
}
|
|
|
|
return TF_RENEW;
|
|
|
|
close:
|
|
if (bulk) {
|
|
// Ask user to free the bulk and userdata
|
|
bulk->write(bulk, NULL, 0);
|
|
msg->userdata = NULL;
|
|
}
|
|
return TF_CLOSE;
|
|
}
|
|
|
|
|
|
void bulkwrite_start(TinyFrame *tf, BulkWrite *bulk)
|
|
{
|
|
assert_param(bulk);
|
|
assert_param(bulk->len);
|
|
assert_param(bulk->write);
|
|
|
|
bulk->offset = 0;
|
|
|
|
{
|
|
uint8_t buf[8];
|
|
PayloadBuilder pb = pb_start(buf, 8, NULL);
|
|
pb_u32(&pb, bulk->len); // total expected len
|
|
pb_u32(&pb, TF_MAX_PAYLOAD_RX); // max chunk size
|
|
|
|
// We use userdata1 to hold a reference to the bulk transfer
|
|
TF_Msg msg = {
|
|
.type = MSG_BULK_WRITE_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, bulkwrite_lst, BULK_LST_TIMEOUT_MS);
|
|
}
|
|
}
|
|
|