summaryrefslogtreecommitdiffstats
path: root/host/cryptopad.c
blob: 42b1d36f2f1d407c37efb76a891fe0a7ededc82b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#include "project.h"
#include "../common/vendor_req.h"

#define TIMEOUT 4000

struct libusb_device * find_device(libusb_context *ctx)
{
        libusb_device **list;
        ssize_t num_devs;
        ssize_t i;

        num_devs = libusb_get_device_list(ctx, &list);
        for (i = 0; i < num_devs; ++i) {
                struct libusb_device_descriptor desc;
                struct libusb_device *dev = list[i];

                if (libusb_get_device_descriptor(dev, &desc))
                        continue;


		if (desc.idVendor!=0x1d6b) continue;
		if (desc.idProduct!=0x1932) continue;
		
		return dev;

        }
        libusb_free_device_list(list, 0);

	return NULL;
}


static void usage(char *n)
{
fprintf(stderr,"Usage:\n"
	"%s [-g [-n secs]] [-s [-p pwd]] [-h]\n"
	"        -h       display this message\n"
	"        -g       ask cryptopad to type password\n"
	"        -n secs  wait secs before typing password\n"
	"        -s       set password in cryptopad\n"
	"        -p pwd   password to set, prompts otherwise\n"
	"        -w       wipe password from cryptopad\n",
	n);

exit(1);
}




int main(int argc,char *argv[])
{
        libusb_context *ctx;
	libusb_device *dev;
	libusb_device_handle *devh;
	int wflag=0,gflag=0,sflag=0;
	char *pass=NULL;
        int c,ret;
        int delay=0;


        while((c=getopt(argc,argv,"hwgsp:n:"))!=-1) {
	switch(c) {
		case 'n':
		if (optarg) delay=atoi(optarg);
		break;
		case 'g':
		gflag++;
		break;
		case 's':
		sflag++;
		break;
		case 'w':
		wflag++;
		break;
		case 'p':
		pass=optarg;
		break;
		default:
		usage(argv[0]);
	}
	}

        if ((ret=libusb_init(&ctx)))
                errx(EX_IOERR, "unable to initialize libusb: %i", ret);

	dev=find_device(ctx);

	if (!dev) err(1,"no cryptopad found");

	if ((ret=libusb_open(dev, &devh))) err(1,"unable to open usb device: %i",ret);


	if (sflag) {
		char *pwd;
		if (!pass) pass=getpass("Enter password:");

		pwd=malloc(strlen(pass)+2);
		strcpy(pwd,pass);
		strcat(pwd,"\n");

		libusb_control_transfer( devh, LIBUSB_ENDPOINT_OUT | LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE, VENDOR_REQ_SET_KEY, 0, 0, pwd, strlen(pwd), TIMEOUT );
	}


	if (gflag) 
		libusb_control_transfer( devh, LIBUSB_ENDPOINT_OUT | LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE, VENDOR_REQ_SEND_KEY, delay, 0, NULL, 0, TIMEOUT );

	if (wflag) 
		libusb_control_transfer( devh, LIBUSB_ENDPOINT_OUT | LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE, VENDOR_REQ_CLEAR_KEY, 0, 0, NULL, 0, TIMEOUT );


	libusb_close(devh);

	return 0;
}