common/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 #include "string.h"
  35 #include "proxmark3.h"
  36
  37 bool cmd_receive(UsbCommand* cmd) {
  38
  39   // Check if there is a usb packet available
  40   if (!usb_poll()) return false;
  41
  42   // Try to retrieve the available command frame
  43   size_t rxlen = usb_read((byte_t*)cmd,sizeof(UsbCommand));
  44
  45   // Check if the transfer was complete
  46   if (rxlen != sizeof(UsbCommand)) return false;
  47
  48   // Received command successfully
  49   return true;
  50 }
  51
  52 bool cmd_send(uint32_t cmd, uint32_t arg0, uint32_t arg1, uint32_t arg2, void* data, size_t len) {
  53   UsbCommand txcmd;
  54
  55   for (size_t i=0; i<sizeof(UsbCommand); i++) {
  56     ((byte_t*)&txcmd)[i] = 0x00;
  57   }
  58
  59   // Compose the outgoing command frame
  60   txcmd.cmd = cmd;
  61   txcmd.arg[0] = arg0;
  62   txcmd.arg[1] = arg1;
  63   txcmd.arg[2] = arg2;
  64
  65   // Add the (optional) content to the frame, with a maximum size of USB_CMD_DATA_SIZE
  66   if (data && len) {
  67     len = MIN(len,USB_CMD_DATA_SIZE);
  68     for (size_t i=0; i<len; i++) {
  69       txcmd.d.asBytes[i] = ((byte_t*)data)[i];
  70     }
  71   }
  72
  73   // Send frame and make sure all bytes are transmitted
  74   if (usb_write((byte_t*)&txcmd,sizeof(UsbCommand)) != 0) return false;
  75
  76   return true;
  77 }
  78
  79