avr/bootloader
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add flash reading functionality

Browse Source
This commit is contained in:
BlackMark 2020-04-11 10:35:52 +02:00
parent 9ba16e71e1
commit 98d6c9720f
1 changed files with 47 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
stk500v2/main.cpp
View File

@ -3,6 +3,7 @@
#include "uart/uart.hpp" #include "uart/uart.hpp"
#include <avr/boot.h> #include <avr/boot.h>
#include <avr/pgmspace.h>
#include "command.hpp" #include "command.hpp"
@ -110,6 +111,20 @@ static inline bool isReadLockIsp(const Message &msg)
return false; return false;
} }
static inline bool isLoadAddress(const Message &msg)
{
if (msg.size == 5 && msg.body[0] == CMD_LOAD_ADDRESS)
return true;
return false;
}
static inline bool isReadFlashIsp(const Message &msg)
{
if (msg.size == 4 && msg.body[0] == CMD_READ_FLASH_ISP)
return true;
return false;
}
static inline bool isLeaveProgmodeIsp(const Message &msg) static inline bool isLeaveProgmodeIsp(const Message &msg)
{ {
if (msg.size == 3 && msg.body[0] == CMD_LEAVE_PROGMODE_ISP) if (msg.size == 3 && msg.body[0] == CMD_LEAVE_PROGMODE_ISP)
@ -226,6 +241,28 @@ static inline void formatReadLockIspAnswer(Message &msg)
msg.checksum = calcChecksum(msg); msg.checksum = calcChecksum(msg);
} }
static inline void formatLoadAddressAnswer(Message &msg)
{
msg.size = 2;
msg.body[1] = STATUS_CMD_OK;
msg.checksum = calcChecksum(msg);
}
static inline void formatReadFlashIspAnswer(Message &msg, uint32_t &addr)
{
const uint16_t numBytes = static_cast<uint16_t>(msg.body[1]) << 8 | msg.body[2];
msg.size = 3 + numBytes;
msg.body[1] = STATUS_CMD_OK;
for (uint16_t i = 0; i < numBytes; ++i) {
msg.body[i + 2] = pgm_read_byte(static_cast<uint16_t>(addr + i));
}
addr += numBytes;
msg.body[numBytes + 2] = STATUS_CMD_OK;
msg.checksum = calcChecksum(msg);
}
static inline void formatLeaveProgmodeIspAnswer(Message &msg) static inline void formatLeaveProgmodeIspAnswer(Message &msg)
{ {
msg.size = 2; msg.size = 2;
@ -245,6 +282,8 @@ static inline void formatErrorAnswer(Message &msg)
static inline void handleMessage(Message &msg) static inline void handleMessage(Message &msg)
{ {
static uint32_t s_address = 0;
if (isSignOn(msg)) if (isSignOn(msg))
formatSignOnAnswer(msg); formatSignOnAnswer(msg);
else if (isGetParameter(msg)) else if (isGetParameter(msg))
@ -259,6 +298,14 @@ static inline void handleMessage(Message &msg)
formatReadFuseIspAnswer(msg); formatReadFuseIspAnswer(msg);
else if (isReadLockIsp(msg)) else if (isReadLockIsp(msg))
formatReadLockIspAnswer(msg); formatReadLockIspAnswer(msg);
else if (isLoadAddress(msg)) {
s_address = msg.body[1];
s_address = (s_address << 8) | msg.body[2];
s_address = (s_address << 8) | msg.body[3];
s_address = (s_address << 8) | msg.body[4];
formatLoadAddressAnswer(msg);
} else if (isReadFlashIsp(msg))
formatReadFlashIspAnswer(msg, s_address);
else if (isLeaveProgmodeIsp(msg)) else if (isLeaveProgmodeIsp(msg))
formatLeaveProgmodeIspAnswer(msg); formatLeaveProgmodeIspAnswer(msg);
else else