Linux 下基于socket的简单网络聊天室
原文http://blog.sina.com.cn/s/blog_6f181ff8010150vf.html 经过亲测调试修改了部分内容
服务端程序 server.c
#include
#include // 包含套接字函数库
#include
#include // 包含AF_INET相关结构
#include // 包含AF_INET相关操作的函数
#include
#include
#include
#include
#include
#define PORT 8889
#define MYKEY 12345
#define SIZE 10240
int main()
{
int shmid;
char *shmaddr; //定义子进程共用的共享内存
shmid = shmget(MYKEY, SIZE, IPC_CREAT | 0600);
shmaddr= (char *) shmat(shmid, 0, 0);
if(shmid==-1)
{
printf("shmid error\n");
}
memset(shmaddr,0,SIZE);
int i=0;
char buf[100];
memset(buf,0,100);
int server_sockfd,client_sockfd;
int server_len,client_len;
struct sockaddr_in server_sockaddr,client_sockaddr;
server_sockfd = socket(AF_INET,SOCK_STREAM, 0); // 定义套接字类型
server_sockaddr.sin_family=AF_INET;
server_sockaddr.sin_port=htons(PORT);
server_sockaddr.sin_addr.s_addr=INADDR_ANY;
server_len=sizeof(server_sockaddr);
//允许重复使用本地地址和套接字绑定
int j=1;
setsockopt(server_sockfd,SOL_SOCKET,SO_REUSEADDR,&j,sizeof(j));
//绑定端口
if(bind(server_sockfd,(struct sockaddr *)&server_sockaddr,server_len)==-1)
{
perror("bind:");
exit(1);
}
if(listen(server_sockfd,5)==-1)
{
perror("listen:");
exit(1);
}
printf("Listening...\n");
client_len=sizeof(client_sockaddr);
pid_t ppid,pid;
while(1)
{
if((client_sockfd=accept(server_sockfd,(struct sockaddr *)&client_sockaddr,&client_len))==-1)
{
perror("accept error:");
exit(1);
}
printf("%s登录服务器\n",inet_ntoa(client_sockaddr.sin_addr));
ppid=fork();
if(ppid==-1)
{
printf("fork 1 failed:");
}
if(ppid==0) //子进程用于接收客户端信息并发送
{
pid=fork();
if(pid==-1)
{
printf("fork 2 failed:");
exit(1);
}
int recvbytes;
if(pid==0) //子子进程用于接收消息
{
while(1)
{
if((recvbytes=recv(client_sockfd,buf,100,0))==-1)
{
perror("read client_sockfd failed:");
}
// printf("recvbytes=%d\n",recvbytes);
usleep(10000);
printf("client send buf=%s\n",buf);
for(i=0;i<1000;i++)
{
if(*(shmaddr+100*i)==0)
{
strcpy(shmaddr+100*i,buf);
break;
}
}
}
}
if(pid>0) //子进程用于发送消息
{
while(1)
{
if(*(shmaddr+i*100)!=0)
{
// strcpy(&buf,shmaddr+100*i);
// buf++;
write(client_sockfd,shmaddr,SIZE);
// send(client_sockfd,buf,strlen(buf),0);
// printf("the server is send buf=%c",buf);
// printf("send client :%s\n",(shmaddr+i*100)) ;
i++;
}
}
}
}
if(ppid>0) //总父进程返回等待接收消息
{
close(client_sockfd);
}
}
} 客户端 client.c
#include
#include // 包含套接字函数库
#include
#include // 包含AF_INET相关结构
#include // 包含AF_INET相关操作的函数
#include
#include
#include
#define PORT 8889
#define IP_ADDR "127.0.0.1"
#define SIZE 10240
int main()
{
struct tm *timeptr;
time_t timeval;
char tm[50];
//(void)time(&timeval);
//printf("the date is %s\n",ctime(&timeval));
// printf("The time is %s\n",tm);
int sockfd; // 用于保存客户套接字标识符
int len; // 用于客户消息长度
struct sockaddr_in address; // 定义客户套接字地址
int result;
sockfd = socket(AF_INET,SOCK_STREAM, 0); // 定义套接字类型
address.sin_family = AF_INET; // 定义套接字地址中的域
address.sin_addr.s_addr = inet_addr(IP_ADDR); // 定义套接字地址
address.sin_port = htons(PORT); // 定义套接字端口
char buf[100]; // 定义要传送的消息
memset(buf,0,100);
char str[120]; //存贮输入的语句
char shmaddr[SIZE]; //接受服务器发送的全部聊天数据
int i=0;
char myname[100];
char say[10]={"[-_-]"};
char blank[10] = {" "};
printf("欢迎来到聊天室，请输入你的姓名:\n");
scanf("%s",myname);
len = sizeof(address);
result = connect(sockfd, (struct sockaddr *) &address, len); // 请求连接
if (result == -1)
{
perror("Connect failed");
return 1;
}
printf("%s成功登录服务器:\n",myname);
pid_t pid;
pid=fork();
if(pid==-1)
{
printf("fork failed");
}
int sendbytes=0;
if(pid==0) //子进程用于发送数据
{
while(1)
{
scanf("%s",str);
(void)time(&timeval);
strcpy(tm,ctime(&timeval));
strcpy(buf,myname); //姓名传入buf中
strcat(buf,blank);
strcat(buf,tm); //时间传入buf中
strcat(buf,say);
strcat(buf,str); //语句传入bufz中
if((sendbytes=write(sockfd, buf, 100))==-1)
{
perror("send to server failed:");
} // 向服务器传送消息
usleep(1000);
memset(buf,0,100);
memset(tm,0,50);
}
}
if(pid>0) //父进程用于接受消息并读取
{
while(1)
{
read(sockfd,shmaddr,SIZE);
// printf("server send shmaddr=%s\n",shmaddr);
if(*(shmaddr+i*100)!=0)
{
printf("%s\n",(shmaddr+i*100)) ;
i++;
printf("\n");
}
usleep(1000);
}
}
close(sockfd);
return 0;
} 在ubuntu14下编译运行通过
还有一些小问题，正在研究，初学，欢迎交流！