1*a61ed2ceSHans Rosenfeld /*
2*a61ed2ceSHans Rosenfeld  * This file and its contents are supplied under the terms of the
3*a61ed2ceSHans Rosenfeld  * Common Development and Distribution License ("CDDL"), version 1.0.
4*a61ed2ceSHans Rosenfeld  * You may only use this file in accordance with the terms of version
5*a61ed2ceSHans Rosenfeld  * 1.0 of the CDDL.
6*a61ed2ceSHans Rosenfeld  *
7*a61ed2ceSHans Rosenfeld  * A full copy of the text of the CDDL should have accompanied this
8*a61ed2ceSHans Rosenfeld  * source.  A copy of the CDDL is also available via the Internet at
9*a61ed2ceSHans Rosenfeld  * http://www.illumos.org/license/CDDL.
10*a61ed2ceSHans Rosenfeld  */
11*a61ed2ceSHans Rosenfeld 
12*a61ed2ceSHans Rosenfeld /*
13*a61ed2ceSHans Rosenfeld  * Copyright 2019, Joyent, Inc.
14*a61ed2ceSHans Rosenfeld  */
15*a61ed2ceSHans Rosenfeld 
16*a61ed2ceSHans Rosenfeld /*
17*a61ed2ceSHans Rosenfeld  * Open a YubiKey class device and get the basic information applet
18*a61ed2ceSHans Rosenfeld  * through an APDU while using poll(2) to check device readyness.
19*a61ed2ceSHans Rosenfeld  */
20*a61ed2ceSHans Rosenfeld 
21*a61ed2ceSHans Rosenfeld #include <err.h>
22*a61ed2ceSHans Rosenfeld #include <stdlib.h>
23*a61ed2ceSHans Rosenfeld #include <sys/types.h>
24*a61ed2ceSHans Rosenfeld #include <sys/stat.h>
25*a61ed2ceSHans Rosenfeld #include <fcntl.h>
26*a61ed2ceSHans Rosenfeld #include <strings.h>
27*a61ed2ceSHans Rosenfeld #include <unistd.h>
28*a61ed2ceSHans Rosenfeld #include <errno.h>
29*a61ed2ceSHans Rosenfeld #include <poll.h>
30*a61ed2ceSHans Rosenfeld 
31*a61ed2ceSHans Rosenfeld #include <sys/usb/clients/ccid/uccid.h>
32*a61ed2ceSHans Rosenfeld 
33*a61ed2ceSHans Rosenfeld static const uint8_t yk_req[] = {
34*a61ed2ceSHans Rosenfeld 	0x00, 0xa4, 0x04, 0x00, 0x07, 0xa0, 0x00, 0x00, 0x05, 0x27, 0x20, 0x01
35*a61ed2ceSHans Rosenfeld };
36*a61ed2ceSHans Rosenfeld 
37*a61ed2ceSHans Rosenfeld int
main(int argc,char * argv[])38*a61ed2ceSHans Rosenfeld main(int argc, char *argv[])
39*a61ed2ceSHans Rosenfeld {
40*a61ed2ceSHans Rosenfeld 	int fd, ret;
41*a61ed2ceSHans Rosenfeld 	struct pollfd pfds[1];
42*a61ed2ceSHans Rosenfeld 	uccid_cmd_txn_begin_t begin;
43*a61ed2ceSHans Rosenfeld 	uint8_t buf[UCCID_APDU_SIZE_MAX];
44*a61ed2ceSHans Rosenfeld 
45*a61ed2ceSHans Rosenfeld 	if (argc != 2) {
46*a61ed2ceSHans Rosenfeld 		errx(EXIT_FAILURE, "missing required ccid path");
47*a61ed2ceSHans Rosenfeld 	}
48*a61ed2ceSHans Rosenfeld 
49*a61ed2ceSHans Rosenfeld 	if ((fd = open(argv[1], O_RDWR)) < 0) {
50*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "failed to open %s", argv[1]);
51*a61ed2ceSHans Rosenfeld 	}
52*a61ed2ceSHans Rosenfeld 
53*a61ed2ceSHans Rosenfeld 	bzero(&begin, sizeof (begin));
54*a61ed2ceSHans Rosenfeld 	begin.uct_version = UCCID_CURRENT_VERSION;
55*a61ed2ceSHans Rosenfeld 	if (ioctl(fd, UCCID_CMD_TXN_BEGIN, &begin) != 0) {
56*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "failed to issue begin ioctl");
57*a61ed2ceSHans Rosenfeld 	}
58*a61ed2ceSHans Rosenfeld 
59*a61ed2ceSHans Rosenfeld 	pfds[0].fd = fd;
60*a61ed2ceSHans Rosenfeld 	pfds[0].events = POLLOUT | POLLIN | POLLRDNORM;
61*a61ed2ceSHans Rosenfeld 	pfds[0].revents = 0;
62*a61ed2ceSHans Rosenfeld 
63*a61ed2ceSHans Rosenfeld 	ret = poll(pfds, 1, 0);
64*a61ed2ceSHans Rosenfeld 	if (ret != 1) {
65*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "poll didn't return 1, returned %d "
66*a61ed2ceSHans Rosenfeld 		    "(errno %d)", ret, errno);
67*a61ed2ceSHans Rosenfeld 	}
68*a61ed2ceSHans Rosenfeld 
69*a61ed2ceSHans Rosenfeld 	if ((pfds[0].revents & POLLOUT) != POLLOUT) {
70*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "expecting pollout, got %d", pfds[0].revents);
71*a61ed2ceSHans Rosenfeld 	}
72*a61ed2ceSHans Rosenfeld 
73*a61ed2ceSHans Rosenfeld 	if ((ret = write(fd, yk_req, sizeof (yk_req))) < 0) {
74*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "failed to write data");
75*a61ed2ceSHans Rosenfeld 	}
76*a61ed2ceSHans Rosenfeld 
77*a61ed2ceSHans Rosenfeld 	pfds[0].revents = 0;
78*a61ed2ceSHans Rosenfeld 
79*a61ed2ceSHans Rosenfeld 	ret = poll(pfds, 1, -1);
80*a61ed2ceSHans Rosenfeld 	if (ret != 1) {
81*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "poll didn't return 1, returned %d "
82*a61ed2ceSHans Rosenfeld 		    "(errno %d)", ret, errno);
83*a61ed2ceSHans Rosenfeld 	}
84*a61ed2ceSHans Rosenfeld 
85*a61ed2ceSHans Rosenfeld 	if ((pfds[0].revents & (POLLIN | POLLRDNORM)) !=
86*a61ed2ceSHans Rosenfeld 	    (POLLIN | POLLRDNORM)) {
87*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "expecting pollin|pollrdnorm, got %d",
88*a61ed2ceSHans Rosenfeld 		    pfds[0].revents);
89*a61ed2ceSHans Rosenfeld 	}
90*a61ed2ceSHans Rosenfeld 
91*a61ed2ceSHans Rosenfeld 	if ((ret = read(fd, buf, sizeof (buf))) < 0) {
92*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "failed to read data");
93*a61ed2ceSHans Rosenfeld 	}
94*a61ed2ceSHans Rosenfeld 
95*a61ed2ceSHans Rosenfeld 	pfds[0].revents = 0;
96*a61ed2ceSHans Rosenfeld 
97*a61ed2ceSHans Rosenfeld 	ret = poll(pfds, 1, 0);
98*a61ed2ceSHans Rosenfeld 	if (ret != 1) {
99*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "poll didn't return 1, returned %d "
100*a61ed2ceSHans Rosenfeld 		    "(errno %d)", ret, errno);
101*a61ed2ceSHans Rosenfeld 	}
102*a61ed2ceSHans Rosenfeld 
103*a61ed2ceSHans Rosenfeld 	if ((pfds[0].revents & POLLOUT) != POLLOUT) {
104*a61ed2ceSHans Rosenfeld 		err(EXIT_FAILURE, "expecting pollout, got %d", pfds[0].revents);
105*a61ed2ceSHans Rosenfeld 	}
106*a61ed2ceSHans Rosenfeld 
107*a61ed2ceSHans Rosenfeld 	return (0);
108*a61ed2ceSHans Rosenfeld }