URL
https://opencores.org/ocsvn/openarty/openarty/trunk
Subversion Repositories openarty
[/] [openarty/] [trunk/] [sw/] [host/] [netuart.cpp] - Rev 50
Go to most recent revision | Compare with Previous | Blame | View Log
#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); }
Go to most recent revision | Compare with Previous | Blame | View Log