项目 - 员工信息管理系统

员工管理系统工作流程图:

项目 - 员工信息管理系统_第1张图片

员工管理系统-服务器端流程图:

项目 - 员工信息管理系统_第2张图片

 员工管理系统-客户端流程图:

项目 - 员工信息管理系统_第3张图片

项目名称:员工管理系统

项目时间:3天

项目描述:

1)服务器负责管理所有员工表单(以数据库形式或文件形式都可),其他客户端可通过网络连接服务器来查询员工表单。

2)需要账号密码登陆,其中需要区分管理员账号还是普通用户账号。

3)管理员账号可以查看、修改员工表单,管理员要负责管理所有的普通用户。

4)普通用户只能查询修改与本人有关的相关信息,其他员工信息(出于保密原则)不得泄露。

5)有查询历史记录功能。

6)能同时处理多台客户端的请求功能。

【1】TCP通信的编程步骤
  1.服务器:
	1)创建套接字
	2)绑定ip和端口号
	3)监听
	4)等待客户端连接
	
	int main()
	{
		//1.创建套接字
		int sockfd = socket();
		//2.初始化通信结构
		struct sockaddr_in addr;
		addr.sin_family=AF_INET;
		addr.sin_port = port;
		addr.sin_addr=addr;
		bind(sockfd, &addr);
		//3.监听
		listen();
		//4.连接
		while(1)
		{
			int connfd = accept();
			//5.循环数据收发
			while(1)
			{
				recv();
				send();
			}
		}
		close(sockfd);
		close(connfd);
	}
  2.客户端:
	1)创建套接字
	2)连接服务器
	
	int main()
	{
		//1.创建套接字
		int sockfd = socket();
		//2.初始化通信结构
		struct sockaddr_in addr;
		addr.sin_family=AF_INET;
		addr.sin_port = port;
		addr.sin_addr=addr;
		//3.连接
		connect();
		//5.循环数据收发
		while(1)
		{
			send();
			recv();
		}
	}
	
【2】服务器模型
 1.循环服务器
 2.并发服务器
	1)多线程
	2)多进程
	3)IO多路复用:
		a. select:
		基本思想:
		1. 先构造一张有关文件描述符的表(集合、数组); fd_set fd;
		2. 将你关心的文件描述符加入到这个表中;FD_SET();
		3. 然后调用一个函数。 select / poll 
		4. 当这些文件描述符中的一个或多个已准备好进行I/O操作的时候
				该函数才返回(阻塞)。
		5. 判断是哪一个或哪些文件描述符产生了事件(IO操作);
		6. 做对应的逻辑处理;
		****select函数返回之后,会自动将除了产生事件的文件描述符以外的位全部清空;
		程序步骤:
		1.把关心的文件描述符放入集合--FD_SET
		2.监听集合中的文件描述符--select
		3.依次判断哪个文件描述符有数据--FD_ISSET
		4.依次处理有数据的文件描述符的数据
		伪代码:
		fd_set fd;
		FD_SET(sockfd);
		while(1) {
			设置监听读写文件描述符集合(FD_*);
			调用select;
			select();	
			如果是监听套接字就绪,说明有新的连接请求 
			if(sockfd)
			{
				建立连接();
				int connfd = accept();
				加入到监听文件描述符集合;
				FD_SET(connfd);
			}否则说明是一个已经连接过的描述符 
			else
			{
				进行操作(send或者recv);
				recv();
				send();
			}
		} 
		select弊端: 
			1. 一个进程最多只能监听1024个文件描述符 (千级别)
			2. select是一种轮询的机制;
			3. 涉及到用户态和内核态的数据拷贝;
		
		b. poll
			1. 优化文件描述符个数的限制;
			2. poll是一种轮询的机制;
			3. 涉及到用户态和内核态的数据拷贝;
			函数接口:	   
			int poll(struct pollfd *fds, nfds_t nfds, int timeout);
			参数:
				struct pollfd *fds
				关心的文件描述符数组struct pollfd fds[N];
				nfds:个数
				timeout: 超市检测
						毫秒级的:如果填1000,1秒
								  如果-1,阻塞
								  
			问题:
			我想检测是键盘事件(标准输入 文件描述如为0 ),
			还是鼠标事件(文件描述符是/dev/input/mouse1);				  				
			1. 创建一个结构体数组
				struct pollfd fds[2];
			2. 将你关心的文件描述符加入到结构体成员中
				struct pollfd {
				   int   fd;         // 关心的文件描述符;
				   short events;     // 关心的事件,读
				   short revents;    // 如果产生事件,则会自动填充该成员的值
				};
			
				// 键盘
				fds[0].fd = 0;
				fds[0].events = POLLIN;
				
				//鼠标
				fds[1].fd = mouse1_fd;
				fds[1].events = POLLIN;
			
			3. 调用poll函数
				如果返回表示有事件产生;
				poll(fds,2,1000)
			4. 判断具体是哪个文件描述符产生了事件
				if(fds[0].revents == POLLIN)
				{
					....
				}
			
		c. epoll
			1. 没有文件描述符的限制
			2. 异步IO,当有事件产生,文件描述符主动调用callback
			3. 不用数据拷贝;
			
		3个功能函数:
			#include 
			int epoll_create(int size);//创建红黑树根节点
			//成功时返回epoll文件描述符,失败时返回-1
			
			//控制epoll属性
			int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
			epfd:epoll_create函数的返回句柄。
			op:表示动作类型。有三个宏 来表示:
			EPOLL_CTL_ADD:注册新的fd到epfd中
			EPOLL_CTL_MOD:修改已注册fd的监听事件
			EPOLL_CTL_DEL:从epfd中删除一个fd
			FD:需要监听的fd。
			event:告诉内核需要监听什么事件
				EPOLLIN:表示对应文件描述符可读
				EPOLLOUT:可写
				EPOLLPRI:有紧急数据可读;
				EPOLLERR:错误;
				EPOLLHUP:被挂断;
				EPOLLET:触发方式,电平触发;
					 ET模式:表示状态的变化;
			//成功时返回0,失败时返回-1
			
			//等待事件到来
			int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);
			功能:等待事件的产生,类似于select嗲用
			epfd:句柄;
			events:用来从内核得到事件的集合;
			maxevents:表示每次能处理事件最大个数;
			timeout:超时时间,毫秒,0立即返回,-1阻塞		
			//成功时返回发生事件的文件描述数,失败时返回-1
		伪代码:
		1.定义epoll事件,创建epoll的fd
			int epfd,epct,i;
			struct epoll_event event;       //定义epoll 事件
			struct epoll_event events[20];  //定义epoll 事件集合
			epfd = epoll_create(1); // 创建epoll 的fd 
		2.填充事件
			event.data.fd = serverFd;           //填充事件的fd
			event.events = EPOLLIN | EPOLLET;   //填充 事件类型 
			epoll_ctl(epfd,EPOLL_CTL_ADD,serverFd,&event);  //把serverFd(监听FD)注册到epfd中
		3.监听事件
			while(1){		
				epct = epoll_wait(epfd,events,20,-1); // 等待事件到来,阻塞模式 
				for(i=0;i

代码实现:

服务器端(common.h文件,客户端服务器通用):


#ifndef _COMMON_H_
#define _COMMON_H_

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define STAFF_DATABASE 	 "staff_manage_system.db"

#define USER_LOGIN 		0x00000000  // login	登陆    0x00000001
#define USER_MODIFY 	0x00000001  // user-modification  修改
#define USER_QUERY 		0x00000002  // user-query   查询

#define ADMIN_LOGIN 	0x10000000  // login	         登陆    0x00000001
#define ADMIN_MODIFY 	0x10000001 // admin_modification 修改
#define ADMIN_ADDUSER 	0x10000002 // admin_adduser      添加
#define ADMIN_DELUSER 	0x10000004 // admin_deluser      删除
#define ADMIN_QUERY 	0x10000008  //hitory_query       查找
#define ADMIN_HISTORY 	0x10000010  //hitory_history     历史

#define ADMIN_LIST      0x10000011 //列表

#define QUIT 			0x11111111

#define ADMIN 0	//管理员
#define USER  1	//用户

#define PASSLEN 8
#define NAMELEN 16
#define DATALEN 128

/*员工基本信息*/
typedef struct staff_info
{
	int  no; 					//员工编号
	int  usertype;  			//ADMIN 1	USER 2	 
	char name[NAMELEN];			//姓名
	char passwd[PASSLEN]; 		//密码
	int  age; 					// 年龄
	char phone[NAMELEN];		//电话
	char addr[DATALEN]; 		// 地址
	char work[DATALEN]; 		//职位
	char date[DATALEN];			//入职年月
	int level;					// 等级
	int salary;					// 工资
	
}staff_info_t;

/*定义双方通信的结构体信息*/
typedef struct 
{
	int  msgtype;       	 //请求的消息类型
	int  usertype;   		 //ADMIN 1	USER 2	   
	char username[NAMELEN];  //姓名
	char passwd[PASSLEN];	 //登陆密码
	char recvmsg[DATALEN];   //通信的消息
	int  flags;      		 //标志位
	staff_info_t info;       //员工信息
}MSG;

#define S_IP "10.0.12.9"
#define IP "124.223.177.144"
#define PORT "8888"

char user_name[NAMELEN];	 //姓名
char user_passwd[PASSLEN];	 //登陆密码

char history[DATALEN+100];




#endif

服务器端(server.h文件):

int process_user_or_admin_login_request(int acceptfd, MSG *msg);
int process_user_modify_request(int acceptfd, MSG *msg);
int process_user_query_request(int acceptfd, MSG *msg);
int process_admin_modify_request(int acceptfd, MSG *msg);
int process_admin_adduser_request(int acceptfd, MSG *msg);
int process_admin_deluser_request(int acceptfd, MSG *msg);
int process_admin_query_request(int acceptfd, MSG *msg);
int process_admin_history_request(int acceptfd, MSG *msg);
int process_admin_list_request(int acceptfd, MSG *msg);
int process_client_quit_request(int acceptfd, MSG *msg);
int process_client_request(int acceptfd, MSG *msg);
int process_recv_history_request(char *name,char *str);

服务器端(server.c文件):


#include "common.h"
#include "server.h"

sqlite3 *db = NULL;

//登录
int process_user_or_admin_login_request(int acceptfd, MSG *msg)
{

	//封装sql命令,表中查询用户名和密码-存在-登录成功-发送响应-失败-发送失败响应
	char sql[DATALEN] = {0};
	char *errmsg;
	char **result;
	int nrow, ncolumn;

	msg->info.usertype = msg->usertype;
	strcpy(msg->info.name, msg->username);
	strcpy(msg->info.passwd, msg->passwd);

	strcpy(user_name, msg->username);
	strcpy(user_passwd, msg->username);

	printf("usertype: %#x ,username: %s ,userpasswd: %s.\n", msg->info.usertype, msg->info.name, msg->info.passwd);
	sprintf(sql, "select * from usrinfo where usertype=%d and name='%s' and passwd='%s';", msg->info.usertype, msg->info.name, msg->info.passwd);

	if (sqlite3_get_table(db, sql, &result, &nrow, &ncolumn, &errmsg) != SQLITE_OK)
	{
		printf("---****----%s.\n", errmsg);
	}
	else
	{
		// printf("----nrow-----%d,ncolumn-----%d.\n",nrow,ncolumn);
		if (nrow == 0)
		{
			//查询到行为0
			strcpy(msg->recvmsg, "name or passwd failed.\n");
			send(acceptfd, msg, sizeof(MSG), 0);
			process_recv_history_request(msg->username, "longin error!");
		}
		else
		{
			strcpy(msg->recvmsg, "OK");
			send(acceptfd, msg, sizeof(MSG), 0);
			process_recv_history_request(msg->username, "longin success!");
		}
	}
	return 0;
}

//用户修改
int process_user_modify_request(int acceptfd, MSG *msg)
{
	char cmd[NAMELEN] = "";
	char *errmsg;
	char sql[DATALEN + 100] = "";
	int nrow = 0, ncolumn;
	char **result;

	switch (msg->flags)
	{
	case 1:
		sprintf(sql, "select * from usrinfo where taffon=%d;",msg->info.no);

		if (sqlite3_get_table(db, sql, &result, &nrow, &ncolumn, &errmsg) == SQLITE_OK)
		{
			memset(msg->recvmsg, 0, DATALEN);
			strcpy(msg->recvmsg, "no");
			send(acceptfd, msg, sizeof(MSG), 0);
			sqlite3_free_table(result);
			return -1;
		}
		sprintf(sql, "UPDATE usrinfo SET taffon=%d WHERE name='%s' AND passwd='%s';", msg->info.no, msg->username, msg->passwd);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change taffon -> %d", msg->info.no);
		process_recv_history_request(msg->username, history);
		
		break;
	case 2:
		// sprintf(sql, "UPDATE usrinfo SET usertype=%d WHERE name='%s' AND passwd='%s';",msg->info.usertype,msg->username,msg->passwd);
		{
			memset(msg, 0, sizeof(MSG));
			strcpy(msg->recvmsg, "no");
			if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
			{
				printf("Sending the account to the client error\n");
				return -1;
			}
			return -1;
		}
		break;
	case 3:
		sprintf(sql, "UPDATE usrinfo SET name='%s' WHERE name='%s' AND passwd='%s';", msg->info.name, msg->username, msg->passwd);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change name -> %s", msg->info.name);
		process_recv_history_request(msg->username, history);
		
		break;
	case 4:
		sprintf(sql, "UPDATE usrinfo SET passwd='%s' WHERE name='%s' AND passwd='%s';", msg->info.passwd, msg->username, msg->passwd);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change passwd -> %s", msg->info.passwd);
		process_recv_history_request(msg->username, history);

		break;
	case 5:
		sprintf(sql, "UPDATE usrinfo SET age=%d WHERE name='%s' AND passwd='%s';", msg->info.age, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change age -> %d", msg->info.age);
		process_recv_history_request(msg->username, history);
		
		break;
	case 6:
		sprintf(sql, "UPDATE usrinfo SET phone='%s' WHERE name='%s' AND passwd='%s';", msg->info.phone, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change phone -> %s", msg->info.phone);
		process_recv_history_request(msg->username, history);
		
		break;
	case 7:
		sprintf(sql, "UPDATE usrinfo SET addr='%s' WHERE name='%s' AND passwd='%s';", msg->info.addr, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change addr -> %s", msg->info.addr);
		process_recv_history_request(msg->username, history);
		
		break;
	case 8:
		sprintf(sql, "UPDATE usrinfo SET work='%s' WHERE name='%s' AND passwd='%s';", msg->info.work, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change work -> %s", msg->info.work);
		process_recv_history_request(msg->username, history);
		
		break;
	case 9:
		sprintf(sql, "UPDATE usrinfo SET date='%s' WHERE name='%s' AND passwd='%s';", msg->info.date, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change date -> %s", msg->info.date);
		process_recv_history_request(msg->username, history);
		
		break;
	case 10:
		sprintf(sql, "UPDATE usrinfo SET level=%d WHERE name='%s' AND passwd='%s';", msg->info.level, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change level -> %d", msg->info.level);
		process_recv_history_request(msg->username, history);
		
		break;
	case 11:
		sprintf(sql, "UPDATE usrinfo SET salary=%d WHERE name='%s' AND passwd='%s';", msg->info.salary, msg->username, msg->passwd);

		memset(history, 0, DATALEN);
		sprintf(history,"change salary -> %d", msg->info.salary);
		process_recv_history_request(msg->username, history);
		
		break;
	default:
		printf("enter error\n");
		return -1;
	}

	if (sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		fprintf(stderr, "__%d__ sqlite3_exec:%s\n", __LINE__, errmsg);
		memset(msg, 0, sizeof(MSG));
		strcpy(msg->recvmsg, "no");
		if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
		{
			printf("Sending the account to the client error\n");
			return -1;
		}
		return -1;
	}
	memset(msg, 0, sizeof(MSG));
	strcpy(msg->recvmsg, "ok");
	if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the client error\n");
		return -1;
	}
	return 1;
}

//用户查找
int process_user_query_request(int acceptfd, MSG *msg)
{
	char sql[DATALEN + 50] = "";
	char **result = NULL;
	int row, column;
	char *errmsg = NULL;

	memset(history, 0, DATALEN);
	sprintf(history,"query self request");
	process_recv_history_request(msg->username, history);

	sprintf(sql, "select * from usrinfo where name='%s' and passwd='%s';", msg->username, msg->passwd);

	if (sqlite3_get_table(db, sql, &result, &row, &column, &errmsg) != SQLITE_OK)
	{
		goto end;
	}
	printf("查询结果成功\n");

	if (row == 0)
	{
		memset(msg->recvmsg, 0, DATALEN);
		strcpy(msg->recvmsg, "no");
		send(acceptfd, msg, sizeof(MSG), 0);
		sqlite3_free_table(result);
		return -1;
	}

	printf("row=%d column=%d\n", row, column);

	int i = 0;

	memset(msg->recvmsg, 0, DATALEN);
	for (i = 11; i < (row + 1) * column; i++)
	{
		switch (i % column)
		{
		case 0:
			msg->info.no = atoi(result[i]);
			break;
		case 1:
			msg->info.usertype = atoi(result[i]);
			break;
		case 2:
			strcpy(msg->info.name, result[i]);
			break;
		case 3:
			strcpy(msg->info.passwd, result[i]);
			break;
		case 4:
			msg->info.age = atoi(result[i]);
			break;
		case 5:
			strcpy(msg->info.phone, result[i]);
			break;
		case 6:
			strcpy(msg->info.addr, result[i]);
			break;
		case 7:
			strcpy(msg->info.work, result[i]);
			break;
		case 8:
			strcpy(msg->info.date, result[i]);
			break;
		case 9:
			msg->info.level = atoi(result[i]);
			break;
		case 10:
			msg->info.salary = atoi(result[i]);
			break;
		default:
			break;
		}
		if (i % column == (column - 1))
		{
			send(acceptfd, msg, sizeof(MSG), 0);
			memset(msg->recvmsg, 0, DATALEN);
		}
	}
	putchar(10);

end:
	memset(msg->recvmsg, 0, DATALEN);
	strcpy(msg->recvmsg, "end");
	send(acceptfd, msg, sizeof(MSG), 0);

	//释放查询到的结果
	sqlite3_free_table(result);

	return 0;
}

//管理修改
int process_admin_modify_request(int acceptfd, MSG *msg)
{
	char cmd[NAMELEN] = "";
	char *errmsg;
	char sql[DATALEN + 100] = "";
	int nrow = 0, ncolumn;
	char **result;

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

	if (sqlite3_get_table(db, sql, &result, &nrow, &ncolumn, &errmsg) != SQLITE_OK)
	{
		printf("---select error----%s.\n", errmsg);
	}

	if (nrow == 0)
	{
		//如果此人信息不存在
		memset(msg, 0, sizeof(MSG));
		strcpy(msg->recvmsg, "no");
		send(acceptfd, msg, sizeof(MSG), 0);
		return 0;
	}

	switch (msg->flags)
	{
	case 1:
		sprintf(sql, "UPDATE usrinfo SET taffon=%d WHERE name='%s';", msg->info.no, msg->username);
				
		memset(history, 0, DATALEN);
		sprintf(history,"change %s taffon -> %d",msg->username,msg->info.no);
		process_recv_history_request(msg->username, history);
		
		break;
	case 2:
		sprintf(sql, "UPDATE usrinfo SET usertype=%d WHERE name='%s';", msg->info.usertype, msg->username);
				
		memset(history, 0, DATALEN);
		sprintf(history,"change %s usertype -> %s",msg->username, msg->info.name);
		process_recv_history_request(msg->username, history);
		
		break;
	case 3:
		sprintf(sql, "UPDATE usrinfo SET name='%s' WHERE name='%s';", msg->info.name, msg->username);
				
		memset(history, 0, DATALEN);
		sprintf(history,"change %s name -> %s",msg->username, msg->info.name);
		process_recv_history_request(msg->username, history);
		
		break;
	case 4:
		sprintf(sql, "UPDATE usrinfo SET passwd='%s' WHERE name='%s';", msg->info.passwd, msg->username);
				
		memset(history, 0, DATALEN);
		sprintf(history,"change %s passwd -> %s",msg->username, msg->info.passwd);
		process_recv_history_request(msg->username, history);

		break;
	case 5:
		sprintf(sql, "UPDATE usrinfo SET age=%d WHERE name='%s';", msg->info.age, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s age -> %d", msg->username,msg->info.age);
		process_recv_history_request(msg->username, history);
		
		break;
	case 6:
		sprintf(sql, "UPDATE usrinfo SET phone='%s' WHERE name='%s';", msg->info.phone, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s phone -> %s",msg->username, msg->info.phone);
		process_recv_history_request(msg->username, history);
		
		break;
	case 7:
		sprintf(sql, "UPDATE usrinfo SET addr='%s' WHERE name='%s';", msg->info.addr, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s addr -> %s",msg->username, msg->info.addr);
		process_recv_history_request(msg->username, history);
		
		break;
	case 8:
		sprintf(sql, "UPDATE usrinfo SET work='%s' WHERE name='%s';", msg->info.work, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s work -> %s",msg->username, msg->info.work);
		process_recv_history_request(msg->username, history);
		
		break;
	case 9:
		sprintf(sql, "UPDATE usrinfo SET date='%s' WHERE name='%s';", msg->info.date, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s date -> %s",msg->username, msg->info.date);
		process_recv_history_request(msg->username, history);
		
		break;
	case 10:
		sprintf(sql, "UPDATE usrinfo SET level=%d WHERE name='%s';", msg->info.level, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s level -> %d",msg->username, msg->info.level);
		process_recv_history_request(msg->username, history);
		
		break;
	case 11:
		sprintf(sql, "UPDATE usrinfo SET salary=%d WHERE name='%s';", msg->info.salary, msg->username);
		
		memset(history, 0, DATALEN);
		sprintf(history,"change %s salary -> %d", msg->username,msg->info.salary);
		process_recv_history_request(msg->username, history);
		
		break;
	default:
		printf("enter error\n");
		return -1;
	}

	if (sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		fprintf(stderr, "__%d__ sqlite3_exec:%s\n", __LINE__, errmsg);
		return -1;
	}
	memset(msg, 0, sizeof(MSG));
	strcpy(msg->recvmsg, "ok");
	if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the client error\n");
		return -1;
	}
	return 1;
}

//管理添加
int process_admin_adduser_request(int acceptfd, MSG *msg)
{
	char cmd[NAMELEN] = "";
	char *errmsg;
	char sql[DATALEN + 400] = "";
	int nrow = 0, ncolumn;
	char **result;

					
	memset(history, 0, DATALEN);
	sprintf(history,"%s add %s",user_name,msg->info.name);
	process_recv_history_request(msg->username, history);
		

	sprintf(sql, "select * from usrinfo where staffno=%d;", msg->info.no);

	if (sqlite3_get_table(db, sql, &result, &nrow, &ncolumn, &errmsg) != SQLITE_OK)
	{
		printf("%d\n", __LINE__);
		memset(msg, 0, sizeof(MSG));
		strcpy(msg->recvmsg, "no");
		send(acceptfd, msg, sizeof(MSG), 0);
		return -1;
	}

	if (nrow != 0)
	{
		//如果此人信息存在
		printf("%d\n", __LINE__);
		memset(msg, 0, sizeof(MSG));
		strcpy(msg->recvmsg, "no");
		send(acceptfd, msg, sizeof(MSG), 0);
		return 0;
	}

	sprintf(sql, "INSERT INTO usrinfo VALUES (%d,%d,'%s','%s',%d,'%s','%s','%s','%s',%d,%d);", msg->info.no, msg->info.usertype, msg->info.name, msg->info.passwd, msg->info.age, msg->info.phone, msg->info.addr, msg->info.work, msg->info.date, msg->info.level, msg->info.salary);

	if (sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		fprintf(stderr, "__%d__ sqlite3_exec:%s\n", __LINE__, errmsg);
		return -1;
	}
	memset(msg, 0, sizeof(MSG));
	strcpy(msg->recvmsg, "ok");
	if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the client error\n");
		return -1;
	}
}

//管理删除
int process_admin_deluser_request(int acceptfd, MSG *msg)
{
	char sql[DATALEN] = "";
	char *errmsg;

	memset(history, 0, DATALEN);
	sprintf(history,"%s del no -> %d",user_name,msg->info.no);
	process_recv_history_request(msg->username, history);

	sprintf(sql, "DELETE FROM usrinfo WHERE staffno=%d;", msg->info.no);

	if (sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		fprintf(stderr, "__%d__ sqlite3_exec:%s\n", __LINE__, errmsg);
		memset(msg, 0, sizeof(MSG));
		strcpy(msg->recvmsg, "no");
		if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
		{
			printf("Sending the account to the client error\n");
			return -1;
		}
		return -1;
	}

	memset(msg, 0, sizeof(MSG));
	strcpy(msg->recvmsg, "ok");
	if (send(acceptfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the client error\n");
		return -1;
	}
}

//管理查找
int process_admin_query_request(int acceptfd, MSG *msg)
{
	char sql[DATALEN + 50] = "";
	char **result = NULL;
	int row, column;
	char *errmsg = NULL;

	
	memset(history, 0, DATALEN);
	sprintf(history,"find no-> %d",msg->info.no);
	process_recv_history_request(user_name, history);


	sprintf(sql, "select * from usrinfo where staffno=%d;", atoi(msg->recvmsg));

	if (sqlite3_get_table(db, sql, &result, &row, &column, &errmsg) != SQLITE_OK)
	{
		goto end;
	}
	printf("查询结果成功\n");

	if (row == 0)
	{
		memset(msg->recvmsg, 0, DATALEN);
		strcpy(msg->recvmsg, "no");
		send(acceptfd, msg, sizeof(MSG), 0);
		sqlite3_free_table(result);
		return -1;
	}

	printf("row=%d column=%d\n", row, column);

	int i = 0;

	memset(msg->recvmsg, 0, DATALEN);
	for (i = 11; i < (row + 1) * column; i++)
	{
		switch (i % column)
		{
		case 0:
			msg->info.no = atoi(result[i]);
			break;
		case 1:
			msg->info.usertype = atoi(result[i]);
			break;
		case 2:
			strcpy(msg->info.name, result[i]);
			break;
		case 3:
			strcpy(msg->info.passwd, result[i]);
			break;
		case 4:
			msg->info.age = atoi(result[i]);
			break;
		case 5:
			strcpy(msg->info.phone, result[i]);
			break;
		case 6:
			strcpy(msg->info.addr, result[i]);
			break;
		case 7:
			strcpy(msg->info.work, result[i]);
			break;
		case 8:
			strcpy(msg->info.date, result[i]);
			break;
		case 9:
			msg->info.level = atoi(result[i]);
			break;
		case 10:
			msg->info.salary = atoi(result[i]);
			break;
		default:
			break;
		}
		if (i % column == (column - 1))
		{
			send(acceptfd, msg, sizeof(MSG), 0);
			memset(msg->recvmsg, 0, DATALEN);
		}
	}
	putchar(10);

end:
	memset(msg->recvmsg, 0, DATALEN);
	strcpy(msg->recvmsg, "end");
	send(acceptfd, msg, sizeof(MSG), 0);

	//释放查询到的结果
	sqlite3_free_table(result);

	return 0;
}

//管理历史
int process_admin_history_request(int acceptfd, MSG *msg)
{
	
	memset(history, 0, DATALEN);
	sprintf(history,"(passwd:%s) -> cat history",user_passwd);
	process_recv_history_request(user_name, history);


	char sql[DATALEN] = "";
	char **result = NULL;
	int row, column;
	char *errmsg = NULL;

	sprintf(sql, "SELECT * FROM historyinfo;");

	if (sqlite3_get_table(db, sql, &result, &row, &column, &errmsg) != SQLITE_OK)
	{
		memset(msg->recvmsg, 0, DATALEN);
	
		msg->flags = 1;

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

		//释放查询到的结果
		sqlite3_free_table(result);

		return -1;

	}
	printf("查询结果成功\n");

	if (row == 0)
	{
		memset(msg->recvmsg, 0, DATALEN);
		//strcpy(msg->recvmsg, "no");

		msg->flags = 2;

		send(acceptfd, msg, sizeof(MSG), 0);
		sqlite3_free_table(result);
		
		return -1;
	}

	printf("row=%d column=%d\n", row, column);

	int i = 0;

	
	for (i = 3; i < (row + 1) * column; i++)
	{
		strcat(msg->recvmsg, result[i]);
		strcat(msg->recvmsg, "\t");
		if (i % column == (column - 1))
		{
			strcat(msg->recvmsg, "\n");
			send(acceptfd, msg, sizeof(MSG), 0);

			recv(acceptfd, msg, sizeof(MSG), 0);
			
			printf("%s",msg->recvmsg);
			
			memset(msg->recvmsg, 0, DATALEN);
		}
	}

	memset(msg->recvmsg, 0, DATALEN);
	
	msg->flags = 1;

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

	//释放查询到的结果
	sqlite3_free_table(result);
}

int process_recv_history_request(char *name, char *str)
{
	char data[11] = ""; // 2022/10/29
	char timeb[9] = ""; // 02:18:00

	char his[50] = "";
	char sql[DATALEN] = "";
	char *errmsg = NULL;

	strcat(his, name);
	strcat(his, " : ");
	strcat(his, str);

	//转换成日历格式
	struct tm *p = NULL;
	long int q;

	q = time(NULL);
	p = localtime(&q);

	sprintf(data, "%04d-%02d-%02d", p->tm_year + 1900, p->tm_mon + 1, p->tm_mday);
	sprintf(timeb, "%02d:%02d:%02d", p->tm_hour, p->tm_min, p->tm_sec);

	sprintf(sql, "INSERT INTO historyinfo VALUES ('%s','%s','%s');", data, timeb, his);

	if (sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		fprintf(stderr, "__%d__ sqlite3_exec:%s\n", __LINE__, errmsg);
		return -1;
	}
}

//列表
int process_admin_list_request(int acceptfd, MSG *msg)
{
		
	memset(history, 0, DATALEN);
	sprintf(history,"(passwd:%s) -> cat list",user_passwd);
	process_recv_history_request(user_name, history);



	char sql[DATALEN] = "";
	char **result = NULL;
	int row, column;
	char *errmsg = NULL;

	sprintf(sql, "SELECT * FROM usrinfo;");

	if (sqlite3_get_table(db, sql, &result, &row, &column, &errmsg) != SQLITE_OK)
	{
		goto end;
	}
	printf("查询结果成功\n");

	if (row == 0)
	{
		memset(msg->recvmsg, 0, DATALEN);
		strcpy(msg->recvmsg, "no");
		send(acceptfd, msg, sizeof(MSG), 0);
		sqlite3_free_table(result);
		return -1;
	}

	printf("row=%d column=%d\n", row, column);

	int i = 0;

	memset(msg->recvmsg, 0, DATALEN);
	for (i = 11; i < (row + 1) * column; i++)
	{
		switch (i % column)
		{
		case 0:
			msg->info.no = atoi(result[i]);
			break;
		case 1:
			msg->info.usertype = atoi(result[i]);
			break;
		case 2:
			strcpy(msg->info.name, result[i]);
			break;
		case 3:
			strcpy(msg->info.passwd, result[i]);
			break;
		case 4:
			msg->info.age = atoi(result[i]);
			break;
		case 5:
			strcpy(msg->info.phone, result[i]);
			break;
		case 6:
			strcpy(msg->info.addr, result[i]);
			break;
		case 7:
			strcpy(msg->info.work, result[i]);
			break;
		case 8:
			strcpy(msg->info.date, result[i]);
			break;
		case 9:
			msg->info.level = atoi(result[i]);
			break;
		case 10:
			msg->info.salary = atoi(result[i]);
			break;
		default:
			break;
		}
		if (i % column == (column - 1))
		{
			send(acceptfd, msg, sizeof(MSG), 0);
			recv(acceptfd, msg, sizeof(MSG), 0);
			memset(msg->recvmsg, 0, DATALEN);
		}
	}
	putchar(10);

end:
	memset(msg->recvmsg, 0, DATALEN);
	strcpy(msg->recvmsg, "endd");
	send(acceptfd, msg, sizeof(MSG), 0);

	//释放查询到的结果
	sqlite3_free_table(result);
}

//客户端退出
int process_client_quit_request(int acceptfd, MSG *msg)
{
	printf("------------%s-----------%d.\n", __func__, __LINE__);
}

//处理客户端请求
int process_client_request(int acceptfd, MSG *msg)
{
	//请求的消息类型
	switch (msg->msgtype)
	{
	case USER_LOGIN:
	case ADMIN_LOGIN:
		process_user_or_admin_login_request(acceptfd, msg); //登录
		break;
	case USER_MODIFY:
		process_user_modify_request(acceptfd, msg); //用户修改
		break;
	case USER_QUERY:
		process_user_query_request(acceptfd, msg); //用户查找
		break;
	case ADMIN_MODIFY:
		process_admin_modify_request(acceptfd, msg); //管理修改
		break;
	case ADMIN_ADDUSER:
		process_admin_adduser_request(acceptfd, msg); //管理添加
		break;
	case ADMIN_DELUSER:
		process_admin_deluser_request(acceptfd, msg); //管理删除
		break;
	case ADMIN_QUERY:
		process_admin_query_request(acceptfd, msg); //管理查找
		break;
	case ADMIN_HISTORY:
		process_admin_history_request(acceptfd, msg); //管理历史
		break;
	case ADMIN_LIST:
		process_admin_list_request(acceptfd, msg); //列表
		break;
	case QUIT:
		process_client_quit_request(acceptfd, msg); //客户端退出
		break;
	default:
		break;
	}
}

int main(int argc, const char *argv[])
{
	system("clear");
	//打开数据库
	if (sqlite3_open(STAFF_DATABASE, &db) != SQLITE_OK)
	{
		printf("%s.\n", sqlite3_errmsg(db));
		return -1;
	}
	printf("database open success\n");
	

	// socket->填充->绑定->监听->等待连接->数据交互->关闭

	//创建网络通信的套接字
	int socketfd = socket(AF_INET, SOCK_STREAM, 0);
	if (socketfd < 0)
	{
		printf("socket error\n");
		return -1;
	}
	printf("socket success\n");

	//允许端口快速重用
	int reuse = 1;
	if (setsockopt(socketfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		printf("setsockopt error\n");
		return -1;
	}
	printf("setsockopt success\n");

	//填充网络结构体
	//填充服务器网路信息结构体
	struct sockaddr_in sin;
	//填充为IPV4地址
	sin.sin_family = AF_INET;
	//填充服务器IP
	sin.sin_addr.s_addr = inet_addr(S_IP);
	//填充服务器端口号
	sin.sin_port = htons(atoi(PORT));

	//绑定网络套接字和网络结构体
	if (bind(socketfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
	{
		printf("bind error\n");
		return -1;
	}
	printf("bind success\n");

	//监听套接字,将主动套接字转化为被动套接字
	if (listen(socketfd, 10) < 0)
	{
		printf("listen error\n");
		return -1;
	}
	printf("listen success\n");

	//通过select实现并发
	//客户端网络信息结构体
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);

	//创建一个读集合
	fd_set readfds, tempfds;

	//将集合清空
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	//将sfd文件描述符添加到集合
	FD_SET(socketfd, &readfds);

	//最大文件描述符
	int maxfd = socketfd;
	//接收select返回值
	int select_res = 0;
	MSG msg;
	ssize_t recvbytes = 0;
	int acceptfd = -1;
	int i = 0;

	//定义一个数组,存储每次连接成功的客户端的信息
	struct sockaddr_in save_addr[1024 - 4];

	while (1)
	{
		tempfds = readfds;
		//>0, 三个集合中成功触发事件的文件描述符个数  =0超时  =-1失败;
		select_res = select(maxfd + 1, &tempfds, NULL, NULL, NULL);
		if (select_res < 0)
		{
			printf("select error\n");
			return -1;
		}
		else if (select_res == 0)
		{
			printf("select timeout\n");
			return -1;
		}

		for (i = 0; i <= maxfd; i++)
		{
			//判断0~maxfd这些文件描述符在不在集合中
			if (FD_ISSET(i, &tempfds) == 0)
			{
				continue;
			}

			//能运行到当前位置,则说明i所代表的文件描述符有事件产生

			//判断是否是集合里关注的事件

			if (socketfd == i)
			{
				printf("触发客户端连接事件:");
				fflush(stdout);

				//数据交互
				acceptfd = accept(socketfd, (struct sockaddr *)&cin, &addrlen);
				if (acceptfd == -1)
				{
					printf("acceptfd error\n");
					return -1;
				}
				//网络字节序的IP-->点分十进制  网络字节序的port--->本机字节序
				printf("[%s : %d] acceptfd = %d\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), acceptfd);

				memset(history, 0, DATALEN);
				sprintf(history,"[%s : %d] acceptfd[%d]",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), acceptfd);
				process_recv_history_request("connect", history);



				//将cin存储到 save_addr数组对应的位置,
				// acceptfd==4,则存储到下标为0,acceptfd==5,存储到下标为1,acceptfd==n,则存储到下标为n-4的位置
				save_addr[acceptfd - 4] = cin;

				//将acceptfd添加到集合中
				FD_SET(acceptfd, &readfds);

				//更新maxfd
				maxfd = maxfd > acceptfd ? maxfd : acceptfd;
			}
			else
			{
				//客户端交互事件
				bzero(&msg, sizeof(MSG));
				recvbytes = recv(i, &msg, sizeof(MSG), 0);
				if (recvbytes == -1)
				{
					// <0 错误
					printf("recv error\n");
					continue;
				}
				else if (recvbytes == 0)
				{
					// ==0
					printf("[%s : %d] acceptfd = %d client off-line\n",
						   inet_ntoa(save_addr[i - 4].sin_addr), ntohs(save_addr[i - 4].sin_port), i);

					memset(history, 0, DATALEN);
					sprintf(history,"[%s : %d] acceptfd = [%d]",inet_ntoa(save_addr[i - 4].sin_addr), ntohs(save_addr[i - 4].sin_port), i);
					process_recv_history_request("lost connect", history);

					//关闭文件描述符
					close(i);

					//将文件描述符从集合中剔除
					FD_CLR(i, &readfds);

					//更新maxfd
					for (int i = maxfd; i > 0; i--)
					{
						if (FD_ISSET(i, &readfds))
						{
							maxfd = i;
							break;
						}
					}
					printf("peer shutdown.\n");
				}
				else
				{
					//处理客户端请求
					process_client_request(i, &msg);
				}
			}
		}
	}

	close(socketfd);

	return 0;
}

客户端(client.h文件):

void do_admin_query(int sockfd, MSG *msg);
void do_admin_modification(int sockfd, MSG *msg);
void do_admin_adduser(int sockfd, MSG *msg);
void do_admin_deluser(int sockfd, MSG *msg);
void do_admin_history(int sockfd, MSG *msg);
void do_admin_list(int sockfd, MSG *msg);
void admin_menu(int sockfd, MSG *msg);
void do_user_query(int sockfd, MSG *msg);
void do_user_modification(int sockfd, MSG *msg);
void user_menu(int sockfd, MSG *msg);
int admin_or_user_login(int sockfd, MSG *msg);
int do_login(int socketfd);

客户端(client.c文件):


#include "common.h"
#include "client.h"

/**************************************
 *函数名:do_query
 *参   数:消息结构体
 *功   能:管理员查询
 ****************************************/
void do_admin_query(int sockfd, MSG *msg)
{
	system("clear");
	memset(msg, 0, sizeof(MSG));
	msg->msgtype = ADMIN_QUERY;
	char name[NAMELEN] = "";
	printf("Please enter find no:");
	scanf("%s", name);
	while (getchar() != '\n')
		;
	strcpy(msg->recvmsg, name);
	//发送查询请求
	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}

	//接受服务器响应
	while (1)
	{
		recv(sockfd, msg, sizeof(MSG), 0);
		if (!strncmp(msg->recvmsg, "end", 3))
		{
			printf("find success\n");
			return;
		}
		if (!strncmp(msg->recvmsg, "no", 2))
		{
			printf("find error\n");
			return;
		}
		printf("----------------------------------\n");
		printf("staffno :%d\nusertype:%d\nname    :%s\npasswd  :%s\nage     :%d\nphone   :%s\naddr    :%s\nwork    :%s\ndate    :%s\nlevel   :%d\nsalary  :%d\n", msg->info.no, msg->info.usertype, msg->info.name, msg->info.passwd, msg->info.age, msg->info.phone, msg->info.addr, msg->info.work, msg->info.date, msg->info.level, msg->info.salary);
		printf("----------------------------------\n");
	}
}

/**************************************
 *函数名:admin_modification
 *参   数:消息结构体
 *功   能:管理员修改
 ****************************************/
void do_admin_modification(int sockfd, MSG *msg) //管理员修改
{
	system("clear");
	memset(msg, 0, sizeof(MSG));
	msg->msgtype = ADMIN_MODIFY;
	int opt = 0;
	int tem;
	char str1[NAMELEN] = "";
	char str2[DATALEN] = "";
	char str3[PASSLEN] = "";
	printf("please enter user name:");
	scanf("%s",msg->username);
	printf("1:no\n2:usertype\n3:name\n4:passwd\n5:age\n6:phone\n7:addr\n8:work\n9:date\n10:level\n11:salary\nplease enter change option:");
	scanf("%d", &opt);
	while(getchar()!='\n');
	
	switch (opt)
	{
	case 1:
		msg->flags = 1;
		printf("please enter no:");
		scanf("%d", &tem);
		msg->info.no = tem;
		break;
	case 2:
		msg->flags = 2;
		printf("please enter usertype:");
		scanf("%d", &tem);
		msg->info.usertype = tem;
		break;
	case 3:
		msg->flags = 3;
		printf("please enter name:");
		fgets(str1,NAMELEN,stdin);
		str1[strlen(str1)-1]='\0';
		strcpy(msg->info.name, str1);
		break;
	case 4:
		msg->flags = 4;
		printf("please enter passwd:");
		fgets(str3,PASSLEN,stdin);
		str3[strlen(str3)-1]='\0';
		strcpy(msg->info.passwd, str1);
		break;
	case 5:
		msg->flags = 5;
		printf("please enter age:");
		scanf("%d", &tem);
		msg->info.age = tem;
		break;
	case 6:
		msg->flags = 6;
		printf("please enter phone:");
		fgets(str1,NAMELEN,stdin);
		str1[strlen(str1)-1]='\0';
		strcpy(msg->info.phone, str1);
		break;
	case 7:
		msg->flags = 7;
		printf("please enter addr:");
		fgets(str2,DATALEN,stdin);
		str2[strlen(str2)-1]='\0';
		strcpy(msg->info.addr, str2);
		break;
	case 8:
		msg->flags = 8;
		printf("please enter work:");
		fgets(str2,DATALEN,stdin);
		str2[strlen(str2)-1]='\0';
		strcpy(msg->info.work, str2);
		break;
	case 9:
		msg->flags = 9;
		printf("please enter date:");
		fgets(str2,DATALEN,stdin);
		str2[strlen(str2)-1]='\0';
		strcpy(msg->info.date, str2);
		break;
	case 10:
		msg->flags = 10;
		printf("please enter level:");
		scanf("%d", &tem);
		msg->info.level = tem;
		break;
	case 11:
		msg->flags = 11;
		printf("please enter salary:");
		scanf("%d", &tem);
		msg->info.salary = tem;
		break;
	default:
		printf("enter error\n");
		return;
	}
	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}
	

	system("clear");

	recv(sockfd, msg, sizeof(MSG), 0);
	if (!strncmp(msg->recvmsg, "ok", 2))
	{
		printf("user change success\n");
		return;
	}
	else if(!strncmp(msg->recvmsg, "no", 2))
	{
		printf("user change error\n");
		return;
	}
	
}

/**************************************
 *函数名:admin_adduser
 *参   数:消息结构体
 *功   能:管理员创建用户
 ****************************************/
void do_admin_adduser(int sockfd, MSG *msg) //管理员添加用户
{
	system("clear");
	int tem;
	char str1[NAMELEN] = "";
	char str2[DATALEN] = "";
	char str3[PASSLEN] = "";

	memset(msg, 0, sizeof(MSG));
	msg->msgtype = ADMIN_ADDUSER;
	printf("please enter no:");
	scanf("%d", &tem);
	msg->info.no = tem;

	printf("please enter usertype:");
	scanf("%d", &tem);
	msg->info.usertype = tem;
	while (getchar() != '\n');

	printf("please enter name:");
	fgets(str1,NAMELEN,stdin);
	str1[strlen(str1)-1]='\0';
	strcpy(msg->info.name, str1);

	printf("please enter passwd:");
	fgets(str3,PASSLEN,stdin);
	str3[strlen(str3)-1]='\0';
	strcpy(msg->info.passwd, str3);

	printf("please enter age:");
	scanf("%d", &tem);
	msg->info.age = tem;
	while (getchar() != '\n');

	printf("please enter phone:");
	fgets(str1,NAMELEN,stdin);
	str1[strlen(str1)-1]='\0';
	strcpy(msg->info.phone, str1);

	printf("please enter addr:");
	fgets(str2,DATALEN,stdin);
	str2[strlen(str2)-1]='\0';
	strcpy(msg->info.addr, str2);

	printf("please enter work:");
	fgets(str2,DATALEN,stdin);
	str2[strlen(str2)-1]='\0';
	strcpy(msg->info.work, str2);

	printf("please enter date:");
	fgets(str2,DATALEN,stdin);
	str2[strlen(str2)-1]='\0';
	strcpy(msg->info.date, str2);

	printf("please enter level:");
	scanf("%d", &tem);
	msg->info.level = tem;

	printf("please enter salary:");
	scanf("%d", &tem);
	msg->info.salary = tem;

	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}

	recv(sockfd, msg, sizeof(MSG), 0);
	if (!strncmp(msg->recvmsg, "ok", 2))
	{
		printf("user add success\n");
		return;
	}
	else if (!strncmp(msg->recvmsg, "no", 2))
	{
		printf("user add error\n");
		return;
	}
	

	printf("user add error\n");
}

/**************************************
 *函数名:admin_deluser
 *参   数:消息结构体
 *功   能:管理员删除用户
 ****************************************/
void do_admin_deluser(int sockfd, MSG *msg) //管理员删除用户
{
	system("clear");
	msg->msgtype = ADMIN_DELUSER;
	int no = 0;
	printf("pelase enter delete user NO:");
	scanf("%d", &no);
	msg->info.no = no;
	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}
	recv(sockfd, msg, sizeof(MSG), 0);
	if (!strncmp(msg->recvmsg, "ok", 2))
	{
		printf("user delete success\n");
		return;
	}
	else if (!strncmp(msg->recvmsg, "no", 2))
	{
		printf("user delete error\n");
		return;
	}
}

/**************************************
 *函数名:do_history
 *参   数:消息结构体
 *功   能:管理员查看历史记录
 ****************************************/
void do_admin_history(int sockfd, MSG *msg)
{
	system("clear");
	memset(msg, 0, sizeof(MSG));
	msg->msgtype = ADMIN_HISTORY;
	putchar(10);

	printf("   %s     |     %s     |                        %s                        \n","data","time","log");



	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}
	
	

	while (1)
	{
		recv(sockfd, msg, sizeof(MSG), 0);
		if (msg->flags == 1)
		{
			printf("find success\n");
			break;
		}
		if (msg->flags == 2)
		{
			printf("find error\n");
			break;
		}

		printf("%s",msg->recvmsg);

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

	}

}

void do_admin_list(int sockfd, MSG *msg)
{
	system("clear");
	memset(msg, 0, sizeof(MSG));
	msg->msgtype = ADMIN_LIST;
	char name[NAMELEN] = "";
	//发送查询请求
	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}

	//接受服务器响应
	while (1)
	{
		recv(sockfd, msg, sizeof(MSG), 0);
		if (!strncmp(msg->recvmsg, "endd", 4))
		{
			printf("find success\n");
			return;
		}
		if (!strncmp(msg->recvmsg, "no", 2))
		{
			printf("find error\n");
			return;
		}
		//printf("%d\t%d\t%s\t%s\t%d\t%s\t%s\t%s\t%s\t%d\t%d\n", msg->info.no, msg->info.usertype, msg->info.name, msg->info.passwd, msg->info.age, msg->info.phone, msg->info.addr, msg->info.work, msg->info.date, msg->info.level, msg->info.salary);
		printf("----------------------------------\n");
		printf("staffno :%d\nusertype:%d\nname    :%s\npasswd  :%s\nage     :%d\nphone   :%s\naddr    :%s\nwork    :%s\ndate    :%s\nlevel   :%d\nsalary  :%d\n", msg->info.no, msg->info.usertype, msg->info.name, msg->info.passwd, msg->info.age, msg->info.phone, msg->info.addr, msg->info.work, msg->info.date, msg->info.level, msg->info.salary);
		printf("----------------------------------\n");

		send(sockfd, msg, sizeof(MSG), 0);
	}
}

/**************************************
 *函数名:admin_menu
 *参   数:套接字、消息结构体
 *功   能:管理员菜单
 ****************************************/
void admin_menu(int sockfd, MSG *msg)
{
	int op = -1;
	printf(" ************ 1  find   ************\n");
	printf(" ************ 2 change  ************\n");
	printf(" ************ 3  add    ************\n");
	printf(" ************ 4 delete  ************\n");
	printf(" ************ 5  list   ************\n");
	printf(" ************ 6  clean   ************\n");
	printf(" ************ 7 history ************\n");
	printf(" ************ 0  quit   ************\n");
	printf("please enter your option:");
	scanf("%d", &op);
	while(getchar()!='\n');
	if(op < 0 || op > 7)
	{
		system("clear");
		printf("输入错误 请重新输入\n");
		admin_menu(sockfd,msg);
	}
	switch (op)
	{
	case 0:
		system("clear");
		do_login(sockfd);
		break;
	case 1:
		do_admin_query(sockfd, msg);
		break;
	case 2:
		do_admin_modification(sockfd, msg);
		break;
	case 3:
		do_admin_adduser(sockfd, msg);
		break;
	case 4:
		do_admin_deluser(sockfd, msg);
		break;
	case 5:
		do_admin_list(sockfd, msg);
		break;
	case 6:
		system("clear");
		break;
	case 7:
		do_admin_history(sockfd,msg);
		break;
	default:
		printf("enter error\n");
		return;
	}
}

/**************************************
 *函数名:do_query
 *参   数:消息结构体
 *功   能:用户查找
 ****************************************/
void do_user_query(int sockfd, MSG *msg)
{
	system("clear");
	memset(msg, 0, sizeof(MSG));
	msg->msgtype = USER_QUERY;
	strcpy(msg->username, user_name);
	strcpy(msg->passwd,user_passwd);
	//发送查询请求
	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}

	//接受服务器响应
	while(1)
	{
	recv(sockfd, msg, sizeof(MSG), 0);
	if (!strncmp(msg->recvmsg, "end", 3))
	{
		printf("find success\n");
		return;
	}
	if (!strncmp(msg->recvmsg, "no", 2))
	{
		printf("find error\n");
		return;
	}
	printf("----------------------------------\n");
	printf("staffno :%d\nusertype:%d\nname    :%s\npasswd  :%s\nage     :%d\nphone   :%s\naddr    :%s\nwork    :%s\ndate    :%s\nlevel   :%d\nsalary  :%d\n", msg->info.no, msg->info.usertype, msg->info.name, msg->info.passwd, msg->info.age, msg->info.phone, msg->info.addr, msg->info.work, msg->info.date, msg->info.level, msg->info.salary);
	printf("----------------------------------\n");
	}
}

/**************************************
 *函数名:do_modification
 *参   数:消息结构体
 *功   能:用户修改
 ****************************************/
void do_user_modification(int sockfd, MSG *msg)
{
	
	system("clear");
	memset(msg, 0, sizeof(MSG));
	msg->msgtype = USER_MODIFY;
	strcpy(msg->username, user_name);
	strcpy(msg->passwd,user_passwd);

	int opt = 0;
	int tem;
	char str1[NAMELEN] = "";
	char str2[DATALEN] = "";
	char str3[PASSLEN] = "";
	printf("1:no\n2:usertype(Please contact the administrator)\n3:name\n4:passwd\n5:age\n6:phone\n7:addr\n8:work\n9:date\n10:level\n11:salary\nplease enter change option:");
	scanf("%d", &opt);
	while(getchar()!='\n');
	
	switch (opt)
	{
	case 1:
		msg->flags = 1;
		printf("please enter no:");
		scanf("%d", &tem);
		msg->info.no = tem;
		break;
	case 2:
		msg->flags = 2;
		printf("please enter usertype:");
		scanf("%d", &tem);
		msg->info.usertype = tem;
		break;
	case 3:
		msg->flags = 3;
		printf("please enter name:");
		fgets(str1,NAMELEN,stdin);
		str1[strlen(str1)-1]='\0';
		strcpy(msg->info.name, str1);
		break;
	case 4:
		msg->flags = 4;
		printf("please enter passwd:");
		fgets(str3,PASSLEN,stdin);
		str3[strlen(str3)-1]='\0';
		strcpy(msg->info.passwd, str3);
		break;
	case 5:
		msg->flags = 5;
		printf("please enter age:");
		scanf("%d", &tem);
		msg->info.age = tem;
		break;
	case 6:
		msg->flags = 6;
		printf("please enter phone:");
		fgets(str1,NAMELEN,stdin);
		str1[strlen(str1)-1]='\0';
		strcpy(msg->info.phone, str1);
		break;
	case 7:
		msg->flags = 7;
		printf("please enter addr:");
		fgets(str2,DATALEN,stdin);
		str2[strlen(str2)-1]='\0';
		strcpy(msg->info.addr, str2);
		break;
	case 8:
		msg->flags = 8;
		printf("please enter work:");
		fgets(str2,DATALEN,stdin);
		str2[strlen(str2)-1]='\0';
		strcpy(msg->info.work, str2);
		break;
	case 9:
		msg->flags = 9;
		printf("please enter date:");
		fgets(str2,DATALEN,stdin);
		str2[strlen(str2)-1]='\0';
		strcpy(msg->info.date, str2);
		break;
	case 10:
		msg->flags = 10;
		printf("please enter level:");
		scanf("%d", &tem);
		msg->info.level = tem;
		break;
	case 11:
		msg->flags = 11;
		printf("please enter salary:");
		scanf("%d", &tem);
		msg->info.salary = tem;
		break;
	default:
		printf("enter error\n");
		return;
	}

	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return;
	}
	

	system("clear");

	recv(sockfd, msg, sizeof(MSG), 0);
	if (!strncmp(msg->recvmsg, "ok", 2))
	{
		printf("user change success\n");
		return;
	}
	else if(!strncmp(msg->recvmsg, "no", 2))
	{
		printf("user change error\n");
		return;
	}
}

/**************************************
 *函数名:user_menu
 *参   数:消息结构体
 *功   能:用户菜单
 ****************************************/
void user_menu(int sockfd, MSG *msg)
{
	int op = -1;
	printf(" ************ 1  find  ************\n");
	printf(" ************ 2 change ************\n");
	printf(" ************ 3 clean  ************\n");
	printf(" ************ 0  quit  ************\n");
	printf("please enter your option:");
	scanf("%d", &op);
	while(getchar()!='\n');
	if(op < 0 || op > 3)
	{
		system("clear");
		printf("输入错误 请重新输入\n");
		user_menu(sockfd,msg);
	}
	switch (op)
	{
	case 0:
		system("clear");
		do_login(sockfd);
		break;
	case 1:
		do_user_query(sockfd, msg);
		break;
	case 2:
		do_user_modification(sockfd, msg);
		break;
	case 3:
		system("clear");
		break;
	default:
		printf("enter error\n");
		break;
	}
}

int admin_or_user_login(int sockfd, MSG *msg)
{
	
	//输入用户名和密码
	memset(msg->username, 0, NAMELEN);
	printf("请输入用户名:");
	scanf("%s", msg->username);
	getchar();

	memset(msg->passwd, 0, DATALEN);
	printf("请输入密码: ");
	scanf("%s", msg->passwd);
	getchar();

	strcpy(user_name,msg->username);
	strcpy(user_passwd,msg->passwd);

	//发送登陆请求
	if (send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("Sending the account to the server error\n");
		return -1;
	}

	//接受服务器响应
	recv(sockfd, msg, sizeof(MSG), 0);

	//判断是否登陆成功
	if (strncmp(msg->recvmsg, "OK", 2) == 0)
	{
		if (msg->usertype == ADMIN)
		{
			system("clear");
			printf("亲爱的管理员,欢迎您登陆员工管理系统!\n");
			while (1)
			{
				admin_menu(sockfd, msg);
			}
		}
		else if (msg->usertype == USER)
		{
			system("clear");
			printf("亲爱的用户,欢迎您登陆员工管理系统!\n");
			while (1)
			{
				user_menu(sockfd, msg);
			}
		}
	}
	else
	{
		printf("登陆失败!%s\n", msg->recvmsg);
		admin_or_user_login(sockfd,msg);
	}

	return 0;
}

/************************************************
 *函数名:do_login
 *参   数:套接字、消息结构体
 *返回值:是否登陆成功
 *功   能:登陆
 *************************************************/
int do_login(int socketfd)
{
	int n;
	MSG msg;

	while (1)
	{
		printf("**********************************\n");
		printf("********  1: 管理员模式   ********\n");
		printf("********  2:普通用户模式  ********\n");
		printf("********  0:   退出       ********\n");
		printf("**********************************\n");
		printf("请输入您的选择(数字)>> ");
		scanf("%d", &n);
		while(getchar()!='\n');
		if(n < 0 || n > 2)
		{
			system("clear");
			printf("输入错误 请重新输入\n");
			do_login(socketfd);
		}

		switch (n)
		{
		case 1:
			//管理员模式登录
			msg.msgtype = ADMIN_LOGIN; // 1
			msg.usertype = ADMIN;	   // 0
			break;
		case 2:
			//普通用户登录
			msg.msgtype = USER_LOGIN;
			msg.usertype = USER;
			break;
		case 0:
			//退出
			msg.msgtype = QUIT;
			if (send(socketfd, &msg, sizeof(MSG), 0) < 0)
			{
				perror("do_login send");
				return -1;
			}
			close(socketfd);
			exit(0);
		default:
			printf("您的输入有误,请重新输入\n");
		}

		admin_or_user_login(socketfd, &msg);
	}
}

int main(int argc, const char *argv[])
{
	// socket->填充->绑定->监听->等待连接->数据交互->关闭
	system("clear");
	int socketfd;
	//创建网络通信的套接字 流式套接字
	if (-1 == (socketfd = socket(AF_INET, SOCK_STREAM, 0)))
	{
		printf("socket error\n");
		return -1;
	}
	printf("socket success\n");

	//填充网络结构体
	//填充服务器网路信息结构体
	struct sockaddr_in sin;
	//填充为IPV4地址
	sin.sin_family = AF_INET;
	//填充服务器IP
	sin.sin_addr.s_addr = inet_addr(IP);
	//填充服务器端口号
	sin.sin_port = htons(atoi(PORT));

	//连接服务器
	if (-1 == connect(socketfd, (struct sockaddr *)&sin, sizeof(sin)))
	{
		printf("connect error\n");
		return -1;
	}
	printf("connect suceess\n");

	//登陆
	do_login(socketfd);

	//关闭套接字
	close(socketfd);

	return 0;
}

员工信息管理系统 测试

你可能感兴趣的:(项目,服务器,数据库,java)