mirror of https://github.com/jeelabs/esp-link.git
Bits and pieces for flashing an Arduino Mega. (#290)
dannybackx
7 years ago
committed by
Thorsten von Eicken
parent
c0cd698860
commit
bc87885a5f
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,14 @@ |
||||
// Copyright (c) 2016-2017 by Danny Backx, see LICENSE.txt in the esp-link repo
|
||||
|
||||
#ifndef CGIMEGA_H |
||||
#define CGIMEGA_H |
||||
|
||||
#include <httpd.h> |
||||
|
||||
int ICACHE_FLASH_ATTR cgiMegaSync(HttpdConnData *connData); |
||||
int ICACHE_FLASH_ATTR cgiMegaData(HttpdConnData *connData); |
||||
int ICACHE_FLASH_ATTR cgiMegaRead(HttpdConnData *connData); |
||||
int ICACHE_FLASH_ATTR cgiMegaFuse(HttpdConnData *connData); |
||||
int ICACHE_FLASH_ATTR cgiMegaRebootMCU(HttpdConnData *connData); |
||||
|
||||
#endif |
||||
@ -0,0 +1,149 @@ |
||||
// Copyright (c) 2015 by Thorsten von Eicken, see LICENSE.txt in the esp-link repo
|
||||
// Copyright (c) 2016-2017 by Danny Backx
|
||||
|
||||
#include <esp8266.h> |
||||
#include <osapi.h> |
||||
#include "cgi.h" |
||||
#include "config.h" |
||||
#include "uart.h" |
||||
#include "stk500v2.h" |
||||
#include "serbridge.h" |
||||
#include "serled.h" |
||||
|
||||
#include "pgmshared.h" |
||||
|
||||
struct optibootData *optibootData; |
||||
|
||||
char responseBuf[RESP_SZ]; // buffer to accumulate responses from optiboot
|
||||
short responseLen = 0; // amount accumulated so far
|
||||
char errMessage[ERR_MAX]; // error message
|
||||
|
||||
// verify that N chars are hex characters
|
||||
bool ICACHE_FLASH_ATTR checkHex(char *buf, short len) { |
||||
while (len--) { |
||||
char c = *buf++; |
||||
if ((c >= '0' && c <= '9') || (c >= 'A' && c <= 'F') || (c >= 'a' && c <= 'f')) |
||||
continue; |
||||
DBG("OB non-hex\n"); |
||||
os_sprintf(errMessage, "Non hex char in POST record: '%c'/0x%02x", c, c); |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
// get hex value of some hex characters
|
||||
uint32_t ICACHE_FLASH_ATTR getHexValue(char *buf, short len) { |
||||
uint32_t v = 0; |
||||
while (len--) { |
||||
v = (v<<4) | (uint32_t)(*buf & 0xf); |
||||
if (*buf > '9') v += 9; |
||||
buf++; |
||||
} |
||||
return v; |
||||
} |
||||
|
||||
// verify checksum
|
||||
static bool ICACHE_FLASH_ATTR verifyChecksum(char *buf, short len) { |
||||
uint8_t sum = 0; |
||||
while (len >= 2) { |
||||
sum += (uint8_t)getHexValue(buf, 2); |
||||
buf += 2; |
||||
len -= 2; |
||||
} |
||||
return sum == 0; |
||||
} |
||||
|
||||
// We've not built one function that works on both, but kept the different functions.
|
||||
// This calls one or the other
|
||||
static bool ICACHE_FLASH_ATTR programPage() { |
||||
if (optibootData->mega) |
||||
return megaProgramPage(); |
||||
return optibootProgramPage(); |
||||
} |
||||
|
||||
// Process a hex record -- assumes that the records starts with ':' & hex length
|
||||
bool ICACHE_FLASH_ATTR processRecord(char *buf, short len) { |
||||
buf++; len--; // skip leading ':'
|
||||
// check we have all hex chars
|
||||
if (!checkHex(buf, len)) return false; |
||||
// verify checksum
|
||||
if (!verifyChecksum(buf, len)) { |
||||
buf[len] = 0; |
||||
os_sprintf(errMessage, "Invalid checksum for record %s", buf); |
||||
return false; |
||||
} |
||||
// dispatch based on record type
|
||||
uint8_t type = getHexValue(buf+6, 2); |
||||
switch (type) { |
||||
case 0x00: { // Intel HEX data record
|
||||
//DBG("OB REC data %ld pglen=%d\n", getHexValue(buf, 2), optibootData->pageLen);
|
||||
uint32_t addr = getHexValue(buf+2, 4); |
||||
// check whether we need to program previous record(s)
|
||||
if (optibootData->pageLen > 0 && |
||||
addr != ((optibootData->address+optibootData->pageLen)&0xffff)) { |
||||
//DBG("OB addr chg\n");
|
||||
//DBG("processRecord addr chg, len %d, addr 0x%04x\n", optibootData->pageLen, addr);
|
||||
if (!programPage()) return false; |
||||
} |
||||
// set address, unless we're adding to the end (programPage may have changed pageLen)
|
||||
if (optibootData->pageLen == 0) { |
||||
optibootData->address = (optibootData->address & 0xffff0000) | addr; |
||||
//DBG("OB set-addr 0x%lx\n", optibootData->address);
|
||||
} |
||||
// append record
|
||||
uint16_t recLen = getHexValue(buf, 2); |
||||
for (uint16_t i=0; i<recLen; i++) |
||||
optibootData->pageBuf[optibootData->pageLen++] = getHexValue(buf+8+2*i, 2); |
||||
// program page, if we have a full page
|
||||
if (optibootData->pageLen >= optibootData->pgmSz) { |
||||
//DBG("OB full\n");
|
||||
DBG("processRecord %d, call programPage() %08x\n", optibootData->pgmSz, optibootData->address + optibootData->segment); |
||||
if (!programPage()) return false; |
||||
} |
||||
break; } |
||||
case 0x01: // Intel HEX EOF record
|
||||
DBG("OB EOF\n"); |
||||
// program any remaining partial page
|
||||
#if 1 |
||||
if (optibootData->pageLen > 0) { |
||||
// DBG("processRecord remaining partial page, len %d, addr 0x%04x\n", optibootData->pageLen, optibootData->address + optibootData->segment);
|
||||
if (!programPage()) return false; |
||||
} |
||||
optibootData->eof = true; |
||||
#else |
||||
if (optibootData->pageLen > 0) { |
||||
// HACK : fill up with 0xFF
|
||||
while (optibootData->pageLen < 256) { |
||||
optibootData->pageBuf[optibootData->pageLen++] = 0xFF; |
||||
} |
||||
if (!programPage()) return false; |
||||
} |
||||
optibootData->eof = true; |
||||
#endif |
||||
break; |
||||
case 0x04: // Intel HEX address record
|
||||
DBG("OB address 0x%x\n", getHexValue(buf+8, 4) << 16); |
||||
// program any remaining partial page
|
||||
if (optibootData->pageLen > 0) { |
||||
DBG("processRecord 0x04 remaining partial page, len %d, addr 0x%04x\n", |
||||
optibootData->pageLen, optibootData->address + optibootData->segment); |
||||
if (!programPage()) return false; |
||||
} |
||||
optibootData->address = getHexValue(buf+8, 4) << 16; |
||||
break; |
||||
case 0x05: // Intel HEX start address (MDK-ARM only)
|
||||
// ignore, there's no way to tell optiboot that...
|
||||
break; |
||||
case 0x02: // Intel HEX extended segment address record
|
||||
// Depending on the case, just ignoring this record could solve the problem
|
||||
// optibootData->segment = getHexValue(buf+8, 4) << 4;
|
||||
DBG("OB segment 0x%08X\n", optibootData->segment); |
||||
return true; |
||||
default: |
||||
// DBG("OB bad record type\n");
|
||||
DBG(errMessage, "Invalid/unknown record type: 0x%02x, packet %s", type, buf); |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
@ -0,0 +1,54 @@ |
||||
// Copyright (c) 2015 by Thorsten von Eicken, see LICENSE.txt in the esp-link repo
|
||||
// Copyright (c) 2017 by Danny Backx
|
||||
|
||||
#ifndef _PGM_SHARED_H_ |
||||
#define _PGM_SHARED_H_ |
||||
|
||||
#include "user_config.h" |
||||
|
||||
#define RESP_SZ 64 |
||||
#define ERR_MAX 128 |
||||
|
||||
extern char responseBuf[RESP_SZ]; |
||||
extern short responseLen; |
||||
extern char errMessage[ERR_MAX]; |
||||
|
||||
// structure used to remember request details from one callback to the next
|
||||
// allocated dynamically so we don't burn so much static RAM
|
||||
extern struct optibootData { |
||||
char *saved; // buffer for saved incomplete hex records
|
||||
char *pageBuf; // buffer for received data to be sent to AVR
|
||||
uint16_t pageLen; // number of bytes in pageBuf
|
||||
uint16_t pgmSz; // size of flash page to be programmed at a time
|
||||
uint32_t pgmDone; // number of bytes programmed
|
||||
uint32_t address; // address to write next page to
|
||||
uint32_t segment; // for extended segment addressing, added to the address field
|
||||
uint32_t startTime; // time of program POST request
|
||||
HttpdConnData *conn; // request doing the programming, so we can cancel it
|
||||
bool eof; // got EOF record
|
||||
|
||||
// Whether to use the Mega (STK500v2) protocol
|
||||
bool mega; |
||||
|
||||
// STK500v2 variables
|
||||
int hardwareVersion, |
||||
firmwareVersionMajor, |
||||
firmwareVersionMinor, |
||||
vTarget; |
||||
uint8_t signature[3]; |
||||
uint8_t lfuse, hfuse, efuse; |
||||
} *optibootData; |
||||
|
||||
bool ICACHE_FLASH_ATTR checkHex(char *buf, short len); |
||||
uint32_t ICACHE_FLASH_ATTR getHexValue(char *buf, short len); |
||||
bool ICACHE_FLASH_ATTR processRecord(char *buf, short len); |
||||
bool megaProgramPage(void); |
||||
bool optibootProgramPage(void); |
||||
|
||||
#ifdef OPTIBOOT_DBG |
||||
#define DBG(format, ...) do { os_printf(format, ## __VA_ARGS__); } while(0) |
||||
#else |
||||
#define DBG(format, ...) do { } while(0) |
||||
#endif |
||||
|
||||
#endif |
||||
@ -0,0 +1,114 @@ |
||||
//**** ATMEL AVR - A P P L I C A T I O N N O T E ************************
|
||||
//*
|
||||
//* Title: AVR068 - STK500 Communication Protocol
|
||||
//* Filename: command.h
|
||||
//* Version: 1.0
|
||||
//* Last updated: 31.01.2005
|
||||
//*
|
||||
//* Support E-mail: avr@atmel.com
|
||||
//*
|
||||
//**************************************************************************
|
||||
|
||||
// *****************[ STK message constants ]***************************
|
||||
|
||||
#define MESSAGE_START 0x1B //= ESC = 27 decimal
|
||||
#define TOKEN 0x0E |
||||
|
||||
// *****************[ STK general command constants ]**************************
|
||||
|
||||
#define CMD_SIGN_ON 0x01 |
||||
#define CMD_SET_PARAMETER 0x02 |
||||
#define CMD_GET_PARAMETER 0x03 |
||||
#define CMD_SET_DEVICE_PARAMETERS 0x04 |
||||
#define CMD_OSCCAL 0x05 |
||||
#define CMD_LOAD_ADDRESS 0x06 |
||||
#define CMD_FIRMWARE_UPGRADE 0x07 |
||||
|
||||
|
||||
// *****************[ STK ISP command constants ]******************************
|
||||
|
||||
#define CMD_ENTER_PROGMODE_ISP 0x10 |
||||
#define CMD_LEAVE_PROGMODE_ISP 0x11 |
||||
#define CMD_CHIP_ERASE_ISP 0x12 |
||||
#define CMD_PROGRAM_FLASH_ISP 0x13 |
||||
#define CMD_READ_FLASH_ISP 0x14 |
||||
#define CMD_PROGRAM_EEPROM_ISP 0x15 |
||||
#define CMD_READ_EEPROM_ISP 0x16 |
||||
#define CMD_PROGRAM_FUSE_ISP 0x17 |
||||
#define CMD_READ_FUSE_ISP 0x18 |
||||
#define CMD_PROGRAM_LOCK_ISP 0x19 |
||||
#define CMD_READ_LOCK_ISP 0x1A |
||||
#define CMD_READ_SIGNATURE_ISP 0x1B |
||||
#define CMD_READ_OSCCAL_ISP 0x1C |
||||
#define CMD_SPI_MULTI 0x1D |
||||
|
||||
// *****************[ STK PP command constants ]*******************************
|
||||
|
||||
#define CMD_ENTER_PROGMODE_PP 0x20 |
||||
#define CMD_LEAVE_PROGMODE_PP 0x21 |
||||
#define CMD_CHIP_ERASE_PP 0x22 |
||||
#define CMD_PROGRAM_FLASH_PP 0x23 |
||||
#define CMD_READ_FLASH_PP 0x24 |
||||
#define CMD_PROGRAM_EEPROM_PP 0x25 |
||||
#define CMD_READ_EEPROM_PP 0x26 |
||||
#define CMD_PROGRAM_FUSE_PP 0x27 |
||||
#define CMD_READ_FUSE_PP 0x28 |
||||
#define CMD_PROGRAM_LOCK_PP 0x29 |
||||
#define CMD_READ_LOCK_PP 0x2A |
||||
#define CMD_READ_SIGNATURE_PP 0x2B |
||||
#define CMD_READ_OSCCAL_PP 0x2C |
||||
|
||||
#define CMD_SET_CONTROL_STACK 0x2D |
||||
|
||||
// *****************[ STK HVSP command constants ]*****************************
|
||||
|
||||
#define CMD_ENTER_PROGMODE_HVSP 0x30 |
||||
#define CMD_LEAVE_PROGMODE_HVSP 0x31 |
||||
#define CMD_CHIP_ERASE_HVSP 0x32 |
||||
#define CMD_PROGRAM_FLASH_HVSP 0x33 |
||||
#define CMD_READ_FLASH_HVSP 0x34 |
||||
#define CMD_PROGRAM_EEPROM_HVSP 0x35 |
||||
#define CMD_READ_EEPROM_HVSP 0x36 |
||||
#define CMD_PROGRAM_FUSE_HVSP 0x37 |
||||
#define CMD_READ_FUSE_HVSP 0x38 |
||||
#define CMD_PROGRAM_LOCK_HVSP 0x39 |
||||
#define CMD_READ_LOCK_HVSP 0x3A |
||||
#define CMD_READ_SIGNATURE_HVSP 0x3B |
||||
#define CMD_READ_OSCCAL_HVSP 0x3C |
||||
|
||||
// *****************[ STK status constants ]***************************
|
||||
|
||||
// Success
|
||||
#define STATUS_CMD_OK 0x00 |
||||
|
||||
// Warnings
|
||||
#define STATUS_CMD_TOUT 0x80 |
||||
#define STATUS_RDY_BSY_TOUT 0x81 |
||||
#define STATUS_SET_PARAM_MISSING 0x82 |
||||
|
||||
// Errors
|
||||
#define STATUS_CMD_FAILED 0xC0 |
||||
#define STATUS_CKSUM_ERROR 0xC1 |
||||
#define STATUS_CMD_UNKNOWN 0xC9 |
||||
|
||||
// *****************[ STK parameter constants ]***************************
|
||||
#define PARAM_BUILD_NUMBER_LOW 0x80 |
||||
#define PARAM_BUILD_NUMBER_HIGH 0x81 |
||||
#define PARAM_HW_VER 0x90 |
||||
#define PARAM_SW_MAJOR 0x91 |
||||
#define PARAM_SW_MINOR 0x92 |
||||
#define PARAM_VTARGET 0x94 |
||||
#define PARAM_VADJUST 0x95 |
||||
#define PARAM_OSC_PSCALE 0x96 |
||||
#define PARAM_OSC_CMATCH 0x97 |
||||
#define PARAM_SCK_DURATION 0x98 |
||||
#define PARAM_TOPCARD_DETECT 0x9A |
||||
#define PARAM_STATUS 0x9C |
||||
#define PARAM_DATA 0x9D |
||||
#define PARAM_RESET_POLARITY 0x9E |
||||
#define PARAM_CONTROLLER_INIT 0x9F |
||||
|
||||
// *****************[ STK answer constants ]***************************
|
||||
|
||||
#define ANSWER_CKSUM_ERROR 0xB0 |
||||
|
||||
@ -0,0 +1,114 @@ |
||||
#! /bin/bash |
||||
# |
||||
# Flash an Arduino Mega with STK500v2 using the esp-link built-in programmer |
||||
# Basically we first reset the AVR and get in sync, and then send the hex file |
||||
# |
||||
# ---------------------------------------------------------------------------- |
||||
# "THE BEER-WARE LICENSE" (Revision 42): |
||||
# Thorsten von Eicken wrote this file. As long as you retain |
||||
# this notice you can do whatever you want with this stuff. If we meet some day, |
||||
# and you think this stuff is worth it, you can buy me a beer in return. |
||||
# |
||||
# Danny Backx wrote the changes for Arduino Mega. |
||||
# ---------------------------------------------------------------------------- |
||||
|
||||
show_help() { |
||||
cat <<EOT |
||||
Usage: ${0##*/} [-options...] hostname sketch.hex |
||||
Flash the Mega running optiboot attached to esp-link with the sketch. |
||||
Note : this is for stk500v2 MCUs, use avrflash for Arduino Uno etc instead. |
||||
-v Be verbose |
||||
-h show this help |
||||
|
||||
Example: ${0##*/} -v esp-link mysketch.hex |
||||
${0##*/} 192.168.4.1 mysketch.hex |
||||
EOT |
||||
} |
||||
|
||||
if ! which curl >/dev/null; then |
||||
echo "ERROR: Cannot find curl: it is required for this script." >&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
start=`date +%s` |
||||
|
||||
# ===== Parse arguments |
||||
|
||||
verbose= |
||||
|
||||
while getopts "hvx:" opt; do |
||||
case "$opt" in |
||||
h) show_help; exit 0 ;; |
||||
v) verbose=1 ;; |
||||
x) foo="$OPTARG" ;; |
||||
'?') show_help >&2; exit 1 ;; |
||||
esac |
||||
done |
||||
|
||||
# Shift off the options and optional --. |
||||
shift "$((OPTIND-1))" |
||||
|
||||
# Get the fixed arguments |
||||
if [[ $# != 2 ]]; then |
||||
show_help >&2 |
||||
exit 1 |
||||
fi |
||||
hostname=$1 |
||||
hex=$2 |
||||
|
||||
re='[-A-Za-z0-9.]+' |
||||
if [[ ! "$hostname" =~ $re ]]; then |
||||
echo "ERROR: hostname ${hostname} is not a valid hostname or ip address" >&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
if [[ ! -r "$hex" ]]; then |
||||
echo "ERROR: cannot read hex file ($hex)" >&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
# ===== Get AVR in sync |
||||
|
||||
[[ -n "$verbose" ]] && echo "Resetting AVR with http://$hostname/pgmmega/sync" >&2 |
||||
v=; [[ -n "$verbose" ]] && v=-v |
||||
sync=`curl -m 10 $v -s -w '%{http_code}' -XPOST "http://$hostname/pgmmega/sync"` |
||||
if [[ $? != 0 || "$sync" != 204 ]]; then |
||||
echo "Error resetting AVR" >&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
while true; do |
||||
# sync=`curl -m 10 $v -s "http://$hostname/pgmmega/sync"` |
||||
sync=`curl $v -s "http://$hostname/pgmmega/sync"` |
||||
if [[ $? != 0 ]]; then |
||||
echo "Error checking sync" >&2 |
||||
exit 1 |
||||
fi |
||||
case "$sync" in |
||||
SYNC*) |
||||
echo "AVR in $sync" >&2 |
||||
break;; |
||||
"NOT READY"*) |
||||
[[ -n "$verbose" ]] && echo " Waiting for sync..." >&2 |
||||
;; |
||||
*) |
||||
echo "Error checking sync: $sync" >&2 |
||||
exit 1 |
||||
;; |
||||
esac |
||||
sleep 0.1 |
||||
done |
||||
|
||||
# ===== Send HEX file |
||||
|
||||
[[ -n "$verbose" ]] && echo "Sending HEX file for programming" >&2 |
||||
sync=`curl -m 20 $v -s -g -d "@$hex" "http://$hostname/pgmmega/upload"` |
||||
echo $sync |
||||
if [[ $? != 0 || ! "$sync" =~ ^Success ]]; then |
||||
echo "Error programming AVR" >&2 |
||||
exit 1 |
||||
fi |
||||
|
||||
sec=$(( `date +%s` - $start )) |
||||
echo "Success, took $sec seconds" >&2 |
||||
exit 0 |
||||
Loading…
Reference in new issue