프로젝트: 채팅방(Tcp Select 구조) - 서버
64299 단어 프로젝트
#ifndef _CHATROOM1_H_
#define _CHATROOM1_H_
//************
#include
#include
#include
#include
//************
#define PORT 8888
#define SUCCESS 10000000
#define FAILURE 10000001
#define RESIGER 1
#define LOGIN 2
#define OK 3 //
#define ERROR 4
#define FIND 5
#define SAME 6
#define SEND 7 //
#define UN 8 //
#define MSG 9 //
#define NUM 10 //
#define EXIT 11
#define GROUP 12
#define ERR 13 //
#define ES 14 //
#define PS 15 //
#define VIP 16
#define TALK 17 //
#define SHUTUP 18 //
#define OKK 19
#define OUTLINE 20
#define OKKK 21
#define SENDFILE 22
#define UNRE 23
#define REOK 24
//**************
pthread_t tid = 0;
//*************
struct online //
{
int sockfd;
char name[20];
struct online* next;
};
typedef struct online* LinkList;
struct senduser //
{
char id[20];
char name[20];
char toname[20];
char password[20];
char code[20]; //
char msg[200];
char onlinenum[30][30]; //
char filename[20];
char filemsg[200];
int e_s;
int p_s;
int vip; //1 vip 0
int state; //1 0
int number; //
int cmd;
};
#endif
서버:
#include "chatroom1.h"
int InitList(LinkList *l)
{
(*l) = (LinkList)malloc(sizeof(struct online));
(*l)->next = NULL;
}
int main()
{
//******************* ****************
sqlite3 *ppdb;
char sql[200] = {0};
int opt = 1, i = 0, j;
int sockfd, ret, fd[100] = {0};
struct senduser Recvbuf = {0};
struct sockaddr_in server_addr;
struct sockaddr_in client_addr;
fd_set Readfd, Tmpfd;
int maxfd;
sockfd = socket(PF_INET, SOCK_STREAM, 0);
if(sockfd == -1)
{
perror("socket");
exit(1);
}
bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = PF_INET;
server_addr.sin_port = htons(PORT);
server_addr.sin_addr.s_addr = inet_addr("192.168.1.119");
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
printf("start server!
");
//************ ******************
LinkList online;
InitList(&online);
//****************** *****************************************
ret = sqlite3_open("user.db",&ppdb);
if(ret != SQLITE_OK)
{
perror("open");
exit(1);
}
printf("sqlite3 success!
");
sprintf(sql, "create table if not exists user(id text primary key, password text, name text, code text, vip integer, state integer);");
ret = sqlite3_exec(ppdb, sql, NULL, NULL, NULL);
if(ret != SQLITE_OK)
{
perror("exec1");
exit(1);
}
printf("create user db success!
");
//****************** ***************************************
ret = bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr));
if(-1 == ret)
{
perror("bind");
exit(1);
}
ret = listen(sockfd, 8);
if(-1 == ret)
{
perror("listen");
exit(1);
}
FD_ZERO(&Readfd);
FD_ZERO(&Tmpfd);
FD_SET(sockfd, &Readfd);
maxfd = sockfd;
while(1)
{
Tmpfd = Readfd;
ret = select(maxfd + 1, &Tmpfd, NULL, NULL, NULL);
if(-1 == ret)
{
perror("select");
exit(1);
}
//**************************** ************************************
if(FD_ISSET(sockfd, &Tmpfd))
{
int length = sizeof(client_addr);
fd[i] = accept(sockfd, (struct sockaddr*)&client_addr, &length);
if(fd[i] < 0)
{
perror("accept");
exit(1);
}
printf("Accept %d, Port %d
", fd[i], client_addr.sin_port);
FD_SET(fd[i], &Readfd);
maxfd = fd[i];
i++;
}
else
{
for(j = 0; j < i; j++)
{
if(FD_ISSET(fd[j], &Tmpfd))
{
ret = recv(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret < 0)
{
perror("recv");
return FAILURE;
}
//************************************ ************************
if(RESIGER == Recvbuf.cmd)
{
memset(sql, 0, sizeof(sql));
sprintf(sql, "insert into user(id, password, name, code, vip, state) values ('%s', '%s', '%s', '%s', '%d', '%d');", Recvbuf.id, Recvbuf.password, Recvbuf.name, Recvbuf.code, Recvbuf.vip, Recvbuf.state);
ret = sqlite3_exec(ppdb, sql, NULL, NULL, NULL);
if(ret != SQLITE_OK)
{
perror("exec2");
Recvbuf.cmd = UNRE;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
}
else
{
Recvbuf.cmd = REOK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
}
}
//************************************ *********************************
if(Recvbuf.cmd == LOGIN)
{
char **Result = NULL;
int nrow;
memset(sql, 0, sizeof(sql));
sprintf(sql, "select id, password, name, vip, state from user where id = '%s' and password = '%s';", Recvbuf.id, Recvbuf.password);
ret = sqlite3_get_table(ppdb, sql, &Result, &nrow, NULL, NULL);
if(ret != SQLITE_OK)
{
perror(" ");
return FAILURE;
}
if(nrow == 1)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Result[7]) == 0)
{
Recvbuf.cmd = SAME;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("samesend");
return FAILURE;
}
break;
}
m = m->next;
}
if(Recvbuf.cmd != SAME)
{
strcpy(Recvbuf.name, Result[7]);
Recvbuf.vip = *(Result[8]) - 48;
Recvbuf.state = *(Result[9]) - 48;
LinkList s = (LinkList)malloc(sizeof(struct online));
strcpy(s->name ,Recvbuf.name);
s->sockfd = fd[j];
s->next = online->next;
online->next = s;
Recvbuf.cmd = OK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("Loginsend");
return FAILURE;
}
}
}
else
{
Recvbuf.cmd = ERROR;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("loginsend2");
return FAILURE;
}
}
}
//*********************************** *********************
if(Recvbuf.cmd == FIND)
{
char **Result = NULL;
int nrow;
memset(sql, 0, sizeof(sql));
sprintf(sql, "select id, code from user where id = '%s' and code = '%s';", Recvbuf.id, Recvbuf.code);
ret = sqlite3_get_table(ppdb, sql, &Result, &nrow, NULL, NULL);
if(ret != SQLITE_OK)
{
perror(" 2");
return FAILURE;
}
if(nrow == 1)
{
Recvbuf.cmd = OK;
memset(sql, 0, sizeof(sql));
sprintf(sql, "update user set password = '%s' where id = '%s';", Recvbuf.password, Recvbuf.id);
ret = sqlite3_exec(ppdb, sql, NULL, NULL, NULL);
if(ret != SQLITE_OK)
{
perror(" ");
exit(1);
}
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("findsend1");
exit(1);
}
}
else
{
Recvbuf.cmd = ERROR;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("findsend");
return FAILURE;
}
}
}
//********************************** ************************
if(Recvbuf.cmd == SEND)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Recvbuf.toname) == 0)
{
break;
}
m = m->next;
}
if(m == NULL)
{
Recvbuf.cmd = UN;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv");
return FAILURE;
}
}
else
{
Recvbuf.cmd = MSG;
ret = send(m->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendmsg");
return FAILURE;
}
Recvbuf.cmd = OK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv2");
return FAILURE;
}
}
}
//********************************** *******************
if(Recvbuf.cmd == NUM)
{
int i = 0;
Recvbuf.number = 0;
LinkList m = online->next;
while(m != NULL)
{
Recvbuf.number++;
strcpy(Recvbuf.onlinenum[i], m->name);
i++;
m = m->next;
}
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendnum");
return FAILURE;
}
}
//********************************* *********************
if(Recvbuf.cmd == EXIT)
{
LinkList tmp = online;
while(tmp->next != NULL)
{
if(strcmp(Recvbuf.name, tmp->next->name) == 0)
{
LinkList p = tmp->next;
tmp->next = p->next;
free(p);
break;
}
tmp = tmp->next;
}
}
//******************************** *********************
if(Recvbuf.cmd == GROUP)
{
int flag = 0;
LinkList q = online->next;
while(q != NULL)
{
if(q->sockfd != fd[j])
{
flag = 1;
ret = send(q->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendgroup");
return FAILURE;
}
}
q = q->next;
}
if(flag == 0)
{
Recvbuf.cmd = ERR;
send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
}
}
//******************************** *************
if(Recvbuf.cmd == PS)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Recvbuf.toname) == 0)
{
break;
}
m = m->next;
}
if(m == NULL)
{
Recvbuf.cmd = UN;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv");
return FAILURE;
}
}
else
{
ret = send(m->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendmsg");
return FAILURE;
}
Recvbuf.cmd = OK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv2");
return FAILURE;
}
}
}
//********************** *******************************
if(Recvbuf.cmd == ES)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Recvbuf.toname) == 0)
{
break;
}
m = m->next;
}
if(m == NULL)
{
Recvbuf.cmd = UN;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv");
return FAILURE;
}
}
else
{
ret = send(m->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendmsg");
return FAILURE;
}
Recvbuf.cmd = OK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv2");
return FAILURE;
}
}
}
//******************** **********************
if(Recvbuf.cmd == VIP)
{
memset(sql, 0, sizeof(sql));
sprintf(sql, "update user set vip = '%d' where name = '%s';", Recvbuf.vip, Recvbuf.name);
ret = sqlite3_exec(ppdb, sql, NULL, NULL, NULL);
if(ret != SQLITE_OK)
{
perror("vip");
return FAILURE;
}
}
//***************** ******************************
if(Recvbuf.cmd == TALK)
{
memset(sql, 0, sizeof(sql));
sprintf(sql, "update user set state = '%d' where name = '%s';", Recvbuf.state, Recvbuf.name);
ret = sqlite3_exec(ppdb, sql, NULL, NULL, NULL);
if(ret != SQLITE_OK)
{
perror("talk");
return FAILURE;
}
}
//************************** *************************
if(Recvbuf.cmd == SHUTUP)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Recvbuf.toname) == 0)
{
break;
}
m = m->next;
}
if(m == NULL)
{
Recvbuf.cmd = UN;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv");
return FAILURE;
}
}
else
{
memset(sql, 0, sizeof(sql));
sprintf(sql, "update user set state = 1 where name = '%s';", Recvbuf.toname);
ret = sqlite3_exec(ppdb, sql, NULL, NULL, NULL);
if(ret != SQLITE_OK)
{
perror("shutup");
return FAILURE;
}
ret = send(m->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendshutup");
return FAILURE;
}
Recvbuf.cmd = OKK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("OK1");
return FAILURE;
}
}
}
//*********************** *******************************
if(Recvbuf.cmd == OUTLINE)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Recvbuf.toname) == 0)
{
break;
}
m = m->next;
}
if(m == NULL)
{
Recvbuf.cmd = UN;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendrecv");
return FAILURE;
}
}
else
{
ret = send(m->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendout");
return FAILURE;
}
LinkList tmp = online;
while(tmp->next != NULL)
{
if(strcmp(Recvbuf.toname, tmp->next->name) == 0)
{
LinkList p = tmp->next;
tmp->next = p->next;
free(p);
break;
}
tmp = tmp->next;
}
Recvbuf.cmd = OKKK;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("okkk");
return FAILURE;
}
}
}
//*********************** ************************
if(Recvbuf.cmd == SENDFILE)
{
LinkList m = online->next;
while(m != NULL)
{
if(strcmp(m->name, Recvbuf.toname) == 0)
{
break;
}
m = m->next;
}
if(m == NULL)
{
Recvbuf.cmd = UN;
ret = send(fd[j], &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendfile1");
return FAILURE;
}
}
else
{
ret = send(m->sockfd, &Recvbuf, sizeof(Recvbuf), 0);
if(ret == -1)
{
perror("sendfile2");
return FAILURE;
}
}
Recvbuf.cmd = 0;
}
//***************************************************************
}
}
}
}이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
[Docker] - 젠킨스 CI/CD (2) - 실습📖 젠킨스 서버(컨테이너)환경 및 Jobs 구성 요약 Jenkins Version : Jenkins 2.332.2 (Docker Img : jenkins/jenkins:lts - ) Installed App : Do...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.
좋은 웹페이지 즐겨찾기
