day4.。。。。。。。。。。。。。。。。。

服务端：

enum Type{
	TYPE_REGIST,
	TYPE_LOGIN,
	TYPE_CHAT,
	TYPE_FILE_UPLOAD_REQUEST
};

 typedef struct Pack{
	enum Type type;
	char name[20];
	char pswd[20];
	char tarname[20];
	char text[1024];
	char filename[20];
	long filesize;
}pack_t;

typedef struct User{
	char name[20];
	char pswd[20];
	int sock;
	int hasMsg;
	char msg[1024];
	int file;
	char file2[1024];
}user_t;

user_t user_arr[50] = {0};
int user_len = 0;

void read_data(int client);
void insert_user(user_t user);
int find_user(const char* username);

int main(int argc, const char *argv[])
{
	if(argc != 2){
	printf("请输入端口号\n");
	return 1;
	}

	int port = atoi(argv[1]);

	int server =socket(AF_INET,SOCK_STREAM,0);

	addr_in_t addr = {0};
	addr.sin_family = AF_INET;
	addr.sin_port = htons(port);
	addr.sin_addr.s_addr= inet_addr("0.0.0.0");

	if(bind(server,(addr_t*)&addr,sizeof(addr)) == -1){
		perror("bind");
		return 1;
	}

	listen(server,10);

	int epfd =epoll_create1(EPOLL_CLOEXEC);

	struct epoll_event epoll_stdin ={
		.events = EPOLLIN,
		.data.fd = 0
	};

	struct epoll_event epoll_server = {
		.events = EPOLLIN,
		.data.fd = server
	};
	
	epoll_ctl(epfd,EPOLL_CTL_ADD,0,&epoll_stdin);
	epoll_ctl(epfd,EPOLL_CTL_ADD,server,&epoll_server);

	struct epoll_event arr[50] = {0};

	while(1){
		int len = epoll_wait(epfd,arr,50,-1);
		for(int i=0;i<len;i++){
			int fd = arr[i].data.fd;
			if(fd == server){
				printf("有新客户端连接\n");
				int client= accept(server,0,0);
				struct epoll_event epoll_client = {
					.events = EPOLLIN,
					.data.fd = client
				};
				epoll_ctl(epfd,EPOLL_CTL_ADD,client,&epoll_client);

			}else if(fd == 0){
				char buf[64] = "";
				scanf("%s",buf);
				while(getchar()!=10);
				printf("键盘输入数据:%s\n",buf);
			}else{
				read_data(fd);
			}
		}
	}
	return 0;
}

void insert_user(user_t user){
	user_arr[user_len] = user;
	user_len ++;
}

 int find_user(const char* username){
	for(int i=0;i<user_len;i++){
		if(strcmp(username,user_arr[i].name) == 0){
			return i;
		}
	}
	 return -1;
 }

 void read_data(int client){
	pack_t pack = {0};
	int res = read(client,&pack,sizeof(pack));
	switch(pack.type){
		case TYPE_REGIST:{
		int res = find_user(pack.name);
		char* msg = NULL;
		if(res == -1){ 
			user_t user = {0};
			strcpy(user.name,pack.name);
			strcpy(user.pswd,pack.pswd);
			insert_user(user);
			msg ="注册成功";
		}else{
			msg ="该账号已存在";
		}
		strcpy(pack.text,msg);
		write(client,&pack,sizeof(pack));
		break;
	}
	case TYPE_LOGIN:{
		int res = find_user(pack.name);
		char* msg = NULL;
		if(res == -1){
			msg ="该账号不存在";
		}else{
			user_t user = user_arr[res];
			if(strcmp(user.pswd,pack.pswd) == 0){
				msg ="登录成功";
				user_arr[res].sock = client;

				if(user_arr[res].hasMsg == 1){
					pack_t pack = {0};
					pack.type = TYPE_CHAT;
					strcpy(pack.text,user_arr[res].msg);
					write(client,&pack,sizeof(pack));
					user_arr[res].hasMsg = 0;

				}else if(user_arr[res].file = 1){//当前用户存在未接收文件
					pack_t pack = {0};
					pack.type = TYPE_FILE_UPLOAD_REQUEST;
					strcpy(pack.text,user_arr[res].file2);
					write(client,&pack,sizeof(pack));
					user_arr[res].file = 0;
				}
			}else{
				msg = "密码错误";
			}
		}
		strcpy(pack.text,msg);
		write(client,&pack,sizeof(pack));
		break;
	}
	case TYPE_CHAT:{
		char* msg = NULL;
		int res = find_user(pack.tarname);
		if(res == -1){
			msg ="该用户不存在";
			strcpy(pack.text,msg);
			write(client,&pack,sizeof(pack));
		}else{
			user_t user=user_arr[res];
			if(user.sock == 0){
			//	msg ="用户未登录";
			//	strcpy(pack.text,msg);
			//	write(client,&pack,sizeof(pack));
			user_arr[res].hasMsg = 1;
			strcpy(user_arr[res].msg,pack.text);
			}else{
				int tarsock = user.sock;
				write(tarsock,&pack,sizeof(pack));
			}
		}
		break;
	}
		case TYPE_FILE_UPLOAD_REQUEST:{
				//printf("接收到客户端文件上传请求:%s %s\n",pack.tarname,pack.filename);
				int res = find_user(pack.tarname);
				if(res == -1){
					// 没注册 今晚作业
					char buf2[64] = "该用户不存在";
					write(client,buf2,64);
				}else{
					if(user_arr[res].sock == 0){
						// 不在线，也是今晚作业
						user_arr[res].file = 1;// 表明当前用户存在了未接收文件
						strcpy(user_arr[res].file2,pack.text); // 将准备发送给该用户的文件，缓存到user_arr[res].file2里面去
					
					}else{
						// 目标用户在线，直接将接收到的pack包转发给目标用户
						printf("转发文件中...\n");
						write(user_arr[res].sock,&pack,sizeof(pack));
					}
				}				
				break;
			}
 }


 }

客户端：

enum Type{
	TYPE_REGIST,
	TYPE_LOGIN,
	TYPE_CHAT,
	TYPE_FILE_UPLOAD_REQUEST
};

typedef struct Pack{
	enum Type type;
	char name[20];
	char pswd[20];
	char tarname[20];
	char text[1024];
	char filename[20];
	long filesize;
}pack_t;

void* thread_main(void* arg){
	int client = *(int*)arg;
	while(1){
		pack_t pack = {0};
		int res = read(client,&pack,sizeof(pack));
		if(res == 0){break;}
		switch(pack.type){
			case TYPE_REGIST:{
				printf("%s\n",pack.text);
				break;
			}
			case TYPE_LOGIN:{
				printf("%s\n",pack.text);
				break;
			}
			case TYPE_CHAT:{
				printf("接收到消息:%s\n",pack.text);
				break;
			}
			case TYPE_FILE_UPLOAD_REQUEST:{
				char filename[128] = "./client_file_system/";
				// 获取文件名，方便打开文件
				strcat(filename , pack.filename);
				int fd = open(filename,O_CREAT | O_TRUNC | O_WRONLY,0666);
				// 获取文件长度
				long filesize = pack.filesize;
				long readed_size = 0;

				//printf("接收到文件名和文件大小\n");
				while(1){
					pack_t filepack = {0};
					int res = read(client,&filepack,sizeof(filepack)); // 读取别的客户端发来的文件内容
					if(res != sizeof(filepack)){
						printf("发生分包\n");
					}
					int size = strlen(filepack.text);

					//printf("接受到文件信息\n");
					write(fd,filepack.text,size);// 将接受到的文件内容，写入文件中去
					readed_size += size;
					if(readed_size >= filesize){
						close(fd);
						break;
					}

				}
				break;
			}
		}
	}
}

int main(int argc, const char *argv[])
{
	if(argc != 2){
		printf("请输入端口号\n");
		return 1;
	}

	int port = atoi(argv[1]);

	int client = socket(AF_INET,SOCK_STREAM,0);
	addr_in_t addr ={0};
	addr.sin_family = AF_INET;
	addr.sin_port = htons(port);
	addr.sin_addr.s_addr = inet_addr("192.168.126.71");

	if(connect(client,(addr_t*)&addr,sizeof(addr)) == -1){
		perror("connect");
		return 1;
	}

	pthread_t id;
	pthread_create(&id,0,thread_main,&client);
	pthread_detach(id);

	while(1){
		int ch= -1;
		printf("1:注册\n");
		printf("2:登录\n");
		printf("3:聊天\n");
		printf("4:发送文件\n");
		printf("0:退出\n");
		printf("请选择:");
		scanf("%d",&ch);
		while(getchar()!=10);
		switch(ch){
			case 1:{
				pack_t pack = {0};
				printf("请输入账号:");
				scanf("%s",pack.name);
				while(getchar()!=10);
				printf("请输入密码:");
				scanf("%s",pack.pswd);
				while(getchar()!=10);
				pack.type = TYPE_REGIST;
				write(client,&pack,sizeof(pack));
				break;
			}
			case 2:{
				pack_t pack = {0};
				printf("请输入账号:");
				scanf("%s",pack.name);
				while(getchar()!=10);
				printf("请输入密码:");
				scanf("%s",pack.pswd);
				while(getchar()!=10);
				pack.type = TYPE_LOGIN;
				write(client,&pack,sizeof(pack));				
				break;
			}
			case 3:{
				pack_t pack = {0};
				pack.type = TYPE_CHAT;
				scanf("%s %s",pack.tarname,pack.text);
				printf("tarname =%s\n",pack.tarname);
				while(getchar()!=10);
				write(client,&pack,sizeof(pack));
				break;
			}
			case 4:{
				pack_t pack = {0};
				pack.type = TYPE_FILE_UPLOAD_REQUEST;
				printf("请输入接受文件的用户名:");
				char tarname[20] = "";
				scanf("%19s",pack.tarname);
				strcpy(tarname,pack.tarname);
				while(getchar()!=10);
				printf("请输入想要发送的文件名:");
				scanf("%19s",pack.filename);
				while(getchar()!=10);
				int fd = open(pack.filename,O_RDONLY);
				if(fd == -1){
					printf("该文件不存在\n");
					break;
				}
				struct stat buf = {0};
				stat(pack.filename,&buf);
				pack.filesize = buf.st_size;
				write(client,&pack,sizeof(pack));

				while(1){
					pack_t pack = {0};
					pack.type = TYPE_FILE_UPLOAD_REQUEST;
					strcpy(pack.tarname,tarname);
					int res = read(fd,pack.text,1023);
					if(res == 0){break;}
					write(client,&pack,sizeof(pack));
				}
				close(fd);
				break;
			}
			case 0:{
				break;
			}
		}
	}
	return 0;
}