tcp客户端接收文件

/***client.c***/
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <netdb.h>
#define PORT    60000
#define BUFFER_SIZE 1024
 

#define FILE_NAME_MAX_SIZE 512

 
int main(int argc, char *argv[])
{
    int sockfd,sendbytes;
    char buf[BUFFER_SIZE];
    struct hostent *host;
    struct sockaddr_in serv_addr;
    int port;
    
    if(argc < 2)
    {
        fprintf(stderr,"Usage: ./client Hostname(or ip address) \n");
        exit(1);
    }
    /*地址解析函数*/
    if ((host = gethostbyname(argv[1])) == NULL)
    {
        perror("Server IP Address Error. \n");
        exit(1);
    }

    if((port = atoi(argv[2]))==NULL)
    {
    	perror("port Error. \n");
        exit(1);
    }
    printf("port is %d\n",port);
    
    memset(buf, 0, sizeof(buf));  //将buf全部填充0
//    sprintf(buf, "%s", argv[2]);  //将输入的第2个参数写到buf中
    /*创建socket*/
    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
    {
        perror("Create Socket Failed! \n");//创建Socket失败
        exit(1);
    }
    
    /*设置sockaddr_in 结构体中相关参数*/
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(port);
    serv_addr.sin_addr = *((struct in_addr *)host->h_addr);
    bzero(&(serv_addr.sin_zero), 8);
    /*调用connect函数主动发起对服务器端的连接*/
    if(connect(sockfd,(struct sockaddr *)&serv_addr,sizeof(struct sockaddr))== -1)
    {
        perror("Connect To Server IP Failed! \n"); //连接服务器失败
        exit(1);
    }
 

    printf("Connected! \n");
    char file_name[FILE_NAME_MAX_SIZE+1]; 
    bzero(file_name, FILE_NAME_MAX_SIZE+1);
    printf("Please Input File Name On Server:\t");
    scanf("%s",file_name);
    char buffer[BUFFER_SIZE];
    bzero(buffer,BUFFER_SIZE);
 
    //复制文件名到buffer中
    strncpy(buffer,file_name,strlen(file_name)>BUFFER_SIZE?BUFFER_SIZE:strlen(file_name));
 
    if(send(sockfd,buffer,BUFFER_SIZE,0)<0) //向服务器发送buffer中的数据，即发送文件名给服务器
    {
        perror("Send File Name Failed! \n");  //发送文件名给服务器失败
        exit(1);
    }
 
    FILE *fp=fopen(file_name,"w"); //打开文件，准备写入
    if(NULL==fp)
    {
        printf("File %s Can Not Open To Write! \n",file_name);//新建或打开文件进行写操作失败
        exit(1);
    }
 
    bzero(buffer,BUFFER_SIZE);
    int length=0;
    while((length=recv(sockfd,buffer,BUFFER_SIZE,0)) >0)   //将从服务器接收数据缓存到buffer中
    {        //每接收一段数据，便将其写入文件中，循环直到文件接收并写完为止
        printf("the receive length is %d\n",length);
	if(fwrite(buffer,sizeof(char),length,fp)<length) 
        {    //如果实际写入的字节数少于服务器传来的字节数，则写入错误
            printf("File %s Write Failed! \n",file_name);//文件写入失败
            break;
        }
        bzero(buffer,BUFFER_SIZE);   //将buffe清零
    }
    printf("Receive File: %s  \n",file_name);  //从服务器得到文件:%s
    fclose(fp);   //关闭文件
 
    system("bash ./send.sh boot1.bin");
 
/***************************************************
    //发送消息给服务器端
    if ((sendbytes = send(sockfd, buf, strlen(buf), 0)) == -1)
    {
        perror("send");
        exit(1);
    }
****************************************************/
    close(sockfd);    //关闭Socket
    exit(0);          //正常退出
}