Projects/AVRISP-MKII/AVRISP-MKII.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2017.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6            www.lufa-lib.org
   7 */
   8
   9 /*
  10   Copyright 2017  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, distribute, and sell this
  13   software and its documentation for any purpose is hereby granted
  14   without fee, provided that the above copyright notice appear in
  15   all copies and that both that the copyright notice and this
  16   permission notice and warranty disclaimer appear in supporting
  17   documentation, and that the name of the author not be used in
  18   advertising or publicity pertaining to distribution of the
  19   software without specific, written prior permission.
  20
  21   The author disclaims all warranties with regard to this
  22   software, including all implied warranties of merchantability
  23   and fitness.  In no event shall the author be liable for any
  24   special, indirect or consequential damages or any damages
  25   whatsoever resulting from loss of use, data or profits, whether
  26   in an action of contract, negligence or other tortious action,
  27   arising out of or in connection with the use or performance of
  28   this software.
  29 */
  30
  31 /** \file
  32  *
  33  *  Main source file for the AVRISP project. This file contains the main tasks of
  34  *  the project and is responsible for the initial application hardware configuration.
  35  */
  36
  37 #include "AVRISP-MKII.h"
  38
  39 #if (BOARD != BOARD_NONE)
  40   /* Some board hardware definitions (e.g. the Arduino Micro) have their LEDs defined on the same pins
  41      as the ISP, PDI or TPI interfaces (see the accompanying project documentation). If a board other
  42      than NONE is selected (to enable the LED driver with the programmer) you should double-check that
  43      no conflicts will occur. If there is a conflict, turn off the LEDs (set BOARD to NONE in the makefile)
  44      or define a custom board driver (see the LUFA manual) with alternative LED mappings.
  45   */
  46   #warning Board specific drivers have been selected; make sure the board LED driver does not conflict with the programmer ISP/PDI/TPI interfaces.
  47 #endif
  48
  49 /** Main program entry point. This routine contains the overall program flow, including initial
  50  *  setup of all components and the main program loop.
  51  */
  52 int main(void)
  53 {
  54         SetupHardware();
  55         V2Protocol_Init();
  56
  57         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  58         GlobalInterruptEnable();
  59
  60         for (;;)
  61         {
  62                 #if (BOARD == BOARD_USBTINYMKII)
  63                 /* On the USBTINY-MKII target, there is a secondary LED which indicates the current selected power
  64                    mode - either VBUS, or sourced from the VTARGET pin of the programming connectors */
  65                 LEDs_ChangeLEDs(LEDMASK_VBUSPOWER, (PIND & (1 << 0)) ? 0 : LEDMASK_VBUSPOWER);
  66                 #endif
  67
  68                 AVRISP_Task();
  69                 USB_USBTask();
  70         }
  71 }
  72
  73 /** Configures the board hardware and chip peripherals for the demo's functionality. */
  74 void SetupHardware(void)
  75 {
  76 #if (ARCH == ARCH_AVR8)
  77         /* Disable watchdog if enabled by bootloader/fuses */
  78         MCUSR &= ~(1 << WDRF);
  79         wdt_disable();
  80
  81         /* Disable clock division */
  82         clock_prescale_set(clock_div_1);
  83 #endif
  84
  85         /* Hardware Initialization */
  86         LEDs_Init();
  87         #if defined(RESET_TOGGLES_LIBUSB_COMPAT)
  88         UpdateCurrentCompatibilityMode();
  89         #endif
  90
  91         /* USB Stack Initialization */
  92         USB_Init();
  93 }
  94
  95 /** Event handler for the library USB Connection event. */
  96 void EVENT_USB_Device_Connect(void)
  97 {
  98         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
  99 }
 100
 101 /** Event handler for the library USB Disconnection event. */
 102 void EVENT_USB_Device_Disconnect(void)
 103 {
 104         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 105 }
 106
 107 /** Event handler for the library USB Configuration Changed event. */
 108 void EVENT_USB_Device_ConfigurationChanged(void)
 109 {
 110         bool ConfigSuccess = true;
 111
 112         /* Setup AVRISP Data OUT endpoint */
 113         ConfigSuccess &= Endpoint_ConfigureEndpoint(AVRISP_DATA_OUT_EPADDR, EP_TYPE_BULK, AVRISP_DATA_EPSIZE, 1);
 114
 115         /* Setup AVRISP Data IN endpoint if it is using a physically different endpoint */
 116         if ((AVRISP_DATA_IN_EPADDR & ENDPOINT_EPNUM_MASK) != (AVRISP_DATA_OUT_EPADDR & ENDPOINT_EPNUM_MASK))
 117           ConfigSuccess &= Endpoint_ConfigureEndpoint(AVRISP_DATA_IN_EPADDR, EP_TYPE_BULK, AVRISP_DATA_EPSIZE, 1);
 118
 119         /* Indicate endpoint configuration success or failure */
 120         LEDs_SetAllLEDs(ConfigSuccess ? LEDMASK_USB_READY : LEDMASK_USB_ERROR);
 121 }
 122
 123 /** Processes incoming V2 Protocol commands from the host, returning a response when required. */
 124 void AVRISP_Task(void)
 125 {
 126         /* Device must be connected and configured for the task to run */
 127         if (USB_DeviceState != DEVICE_STATE_Configured)
 128           return;
 129
 130         V2Params_UpdateParamValues();
 131
 132         Endpoint_SelectEndpoint(AVRISP_DATA_OUT_EPADDR);
 133
 134         /* Check to see if a V2 Protocol command has been received */
 135         if (Endpoint_IsOUTReceived())
 136         {
 137                 LEDs_SetAllLEDs(LEDMASK_BUSY);
 138
 139                 /* Pass off processing of the V2 Protocol command to the V2 Protocol handler */
 140                 V2Protocol_ProcessCommand();
 141
 142                 LEDs_SetAllLEDs(LEDMASK_USB_READY);
 143         }
 144 }
 145
 146 /** This function is called by the library when in device mode, and must be overridden (see library "USB Descriptors"
 147  *  documentation) by the application code so that the address and size of a requested descriptor can be given
 148  *  to the USB library. When the device receives a Get Descriptor request on the control endpoint, this function
 149  *  is called so that the descriptor details can be passed back and the appropriate descriptor sent back to the
 150  *  USB host.
 151  *
 152  *  \param[in]  wValue                 Descriptor type and index to retrieve
 153  *  \param[in]  wIndex                 Sub-index to retrieve (such as a localized string language)
 154  *  \param[out] DescriptorAddress      Address of the retrieved descriptor
 155  *
 156  *  \return Length of the retrieved descriptor in bytes, or NO_DESCRIPTOR if the descriptor was not found
 157  */
 158 uint16_t CALLBACK_USB_GetDescriptor(const uint16_t wValue,
 159                                     const uint16_t wIndex,
 160                                     const void** const DescriptorAddress)
 161 {
 162         return AVRISP_GetDescriptor(wValue, wIndex, DescriptorAddress);
 163 }
 164