X-Git-Url: http://git.zerfleddert.de/cgi-bin/gitweb.cgi/proxmark3-svn/blobdiff_plain/b811cc51f9a21324dc7589d1254e767374488a20..6658905f18a1eebc148836f26c731dea9c1377dc:/linux/usb.c

diff --git a/linux/usb.c b/linux/usb.c
new file mode 100644
index 00000000..8145b64a
--- /dev/null
+++ b/linux/usb.c
@@ -0,0 +1,171 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <usb.h>
+#include <strings.h>
+#include <errno.h>
+
+#include "translate.h"
+#include "../winsrc/prox.h"
+#include "proxmark3.h"
+
+usb_dev_handle *devh = NULL;
+static unsigned int claimed_iface = 0;
+unsigned char return_on_error = 0;
+unsigned char error_occured = 0;
+
+void SendCommand(UsbCommand *c, BOOL wantAck) {
+	int ret;
+
+#if 0
+	printf("Sending %d bytes\n", sizeof(UsbCommand));
+#endif
+	ret = usb_bulk_write(devh, 0x01, (char*)c, sizeof(UsbCommand), 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(!(devh=OpenProxmark(0))) { sleep(1); }
+		printf(PROXPROMPT);
+		fflush(NULL);
+		
+		return;
+	}
+
+	if(wantAck) {
+		UsbCommand ack;
+		ReceiveCommand(&ack);
+		if(ack.cmd != CMD_ACK) {
+			printf("bad ACK\n");
+			exit(-1);
+		}
+	}
+}
+
+int ReceiveCommandP(UsbCommand *c) {
+	int ret;
+
+	bzero(c, sizeof(UsbCommand));
+	ret = usb_bulk_read(devh, 0x82, (char*)c, sizeof(UsbCommand), 500);
+	if (ret<0) {
+		if (ret != -ETIMEDOUT) {
+			error_occured = 1;
+			if (return_on_error)
+				return 0;
+
+			fprintf(stderr, "read failed: %s(%d)!\nTrying to reopen device...\n",
+				usb_strerror(), ret);
+
+			if (devh) {
+				usb_close(devh);
+				devh = NULL;
+			}
+			while(!(devh=OpenProxmark(0))) { sleep(1); }
+			printf(PROXPROMPT);
+			fflush(NULL);
+
+			return 0;
+		}
+	} else {
+		if (ret && (ret < sizeof(UsbCommand))) {
+			fprintf(stderr, "Read only %d instead of requested %d bytes!\n",
+				ret, (int)sizeof(UsbCommand));
+		}
+
+#if 0
+		{
+			int i;
+
+			printf("Read %d bytes\n", ret);
+			for (i = 0; i < ret; i++) {
+				printf("0x%02X ", ((unsigned char*)c)[i]);
+				if (!((i+1)%8))
+					printf("\n");
+			}
+			printf("\n");
+		}
+#endif
+	}
+
+	return ret;
+}
+
+void ReceiveCommand(UsbCommand *c) {
+	while(ReceiveCommandP(c)<0) {}
+}
+
+usb_dev_handle* findProxmark(int verbose, unsigned int *iface) {
+	struct usb_bus *busses, *bus;
+	usb_dev_handle *handle = NULL;
+
+	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 failed: %s!\n", usb_strerror());
+					return NULL;
+				}
+
+				*iface = dev->config[0].interface[0].altsetting[0].bInterfaceNumber;
+
+				return handle;
+			}
+		}
+	}
+
+	return NULL;
+}
+
+usb_dev_handle* OpenProxmark(int verbose) {
+	int ret;
+	usb_dev_handle *handle = NULL;
+	unsigned int iface;
+
+#ifndef __APPLE__
+	handle = findProxmark(verbose, &iface);
+	if (!handle)
+		return NULL;
+
+	/* Whatever... */
+	usb_reset(handle);
+#endif
+
+	handle = findProxmark(verbose, &iface);
+	if (!handle)
+		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);
+}