Linux C语言下TCP传文件，并显示传输进度条

1.本文介绍将文件通过sever传到client端，并显示传输进度条


2.打开一个终端，执行./file_server在file_server所在目录下放好要传的文件tianitande.mp3可通过宏FILE_NAME来改变或者做成参数传进去


3.将file_client拷贝到file_server目录外的其他目录，执行file_client便会将tianitande.mp3文件从file_client所在目录下了。


服务器端：

/*
* =====================================================================================
*
* Filename: file_server.c
*
* Description:
*
* Version: 1.0
* Created: 04/05/2017 14:12:16 PM
* Revision: none
* Compiler: gcc file_server.c -o file_server
*
* Author: LI JUN LIANG
* Organization:
*
* =====================================================================================

*/

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h> //close()
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <assert.h>

#define SERVER_PORT 6667
#define MAX_CONNETCION_COUNT 20 //最大连接数
#define BUFFER_SIZE 1024
#define MAX_SIZE_OF_FILE_NAME 512
#define FILE_NAME "tiantiande.mp3"

struct TCP_STRUCT_DATA
{
int m_cmd;
int m_data_len;
};

enum TCP_CMD
{
GET_FILE = 0,
//...
};

long get_file_size(char *file_name)
{
assert(file_name != NULL);
if(access(file_name, F_OK))
{
printf("####L(%d) file:%s not exist!", __LINE__, file_name);
return -1;
}

FILE *fp = fopen(file_name, "rb");
if(NULL == fp )
{
printf("file:%s can not open!\n", file_name);
return -1;
}

long file_size = 0;
fseek(fp, 0, SEEK_END); //把文件内部指针移动到文件尾部
file_size = ftell(fp); //返回指针偏离文件头的位置(即文件中字符个数)
fseek(fp, 0, SEEK_SET); //把文件内部指针移回到文件头部

fclose(fp);

return file_size;
}

int main(int argc, char *argv[])
{
struct sockaddr_in server_addr;
bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = htons(INADDR_ANY);//允许任何IP连接server
//server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
server_addr.sin_port = htons(SERVER_PORT);

int server_socket = socket(PF_INET, SOCK_STREAM, 0); //创建TCP socket套接字
if( server_socket < 0)
{
printf("####L(%d) create socket failed!", __LINE__);
exit(1);
}

if( bind(server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr))) //绑定端口
{
printf("####L(%d) bind port : %d failed!", __LINE__, SERVER_PORT);
exit(1);
}

if ( listen(server_socket, MAX_CONNETCION_COUNT) )//监听
{
printf("####L(%d) server listen failed!", __LINE__);
exit(1);
}

while (1) //服务器端一直运行
{
struct sockaddr_in client_addr;
socklen_t length = sizeof(client_addr);

int new_server_socket = accept(server_socket, (struct sockaddr *)&client_addr, &length);
if ( new_server_socket < 0)
{
printf("####L(%d) server accept failed!\n", __LINE__);
break;
}

char data_tmp[512] = {0};//要发的数据部分内容
long file_size = get_file_size(FILE_NAME);
if(-1 == file_size)
{
printf("####L(%d) get_file_size err!\n", __LINE__);
break;
}

sprintf(data_tmp, "<file><name>%s</name><size>%ld</size></file>", FILE_NAME, file_size); //要发送的数据部分内容

struct TCP_STRUCT_DATA struct_data;//数据头
memset(&struct_data, 0x0, sizeof(struct_data));
struct_data.m_cmd = GET_FILE; //获取文件命令
struct_data.m_data_len = strlen(data_tmp); //数据部分真实的长度
//struct_data.m_data_len =sizeof(data_tmp);//512是错误的

/*关键部分*/
char buffer[BUFFER_SIZE];
bzero(buffer, BUFFER_SIZE);
int send_len = sizeof(struct_data);
memcpy(buffer, &struct_data, send_len);

//向服务器发送buffer中的数据
int len = send(new_server_socket, buffer, send_len, 0); //发数据头部分
if(len < 0)
{
printf("####L(%d) send err...\n", __LINE__);//发送失败
break;
}

usleep(100 * 1000); //要休眠一下 否则第二次发过去的数据可能来不及接收到
len = send(new_server_socket, data_tmp, struct_data.m_data_len, 0);
if(len < 0)
{
printf("####L(%d) send err...\n", __LINE__);//发送失败
break;
}

usleep(100 * 1000); //要休眠一下 否则第二次发过去的数据可能来不及接收到

FILE *fp = fopen(FILE_NAME, "rb");//以二进制方式打开文件
if(NULL == fp )
{
printf("####L(%d) file:%s not found!\n", __LINE__, FILE_NAME);
}
else
{
int file_block_length = 0;
bzero(buffer, BUFFER_SIZE);//缓冲区清0
while( (file_block_length = fread(buffer, sizeof(char), BUFFER_SIZE, fp)) > 0) //读文件
{
printf("####L(%d) file_block_length = %d\n", __LINE__, file_block_length);
if(send(new_server_socket, buffer, file_block_length, 0) < 0) //发送文件数据
{
printf("####L(%d) send file:%s failed\n", __LINE__, FILE_NAME);
break;
}
bzero(buffer, BUFFER_SIZE);
}
fclose(fp);
printf("####L(%d) file:%s transfer finished!\n", __LINE__, FILE_NAME);
}
//关闭与客户端的连接
close(new_server_socket);
}
//关闭监听用的socket
close(server_socket);
return 0;
}
//end flie_server.c

客户端：
/*
* =====================================================================================
*
* Filename: file_client.c
*
* Description:
*
* Version: 1.0
* Created: 04/05/2017 13:43:29 PM
* Revision: none
* Compiler: gcc file_client.c -o file_client
*
* Author: LI JUN LIANG
* Organization:
*
* =====================================================================================
*/
#include <netinet/in.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define SERVER_PORT 6667
#define BUFFER_SIZE 1024
#define MAX_SIZE_OF_FILE_NAME 512
#define FILE_NAME "tiantiande.mp3"
#define DEBUG_LOG 0

struct TCP_STRUCT_DATA
{
int m_cmd;
int m_data_len;
};

enum TCP_CMD
{
GET_FILE = 0,
//...
};

void parse_xml(char *back_str, const char *xml_str, const char *pre_str, const char *suf_str)
{
if(DEBUG_LOG)
printf("####L(%d) xml_str:%s pre_str:%s suf_str:%s \n", __LINE__, xml_str, pre_str, suf_str);
int offset = strlen(pre_str);
char *ptr_first = strstr(xml_str, pre_str);
char *ptr_end = strstr(xml_str, suf_str);

if((ptr_first != NULL) && (ptr_end != NULL))
{
ptr_first = ptr_first + offset;
strncpy(back_str, ptr_first, (ptr_end - ptr_first));
if(DEBUG_LOG)
printf("####L(%d) xml data back_str:%s\n", __LINE__, back_str);
}
}

int main(int argc, char *argv[])
{
if (argc != 2)
{
printf("####L(%d) usage: ./%s serverIpAddress\n", __LINE__, argv[0]);
exit(1);
}

char file_name[16] = {0};//文件名称
long file_size = 0;//文件总大小
long recv_sum = 0;//统计接收文件总大小
int progress = 50; //进度条的长度
int current = 0; //当前进度
int length, i;
struct sockaddr_in client_addr;
bzero(&client_addr, sizeof(client_addr));
client_addr.sin_family = AF_INET;
client_addr.sin_addr.s_addr = htons(INADDR_ANY);
client_addr.sin_port = htons(0); //0:自动分配一个空闲端口
int client_socket = socket(AF_INET, SOCK_STREAM, 0); //创建TCP socket套接字
if( client_socket < 0)
{
printf("####L(%d) create socket failed!\n", __LINE__);
exit(1);
}

if( bind(client_socket, (struct sockaddr *)&client_addr, sizeof(client_addr))) //绑定
{
printf("####L(%d) client bind port failed!\n", __LINE__);
exit(1);
}

struct sockaddr_in server_addr;
bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = AF_INET;
if(0 == inet_aton(argv[1], &server_addr.sin_addr)) //服务器的IP地址来自程序的参数
{
printf("####L(%d) Server IP Address Error!\n", __LINE__);
exit(1);
}
server_addr.sin_port = htons(SERVER_PORT);
socklen_t server_addr_length = sizeof(server_addr);

if(connect(client_socket, (struct sockaddr *)&server_addr, server_addr_length) < 0)
{
printf("####L(%d) Can Not Connect To %s!\n", __LINE__, argv[1]);
exit(1);
}

char buffer[BUFFER_SIZE];
bzero(buffer, BUFFER_SIZE);
length = recv(client_socket, buffer, BUFFER_SIZE, 0);
if (length < 0)
{
printf("####L(%d) server recieve data failed!\n", __LINE__);
exit(1);
}

/*关键代码*/
struct TCP_STRUCT_DATA struct_data;
memset(&struct_data, 0x0, sizeof(struct_data));
memcpy(&struct_data, buffer, sizeof(buffer)); //把结构体的数据通过memcpy的方式拷贝到struct_data中
printf("####L(%d) cmd:%d data_len:%d\n", __LINE__, struct_data.m_cmd, struct_data.m_data_len);

if(struct_data.m_data_len > 0) //m_data_len>0 则仍有数据要接收
{
printf("####L(%d) going to recv data...\n", __LINE__);

length = recv(client_socket, buffer, struct_data.m_data_len, 0);
if (length < 0)
{
printf("####L(%d) server recieve data failed!\n", __LINE__);
exit(1);
}
printf("####L(%d) data:%s\n", __LINE__, buffer); //打印要接收的数据部分
//解析xml

char file_size_tmp[8] = {0};
parse_xml(file_name, buffer, "<name>", "</name>");
parse_xml(file_size_tmp, buffer, "<size>", "</size>");
if(NULL != file_name)
{
file_size = atol(file_size_tmp);
printf("####L(%d) name:%s, size:%ld\n", __LINE__, file_name, file_size);
}
else
{
printf("####L(%d) parse xml err！\n", __LINE__);
exit(1);
}

}

if(0 == access(file_name, F_OK))
{
/*如果文件已经存在 先删*/
if( 0 != remove(file_name) )
{
printf("####L(%d) rm file:%s err！\n", __LINE__, file_name);
}
}

FILE *fp = fopen(file_name, "wb");
if(NULL == fp )
{
printf("####L(%d) file:%s can not open to write\n", __LINE__, file_name);
exit(1);
}

//开始接收文件
printf("####L(%d) now receiving file...\n", __LINE__);
while( ( length = recv(client_socket, buffer, BUFFER_SIZE, 0) ))
{
if(length < 0)
{
printf("####L(%d) current length < 0 !\n", __LINE__);
fclose(fp);
exit(1);
}
int write_length = fwrite(buffer, sizeof(char), length, fp); //写文件
if (write_length < length)
{
printf("####L(%d) file:%s write failed write_length<length!\n", __LINE__, file_name);
fclose(fp);
exit(1);
}

bzero(buffer, BUFFER_SIZE);
recv_sum += length;//累计接收文件大小

//计算当前进度
current = recv_sum / (file_size / progress);
//打印进度条
printf("\r");
printf("[");
for(i = 0; i < progress; i++)
{
if(i < current)
printf("="); //已接收部分
else
printf("+"); //未接收部分
}
printf("]");
printf(" %8ld/%ld %6.2f%%", recv_sum, file_size, (float)recv_sum / file_size * 100);

if(recv_sum == file_size)
{
//接收完成
printf("\n####L(%d) recv file finished ...####\n", __LINE__);
usleep(500 * 1000);
break;
}
else if(recv_sum > file_size)
{
printf("\n####L(%d) recv file err recv_sum[%ld] ...####\n", __LINE__, recv_sum);
break;
}
usleep(1);
}

fclose(fp);
//关闭socket
close(client_socket);
return 0;
}
//end flie_client.c