+++ /dev/null
-//-----------------------------------------------------------------------------
-// Copyright (C) 2009 Michael Gernoth <michael at gernoth.net>
-// Copyright (C) 2010 iZsh <izsh at fail0verflow.com>
-//
-// This code is licensed to you under the terms of the GNU GPL, version 2 or,
-// at your option, any later version. See the LICENSE.txt file for the text of
-// the license.
-//-----------------------------------------------------------------------------
-// USB utilities
-//-----------------------------------------------------------------------------
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <stdbool.h>
-#include <unistd.h>
-#include <usb.h>
-#include <strings.h>
-#include <errno.h>
-
-#include "sleep.h"
-#include "proxusb.h"
-#include "proxmark3.h"
-#include "usb_cmd.h"
-
-// It seems to be missing for mingw
-#ifndef ETIMEDOUT
-#define ETIMEDOUT 116
-#endif
-
-usb_dev_handle *devh = NULL;
-static unsigned int claimed_iface = 0;
-unsigned char return_on_error = 0;
-unsigned char error_occured = 0;
-extern unsigned int current_command;
-
-void SendCommand_(HidCommand *c)
-{
- int ret;
-
-#if 0
- printf("Sending %d bytes\n", sizeof(HidCommand));
-#endif
- current_command = c->cmd;
- ret = usb_bulk_write(devh, 0x01, (char*)c, sizeof(HidCommand), 1000);
- if (ret<0) {
- error_occured = 1;
- if (return_on_error)
- return;
-
- fprintf(stderr, "write failed: %s!\nTrying to reopen device...\n",
- usb_strerror());
-
- if (devh) {
- usb_close(devh);
- devh = NULL;
- }
- while(!OpenProxmark(0)) { sleep(1); }
- printf(PROXPROMPT);
- fflush(NULL);
-
- return;
- }
-}
-
-bool ReceiveCommandPoll(HidCommand *c)
-{
- int ret;
-
- memset(c, 0, sizeof (HidCommand));
- ret = usb_bulk_read(devh, 0x82, (char*)c, sizeof(HidCommand), 500);
- if (ret<0) {
- if (ret != -ETIMEDOUT) {
- error_occured = 1;
- if (return_on_error)
- return false;
-
- fprintf(stderr, "read failed: %s(%d)!\nTrying to reopen device...\n",
- usb_strerror(), ret);
-
- if (devh) {
- usb_close(devh);
- devh = NULL;
- }
- while(!OpenProxmark(0)) { sleep(1); }
- printf(PROXPROMPT);
- fflush(NULL);
-
- return false;
- }
- } else {
- if (ret && (ret < sizeof(HidCommand))) {
- fprintf(stderr, "Read only %d instead of requested %d bytes!\n",
- ret, (int)sizeof(HidCommand));
- }
- }
-
- return ret > 0;
-}
-
-void ReceiveCommand(HidCommand *c)
-{
-// printf("%s()\n", __FUNCTION__);
- int retval = 0;
- do {
- retval = ReceiveCommandPoll(c);
- if (retval != 1) printf("ReceiveCommandPoll returned %d\n", retval);
- } while(retval<0);
-// printf("recv %x\n", c->cmd);
-}
-
-usb_dev_handle* findProxmark(int verbose, unsigned int *iface)
-{
- struct usb_bus *busses, *bus;
- usb_dev_handle *handle = NULL;
- struct prox_unit units[50];
- int iUnit = 0;
-
- usb_find_busses();
- usb_find_devices();
-
- busses = usb_get_busses();
-
- for (bus = busses; bus; bus = bus->next) {
- struct usb_device *dev;
-
- for (dev = bus->devices; dev; dev = dev->next) {
- struct usb_device_descriptor *desc = &(dev->descriptor);
-
- if ((desc->idProduct == 0x4b8f) && (desc->idVendor == 0x9ac4)) {
- handle = usb_open(dev);
- if (!handle) {
- if (verbose)
- fprintf(stderr, "open fabiled: %s!\n", usb_strerror());
- //return NULL;
- continue;
- }
- *iface = dev->config[0].interface[0].altsetting[0].bInterfaceNumber;
-
- struct prox_unit unit = {handle, {0}};
- usb_get_string_simple(handle, desc->iSerialNumber, unit.serial_number, sizeof(unit.serial_number));
- units[iUnit++] = unit;
-
- //return handle;
- }
- }
- }
-
- if (iUnit > 0) {
- int iSelection = 0;
-
- fprintf(stdout, "\nConnected units:\n");
-
- for (int i = 0; i < iUnit; i++) {
- struct usb_device * dev = usb_device(units[i].handle);
- fprintf(stdout, "\t%d. SN: %s [%s/%s]\n", i+1, units[i].serial_number, dev->bus->dirname, dev->filename);
- }
- if (iUnit > 1) {
- while (iSelection < 1 || iSelection > iUnit) {
- fprintf(stdout, "Which unit do you want to connect to? ");
- fscanf(stdin, "%d", &iSelection);
- }
- }
- else
- iSelection = 1;
- iSelection --;
-
- for (int i = 0; i < iUnit; i++) {
- if (iSelection == i) continue;
- usb_close(units[i].handle);
- units[i].handle = NULL;
- }
-
- return units[iSelection].handle;
- }
-
- return NULL;
-}
-
-usb_dev_handle* OpenProxmark(int verbose)
-{
- int ret;
- usb_dev_handle *handle = NULL;
- unsigned int iface;
-
- handle = findProxmark(verbose, &iface);
- if (!handle)
- return NULL;
-
-#ifdef __linux__
- /* detach kernel driver first */
- ret = usb_detach_kernel_driver_np(handle, iface);
- /* don't complain if no driver attached */
- if (ret<0 && ret != -61 && verbose)
- fprintf(stderr, "detach kernel driver failed: (%d) %s!\n", ret, usb_strerror());
-#endif
-
- // Needed for Windows. Optional for Mac OS and Linux
- ret = usb_set_configuration(handle, 1);
- if (ret < 0) {
- if (verbose)
- fprintf(stderr, "configuration set failed: %s!\n", usb_strerror());
- return NULL;
- }
-
- ret = usb_claim_interface(handle, iface);
- if (ret < 0) {
- if (verbose)
- fprintf(stderr, "claim failed: %s!\n", usb_strerror());
- return NULL;
- }
- claimed_iface = iface;
- devh = handle;
- return handle;
-}
-
-void CloseProxmark(void)
-{
- usb_release_interface(devh, claimed_iface);
- usb_close(devh);
- devh = NULL;
-}