X-Git-Url: http://git.linex4red.de/pub/USBaspLoader.git/blobdiff_plain/2bdd5099cb391ff24ba3efa351b42e3745a90dc3..dda3e71306415ec5fd744d0daa2fafeddec99961:/firmware/main.c?ds=inline

diff --git a/firmware/main.c b/firmware/main.c
index 98703f9..c609439 100644
--- a/firmware/main.c
+++ b/firmware/main.c
@@ -493,12 +493,7 @@ static uchar    replyBuffer[4];
     }else if(rq->bRequest == USBASP_FUNC_DISCONNECT){
 
 #if BOOTLOADER_CAN_EXIT
-#	if (BOOTLOADER_LOOPCYCLES_TIMEOUT)
-      timeout_remaining = BOOTLOADER_LOOPCYCLES_TIMEOUT;
-      stayinloader	    = (0xfe);
-#	else
       stayinloader	   &= (0xfe);
-#	endif
 #endif
     }else{
         /* ignore: others, but could be USBASP_FUNC_CONNECT */
@@ -766,10 +761,14 @@ int __attribute__((__noreturn__)) main(void)
         initForUsbConnectivity();
         do{
 #if ((BOOTLOADER_LOOPCYCLES_TIMEOUT) && (BOOTLOADER_CAN_EXIT))
-	__loopscycles++;
-	if (!(__loopscycles)) {
-	  if(timeout_remaining)	timeout_remaining--;
-	  else			stayinloader&=0xf1;
+	if (stayinloader & 0x1) {
+	  timeout_remaining = BOOTLOADER_LOOPCYCLES_TIMEOUT;
+	} else {
+	  __loopscycles++;
+	  if (!(__loopscycles)) {
+	    if(timeout_remaining)	timeout_remaining--;
+	    else			stayinloader&=0xf1;
+	  }
 	}
 #endif
             usbPoll();