程序功能
要求:编写一个网络聊天程序,要求采用数据流的套接口编程。程序分为服务器端与客户端。服务器端最大同时连接10个客户端。
服务器端可以响应多个客户端的请求,每个客户端之间可以相互通信,由服务器实现转发。服务器端显示所有客户端的通信,并有脏话过滤功能(比如用户输入“Fuck”,过滤后为“**”)。客户端通过用户名实现不同用户间通信(发送消息格式“用户名” “消息内容””)。
程序流程图
程序源代码
服务端
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <pthread.h>
#include <stdio.h>
#include <string.h>
#include <arpa/inet.h>
#define PORT 8848
#define BACKLOG 10
#define MAXDATASIZE 128
#define MAXUSER 9
void fuck_filter(const char* buf);
int select_user(char uname);
void process_cli(int connectfd, struct sockaddr_in client);
void* creat_conn(void* arg);
struct ARG {
int connfd;
struct sockaddr_in client;
};
struct USER {
int connfd;
char name;
}user[MAXUSER+1];
static int u_len = 0;
/////////////////////////////////
int main(void)
{
int listenfd, connectfd;
pthread_t thread; //id of thread
struct ARG *arg;
struct sockaddr_in server; //server's address info
struct sockaddr_in client; //client's
int sin_size;
int opt;
//create tcp socket
printf("socket.... ");
if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
perror("creating socket failed.");
exit(1);
}
opt = SO_REUSEADDR;
setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
/*setsockopt()用来设置参数s所指定的socket状态。
参数level代表欲设置的网络层,一般设成SOL_SOCKET以存取socket层。
参数optname代表欲设置的选项,有下列几种数值:SO_REUSEADDR 允许在bind()过程中本地地址可重复使用
参数 optval代表欲设置的值,参数optlen则为optval的长度。
返回值 成功则返回0,若有错误则返回-1,错误原因存于errno。*/
bzero(&server,sizeof(server));
server.sin_family = AF_INET;
server.sin_port = htons(PORT);
server.sin_addr.s_addr = htonl(INADDR_ANY);
printf("bind.... ");
if(bind(listenfd,(struct sockaddr *)&server,sizeof(struct sockaddr)) == -1){
perror("bind error.");
exit(1);
}
printf("listen.... ");
if(listen(listenfd,BACKLOG) == -1) {
perror("listen() error ");
exit(1);
}
sin_size = sizeof(struct sockaddr_in);
sleep(1);
while(1){
//accept() using main thread
printf("accepting....\n ");
if((connectfd = accept(listenfd,(struct sockaddr *)&client,(socklen_t*)&sin_size)) == -1) {
perror("accept() error ");
exit(1);
}
//超过系统最大用户容量的处理???
printf("cread ptheard arg\n");
arg = malloc(sizeof(struct ARG));
arg->connfd = connectfd;
memcpy((void *)&arg->client, &client, sizeof(client));
//invoke start_routine to handle this thread
if(pthread_create(&thread, NULL, creat_conn, (void*)arg)){
perror("pthread_creat() error");
exit(1);
}
}
close(listenfd);
return (0);
}
//////////////////////////
void process_cli(int connectfd, struct sockaddr_in client)
{
int num,ufd,i;
int tid = pthread_self();
char recvbuf[MAXDATASIZE], sendbuf[MAXDATASIZE];
char cli_name[MAXDATASIZE];
char name;
printf("IP: %s. \n",inet_ntoa(client.sin_addr) );
//get client's name from client
user[u_len].connfd = connectfd;
user[u_len].name = 'a' + u_len;
printf("u_fd:%d, u_name:%c\n",user[u_len].connfd,user[u_len].name);
//******???
if(u_len++ > MAXUSER) u_len = 0;
while(1){
memset(recvbuf,0,MAXDATASIZE);
num = recv(connectfd, recvbuf, MAXDATASIZE, 0);
printf("num=%d\n",num);
if(num == 0 || num == -1) {
close(connectfd);
perror("Client disconnected. ");
break;
}
printf("[%d]INFO: %s \n", tid, recvbuf );
//脏话过滤
fuck_filter(recvbuf);
//选择用户
name = recvbuf[0];
//群发处理
if(name == 'x')
{
for(i=0;i<= MAXUSER;i++)
{
if(user[i].connfd>0)
send(user[i].connfd, recvbuf, strlen(recvbuf), 0);
}
}
ufd = select_user(name);//得到用户的socket
if(ufd > 0){
//通过ufd,向指定用户发送
send(ufd, recvbuf, strlen(recvbuf), 0);
}
}
}
//////////////////////////////
int select_user(char uname)
{
int ufd = 0;
int i;
for(i=0;i<= MAXUSER;i++)
{
if(uname==user[i].name)
return user[i].connfd;
}
return ufd;
}
/////////////////////////////
void* creat_conn(void* arg)
{
int tid = pthread_self();
struct ARG * n_arg = arg;
// printf("\nTid[%d]", tid);
process_cli(n_arg->connfd, n_arg->client);
}
///////////////////////////
void fuck_filter(const char* buf)
{
char buf_fuck[] = "fuck";
char * p = NULL;
do{
p = strstr(buf,buf_fuck);
// printf("==%d==",p);
if(p != 0){
*p = '*';
*(++p) = '*';
*(++p) = '*';
*(++p) = '*';
}
}while(p != 0);
}
客户端
/* cthread.c */
#include <stdio.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <stdlib.h>
#include <pthread.h>
#define PORT 8848
#define MAXDATASIZE 128
void* recv_proc(void* arg);
void* send_proc(void* arg);
int get_line(char *msg);
int main(int argc, char *argv[])
{
int fd;
pthread_t th1,th2;
struct hostent *he;
struct sockaddr_in server; //server's address info
char *defaultIP = "127.0.0.1";
//???
if(argc != 2) {
if((he = gethostbyname(defaultIP)) == NULL){
perror("gethostbyname() error");
exit(1);
}
}else{
if((he = gethostbyname(argv[1])) == NULL){
perror("gethostbyname() error");
exit(1);
}
}
if((fd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
perror("socket() error");
exit(1);
}
bzero(&server , sizeof(server));
server.sin_family = AF_INET;
server.sin_port = htons(PORT);
server.sin_addr = *((struct in_addr *)he->h_addr);
if(connect(fd, (struct sockaddr *)&server,sizeof(struct sockaddr)) == -1){
perror("connect() error");
exit(1);
}
if(pthread_create(&th1, NULL, recv_proc, (void*)&fd)){
perror("recv pthread_creat() error");
close(fd);
exit(1);
}
if(pthread_create(&th2, NULL, send_proc, (void*)&fd)){
perror("send pthread_creat() error");
close(fd);
exit(1);
}
pthread_join(th1,NULL);
pthread_join(th2,NULL);
return 0;
}
void* recv_proc(void* arg)
{
char recvline[MAXDATASIZE];
int numbytes;
int sockfd;
printf("recv from server. \n");
sockfd = *((int*)arg);
while(1){
if((numbytes = recv(sockfd, recvline, MAXDATASIZE, 0)) > 0){
recvline[numbytes] = '\0';
printf("\n[recvice:] %s\n", recvline);
//clean output buffer
memset(recvline, '0', strlen(recvline)-1);
}else{
//printf("recv error/null. ");
}
}
}
void* send_proc(void* arg)
{
char sendline[MAXDATASIZE];
int numbytes;
int len;
int sockfd;
printf("send to server. \n");
sockfd = *((int*)arg);
while(1)
{
//send message to server
//when the string is not NULL , send another!
while((len =get_line(sendline)) != 0){
sendline[len] = '\0';
send(sockfd, sendline, strlen(sendline), 0);
}
}
}
int get_line(char *msg)
{
int i=0;
char temp;
printf("Input message:");
fflush(stdout);
while (1){
temp = getchar();
if (temp == '\r' || temp == '\n') {return i ;}
msg[i]=temp;
if(msg[i]==13){
msg[i]=0;
break;
}
fflush(stdout);
i++;
}
}