X-Git-Url: http://git.linex4red.de/pub/USBasp.git/blobdiff_plain/782614dbb55addcae8c9d4d9e1ce3dec81287282..1ab76048c9c727a588a0cc0fab285222b5916df5:/Projects/Incomplete/StandaloneProgrammer/StandaloneProgrammer.c?ds=sidebyside

diff --git a/Projects/Incomplete/StandaloneProgrammer/StandaloneProgrammer.c b/Projects/Incomplete/StandaloneProgrammer/StandaloneProgrammer.c
index 3f5d48b00..86318e9cb 100644
--- a/Projects/Incomplete/StandaloneProgrammer/StandaloneProgrammer.c
+++ b/Projects/Incomplete/StandaloneProgrammer/StandaloneProgrammer.c
@@ -1,13 +1,13 @@
 /*
              LUFA Library
-     Copyright (C) Dean Camera, 2011.
+     Copyright (C) Dean Camera, 2012.
 
   dean [at] fourwalledcubicle [dot] com
            www.lufa-lib.org
 */
 
 /*
-  Copyright 2011  Dean Camera (dean [at] fourwalledcubicle [dot] com)
+  Copyright 2012  Dean Camera (dean [at] fourwalledcubicle [dot] com)
 
   Permission to use, copy, modify, distribute, and sell this
   software and its documentation for any purpose is hereby granted
@@ -43,7 +43,6 @@ FILE DiskStream = FDEV_SETUP_STREAM(NULL, Disk_getchar, _FDEV_SETUP_READ);
 /** Petite FAT Fs structure to hold the internal state of the FAT driver for the Dataflash contents. */
 FATFS DiskFATState;
 
-
 /** Stream character fetching routine for the FAT driver so that characters from the currently open file can be
  *  read in sequence when applied to a stdio stream.
  */
@@ -115,18 +114,15 @@ int main(void)
 	{
 		Programmer_Task();
 
+		#if defined(USB_CAN_BE_HOST)
 		if (USB_CurrentMode == USB_MODE_Host)
-		{
-			#if defined(USB_CAN_BE_HOST)
-			DiskHost_USBTask();
-			#endif
-		}
-		else
-		{
-			#if defined(USB_CAN_BE_DEVICE)
-			DiskDevice_USBTask();
-			#endif
-		}
+		  DiskHost_USBTask();
+		#endif
+		
+		#if defined(USB_CAN_BE_DEVICE)
+		if (USB_CurrentMode == USB_MODE_Device)
+		  DiskDevice_USBTask();
+		#endif
 
 		USB_USBTask();
 	}
@@ -159,6 +155,13 @@ void SetupHardware(void)
 	Serial_CreateStream(NULL);
 
 	#if defined(USB_CAN_BE_DEVICE)
+	/* Check if the Dataflash is working, abort if not */
+	if (!(DataflashManager_CheckDataflashOperation()))
+	{
+		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
+		for(;;);
+	}
+
 	/* Clear Dataflash sector protections, if enabled */
 	DataflashManager_ResetDataflashProtections();
 	#endif