TCP实现文件传输的代码实现与启发

[root@localhost root]# cat /etc/hosts
# Do not remove the following line, or various programs
# that require network functionality will fail.
127.0.0.1               localhost.localdomain localhost

192.168.1.21    localhost.localdomain Server

/**********************FileServer.c******************************/
//Author: May be duanjigang, ^_^
#include<stdio.h>
#include<sys/types.h>
#include<sys/fcntl.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netdb.h>
#define command "clear"
//这里其实应改写成ifdef _WIN32 define command "cls" else define "clear"，更合理些
#include<errno.h>
#define  big_size  2048
#define small_size  50
int main(void)
{
        char c= 0,
        buf[big_size],//读文件的缓冲
        file[small_size],//文件名
  host_name[small_size];//主机名
        int fromlen = 0,source = 0;
        register int k,s,ns;
        struct sockaddr_in sin;
        struct hostent * hp;
        system(command);//清屏
        printf("输入要传输的文件名:");
        memset(file, 0, small_size);
        scanf("%s",file);
        if ((source = open(file, O_RDONLY) ) < 0)
        {
                perror("源文件打开出错");
                exit(1);
        }
        memset(host_name, 0, small_size);
  //写入server的名子，然后gethostbyname将会根据这个名子去找主机的信息
  sprintf(host_name, "%s", "Server");
        if((hp = gethostbyname(host_name)) == NULL)
        {
                   perror("返回主机地址信息错!!!");
                   exit(2);
        }
        if( (s = socket(AF_INET, SOCK_STREAM, 0)) < 0)
        {
                   printf("获取SOCKET号失败!!!");
                   exit(3);
        }
        sin.sin_family=AF_INET;
        sin.sin_port=htons(1500);
        memcpy(&sin.sin_addr, hp->h_addr, hp->h_length);
        if(bind(s, (struct sockaddr*)&sin, sizeof(sin)) < 0 )
        {
                   printf("不能将服务器地址捆绑到SOCKET号上!!!\n");
                   close(s);
                   exit(4);
        }
         if(listen(s,5) < 0)
         {
                   fprintf(stderr, "sever:listen\n");
                   exit(5);
         }
          while(1)
          {
                   printf("等待传送文件中.......\n");
                   if((ns = accept(s, NULL, NULL)) < 0 )
                   {
                           fprintf(stderr, "sever:accept\n");
                           exit(6);
                   }
                   lseek(source,0L,0);
                   printf("开始传送文件.....\n");
        //先把文件名发过求
                   write(ns,file,sizeof(file));
                   printf("%s\n", file);
                   //发送文件名
                   memset(buf, 0, big_size);
                   while( (k = read(source, buf, sizeof(buf))) > 0)
                   {
                             //循环发送直到，文件结束
                             write(ns,buf,k);
                       memset(buf, 0, big_size);
                   }
                   printf("\n传输完毕\n");
                   close(ns);
           }
           close(source);
           exit(0);
}  
[code]
客户端，即要下载文件的机子端程序
[code]
/*****************************FileClient.c********************************/
#include<stdio.h>
#include<sys/types.h>

#include<sys/fcntl.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netdb.h>
#define command "clear"

#include<errno.h>
#include <string.h>

#define big_size 1024
#define small_size 50

int main(void)
{
        char buf[big_size] ={0},file[small_size]={0},
         host_name[small_size]={0};
        int target,ret;
        register int bytes,sockfd;
        struct sockaddr_in sin ;
        struct hostent *hp;
        system(command);
               sprintf(host_name, "%s", "Server");
        if((hp = gethostbyname(host_name)) == NULL)
        {
                fprintf(stderr, "返回服务器地址信息错!!!\n");
                exit(1);
        }
        if((sockfd = socket(AF_INET, SOCK_STREAM, 0 )) < 0)
        {
                fprintf(stderr, "获取SOCKET号失败!!!\n");
                exit(2);
        }
       
        memset((char*)&sin, '\0', sizeof(sin));
       
        sin.sin_family = AF_INET;
        sin.sin_port = htons(1500);
  
        sin.sin_addr = *(struct in_addr*)hp->h_addr;
        if((ret = connect(sockfd, (struct sockaddr*)&sin, sizeof(sin))) == -1)
        {
                fprintf(stderr, "不能与服务器连接!!!\n");
                exit(3);
        }
        memset(buf, 0, big_size);
        while(read(sockfd, file, sizeof(file)) < 0);       
        sprintf(buf, "%s_%s", "receive", file);

        if((target=open(buf, O_WRONLY|O_CREAT|O_TRUNC,0644)) < 0)
        {
                perror("不能打开目标文件!!");
                exit(4);
        }
       
        memset(buf, 0, big_size);
        while((bytes = read(sockfd, buf, sizeof(buf))) > 0)
        {
                write(target,buf,bytes);
                memset(buf, 0, big_size);       
        }
        printf("接收文件成功!!!\n");
        close(sockfd);
        close(target);
}