用socket实现简单的文件传输

http://atu82.bokee.com/4667560.html

服务器端：
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#define SERVPORT 6666
#define MAXDATASIZE 1024 /*每次最大数据传输量 */
#define FILETOWRITE "/home/xufeng/xf_write"
#define BACKLOG 10

int main()
{
 FILE *fpw;
 int sockfd,client_fd;
 int recvbytes;
 struct sockaddr_in my_addr; /* 本机地址信息 */
 struct sockaddr_in remote_addr; /* 客户端地址信息 */
 unsigned char buf[MAXDATASIZE];
 char control[4]="YES";
 memset(buf, '\0', sizeof(buf));
 

 if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
 { 
  perror("socket initialization Error.");
  exit(1);
 }
 
 my_addr.sin_family=AF_INET;   
 my_addr.sin_port=htons(SERVPORT); 
 my_addr.sin_addr.s_addr = INADDR_ANY; 
 bzero(&(my_addr.sin_zero),8); 

 if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof(struct sockaddr)) == -1)
 {
  perror("bind Error");
  exit(1);
 }
 
 if (listen(sockfd, BACKLOG) == -1)
 {
  perror("listen Error!");
  exit(1);
 }
 
 while(1)
 {
  int sin_size = sizeof(struct sockaddr_in);
  if ((client_fd = accept(sockfd, (struct sockaddr *)&remote_addr, &sin_size)) == -1)
  {
   perror("accept Error");
   continue;
  }
   
  printf("received a connection from %s\n", inet_ntoa(remote_addr.sin_addr)); //inet_ntoa（将网络二进制的数字转换成网络地址）
  
  if ((fpw=fopen(FILETOWRITE, "ab+")) == NULL)
  {
   printf("File to Write Open error.\n");
   exit(1);
  }
  if (send(client_fd, control,strlen(control), 0) == -1)
  {
   perror("send Error");
   /* 这个通讯已经结束     */
   exit(0);
  }
  printf("Send YES: clear to receive data\n");
  while(1)
  {
   if ((recvbytes=recv(client_fd, buf,MAXDATASIZE, 0)) ==-1)
   {
    perror("recv Error!");
    exit(1);
   }
   buf[recvbytes] = '\0';
   //printf("Received:\n%s\nlength: %d\n",buf, strlen(buf));

   // open file to write
   //printf("begin writing...\n");
   fwrite(buf, 1, recvbytes, fpw);
   memset(buf, '\0', sizeof(buf));
   if(recvbytes < 1000)
    break;
   if (send(client_fd, control,strlen(control), 0) == -1)
   {
    perror("send Error");
    /* 这个通讯已经结束     */
    exit(0);
   }
   printf("Send YES: clear to receive data\n");
  }
  //printf("end writing\n");
  /* 循环下一个     */  
  close(client_fd);
  fclose(fpw);
 }
 
}

客户端：
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>

#define SERVPORT 6666
#define MAXDATASIZE 1024 /*每次最大数据传输量 */
#define FILETOREAD "/home/xufeng/xf_read"

int main()
{
 FILE *fpr;
 int i=0;
 unsigned char buf[MAXDATASIZE];
 int readlength;
 int sockfd;
 struct sockaddr_in serv_addr;
 int recvbytes;

 memset(buf, '\0', sizeof(buf));
 if ((fpr=fopen(FILETOREAD, "rb")) == NULL)
 {
  printf("File to Read Open error.\n");
  exit(1);
 } 
 
 if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
 {
  perror("socket initialization failure.\n");
  exit(1);
 }
 
 memset(&serv_addr, 0, sizeof(serv_addr));
 serv_addr.sin_family=AF_INET;
 serv_addr.sin_port=htons(SERVPORT);
 serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
 bzero(&(serv_addr.sin_zero),8);

 if (connect(sockfd, (struct sockaddr *)&serv_addr, sizeof(struct sockaddr)) == -1)
 {
  perror("connect Error!");
  exit(1);
 }

 while(1)
 {
  if ((recvbytes=recv(sockfd, buf, 3, 0)) ==-1)
  {
   perror("recv Error!");
   exit(1);
  }
  buf[recvbytes]='\0';
  printf("receive control signal: %s\n", buf);
  if(strcmp(buf, "YES")==0)
  {
   readlength = fread(buf, 1, 1000, fpr); 
   buf[readlength]='\0';
   //printf("send:\n%s\nlength: %d\n", buf, strlen(buf));
   if(readlength < 1000)
    break;
   if (send(sockfd, buf, readlength, 0) == -1)
   {
    perror("send error\n");
    exit(1);
   }
   memset(buf, '\0', sizeof(buf));
  }
 }
 //fread(buf, 1, readlength, fpr);
 if (send(sockfd, buf, readlength, 0) == -1)
 {
  perror("send error\n");
  exit(1);
 }
 //printf("send:\n%s\nlength: %d\n", buf, strlen(buf));
 memset(buf, '\0', sizeof(buf));
 close(sockfd);
 fclose(fpr);
}

这是我在学习socket的时候编写的简单的文件上传程序,在本机上实现模拟文件传输.
原理是这样的:客户端打开一个文件,每次读1000个字节,然后通过socket传送到服务器,服务器接受字节流,创建一个新文件,再写文件.
写程序的过程中遇到一个问题:当传送小文件的时候没有什么事情,但是当传送较大的文件的时候,就会得到无法预料的结果,有可能造成socket pipe broken(socket管道破裂),推测原因,可能是由于文件读的速度远远快于文件写的速度.后来我加了一段简单的流量控制程序,类似于停止等待协议,即发送方发完数据后等待接受方传送过来的控制信令(本程序采用了一个字符串"YES")代表发送方可以继续发送字节流.这种方法可以解决不能传送大文件的问题,只是采用类似停止等待协议的思想本身效率可能不高.

不知道FTP协议采用的哪种流量控制方法?希望有人指点.