Projects/Unfinished/AVRISP/Lib/V2Protocol.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2009.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6       www.fourwalledcubicle.com
   7 */
   8
   9 /*
  10   Copyright 2009  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, and distribute this software
  13   and its documentation for any purpose and without fee is hereby
  14   granted, provided that the above copyright notice appear in all
  15   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 disclaim 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  *  V2Protocol handler, to process V2 Protocol commands used in Atmel programmer devices.
  34  */
  35
  36 #define  INCLUDE_FROM_V2PROTOCOL_C
  37 #include "V2Protocol.h"
  38
  39 /* Table of masks for SPI_Init() from a given PARAM_SCK_DURATION value */
  40 static const uint8_t SPIMaskFromSCKDuration[MAX_SPI_SETTINGS] =
  41         {
  42                 #if (F_CPU == 8000000)
  43                 SPI_SPEED_FCPU_DIV_2,
  44                 #endif
  45                 SPI_SPEED_FCPU_DIV_2, SPI_SPEED_FCPU_DIV_4, SPI_SPEED_FCPU_DIV_8,
  46                 SPI_SPEED_FCPU_DIV_16, SPI_SPEED_FCPU_DIV_32, SPI_SPEED_FCPU_DIV_64
  47                 #if (F_CPU == 16000000)
  48                 , SPI_SPEED_FCPU_DIV_128
  49                 #endif
  50         };
  51
  52 static void V2Protocol_ReconfigureSPI(void)
  53 {
  54         uint8_t SCKDuration = V2Params_GetParameterValue(PARAM_SCK_DURATION);
  55
  56         if (SCKDuration >= MAX_SPI_SETTINGS)
  57           SCKDuration = (MAX_SPI_SETTINGS - 1);
  58
  59         SPI_Init(SPIMaskFromSCKDuration[SCKDuration], true);
  60 }
  61
  62 static void V2Protocol_ChangeTargetResetLine(bool ResetTarget)
  63 {
  64         if (ResetTarget)
  65         {
  66                 RESET_LINE_DDR  |= RESET_LINE_MASK;
  67
  68                 if (!(V2Params_GetParameterValue(PARAM_RESET_POLARITY)))
  69                   RESET_LINE_PORT |= RESET_LINE_MASK;
  70         }
  71         else
  72         {
  73                 RESET_LINE_PORT &= ~RESET_LINE_MASK;
  74                 RESET_LINE_DDR  &= ~RESET_LINE_MASK;
  75         }
  76 }
  77
  78 void V2Protocol_ProcessCommand(void)
  79 {
  80         uint8_t V2Command = Endpoint_Read_Byte();
  81
  82         switch (V2Command)
  83         {
  84                 case CMD_SIGN_ON:
  85                         V2Protocol_Command_SignOn();
  86                         break;
  87                 case CMD_SET_PARAMETER:
  88                 case CMD_GET_PARAMETER:
  89                         V2Protocol_Command_GetSetParam(V2Command);
  90                         break;
  91                 case CMD_SPI_MULTI:
  92                         V2Protocol_Command_SPIMulti();
  93                         break;
  94                 default:
  95                         V2Protocol_Command_Unknown(V2Command);
  96                         break;
  97         }
  98
  99         printf("COMMAND 0x%02x\r\n", V2Command);
 100
 101         Endpoint_WaitUntilReady();
 102         Endpoint_SetEndpointDirection(ENDPOINT_DIR_OUT);
 103 }
 104
 105 static void V2Protocol_Command_Unknown(uint8_t V2Command)
 106 {
 107         while (Endpoint_BytesInEndpoint() == AVRISP_DATA_EPSIZE)
 108         {
 109                 Endpoint_ClearOUT();
 110                 while (!(Endpoint_IsOUTReceived()));
 111         }
 112
 113         Endpoint_ClearOUT();
 114         Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);
 115
 116         Endpoint_Write_Byte(V2Command);
 117         Endpoint_Write_Byte(STATUS_CMD_UNKNOWN);
 118         Endpoint_ClearIN();
 119 }
 120
 121 static void V2Protocol_Command_SignOn(void)
 122 {
 123         Endpoint_ClearOUT();
 124         Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);
 125         Endpoint_WaitUntilReady();
 126
 127         V2Protocol_ReconfigureSPI();
 128
 129         Endpoint_Write_Byte(CMD_SIGN_ON);
 130         Endpoint_Write_Byte(STATUS_CMD_OK);
 131         Endpoint_Write_Byte(PROGRAMMER_ID_LEN);
 132         Endpoint_Write_Stream_LE(PROGRAMMER_ID, PROGRAMMER_ID_LEN);
 133         Endpoint_ClearIN();
 134 }
 135
 136 static void V2Protocol_Command_GetSetParam(uint8_t V2Command)
 137 {
 138         uint8_t ParamID    = Endpoint_Read_Byte();
 139         uint8_t ParamValue = Endpoint_Read_Byte();
 140
 141         Endpoint_ClearOUT();
 142         Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);
 143         Endpoint_WaitUntilReady();
 144
 145         uint8_t ParamPrivs = V2Params_GetParameterPrivellages(ParamID);
 146
 147         Endpoint_Write_Byte(V2Command);
 148
 149         if ((V2Command == CMD_SET_PARAMETER) && (ParamPrivs & PARAM_PRIV_WRITE))
 150         {
 151                 Endpoint_Write_Byte(STATUS_CMD_OK);
 152                 V2Params_SetParameterValue(ParamID, ParamValue);
 153         }
 154         else if ((V2Command == CMD_GET_PARAMETER) && (ParamPrivs & PARAM_PRIV_READ))
 155         {
 156                 Endpoint_Write_Byte(STATUS_CMD_OK);
 157                 Endpoint_Write_Byte(V2Params_GetParameterValue(ParamID));
 158         }
 159         else
 160         {
 161                 Endpoint_Write_Byte(STATUS_CMD_FAILED);
 162         }
 163
 164         Endpoint_ClearIN();
 165 }
 166
 167 static void V2Protocol_Command_SPIMulti(void)
 168 {
 169         uint8_t TxBytes = Endpoint_Read_Byte();
 170         uint8_t RxBytes = Endpoint_Read_Byte();
 171         uint8_t RxStartAddr = Endpoint_Read_Byte();
 172         uint8_t TxData[255];
 173
 174         Endpoint_Read_Stream_LE(TxData, TxBytes);
 175
 176         Endpoint_ClearOUT();
 177         Endpoint_SetEndpointDirection(ENDPOINT_DIR_IN);
 178         Endpoint_WaitUntilReady();
 179
 180         Endpoint_Write_Byte(CMD_SPI_MULTI);
 181         Endpoint_Write_Byte(STATUS_CMD_OK);
 182
 183         uint8_t CurrTxPos = 0;
 184         uint8_t CurrRxPos = 0;
 185
 186         while (CurrTxPos < RxStartAddr)
 187         {
 188                 if (CurrTxPos < TxBytes)
 189                   SPI_SendByte(TxData[CurrTxPos]);
 190                 else
 191                   SPI_SendByte(0);
 192
 193                 CurrTxPos++;
 194         }
 195
 196         while (CurrRxPos < RxBytes)
 197         {
 198                 if (CurrTxPos < TxBytes)
 199                   Endpoint_Write_Byte(SPI_TransferByte(TxData[CurrTxPos++]));
 200                 else
 201                   Endpoint_Write_Byte(SPI_ReceiveByte());
 202
 203                 CurrRxPos++;
 204         }
 205
 206         Endpoint_Write_Byte(STATUS_CMD_OK);
 207         Endpoint_ClearIN();
 208 }