需求分析
需要在两台 linux 虚拟机之间传送文件
程序功能:
(1)客户端:
a.可以向服务端发送一个本目录下指定的文件,文件名由用户输入;
b.可以从服务端下载一个服务端目录下的文件,先从服务端获得文件名列表,再由用户输入需要的文件名。
(2)服务端:
a.可以监听来自客户端的连接请求;
b.可以接收客户端传送的文件;
c.可以向客户端传送一个指定的文件,文件由客户端给出。
(3)传送文件要求:任何二进制文件。
上一篇文章已经介绍过该程序的基本流程,下面给出源代码,
文章链接:https://blog.csdn.net/weixin_45937957/article/details/123362167
说明:可以在一台虚拟机开两个终端运行,也可以在两台虚拟机分别运行,可以多个客户端连接一个服务端
其中
#define SERVER_IP “192.168.62.129”
#define SERVER_PORT 1234
#define BUFFER_SIZE 1024
#define LOCAL_PATH “./local_client_text/” //客户端本地文件夹
#define DOWNLOAD_PATH “./download_text/” //客户端下载路径(从服务端下载)
根据实际情况填写
我的文件夹:
client.c:
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <dirent.h>
#define SERVER_IP "192.168.62.129"
#define SERVER_PORT 1234
#define BUFFER_SIZE 1024
#define LOCAL_PATH "./local_client_text/" //客户端本地文件夹
#define DOWNLOAD_PATH "./download_text/" //客户端下载路径(从服务端下载)
int connect_to_Server(char* serverIP, int port) {
struct sockaddr_in serverAddr;
int Socket;
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(port);
serverAddr.sin_addr.s_addr = inet_addr(serverIP);
if((Socket = socket(AF_INET, SOCK_STREAM, 0)) < 0){
printf("sock生成出错!\n");
return -1;
}
if(connect(Socket, (struct sockaddr*)&serverAddr, sizeof(serverAddr)) < 0) {
printf("connect失败!\n");
return -1;
}
return Socket;
}
int get_File_Num(char* pathname) {
struct dirent* ptr;
DIR* path = NULL;
path = opendir(pathname);
int cnt = 0;
while((ptr=readdir(path)) != NULL) {
if(strcmp(ptr->d_name,".")==0||strcmp(ptr->d_name,"..")==0)
continue;
if(ptr->d_type==DT_REG)
cnt++;
}
return cnt;
}
int pack_send_File(int Socket, FILE* fp) {
char send_buf[BUFFER_SIZE] = {0};
int len_file, len_block;
int totalBlock;
fseek(fp,0,2);//将指针放到文件末尾
len_file = ftell(fp); //统计文本的数据长度
rewind(fp); //让指针指向文件开始位置
printf("文件的总长度len_file:%d\n",len_file);
totalBlock = len_file % BUFFER_SIZE == 0 ? len_file / BUFFER_SIZE : ((len_file / BUFFER_SIZE) + 1); //除最后一快的总块数
printf("文件的总块数totalBlock:%d\n",totalBlock);
sprintf(send_buf, "%d", totalBlock);
send(Socket,(char*)send_buf,sizeof(send_buf),0);
for(int i = 1; i <= totalBlock; i++) {
len_block = fread(send_buf, 1, BUFFER_SIZE, fp);
send_buf[len_block]='\0';
char temp[100] = {0};
sprintf(temp, "%d", len_block);
printf("文件第%d块的长度len_block:%s\n",i,temp);
send(Socket,(char*)temp,sizeof(temp),0);
printf("第%d块的内容:%s\n",i,send_buf);
send(Socket,send_buf,len_block,0);
}
fclose(fp);
}
int submit_File(int Socket) {//提交单个文件
//将文件内容发送给服务端
FILE *fp;
char send_buf[BUFFER_SIZE] = {0};
char filename[100] = {0};
char* pre_filename = (char*)malloc(sizeof(char)*100);
while(1){
strcpy(pre_filename,LOCAL_PATH);
printf("请输入你要提交的文件名: ");
scanf("%[^\n]%*c",filename);
strcat(pre_filename,filename);
fp = fopen(pre_filename, "rb");
if(fp == NULL) {
printf("您提供的文件不存在,请重新输入!\n");
continue;
}
break;
}
printf("已找到您的文件:路径是%s\n",pre_filename);
pack_send_File(Socket, fp);
printf("文件上传成功!\n");
// close(Socket);
}
int submit_Files(int Socket) {//提交多个文件,想法是提交某个文件夹下的所有文件
FILE* fp;
char filename[100] = {0}, pre_filename[100] = {0};
struct dirent* ptr;
DIR* path = NULL;
char send_buf[BUFFER_SIZE] = {0};
path = opendir((char*)LOCAL_PATH);
while((ptr=readdir(path)) != NULL) {
if(strcmp(ptr->d_name,".")==0||strcmp(ptr->d_name,"..")==0) {
continue;
}
if(ptr->d_type==DT_REG) {
printf("%s\n",ptr->d_name);
strcpy(pre_filename,LOCAL_PATH);
strcat(pre_filename,ptr->d_name);
fp = fopen(pre_filename, "rb");
pack_send_File(Socket, fp);
printf("文件%s上传成功!\n",ptr->d_name);
}
}
}
int receive_File(int Socket, char* filename)
{
char recv_buf[BUFFER_SIZE] = {0};
char pathname[200] = {0};
int totalBlock, lenBlock;
FILE *fp;
strcpy(pathname, DOWNLOAD_PATH);
strcat(pathname, filename);
printf("从服务器下载的文件存储在:%s\n", pathname);
fp = fopen(pathname, "wb");
// if (fp != NULL)
// printf("成功创建文件%s\n", pathname);
recv(Socket, recv_buf, sizeof(recv_buf), 0);
totalBlock = atoi(recv_buf);
// printf("文件有%d个块\n", totalBlock);
for (int j = 1; j <= totalBlock; j++)
{
char temp[100] = {0};
recv(Socket, temp, sizeof(temp), 0);
lenBlock = atoi(temp);
// printf("第%d个块长度为%d\n", j, lenBlock);
if ((recv(Socket, recv_buf, lenBlock, 0)) != -1){
// printf("..........接收成功..........\n");
}
else{
printf("接收失败!\n");
break;
}
fwrite(recv_buf, 1, lenBlock, fp);
}
fclose(fp);
char *p = (char *)"文件保存完毕";
printf("%s\n\n", p);
// close(clientSocket);
}
int Menu()
{
int ch;
printf("**********************┍ -----------------------┑ *********************\n");
printf("********************** 欢迎使用通信管理系统 *********************\n");
printf("**********************┕------------------------┙ *********************\n");
printf("xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx菜单xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n");
printf("======================================================================\n");
printf("....... 1.向服务端发送文件 .......\n");
printf("....... 2.获取服务端文件表名 .......\n");
printf("....... 3.从服务端下载文件 .......\n");
printf("....... 0.退出系统 .......\n");
printf("Please Input your choice:\n");
scanf("%d", &ch);
return ch;
}
int Send_To_File_Main(int serverSocket){
int choice;//选择上传单个文件还是多个
char send_buf[BUFFER_SIZE] = {0};
printf("请选择上传单个文件还是多个文件,单个文件输入0,多个文件输入1:\n");
while(1) {
scanf("%d",&choice);
getchar();
if(choice == 0 || choice == 1)
break;
else
printf("请正确输入0或1");
}
if(choice == 0) {
strcpy(send_buf, "1");
send(serverSocket, send_buf, sizeof(send_buf), 0);
submit_File(serverSocket);
} else {
int filenum = get_File_Num((char*)LOCAL_PATH);
sprintf(send_buf, "%d", filenum);
send(serverSocket, send_buf, sizeof(send_buf), 0);
submit_Files(serverSocket);
}
}
void choose(int ch, int socket)
{
if (ch != 0){ //发送客户端的需求类型
char send_type[BUFFER_SIZE] = {0};
sprintf(send_type, "%d", ch);
send(socket, send_type, sizeof(send_type), 0);
}
switch (ch)
{
case 1: //向服务端发送文件
Send_To_File_Main(socket);
break;
case 2: //获取服务端文件表名
{
char recvBuf[1024] = {0};
//读取服务器传回的数据
int len = read(socket, recvBuf, sizeof(recvBuf));
if(len == -1){
printf("获取服务端文件表名失败");
}else if(len > 0){
printf("recv server data : %s\n", recvBuf);
}else if(len == 0){
// 表示服务器端断开连接
printf("server closed...\n");
}
break;
}
case 3: //从服务端下载文件
{
char filename[100] = {0};
printf("请输入你要下载的文件名: ");
scanf("%s",filename);
write(socket, filename, strlen(filename)); //向服务端说明想要下载的文件名
receive_File(socket, filename);
break;
}
case 0:
printf("End of program!\n");
exit(0);
default: printf("Input error!\n");
}
}
int main() {
printf("Hello, there is client system! We will connect server system(192.168.62.129) automatically\n");
while (1) {
int ch = Menu(); //Menu的选择
if (ch != 0)
{
int serverSocket = connect_to_Server((char*)SERVER_IP,SERVER_PORT);
if(serverSocket == -1) return 0;
printf("已连接到服务器...\n");
choose(ch, serverSocket);
ch = -1;
close(serverSocket);
}else{
choose(ch, 0);
}
}
//shutdown(clientSocket, SHUT_WR);
//recv(clientSocket, recv_buf, strlen(recv_buf), MSG_DONTWAIT);
//printf("%s\n",recv_buf);
return 0;
}
server.c
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <dirent.h>
#define SERVER_IP "192.168.62.129" //服务器IP地址
#define SERVER_PORT 1234
#define BUFFER_SIZE 1024
#define RECV_SERVER_PATH "./recv_server_text/" //服务器接收客户端文件夹路径
#define SEND_SERVER_PATH "./send_server_text/" //服务器发送文件夹路径
#define FILES_NAME "欢迎查询服务端文件系统,服务端文件表名分别为:1,2,3,4,5,6,7" //服务端文件表名
int init_Server(int port)
{
int serverSocket;
struct sockaddr_in server_addr;
// socket初始化
if ((serverSocket = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("socket");
return -1;
}
//服务端地址类型,地址及端口初始化
bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(port);
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(serverSocket, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0)
{
perror("connect");
return -1;
}
//设置服务器上的socket为监听状态
if (listen(serverSocket, 5) < 0)
{
perror("listen");
return -1;
}
return serverSocket;
}
int get_File_Num(char *pathname)
{
struct dirent *ptr;
DIR *path = NULL;
path = opendir(pathname);
int cnt = 0;
while ((ptr = readdir(path)) != NULL)
{
if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)
continue;
if (ptr->d_type == DT_REG)
cnt++;
}
return cnt;
}
int pack_send_File(int Socket, FILE* fp) {
char send_buf[BUFFER_SIZE] = {0};
int len_file, len_block;
int totalBlock;
fseek(fp,0,2);//将指针放到文件末尾
len_file = ftell(fp); //统计文本的数据长度
rewind(fp); //让指针指向文件开始位置
printf("文件的总长度len_file:%d\n",len_file);
totalBlock = len_file % BUFFER_SIZE == 0 ? len_file / BUFFER_SIZE : ((len_file / BUFFER_SIZE) + 1); //除最后一快的总块数
printf("文件的总块数totalBlock:%d\n",totalBlock);
sprintf(send_buf, "%d", totalBlock);
send(Socket,(char*)send_buf,sizeof(send_buf),0);
for(int i = 1; i <= totalBlock; i++) {
len_block = fread(send_buf, 1, BUFFER_SIZE, fp);
send_buf[len_block]='\0';
char temp[100] = {0};
sprintf(temp, "%d", len_block);
printf("文件第%d块的长度len_block:%s\n",i,temp);
send(Socket,(char*)temp,sizeof(temp),0);
printf("第%d块的内容:%s\n",i,send_buf);
send(Socket,send_buf,len_block,0);
}
fclose(fp);
}
int receive_File(int clientSocket)
{
char recv_buf[BUFFER_SIZE] = {0};
char filename[200] = {0}, pathname[200] = {0};
int cnt;
int totalBlock, lenBlock;
recv(clientSocket, recv_buf, sizeof(recv_buf), 0); //先接收客户端要传多少个文件
// printf("%d\n",(int)strlen(recv_buf));
int filenum = atoi(recv_buf);
printf("客户端要发送%d个文件\n", filenum);
for (int i = 1; i <= filenum; i++)
{
//遍历指定目录下文件的个数,以确定新接收文件的名字
cnt = get_File_Num((char *)RECV_SERVER_PATH); //得到服务器目录下的文件个数,方便编号命名
FILE *fp;
memset(filename, 0, sizeof(filename));
sprintf(filename, "%d", cnt + 1);
strcpy(pathname, RECV_SERVER_PATH);
strcat(pathname, filename);
printf("%s\n", pathname);
fp = fopen(pathname, "wb");
if (fp != NULL)
printf("成功创建文件%s\n", pathname);
recv(clientSocket, recv_buf, sizeof(recv_buf), 0);
totalBlock = atoi(recv_buf);
// printf("第%d个文件有%d个块\n", i, totalBlock);
for (int j = 1; j <= totalBlock; j++)
{
char temp[100] = {0};
recv(clientSocket, temp, sizeof(temp), 0);
lenBlock = atoi(temp);
// printf("第%d个块长度为%d\n", j, lenBlock);
if ((recv(clientSocket, recv_buf, lenBlock, 0)) != -1){
// printf("..........接收成功..........\n");
}
else{
printf("接收失败!\n");
break;
}
fwrite(recv_buf, 1, lenBlock, fp);
}
fclose(fp);
char *p = (char *)"文件保存完毕";
printf("%s\n\n", p);
}
// close(clientSocket);
}
int submit_File(int Socket, char* filename) {//提交单个文件
//将文件内容发送给客户端
FILE *fp;
char send_buf[BUFFER_SIZE] = {0};
char* pre_filename = (char*)malloc(sizeof(char)*100);
strcpy(pre_filename,SEND_SERVER_PATH);
strcat(pre_filename,filename);
fp = fopen(pre_filename, "rb");
if(fp == NULL) {
printf("文件不存在,请客户端重新输入!\n");
}else{
printf("已找到文件");
pack_send_File(Socket, fp);
printf("文件上传成功!\n");
}
// close(Socket);
}
void KKL_Implement() {}
void choose(int ch, int socket){
switch (ch)
{
case 1: //向服务端发送文件
//将客户端发送过来的消息写入文件
receive_File(socket);
KKL_Implement(); //实现服务器端KKL协议的接口
close(socket);
break;
case 2: //获取服务端文件表名
{
char *data = FILES_NAME;
write(socket, data , strlen(data));
close(socket);
break;
}
case 3: //从服务端下载文件
{
//获取客户端想要下载的文件名
char filename[100] = {0};
int len = read(socket, filename, sizeof(filename));
if(len == -1){
printf("获取download文件表名失败");
}else if(len > 0){
printf("recv client data : %s\n", filename);
}
//传输客户端想要的文件
submit_File(socket, filename);
close(socket);
break;
}
default: printf("Input error!\n");
}
}
int main()
{
FILE *fp;
int serverSocket;
struct sockaddr_in clientAddr;
int addr_len = sizeof(clientAddr);
int clientSocket;
// char send_buf[BUFFER_SIZE] = {0}, recv_buf[BUFFER_SIZE] = {0};
// int iDataNum;
serverSocket = init_Server(SERVER_PORT);
if (serverSocket < 0)
{
printf("服务器初始化失败!\n");
return 0;
}
printf("服务器初始化成功,监听端口:%d\n", SERVER_PORT);
//等待用户连接
while (1)
{
clientSocket = accept(serverSocket, (struct sockaddr *)&clientAddr, (socklen_t *)&addr_len);
if (clientSocket < 0)
{
printf("服务端accept失败,程序退出。\n");
return 0;
}
printf("客户已连接:\n");
printf("IP is %s\n", inet_ntoa(clientAddr.sin_addr)); // inet_ntoa ip地址转换函数,将网络字节序IP转换为点分十进制IP
printf("Port is %d\n", htons(clientAddr.sin_port));
printf("等待消息...\n");
//解析客户端的需求类型
char recv_type[BUFFER_SIZE] = {0} ;
recv(clientSocket, recv_type, sizeof(recv_type), 0); //客户端的需求是1,2,3?
int tpye = atoi(recv_type);
printf("客户端的需求是: %d\n", tpye);
choose(tpye, clientSocket);
}
// clientSocket = accept(serverSocket, (struct sockaddr *)&clientAddr, (socklen_t *)&addr_len);
// if (clientSocket < 0)
// {
// printf("服务端accept失败,程序退出。\n");
// return 0;
// }
// printf("客户已连接:\n");
// printf("IP is %s\n", inet_ntoa(clientAddr.sin_addr)); // inet_ntoa ip地址转换函数,将网络字节序IP转换为点分十进制IP
// printf("Port is %d\n", htons(clientAddr.sin_port));
// printf("等待消息...\n");
// //将客户端发送过来的消息写入文件
// receive_File(clientSocket);
// KKL_Implement(); //实现服务器端KKL协议的接口
close(serverSocket);
return 0;
}