mirror of https://github.com/GOSTSec/sgminer
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
506 lines
11 KiB
506 lines
11 KiB
/* |
|
* libusbx example program to manipulate U.are.U 4000B fingerprint scanner. |
|
* Copyright © 2007 Daniel Drake <dsd@gentoo.org> |
|
* |
|
* Basic image capture program only, does not consider the powerup quirks or |
|
* the fact that image encryption may be enabled. Not expected to work |
|
* flawlessly all of the time. |
|
* |
|
* This library is free software; you can redistribute it and/or |
|
* modify it under the terms of the GNU Lesser General Public |
|
* License as published by the Free Software Foundation; either |
|
* version 2.1 of the License, or (at your option) any later version. |
|
* |
|
* This library 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 |
|
* Lesser General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Lesser General Public |
|
* License along with this library; if not, write to the Free Software |
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
*/ |
|
|
|
#include <errno.h> |
|
#include <signal.h> |
|
#include <string.h> |
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
|
|
#include "libusb.h" |
|
|
|
#define EP_INTR (1 | LIBUSB_ENDPOINT_IN) |
|
#define EP_DATA (2 | LIBUSB_ENDPOINT_IN) |
|
#define CTRL_IN (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_IN) |
|
#define CTRL_OUT (LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT) |
|
#define USB_RQ 0x04 |
|
#define INTR_LENGTH 64 |
|
|
|
enum { |
|
MODE_INIT = 0x00, |
|
MODE_AWAIT_FINGER_ON = 0x10, |
|
MODE_AWAIT_FINGER_OFF = 0x12, |
|
MODE_CAPTURE = 0x20, |
|
MODE_SHUT_UP = 0x30, |
|
MODE_READY = 0x80, |
|
}; |
|
|
|
static int next_state(void); |
|
|
|
enum { |
|
STATE_AWAIT_MODE_CHANGE_AWAIT_FINGER_ON = 1, |
|
STATE_AWAIT_IRQ_FINGER_DETECTED, |
|
STATE_AWAIT_MODE_CHANGE_CAPTURE, |
|
STATE_AWAIT_IMAGE, |
|
STATE_AWAIT_MODE_CHANGE_AWAIT_FINGER_OFF, |
|
STATE_AWAIT_IRQ_FINGER_REMOVED, |
|
}; |
|
|
|
static int state = 0; |
|
static struct libusb_device_handle *devh = NULL; |
|
static unsigned char imgbuf[0x1b340]; |
|
static unsigned char irqbuf[INTR_LENGTH]; |
|
static struct libusb_transfer *img_transfer = NULL; |
|
static struct libusb_transfer *irq_transfer = NULL; |
|
static int img_idx = 0; |
|
static int do_exit = 0; |
|
|
|
static int find_dpfp_device(void) |
|
{ |
|
devh = libusb_open_device_with_vid_pid(NULL, 0x05ba, 0x000a); |
|
return devh ? 0 : -EIO; |
|
} |
|
|
|
static int print_f0_data(void) |
|
{ |
|
unsigned char data[0x10]; |
|
int r; |
|
unsigned int i; |
|
|
|
r = libusb_control_transfer(devh, CTRL_IN, USB_RQ, 0xf0, 0, data, |
|
sizeof(data), 0); |
|
if (r < 0) { |
|
fprintf(stderr, "F0 error %d\n", r); |
|
return r; |
|
} |
|
if ((unsigned int) r < sizeof(data)) { |
|
fprintf(stderr, "short read (%d)\n", r); |
|
return -1; |
|
} |
|
|
|
printf("F0 data:"); |
|
for (i = 0; i < sizeof(data); i++) |
|
printf("%02x ", data[i]); |
|
printf("\n"); |
|
return 0; |
|
} |
|
|
|
static int get_hwstat(unsigned char *status) |
|
{ |
|
int r; |
|
|
|
r = libusb_control_transfer(devh, CTRL_IN, USB_RQ, 0x07, 0, status, 1, 0); |
|
if (r < 0) { |
|
fprintf(stderr, "read hwstat error %d\n", r); |
|
return r; |
|
} |
|
if ((unsigned int) r < 1) { |
|
fprintf(stderr, "short read (%d)\n", r); |
|
return -1; |
|
} |
|
|
|
printf("hwstat reads %02x\n", *status); |
|
return 0; |
|
} |
|
|
|
static int set_hwstat(unsigned char data) |
|
{ |
|
int r; |
|
|
|
printf("set hwstat to %02x\n", data); |
|
r = libusb_control_transfer(devh, CTRL_OUT, USB_RQ, 0x07, 0, &data, 1, 0); |
|
if (r < 0) { |
|
fprintf(stderr, "set hwstat error %d\n", r); |
|
return r; |
|
} |
|
if ((unsigned int) r < 1) { |
|
fprintf(stderr, "short write (%d)", r); |
|
return -1; |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static int set_mode(unsigned char data) |
|
{ |
|
int r; |
|
printf("set mode %02x\n", data); |
|
|
|
r = libusb_control_transfer(devh, CTRL_OUT, USB_RQ, 0x4e, 0, &data, 1, 0); |
|
if (r < 0) { |
|
fprintf(stderr, "set mode error %d\n", r); |
|
return r; |
|
} |
|
if ((unsigned int) r < 1) { |
|
fprintf(stderr, "short write (%d)", r); |
|
return -1; |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static void LIBUSB_CALL cb_mode_changed(struct libusb_transfer *transfer) |
|
{ |
|
if (transfer->status != LIBUSB_TRANSFER_COMPLETED) { |
|
fprintf(stderr, "mode change transfer not completed!\n"); |
|
do_exit = 2; |
|
} |
|
|
|
printf("async cb_mode_changed length=%d actual_length=%d\n", |
|
transfer->length, transfer->actual_length); |
|
if (next_state() < 0) |
|
do_exit = 2; |
|
} |
|
|
|
static int set_mode_async(unsigned char data) |
|
{ |
|
unsigned char *buf = (unsigned char*) malloc(LIBUSB_CONTROL_SETUP_SIZE + 1); |
|
struct libusb_transfer *transfer; |
|
|
|
if (!buf) |
|
return -ENOMEM; |
|
|
|
transfer = libusb_alloc_transfer(0); |
|
if (!transfer) { |
|
free(buf); |
|
return -ENOMEM; |
|
} |
|
|
|
printf("async set mode %02x\n", data); |
|
libusb_fill_control_setup(buf, CTRL_OUT, USB_RQ, 0x4e, 0, 1); |
|
buf[LIBUSB_CONTROL_SETUP_SIZE] = data; |
|
libusb_fill_control_transfer(transfer, devh, buf, cb_mode_changed, NULL, |
|
1000); |
|
|
|
transfer->flags = LIBUSB_TRANSFER_SHORT_NOT_OK |
|
| LIBUSB_TRANSFER_FREE_BUFFER | LIBUSB_TRANSFER_FREE_TRANSFER; |
|
return libusb_submit_transfer(transfer); |
|
} |
|
|
|
static int do_sync_intr(unsigned char *data) |
|
{ |
|
int r; |
|
int transferred; |
|
|
|
r = libusb_interrupt_transfer(devh, EP_INTR, data, INTR_LENGTH, |
|
&transferred, 1000); |
|
if (r < 0) { |
|
fprintf(stderr, "intr error %d\n", r); |
|
return r; |
|
} |
|
if (transferred < INTR_LENGTH) { |
|
fprintf(stderr, "short read (%d)\n", r); |
|
return -1; |
|
} |
|
|
|
printf("recv interrupt %04x\n", *((uint16_t *) data)); |
|
return 0; |
|
} |
|
|
|
static int sync_intr(unsigned char type) |
|
{ |
|
int r; |
|
unsigned char data[INTR_LENGTH]; |
|
|
|
while (1) { |
|
r = do_sync_intr(data); |
|
if (r < 0) |
|
return r; |
|
if (data[0] == type) |
|
return 0; |
|
} |
|
} |
|
|
|
static int save_to_file(unsigned char *data) |
|
{ |
|
FILE *fd; |
|
char filename[64]; |
|
|
|
snprintf(filename, sizeof(filename), "finger%d.pgm", img_idx++); |
|
fd = fopen(filename, "w"); |
|
if (!fd) |
|
return -1; |
|
|
|
fputs("P5 384 289 255 ", fd); |
|
(void) fwrite(data + 64, 1, 384*289, fd); |
|
fclose(fd); |
|
printf("saved image to %s\n", filename); |
|
return 0; |
|
} |
|
|
|
static int next_state(void) |
|
{ |
|
int r = 0; |
|
printf("old state: %d\n", state); |
|
switch (state) { |
|
case STATE_AWAIT_IRQ_FINGER_REMOVED: |
|
state = STATE_AWAIT_MODE_CHANGE_AWAIT_FINGER_ON; |
|
r = set_mode_async(MODE_AWAIT_FINGER_ON); |
|
break; |
|
case STATE_AWAIT_MODE_CHANGE_AWAIT_FINGER_ON: |
|
state = STATE_AWAIT_IRQ_FINGER_DETECTED; |
|
break; |
|
case STATE_AWAIT_IRQ_FINGER_DETECTED: |
|
state = STATE_AWAIT_MODE_CHANGE_CAPTURE; |
|
r = set_mode_async(MODE_CAPTURE); |
|
break; |
|
case STATE_AWAIT_MODE_CHANGE_CAPTURE: |
|
state = STATE_AWAIT_IMAGE; |
|
break; |
|
case STATE_AWAIT_IMAGE: |
|
state = STATE_AWAIT_MODE_CHANGE_AWAIT_FINGER_OFF; |
|
r = set_mode_async(MODE_AWAIT_FINGER_OFF); |
|
break; |
|
case STATE_AWAIT_MODE_CHANGE_AWAIT_FINGER_OFF: |
|
state = STATE_AWAIT_IRQ_FINGER_REMOVED; |
|
break; |
|
default: |
|
printf("unrecognised state %d\n", state); |
|
} |
|
if (r < 0) { |
|
fprintf(stderr, "error detected changing state\n"); |
|
return r; |
|
} |
|
|
|
printf("new state: %d\n", state); |
|
return 0; |
|
} |
|
|
|
static void LIBUSB_CALL cb_irq(struct libusb_transfer *transfer) |
|
{ |
|
unsigned char irqtype = transfer->buffer[0]; |
|
|
|
if (transfer->status != LIBUSB_TRANSFER_COMPLETED) { |
|
fprintf(stderr, "irq transfer status %d?\n", transfer->status); |
|
do_exit = 2; |
|
libusb_free_transfer(transfer); |
|
irq_transfer = NULL; |
|
return; |
|
} |
|
|
|
printf("IRQ callback %02x\n", irqtype); |
|
switch (state) { |
|
case STATE_AWAIT_IRQ_FINGER_DETECTED: |
|
if (irqtype == 0x01) { |
|
if (next_state() < 0) { |
|
do_exit = 2; |
|
return; |
|
} |
|
} else { |
|
printf("finger-on-sensor detected in wrong state!\n"); |
|
} |
|
break; |
|
case STATE_AWAIT_IRQ_FINGER_REMOVED: |
|
if (irqtype == 0x02) { |
|
if (next_state() < 0) { |
|
do_exit = 2; |
|
return; |
|
} |
|
} else { |
|
printf("finger-on-sensor detected in wrong state!\n"); |
|
} |
|
break; |
|
} |
|
if (libusb_submit_transfer(irq_transfer) < 0) |
|
do_exit = 2; |
|
} |
|
|
|
static void LIBUSB_CALL cb_img(struct libusb_transfer *transfer) |
|
{ |
|
if (transfer->status != LIBUSB_TRANSFER_COMPLETED) { |
|
fprintf(stderr, "img transfer status %d?\n", transfer->status); |
|
do_exit = 2; |
|
libusb_free_transfer(transfer); |
|
img_transfer = NULL; |
|
return; |
|
} |
|
|
|
printf("Image callback\n"); |
|
save_to_file(imgbuf); |
|
if (next_state() < 0) { |
|
do_exit = 2; |
|
return; |
|
} |
|
if (libusb_submit_transfer(img_transfer) < 0) |
|
do_exit = 2; |
|
} |
|
|
|
static int init_capture(void) |
|
{ |
|
int r; |
|
|
|
r = libusb_submit_transfer(irq_transfer); |
|
if (r < 0) |
|
return r; |
|
|
|
r = libusb_submit_transfer(img_transfer); |
|
if (r < 0) { |
|
libusb_cancel_transfer(irq_transfer); |
|
while (irq_transfer) |
|
if (libusb_handle_events(NULL) < 0) |
|
break; |
|
return r; |
|
} |
|
|
|
/* start state machine */ |
|
state = STATE_AWAIT_IRQ_FINGER_REMOVED; |
|
return next_state(); |
|
} |
|
|
|
static int do_init(void) |
|
{ |
|
unsigned char status; |
|
int r; |
|
|
|
r = get_hwstat(&status); |
|
if (r < 0) |
|
return r; |
|
|
|
if (!(status & 0x80)) { |
|
r = set_hwstat(status | 0x80); |
|
if (r < 0) |
|
return r; |
|
r = get_hwstat(&status); |
|
if (r < 0) |
|
return r; |
|
} |
|
|
|
status &= ~0x80; |
|
r = set_hwstat(status); |
|
if (r < 0) |
|
return r; |
|
|
|
r = get_hwstat(&status); |
|
if (r < 0) |
|
return r; |
|
|
|
r = sync_intr(0x56); |
|
if (r < 0) |
|
return r; |
|
|
|
return 0; |
|
} |
|
|
|
static int alloc_transfers(void) |
|
{ |
|
img_transfer = libusb_alloc_transfer(0); |
|
if (!img_transfer) |
|
return -ENOMEM; |
|
|
|
irq_transfer = libusb_alloc_transfer(0); |
|
if (!irq_transfer) |
|
return -ENOMEM; |
|
|
|
libusb_fill_bulk_transfer(img_transfer, devh, EP_DATA, imgbuf, |
|
sizeof(imgbuf), cb_img, NULL, 0); |
|
libusb_fill_interrupt_transfer(irq_transfer, devh, EP_INTR, irqbuf, |
|
sizeof(irqbuf), cb_irq, NULL, 0); |
|
|
|
return 0; |
|
} |
|
|
|
static void sighandler(int signum) |
|
{ |
|
do_exit = 1; |
|
} |
|
|
|
int main(void) |
|
{ |
|
struct sigaction sigact; |
|
int r = 1; |
|
|
|
r = libusb_init(NULL); |
|
if (r < 0) { |
|
fprintf(stderr, "failed to initialise libusb\n"); |
|
exit(1); |
|
} |
|
|
|
r = find_dpfp_device(); |
|
if (r < 0) { |
|
fprintf(stderr, "Could not find/open device\n"); |
|
goto out; |
|
} |
|
|
|
r = libusb_claim_interface(devh, 0); |
|
if (r < 0) { |
|
fprintf(stderr, "usb_claim_interface error %d\n", r); |
|
goto out; |
|
} |
|
printf("claimed interface\n"); |
|
|
|
r = print_f0_data(); |
|
if (r < 0) |
|
goto out_release; |
|
|
|
r = do_init(); |
|
if (r < 0) |
|
goto out_deinit; |
|
|
|
/* async from here onwards */ |
|
|
|
r = alloc_transfers(); |
|
if (r < 0) |
|
goto out_deinit; |
|
|
|
r = init_capture(); |
|
if (r < 0) |
|
goto out_deinit; |
|
|
|
sigact.sa_handler = sighandler; |
|
sigemptyset(&sigact.sa_mask); |
|
sigact.sa_flags = 0; |
|
sigaction(SIGINT, &sigact, NULL); |
|
sigaction(SIGTERM, &sigact, NULL); |
|
sigaction(SIGQUIT, &sigact, NULL); |
|
|
|
while (!do_exit) { |
|
r = libusb_handle_events(NULL); |
|
if (r < 0) |
|
goto out_deinit; |
|
} |
|
|
|
printf("shutting down...\n"); |
|
|
|
if (irq_transfer) { |
|
r = libusb_cancel_transfer(irq_transfer); |
|
if (r < 0) |
|
goto out_deinit; |
|
} |
|
|
|
if (img_transfer) { |
|
r = libusb_cancel_transfer(img_transfer); |
|
if (r < 0) |
|
goto out_deinit; |
|
} |
|
|
|
while (irq_transfer || img_transfer) |
|
if (libusb_handle_events(NULL) < 0) |
|
break; |
|
|
|
if (do_exit == 1) |
|
r = 0; |
|
else |
|
r = 1; |
|
|
|
out_deinit: |
|
libusb_free_transfer(img_transfer); |
|
libusb_free_transfer(irq_transfer); |
|
set_mode(0); |
|
set_hwstat(0x80); |
|
out_release: |
|
libusb_release_interface(devh, 0); |
|
out: |
|
libusb_close(devh); |
|
libusb_exit(NULL); |
|
return r >= 0 ? r : -r; |
|
}
|
|
|