多线程文件传输服务端代码

//server.c
#include<stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <pthread.h>
#include<semaphore.h>

// 定义包的大小
#define PACK_SIZE 2048*2048

//定义线程数目
#define THREAD_NUM 50
pthread_t a_thread[THREAD_NUM];

void* WorkThread(void* args);

struct TrdItem
{
 int sockfd;
 struct sockaddr_in addr;
};

int main(int argc, char *argv[])
{
 //设置缓冲输出
 setvbuf(stdout, NULL, _IONBF, 0);
 fflush(stdout);

 int sockfd,new_fd;
 struct sockaddr_in server_addr;
 struct sockaddr_in client_addr;
 int sin_size,portnumber;
 char hello[]="Hello! Are You Fine?\n";
 
 //服务器端口号
 portnumber = atoi("8080");

 /* 服务器端开始建立socket描述符 */
 if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
 {
  printf("socket error!\n");
  exit(1);
 }

 /* 服务器端填充 sockaddr结构  */
 bzero(&server_addr,sizeof(struct sockaddr_in));
 server_addr.sin_family=AF_INET;
 server_addr.sin_addr.s_addr=htonl(INADDR_ANY);
 server_addr.sin_port=htons(portnumber);

 /* 捆绑sockfd描述符  */
 if(bind(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)
 {
  printf("blind error!\n");
  exit(1);
 }

 /* 监听sockfd描述符  */
 if(listen(sockfd,5)==-1)
 {
  printf("listen error!\n");
  exit(1);
 }
 printf("server is listening\n");

 struct TrdItem item;
 item.sockfd = sockfd;
 item.addr = client_addr;

 int i = 0;
 for(i=0; i<THREAD_NUM; ++i)
 {
  int err = 0;
  err = pthread_create(&a_thread[i], NULL, WorkThread, (void*)&item);
  if(err != 0)
   printf("Cannot create thread ... %s\n", strerror(err));
 }

 for(i=0; i<THREAD_NUM; ++i)
 {
  int err = 0;
  err = pthread_join(a_thread[i] , NULL);
  if(err != 0)
   printf("the thread id : %d ends failes, %s\n", i, strerror(err));
 }
 sleep(5);
 close(sockfd);
 exit(0);
}

void* WorkThread(void* args)
{
 while(1)
 {
  int sin_size = 0;
  int new_fd = 0;
  struct TrdItem t_item = *(struct TrdItem*)args;
  sin_size=sizeof(struct sockaddr_in);
  if( ( new_fd = accept(t_item.sockfd,(struct sockaddr *)(&t_item.addr),(socklen_t*)&sin_size ) ) == -1)
  {
   printf("accept error\n");
   exit(1);
  }
  printf("connection success\n");

  char hello[]="prepare for transmittint\n";
  if(write(new_fd,hello,strlen(hello))==-1)
  {
   printf("write error\n");
   exit(1);
  }

  long int read_size = 0;
  unsigned long file_len  = 0;
  int order_id  = 0;
  char file_name[128] = {'\0'};
  char file_info[1024] = {'\0'};

  // 读取指令
  printf("\n\nWaiting for read file info!\n");
  int nn = 0;
  if(nn = read(new_fd, file_info, 1024))
  {
   // 指令ID
   int id_h = (int)file_info[0]<<8;
   order_id = id_h + (int)file_info[1];

   // 文件长度
   // 高16位
   unsigned long len_hig_1 = 0;
   memcpy(&len_hig_1, &file_info[2], sizeof(file_info[2]));

   unsigned long len_hig_2 = 0;
   memcpy(&len_hig_2, &file_info[3], sizeof(file_info[3]));

   unsigned long len_hig = len_hig_1 * 256 + len_hig_2;

   // 低16位
   unsigned long len_low_1 = 0;
   memcpy(&len_low_1, &file_info[4], sizeof(file_info[4]));

   unsigned long len_low_2 = 0;
   memcpy(&len_low_2, &file_info[5], sizeof(file_info[5]));

   int len_low = len_low_1 * 256 + len_low_2;
   file_len = len_hig * 256 * 256 + len_low;

   // 文件名称
   strncpy(file_name, &file_info[6], strlen(&file_info[6]));

   printf("order = %d, %lu, %s\n", order_id, file_len, file_name);
   if((strlen(file_name) == 0) || (file_len == 0))
   {
    printf("read file info error\n");
    close(new_fd);
   }
  }
  else
  {
   printf("read file info error\n");
   close(new_fd);
   close(t_item.sockfd);
   exit(0);
  }

  // 写入文件
  printf("\n\nWaiting for read file content!\n");
  FILE* pf = fopen(file_name, "wb+");
  if(pf == NULL)
  {
   printf("open file error\n");
   close(new_fd);
  }
  char buff[PACK_SIZE] = {'\0'};
  while(read_size <= file_len)
  {
   int rlen = read(new_fd, buff, PACK_SIZE);
   if(rlen)
   {
    int wn = fwrite(buff, sizeof(char), rlen, pf);
    read_size += rlen;
   }
   else
   {
    printf("read over\n");
    break;
   }
  }
  printf("File Name = %s, File len = %ld , Already read size = %ld\n", file_name, file_len, read_size);

  /* 这个通讯已经结束     */
  fclose(pf);
  close(new_fd);
 }
 return (void*)NULL;
}

转载于:https://my.oschina.net/yuansusu/blog/370368