Subversion Repositories openarty

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <termios.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#include <poll.h>
#include <signal.h>
#include <ctype.h>
#include <assert.h>
#include <errno.h>
 
#include "port.h"
 
void	sigstop(int v) {
	fprintf(stderr, "SIGSTOP!!\n");
	exit(0);
}
void	sighup(int v) {
	fprintf(stderr, "SIGHUP!!\n");
	exit(0);
}
void	sigint(int v) {
	fprintf(stderr, "SIGINT!!\n");
	exit(0);
}
void	sigsegv(int v) {
	fprintf(stderr, "SIGSEGV!!\n");
	exit(0);
}
void	sigbus(int v) {
	fprintf(stderr, "SIGBUS!!\n");
	exit(0);
}
void	sigpipe(int v) {
	fprintf(stderr, "SIGPIPE!!\n");
	exit(0);
}
 
int	setup_listener(const int port) {
	int	skt;
	struct  sockaddr_in     my_addr;
 
	printf("Listening on port %d\n", port);
 
	skt = socket(AF_INET, SOCK_STREAM, 0);
	if (skt < 0) {
		perror("Could not allocate socket: ");
		exit(-1);
	}
 
	// Set the reuse address option
	{
		int optv = 1, er;
		er = setsockopt(skt, SOL_SOCKET, SO_REUSEADDR, &optv, sizeof(optv));
		if (er != 0) {
			perror("SockOpt Err:");
			exit(-1);
		}
	}
 
	memset(&my_addr, 0, sizeof(struct sockaddr_in)); // clear structure
	my_addr.sin_family = AF_INET;
	my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	my_addr.sin_port = htons(port);
 
	if (bind(skt, (struct sockaddr *)&my_addr, sizeof(my_addr))!=0) {
		perror("BIND FAILED:");
		exit(-1);
	}
 
	if (listen(skt, 1) != 0) {
		perror("Listen failed:");
		exit(-1);
	}
 
	return skt;
}
 
class	LINBUFS {
public:
	char	m_iline[512], m_oline[512];
	char	m_buf[256];
	int	m_ilen, m_olen;
	bool	m_connected;
 
	LINBUFS(void) {
		m_ilen = 0; m_olen = 0; m_connected = false;
	}
};
 
bool	check_incoming(LINBUFS &lb, int ttyfd, int confd, int timeout) {
	struct	pollfd	p[2];
	int	pv, nfds;
 
	p[0].fd = ttyfd;
	p[0].events = POLLIN | POLLERR;
	if (confd >= 0) {
		p[1].fd = confd;
		p[1].events = POLLIN | POLLRDHUP | POLLERR;
		nfds = 2;
	} else nfds = 1;
 
	if ((pv=poll(p, nfds, timeout)) < 0) {
		perror("Poll Failed!  O/S Err:");
		exit(-1);
	}
	if (p[0].revents & POLLIN) {
		int nr = read(ttyfd, lb.m_buf, 256);
		if (nr > 0) {
			// printf("%d read from TTY\n", nr);
			if (confd >= 0) {
				int	nw;
				nw = write(confd, lb.m_buf, nr);
				if(nw != nr) {
					// This fails when the other end resets
					// the connection.  Thus, we'll just
					// kindly close the connection and skip
					// the assert that once was at the end.
					fprintf(stderr, "ERR: Could not write return string to buffer\n");
					perror("O/S Err:");
					close(confd);
					confd = -1;
					lb.m_connected = false;
					nfds = 1;
					// assert(nw == nr);
				}
			}
		} for(int i=0; i<nr; i++) {
			lb.m_iline[lb.m_ilen++] = lb.m_buf[i];
			if ((lb.m_iline[lb.m_ilen-1]=='\n')||(lb.m_iline[lb.m_ilen-1]=='\r')||(lb.m_ilen>=sizeof(lb.m_iline)-1)) {
				if (lb.m_ilen >= sizeof(lb.m_iline)-1)
					lb.m_iline[lb.m_ilen] = '\0';
				else
					lb.m_iline[lb.m_ilen-1] = '\0';
				if (lb.m_ilen > 1)
					printf("%c %s\n",
						(confd>=0)?'>':'#', lb.m_iline);
				lb.m_ilen = 0;
			}
		}
	} else if (p[0].revents)
		printf("UNKNOWN TTY EVENT: %d\n", p[0].revents);
 
	if((nfds>1)&&(p[1].revents & POLLIN)) {
		int nr = read(confd, lb.m_buf, 256);
		if (nr == 0) {
			lb.m_connected = false;
			if (lb.m_olen > 0) {
				lb.m_oline[lb.m_olen] = '\0';
				printf("< %s\n", lb.m_oline);
			} lb.m_olen = 0;
			// printf("Disconnect\n");
			close(confd);
		} else if (nr > 0) {
			// printf("%d read from SKT\n", nr);
			int nw = 0, ttlw=0;
 
			errno = 0;
			do {
				nw = write(ttyfd, &lb.m_buf[ttlw], nr-ttlw);
 
				if ((nw < 0)&&(errno == EAGAIN)) {
					nw = 0;
					usleep(10);
				} else if (nw < 0) {
					fprintf(stderr, "ERR: %4d\n", errno);
					perror("O/S Err: ");
					assert(nw > 0);
					break;
				}
				// if (nw != nr-ttlw)
					// printf("Only wrote %d\n", nw);
				ttlw += nw;
			} while(ttlw < nr);
		} for(int i=0; i<nr; i++) {
			lb.m_oline[lb.m_olen++] = lb.m_buf[i];
			assert(lb.m_buf[i] != '\0');
			if ((lb.m_oline[lb.m_olen-1]=='\n')||(lb.m_oline[lb.m_olen-1]=='\r')||(lb.m_olen >= sizeof(lb.m_oline)-1)) {
				if (lb.m_olen >= sizeof(lb.m_oline)-1)
					lb.m_oline[lb.m_olen] = '\0';
				else
					lb.m_oline[lb.m_olen-1] = '\0';
				if (lb.m_olen > 1)
					printf("< %s\n", lb.m_oline);
				lb.m_olen = 0;
			}
		}
	} else if ((nfds>1)&&(p[1].revents)) {
		printf("UNKNOWN SKT EVENT: %d\n", p[1].revents);
	}
 
	return (pv > 0);
}
 
int	myaccept(int skt, int timeout) {
	int	con = -1;
	struct	pollfd	p[1];
	int	pv;
 
	p[0].fd = skt;
	p[0].events = POLLIN | POLLERR;
	if ((pv=poll(p, 1, timeout)) < 0) {
		perror("Poll Failed!  O/S Err:");
		exit(-1);
	} if (p[0].revents & POLLIN) {
		con = accept(skt, 0, 0);
		if (con < 0) {
			perror("Accept failed!  O/S Err:");
			exit(-1);
		}
	} return con;
}
 
int	main(int argc, char **argv) {
	// First, accept a network connection
#ifndef	LOW_SPEED
	int	skt = setup_listener(FPGAPORT);
#else
	int	skt = setup_listener(FPGAPORT+1);
#endif
	int	tty;
	bool	done = false;
 
	signal(SIGSTOP, sigstop);
	signal(SIGBUS, sigbus);
	signal(SIGSEGV, sigsegv);
	signal(SIGPIPE, SIG_IGN);
	signal(SIGINT, sigint);
	signal(SIGHUP, sighup);
 
	if ((argc > 1)&&(NULL != strstr(argv[1], "/ttyUSB"))) {
		// printf("Opening %s\n", argv[1]);
		tty = open(argv[1], O_RDWR | O_NONBLOCK);
	} else if (argc == 1) {
		const	char *deftty = "/dev/ttyUSB2";
		// printf("Opening %s\n", deftty);
		tty = open(deftty, O_RDWR | O_NONBLOCK);
	} else {
		printf("Unknown argument: %s\n", argv[1]);
		exit(-2);
	}
 
	if (tty < 0) {
		printf("Could not open tty\n");
		perror("O/S Err:");
		exit(-1);
	} else if (isatty(tty)) {
		struct	termios	tb;
 
		printf("Setting up TTY\n");
		if (tcgetattr(tty, &tb) < 0) {
			printf("Could not get TTY attributes\n");
			perror("O/S Err:");
			exit(-2);
		}
#ifndef	LOW_SPEED
		// Set 8 bits, 4MBaud, no parity, 1 stop bit
		// const char	set_highspeed[] = "00000600000PG00006";
		// Set 7 bits, 4MBaud, no parity, 1 stop bit
		if (false) {
		const char	set_highspeed[] = "0000060G000P";
		const char	read_qspic[] = "G0000D";
		const char	newline[] = "\n";
		::write(tty, newline, sizeof(newline));
		::write(tty, read_qspic, sizeof(read_qspic));
		::write(tty, set_highspeed, sizeof(set_highspeed));
		::write(tty, newline, sizeof(newline));
		printf("< "); fflush(stdout);
		::write(STDOUT_FILENO, read_qspic, sizeof(read_qspic));
		::write(STDOUT_FILENO, set_highspeed, sizeof(set_highspeed));
		::write(STDOUT_FILENO, newline, sizeof(newline));
		printf("\n"); usleep(400);
		tcdrain(tty);
		}
#endif
 
		cfmakeraw(&tb); // Sets no parity, 8 bits, one stop bit
		tb.c_cflag &= (~(CRTSCTS)); // Sets no parity, 8 bit
		tb.c_cflag &= (~(CSTOPB)); // One stop bit
// #define	LOW_SPEED
#ifndef	LOW_SPEED
		// Switch to 7 bit
		tb.c_cflag &= ~(CSIZE);
		tb.c_cflag |= CS7;
		// And 4 MBaud
		cfsetispeed(&tb, B1000000);
		cfsetospeed(&tb, B1000000);
#else
		// Set the speed to 115200 baud
		cfsetispeed(&tb, B115200);
		cfsetospeed(&tb, B115200);
#endif
		if (tcsetattr(tty, TCSANOW, &tb) < 0) {
			printf("Could not set any TTY attributes\n");
			perror("O/S Err:");
		}
		tcflow(tty, TCOON);
	}
 
	LINBUFS	lb;
	while(!done) {
		int	con;
 
		// Accept a connection before going on
		// Let's call poll(), so we can still read any
		// tty messages even when not accepted
		con = myaccept(skt, 50);
		if (con >= 0) {
			lb.m_connected = true;
 
			/*
			// Set our new socket as non-blocking
			int flags = fcntl(fd, F_GETFL, 0);
			flags |= O_NONBLOCK;
			fcntl(fd, F_SETFL, flags);
			*/
 
			// printf("Received a new connection\n");
		}
 
		// Flush any buffer within the TTY
		while(check_incoming(lb, tty, -1, 0))
			;
 
		// Now, process that connection until it's gone
		while(lb.m_connected) {
			check_incoming(lb, tty, con, -1);
		}
	}
 
	printf("Closing our socket\n");
	close(skt);
}