mirror of
https://github.com/ole00/afterburner.git
synced 2024-11-24 19:30:54 +00:00
300 lines
8.6 KiB
C
300 lines
8.6 KiB
C
#ifndef _SERIAL_PORT_H_
|
|
#define _SERIAL_PORT_H_
|
|
|
|
|
|
char guessedSerialDevice[512] = {0};
|
|
|
|
#ifdef _USE_WIN_API_
|
|
|
|
#include <windows.h>
|
|
|
|
#define SerialDeviceHandle HANDLE
|
|
#define DEFAULT_SERIAL_DEVICE_NAME "COM1"
|
|
#define INVALID_HANDLE INVALID_HANDLE_VALUE
|
|
|
|
#ifdef NO_CLOSE
|
|
static SerialDeviceHandle serH = INVALID_HANDLE;
|
|
#endif
|
|
|
|
// ideas: https://stackoverflow.com/questions/1388871/how-do-i-get-a-list-of-available-serial-ports-in-win32
|
|
static void serialDeviceGuessName(char** deviceName) {
|
|
char buf[64 * 1024] = {0};
|
|
int size = QueryDosDevice(NULL, buf, 64 * 1024);
|
|
int topComNum = 0;
|
|
|
|
// buffer was filled in
|
|
if (size > 0) {
|
|
char* text = buf;
|
|
int pos = 0;
|
|
int start = 0;
|
|
|
|
// search the received buffer for COM string
|
|
while (pos < size) {
|
|
int nameLen;
|
|
start = pos;
|
|
// find the string terminator
|
|
while(pos < size && buf[pos] != 0) {
|
|
pos++;
|
|
}
|
|
nameLen = pos - start;
|
|
// COM port found
|
|
if (nameLen >= 4 && nameLen <=6 && 0 == strncmp(text, "COM", 3) && text[3] >= '0' && text[3] <= '9') {
|
|
int comNum = atoi(text + 3);
|
|
if (comNum > topComNum) {
|
|
topComNum = comNum;
|
|
strcpy(guessedSerialDevice, text);
|
|
}
|
|
}
|
|
pos++;
|
|
text = &buf[pos];
|
|
}
|
|
|
|
// if we have found a COM port then pass it back as the result
|
|
if (topComNum > 0) {
|
|
*deviceName = guessedSerialDevice;
|
|
}
|
|
}
|
|
}
|
|
|
|
// https://www.xanthium.in/Serial-Port-Programming-using-Win32-API
|
|
|
|
static inline SerialDeviceHandle serialDeviceOpen(char* deviceName) {
|
|
SerialDeviceHandle h;
|
|
|
|
#ifdef NO_CLOSE
|
|
if (serH != INVALID_HANDLE) {
|
|
return serH;
|
|
}
|
|
#endif
|
|
|
|
h = CreateFile(
|
|
deviceName, //port name
|
|
GENERIC_READ | GENERIC_WRITE, //Read/Write
|
|
0, // No Sharing
|
|
NULL, // No Security
|
|
OPEN_EXISTING,// Open existing port only
|
|
0, // Non Overlapped I/O
|
|
NULL); // Null for Comm Devices
|
|
|
|
if (h != INVALID_HANDLE) {
|
|
BOOL result;
|
|
COMMTIMEOUTS timeouts = { 0 };
|
|
|
|
DCB dcbSerialParams = { 0 }; // Initializing DCB structure
|
|
dcbSerialParams.DCBlength = sizeof(dcbSerialParams);
|
|
|
|
result = GetCommState(h, &dcbSerialParams);
|
|
|
|
if (!result) {
|
|
return INVALID_HANDLE;
|
|
}
|
|
|
|
dcbSerialParams.BaudRate = CBR_57600; // Setting BaudRate
|
|
dcbSerialParams.ByteSize = 8; // Setting ByteSize = 8
|
|
dcbSerialParams.StopBits = ONESTOPBIT;// Setting StopBits = 1
|
|
dcbSerialParams.Parity = NOPARITY; // Setting Parity = None
|
|
dcbSerialParams.fOutxCtsFlow = FALSE;
|
|
dcbSerialParams.fOutxDsrFlow = FALSE;
|
|
dcbSerialParams.fDtrControl = DTR_CONTROL_DISABLE;
|
|
dcbSerialParams.fOutX = FALSE;
|
|
dcbSerialParams.fInX = FALSE;
|
|
dcbSerialParams.fRtsControl = RTS_CONTROL_DISABLE;
|
|
dcbSerialParams.fBinary = TRUE;
|
|
|
|
|
|
result = SetCommState(h, &dcbSerialParams);
|
|
if (!result) {
|
|
return INVALID_HANDLE;
|
|
}
|
|
|
|
timeouts.ReadIntervalTimeout = 30; // in milliseconds
|
|
timeouts.ReadTotalTimeoutConstant = 30; // in milliseconds
|
|
timeouts.ReadTotalTimeoutMultiplier = 10; // in milliseconds
|
|
timeouts.WriteTotalTimeoutConstant = 50; // in milliseconds
|
|
timeouts.WriteTotalTimeoutMultiplier = 10; // in milliseconds
|
|
|
|
result = SetCommTimeouts(h, &timeouts);
|
|
if (!result) {
|
|
return INVALID_HANDLE;
|
|
}
|
|
//ensure no leftover bytes exist on the serial line
|
|
result = PurgeComm(h, PURGE_RXCLEAR | PURGE_TXCLEAR | PURGE_RXABORT | PURGE_TXABORT);
|
|
#ifdef NO_CLOSE
|
|
serH = h;
|
|
#endif
|
|
return h;
|
|
} else {
|
|
return INVALID_HANDLE;
|
|
}
|
|
|
|
}
|
|
|
|
void serialDeviceCheckName(char* name, int maxSize) {
|
|
int nameLen = strlen(name);
|
|
//convert comxx to COMxx
|
|
if (strncmp(name, "com", 3) == 0 && (nameLen > 3 && name[3] >= '0' && name[3] <= '9')) {
|
|
name[0] = 'C';
|
|
name[1] = 'O';
|
|
name[2] = 'M';
|
|
}
|
|
|
|
// convert COMxx and higher to \\\\.\\COMxx
|
|
if (strncmp(name, "COM", 3) == 0 && (nameLen > 4 && name[3] >= '0' && name[3] <= '9')) {
|
|
if (nameLen + 4 < maxSize) {
|
|
int i;
|
|
for (i = nameLen - 1; i >= 0; i--) {
|
|
name[ i + 4] = name[i];
|
|
}
|
|
name[0] = '\\';
|
|
name[1] = '\\';
|
|
name[2] = '.';
|
|
name[3] = '\\';
|
|
}
|
|
}
|
|
}
|
|
|
|
static inline void serialDeviceClose(SerialDeviceHandle deviceHandle) {
|
|
#ifndef NO_CLOSE
|
|
CloseHandle(deviceHandle);
|
|
#endif
|
|
}
|
|
|
|
static inline int serialDeviceWrite(SerialDeviceHandle deviceHandle, char* buffer, int bytesToWrite) {
|
|
DWORD written = 0;
|
|
WriteFile(deviceHandle, buffer, bytesToWrite, &written, NULL);
|
|
return (int) written;
|
|
}
|
|
|
|
static inline int serialDeviceRead(SerialDeviceHandle deviceHandle, char* buffer, int bytesToRead) {
|
|
DWORD read = 0;
|
|
ReadFile(deviceHandle, buffer, bytesToRead, &read, NULL);
|
|
return (int) read;
|
|
}
|
|
|
|
#else
|
|
|
|
#include <ctype.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
#include <termios.h>
|
|
|
|
|
|
#define SerialDeviceHandle int
|
|
#define DEFAULT_SERIAL_DEVICE_NAME "/dev/ttyUSB0"
|
|
|
|
#define INVALID_HANDLE -1
|
|
#ifdef NO_CLOSE
|
|
static SerialDeviceHandle serH = INVALID_HANDLE;
|
|
#endif
|
|
|
|
|
|
#ifdef _OSX_
|
|
#define CHECK_SERIAL() (text != NULL)
|
|
#define LIST_DEVICES "ls -1 /dev/tty.usb* /dev/tty.wchusb* 2>/dev/null"
|
|
#else
|
|
#define CHECK_SERIAL() (text != NULL && 0 == strncmp(text + 18, "usb-", 4))
|
|
#define LIST_DEVICES "ls -1 /dev/serial/by-id/* 2>/dev/null"
|
|
#endif
|
|
|
|
// ideas: https://unix.stackexchange.com/questions/647235/debian-how-to-identify-usb-devices-with-similar-dev-tty-file
|
|
// https://pubs.opengroup.org/onlinepubs/009696799/functions/popen.html
|
|
static void serialDeviceGuessName(char** deviceName) {
|
|
int i;
|
|
char text[512];
|
|
int topTtyNum = 0;
|
|
|
|
FILE* f = popen(LIST_DEVICES, "r");
|
|
if (f == NULL) {
|
|
return;
|
|
}
|
|
i = 1;
|
|
while (fgets(text, 512, f) != NULL) {
|
|
// filer out non USB devices and prioritise those with Arduino name
|
|
if (CHECK_SERIAL()) {
|
|
int ttyNum = i;
|
|
int textLen = strlen(text);
|
|
// prefer Arduino over generic USB serial ports (does not work on OSX)
|
|
if (textLen > 29 && 0 == strncmp(text + 18, "usb-Arduino", 11)) {
|
|
ttyNum += 1000;
|
|
}
|
|
if (ttyNum > topTtyNum) {
|
|
topTtyNum = ttyNum;
|
|
strcpy(guessedSerialDevice, text);
|
|
//strip the new-line trailing markers
|
|
if (guessedSerialDevice[textLen - 1] == '\n' || guessedSerialDevice[textLen - 1] == '\r') {
|
|
guessedSerialDevice[textLen - 1] = 0;
|
|
}
|
|
}
|
|
}
|
|
i++;
|
|
}
|
|
pclose(f);
|
|
|
|
if (topTtyNum > 0) {
|
|
*deviceName = guessedSerialDevice;
|
|
}
|
|
}
|
|
|
|
static inline SerialDeviceHandle serialDeviceOpen(char* deviceName) {
|
|
|
|
SerialDeviceHandle h;
|
|
#ifdef NO_CLOSE
|
|
if (serH != INVALID_HANDLE) {
|
|
return serH;
|
|
}
|
|
#endif
|
|
|
|
h = open(deviceName, O_RDWR | O_NOCTTY | O_NONBLOCK);
|
|
|
|
if (h != INVALID_HANDLE) {
|
|
//set the serial port parameters
|
|
struct termios serial;
|
|
|
|
memset(&serial, 0, sizeof(struct termios));
|
|
cfsetispeed(&serial, B57600);
|
|
cfsetospeed(&serial, B57600);
|
|
|
|
serial.c_cflag |= CS8; // no parity, 1 stop bit
|
|
serial.c_cflag |= CREAD | CLOCAL;
|
|
|
|
if (0 != tcsetattr(h, TCSANOW, &serial)) {
|
|
close(h);
|
|
printf("Error: failed to set serial parameters %i, %s\n", errno, strerror(errno));
|
|
return INVALID_HANDLE;
|
|
}
|
|
|
|
//ensure no leftover bytes exist on the serial line
|
|
tcdrain(h);
|
|
tcflush(h, TCIOFLUSH); //flush both queues
|
|
#ifdef NO_CLOSE
|
|
serH = h;
|
|
#endif
|
|
return h;
|
|
} else {
|
|
return INVALID_HANDLE;
|
|
}
|
|
}
|
|
|
|
void serialDeviceCheckName(char* name, int maxSize) {
|
|
//nothing to check
|
|
}
|
|
|
|
static inline void serialDeviceClose(SerialDeviceHandle deviceHandle) {
|
|
#ifndef NO_CLOSE
|
|
close(deviceHandle);
|
|
#endif
|
|
}
|
|
|
|
static inline int serialDeviceWrite(SerialDeviceHandle deviceHandle, char* buffer, int bytesToWrite) {
|
|
return write(deviceHandle, buffer, bytesToWrite);
|
|
}
|
|
|
|
static inline int serialDeviceRead(SerialDeviceHandle deviceHandle, char* buffer, int bytesToRead) {
|
|
return read(deviceHandle, buffer, bytesToRead);
|
|
}
|
|
#endif
|
|
|
|
#endif /* _SERIAL_PORT_H_ */
|
|
|