[flashrom] [PATCH 2/2] Introduce serialport_config().

[Stefan Tauner]

This allows to easily reconfigure a serial port as needed in
the Bus Pirate speedup.

Signed-off-by: Stefan Tauner 
---
 programmer.h |    1 +
 serial.c     |  105 +++++++++++++++++++++++++++++++++-------------------------
 2 files changed, 60 insertions(+), 46 deletions(-)

diff --git a/programmer.h b/programmer.h
index 2ea7b48..6361eb8 100644
--- a/programmer.h
+++ b/programmer.h
@@ -656,6 +656,7 @@ typedef int fdtype;

void sp_flush_incoming(void);
 fdtype sp_openserport(char *dev, unsigned int baud);
+int serialport_config(fdtype fd, unsigned int baud);
 void __attribute__((noreturn)) sp_die(char *msg);
 extern fdtype sp_fd;
 /* expose serialport_shutdown as it's currently used by buspirate */
diff --git a/serial.c b/serial.c
index 5f74fd1..0504aaa 100644
--- a/serial.c
+++ b/serial.c
@@ -160,35 +160,18 @@ static void msg_perr_strerror(const char *msg)
 #endif
 }

-fdtype sp_openserport(char *dev, unsigned int baud)
+int serialport_config(fdtype fd, unsigned int baud)
 {
-#ifdef _WIN32
-	HANDLE fd;
-	char *dev2 = dev;
-	if ((strlen(dev) > 3) &&
-	    (tolower((unsigned char)dev[0]) == 'c') &&
-	    (tolower((unsigned char)dev[1]) == 'o') &&
-	    (tolower((unsigned char)dev[2]) == 'm')) {
-		dev2 = malloc(strlen(dev) + 5);
-		if (!dev2) {
-			msg_perr_strerror("Out of memory: ");
-			return SER_INV_FD;
-		}
-		strcpy(dev2, "\\.\");
-		strcpy(dev2 + 4, dev);
-	}
-	fd = CreateFile(dev2, GENERIC_READ | GENERIC_WRITE, 0, NULL,
-			OPEN_EXISTING, 0, NULL);
-	if (dev2 != dev)
-		free(dev2);
-	if (fd == INVALID_HANDLE_VALUE) {
-		msg_perr_strerror("Cannot open serial port: ");
-		return SER_INV_FD;
+	if (fd == SER_INV_FD) {
+		msg_perr("%s: File descriptor is invalid.\n", __func__);
+		return 1;
    }
+
+#ifdef _WIN32
    DCB dcb;
    if (!GetCommState(fd, &dcb)) {
    	msg_perr_strerror("Could not fetch original serial port configuration: ");
-		goto out_close;
+		return 1;
    }
    const struct baudentry *entry = round_baud(baud);
    dcb.BaudRate = entry->flag;
@@ -197,35 +180,25 @@ fdtype sp_openserport(char *dev, unsigned int baud)
    dcb.StopBits = ONESTOPBIT;
    if (!SetCommState(fd, &dcb)) {
    	msg_perr_strerror("Could not change serial port configuration: ");
-		goto out_close;
+		return 1;
    }
    if (!GetCommState(fd, &dcb)) {
    	msg_perr_strerror("Could not fetch new serial port configuration: ");
-		goto out_close;
+		return 1;
    }
    msg_pdbg("Baud rate is %ld.\n", dcb.BaudRate);
-	return fd;
-out_close:
-	CloseHandle(sp_fd);
-	return SER_INV_FD;
 #else
    struct termios wanted, observed;
-	int fd;
-	fd = open(dev, O_RDWR | O_NOCTTY | O_NDELAY);
-	if (fd < 0) {
-		msg_perr_strerror("Cannot open serial port: ");
-		return SER_INV_FD;
-	}
    fcntl(fd, F_SETFL, 0);
    if (tcgetattr(fd, &observed) != 0) {
    	msg_perr_strerror("Could not fetch original serial port configuration: ");
-		goto out_close;
+		return 1;
    }
    wanted = observed;
    const struct baudentry *entry = round_baud(baud);
    if (cfsetispeed(&wanted, entry->flag) != 0 || cfsetospeed(&wanted, entry->flag) != 0) {
    	msg_perr_strerror("Could not set serial baud rate: ");
-		goto out_close;
+		return 1;
    }
    wanted.c_cflag &= ~(PARENB | CSTOPB | CSIZE | CRTSCTS);
    wanted.c_cflag |= (CS8 | CLOCAL | CREAD);
@@ -234,11 +207,11 @@ out_close:
    wanted.c_oflag &= ~OPOST;
    if (tcsetattr(fd, TCSANOW, &wanted) != 0) {
    	msg_perr_strerror("Could not change serial port configuration: ");
-		goto out_close;
+		return 1;
    }
    if (tcgetattr(fd, &observed) != 0) {
    	msg_perr_strerror("Could not fetch new serial port configuration: ");
-		goto out_close;
+		return 1;
    }
    if (observed.c_cflag != wanted.c_cflag ||
        observed.c_lflag != wanted.c_lflag ||
@@ -246,14 +219,54 @@ out_close:
        observed.c_oflag != wanted.c_oflag ||
        cfgetispeed(&observed) != cfgetispeed(&wanted)) {
    	msg_perr("%s: Some requested options did not stick.\n", __func__);
-		goto out_close;
+		return 1;
    }
-	msg_pdbg("Baud rate is %d.\n", entry->baud);
-	return fd;
+	msg_pdbg("Baud rate is %d now.\n", entry->baud);
+#endif
+	return 0;
+}

-out_close:
-	close(sp_fd);
-	return SER_INV_FD;
+fdtype sp_openserport(char *dev, unsigned int baud)
+{
+	fdtype fd;
+#ifdef _WIN32
+	char *dev2 = dev;
+	if ((strlen(dev) > 3) &&
+	    (tolower((unsigned char)dev[0]) == 'c') &&
+	    (tolower((unsigned char)dev[1]) == 'o') &&
+	    (tolower((unsigned char)dev[2]) == 'm')) {
+		dev2 = malloc(strlen(dev) + 5);
+		if (!dev2) {
+			msg_perr_strerror("Out of memory: ");
+			return SER_INV_FD;
+		}
+		strcpy(dev2, "\\.\");
+		strcpy(dev2 + 4, dev);
+	}
+	fd = CreateFile(dev2, GENERIC_READ | GENERIC_WRITE, 0, NULL,
+			OPEN_EXISTING, 0, NULL);
+	if (dev2 != dev)
+		free(dev2);
+	if (fd == INVALID_HANDLE_VALUE) {
+		msg_perr_strerror("Cannot open serial port: ");
+		return SER_INV_FD;
+	}
+	if (serialport_config(fd, baud) != 0) {
+		CloseHandle(fd);
+		return SER_INV_FD;
+	}
+	return fd;
+#else
+	fd = open(dev, O_RDWR | O_NOCTTY | O_NDELAY);
+	if (fd < 0) {
+		msg_perr_strerror("Cannot open serial port: ");
+		return SER_INV_FD;
+	}
+	if (serialport_config(fd, baud) != 0) {
+		close(fd);
+		return SER_INV_FD;
+	}
+	return fd;
 #endif
 }