diff options
Diffstat (limited to 'bootloaders')
| -rw-r--r-- | bootloaders/diskloader/Arduino_Leonardo.inf | 106 | ||||
| -rw-r--r-- | bootloaders/diskloader/DiskLoader.hex | 102 | ||||
| -rw-r--r-- | bootloaders/diskloader/Makefile | 99 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/DiskLoader.cpp | 251 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/Platform.h | 56 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/SCSI.cpp | 213 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/ThinFAT.cpp | 130 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/USBCore.cpp | 526 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/USBCore.h | 261 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/USBDesc.cpp | 88 | ||||
| -rw-r--r-- | bootloaders/diskloader/src/USBDesc.h | 81 | 
11 files changed, 1913 insertions, 0 deletions
| diff --git a/bootloaders/diskloader/Arduino_Leonardo.inf b/bootloaders/diskloader/Arduino_Leonardo.inf new file mode 100644 index 0000000..8c3feb3 --- /dev/null +++ b/bootloaders/diskloader/Arduino_Leonardo.inf @@ -0,0 +1,106 @@ +;************************************************************ +; Windows USB CDC ACM Setup File +; Copyright (c) 2000 Microsoft Corporation + + +[Version] +Signature="$Windows NT$" +Class=Ports +ClassGuid={4D36E978-E325-11CE-BFC1-08002BE10318} +Provider=%MFGNAME% +LayoutFile=layout.inf +CatalogFile=%MFGFILENAME%.cat +DriverVer=11/15/2007,5.1.2600.0 + +[Manufacturer] +%MFGNAME%=DeviceList, NTamd64 + +[DestinationDirs] +DefaultDestDir=12 + + +;------------------------------------------------------------------------------ +;  Windows 2000/XP/Vista-32bit Sections +;------------------------------------------------------------------------------ + +[DriverInstall.nt] +include=mdmcpq.inf +CopyFiles=DriverCopyFiles.nt +AddReg=DriverInstall.nt.AddReg + +[DriverCopyFiles.nt] +usbser.sys,,,0x20 + +[DriverInstall.nt.AddReg] +HKR,,DevLoader,,*ntkern +HKR,,NTMPDriver,,%DRIVERFILENAME%.sys +HKR,,EnumPropPages32,,"MsPorts.dll,SerialPortPropPageProvider" + +[DriverInstall.nt.Services] +AddService=usbser, 0x00000002, DriverService.nt + +[DriverService.nt] +DisplayName=%SERVICE% +ServiceType=1 +StartType=3 +ErrorControl=1 +ServiceBinary=%12%\%DRIVERFILENAME%.sys + +;------------------------------------------------------------------------------ +;  Vista-64bit Sections +;------------------------------------------------------------------------------ + +[DriverInstall.NTamd64] +include=mdmcpq.inf +CopyFiles=DriverCopyFiles.NTamd64 +AddReg=DriverInstall.NTamd64.AddReg + +[DriverCopyFiles.NTamd64] +%DRIVERFILENAME%.sys,,,0x20 + +[DriverInstall.NTamd64.AddReg] +HKR,,DevLoader,,*ntkern +HKR,,NTMPDriver,,%DRIVERFILENAME%.sys +HKR,,EnumPropPages32,,"MsPorts.dll,SerialPortPropPageProvider" + +[DriverInstall.NTamd64.Services] +AddService=usbser, 0x00000002, DriverService.NTamd64 + +[DriverService.NTamd64] +DisplayName=%SERVICE% +ServiceType=1 +StartType=3 +ErrorControl=1 +ServiceBinary=%12%\%DRIVERFILENAME%.sys + + +;------------------------------------------------------------------------------ +;  Vendor and Product ID Definitions +;------------------------------------------------------------------------------ +; When developing your USB device, the VID and PID used in the PC side +; application program and the firmware on the microcontroller must match. +; Modify the below line to use your VID and PID.  Use the format as shown below. +; Note: One INF file can be used for multiple devices with different VID and PIDs. +; For each supported device, append ",USB\VID_xxxx&PID_yyyy" to the end of the line. +;------------------------------------------------------------------------------ +[SourceDisksFiles] +[SourceDisksNames] +[DeviceList] +%DESCRIPTION%=DriverInstall, USB\VID_2341&PID_0020&MI_00 + +[DeviceList.NTamd64] +%DESCRIPTION%=DriverInstall, USB\VID_2341&PID_0020&MI_00 + + +;------------------------------------------------------------------------------ +;  String Definitions +;------------------------------------------------------------------------------ +;Modify these strings to customize your device +;------------------------------------------------------------------------------ +[Strings] +MFGFILENAME="CDC_vista" +DRIVERFILENAME ="usbser" +MFGNAME="Arduino LLC (www.arduino.cc)" +INSTDISK="Arduino Leonardo Driver Installer" +DESCRIPTION="Arduino Lenonardo" +SERVICE="USB RS-232 Emulation Driver"
\ No newline at end of file diff --git a/bootloaders/diskloader/DiskLoader.hex b/bootloaders/diskloader/DiskLoader.hex new file mode 100644 index 0000000..ced02bd --- /dev/null +++ b/bootloaders/diskloader/DiskLoader.hex @@ -0,0 +1,102 @@ +:1078000011241FBECFEFDAE0DEBFCDBFD0C04101F3
 +:1078100042144505560455026403740300001E9586
 +:1078200087020110030000C18081C106C0FF0A0069
 +:107830000CA10185037508150026FF00954009017C
 +:107840008102954009029102C0040309041A033021
 +:1078500000300030003000300030003000300031A7
 +:1078600000370030003100120100020200004041E8
 +:1078700023200000010000010112010002000000AD
 +:1078800040412320000001000001010902640003BF
 +:1078900001008032080B000202020100090400000E
 +:1078A0000102020000052400100105240101010469
 +:1078B0002402020524060001070581031000400987
 +:1078C000040100020A000000070502024000000750
 +:1078D00005830240000009040200010300000009C2
 +:1078E0002101010001221E00070584034000400021
 +:1078F00020918A0130918B012F5F3F4F30938B0194
 +:1079000020938A01832F9927982F87FD9095990FAF
 +:10791000921710F447980895479A089581E080934C
 +:10792000E0001092E200EE27FF27099408950F93DC
 +:107930001F93CF93DF93982FEB01042F10E088E77C
 +:107940006030780730F411E083E0FB01809357004A
 +:10795000E895892F68E071E0402FE2D0112311F102
 +:1079600007B600FCFDCF402F4695FE01A8E0B1E030
 +:1079700020E031E009C08D919D910C0130935700BA
 +:10798000E895112432962F5F2417A8F385E0FE01B5
 +:1079900080935700E89507B600FCFDCF81E1809306
 +:1079A0005700E895DF91CF911F910F910895769A36
 +:1079B000289A47983F9A209A6E9A90E8909361008F
 +:1079C0001092610081E885BF95BF91D084E18093DA
 +:1079D000880180E180938901E0E0F0E08591949155
 +:1079E0008F5F9F4F19F081E080938D01EE24FF247B
 +:1079F000BB24B39421E0C22ED12CE1D1082F8EE01C
 +:107A000098E7FC012491319602964491201711F0D9
 +:107A10002223B9F710928F0110928E0182E068E064
 +:107A200071E07ED0013479F46091080160586330D0
 +:107A300028F0683111F064E001C063E0C62FD0E0A7
 +:107A4000CF5DD7480EC0063571F4809108018033B0
 +:107A500011F011E022C080910A01C82FD0E0C25E6F
 +:107A6000D74811E022C0053721F413E0CEE1D8E772
 +:107A70001CC0053539F4E0900801F0900901EE0CC6
 +:107A8000FF1C0AC0043631F482E0B70140910901BD
 +:107A90004EDF02C0043721F010E0C5E2D8E705C090
 +:107AA00010910901E701E10EF11C89D18097B1F431
 +:107AB00083E068E871E041E050E049D0112329F00B
 +:107AC000412F50E083E8BE0142D083E469E871E0D1
 +:107AD00041E050E03CD0013509F08FCFD0928F01CA
 +:107AE000C0928E01B0928D01EE24FF2486CF1092B9
 +:107AF0008F0110928E0110928C0110928D0181E005
 +:107B00008093D70080EA8093D80082E189BD09B4D0
 +:107B100000FEFDCF80E98093D8001092E000089528
 +:107B2000FB018093E9009BE60FC08091E80085FF90
 +:107B3000FCCF28988091F10081938091E80085FD29
 +:107B400002C09093E8004150442379F70895982F9C
 +:107B5000FB01282F207287708093E9003AE315C05B
 +:107B60008091E80085FFFCCF97FF02C0849101C09F
 +:107B700080813196211180E076988093F100809188
 +:107B8000E80085FD02C03093E800415050408FEF7F
 +:107B90004F3F580729F796FF03C08AE38093E80018
 +:107BA000089580919101813281F4769820E030E04F
 +:107BB00003C080818093F100F901E050FF4F2F5FF7
 +:107BC0003F4F28303105A9F71FC08032B9F48091AA
 +:107BD000E80082FFFCCF289820E030E003C08091CD
 +:107BE000F1008083F901E050FF4F2F5F3F4F2830B5
 +:107BF0003105A9F78BEF8093E80006C0823221F4AB
 +:107C0000809192018093070181E008952091930172
 +:107C1000223231F1213081F48091960190919701C7
 +:107C2000089711F42093980180919801882309F016
 +:107C30003BC0E9E7F8E73AC0223021F484E6EBE8FC
 +:107C4000F8E711C0233071F580919201882319F46F
 +:107C5000E9E4F8E72BC0813029F5EDE4F8E726C028
 +:107C600081E0EBE2F8E790919601282F981708F44D
 +:107C7000292F90E03EEF8091E8008570E1F380913C
 +:107C8000E80082FD0FC09F5F849176988093F10099
 +:107C9000892F8F7311F43093E8003196921758F3BF
 +:107CA00081E0089580E00895E7E6F8E78491DBCF6E
 +:107CB0001092E9008091E80083FF62C02898E0E913
 +:107CC000F1E003C08091F100819381E0E839F80789
 +:107CD000C9F782EF8093E8008091900187FF05C08B
 +:107CE0008091E80080FFFCCF03C08EEF8093E80016
 +:107CF00080919101853051F48091E80080FFFCCFA4
 +:107D00008091920180688093E30033C08930E1F470
 +:107D10008091920180938C01E7E2F8E791E031E0F5
 +:107D200026E39093E9003093EB0084918093EC007C
 +:107D30002093ED009F5F3196953099F78EE7809301
 +:107D4000EA001092EA0015C0883031F480918C016D
 +:107D500076988093F1000DC0882321F476981092D4
 +:107D6000F10007C0863011F451DF01C01ADF88230B
 +:107D700021F08EEF8093E800089581E28093EB007C
 +:107D800008958091E1001092E100282F83FF0CC03C
 +:107D90001092E90081E08093EB001092EC0082E306
 +:107DA0008093ED0010928C0122FF08C080918D011C
 +:107DB000882321F410928F0110928E0108951F9351
 +:107DC00012E076DFDEDF1093E9008091E80085FFA6
 +:107DD0000EC028989091F1008091E80085FD03C0C5
 +:107DE0008BE68093E800892F90E01F91089584E648
 +:107DF00000000000000000008150D1F7289A769A18
 +:107E000080918E0190918F01019790938F018093C3
 +:107E10008E0180918E0190918F01892B19F4769AB1
 +:087E2000289A7CDD65DDCDCF61
 +:087E280000E100000000000071
 +:040000030000780081
 +:00000001FF
 diff --git a/bootloaders/diskloader/Makefile b/bootloaders/diskloader/Makefile new file mode 100644 index 0000000..ca6e11e --- /dev/null +++ b/bootloaders/diskloader/Makefile @@ -0,0 +1,99 @@ +############################################################################### +# Makefile for DiskLoader +############################################################################### + +## General Flags +PROJECT = DiskLoader +TARGET = DiskLoader.elf +CC = avr-gcc + +# BOARD2 +MCU = atmega32u4 +AVR_FREQ   = 16000000L + + +# Change if your programmer is different +AVRDUDE_PROGRAMMER = avrispmkII +AVRDUDE_PORT = usb	    + +# program name should not be changed... +PROGRAM    = DiskLoader + +AVRDUDE = avrdude +AVRDUDE_FLAGS = -p $(MCU) -P $(AVRDUDE_PORT) -c $(AVRDUDE_PROGRAMMER) -p $(MCU) + +## Options common to compile, link and assembly rules +COMMON = -mmcu=$(MCU) + +override CFLAGS = -g -Wall -Os -mmcu=$(MCU) -DF_CPU=$(AVR_FREQ) $(DEFS) -ffunction-sections -gdwarf-2 -fdata-sections -fno-split-wide-types + +## Assembly specific flags +ASMFLAGS = $(COMMON) +ASMFLAGS += $(CFLAGS) +ASMFLAGS += -x assembler-with-cpp -Wa,-gdwarf2 + +## Linker flags +LDFLAGS = $(COMMON) +LDFLAGS += -Wl,-gc-sections,-Map=DiskLoader.map,--section-start=.text=0x7800,--relax +LDFLAGS += -nodefaultlibs -nostartfiles + + +## Intel Hex file production flags +HEX_EEPROM_FLAGS = -j .eeprom +HEX_EEPROM_FLAGS += --set-section-flags=.eeprom="alloc,load" +HEX_EEPROM_FLAGS += --change-section-lma .eeprom=0 --no-change-warnings + +## Objects explicitly added by the user +LINKONLYOBJECTS =  + +MODULES   := . +SRC_DIR   := $(addprefix src/,$(MODULES)) +BUILD_DIR := $(addprefix build/,$(MODULES)) + +SRC       := $(foreach sdir,$(SRC_DIR),$(wildcard $(sdir)/*.cpp)) +OBJ       := $(patsubst src/%.cpp,build/%.o,$(SRC)) +DEP		  := $(OBJ:%.o=%.d) +INCLUDES  := $(addprefix -I,$(SRC_DIR)) + +vpath %.cpp $(SRC_DIR) + +.PHONY: all checkdirs clean + +all: checkdirs $(TARGET) DiskLoader.hex DiskLoader.lss size + +-include $(DEP) + +checkdirs: $(BUILD_DIR) + +$(BUILD_DIR): +	@mkdir -p $@ + +clean: +	@rm -rf build/ +	@rm -f *.hex +	@rm -f *.elf +	@rm -f *.lss +	@rm -f *.map + +define make-goal +$1/%.o: %.cpp +	$(CC) $(INCLUDES) $(CFLAGS) -c $$< -MD -o $$@ +endef + +$(foreach bdir,$(BUILD_DIR),$(eval $(call make-goal,$(bdir)))) + +$(TARGET): $(OBJ) +	$(CC) $(LDFLAGS) $(LINKONLYOBJECTS) $(LIBDIRS) $(LIBS) $^ -o $@ + +%.hex: $(TARGET) +	avr-objcopy -O ihex $(HEX_FLASH_FLAGS)  $< $@ +	 +%.lss: $(TARGET) +	avr-objdump -h -S $< > $@ + +size: $(TARGET) +	@echo +#	@avr-size -C --mcu=${MCU} ${TARGET}.elf + +program: $(TARGET).hex +	$(AVRDUDE) $(AVRDUDE_FLAGS) -B 5 -u -U flash:w:$(TARGET).hex
\ No newline at end of file diff --git a/bootloaders/diskloader/src/DiskLoader.cpp b/bootloaders/diskloader/src/DiskLoader.cpp new file mode 100644 index 0000000..9ecbc4b --- /dev/null +++ b/bootloaders/diskloader/src/DiskLoader.cpp @@ -0,0 +1,251 @@ + + +#include "Platform.h" + +//	This bootloader creates a composite Serial/Mass storage device +// +//	The serial interface supports a STK500v1 protocol that is very similar to optiboot +//	The mass storage interface emulates a disk image that is used to read or write firmware.bin files +//	Delete firmware.bin and copy a new binary (not .hex) file to flash the firmware i.e: +// +//	cp firmware.bin f:/. -f +// +//	Ejecting the disk will run the firmware +// +//	The bootloader will timeout and start the firmware after a few hundred milliseconds +//	if a usb connection is not detected. +//	 +//	The tweakier code is to keep the bootloader below 2k (no interrupt table, for example) + +extern "C" +void entrypoint(void) __attribute__ ((naked)) __attribute__ ((section (".vectors"))); +void entrypoint(void) +{ +	asm volatile ( +		"eor	r1,		r1\n"	// Zero register +		"out	0x3F,	r1\n"	// SREG +		"ldi	r28,	0xFF\n" +		"ldi	r29,	0x0A\n" +		"out	0x3E,	r29\n"	// SPH +		"out	0x3D,	r28\n"	// SPL +		"rjmp	main"			// Stack is all set up, start the main code +		::); +} + +u8 _flashbuf[128]; +u8 _inSync; +u8 _ok; +extern volatile u8 _ejected; +extern volatile u16 _timeout; + +void Program(u8 ep, u16 page, u8 count) +{ +	u8 write = page < 30*1024;		// Don't write over firmware please +	if (write) +		boot_page_erase(page); + +	Recv(ep,_flashbuf,count);		// Read while page is erasing + +	if (!write) +		return; + +	boot_spm_busy_wait();			// Wait until the memory is erased. + +	count >>= 1; +	u16* p = (u16*)page; +	u16* b = (u16*)_flashbuf; +	for (u8 i = 0; i < count; i++) +		boot_page_fill(p++, b[i]); + +    boot_page_write(page); +    boot_spm_busy_wait(); +    boot_rww_enable (); +} + + +int USBGetChar(); +#define getch USBGetChar + +#define HW_VER	 0x02 +#define SW_MAJOR 0x01 +#define SW_MINOR 0x10 + +#define STK_OK              0x10 +#define STK_INSYNC          0x14  // ' ' +#define CRC_EOP             0x20  // 'SPACE' +#define STK_GET_SYNC        0x30  // '0' + +#define STK_GET_PARAMETER   0x41  // 'A' +#define STK_SET_DEVICE      0x42  // 'B' +#define STK_SET_DEVICE_EXT  0x45  // 'E' +#define STK_LOAD_ADDRESS    0x55  // 'U' +#define STK_UNIVERSAL       0x56  // 'V' +#define STK_PROG_PAGE       0x64  // 'd' +#define STK_READ_PAGE       0x74  // 't' +#define STK_READ_SIGN       0x75  // 'u' + +extern const u8 _readSize[] PROGMEM; +const u8 _readSize[] =  +{ +	STK_GET_PARAMETER,	1, +	STK_SET_DEVICE,		20, +	STK_SET_DEVICE_EXT,	5, +	STK_UNIVERSAL,		4, +	STK_LOAD_ADDRESS,	2, +	STK_PROG_PAGE,		3, +	STK_READ_PAGE,		3, +	0,0 +}; + +extern const u8 _consts[] PROGMEM; +const u8 _consts[] =  +{ +	SIGNATURE_0, +	SIGNATURE_1, +	SIGNATURE_2, +	HW_VER,		// Hardware version +	SW_MAJOR,	// Software major version +	SW_MINOR,	// Software minor version +	0x03,		// Unknown but seems to be required by avr studio 3.56 +	0x00,		//  +}; + + +void USBInit(void); +extern u8 _sector3[512]; + +int main(void) __attribute__ ((naked)); + +//	STK500v1 main loop, very similar to optiboot in protocol and implementation +int main() +{ +	TXLED0; +	RXLED0; +	LED0; +	BOARD_INIT(); +	USBInit(); + +#ifdef WRITABLE_DIRECTORY +	_sector3[0] = 0; +#endif + +	_inSync = STK_INSYNC; +	_ok = STK_OK; +#ifndef MSC_ENABLED +	if (pgm_read_word(0) != -1) +		_ejected = 1; +#endif +	for(;;) +	{ +		u8* packet = _flashbuf; +		u16 address = 0; +		for (;;) +		{ +			u8 cmd = getch(); + +			//	Read packet contents +			u8 len; +			const u8* rs = _readSize; +			for(;;) +			{ +				u8 c = pgm_read_byte(rs++); +				len = pgm_read_byte(rs++); +				if (c == cmd || c == 0) +					break; +			} +#ifndef MSC_ENABLED +			_timeout = 0; +#endif +			//	Read params +			Recv(CDC_RX,packet,len); + +			//	Send a response +			u8 send = 0; +			const u8* pgm = _consts+7;			// 0 +			if (STK_GET_PARAMETER == cmd) +			{ +				u8 i = packet[0] - 0x80; +				if (i > 2) +					i = (i == 0x18) ? 3 : 4;	// 0x80:HW_VER,0x81:SW_MAJOR,0x82:SW_MINOR,0x18:3 or 0 +				pgm = _consts + i + 3; +				send = 1; +			} + +			else if (STK_UNIVERSAL == cmd) +			{ +				if (packet[0] == 0x30) +					pgm = _consts + packet[2];	// read signature +				send = 1; +			} +			 +			//	Read signature bytes +			else if (STK_READ_SIGN == cmd) +			{ +				pgm = _consts; +				send = 3; +			} + +			else if (STK_LOAD_ADDRESS == cmd) +			{ +				address = *((u16*)packet);		// word addresses +				address += address; +			} + +			else if (STK_PROG_PAGE == cmd) +			{ +				Program(CDC_RX,address,packet[1]); +			} + +			else if (STK_READ_PAGE == cmd) +			{ +				send = packet[1]; +				pgm = (const u8*)address; +				address += send; // not sure of this is required +			} + +			// Check sync +			if (getch() != ' ') +				break; +			Transfer(CDC_TX,&_inSync,1); + +			// Send result +			if (send) +				Transfer(CDC_TX|TRANSFER_PGM,pgm,send);	// All from pgm memory + +			//	Send ok +			Transfer(CDC_TX|TRANSFER_RELEASE,&_ok,1); + +			if (cmd == 'Q') +				break; +		} +		_timeout = 1;		// signal to exit the bootloader +		_ejected = 1; +	} +} + +//	Nice breathing LED indicates we are in the firmware +u16 _pulse; +void LEDPulse() +{ +	_pulse++; +	u8 p = _pulse >> 8; +	if (p > 127) +		p = 255-p; +	p += p; +	if (((u8)_pulse) > p) +		LED0; +	else +		LED1; +} + +void Reboot() +{ +	/* TODO - ZE - this should probably be a WDT reset instead, right? */ +	UDCON = 1;		// Detatch USB +	UDIEN = 0; +	asm volatile (	// Reset vector to run firmware +		"clr r30\n" +		"clr r31\n" +		"ijmp\n" +	::); +} diff --git a/bootloaders/diskloader/src/Platform.h b/bootloaders/diskloader/src/Platform.h new file mode 100644 index 0000000..004933e --- /dev/null +++ b/bootloaders/diskloader/src/Platform.h @@ -0,0 +1,56 @@ + +#include <inttypes.h> +#include <avr/io.h> +#include <avr/pgmspace.h> +#include <avr/boot.h> +#include <util/delay.h> +#include <avr/interrupt.h> + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +typedef unsigned char u8; +typedef unsigned short u16; +typedef unsigned long u32; + +//#define MICROTOUCH + +//#define WRITABLE_DIRECTORY	// undef saved 56 + 512 RAM + +#define FAT_DISK_LABEL	'b','o','o','t','l','o','a','d','e','r',' '		// 11 chars (undef saves 12) +#define FAT_FILE_NAME	'F','I','R','M','W','A','R','E','B','I','N'		// 11 chars +#define MSC_DISK_SERIAL	'0','0','0','0','0','0','0','0','1','7','0','1'	// 12 chars + + +#define CPU_PRESCALE(n)	(CLKPR = 0x80, CLKPR = (n)) +#define DISABLE_JTAG()  MCUCR = (1 << JTD) | (1 << IVCE) | (0 << PUD); MCUCR = (1 << JTD) | (0 << IVSEL) | (0 << IVCE) | (0 << PUD); + +#define USB_VID 0x2341	// arduino LLC vid +#define USB_PID 0x0020	// educated guess + +#define FAT_OEM_NAME	'l','e','o','n','a','r','d','o'					// 8 chars +#define BOARD_INIT()	DDRC |= (1<<7); DDRB |= (1<<0); DDRE |= (1<<6); CPU_PRESCALE(0); DISABLE_JTAG(); +#define LED0			PORTC &= ~(1<<7) +#define LED1			PORTC |= (1<<7) +#define TXLED0			PORTE |= (1<<6) +#define TXLED1			PORTE &= ~(1<<6) +#define RXLED0			PORTB |= (1<<0) +#define RXLED1			PORTB &= ~(1<<0) + +#define TRANSFER_PGM		0x80 +#define TRANSFER_RELEASE	0x40 +#define TRANSFER_ZERO		0x20 + +void Transfer(u8 ep, const u8* data, int len); +void Recv(u8 ep, u8* dst, u8 len); +void Program(u8 ep, u16 page, u8 count); + +#define CDC_ENABLED +//#define MSC_ENABLED +//#define MSC_NO_MEDIA	// MSC is enabled but not media inserted + +#include "USBCore.h" +#include "USBDesc.h" + + diff --git a/bootloaders/diskloader/src/SCSI.cpp b/bootloaders/diskloader/src/SCSI.cpp new file mode 100644 index 0000000..dbbf9cf --- /dev/null +++ b/bootloaders/diskloader/src/SCSI.cpp @@ -0,0 +1,213 @@ + + +/* Copyright (c) 2010, Peter Barrett   +**   +** Permission to use, copy, modify, and/or distribute this software for   +** any purpose with or without fee is hereby granted, provided that the   +** above copyright notice and this permission notice appear in all copies.   +**  +** THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL   +** WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED   +** WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR   +** BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES   +** OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,   +** WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,   +** ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS   +** SOFTWARE.   +*/ + +#include "Platform.h" + +#ifdef MSC_ENABLED + +//	SCSI Commands +#define TEST_UNIT_READY            0x00 +#define REQUEST_SENSE              0x03 +#define FORMAT_UNIT                0x04 +#define INQUIRY                    0x12 +#define MODE_SENSE				   0x1A +#define START_STOP_UNIT            0x1B +#define MEDIA_REMOVAL              0x1E +#define READ_FORMAT_CAPACITIES     0x23 +#define READ_CAPACITY              0x25 +#define READ10                     0x28 +#define WRITE10                    0x2A +#define VERIFY10                   0x2F + + +//	In FakeFAT +void SendSector(u8 b); +void RecvSector(u8 b); + +//	In USBCore +u8 RecvPacket(u8 ep, u8* dst, u8 len); +u8 SendPacket(u8 ep, const u8* src, u8 len); +void Reboot(); + +#define MSC_TX MSC_ENDPOINT_IN +#define MSC_RX MSC_ENDPOINT_OUT + +extern const u8 P_Inquiry[36] PROGMEM; +extern const u8 P_RequestSense[18] PROGMEM; +extern const u8 P_ReadCapacity[8-3] PROGMEM;  +extern const u8 P_ModeSense[4] PROGMEM;  + +typedef struct +{ +	u8 signature[4]; +	u8 tag[4]; +	u32 transferLength; +	u8 flags; +	u8 lun; +	u8 cblength; +	u8 cb[16]; +} CBW; + +const u8 P_Inquiry[36] = +{ +	0x00,	// Direct Access Device +	0x80,	// Removable Medium +	0x00,	// Version +	0x01, + +	36-4,	// Additional Length +	0x80,	// Storage Controller Component +	0x00, +	0x00, + +	'r','o','s','s','u','m','u','r',	// Vendor + +	'm','i','c','r','o','t','o','u',	// Product +	'c','h',' ',' ',' ',' ',' ',' ', + +	'2','.','4',' '						// Revision +}; + +const u8 P_RequestSense[18] =  +{ +	0x70,		// Response Code  +	0x00, +//	0x02,		// Sense Key  +	0x00, +	0x00, + +	0x00, +	0x00, +	0x00, +	0x0A,       //	Additional Length + +	0x00, +	0x00, +	0x00, +	0x00, + +	0x30,        //	ASC +	0x01,        //	ASCQ + +	0x00, +	0x00, +	0x00, +	0x00 +}; + +//	Save 3 bytes if merging +const u8 P_ReadCapacity[8-3] =  +{ +//	0x00, +//	0x00, +//	0x00,	// Steal 3 zero bytes from end of P_RequestSense +	64+4,	// block count 64 + 4 +	0x00, +	0x00, +	0x02,	// block size = 512 +	0x00 +}; + +const u8 P_ModeSense[4] =  +{ +	0x03, +	0x00, +	0x00, +	0x00 +}; + +extern volatile u8 _ejected; + +void SCSI(CBW& cbw) +{ +	const u8* pgm = 0; +	u8 status = 0; +	u8 blen = 0; + +	u8 cmd = cbw.cb[0]; +	if (START_STOP_UNIT == cmd) +	{ +		_ejected = 1;		// Stopped; going away +	} +	else if (TEST_UNIT_READY == cmd || MEDIA_REMOVAL == cmd) +	{ +#ifdef MSC_NO_MEDIA +		status = 1; +#else +		status = _ejected; +#endif +	} +	else if (REQUEST_SENSE == cmd) +	{ +		pgm = P_RequestSense; +		blen = sizeof(P_RequestSense); +	} +	else if (INQUIRY == cmd) +	{ +		pgm = P_Inquiry; +		blen = sizeof(P_Inquiry); +	} +	else if (WRITE10 == cmd || READ10 == cmd) +	{ +		// Write 10 byte form +		// Read 10 byte form +		{ +			u8 lba = cbw.cb[5]; +			u8 count = cbw.cb[8]; +			u8 read = READ10 == cbw.cb[0]; +			while (count--) +			{ +				if (read) +					SendSector(lba); +				else +					RecvSector(lba); +				lba++; +				cbw.transferLength -= 512;	// 24 bytes +			} +		} +	} +	else if (MODE_SENSE == cmd) +	{ +		pgm = P_ModeSense; +		blen = 4; +	} +	else +	{ +		pgm = P_ReadCapacity-3; +		blen = 8; +		if (READ_CAPACITY != cmd) +			status = 1;	// Faking a data phase for unknown commands +	} + +	if (pgm) +		Transfer(MSC_TX | TRANSFER_PGM | TRANSFER_RELEASE,pgm,blen); + +	cbw.flags = status; +	cbw.transferLength -= blen; +	cbw.signature[3] = 'S'; +	Transfer(MSC_TX | TRANSFER_RELEASE,(u8*)&cbw,13); +} + +CBW cbw;	// 38 byte savings by making this global +void SCSITask() +{ +	Recv(MSC_RX,(u8*)&cbw,sizeof(CBW)); +	SCSI(cbw); +} + +#endif
\ No newline at end of file diff --git a/bootloaders/diskloader/src/ThinFAT.cpp b/bootloaders/diskloader/src/ThinFAT.cpp new file mode 100644 index 0000000..c238741 --- /dev/null +++ b/bootloaders/diskloader/src/ThinFAT.cpp @@ -0,0 +1,130 @@ + + +/* Copyright (c) 2011, Peter Barrett   +**   +** Permission to use, copy, modify, and/or distribute this software for   +** any purpose with or without fee is hereby granted, provided that the   +** above copyright notice and this permission notice appear in all copies.   +**  +** THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL   +** WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED   +** WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR   +** BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES   +** OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,   +** WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,   +** ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS   +** SOFTWARE.   +*/ + +#include "Platform.h" + +#ifdef MSC_ENABLED + +extern const u8 Sector_0[] PROGMEM; +extern const u8 Sector_1[] PROGMEM; +extern const u8 Sector_2[] PROGMEM; +extern const u8 Sector_3[] PROGMEM; +extern const u8 Sector_Size[] PROGMEM; + + +const u8 Sector_0[27] = +{ +	0xEB,0x3C,0x90, +	FAT_OEM_NAME, +	0x00,0x02,	// Bytes per sector 512 +	0x40,		// Sectors per Cluster 32k per cluster +	0x01,0x00,	// Reserved sectors +	0x02,		// FATSs +	0x10,0x00,	// root entries +	4+64,0x00,	// number of sectors: 30k file +	0xF8,		// Media type fixed +	0x01,0x00,	// Sectors per FAT +	0x01,0x00,	// Sectors per head +	0x01,//0x00,	// Heads per cylinder +	//0x00,0x00,0x00,0x00,	// Hidden +	//0x00,0x00,0x00,0x00,	// long number of sectors +	//0x00,0x00,				// Ext flags +	// 38 bytes +}; + +//	FAT12 clusters for 32k file +const u8 Sector_1[] = +{ +#if 0 +	0xF8,0xFF,0xFF, +	0xFF,0x0F//,0x00 +#endif +}; + +const u8 Sector_2[] = +{ +}; + +const u8 Sector_3[] = +{ +#if 0 +	FAT_FILE_NAME, 0x00,0x00,0x00,0x00,0x00,	// Uppercase name please +	0x43,0x3E,0x43,0x3E,0x00,0x00,0xAB,0x8C,0x40,0x3E,0x02,0x00,0x00,0x80,//0x00,0x00 +	// 0x00,0x00, +#endif + +#ifdef FAT_DISK_LABEL +	 FAT_DISK_LABEL,0x28 +#endif +}; + +#if 0 +const u8 Sector_Size[] = +{ +	0, +	sizeof(Sector_0), +	sizeof(Sector_0) + sizeof(Sector_1), +	sizeof(Sector_0) + sizeof(Sector_1) + sizeof(Sector_2), +	sizeof(Sector_0) + sizeof(Sector_1) + sizeof(Sector_2) + sizeof(Sector_3) +}; +#endif + +const u8* LBAToROM(u8 b) +{ +	b -= 4; +	return (const u8*)(b << 9); +} + +void SendSector(u8 b) +{ +	if (b == 0) +	{ +		u8 n = sizeof(Sector_0); +		Transfer(MSC_TX | TRANSFER_PGM,Sector_0,n); +		Transfer(MSC_TX | TRANSFER_ZERO,0,512-n); +		return; +	} +	 +	if ((b < 4) || (b >= 64)) +	{ +		Transfer(MSC_TX | TRANSFER_ZERO,0,512); +		return; +	} +	 +		//	Copy body of file from ROM +	Transfer(MSC_TX | TRANSFER_PGM,LBAToROM(b),512); +} + +#define GENERATE_INVALID_ROM_ADDRESS 60 + +void RecvSector(u8 lba) +{ +	if ((lba < 4) || (lba >= 64)) +		lba = GENERATE_INVALID_ROM_ADDRESS;	// Only care about the first 30k + +	//	Write sectors to flash +	u16 addr = (u16)LBAToROM(lba);		// may generate out of range addresses, Program will fix +	u8 i = 4;							// 4x128 is a sector +	while (i--) +	{ +		Program(MSC_RX,addr,128); +		addr += 128; +	} +} + +#endif
\ No newline at end of file diff --git a/bootloaders/diskloader/src/USBCore.cpp b/bootloaders/diskloader/src/USBCore.cpp new file mode 100644 index 0000000..8b071a1 --- /dev/null +++ b/bootloaders/diskloader/src/USBCore.cpp @@ -0,0 +1,526 @@ + + +/* Copyright (c) 2010, Peter Barrett   +**   +** Permission to use, copy, modify, and/or distribute this software for   +** any purpose with or without fee is hereby granted, provided that the   +** above copyright notice and this permission notice appear in all copies.   +**  +** THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL   +** WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED   +** WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR   +** BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES   +** OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,   +** WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,   +** ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS   +** SOFTWARE.   +*/ + +#include "Platform.h" + +#define MSC_TX MSC_ENDPOINT_IN +#define MSC_RX MSC_ENDPOINT_OUT +#define CDC_TX CDC_ENDPOINT_IN +#define CDC_RX CDC_ENDPOINT_OUT + +#define EP_TYPE_CONTROL				0x00 +#define EP_TYPE_BULK_IN				0x81 +#define EP_TYPE_BULK_OUT			0x80 +#define EP_TYPE_INTERRUPT_IN		0xC1 +#define EP_TYPE_INTERRUPT_OUT		0xC0 +#define EP_TYPE_ISOCHRONOUS_IN		0x41 +#define EP_TYPE_ISOCHRONOUS_OUT		0x40 + +void Reboot(); + +//================================================================== +//================================================================== + +typedef struct +{ +	u32	dwDTERate; +	u8	bCharFormat; +	u8 	bParityType; +	u8 	bDataBits; +	u8	lineState; +} LineInfo; + +static volatile LineInfo _usbLineInfo = { 57600, 0x00, 0x00, 0x00, 0x00 }; + +//================================================================== +//================================================================== + +//	4 bytes of RAM +volatile u8 _usbConfiguration; +volatile u8 _ejected; +volatile u16 _timeout; + +static inline void WaitIN(void) +{ +	while (!(UEINTX & (1<<TXINI))); +} + +static inline void ClearIN(void) +{ +	UEINTX = ~(1<<TXINI); +} + +static inline void WaitOUT(void) +{ +	while (!(UEINTX & (1<<RXOUTI))) +		; +} + +static inline u8 WaitForINOrOUT() +{ +	while (!(UEINTX & ((1<<TXINI)|(1<<RXOUTI)))) +		; +	return (UEINTX & (1<<RXOUTI)) == 0; +} + +static inline void ClearOUT(void) +{ +	UEINTX = ~(1<<RXOUTI); +} + +static +void Send(volatile const u8* data, u8 count) +{ +	TXLED1;					// light the TX LED +	while (count--) +		UEDATX = *data++; +} + +void Recv(volatile u8* data, u8 count) +{ +	RXLED1;					// light the RX LED +	while (count--) +		*data++ = UEDATX; +} + +static inline u8 Recv8() +{ +	RXLED1;					// light the RX LED +	return UEDATX; +} + +static inline void Send8(u8 d) +{ +	TXLED1;					// light the TX LED +	UEDATX = d; +} + +static inline void SetEP(u8 ep) +{ +	UENUM = ep; +} + +static inline u8 FifoByteCount() +{ +	return UEBCLX; +} + +static inline u8 ReceivedSetupInt() +{ +	return UEINTX & (1<<RXSTPI); +} + +static inline void ClearSetupInt() +{ +	UEINTX = ~((1<<RXSTPI) | (1<<RXOUTI) | (1<<TXINI)); +} + +static inline void Stall() +{ +	UECONX = (1<<STALLRQ) | (1<<EPEN); +} + +static inline u8 ReadWriteAllowed() +{ +	return UEINTX & (1<<RWAL); +} + +static inline u8 Stalled() +{ +	return UEINTX & (1<<STALLEDI); +} + +static inline u8 FifoFree() +{ +	return UEINTX & (1<<FIFOCON); +} + +static inline void ReleaseRX() +{ +	UEINTX = 0x6B;	// FIFOCON=0 NAKINI=1 RWAL=1 NAKOUTI=0 RXSTPI=1 RXOUTI=0 STALLEDI=1 TXINI=1 +} + +static inline void ReleaseTX() +{ +	UEINTX = 0x3A;	// FIFOCON=0 NAKINI=0 RWAL=1 NAKOUTI=1 RXSTPI=1 RXOUTI=0 STALLEDI=1 TXINI=0 +} + +static inline u8 FrameNumber() +{ +	return UDFNUML; +} + +//================================================================== +//================================================================== + +#define EP_SINGLE_64 0x32	// EP0 +#define EP_DOUBLE_64 0x36	// Other endpoints + +static void InitEP(u8 index, u8 type, u8 size) +{ +	UENUM = index; +	UECONX = 1; +	UECFG0X = type; +	UECFG1X = size; +} + +//	API +void USBInit(void) +{ +	_timeout = 0; +	_usbConfiguration = 0; +	_ejected = 0; +	 +	UHWCON = 0x01;						// power internal reg (don't need this?) +	USBCON = (1<<USBE)|(1<<FRZCLK);		// clock frozen, usb enabled +	PLLCSR = 0x12;						// Need 16 MHz xtal +	while (!(PLLCSR & (1<<PLOCK)))		// wait for lock pll +		; +	USBCON = ((1<<USBE)|(1<<OTGPADE));	// start USB clock +	UDCON = 0;							// enable attach resistor +} + +u8 USBGetConfiguration(void) +{ +	return _usbConfiguration; +} + +u8 HasData(u8 ep) +{ +	SetEP(ep); +	return ReadWriteAllowed();	// count in fifo +} + +int USBGetChar(); +void Recv(u8 ep, u8* dst, u8 len) +{ +	SetEP(ep); +	while (len--) +	{ +		while (!ReadWriteAllowed()) +			; +		*dst++ = Recv8(); +		if (!ReadWriteAllowed())	// release empty buffer +			ReleaseRX(); +	} +} + +//	Transmit a packet to endpoint +void Transfer(u8 ep, const u8* data, int len) +{ +	u8 zero = ep & TRANSFER_ZERO; +	SetEP(ep & 7); +	while (len--) +	{ +		while (!ReadWriteAllowed()) +			;	// TODO Check for STALL etc + +		u8 d = (ep & TRANSFER_PGM) ? pgm_read_byte(data) : data[0]; +		data++; +		if (zero) +			d = 0; +		Send8(d); + +		if (!ReadWriteAllowed()) +			ReleaseTX(); +	} +	if (ep & TRANSFER_RELEASE) +		ReleaseTX(); +} + +extern const u8 _initEndpoints[] PROGMEM; +const u8 _initEndpoints[] =  +{ +	0, + +#ifdef CDC_ENABLED +	EP_TYPE_INTERRUPT_IN,		// CDC_ENDPOINT_ACM +	EP_TYPE_BULK_OUT,			// CDC_ENDPOINT_OUT +	EP_TYPE_BULK_IN,			// CDC_ENDPOINT_IN +#endif + +	EP_TYPE_INTERRUPT_IN,		// HID_ENDPOINT_INT + +#ifdef MSC_ENABLED +	EP_TYPE_BULK_OUT,			// MSC_ENDPOINT_OUT +	EP_TYPE_BULK_IN				// MSC_ENDPOINT_IN +#endif +}; + +static void InitEndpoints() +{ +	for (u8 i = 1; i < sizeof(_initEndpoints); i++) +	{ +		UENUM = i; +		UECONX = 1; +		UECFG0X = pgm_read_byte(_initEndpoints+i); +		UECFG1X = EP_DOUBLE_64; +	} +	UERST = 0x7E;	// And reset them +	UERST = 0; +} + +typedef struct +{ +	u8 bmRequestType; +	u8 bRequest; +	u8 wValueL; +	u8 wValueH; +	u16 wIndex; +	u16 wLength; +} Setup; +Setup _setup; + +//bool USBHook(Setup& setup) +bool USBHook() +{ +	Setup& setup = _setup; +	u8 r = setup.bRequest; + +	//	CDC Requests +	if (CDC_GET_LINE_CODING == r) +	{ +		Send((const volatile u8*)&_usbLineInfo,7); +	} + +	else if (CDC_SET_LINE_CODING ==  r) +	{ +		WaitOUT(); +		Recv((volatile u8*)&_usbLineInfo,7); +		ClearOUT(); +	} + +	else if (CDC_SET_CONTROL_LINE_STATE == r) +	{ +		_usbLineInfo.lineState = setup.wValueL; +	} + +#ifdef MSC_ENABLED +	//	MSC Requests +	else if (MSC_GET_MAX_LUN == r) +	{ +		Send8(0); +	} +	else if (MSC_RESET == r) +	{ +		// MSC_Reset(); +	} else +		return false;	// unhandled +#endif + +	return true; +} + +extern const u8 _rawHID[] PROGMEM; +#define LSB(_x) ((_x) & 0xFF) +#define MSB(_x) ((_x) >> 8) + +#define RAWHID_USAGE_PAGE	0xFFC0 +#define RAWHID_USAGE		0x0C00 +#define RAWHID_TX_SIZE 64 +#define RAWHID_RX_SIZE 64 + +const u8 _rawHID[] = +{ +	//	RAW HID +	0x06, LSB(RAWHID_USAGE_PAGE), MSB(RAWHID_USAGE_PAGE),	// 30 +	0x0A, LSB(RAWHID_USAGE), MSB(RAWHID_USAGE), + +	0xA1, 0x01,				// Collection 0x01 +    0x85, 0x03,             // REPORT_ID (3) +	0x75, 0x08,				// report size = 8 bits +	0x15, 0x00,				// logical minimum = 0 +	0x26, 0xFF, 0x00,		// logical maximum = 255 + +	0x95, 64,				// report count TX +	0x09, 0x01,				// usage +	0x81, 0x02,				// Input (array) + +	0x95, 64,				// report count RX +	0x09, 0x02,				// usage +	0x91, 0x02,				// Output (array) +	0xC0					// end collection +}; + +u8 _cdcComposite = 0; + +bool SendDescriptor() +{ +	Setup& setup = _setup; +	u8 desc_length = 0; +	const u8* desc_addr = 0; + +	u8 t = setup.wValueH; +	if (0x22 == t) +	{ +		desc_addr = _rawHID; +		desc_length = sizeof(desc_length); +	} else if (USB_DEVICE_DESCRIPTOR_TYPE == t) +	{ +		if (setup.wLength == 8) +			_cdcComposite = 1; +		desc_addr = _cdcComposite ?  (const u8*)&USB_DeviceDescriptorA : (const u8*)&USB_DeviceDescriptor; +	} +	else if (USB_CONFIGURATION_DESCRIPTOR_TYPE == t) +	{ +		desc_addr = (const u8*)&USB_ConfigDescriptor; +		desc_length = sizeof(USB_ConfigDescriptor); +	} +	else if (USB_STRING_DESCRIPTOR_TYPE == t) +	{ +		if (setup.wValueL == 0) +			desc_addr = (const u8*)&STRING_LANGUAGE; +		else if (setup.wValueL == ISERIAL) +			desc_addr = (const u8*)&STRING_SERIAL; +		else +			return false; +	} else  +		return false; + +	if (desc_length == 0) +		desc_length = pgm_read_byte(desc_addr); +	if ((u8)setup.wLength < desc_length)		// bit of a cheat limiting to 255 bytes TODO (saved 8 bytes) +		desc_length = (u8)setup.wLength; + +	//	Send descriptor +	//	EP0 is 64 bytes long +	//	RWAL and FIFOCON don't work on EP0 +	u8 n = 0; +	do +	{ +		if (!WaitForINOrOUT()) +			return false; +		Send8(pgm_read_byte(&desc_addr[n++])); +		u8 clr = n & 0x3F; +		if (!clr) +			ClearIN();	// Fifo is full, release this packet +	} while (n < desc_length); +	return true; +} + +void USBSetupInterrupt() +{ +    SetEP(0); +	if (!ReceivedSetupInt()) +		return; + +	Setup& setup = _setup;	// global saves ~30 bytes +	Recv((u8*)&setup,8); +	ClearSetupInt(); + +	if (setup.bmRequestType & DEVICETOHOST) +		WaitIN(); +	else +		ClearIN(); + +    bool ok = true; +	u8 r = setup.bRequest; +	if (SET_ADDRESS == r) +	{ +		WaitIN(); +		UDADDR = setup.wValueL | (1<<ADDEN); +	} +	else if (SET_CONFIGURATION == r) +	{ +		_usbConfiguration = setup.wValueL; +		InitEndpoints(); +	} +	else if (GET_CONFIGURATION == r) +	{ +		Send8(_usbConfiguration); +	} +	else if (GET_STATUS == r) +	{ +		Send8(0);		// All good as far as I know +	} +	else if (GET_DESCRIPTOR == r) +	{ +		ok = SendDescriptor(); +	} +	else +	{ +		ok = USBHook(); +	} + +	if (ok) +		ClearIN(); +	else +		Stall(); +} + +void USBGeneralInterrupt() +{ +	u8 udint = UDINT; +	UDINT = 0; + +	//	End of Reset +	if (udint & (1<<EORSTI)) +	{ +		InitEP(0,EP_TYPE_CONTROL,EP_SINGLE_64);	// init ep0 +		_usbConfiguration = 0;			// not configured yet +		//UEIENX = 1 << RXSTPE; +	} + +	//	Start of Frame +	if (udint & (1<<SOFI)) +	{ +		if (!_ejected) +			_timeout = 0; +	} +} + +void SCSITask(); + +void LEDPulse(); +int USBGetChar() +{ +	for(;;) +	{ +		USBSetupInterrupt(); +		USBGeneralInterrupt(); + +#ifdef MSC_ENABLED +		//	Service disk +		if (HasData(MSC_RX)) +			SCSITask(); +#endif +		//	Read a char +		if (HasData(CDC_RX)) +		{ +			u8 c = Recv8(); +			if (!ReadWriteAllowed()) +				ReleaseRX(); +			return c; +		} else { +			u8 temp = 0; +			for (temp=100; temp; temp--)  +				asm volatile("nop\n\t" "nop\n\t" "nop\n\t" "nop\n\t"::);			 +			RXLED0;		// we turn the RX and TX LEDs on in the relevant Send or Recv instruction +			TXLED0;		// we turn them off here after some time has passed to ensure a minimum on time.   +		} + +		if (!--_timeout) { +			TXLED0;		// switch off the RX and TX LEDs before starting the user sketch +			RXLED0; +			Reboot();	// USB not connected, run firmware +		} + +		LEDPulse(); +	} +	return -1; +} diff --git a/bootloaders/diskloader/src/USBCore.h b/bootloaders/diskloader/src/USBCore.h new file mode 100644 index 0000000..a37b9f2 --- /dev/null +++ b/bootloaders/diskloader/src/USBCore.h @@ -0,0 +1,261 @@ + +// Copyright (c) 2010, Peter Barrett  +/* +** Permission to use, copy, modify, and/or distribute this software for   +** any purpose with or without fee is hereby granted, provided that the   +** above copyright notice and this permission notice appear in all copies.   +**   +** THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL   +** WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED   +** WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR   +** BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES   +** OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,   +** WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,   +** ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS   +** SOFTWARE.   +*/ + +#ifndef __USBCORE_H__ +#define __USBCORE_H__ + +#define GET_STATUS			0 +#define CLEAR_FEATURE		1 +#define SET_FEATURE			3 +#define SET_ADDRESS			5 +#define GET_DESCRIPTOR		6 +#define GET_CONFIGURATION	8 +#define SET_CONFIGURATION	9 +#define GET_INTERFACE		10 +#define SET_INTERFACE		11 + +// bmRequestType +#define HOSTTODEVICE	0x00 +#define DEVICETOHOST	0x80 +#define STANDARD		0x00 +#define CLASS			0x20 +#define VENDOR			0x40 +#define DEVICE			0x00 +#define INTERFACE		0x01 +#define ENDPOINT		0x02 +#define OTHER			0x03 + +#define CDC_SET_LINE_CODING		0x20 +#define CDC_GET_LINE_CODING		0x21 +#define CDC_SET_CONTROL_LINE_STATE	0x22 + +#define MSC_RESET		0xFF +#define MSC_GET_MAX_LUN	0xFE + + +//	Descriptors + +#define USB_DEVICE_DESC_SIZE 18 +#define USB_CONFIGUARTION_DESC_SIZE 9 +#define USB_INTERFACE_DESC_SIZE 9 +#define USB_ENDPOINT_DESC_SIZE 7 + +#define USB_DEVICE_DESCRIPTOR_TYPE             1 +#define USB_CONFIGURATION_DESCRIPTOR_TYPE      2 +#define USB_STRING_DESCRIPTOR_TYPE             3 +#define USB_INTERFACE_DESCRIPTOR_TYPE          4 +#define USB_ENDPOINT_DESCRIPTOR_TYPE           5 + +#define USB_DEVICE_CLASS_COMMUNICATIONS        0x02 +#define USB_DEVICE_CLASS_HUMAN_INTERFACE       0x03 +#define USB_DEVICE_CLASS_STORAGE               0x08 +#define USB_DEVICE_CLASS_VENDOR_SPECIFIC       0xFF + +#define USB_CONFIG_POWERED_MASK                0x40 +#define USB_CONFIG_BUS_POWERED                 0x80 +#define USB_CONFIG_SELF_POWERED                0xC0 +#define USB_CONFIG_REMOTE_WAKEUP               0x20 + +// bMaxPower in Configuration Descriptor +#define USB_CONFIG_POWER_MA(mA)                ((mA)/2) + +// bEndpointAddress in Endpoint Descriptor +#define USB_ENDPOINT_DIRECTION_MASK            0x80 +#define USB_ENDPOINT_OUT(addr)                 ((addr) | 0x00) +#define USB_ENDPOINT_IN(addr)                  ((addr) | 0x80) + +#define USB_ENDPOINT_TYPE_MASK                 0x03 +#define USB_ENDPOINT_TYPE_CONTROL              0x00 +#define USB_ENDPOINT_TYPE_ISOCHRONOUS          0x01 +#define USB_ENDPOINT_TYPE_BULK                 0x02 +#define USB_ENDPOINT_TYPE_INTERRUPT            0x03 + +#define TOBYTES(x) ((x) & 0xFF),(((x) >> 8) & 0xFF) + +#define CDC_V1_10                               0x0110 +#define CDC_COMMUNICATION_INTERFACE_CLASS       0x02 + +#define CDC_CALL_MANAGEMENT                     0x01 +#define CDC_ABSTRACT_CONTROL_MODEL              0x02 +#define CDC_HEADER                              0x00 +#define CDC_ABSTRACT_CONTROL_MANAGEMENT         0x02 +#define CDC_UNION                               0x06 +#define CDC_CS_INTERFACE                        0x24 +#define CDC_CS_ENDPOINT                         0x25 +#define CDC_DATA_INTERFACE_CLASS                0x0A + +#define MSC_SUBCLASS_SCSI						0x06  +#define MSC_PROTOCOL_BULK_ONLY					0x50  + + + +//	Device +typedef struct { +	u8 len;				// 18 +	u8 dtype;			// 1 USB_DEVICE_DESCRIPTOR_TYPE +	u16 usbVersion;		// 0x200 +	u8	deviceClass; +	u8	deviceSubClass; +	u8	deviceProtocol; +	u8	packetSize0;	// Packet 0 +	u16	idVendor; +	u16	idProduct; +	u16	deviceVersion;	// 0x100 +	u8	iManufacturer; +	u8	iProduct; +	u8	iSerialNumber; +	u8	bNumConfigurations; +} DeviceDescriptor; + +//	Config +typedef struct { +	u8	len;			// 9 +	u8	dtype;			// 2 +	u16 clen;			// total length +	u8	numInterfaces; +	u8	config; +	u8	iconfig; +	u8	attributes; +	u8	maxPower; +} ConfigDescriptor; + +//	String + +//	Interface +typedef struct +{ +	u8 len;		// 9 +	u8 dtype;	// 4 +	u8 number; +	u8 alternate; +	u8 numEndpoints; +	u8 interfaceClass; +	u8 interfaceSubClass; +	u8 protocol; +	u8 iInterface; +} InterfaceDescriptor; + +//	Endpoint +typedef struct +{ +	u8 len;		// 7 +	u8 dtype;	// 5 +	u8 addr; +	u8 attr; +	u16 packetSize; +	u8 interval; +} EndpointDescriptor; + +// Interface Association Descriptor +// Used to bind 2 interfaces together in CDC compostite device +typedef struct +{ +	u8 len;				// 8 +	u8 dtype;			// 11 +	u8 firstInterface; +	u8 interfaceCount; +	u8 functionClass; +	u8 funtionSubClass; +	u8 functionProtocol; +	u8 iInterface; +} IADDescriptor; + +//	CDC CS interface descriptor +typedef struct +{ +	u8 len;		// 5 +	u8 dtype;	// 0x24 +	u8 subtype; +	u8 d0; +	u8 d1; +} CDCCSInterfaceDescriptor; + +typedef struct +{ +	u8 len;		// 4 +	u8 dtype;	// 0x24 +	u8 subtype; +	u8 d0; +} CDCCSInterfaceDescriptor4; + +typedef struct  +{ +	IADDescriptor				iad;	// Only needed on compound device + +	//	Control +	InterfaceDescriptor			cif;	//  +	CDCCSInterfaceDescriptor	header; +	CDCCSInterfaceDescriptor	callManagement; +	CDCCSInterfaceDescriptor4	controlManagement; +	CDCCSInterfaceDescriptor	functionalDescriptor; +	EndpointDescriptor			cifin; + +	//	Data +	InterfaceDescriptor			dif; +	EndpointDescriptor			in; +	EndpointDescriptor			out; +} CDCDescriptor; + +typedef struct  +{ +	InterfaceDescriptor			msc; +	EndpointDescriptor			in; +	EndpointDescriptor			out; +} MSCDescriptor; + +typedef struct +{ +	u8 len;			// 9 +	u8 dtype;		// 0x21 +	u8 addr; +	u8	versionL;	// 0x101 +	u8	versionH;	// 0x101 +	u8	country; +	u8	desctype;	// 0x22 report +	u8	descLenL; +	u8	descLenH; +} HIDDescDescriptor; + +typedef struct  +{ +	InterfaceDescriptor			hid; +	HIDDescDescriptor			desc; +	EndpointDescriptor			in; +} HIDDescriptor; + +#define D_DEVICE(_class,_subClass,_proto,_packetSize0,_vid,_pid,_version,_im,_ip,_is,_configs) \ +	{ 18, 1, 0x200, _class,_subClass,_proto,_packetSize0,_vid,_pid,_version,_im,_ip,_is,_configs } + +#define D_CONFIG(_totalLength,_interfaces) \ +	{ 9, 2, _totalLength,_interfaces, 1, 0, USB_CONFIG_BUS_POWERED, USB_CONFIG_POWER_MA(100) } + +#define D_INTERFACE(_n,_numEndpoints,_class,_subClass,_protocol) \ +	{ 9, 4, _n, 0, _numEndpoints, _class,_subClass, _protocol, 0 } + +#define D_ENDPOINT(_addr,_attr,_packetSize, _interval) \ +	{ 7, 5, _addr,_attr,_packetSize, _interval } + +#define D_IAD(_firstInterface, _count, _class, _subClass, _protocol) \ +	{ 8, 11, _firstInterface, _count, _class, _subClass, _protocol, 0 } + +#define D_HIDREPORT(_descriptorLength) \ +	{ 9, 0x21, 0x1, 0x1, 0, 1, 0x22, _descriptorLength, 0 } + +#define D_CDCCS(_subtype,_d0,_d1)	{ 5, 0x24, _subtype, _d0, _d1 } +#define D_CDCCS4(_subtype,_d0)		{ 4, 0x24, _subtype, _d0 } + +#endif
\ No newline at end of file diff --git a/bootloaders/diskloader/src/USBDesc.cpp b/bootloaders/diskloader/src/USBDesc.cpp new file mode 100644 index 0000000..dfde01d --- /dev/null +++ b/bootloaders/diskloader/src/USBDesc.cpp @@ -0,0 +1,88 @@ + + +/* Copyright (c) 2011, Peter Barrett   +**   +** Permission to use, copy, modify, and/or distribute this software for   +** any purpose with or without fee is hereby granted, provided that the   +** above copyright notice and this permission notice appear in all copies.   +**  +** THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL   +** WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED   +** WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR   +** BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES   +** OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,   +** WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,   +** ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS   +** SOFTWARE.   +*/ + +#include "Platform.h" + +//==================================================================================================== +//==================================================================================================== +//	Actual device descriptors + +const u16 STRING_LANGUAGE[2] = { +	(3<<8) | (2+2), +	0x0409	// English +}; + +#if 0 +const u16 STRING_PRODUCT[] = { +	(3<<8) | (2+2*10), +	'M','i','c','r','o','t','o','u','c','h' +}; +#endif + +const u16 STRING_SERIAL[13] = { +	(3<<8) | (2+2*12), +	MSC_DISK_SERIAL +}; + + +//#ifdef CDC_ENABLED +DeviceDescriptor USB_DeviceDescriptorA = D_DEVICE(0X02,0X00,0X00,64,USB_VID,USB_PID,0x100,0,IPRODUCT,ISERIAL,1); +//#else +DeviceDescriptor USB_DeviceDescriptor = D_DEVICE(0x00,0x00,0x00,64,USB_VID,USB_PID,0x100,0,IPRODUCT,ISERIAL,1); +//#endif + +Config USB_ConfigDescriptor = +{ +	D_CONFIG(sizeof(Config),INTERFACE_COUNT), + +#ifdef CDC_ENABLED +	//	CDC +	{ +		D_IAD(0,2,CDC_COMMUNICATION_INTERFACE_CLASS,CDC_ABSTRACT_CONTROL_MODEL,1), + +		//	CDC communication interface +		D_INTERFACE(CDC_ACM_INTERFACE,1,CDC_COMMUNICATION_INTERFACE_CLASS,CDC_ABSTRACT_CONTROL_MODEL,0), +		D_CDCCS(CDC_HEADER,0x10,0x01),								// Header (1.10 bcd) +		D_CDCCS(CDC_CALL_MANAGEMENT,1,1),							// Device handles call management +		D_CDCCS4(CDC_ABSTRACT_CONTROL_MANAGEMENT,2),				// SET_LINE_CODING, GET_LINE_CODING, SET_CONTROL_LINE_STATE supported +		D_CDCCS(CDC_UNION,CDC_ACM_INTERFACE,CDC_DATA_INTERFACE),	// Communication interface is master, data interface is slave 0 +		D_ENDPOINT(USB_ENDPOINT_IN (CDC_ENDPOINT_ACM),USB_ENDPOINT_TYPE_INTERRUPT,0x10,0x40), + +		//	CDC data interface +		D_INTERFACE(CDC_DATA_INTERFACE,2,CDC_DATA_INTERFACE_CLASS,0,0), +		D_ENDPOINT(USB_ENDPOINT_OUT(CDC_ENDPOINT_OUT),USB_ENDPOINT_TYPE_BULK,0x40,0), +		D_ENDPOINT(USB_ENDPOINT_IN (CDC_ENDPOINT_IN ),USB_ENDPOINT_TYPE_BULK,0x40,0) +	}, +#endif +	// HID +	{ +		D_INTERFACE(HID_INTERFACE,1,3,0,0), +		D_HIDREPORT(30), +		D_ENDPOINT(USB_ENDPOINT_IN (HID_ENDPOINT_INT),USB_ENDPOINT_TYPE_INTERRUPT,0x40,0x40) +	}, + +#ifdef MSC_ENABLED +	// Mass Storage +	{ +		D_INTERFACE(MSC_INTERFACE,2,USB_DEVICE_CLASS_STORAGE,MSC_SUBCLASS_SCSI,MSC_PROTOCOL_BULK_ONLY), +		D_ENDPOINT(USB_ENDPOINT_OUT(MSC_ENDPOINT_OUT),USB_ENDPOINT_TYPE_BULK,0x40,0), +		D_ENDPOINT(USB_ENDPOINT_IN (MSC_ENDPOINT_IN ),USB_ENDPOINT_TYPE_BULK,0x40,0) +	} +#endif +}; + diff --git a/bootloaders/diskloader/src/USBDesc.h b/bootloaders/diskloader/src/USBDesc.h new file mode 100644 index 0000000..a551d69 --- /dev/null +++ b/bootloaders/diskloader/src/USBDesc.h @@ -0,0 +1,81 @@ + + +/* Copyright (c) 2011, Peter Barrett   +**   +** Permission to use, copy, modify, and/or distribute this software for   +** any purpose with or without fee is hereby granted, provided that the   +** above copyright notice and this permission notice appear in all copies.   +**  +** THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL   +** WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED   +** WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR   +** BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES   +** OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,   +** WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,   +** ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS   +** SOFTWARE.   +*/ + + +#ifndef CDC_ENABLED + +#define MSC_INTERFACE		0	// MSC Interface +#define MSC_ENDPOINT_OUT	1 +#define MSC_ENDPOINT_IN		2 + +#define INTERFACE_COUNT		1	// 1 for msc + +#else + +#define CDC_ACM_INTERFACE	0	// CDC ACM +#define CDC_DATA_INTERFACE	1	// CDC Data +#define CDC_ENDPOINT_ACM	1 +#define CDC_ENDPOINT_OUT	2 +#define CDC_ENDPOINT_IN		3 + +#define HID_INTERFACE		2	// HID Interface +#define HID_ENDPOINT_INT	4 + +#ifdef MSC_ENABLED + +#define MSC_INTERFACE		3	// MSC Interface +#define MSC_ENDPOINT_OUT	5 +#define MSC_ENDPOINT_IN		6 +#define INTERFACE_COUNT		4	// 2 for cdc + 1 for hid + 1 for msc + +#else + +#define INTERFACE_COUNT		3	// 2 for cdc + 1 for hid + +#endif + +#endif + +typedef struct +{ +	ConfigDescriptor	config; +#ifdef CDC_ENABLED +	CDCDescriptor		cdc; +#endif +	HIDDescriptor		hid; +#ifdef MSC_ENABLED +	MSCDescriptor		msc; +#endif +} Config; + +extern Config USB_ConfigDescriptor PROGMEM; +extern DeviceDescriptor USB_DeviceDescriptor PROGMEM; +extern DeviceDescriptor USB_DeviceDescriptorA PROGMEM; + +extern const u16 STRING_LANGUAGE[2] PROGMEM; +extern const u16 STRING_SERIAL[13] PROGMEM; + +#define IMANUFACTURER	0 +#define IPRODUCT		0 +#define ISERIAL			1	// Only need this for MSC + + +#define MSC_TX MSC_ENDPOINT_IN +#define MSC_RX MSC_ENDPOINT_OUT +#define CDC_TX CDC_ENDPOINT_IN +#define CDC_RX CDC_ENDPOINT_OUT
\ No newline at end of file | 
