Subversion Repositories f9daq

/*
    C socket server example, handles multiple clients using threads */
 
#include<stdio.h>
#include<string.h> //strlen
#include<stdlib.h> //strlen
#include<sys/socket.h>
#include<arpa/inet.h> //inet_addr
#include<unistd.h>
#//write
#include<pthread.h>
#include "daq.h"
#define PORT 9930
//the thread function



void *connection_handler(void *);
 
int is_connected = 0;
int ncalls=0;
extern int ctrl_c;

int writesock(int sock, int id , char *msg, int len ){
    int hdr[2];
    hdr[0]= id;
    hdr[1]= len + 8;
    write(sock , hdr , 8);
    return write(sock , msg , len);
}


int main(int argc , char *argv[]) {
    int socket_desc , client_sock , c , *new_sock=NULL;
    struct sockaddr_in server , client;

    if (argc >1) return daq_main(argc, argv);
     
    //Create socket
    socket_desc = socket(AF_INET , SOCK_STREAM , 0);
    if (socket_desc == -1)
    {
        printf("Could not create socket");
    }
    puts("Socket created");
     
    //Prepare the sockaddr_in structure
    server.sin_family = AF_INET;
    server.sin_addr.s_addr = INADDR_ANY;
    server.sin_port = htons( PORT );
     
    //Bind
    if( bind(socket_desc,(struct sockaddr *)&server , sizeof(server)) < 0)
    {
        //print the error message
        perror("bind failed. Error");
        return 1;
    }
    puts("bind done");
     
    //Listen
    listen(socket_desc , 3);
     
    //Accept and incoming connection
    puts("Waiting for incoming connections...");
    c = sizeof(struct sockaddr_in);
     
     
    //Accept and incoming connection
    puts("Waiting for incoming connections...");
    c = sizeof(struct sockaddr_in);
    while( (client_sock = accept(socket_desc, (struct sockaddr *)&client, (socklen_t*)&c)) )
    {
        puts("Connection accepted");
         
        pthread_t sniffer_thread;

        if (is_connected){
            char message[0xFF]="Only one client can connect. Disconnect it first!";
            perror(message);
            writesock(client_sock , 0, message , strlen(message));
            close (client_sock);
            continue;
        }
        new_sock = (int *) malloc(1);
        *new_sock = client_sock;
        if( pthread_create( &sniffer_thread , NULL , connection_handler , (void*) new_sock) < 0)
        {
            perror("could not create thread");
            return 1;
        }

         
        //Now join the thread , so that we dont terminate before the thread
        //pthread_join( sniffer_thread , NULL);
       
        puts("Handler assigned");
       
    }
     
    if (client_sock < 0)
    {
        perror("accept failed");
        return 1;
    }
     
    return 0;
}
 


void *daq_handler(void *rdata){
 
 pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
 //Get the socket descriptor
 int sock = *(int*)rdata;
 int *idata = (int *)rdata;
 
 char * settings = (char *) (idata+2);
 daq_init(settings);
 settings +=12;
 int bufsize = 0;  
 char * data = NULL;

 //for(int i =0;i<5;i++) printf("%d %d\n",i, idata[i]);
 while (1){
    int nb = daq_run ( settings, &data, &bufsize );
    int error = 0;
    socklen_t len = sizeof (error);
    int retval = getsockopt (sock, SOL_SOCKET, SO_ERROR, &error, &len);
    if (retval!=0 || error!=0) break;
    writesock(sock , 1, data , nb );
    if (ctrl_c) break;
    //break;
 }
 if (data!=NULL)  free(data);
 fprintf(stderr, "Exiting thread %d\n", sock ) ;
 return NULL;
}

/*
 * This will handle connection for each client
 * */

void *connection_handler(void *socket_desc) {
    //Get the socket descriptor
    int sock = *(int*)socket_desc;
    int read_size;
    char *message , client_message[2000];
    is_connected = 1;
    //Send some messages to the client
    message = "Info from connection handler\n";
    writesock(sock , 0, message , strlen(message));
         
    //Receive a message from client
    pthread_t daq_thread = 0;

    while( (read_size = recv(sock , client_message , 2000 , 0)) > 0 )
    {
        //Send the message back to client
        int * hdr = (int *) client_message;
        printf("Received %d bytes RECID = %d LEN %d\n",read_size, hdr[0], hdr[1]);
        switch (hdr[0]){
           case 0:
             ctrl_c = 0;
             hdr[0]= sock;
             if( pthread_create( &daq_thread , NULL , daq_handler , (void*) &client_message) < 0)
             {
               perror("could not create daq thread");
              }
              break;
           case 1:
             ctrl_c = 1;
             sleep(1);
             if (daq_thread) {
                pthread_cancel(daq_thread);
                daq_thread = 0;
             }

             break;
           default: break;

        }
    }
     
    if(read_size == 0)
    {
        puts("Client disconnected");
        fflush(stdout);
    }
    else if(read_size == -1)
    {
        perror("recv failed");
       
    }
    if (daq_thread) {
      pthread_cancel(daq_thread);
      daq_thread = 0;
    }

         
    //Free the socket pointer
    free(socket_desc);
    socket_desc = 0;
    is_connected = 0;
     
    return 0;
}