LV.8 Linux应用开发综合实战

在线词典项目

应用综合项目

在线英英词典

项目功能描述

1 用户注册和登录验证
2 服务器端将用户信息和历史记录保存在数据库中。客户端输入用户名和密码,服务器端在数据库中查找、匹配,返回结果
3 单词在线翻译
根据客户端输入的单词在字典文件中搜索
4 历史记录查询

项目分析

有道词典功能分析图

在这里插入图片描述

项目流程

定义数据库中表的结构
定义消息结构体
分析服务器端和客户端流程
编码实现

流程示意图-客户端

在这里插入图片描述

流程示意图-服务器

在这里插入图片描述

代码

词典文件(.txt)

格式
在这里插入图片描述

client.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>

#define   N  32

#define  R  1   // user - register
#define  L  2   // user - login
#define  Q  3   // user - query
#define  H  4   // user - history

// 定义通信双方的信息结构体
typedef struct {
	int type;
	char name[N];
	char data[256];
}MSG;


int  do_register(int sockfd, MSG *msg)
{
	msg->type = R;

	printf("Input name:");
	scanf("%s", msg->name);
	getchar();

	printf("Input passwd:");
	scanf("%s", msg->data);

	if(send(sockfd, msg, sizeof(MSG),0) < 0)
	{
		printf("fail to send.\n");
		return -1;
	}

	if(recv(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Fail to recv.\n");
		return -1;
	}

	// ok !  or  usr alread exist.
	printf("%s\n", msg->data);

	return 0;
}

int do_login(int sockfd, MSG *msg)
{
	msg->type = L;

	printf("Input name:");
	scanf("%s", msg->name);
	getchar();

	printf("Input passwd:");
	scanf("%s", msg->data);

	if(send(sockfd, msg, sizeof(MSG),0) < 0)
	{
		printf("fail to send.\n");
		return -1;
	}

	if(recv(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Fail to recv.\n");
		return -1;
	}

	if(strncmp(msg->data, "OK", 3) == 0)
	{
		printf("Login ok!\n");
		return 1;
	}
	else 
	{
		printf("%s\n", msg->data);
	}

	return 0;
}

int do_query(int sockfd, MSG *msg)
{
	msg->type = Q;
	puts("--------------");

	while(1)
	{
		printf("Input word:");
		scanf("%s", msg->data);
		getchar();

		//客户端,输入#号,返回到上一级菜单
		if(strncmp(msg->data, "#", 1) == 0)
			break;

		//将要查询的单词发送给服务器
		if(send(sockfd,msg, sizeof(MSG), 0) < 0)
		{
			printf("Fail to send.\n");
			return -1;
		}

		// 等待接受服务器,传递回来的单词的注释信息
		if(recv(sockfd, msg,sizeof(MSG), 0) < 0)
		{
			printf("Fail to recv.\n");
			return -1;
		}
		printf("%s\n", msg->data);
	}
		
	return 0;
}

int do_history(int sockfd, MSG *msg)
{

	msg->type = H;

	send(sockfd, msg, sizeof(MSG), 0);
	
	// 接受服务器,传递回来的历史记录信息
	while(1)
	{
		recv(sockfd, msg, sizeof(MSG), 0);

		if(msg->data[0] == '\0')
			break;

		//输出历史记录信息
		printf("%s\n", msg->data);
	}

	return 0;
}

// ./server  192.168.3.196  10000
int main(int argc, const char *argv[])
{

	int sockfd;
	struct sockaddr_in  serveraddr;
	int n;
	MSG  msg;

	if(argc != 3)
	{
		printf("Usage:%s serverip  port.\n", argv[0]);
		return -1;
	}

	if((sockfd = socket(AF_INET, SOCK_STREAM,0)) < 0)
	{
		perror("fail to socket.\n");
		return -1;
	}

	bzero(&serveraddr, sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
	serveraddr.sin_port = htons(atoi(argv[2]));

	if(connect(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
	{
		perror("fail to connect");
		return -1;
	}

	while(1)
	{
		printf("*****************************************************************\n");
		printf("* 1.register          2.login              3.quit               *\n");
		printf("*****************************************************************\n");
		printf("Please choose:");

		scanf("%d", &n);
		getchar();

		switch(n)
		{
		case 1:
			do_register(sockfd, &msg);
			break;
		case 2:
			if(do_login(sockfd, &msg) == 1)
			{
				goto next;
			}
			break;
		case 3:
			close(sockfd);
			exit(0);
			break;
		default:
			printf("Invalid data cmd.\n");
		}

	}

next:
	while(1)
	{
		printf("*****************************************************\n");
		printf("* 1.query_word   2.history_record   3.quit          *\n");
		printf("*****************************************************\n");
		printf("Please choose:");
		scanf("%d", &n);
		getchar();

		switch(n)
		{
			case 1:
				do_query(sockfd, &msg);
				break;
			case 2:
				do_history(sockfd, &msg);
				break;
			case 3:
				close(sockfd);
				exit(0);
				break;
			default :
				printf("Invalid data cmd.\n");
		}

	}
	
	return 0;
}

server.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sqlite3.h>
#include <signal.h>
#include <time.h>


#define   N  32

#define  R  1   // user - register
#define  L  2   // user - login
#define  Q  3   // user - query
#define  H  4   // user - history

#define  DATABASE  "my.db"

// 定义通信双方的信息结构体
typedef struct {
	int type;
	char name[N];
	char data[256];
}MSG;



int do_client(int acceptfd, sqlite3 *db);
void do_register(int acceptfd, MSG *msg, sqlite3 *db);
int do_login(int acceptfd, MSG *msg, sqlite3 *db);
int do_query(int acceptfd, MSG *msg, sqlite3 *db);
int do_history(int acceptfd, MSG *msg, sqlite3 *db);
int history_callback(void* arg,int f_num,char** f_value,char** f_name);
int do_searchword(int acceptfd, MSG *msg, char word[]);
int get_date(char *date);

// ./server  192.168.3.196  10000
int main(int argc, const char *argv[])
{

	int sockfd;
	struct sockaddr_in  serveraddr;
	int n;
	MSG  msg;
	sqlite3 *db;
	int acceptfd;
	pid_t pid;

	if(argc != 3)
	{
		printf("Usage:%s serverip  port.\n", argv[0]);
		return -1;
	}

	//打开数据库
	if(sqlite3_open(DATABASE, &db) != SQLITE_OK)
	{
		printf("%s\n", sqlite3_errmsg(db));
		return -1;
	}
	else
	{
		printf("open DATABASE success.\n");
	}

	if((sockfd = socket(AF_INET, SOCK_STREAM,0)) < 0)
	{
		perror("fail to socket.\n");
		return -1;
	}

	bzero(&serveraddr, sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
	serveraddr.sin_port = htons(atoi(argv[2]));

	if(bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
	{
		perror("fail to bind.\n");
		return -1;
	}

	// 将套接字设为监听模式
	if(listen(sockfd, 5) < 0)
	{
		printf("fail to listen.\n");
		return -1;
	}

	//处理僵尸进程
	signal(SIGCHLD, SIG_IGN);

	while(1)
	{
		if((acceptfd = accept(sockfd, NULL, NULL)) < 0)
		{
			perror("fail to accept");
			return -1;
		}

		if((pid = fork()) < 0)
		{
			perror("fail to fork");
			return -1;
		}
		else if(pid == 0)  // 儿子进程
		{
			//处理客户端具体的消息
			close(sockfd);
			do_client(acceptfd, db);

		}
		else  // 父亲进程,用来接受客户端的请求的
		{
			close(acceptfd);
		}
	}
	
	return 0;
}


int do_client(int acceptfd, sqlite3 *db)
{
	MSG msg;
	while(recv(acceptfd, &msg, sizeof(msg), 0) > 0)
	{
	  printf("type:%d\n", msg.type);
	   switch(msg.type)
	   {
	  	 case R:
			 do_register(acceptfd, &msg, db);
			 break;
		 case L:
			 do_login(acceptfd, &msg, db);
			 break;
		 case Q:
			 do_query(acceptfd, &msg, db);
			 break;
		 case H:
			 do_history(acceptfd, &msg, db);
			 break;
		 default:
			 printf("Invalid data msg.\n");
	   }

	}

	printf("client exit.\n");
	close(acceptfd);
	exit(0);

	return 0;
}

void do_register(int acceptfd, MSG *msg, sqlite3 *db)
{
	char * errmsg;
	char sql[128];

	sprintf(sql, "insert into usr values('%s', %s);", msg->name, msg->data);
	printf("%s\n", sql);

	if(sqlite3_exec(db,sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		printf("%s\n", errmsg);
		strcpy(msg->data, "usr name already exist.");
	}
	else
	{
		printf("client  register ok!\n");
		strcpy(msg->data, "OK!");
	}

	if(send(acceptfd, msg, sizeof(MSG), 0) < 0)
	{
		perror("fail to send");
		return ;
	}

	return ;
}

int do_login(int acceptfd, MSG *msg , sqlite3 *db)
{
	char sql[128] = {};
	char *errmsg;
	int nrow;
	int ncloumn;
	char **resultp;

	sprintf(sql, "select * from usr where name = '%s' and pass = '%s';", msg->name, msg->data);
	printf("%s\n", sql);

	if(sqlite3_get_table(db, sql, &resultp, &nrow, &ncloumn, &errmsg)!= SQLITE_OK)
	{
		printf("%s\n", errmsg);
		return -1;
	}
	else
	{
		printf("get_table ok!\n");
	}

	// 查询成功,数据库中拥有此用户
	if(nrow == 1)
	{
		strcpy(msg->data, "OK");
		send(acceptfd, msg, sizeof(MSG), 0);
		return 1;
	}

	if(nrow == 0) // 密码或者用户名错误
	{
		strcpy(msg->data,"usr/passwd wrong.");
		send(acceptfd, msg, sizeof(MSG), 0);
	}

	return 0;
}

int do_searchword(int acceptfd, MSG *msg, char word[])
{
	FILE * fp;
	int len = 0;
	char temp[512] = {};
	int result;
	char *p;


	//打开文件,读取文件,进行比对
	
	if((fp = fopen("dict.txt", "r")) == NULL)
	{
		perror("fail to fopen.\n");
		strcpy(msg->data, "Failed to open dict.txt");
		send(acceptfd, msg, sizeof(MSG), 0);
		return -1;
	}

	//打印出,客户端要查询的单词
	len = strlen(word);
	printf("%s , len = %d\n", word, len);

	//读文件,来查询单词
	while(fgets(temp, 512, fp) != NULL)
	{

	//	printf("temp:%s\n", temp);

		// abandon  ab
		result = strncmp(temp,word,len);

		if(result < 0)
		{
			continue;
		}
		if(result > 0 || ((result == 0) && (temp[len]!=' ')))
		{
			break;
		}

		// 表示找到了,查询的单词
		p = temp + len; //  abandon   v.akdsf dafsjkj 
	//	printf("found word:%s\n", p);
		while(*p == ' ')
		{
			p++;
		}

		// 找到了注释,跳跃过所有的空格
		strcpy(msg->data, p);
		printf("found word:%s\n", msg->data);

		// 注释拷贝完毕之后,应该关闭文件
		fclose(fp);
		return 1;
	}

	fclose(fp);

	return 0;
}

int get_date(char *date)
{
	time_t t;
	struct tm *tp;

	time(&t);

	//进行时间格式转换
	tp = localtime(&t);

	sprintf(date, "%d-%d-%d %d:%d:%d", tp->tm_year + 1900, tp->tm_mon+1, tp->tm_mday, 
			tp->tm_hour, tp->tm_min , tp->tm_sec);
	printf("get date:%s\n", date);

	return 0;
}

int do_query(int acceptfd, MSG *msg , sqlite3 *db)
{
	char word[64];
	int found = 0;
	char date[128] = {};
	char sql[128] = {};
	char *errmsg;

	//拿出msg结构体中,要查询的单词
	strcpy(word, msg->data);

	found = do_searchword(acceptfd, msg, word);
	printf("查询一个单词完毕.\n");

	// 表示找到了单词,那么此时应该将 用户名,时间,单词,插入到历史记录表中去。
	if(found == 1)
	{
		// 需要获取系统时间
		get_date(date);

        sprintf(sql, "insert into record values('%s', '%s', '%s')", msg->name, date, word);

		if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
		{
			printf("%s\n", errmsg);
			return -1;
		}
		else
		{
			printf("Insert record done.\n");
		}

	}
	else  //表示没有找到
	{
		strcpy(msg->data, "Not found!");
	}

	// 将查询的结果,发送给客户端
	send(acceptfd, msg, sizeof(MSG), 0);

	return 0;
}

// 得到查询结果,并且需要将历史记录发送给客户端
int history_callback(void* arg,int f_num,char** f_value,char** f_name)
{
	// record  , name  , date  , word 
	int acceptfd;
	MSG msg;

	acceptfd = *((int *)arg);

	sprintf(msg.data, "%s , %s", f_value[1], f_value[2]);

	send(acceptfd, &msg, sizeof(MSG), 0);

	return 0;
}


int do_history(int acceptfd, MSG *msg, sqlite3 *db)
{
	char sql[128] = {};
	char *errmsg;

	sprintf(sql, "select * from record where name = '%s'", msg->name);

	//查询数据库
	if(sqlite3_exec(db, sql, history_callback,(void *)&acceptfd, &errmsg)!= SQLITE_OK)
	{
		printf("%s\n", errmsg);
	}
	else
	{
		printf("Query record done.\n");
	}

	// 所有的记录查询发送完毕之后,给客户端发出一个结束信息
	msg->data[0] = '\0';

	send(acceptfd, msg, sizeof(MSG), 0);

	return 0;
}



文件上传下载

需求

通过分析现象来推导代码的实现===>ftp服务器的实现(提供参考框架,但是不提供参考代码):

fengjunhui@ubuntu:~/homework/ftpserver$ ls
client.c Makefile server.c

fengjunhui@ubuntu:~/homework/ftpserver$ make
gcc -g -c server.c -o server.o
gcc server.o -o server
gcc -g -c client.c -o client.o
gcc client.o -o client

fengjunhui@ubuntu:~/homework/ftpserver$ ls
client client.c client.o Makefile server server.c server.o
(*.o文件可以删掉)

fengjunhui@ubuntu:~/homework/ftpserver$ mkdir feng
fengjunhui@ubuntu:~/homework/ftpserver$ mv client feng/

fengjunhui@ubuntu:~/homework/ftpserver/feng$ ls
client

服务器端等待链接请求:
fengjunhui@ubuntu:~/homework/ftpserver$ ./server 192.168.1.200 9999 (指定你本机Ubuntu的ip和>5000的端口号)
server sockfd :3

注意:编译生成可执行文件之后./server 和 ./client的执行在不同的目录
客户端登录
fengjunhui@ubuntu:~/homework/ftpserver/feng$ ./client 192.168.1.200 9999 (指定你本机Ubuntu的ip和>5000的端口号)


请输入 help 查看选项


input your choice: >>>

input your choice: >>> help


输入/功能*****************************
list :查看服务器所在目录的所有文件****
get filename 下载服务器目录的文件****
put filename: 上传文件到服务器********
********quit :关闭客户端 *****************************


功能1----------------查看服务器端的文件列表信息

input your choice: >>> list
***Makefile
***server
***server.c

服务器目录已经接收完毕

服务器应答
目录清单已经成功发送

功能2 --------------从服务器段下载文件到客户端

input your choice: >>> get server.c
下载完毕
ls 客户端所在目录可以看到server.c的文件

服务器提示:
文件传送完成

功能3 ------------向服务器端上传文件

input your choice: >>> put hello.c(自己定义一个文件,输出hello world就行)
上传完毕

服务器提示:
接收文件成功
client client.c client.o Makefile server server.c server.o hello.c

功能4--------------客户端退出,服务器继续等待链接

input your choice: >>> quit

服务器端打印客户端退出

要求:
上传或下载的某个功能要采用多进程或多线程的方式实现
每个客户端访问的日期信息要显示在服务器上
要求独立实现,不要问老师,实在不会的也不要问老师,自己尽可能的逼着自己独立去实现,
独立去调试,可以参考level7-8部分在线词典的项目,但是一定要自己吃透。

代码

结构

.
├── client.c
├── common.h
├── feng
│ ├── client
│ ├── core
│ ├── test1.txt
│ └── test.txt
├── Makefile
└── server.c

common.h

#ifndef __COMMON__
#define __COMMON_
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <signal.h>
#include <time.h>
#include  <dirent.h>
#define N 32;


	
#define OVER  8888   // file - upload
#define  P  3   // file - upload
#define  G  2   // file - download
#define  Q  1   // file - query
// 定义通信双方的信息结构体
typedef struct {
	int type;//通信类型
	char data[256];//通信数据
}MSG;

#endif 

client.c

#include "common.h"
//查询服务器同级目录的文件
int  do_list(int sockfd, MSG *msg)
{
	ssize_t ret;
	msg->type = Q;//指定请求的功能

	if(send(sockfd, msg, sizeof(MSG),0) < 0){//第一次发送请求
		printf("fail to send.\n");
		return -1;
	}
	printf("***********************************\n");
	while(ret){
		ret = recv(sockfd, msg, sizeof(MSG), 0); //接收服务器响应
		if(ret < 0 ){
			printf("Fail to recv.\n");
			return -1;
		}

		if(msg->type==OVER){//判断传输是否结束
			break;
		}
		printf("%s\n", msg->data);//输出服务器目录列表
	}
	printf("***********************************\n");
	printf("do_list over\n");	
	return 0;
}
//下载服务器文件功能
int	do_download(int sockfd,MSG *msg){
	ssize_t ret;
	msg->type =G;
	char filename[20]="";
	//	char buf[20];
	int fd;
	printf("指定要下载的文件 get filename>>>");
	scanf("%s",filename);
	strcpy(msg->data,filename);

	if(send(sockfd, msg, sizeof(MSG),0) < 0){//第一次请求服务器
		printf("fail to send.\n");
		return -1;
	}


	if((fd = open(filename,O_RDWR|O_CREAT|O_APPEND,0666))<0){//创建本地空文件
		perror("open");
		return  -1;
	}


	printf("***********************************\n");
	while(ret){
		bzero(&msg->data,sizeof(msg->data));//清空数据暂存区
		ret = recv(sockfd, msg, sizeof(MSG), 0); 
		if(ret < 0 ){
			close(fd);
			printf("Fail to recv.\n");
			return -1;
		}

		if(msg->type==OVER){//判断传输是否结束
			break;
		}else{
			printf("***********msg->data:%s\n",msg->data);
			write(fd,msg->data,strlen(msg->data));
		}

	}
	printf("***********************************\n");
	close(fd);
	return 0;
}
//上传文件到服务器功能
int	do_upload(int sockfd,MSG *msg){
	msg->type = P;
	char filename[20]="";
	int fd;
	printf("指定上传要的文件 get filename>>>");
	scanf("%s",filename);
	strcpy(msg->data,filename);
	if(send(sockfd, msg, sizeof(MSG),0) < 0)
	{
		printf("fail to send.\n");
		return -1;
	}
	//打开要上传的文件
	if((fd = open(msg->data,O_RDONLY))<0){
		perror("open");
		return  -1;
	}		
	while (1) {
		if((read(fd,msg->data,255))){//读取本地文件到暂存区
			printf("mag->data1:%s\n",msg->data);
		}else{
			msg->type=OVER;
			printf("mag->data2:%s\n",msg->data);
			if(send(sockfd,msg, sizeof(MSG), 0) < 0){
				perror("fail to send");
				close(fd);
				return -1;
			}
			break;
		}
		if(send(sockfd, msg, sizeof(MSG), 0) < 0)	{
			perror("fail to send");
			close(fd);
			return -1;
		}	
		bzero(&msg->data,sizeof(msg->data));//清空数据暂存区
	}
	close(fd);
	printf("do_upload\n");
	return 0;
}


int main(int argc, char *argv[])
{
	int sockfd;
	struct sockaddr_in  serveraddr;
	int num;
	MSG  msg;
	char str[10] ="";

	if(argc != 3){
		printf("Usage:%s serverip  port.\n", argv[0]);
		return -1;
	}

	if((sockfd = socket(AF_INET,SOCK_STREAM,0))<0){//创建一个绑定到特定传输服务提供者的套接字。
		perror("fail to socket.\n");
		return -1;
	}
	//允许绑定地址快速重用
	int b_reuse = 1;
	setsockopt (sockfd, SOL_SOCKET, SO_REUSEADDR, &b_reuse, sizeof (int));

	bzero(&serveraddr,sizeof(serveraddr));//将serveraddr变量的每一个字节都置零
	serveraddr.sin_family=AF_INET;//指定使用ipv4协议
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);//将本地字节序IP地址,转换成网络ip字节序地址
	serveraddr.sin_port = htons(atoi(argv[2]));//字符端口转换成数值端口再转换成网络字节序

	//connect()系统调用将文件描述符sockfd引用的套接字连接到由addr指定的地址。参数addrlen指定了addr的大小。addr中的地址格式由套接字sockfd的地址空间决定
	if(connect(sockfd,(struct sockaddr*)&serveraddr,sizeof(serveraddr))<0){
		perror("connect");
		return -1;
	}


next:	
	printf("********************************\n");
	printf("******请输入 help 查看选项******\n");
	printf("********************************\n");
	printf("input your choice:>>>");
	scanf("%s",str);
	while(getchar()!='\n');
	if(strcasecmp(str,"help")){
		printf("invalid input,retype\n");
		goto next;
	}

next1:	
	//选择功能
	printf("*****************************************************\n");
	printf("*******输入/功能*************************************\n");
	printf("*******1.list :查看服务器所在目录的所有文件************\n");
	printf("*******2.get filename: 下载服务器目录的文件*************\n");
	printf("*******3.put filename: 上传文件到服务器****************\n");
	printf("*******4.quit :关闭客户端 *****************************\n");
	printf("*****************************************************\n");



	while(1){
		printf("请输入要执行的功能的序号(1-4)>>>");
		scanf("%d",&num);
		while(getchar()!='\n');
		switch(num){
		case 1:
			do_list(sockfd, &msg);
			goto next1;
		case 2:
			do_download(sockfd,&msg);
			break;
		case 3:
			do_upload(sockfd,&msg);
			break;
		case 4:
			close(sockfd);
			exit(0);
			break;
		default:
			printf("invalid input,retype\n");
		}
	}





	printf("okokokokokokokokokokokokokokokokok\n");
	return 0;
}


server.c

#include "common.h"

int do_list(int acceptfd,MSG *msg){
	DIR *dirp;
	struct dirent *dp;
	char *path = "./";

	if ((dirp  = opendir(path))==NULL){
		perror("opendir");
		return -1;
	}

	while (1) {
		//    printf(“%s\n”, dp->d_name);
		//	msg->data = dp->d_name;
		if((dp = readdir(dirp)) != NULL){
			strcpy(msg->data,dp->d_name);
		}else{
			msg->type=OVER;
			if(send(acceptfd, msg, sizeof(MSG), 0) < 0)	{
				perror("fail to send");
				return -1;
			}	
			break;
		}
		if(send(acceptfd, msg, sizeof(MSG), 0) < 0)	{
			perror("fail to send");
			return -1;
		}	
		printf("msg->data:%s\n",msg->data);
	}
	closedir(dirp);
	printf("do_list over\n");
	return 0;
}

//下载文件
int do_download(int acceptfd,MSG *msg){
	int fd;
	if((fd = open(msg->data,O_RDONLY))<0){
		perror("open");
		strcpy(msg->data,"未找到文件或文件打开失败");
		if(send(acceptfd, msg, sizeof(MSG), 0) < 0)	{
			perror("fail to send");
			return -1;
		}	
		
		msg->type=OVER;
		if(send(acceptfd, msg, sizeof(MSG), 0) < 0)	{
			perror("fail to send");
			return -1;
		}	
		return  -1;
	}		
	while (1) {
		if((read(fd,msg->data,255))){
		printf("mag->data1:%s\n",msg->data);
		}else{
			msg->type=OVER;
		printf("mag->data2:%s\n",msg->data);
			if(send(acceptfd, msg, sizeof(MSG), 0) < 0)	{
				perror("fail to send");
				return -1;
			}	
			break;
		}
		if(send(acceptfd, msg, sizeof(MSG), 0) < 0)	{
			perror("fail to send");
			return -1;
		}	
		bzero(&msg->data,sizeof(msg->data));
	}
	close(fd);
	printf("do_download over\n");
	return 0;
}


//上传文件到服务器功能
int	do_upload(int sockfd,MSG *msg){
	ssize_t ret;
	int fd;

	if((fd = open(msg->data,O_RDWR|O_CREAT,0666))<0){
	    perror("open");
	    return  -1;
	}


	printf("***********************************\n");
	while(ret){
		bzero(&msg->data,sizeof(msg->data));
		ret = recv(sockfd, msg, sizeof(MSG), 0); 
		if(ret < 0 ){
			printf("Fail to recv.\n");
			return -1;
		}

		if(msg->type==OVER){
			break;
		}else{
			printf("***********msg->data:%s\n",msg->data);
			write(fd,msg->data,strlen(msg->data));
		}

	}
	printf("***********************************\n");
	printf("do_upload over\n");
	close(fd);
	return 0;
}


int do_client(int acceptfd)
{
	MSG msg;
	while(recv(acceptfd, &msg, sizeof(msg), 0) > 0)
	{
		printf("type:%d\n", msg.type);
		switch(msg.type)
		{
		case P:
			do_upload(acceptfd,&msg);
			break;
		case G:
			do_download(acceptfd,&msg);
			break;
		case Q:
			do_list(acceptfd,&msg);
			break;
		default:
			printf("Invalid data msg.\n");
		}

	}

	printf("client exit.\n");
	close(acceptfd);
	exit(0);

	return 0;
}



int main(int argc, char *argv[])
{
	int sockfd;
	struct sockaddr_in serveraddr;
	int acceptfd;

	pid_t pid;


	if(argc != 3){
		printf("Usage:%serverip port\n",argv[0]);
		return -1;
	}

	if((sockfd = socket(AF_INET,SOCK_STREAM,0))<0){
		perror("socket");
		return -1;
	}
	//允许绑定地址快速重用
	int b_reuse = 1;
	setsockopt (sockfd, SOL_SOCKET, SO_REUSEADDR, &b_reuse, sizeof (int));
	bzero(&serveraddr,sizeof(serveraddr));

	serveraddr.sin_family = AF_INET;
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
	serveraddr.sin_port = htons(atoi(argv[2]));
	//当使用socket(2)创建套接字时,它存在于名称空间(地址族)中,但没有分配给它的地址。Bind()将addr指定的地址分配给文件描述符sockfd引用的套接字。Addrlen以字节为单位指定addr指向的地址结构的大小。传统上,此操作称为“为套接字分配名称”。
	if(bind(sockfd,(struct sockaddr*)&serveraddr,sizeof(serveraddr))<0){
		perror("fail to bind.\n");
		return -1;
	}

	//将套接字设为监听模式
	if(listen(sockfd, 5) < 0)
	{
		printf("fail to listen.\n");
		return -1;
	}

	printf("服务端已启动******\n");

	//处理僵尸进程
	signal(SIGCHLD,SIG_IGN);

	while(1){
		if((acceptfd = accept(sockfd,NULL,NULL))<0){
			perror("acceptfd");
			return -1;
		}
		if((pid = fork())<0){
			perror("fork");
			return -1;
		}else if(pid == 0 ){
			//处理客户端具体的消息
			close(sockfd);
			do_client(acceptfd);
		}else{
			close(acceptfd);
		}
	}
	return 0;
}


Makefile

cc = gcc
CFLAGS = -Wall  -g

all:client server
client:client.o
server:server.o

client.o:client.c
	$(cc) $(CFLAGS) -c $^ -o $@ 
server.o:server.c
	$(cc) $(CFLAGS) -c $^ -o $@ 

.PHONY:clean
clean:
	rm -r *.o client server;  
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值