source: tools/FAD/simple_daq/simple_daq.cpp@ 254

/********************************************************************\

  Read 8 Channels from FAD-Board
  Write Commands to Socket 0, Commands must be (n * 16) Bit long
  
  Usage: SocketClient [IP-Address]
  
  kw, 05.10
  
  based on:
  Generic example for remote control via a socket interface

  Oliver Grimm, March 2009

\********************************************************************/
#ifndef PATH_MAX
#define PATH_MAX 1000
#endif
#include "simple_daq.h"

int SocketDescriptor[8];

int main(int argc, char *argv[])
{
        char Buffer[MAX_COM_SIZE];
        
        // default server
        char ServerName[IP_ADDR_LENGTH] = IP_ADDR_DEFAULT; // default IP-Address

        int i, read_return;

        int DAQPort = FIRST_DAQPORT;
        struct sockaddr_in SocketAddress[8];
        struct in_addr Serveripv4addr;
        fd_set ReadFileDescriptor;

        FILE* fhandle[8];
        char fname[PATH_MAX];

        int max_fd = 0;

        // Get IP-Address from command line
        if (argc > 1)
        {
                strncpy (ServerName, argv[1], IP_ADDR_LENGTH);
        }
        
        // Convert IP-Addr to binary
        if (inet_pton (AF_INET, ServerName, &Serveripv4addr) <= 0)
        {
                printf ("Error: network address not valid\n");
                exit_program (EXIT_FAILURE);
        }

        // Open sockets
        for (i = 0; i < 8; i++)
        {
                if ((SocketDescriptor[i] = socket (PF_INET, SOCK_STREAM, 0)) == -1)
                {
                        printf ("Error: Could not open socket Nr.: %d\n", i);
                        exit_program (EXIT_FAILURE);
                }
                else
                {
                        if (SocketDescriptor[i] > max_fd)
                        {
                                max_fd = SocketDescriptor[i];
                        }
                }
        }
        
        // Open files for output
        for (i = 0; i < 8; i++)
        {
                sprintf (fname, "socket-%d.dat", i);
                if ((fhandle[i] = fopen (fname, "w")) == NULL)
                {
                        printf ("Error: Could not open file %s\n", fname);
                        exit_program (EXIT_FAILURE);
                }
        }

        // Connect to server
        printf ("Trying to connect to %s...\n", ServerName);
        for (i = 0; i < 8; i++)
        {
                SocketAddress[i].sin_family = PF_INET;
                SocketAddress[i].sin_port = htons ((unsigned short) DAQPort + i);
                SocketAddress[i].sin_addr = Serveripv4addr;
                
                if (connect (SocketDescriptor[i], (struct sockaddr *) &SocketAddress[i], sizeof (SocketAddress[i])) == -1)
                {
                        printf ("Error: Could not connect to server %s (port %d)\n", ServerName, DAQPort + i);
                        exit_program (EXIT_FAILURE);
                }
                else
                {
                        printf ("Connected to %s:%d\n", ServerName, DAQPort + i);
                }
        }

        signal (SIGPIPE, SIG_IGN); // Do not kill process if writing to closed socket
        signal (SIGINT, int_handler); // Cleanup after SIGINT (CTRL-C)

        // Main loop
        while (true)
        {
                fflush (stdout);

                FD_ZERO (&ReadFileDescriptor);
                
                FD_SET(STDIN_FILENO, &ReadFileDescriptor);
                
                for (i = 0; i < 8; i++)
                {
                        FD_SET (SocketDescriptor[i], &ReadFileDescriptor);
                }

                // Wait for data from sockets
                if (select (((int) max_fd) + 1, &ReadFileDescriptor, NULL, NULL, NULL) == -1)
                {
                        perror ("Error with select()\n");
                        break;
                }
                
                memset (Buffer, 0, sizeof (Buffer));

                // Data from STDIN
                if (FD_ISSET (STDIN_FILENO, &ReadFileDescriptor))
                {
                        fgets (Buffer, MAX_COM_SIZE, stdin);
                        // Send command to socket 0
                        cmd_send (Buffer, SocketDescriptor[0]);                 
                }
                // Data from sockets
                else
                {
                        // Check all sockets
                        for (i = 0; i < 8; i++)
                        {
                                if (FD_ISSET (SocketDescriptor[i], &ReadFileDescriptor))
                                { 
                                        // Only for testing
                                        memset (Buffer, 0xAA, sizeof (Buffer));
                                        if ((read_return = read (SocketDescriptor[i], Buffer, MAX_COM_SIZE)) == 0)
                                        {
                                                printf ("Error: Server not existing anymore, exiting...\n");
                                                exit_program (EXIT_FAILURE);
                                        }
        
                                        if (read_return > 0)
                                        {
                                                printf ("Socket [%d]: Read %d Bytes\n", i, read_return);
                                        
                                                fwrite (Buffer, 1, (size_t) read_return, fhandle[i]);
                                                // Important!!!
                                                fflush (fhandle[i]);
                                        }
                                }
                        }
                }
        } // while (TRUE)

        exit (EXIT_SUCCESS);
}

// close sockets and exit
void exit_program (int exit_status)
{
        int i;
        
        printf ("\nClosing Sockets...");
        for (i = 0; i < 8; i++)
        {
                close (SocketDescriptor[i]);
        }
        printf (" done\n");
        
        exit (exit_status);
}

// SIGINT
void int_handler (int sig)
{
        exit_program (EXIT_SUCCESS);
}