OpenCores
URL https://opencores.org/ocsvn/openarty/openarty/trunk

Subversion Repositories openarty

[/] [openarty/] [trunk/] [sw/] [host/] [netuart.cpp] - Diff between revs 4 and 30

Go to most recent revision | Only display areas with differences | Details | Blame | View Log

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

powered by: WebSVN 2.1.0

© copyright 1999-2024 OpenCores.org, equivalent to Oliscience, all rights reserved. OpenCores®, registered trademark.