将 epoll 服务器 客户端拿来用
客户端:写一个界面,里面有注册登录
服务器:处理注册和登录逻辑,注册的话将注册的账号密码写入数据库,登录的话查询数据库中是否存在账号,并验证密码是否正确
额外功能:客户端登录的时候,服务器向客户端发送一个验证码,只有验证码也正确的时候,才能登录成功·
server.c
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;
typedef struct sockaddr_un addr_un_t;
int sqllen;
enum Type{
TYPE_REGIST,
TYPE_LOGIN
};
typedef struct pack{
enum Type type;
char name[20];
char pwd[20];
int auth;
}pack_t;
void read_data(int client);
int callback(void* arg,int argc,char** argv,char** col);
void* thread_main(void* arg)//验证码验证
{
int client=*(int *)arg;
pack_t pack={0};
srand(time(0));
pack.auth=rand()%10000+1;
int test=pack.auth;
write(client,&pack,sizeof(pack));
read(client,&pack,sizeof(pack));
if(test==pack.auth)
{
printf("登录成功");
}else
{
printf("验证码错误\n");
}
}
int main(int argc, const char *argv[])
{
if(argc!=2)
{
printf("请输入端口号\n");
return 0;
}
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");
//为套接字绑定ip和port
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 event_server={.events=EPOLLIN, .data.fd=server};
struct epoll_event event_stdin={.events=EPOLLIN, .data.fd=0};
//添加到监视列表
epoll_ctl(epfd,EPOLL_CTL_ADD,server,&event_server);
epoll_ctl(epfd,EPOLL_CTL_ADD,0,&event_stdin);
//准备数组,存放激活的描述符
struct epoll_event arr[50]={0};
while(1)
{
int len=epoll_wait(epfd,arr,50,-1);
for(int i=0;i
client.c
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;
typedef struct sockaddr_un addr_un_t;
enum Type{
TYPE_REGIST,
TYPE_LOGIN
};
typedef struct pack{
enum Type type;
char name[20];
char pwd[20];
int auth;
}pack_t;
void* thread_main(void* arg)
{
int client=*(int *)arg;
pack_t pack;
read(client,&pack,sizeof(pack));
printf("验证码为:%d\n",pack.auth);
printf("请输入验证码\n");
scanf("%d",&pack.auth);
while(getchar()!=10);
write(client,&pack,sizeof(pack));
}
int main(int argc, const char *argv[])
{
if(argc!=2)
{
printf("请输入端口号\n");
return 0;
}
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.86");
if(connect(client,(addr_t*)&addr,sizeof(addr))==-1)
{
perror("connect");
return 1;
}
while(1)
{
int ch=-1;
printf("请选择\n");
printf("1:注册\n");
printf("2:登录\n");
printf("0:退出\n");
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.pwd);
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.pwd);
while(getchar()!=10);
pack.type = TYPE_LOGIN;
write(client,&pack,sizeof(pack));
pthread_t id;
pthread_create(&id,0,thread_main,&client);
pthread_detach(id);
break;
}
case 0:
{
exit(0);
}
}
}
return 0;
}