Subversion Repositories gecko3

/* GECKO3COM

 *

 * Copyright (C) 2008 by

 *   ___    ____  _   _

 *  (  _`\ (  __)( ) ( )

 *  | (_) )| (_  | |_| |   Berne University of Applied Sciences

 *  |  _ <'|  _) |  _  |   School of Engineering and

 *  | (_) )| |   | | | |   Information Technology

 *  (____/'(_)   (_) (_)

 *

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

//#define USB_DFU_SUPPORT

#include <stdint.h>

#include "fx2regs.h"

#include "isr.h"

#include "usb_requests.h"

#include "usb_common.h"

#include "usb_dfu.h"

#include "debugprint.h"

volatile uint8_t  usb_dfu_status = DFU_STATUS_OK;

volatile enum dfu_state usb_dfu_state = DFU_STATE_appIDLE;

volatile uint8_t usb_dfu_timeout = DFU_TIMEOUT;

uint8_t usb_handle_dfu_packet (void)

{

  if ((bRequestType & bmRT_DIR_MASK) == bmRT_DIR_IN){

    /*********************************

     *    handle the DFU IN requests

     ********************************/

    switch (bRequest){

    case DFU_UPLOAD:

      usb_dfu_status == DFU_STATUS_errSTALLEDPKT;

      return 0;

      break;

    case DFU_GETSTATUS:

      EP0BUF[0] = usb_dfu_status;

      EP0BUF[1] = 0;

      EP0BUF[2] = 0;

      EP0BUF[3] = 0xff;

      EP0BUF[4] = usb_dfu_state;

      EP0BUF[5] = 0;

      EP0BCH = 0;

      EP0BCL = 6;

      break;

    case DFU_GETSTATE:

      EP0BUF[0] = usb_dfu_state;

      EP0BCH = 0;

      EP0BCL = 1;

      break;

    default:

      return 0;

    }

  }

  else if ((bRequestType & bmRT_DIR_MASK) == bmRT_DIR_OUT){

    /***********************************

     *    handle the DFU OUT requests

     **********************************/

    switch (bRequest){

    case DFU_DETACH:

      if(usb_dfu_state == DFU_STATE_appIDLE){

        usb_toggle_dfu_handlers();

        usb_dfu_state = DFU_STATE_appDETACH;

        usb_dfu_timeout = DFU_TIMEOUT;

        /* FIXME start watchdog for usb reset */

      }

      else {

        usb_dfu_status = DFU_STATUS_errSTALLEDPKT;

        return 0;

      }

      break;

    case DFU_DNLOAD:

      if((usb_dfu_state == DFU_STATE_dfuIDLE ||

          usb_dfu_state == DFU_STATE_dfuDNLOAD_IDLE) &&

         wLengthL > 0){

        if(!app_firmware_write()){

          usb_dfu_status = DFU_STATUS_errWRITE;

          usb_dfu_state = DFU_STATE_dfuERROR;

          return 0;

        }

        usb_dfu_state = DFU_STATE_dfuDNLOAD_IDLE;

      }

      else if(usb_dfu_state == DFU_STATE_dfuDNLOAD_IDLE &&

              wLengthL == 0){

        usb_toggle_dfu_handlers();

        usb_dfu_state = DFU_STATE_dfuMANIFEST_WAIT_RST;

      }

      else {

        usb_dfu_status = DFU_STATUS_errSTALLEDPKT;

        return 0;

      }

      break;

    case DFU_CLRSTATUS:

      if(usb_dfu_state == DFU_STATE_dfuERROR){

        usb_dfu_status = DFU_STATUS_OK;

        usb_dfu_state = DFU_STATE_dfuIDLE;

      }

      else {

        usb_dfu_status = DFU_STATUS_errSTALLEDPKT;

        return 0;

      }

      break;

    case DFU_ABORT:

      if(usb_dfu_state != DFU_STATE_appIDLE && \

         usb_dfu_state != DFU_STATE_appDETACH){

        /*FIXME stop all pending operations */

        usb_dfu_status = DFU_STATUS_OK;

        usb_dfu_state = DFU_STATE_dfuIDLE;

      }

      else {

        usb_dfu_status = DFU_STATUS_errSTALLEDPKT;

        return 0;

      }

      break;

    default:

      usb_dfu_status = DFU_STATUS_errSTALLEDPKT;

      return 0;

    }

  }

  else

    return 0;    /* invalid bRequestType */

  return 1;

}