git.zerfleddert.de Git - proxmark3-svn/blob - armsrc/cmd.c

   1 /*
   2  * Proxmark send and receive commands
   3  *
   4  * Copyright (c) 2012, Roel Verdult
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  * notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  * notice, this list of conditions and the following disclaimer in the
  13  * documentation and/or other materials provided with the distribution.
  14  * 3. Neither the name of the copyright holders nor the
  15  * names of its contributors may be used to endorse or promote products
  16  * derived from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ''AS IS'' AND ANY
  19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21  * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
  22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  *
  29  * @file cmd.c
  30  * @brief
  31  */
  32
  33 #include "cmd.h"
  34
  35 #include <stddef.h>
  36 #include <stdint.h>
  37 #include <stdbool.h>
  38 #include "common.h"
  39 #include "usb_cmd.h"
  40 #include "usb_cdc.h"
  41
  42
  43 bool cmd_receive(UsbCommand* cmd) {
  44
  45         // Check if there is a usb packet available
  46         if (!usb_poll())
  47                 return false;
  48
  49         // Try to retrieve the available command frame
  50         size_t rxlen = usb_read((uint8_t*)cmd, sizeof(UsbCommand));
  51
  52         // Check if the transfer was complete
  53         if (rxlen != sizeof(UsbCommand))
  54                 return false;
  55
  56         // Received command successfully
  57         return true;
  58 }
  59
  60
  61 bool cmd_send(uint32_t cmd, uint32_t arg0, uint32_t arg1, uint32_t arg2, void* data, size_t len) {
  62         UsbCommand txcmd;
  63
  64         for (size_t i = 0; i < sizeof(UsbCommand); i++) {
  65                 ((uint8_t*)&txcmd)[i] = 0x00;
  66         }
  67
  68         // Compose the outgoing command frame
  69         txcmd.cmd = cmd;
  70         txcmd.arg[0] = arg0;
  71         txcmd.arg[1] = arg1;
  72         txcmd.arg[2] = arg2;
  73
  74         // Add the (optional) content to the frame, with a maximum size of USB_CMD_DATA_SIZE
  75         if (data && len) {
  76                 len = MIN(len, USB_CMD_DATA_SIZE);
  77                 for (size_t i = 0; i < len; i++) {
  78                         txcmd.d.asBytes[i] = ((uint8_t*)data)[i];
  79                 }
  80         }
  81
  82         // Send frame and make sure all bytes are transmitted
  83         if (usb_write((uint8_t*)&txcmd, sizeof(UsbCommand)) != 0) return false;
  84
  85         return true;
  86 }