mirror of
https://github.com/Proxmark/proxmark3.git
synced 2025-07-06 21:21:17 -07:00
Refactoring uart interface (#341)
* uart: Major cleanups - Adds documentation to the uart API. - Fixes a buffer overflow issue in `uart_receive`, where the maximum parameter was ignored. - Splits the maximum length and bytes recieved variables in `uart_receive`. - Downsizes the receive buffer to the minimum required, saving 16MiB of RAM at runtime. - Refactors the POSIX and Win32 implementations of uart into separate files. - Removes the unused `uart_{get,set}_parity` functions, which were not implemented on Win32.
This commit is contained in:
parent
52244230d3
commit
067bfc8b76
7 changed files with 315 additions and 270 deletions
|
@ -13,13 +13,13 @@ RM = rm -f
|
||||||
MV = mv
|
MV = mv
|
||||||
|
|
||||||
#COMMON_FLAGS = -m32
|
#COMMON_FLAGS = -m32
|
||||||
VPATH = ../common ../zlib
|
VPATH = ../common ../zlib ../uart
|
||||||
OBJDIR = obj
|
OBJDIR = obj
|
||||||
|
|
||||||
LDLIBS = -L/opt/local/lib -L/usr/local/lib -lreadline -lpthread -lm
|
LDLIBS = -L/opt/local/lib -L/usr/local/lib -lreadline -lpthread -lm
|
||||||
LUALIB = ../liblua/liblua.a
|
LUALIB = ../liblua/liblua.a
|
||||||
LDFLAGS = $(COMMON_FLAGS)
|
LDFLAGS = $(COMMON_FLAGS)
|
||||||
CFLAGS = -std=c99 -D_ISOC99_SOURCE -I. -I../include -I../common -I../zlib -I/opt/local/include -I../liblua -Wall $(COMMON_FLAGS) -g -O3
|
CFLAGS = -std=c99 -D_ISOC99_SOURCE -I. -I../include -I../common -I../zlib -I../uart -I/opt/local/include -I../liblua -Wall $(COMMON_FLAGS) -g -O3
|
||||||
CXXFLAGS = -I../include -Wall -O3
|
CXXFLAGS = -I../include -Wall -O3
|
||||||
|
|
||||||
LUAPLATFORM = generic
|
LUAPLATFORM = generic
|
||||||
|
@ -78,7 +78,8 @@ DEPFLAGS = -MT $@ -MMD -MP -MF $(OBJDIR)/$*.Td
|
||||||
# make temporary to final dependeny files after successful compilation
|
# make temporary to final dependeny files after successful compilation
|
||||||
POSTCOMPILE = $(MV) -f $(OBJDIR)/$*.Td $(OBJDIR)/$*.d
|
POSTCOMPILE = $(MV) -f $(OBJDIR)/$*.Td $(OBJDIR)/$*.d
|
||||||
|
|
||||||
CORESRCS = uart.c \
|
CORESRCS = uart_posix.c \
|
||||||
|
uart_win32.c \
|
||||||
util.c \
|
util.c \
|
||||||
util_posix.c
|
util_posix.c
|
||||||
|
|
||||||
|
|
|
@ -19,6 +19,8 @@
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
# define unlink(x)
|
# define unlink(x)
|
||||||
|
#else
|
||||||
|
# include <unistd.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
static serial_port sp;
|
static serial_port sp;
|
||||||
|
@ -52,8 +54,7 @@ void ReceiveCommand(UsbCommand* rxcmd) {
|
||||||
byte_t* prx = prxcmd;
|
byte_t* prx = prxcmd;
|
||||||
size_t rxlen;
|
size_t rxlen;
|
||||||
while (true) {
|
while (true) {
|
||||||
rxlen = sizeof(UsbCommand) - (prx-prxcmd);
|
if (uart_receive(sp, prx, sizeof(UsbCommand) - (prx-prxcmd), &rxlen)) {
|
||||||
if (uart_receive(sp,prx,&rxlen)) {
|
|
||||||
prx += rxlen;
|
prx += rxlen;
|
||||||
if ((prx-prxcmd) >= sizeof(UsbCommand)) {
|
if ((prx-prxcmd) >= sizeof(UsbCommand)) {
|
||||||
return;
|
return;
|
||||||
|
@ -129,7 +130,7 @@ int main(int argc, char **argv)
|
||||||
|
|
||||||
fprintf(stderr,"Waiting for Proxmark to appear on %s",serial_port_name);
|
fprintf(stderr,"Waiting for Proxmark to appear on %s",serial_port_name);
|
||||||
do {
|
do {
|
||||||
sleep(1);
|
msleep(1000);
|
||||||
fprintf(stderr, ".");
|
fprintf(stderr, ".");
|
||||||
} while (!OpenProxmark(0));
|
} while (!OpenProxmark(0));
|
||||||
fprintf(stderr," Found.\n");
|
fprintf(stderr," Found.\n");
|
||||||
|
|
|
@ -57,26 +57,22 @@ struct receiver_arg {
|
||||||
int run;
|
int run;
|
||||||
};
|
};
|
||||||
|
|
||||||
byte_t rx[0x1000000];
|
byte_t rx[sizeof(UsbCommand)];
|
||||||
byte_t* prx = rx;
|
byte_t* prx = rx;
|
||||||
|
|
||||||
static void *uart_receiver(void *targ) {
|
static void *uart_receiver(void *targ) {
|
||||||
struct receiver_arg *arg = (struct receiver_arg*)targ;
|
struct receiver_arg *arg = (struct receiver_arg*)targ;
|
||||||
size_t rxlen;
|
size_t rxlen;
|
||||||
size_t cmd_count;
|
|
||||||
|
|
||||||
while (arg->run) {
|
while (arg->run) {
|
||||||
rxlen = sizeof(UsbCommand);
|
rxlen = 0;
|
||||||
if (uart_receive(sp, prx, &rxlen)) {
|
if (uart_receive(sp, prx, sizeof(UsbCommand) - (prx-rx), &rxlen)) {
|
||||||
prx += rxlen;
|
prx += rxlen;
|
||||||
if (((prx-rx) % sizeof(UsbCommand)) != 0) {
|
if (prx-rx < sizeof(UsbCommand)) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
cmd_count = (prx-rx) / sizeof(UsbCommand);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < cmd_count; i++) {
|
UsbCommandReceived((UsbCommand*)rx);
|
||||||
UsbCommandReceived((UsbCommand*)(rx+(i*sizeof(UsbCommand))));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
prx = rx;
|
prx = rx;
|
||||||
|
|
||||||
|
|
13
uart/README.md
Normal file
13
uart/README.md
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
# uart
|
||||||
|
|
||||||
|
This contains functionality for talking to UART/Serial devices on different platforms. The official client will build either `uart_posix.c` and `uart_win32.c`. Build targets for these files are contained in `client/Makefile`.
|
||||||
|
|
||||||
|
If you want to implement support for other platforms, you need to implement the methods provided in `uart.h`.
|
||||||
|
|
||||||
|
## Implementing a new driver
|
||||||
|
|
||||||
|
Each driver is called with a string, typically containing a path or other reference to a serial port on the host. The methods outlined in `uart.h` need to be implemented.
|
||||||
|
|
||||||
|
The hardware uses `common/usb_cdc.c` to implement a USB CDC endpoint exposed by the Atmel MCU.
|
||||||
|
|
||||||
|
|
|
@ -27,12 +27,10 @@
|
||||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
*
|
*
|
||||||
* @file uart.h
|
* @file uart.h
|
||||||
* @brief
|
|
||||||
*
|
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef _RS232_H_
|
#ifndef _PM3_UART_H_
|
||||||
#define _RS232_H_
|
#define _PM3_UART_H_
|
||||||
|
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
@ -43,45 +41,58 @@
|
||||||
|
|
||||||
typedef unsigned char byte_t;
|
typedef unsigned char byte_t;
|
||||||
|
|
||||||
// Handle platform specific includes
|
/* serial_port is declared as a void*, which you should cast to whatever type
|
||||||
#ifndef _WIN32
|
* makes sense to your connection method. Both the posix and win32
|
||||||
#include <termios.h>
|
* implementations define their own structs in place.
|
||||||
#include <sys/ioctl.h>
|
*/
|
||||||
#include <unistd.h>
|
|
||||||
#include <fcntl.h>
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/stat.h>
|
|
||||||
#include <limits.h>
|
|
||||||
#include <sys/time.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#else
|
|
||||||
#include <windows.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
typedef enum {
|
|
||||||
SP_INVALID = 0x00, // invalid value, error occured
|
|
||||||
SP_NONE = 0x01, // no parity (default)
|
|
||||||
SP_EVEN = 0x02, // even parity
|
|
||||||
SP_ODD = 0x03 // odd parity
|
|
||||||
} serial_port_parity;
|
|
||||||
|
|
||||||
// Define shortcut to types to make code more readable
|
|
||||||
typedef void* serial_port;
|
typedef void* serial_port;
|
||||||
|
|
||||||
|
/* Returned by uart_open if the serial port specified was invalid.
|
||||||
|
*/
|
||||||
#define INVALID_SERIAL_PORT (void*)(~1)
|
#define INVALID_SERIAL_PORT (void*)(~1)
|
||||||
|
|
||||||
|
/* Returned by uart_open if the serial port specified is in use by another
|
||||||
|
* process.
|
||||||
|
*/
|
||||||
#define CLAIMED_SERIAL_PORT (void*)(~2)
|
#define CLAIMED_SERIAL_PORT (void*)(~2)
|
||||||
|
|
||||||
|
/* Given a user-specified port name, connect to the port and return a structure
|
||||||
|
* used for future references to that port.
|
||||||
|
*
|
||||||
|
* On errors, this method returns INVALID_SERIAL_PORT or CLAIMED_SERIAL_PORT.
|
||||||
|
*/
|
||||||
serial_port uart_open(const char* pcPortName);
|
serial_port uart_open(const char* pcPortName);
|
||||||
|
|
||||||
|
/* Closes the given port.
|
||||||
|
*/
|
||||||
void uart_close(const serial_port sp);
|
void uart_close(const serial_port sp);
|
||||||
|
|
||||||
bool uart_set_speed(serial_port sp, const uint32_t uiPortSpeed);
|
/* Reads from the given serial port for up to 30ms.
|
||||||
uint32_t uart_get_speed(const serial_port sp);
|
* pbtRx: A pointer to a buffer for the returned data to be written to.
|
||||||
|
* pszMaxRxLen: The maximum data size we want to be sent.
|
||||||
|
* pszRxLen: The number of bytes that we were actually sent.
|
||||||
|
*
|
||||||
|
* Returns TRUE if any data was fetched, even if it was less than pszMaxRxLen.
|
||||||
|
*
|
||||||
|
* Returns FALSE if there was an error reading from the device. Note that a
|
||||||
|
* partial read may have completed into the buffer by the corresponding
|
||||||
|
* implementation, so pszRxLen should be checked to see if any data was written.
|
||||||
|
*/
|
||||||
|
bool uart_receive(const serial_port sp, byte_t* pbtRx, size_t pszMaxRxLen, size_t* pszRxLen);
|
||||||
|
|
||||||
bool uart_set_parity(serial_port sp, serial_port_parity spp);
|
/* Sends a buffer to a given serial port.
|
||||||
serial_port_parity uart_get_parity(const serial_port sp);
|
* pbtTx: A pointer to a buffer containing the data to send.
|
||||||
|
* szTxLen: The amount of data to be sent.
|
||||||
bool uart_receive(const serial_port sp, byte_t* pbtRx, size_t* pszRxLen);
|
*/
|
||||||
bool uart_send(const serial_port sp, const byte_t* pbtTx, const size_t szTxLen);
|
bool uart_send(const serial_port sp, const byte_t* pbtTx, const size_t szTxLen);
|
||||||
|
|
||||||
#endif // _PROXMARK3_RS232_H_
|
/* Sets the current speed of the serial port, in baud.
|
||||||
|
*/
|
||||||
|
bool uart_set_speed(serial_port sp, const uint32_t uiPortSpeed);
|
||||||
|
|
||||||
|
/* Gets the current speed of the serial port, in baud.
|
||||||
|
*/
|
||||||
|
uint32_t uart_get_speed(const serial_port sp);
|
||||||
|
|
||||||
|
#endif // _PM3_UART_H_
|
||||||
|
|
|
@ -26,17 +26,26 @@
|
||||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
*
|
*
|
||||||
* @file uart.c
|
* @file uart_posix.c
|
||||||
* @brief
|
|
||||||
*
|
*
|
||||||
|
* This version of the library has functionality removed which was not used by
|
||||||
|
* proxmark3 project.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "uart.h"
|
#include "uart.h"
|
||||||
|
|
||||||
// Test if we are dealing with unix operating systems
|
// Test if we are dealing with posix operating systems
|
||||||
#ifndef _WIN32
|
#ifndef _WIN32
|
||||||
|
|
||||||
#include <termios.h>
|
#include <termios.h>
|
||||||
|
#include <sys/ioctl.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <limits.h>
|
||||||
|
#include <sys/time.h>
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
typedef struct termios term_info;
|
typedef struct termios term_info;
|
||||||
typedef struct {
|
typedef struct {
|
||||||
int fd; // Serial port file descriptor
|
int fd; // Serial port file descriptor
|
||||||
|
@ -124,6 +133,99 @@ void uart_close(const serial_port sp) {
|
||||||
free(sp);
|
free(sp);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool uart_receive(const serial_port sp, byte_t* pbtRx, size_t pszMaxRxLen, size_t* pszRxLen) {
|
||||||
|
int res;
|
||||||
|
int byteCount;
|
||||||
|
fd_set rfds;
|
||||||
|
struct timeval tv;
|
||||||
|
|
||||||
|
// Reset the output count
|
||||||
|
*pszRxLen = 0;
|
||||||
|
|
||||||
|
do {
|
||||||
|
// Reset file descriptor
|
||||||
|
FD_ZERO(&rfds);
|
||||||
|
FD_SET(((serial_port_unix*)sp)->fd,&rfds);
|
||||||
|
tv = timeout;
|
||||||
|
res = select(((serial_port_unix*)sp)->fd+1, &rfds, NULL, NULL, &tv);
|
||||||
|
|
||||||
|
// Read error
|
||||||
|
if (res < 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read time-out
|
||||||
|
if (res == 0) {
|
||||||
|
if (*pszRxLen == 0) {
|
||||||
|
// Error, we received no data
|
||||||
|
return false;
|
||||||
|
} else {
|
||||||
|
// We received some data, but nothing more is available
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Retrieve the count of the incoming bytes
|
||||||
|
res = ioctl(((serial_port_unix*)sp)->fd, FIONREAD, &byteCount);
|
||||||
|
if (res < 0) return false;
|
||||||
|
|
||||||
|
// Cap the number of bytes, so we don't overrun the buffer
|
||||||
|
if (pszMaxRxLen - (*pszRxLen) < byteCount) {
|
||||||
|
byteCount = pszMaxRxLen - (*pszRxLen);
|
||||||
|
}
|
||||||
|
|
||||||
|
// There is something available, read the data
|
||||||
|
res = read(((serial_port_unix*)sp)->fd, pbtRx+(*pszRxLen), byteCount);
|
||||||
|
|
||||||
|
// Stop if the OS has some troubles reading the data
|
||||||
|
if (res <= 0) return false;
|
||||||
|
|
||||||
|
*pszRxLen += res;
|
||||||
|
|
||||||
|
if (*pszRxLen == pszMaxRxLen) {
|
||||||
|
// We have all the data we wanted.
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
} while (byteCount);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool uart_send(const serial_port sp, const byte_t* pbtTx, const size_t szTxLen) {
|
||||||
|
int32_t res;
|
||||||
|
size_t szPos = 0;
|
||||||
|
fd_set rfds;
|
||||||
|
struct timeval tv;
|
||||||
|
|
||||||
|
while (szPos < szTxLen) {
|
||||||
|
// Reset file descriptor
|
||||||
|
FD_ZERO(&rfds);
|
||||||
|
FD_SET(((serial_port_unix*)sp)->fd,&rfds);
|
||||||
|
tv = timeout;
|
||||||
|
res = select(((serial_port_unix*)sp)->fd+1, NULL, &rfds, NULL, &tv);
|
||||||
|
|
||||||
|
// Write error
|
||||||
|
if (res < 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write time-out
|
||||||
|
if (res == 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send away the bytes
|
||||||
|
res = write(((serial_port_unix*)sp)->fd,pbtTx+szPos,szTxLen-szPos);
|
||||||
|
|
||||||
|
// Stop if the OS has some troubles sending the data
|
||||||
|
if (res <= 0) return false;
|
||||||
|
|
||||||
|
szPos += res;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
bool uart_set_speed(serial_port sp, const uint32_t uiPortSpeed) {
|
bool uart_set_speed(serial_port sp, const uint32_t uiPortSpeed) {
|
||||||
const serial_port_unix* spu = (serial_port_unix*)sp;
|
const serial_port_unix* spu = (serial_port_unix*)sp;
|
||||||
speed_t stPortSpeed;
|
speed_t stPortSpeed;
|
||||||
|
@ -211,220 +313,5 @@ uint32_t uart_get_speed(const serial_port sp) {
|
||||||
return uiPortSpeed;
|
return uiPortSpeed;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool uart_set_parity(serial_port sp, serial_port_parity spp) {
|
|
||||||
struct termios ti;
|
|
||||||
const serial_port_unix* spu = (serial_port_unix*)sp;
|
|
||||||
if (tcgetattr(spu->fd,&ti) == -1) return false;
|
|
||||||
switch(spp) {
|
|
||||||
case SP_INVALID: return false;
|
|
||||||
case SP_NONE: ti.c_cflag &= ~(PARENB | PARODD); break;
|
|
||||||
case SP_EVEN: ti.c_cflag |= PARENB; ti.c_cflag &= ~(PARODD); break;
|
|
||||||
case SP_ODD: ti.c_cflag |= PARENB | PARODD; break;
|
|
||||||
}
|
|
||||||
return (tcsetattr(spu->fd,TCSANOW,&ti) != -1);
|
|
||||||
}
|
|
||||||
|
|
||||||
serial_port_parity uart_get_parity(const serial_port sp) {
|
|
||||||
struct termios ti;
|
|
||||||
const serial_port_unix* spu = (serial_port_unix*)sp;
|
|
||||||
if (tcgetattr(spu->fd,&ti) == -1) return SP_INVALID;
|
|
||||||
|
|
||||||
if (ti.c_cflag & PARENB) {
|
|
||||||
if (ti.c_cflag & PARODD) {
|
|
||||||
return SP_ODD;
|
|
||||||
} else {
|
|
||||||
return SP_EVEN;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
return SP_NONE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool uart_cts(const serial_port sp) {
|
|
||||||
char status;
|
|
||||||
if (ioctl(((serial_port_unix*)sp)->fd,TIOCMGET,&status) < 0) return false;
|
|
||||||
return (status & TIOCM_CTS);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool uart_receive(const serial_port sp, byte_t* pbtRx, size_t* pszRxLen) {
|
|
||||||
|
|
||||||
int res;
|
|
||||||
int byteCount;
|
|
||||||
fd_set rfds;
|
|
||||||
struct timeval tv;
|
|
||||||
|
|
||||||
// Reset the output count
|
|
||||||
*pszRxLen = 0;
|
|
||||||
|
|
||||||
do {
|
|
||||||
// Reset file descriptor
|
|
||||||
FD_ZERO(&rfds);
|
|
||||||
FD_SET(((serial_port_unix*)sp)->fd,&rfds);
|
|
||||||
tv = timeout;
|
|
||||||
res = select(((serial_port_unix*)sp)->fd+1, &rfds, NULL, NULL, &tv);
|
|
||||||
|
|
||||||
// Read error
|
|
||||||
if (res < 0) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Read time-out
|
|
||||||
if (res == 0) {
|
|
||||||
if (*pszRxLen == 0) {
|
|
||||||
// Error, we received no data
|
|
||||||
return false;
|
|
||||||
} else {
|
|
||||||
// We received some data, but nothing more is available
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Retrieve the count of the incoming bytes
|
|
||||||
res = ioctl(((serial_port_unix*)sp)->fd, FIONREAD, &byteCount);
|
|
||||||
if (res < 0) return false;
|
|
||||||
|
|
||||||
// There is something available, read the data
|
|
||||||
res = read(((serial_port_unix*)sp)->fd,pbtRx+(*pszRxLen),byteCount);
|
|
||||||
|
|
||||||
// Stop if the OS has some troubles reading the data
|
|
||||||
if (res <= 0) return false;
|
|
||||||
|
|
||||||
*pszRxLen += res;
|
|
||||||
|
|
||||||
if(res==byteCount)
|
|
||||||
return true;
|
|
||||||
|
|
||||||
} while (byteCount);
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool uart_send(const serial_port sp, const byte_t* pbtTx, const size_t szTxLen) {
|
|
||||||
int32_t res;
|
|
||||||
size_t szPos = 0;
|
|
||||||
fd_set rfds;
|
|
||||||
struct timeval tv;
|
|
||||||
|
|
||||||
while (szPos < szTxLen) {
|
|
||||||
// Reset file descriptor
|
|
||||||
FD_ZERO(&rfds);
|
|
||||||
FD_SET(((serial_port_unix*)sp)->fd,&rfds);
|
|
||||||
tv = timeout;
|
|
||||||
res = select(((serial_port_unix*)sp)->fd+1, NULL, &rfds, NULL, &tv);
|
|
||||||
|
|
||||||
// Write error
|
|
||||||
if (res < 0) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write time-out
|
|
||||||
if (res == 0) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Send away the bytes
|
|
||||||
res = write(((serial_port_unix*)sp)->fd,pbtTx+szPos,szTxLen-szPos);
|
|
||||||
|
|
||||||
// Stop if the OS has some troubles sending the data
|
|
||||||
if (res <= 0) return false;
|
|
||||||
|
|
||||||
szPos += res;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
// The windows serial port implementation
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
HANDLE hPort; // Serial port handle
|
|
||||||
DCB dcb; // Device control settings
|
|
||||||
COMMTIMEOUTS ct; // Serial port time-out configuration
|
|
||||||
} serial_port_windows;
|
|
||||||
|
|
||||||
void upcase(char *p) {
|
|
||||||
while(*p != '\0') {
|
|
||||||
if(*p >= 97 && *p <= 122) {
|
|
||||||
*p -= 32;
|
|
||||||
}
|
|
||||||
++p;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
serial_port uart_open(const char* pcPortName) {
|
|
||||||
char acPortName[255];
|
|
||||||
serial_port_windows* sp = malloc(sizeof(serial_port_windows));
|
|
||||||
|
|
||||||
// Copy the input "com?" to "\\.\COM?" format
|
|
||||||
sprintf(acPortName,"\\\\.\\%s",pcPortName);
|
|
||||||
upcase(acPortName);
|
|
||||||
|
|
||||||
// Try to open the serial port
|
|
||||||
sp->hPort = CreateFileA(acPortName,GENERIC_READ|GENERIC_WRITE,0,NULL,OPEN_EXISTING,0,NULL);
|
|
||||||
if (sp->hPort == INVALID_HANDLE_VALUE) {
|
|
||||||
uart_close(sp);
|
|
||||||
return INVALID_SERIAL_PORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Prepare the device control
|
|
||||||
memset(&sp->dcb, 0, sizeof(DCB));
|
|
||||||
sp->dcb.DCBlength = sizeof(DCB);
|
|
||||||
if(!BuildCommDCBA("baud=9600 data=8 parity=N stop=1",&sp->dcb)) {
|
|
||||||
uart_close(sp);
|
|
||||||
return INVALID_SERIAL_PORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Update the active serial port
|
|
||||||
if(!SetCommState(sp->hPort,&sp->dcb)) {
|
|
||||||
uart_close(sp);
|
|
||||||
return INVALID_SERIAL_PORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
sp->ct.ReadIntervalTimeout = 0;
|
|
||||||
sp->ct.ReadTotalTimeoutMultiplier = 0;
|
|
||||||
sp->ct.ReadTotalTimeoutConstant = 30;
|
|
||||||
sp->ct.WriteTotalTimeoutMultiplier = 0;
|
|
||||||
sp->ct.WriteTotalTimeoutConstant = 30;
|
|
||||||
|
|
||||||
if(!SetCommTimeouts(sp->hPort,&sp->ct)) {
|
|
||||||
uart_close(sp);
|
|
||||||
return INVALID_SERIAL_PORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
PurgeComm(sp->hPort, PURGE_RXABORT | PURGE_RXCLEAR);
|
|
||||||
|
|
||||||
return sp;
|
|
||||||
}
|
|
||||||
|
|
||||||
void uart_close(const serial_port sp) {
|
|
||||||
CloseHandle(((serial_port_windows*)sp)->hPort);
|
|
||||||
free(sp);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool uart_set_speed(serial_port sp, const uint32_t uiPortSpeed) {
|
|
||||||
serial_port_windows* spw;
|
|
||||||
spw = (serial_port_windows*)sp;
|
|
||||||
spw->dcb.BaudRate = uiPortSpeed;
|
|
||||||
return SetCommState(spw->hPort, &spw->dcb);
|
|
||||||
}
|
|
||||||
|
|
||||||
uint32_t uart_get_speed(const serial_port sp) {
|
|
||||||
const serial_port_windows* spw = (serial_port_windows*)sp;
|
|
||||||
if (!GetCommState(spw->hPort, (serial_port)&spw->dcb)) {
|
|
||||||
return spw->dcb.BaudRate;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool uart_receive(const serial_port sp, byte_t* pbtRx, size_t* pszRxLen) {
|
|
||||||
ReadFile(((serial_port_windows*)sp)->hPort,pbtRx,*pszRxLen,(LPDWORD)pszRxLen,NULL);
|
|
||||||
return (*pszRxLen != 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool uart_send(const serial_port sp, const byte_t* pbtTx, const size_t szTxLen) {
|
|
||||||
DWORD dwTxLen = 0;
|
|
||||||
return WriteFile(((serial_port_windows*)sp)->hPort,pbtTx,szTxLen,&dwTxLen,NULL);
|
|
||||||
return (dwTxLen != 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
136
uart/uart_win32.c
Normal file
136
uart/uart_win32.c
Normal file
|
@ -0,0 +1,136 @@
|
||||||
|
/*
|
||||||
|
* Generic uart / rs232/ serial port library
|
||||||
|
*
|
||||||
|
* Copyright (c) 2013, Roel Verdult
|
||||||
|
* All rights reserved.
|
||||||
|
*
|
||||||
|
* Redistribution and use in source and binary forms, with or without
|
||||||
|
* modification, are permitted provided that the following conditions are met:
|
||||||
|
* 1. Redistributions of source code must retain the above copyright
|
||||||
|
* notice, this list of conditions and the following disclaimer.
|
||||||
|
* 2. Redistributions in binary form must reproduce the above copyright
|
||||||
|
* notice, this list of conditions and the following disclaimer in the
|
||||||
|
* documentation and/or other materials provided with the distribution.
|
||||||
|
* 3. Neither the name of the copyright holders nor the
|
||||||
|
* names of its contributors may be used to endorse or promote products
|
||||||
|
* derived from this software without specific prior written permission.
|
||||||
|
*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ''AS IS'' AND ANY
|
||||||
|
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
||||||
|
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||||
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||||
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*
|
||||||
|
* @file uart_win32.c
|
||||||
|
*
|
||||||
|
* Note: the win32 version of this library has also been seen under the GPLv3+
|
||||||
|
* license as part of the libnfc project, which appears to have additional
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* This version of the library has functionality removed which was not used by
|
||||||
|
* proxmark3 project.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "uart.h"
|
||||||
|
|
||||||
|
// The windows serial port implementation
|
||||||
|
#ifdef _WIN32
|
||||||
|
#include <windows.h>
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
HANDLE hPort; // Serial port handle
|
||||||
|
DCB dcb; // Device control settings
|
||||||
|
COMMTIMEOUTS ct; // Serial port time-out configuration
|
||||||
|
} serial_port_windows;
|
||||||
|
|
||||||
|
void upcase(char *p) {
|
||||||
|
while(*p != '\0') {
|
||||||
|
if(*p >= 97 && *p <= 122) {
|
||||||
|
*p -= 32;
|
||||||
|
}
|
||||||
|
++p;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
serial_port uart_open(const char* pcPortName) {
|
||||||
|
char acPortName[255];
|
||||||
|
serial_port_windows* sp = malloc(sizeof(serial_port_windows));
|
||||||
|
|
||||||
|
// Copy the input "com?" to "\\.\COM?" format
|
||||||
|
sprintf(acPortName,"\\\\.\\%s",pcPortName);
|
||||||
|
upcase(acPortName);
|
||||||
|
|
||||||
|
// Try to open the serial port
|
||||||
|
sp->hPort = CreateFileA(acPortName,GENERIC_READ|GENERIC_WRITE,0,NULL,OPEN_EXISTING,0,NULL);
|
||||||
|
if (sp->hPort == INVALID_HANDLE_VALUE) {
|
||||||
|
uart_close(sp);
|
||||||
|
return INVALID_SERIAL_PORT;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Prepare the device control
|
||||||
|
memset(&sp->dcb, 0, sizeof(DCB));
|
||||||
|
sp->dcb.DCBlength = sizeof(DCB);
|
||||||
|
if(!BuildCommDCBA("baud=9600 data=8 parity=N stop=1",&sp->dcb)) {
|
||||||
|
uart_close(sp);
|
||||||
|
return INVALID_SERIAL_PORT;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update the active serial port
|
||||||
|
if(!SetCommState(sp->hPort,&sp->dcb)) {
|
||||||
|
uart_close(sp);
|
||||||
|
return INVALID_SERIAL_PORT;
|
||||||
|
}
|
||||||
|
|
||||||
|
sp->ct.ReadIntervalTimeout = 0;
|
||||||
|
sp->ct.ReadTotalTimeoutMultiplier = 0;
|
||||||
|
sp->ct.ReadTotalTimeoutConstant = 30;
|
||||||
|
sp->ct.WriteTotalTimeoutMultiplier = 0;
|
||||||
|
sp->ct.WriteTotalTimeoutConstant = 30;
|
||||||
|
|
||||||
|
if(!SetCommTimeouts(sp->hPort,&sp->ct)) {
|
||||||
|
uart_close(sp);
|
||||||
|
return INVALID_SERIAL_PORT;
|
||||||
|
}
|
||||||
|
|
||||||
|
PurgeComm(sp->hPort, PURGE_RXABORT | PURGE_RXCLEAR);
|
||||||
|
|
||||||
|
return sp;
|
||||||
|
}
|
||||||
|
|
||||||
|
void uart_close(const serial_port sp) {
|
||||||
|
CloseHandle(((serial_port_windows*)sp)->hPort);
|
||||||
|
free(sp);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool uart_receive(const serial_port sp, byte_t* pbtRx, size_t pszMaxRxLen, size_t* pszRxLen) {
|
||||||
|
ReadFile(((serial_port_windows*)sp)->hPort,pbtRx,pszMaxRxLen,(LPDWORD)pszRxLen,NULL);
|
||||||
|
return (*pszRxLen != 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool uart_send(const serial_port sp, const byte_t* pbtTx, const size_t szTxLen) {
|
||||||
|
DWORD dwTxLen = 0;
|
||||||
|
return WriteFile(((serial_port_windows*)sp)->hPort,pbtTx,szTxLen,&dwTxLen,NULL);
|
||||||
|
return (dwTxLen != 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool uart_set_speed(serial_port sp, const uint32_t uiPortSpeed) {
|
||||||
|
serial_port_windows* spw;
|
||||||
|
spw = (serial_port_windows*)sp;
|
||||||
|
spw->dcb.BaudRate = uiPortSpeed;
|
||||||
|
return SetCommState(spw->hPort, &spw->dcb);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t uart_get_speed(const serial_port sp) {
|
||||||
|
const serial_port_windows* spw = (serial_port_windows*)sp;
|
||||||
|
if (!GetCommState(spw->hPort, (serial_port)&spw->dcb)) {
|
||||||
|
return spw->dcb.BaudRate;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
Loading…
Add table
Add a link
Reference in a new issue