UDP数据接收服务器

简介

这是我在做一个要用UDP方式进行数据传输时，自己写的一个多线程的UDP数据接收服务器，


它能将接收到的UDP数据包存成文件，并提供数据包接收时间监测；


还支持键盘命令响应，以将数据写到新的文件，和退出程序；


闲言少述，直接上代码;

代码：


/*


 ******************************************************************************


 * \File


 *   udp_server.c


 * \Descript


 *   Receive udp datas and write into file, implement by multi-threads


 * \Author


 *   Anderson Yan


 * \Create


 *   2013-9-18


 ******************************************************************************


 */

#include <stdio.h>


#include <string.h>


#include <stdlib.h>


#include <sys/socket.h>


#include <netinet/in.h>


#include <pthread.h>


#include <sys/time.h>


#include <unistd.h>

#include <getopt.h>


/* For signal response */


#include <signal.h>


#include <limits.h>


#include <unistd.h>


#include <fcntl.h>


#include <sys/ioctl.h>


#include <sys/time.h>


#include <termios.h>

static struct termios oldtty;


static int q_pressed = 0;


static int verbose = 0;


static int using_stdin = 0;


static int run_as_daemon = 0;

static volatile int received_sigterm = 0;

static void term_init(void);


static void term_exit(void);


static void sigterm_handler(int sig);


static int read_key(void);


/* end */

/* For parsing command-line parameters */


extern char* optarg;


extern int   opterr;


struct option opts[] = 


{


  {"port", required_argument, NULL, 'p'},


  {"out",  required_argument, NULL, 'o'},


  {"help", required_argument, NULL, 'h'},


  {0,0,0,0}


};

int parse_params(int agrc, char** argv, int* port, char* f);


/* end */

#define SERVPORT 10000


#define MAXDATASIZE 2048


FILE* fp_udp = NULL;


void *sock_recv_udp(void *arg);

static int bTerminalFlag = 0;

int main(int argc, char** argv)


{


  int file_number = 0;


  int  udp_port = SERVPORT;


  char filename[128] = "test.dat";


  char filename_new[256] = {0};

int res;


  pthread_t tid_sock;


  void *thread_result;

/* 命令行输入参数解析 */


  if (argc < 2)


  {


    printf("Usage: \n");


    printf("-p port : set listern port\n");


    printf("-o file : set output filename\n");


    printf("-h      : print help infomation\n");


    printf("eg :\n");


    printf("$./udp_server -p 10001 -o test.dat\n"); 


    exit(1);


  }


  parse_params(argc, argv, &udp_port, filename);

/* SIGNAL，键盘命令响应的初始化 */


  int key;


  if (!using_stdin)


  {


    if (verbose >=0)


      printf("Press [q] to stop, [?] for help, [n] for creating new file\n");


  }


  term_init();


  /* end */

fp_udp = fopen(filename, "wb");


  if (fp_udp == NULL)


  {


    perror("Open file failed");


    exit(EXIT_FAILURE);


  }

res = pthread_create(&tid_sock, NULL, sock_recv_udp, (void*)&udp_port);


  if (res != 0)


  {


    perror("Thread creation failed");


    exit(EXIT_FAILURE);


  }

/* Response for keyboard */


  for(; received_sigterm == 0;)


  {


    if (verbose != file_number)


    {


      file_number = verbose;


      bTerminalFlag = 0;


      snprintf(filename_new, sizeof(filename_new),"%s.%d",filename, file_number);


      printf("Create new file : %s\n", filename_new);

fp_udp = fopen(filename_new, "wb");


      if (fp_udp == NULL)


      {


        perror("Open file failed");


        exit(EXIT_FAILURE);


      }

res = pthread_create(&tid_sock, NULL, sock_recv_udp, (void*)&udp_port);


      if (res != 0)


      {


        perror("Thread creation failed");


        exit(EXIT_FAILURE);


      }


    }


    /* if 'q' pressed, exits */


    if (!using_stdin)


    {


      if (q_pressed)


        break;

/* read_key() returns 0 on EOF */


      key = read_key();


      if (key == 'q')


      {


        printf("quit\n");


        bTerminalFlag = 1;


        break;


      }


      if (key == 'n')


      {


        verbose++;


        printf("verbose = %d\n", verbose);

bTerminalFlag = 1;


        printf("Waiting for thread to finish...\n");


        res = pthread_join(tid_sock, &thread_result);


        if (res != 0)


        {


          perror("Thread join failed");


          exit(EXIT_FAILURE);


        }


        printf("Thread joined, it returned %s\n", (char*)thread_result);

memset(filename_new, 0, sizeof(filename_new));


        fflush(fp_udp);


        fclose(fp_udp);


      }


      if (key == '?')


      {


        printf("key function\n"


               "? show this help\n"


               "n create new file\n"


               "q quit\n");


      }


    }


  }


  /* end */

printf("Waiting for thread to finish...\n");


  res = pthread_join(tid_sock, &thread_result);


  if (res != 0)


  {


    perror("Thread join failed");


    exit(EXIT_FAILURE);


  }


  printf("Thread joined, it returned %s\n", (char*)thread_result);

fflush(fp_udp);


  fclose(fp_udp);

return 0;


}

void *sock_recv_udp(void *arg)


{


  int port = *(int*)arg;

int sockfd, recvbytes;


  char rcv_buf[MAXDATASIZE], snd_buf[MAXDATASIZE];


  struct sockaddr_in server_addr;


  socklen_t addr_size=sizeof(server_addr);


  int pkt_cnt = 0;

struct timeval start_time, end_time;


  double time_used;

int optval;


  printf("PORT : %d\n", port);

if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)


  {


    perror("socket:");


    exit(1);


  }

server_addr.sin_family = AF_INET;


  server_addr.sin_port = htons(port);


  server_addr.sin_addr.s_addr = htonl(INADDR_ANY);


  memset(&(server_addr.sin_zero), 0, 8);

if (bind(sockfd, (struct sockaddr *)&server_addr, sizeof(struct sockaddr))==-1)


  {


    perror("bind:");


    exit(1);


  }

optval = 1;


  int ret = setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));

while(!bTerminalFlag)


  {


    addr_size = sizeof(struct sockaddr);


    gettimeofday(&start_time, 0);

/* receive the data from "server_addr" by "sockfd" and store into "rcv_buf" */


    if ((recvbytes = recvfrom(sockfd, rcv_buf, MAXDATASIZE, 0,


(struct sockaddr*)&server_addr, &addr_size)) == -1)


    {


      perror("recv:");


      exit(1);


    }

fwrite(rcv_buf, 1, recvbytes, fp_udp);


    fflush(fp_udp);

gettimeofday(&end_time, 0);


    time_used = 1000000 * (end_time.tv_sec - start_time.tv_sec)


                + end_time.tv_usec - start_time.tv_usec;


    time_used /= 1000;


    printf("recv:%s:%d : pkt_cnt = %6d :  %4d bytes : time(ms) = %f\n", 


  inet_ntoa(server_addr.sin_addr), ntohs(server_addr.sin_port), 


  pkt_cnt, recvbytes, time_used);

pkt_cnt++;


  }


  close(sockfd);

pthread_exit("thread sock_recv_udp over\n");


}

完整的代码可以在这里下载:


http://download.csdn.net/detail/fireroll/6286649