2 * Copyright © 2016 Keith Packard <keithp@keithp.com>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
19 #include "libaltos_private.h"
31 #define USB_BUF_SIZE 64
33 struct altos_file_windows {
34 struct altos_file file;
37 /* Data used by the regular I/O */
43 /* Data used by winsock */
50 log_message(char *fmt, ...)
52 static FILE *log = NULL;
56 log = fopen("\\temp\\altos.txt", "w");
62 sprintf (buffer, "%4d-%02d-%02d %2d:%02d:%02d. ",
63 time.wYear, time.wMonth, time.wDay,
64 time.wHour, time.wMinute, time.wSecond);
67 vsprintf(buffer + strlen(buffer), fmt, a);
72 fputs(buffer, stdout);
78 _altos_set_last_windows_error(char *file, int line, DWORD error)
81 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
86 sizeof (message) / sizeof (TCHAR),
88 if (error != ERROR_SUCCESS)
89 log_message ("%s:%d (%d) %s\n", file, line, error, message);
90 altos_set_last_error(error, message);
93 #define altos_set_last_windows_error() _altos_set_last_windows_error(__FILE__, __LINE__, GetLastError())
94 #define altos_set_last_winsock_error() _altos_set_last_windows_error(__FILE__, __LINE__, WSAGetLastError())
96 PUBLIC struct altos_list *
97 altos_list_start(void)
99 struct altos_list *list = calloc(1, sizeof (struct altos_list));
103 list->dev_info = SetupDiGetClassDevs(NULL, "USB", NULL,
104 DIGCF_ALLCLASSES|DIGCF_PRESENT);
105 if (list->dev_info == INVALID_HANDLE_VALUE) {
106 altos_set_last_windows_error();
115 PUBLIC struct altos_list *
116 altos_ftdi_list_start(void)
118 struct altos_list *list = calloc(1, sizeof (struct altos_list));
122 list->dev_info = SetupDiGetClassDevs(NULL, "FTDIBUS", NULL,
123 DIGCF_ALLCLASSES|DIGCF_PRESENT);
124 if (list->dev_info == INVALID_HANDLE_VALUE) {
125 altos_set_last_windows_error();
138 { .windows = "AltusMetrum28", .real = "EasyMega" },
139 { .windows = "AltusMetrum2c", .real = "EasyMotor" },
144 altos_list_next(struct altos_list *list, struct altos_device *device)
146 SP_DEVINFO_DATA dev_info_data;
149 char friendlyname[256];
153 unsigned int vid, pid;
156 DWORD friendlyname_type;
157 DWORD friendlyname_len;
158 char instanceid[1024];
159 DWORD instanceid_len;
162 dev_info_data.cbSize = sizeof (SP_DEVINFO_DATA);
163 while(SetupDiEnumDeviceInfo(list->dev_info, list->index,
168 dev_key = SetupDiOpenDevRegKey(list->dev_info, &dev_info_data,
169 DICS_FLAG_GLOBAL, 0, DIREG_DEV,
171 if (dev_key == INVALID_HANDLE_VALUE) {
172 altos_set_last_windows_error();
181 vid = pid = serial = 0;
182 /* Fetch symbolic name for this device and parse out
183 * the vid/pid/serial info */
184 symbolic_len = sizeof(symbolic);
185 result = RegQueryValueEx(dev_key, "SymbolicName", NULL, NULL,
186 symbolic, &symbolic_len);
188 altos_set_last_windows_error();
190 sscanf((char *) symbolic + sizeof("\\??\\USB#VID_") - 1,
192 sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_") - 1,
194 sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_XXXX#") - 1,
197 if (vid == 0 || pid == 0 || serial == 0) {
198 if (SetupDiGetDeviceInstanceId(list->dev_info,
203 sscanf((char *) instanceid + sizeof("USB\\VID_") - 1,
205 sscanf((char *) instanceid + sizeof("USB\\VID_XXXX&PID_") - 1,
207 sscanf((char *) instanceid + sizeof("USB\\VID_XXXX&PID_XXXX\\") - 1,
210 altos_set_last_windows_error();
213 if (vid == 0 || pid == 0 || serial == 0) {
214 RegCloseKey(dev_key);
219 /* Fetch the com port name */
220 port_len = sizeof (port);
221 result = RegQueryValueEx(dev_key, "PortName", NULL, NULL,
223 RegCloseKey(dev_key);
225 altos_set_last_windows_error();
229 /* Fetch the device description which is the device name,
230 * with firmware that has unique USB ids */
231 friendlyname_len = sizeof (friendlyname);
232 if(!SetupDiGetDeviceRegistryProperty(list->dev_info,
236 (BYTE *)friendlyname,
237 sizeof(friendlyname),
240 altos_set_last_windows_error();
243 for (i = 0; name_map[i].windows; i++)
244 if (!strcmp(name_map[i].windows, friendlyname)) {
245 strcpy(friendlyname, name_map[i].real);
249 device->vendor = vid;
250 device->product = pid;
251 device->serial = serial;
252 strcpy(device->name, friendlyname);
254 strcpy(device->path, (char *) port);
257 result = GetLastError();
258 if (result != ERROR_NO_MORE_ITEMS)
259 altos_set_last_windows_error();
264 altos_list_finish(struct altos_list *list)
266 SetupDiDestroyDeviceInfoList(list->dev_info);
271 altos_queue_read(struct altos_file_windows *file)
275 return LIBALTOS_SUCCESS;
277 if (!ReadFile(file->handle, file->file.in_data, USB_BUF_SIZE, &got, &file->ov_read)) {
278 if (GetLastError() != ERROR_IO_PENDING) {
279 altos_set_last_windows_error();
280 return LIBALTOS_ERROR;
282 file->pend_read = TRUE;
284 file->pend_read = FALSE;
285 file->file.in_read = 0;
286 file->file.in_used = got;
288 return LIBALTOS_SUCCESS;
292 altos_wait_read(struct altos_file_windows *file, int timeout)
297 if (!file->pend_read)
298 return LIBALTOS_SUCCESS;
303 ret = WaitForSingleObject(file->ov_read.hEvent, timeout);
306 if (!GetOverlappedResult(file->handle, &file->ov_read, &got, FALSE)) {
307 if (GetLastError () != ERROR_OPERATION_ABORTED)
308 altos_set_last_windows_error();
309 return LIBALTOS_ERROR;
311 file->pend_read = FALSE;
312 file->file.in_read = 0;
313 file->file.in_used = got;
316 return LIBALTOS_TIMEOUT;
319 altos_set_last_windows_error();
320 return LIBALTOS_ERROR;
322 return LIBALTOS_SUCCESS;
326 altos_fill(struct altos_file *file_common, int timeout)
328 struct altos_file_windows *file = (struct altos_file_windows *) file_common;
332 if (file->file.in_read < file->file.in_used)
333 return LIBALTOS_SUCCESS;
335 file->file.in_read = file->file.in_used = 0;
337 if (file->is_winsock) {
344 /* Check to see if the socket has been closed */
345 if (file->socket == INVALID_SOCKET)
346 return LIBALTOS_ERROR;
348 #define POLL_TIMEOUT 10000
350 /* Poll to see if the socket has been closed
351 * as select doesn't abort when that happens
354 thistimeout = timeout;
355 if (thistimeout > POLL_TIMEOUT)
356 thistimeout = POLL_TIMEOUT;
358 thistimeout = POLL_TIMEOUT;
361 timeval.tv_sec = thistimeout / 1000;
362 timeval.tv_usec = (thistimeout % 1000) * 1000;
365 FD_SET(file->socket, &readfds);
367 ret = select(1, &readfds, NULL, NULL, &timeval);
371 timeout -= thistimeout;
373 return LIBALTOS_TIMEOUT;
380 altos_set_last_winsock_error();
381 return LIBALTOS_ERROR;
386 if (file->socket == INVALID_SOCKET) {
387 altos_set_last_winsock_error();
388 return LIBALTOS_ERROR;
391 ret = recv(file->socket, (char *) file->file.in_data, USB_BUF_SIZE, 0);
394 altos_set_last_winsock_error();
395 return LIBALTOS_ERROR;
397 file->file.in_read = 0;
398 file->file.in_used = ret;
400 if (file->handle == INVALID_HANDLE_VALUE) {
401 altos_set_last_windows_error();
402 return LIBALTOS_ERROR;
405 ret = altos_queue_read(file);
408 ret = altos_wait_read(file, timeout);
413 return LIBALTOS_SUCCESS;
417 altos_flush(struct altos_file *file_common)
419 struct altos_file_windows *file = (struct altos_file_windows *) file_common;
421 unsigned char *data = file->file.out_data;
422 int used = file->file.out_used;
425 if (file->is_winsock) {
428 put = send(file->socket, (char *) data, used, 0);
430 altos_set_last_winsock_error();
431 return LIBALTOS_ERROR;
438 if (!WriteFile(file->handle, data, used, &put, &file->ov_write)) {
439 if (GetLastError() != ERROR_IO_PENDING) {
440 altos_set_last_windows_error();
441 return LIBALTOS_ERROR;
443 ret = WaitForSingleObject(file->ov_write.hEvent, INFINITE);
446 if (!GetOverlappedResult(file->handle, &file->ov_write, &put, FALSE)) {
447 altos_set_last_windows_error();
448 return LIBALTOS_ERROR;
452 altos_set_last_windows_error();
453 return LIBALTOS_ERROR;
460 file->file.out_used = 0;
461 return LIBALTOS_SUCCESS;
465 open_serial(char *full_name)
470 handle = CreateFile(full_name, GENERIC_READ|GENERIC_WRITE,
471 0, NULL, OPEN_EXISTING,
472 FILE_FLAG_OVERLAPPED, NULL);
474 if (handle == INVALID_HANDLE_VALUE) {
475 altos_set_last_windows_error();
476 return INVALID_HANDLE_VALUE;
479 if (!GetCommState(handle, &dcb)) {
480 altos_set_last_windows_error();
482 return INVALID_HANDLE_VALUE;
484 dcb.BaudRate = CBR_9600;
487 dcb.fOutxCtsFlow = FALSE;
488 dcb.fOutxDsrFlow = FALSE;
489 dcb.fDtrControl = DTR_CONTROL_ENABLE;
490 dcb.fDsrSensitivity = FALSE;
491 dcb.fTXContinueOnXoff = FALSE;
494 dcb.fErrorChar = FALSE;
496 dcb.fRtsControl = RTS_CONTROL_ENABLE;
497 dcb.fAbortOnError = FALSE;
501 dcb.Parity = NOPARITY;
502 dcb.StopBits = ONESTOPBIT;
510 if (!SetCommState(handle, &dcb)) {
511 altos_set_last_windows_error();
513 return INVALID_HANDLE_VALUE;
518 PUBLIC struct altos_file *
519 altos_open(struct altos_device *device)
521 struct altos_file_windows *file = calloc (1, sizeof (struct altos_file_windows));
523 COMMTIMEOUTS timeouts;
529 strcpy(full_name, "\\\\.\\");
530 strcat(full_name, device->path);
532 file->handle = INVALID_HANDLE_VALUE;
534 for (i = 0; i < 5; i++) {
535 file->handle = open_serial(full_name);
536 if (file->handle != INVALID_HANDLE_VALUE)
538 altos_set_last_windows_error();
542 if (file->handle == INVALID_HANDLE_VALUE) {
547 /* The FTDI driver doesn't appear to work right unless you open it twice */
548 if (device->vendor == 0x0403) {
549 CloseHandle(file->handle);
550 file->handle = open_serial(full_name);
551 if (file->handle == INVALID_HANDLE_VALUE) {
557 timeouts.ReadIntervalTimeout = MAXDWORD;
558 timeouts.ReadTotalTimeoutMultiplier = MAXDWORD;
559 timeouts.ReadTotalTimeoutConstant = 1 << 30; /* almost forever */
560 timeouts.WriteTotalTimeoutMultiplier = 0;
561 timeouts.WriteTotalTimeoutConstant = 0;
562 SetCommTimeouts(file->handle, &timeouts);
564 file->ov_read.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
565 file->ov_write.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
571 altos_close(struct altos_file *file_common)
573 struct altos_file_windows *file = (struct altos_file_windows *) file_common;
575 if (file->is_winsock) {
576 SOCKET socket = file->socket;
577 if (socket != INVALID_SOCKET) {
578 file->socket = INVALID_SOCKET;
582 HANDLE handle = file->handle;
583 if (handle != INVALID_HANDLE_VALUE) {
584 HANDLE ov_read = file->ov_read.hEvent;
585 HANDLE ov_write = file->ov_write.hEvent;
586 file->handle = INVALID_HANDLE_VALUE;
587 file->ov_read.hEvent = INVALID_HANDLE_VALUE;
588 file->ov_write.hEvent = INVALID_HANDLE_VALUE;
589 PurgeComm(handle, PURGE_RXABORT|PURGE_RXCLEAR|PURGE_TXABORT|PURGE_TXCLEAR);
592 file->handle = INVALID_HANDLE_VALUE;
593 CloseHandle(ov_read);
594 CloseHandle(ov_write);
601 #define LUP_SET (LUP_RETURN_NAME| LUP_CONTAINERS | LUP_RETURN_ADDR | LUP_FLUSHCACHE |\
602 LUP_RETURN_TYPE | LUP_RETURN_BLOB | LUP_RES_SERVICE)
604 struct altos_bt_list {
609 struct altos_bt_list *
610 altos_bt_list_start(int inquiry_time)
612 struct altos_bt_list *bt_list;
613 WSAQUERYSET query_set;
616 /* Windows provides no way to set the time */
618 bt_list = calloc(1, sizeof (struct altos_bt_list));
620 altos_set_last_windows_error();
624 if ((retCode = WSAStartup(MAKEWORD(2,2),&bt_list->WSAData)) != 0) {
625 altos_set_last_winsock_error();
630 memset(&query_set, '\0', sizeof (query_set));
631 query_set.dwSize = sizeof(WSAQUERYSET);
632 query_set.dwNameSpace = NS_BTH;
634 retCode = WSALookupServiceBegin(&query_set, LUP_SET, &bt_list->lookup);
637 altos_set_last_winsock_error();
644 static unsigned char get_byte(BTH_ADDR ba, int shift)
646 return (ba >> ((5 - shift) << 3)) & 0xff;
649 static BTH_ADDR put_byte(unsigned char c, int shift)
651 return ((BTH_ADDR) c) << ((5 - shift) << 3);
655 ba2str(BTH_ADDR ba, char *str)
658 sprintf(str, "%02X:%02X:%02X:%02X:%02X:%02X",
670 unsigned int bytes[6];
672 sscanf(str, "%02x:%02x:%02x:%02x:%02x:%02x",
679 return (put_byte(bytes[0], 0) |
680 put_byte(bytes[1], 1) |
681 put_byte(bytes[2], 2) |
682 put_byte(bytes[3], 3) |
683 put_byte(bytes[4], 4) |
684 put_byte(bytes[5], 5));
688 altos_bt_list_next(struct altos_bt_list *bt_list,
689 struct altos_bt_device *device)
693 DWORD length = sizeof (buffer);;
694 WSAQUERYSET *results = (WSAQUERYSET *)buffer;
695 CSADDR_INFO *addr_info;
697 SOCKADDR_BTH *sockaddr_bth;
699 memset(buffer, '\0', sizeof(buffer));
701 retCode = WSALookupServiceNext(bt_list->lookup, LUP_SET, &length, results);
704 int error = WSAGetLastError();
705 if (error != WSAENOMORE && error != WSA_E_NO_MORE)
706 altos_set_last_winsock_error();
710 if (results->dwNumberOfCsAddrs > 0) {
712 addr_info = results->lpcsaBuffer;
714 strncpy(device->name, results->lpszServiceInstanceName, sizeof(device->name));
715 device->name[sizeof(device->name)-1] = '\0';
717 sockaddr_bth = (SOCKADDR_BTH *) addr_info->RemoteAddr.lpSockaddr;
719 ba2str(sockaddr_bth->btAddr, device->addr);
727 altos_bt_list_finish(struct altos_bt_list *bt_list)
729 WSALookupServiceEnd(bt_list->lookup);
734 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
736 strncpy(device->name, name, sizeof (device->name));
737 device->name[sizeof(device->name)-1] = '\0';
738 strncpy(device->addr, addr, sizeof (device->addr));
739 device->addr[sizeof(device->addr)-1] = '\0';
743 altos_bt_open(struct altos_bt_device *device)
745 struct altos_file_windows *file;
746 SOCKADDR_BTH sockaddr_bth;
749 file = calloc(1, sizeof (struct altos_file_windows));
754 file->is_winsock = TRUE;
755 file->socket = WSASocket(AF_BTH, SOCK_STREAM, BTHPROTO_RFCOMM, NULL, 0, WSA_FLAG_OVERLAPPED);
757 if (file->socket == INVALID_SOCKET) {
758 altos_set_last_winsock_error();
763 memset(&sockaddr_bth, '\0', sizeof (sockaddr_bth));
764 sockaddr_bth.addressFamily = AF_BTH;
765 sockaddr_bth.btAddr = str2ba(device->addr);
766 sockaddr_bth.port = altos_bt_port(device);
768 ret = connect(file->socket, (SOCKADDR *) &sockaddr_bth, sizeof (sockaddr_bth));
771 altos_set_last_winsock_error();
772 closesocket(file->socket);
774 log_message("Connection attempted to address %s port %d\n", device->addr, sockaddr_bth.port);
781 altos_pause_one_second(void)