#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/wait.h>
#define ERR_MSG(msg) do{\
fprintf(stderr, "line: %d ", __LINE__);\
perror(msg);\
}while(0)
#define PORT 8888 //1024~49151
#define IP "192.168.8.177" //ifconfig查看自己的IP,如果实在配不上填127.0.0.1
typedef void (*sighandler_t)(int);
int deal_cli_msg(int newfd, struct sockaddr_in cin);
void handler(int sig)
{
while(waitpid(-1, NULL, WNOHANG) > 0);
return;
}
int main(int argc, const char *argv[])
{
sighandler_t s = signal(17, handler);
if(SIG_ERR == s)
{
ERR_MSG("signal");
return -1;
}
//创建流式套接字
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
//允许端口快速重用
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("允许端口快速重用成功\n");
//填充服务器的IP和端口到地址信息结构体上
//地址信息结构体根据地址族指定 man 7 ip
struct sockaddr_in sin;
sin.sin_family = AF_INET; //必须填AF_INET;
sin.sin_port = htons(PORT); //网络字节序的端口号1024~49151
sin.sin_addr.s_addr = inet_addr(IP); //网络字节序的IP地址
//将服务器的IP和端口绑定到套接字上; 必须绑定
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为被动监听状态
if(listen(sfd, 10) < 0)
{
ERR_MSG("listen");
return -1;
}
printf("listen success\n");
struct sockaddr_in cin; //存储客户端的地址信息结构体
socklen_t addrlen = sizeof(cin); //地址信息结构体的大小
int newfd = 0;
pid_t pid = 0;
while(1)
{
//父进程只负责处理连接
//获取完成连接的客户端信息,并生成一个新的文件描述符
newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
if(newfd < 0)
{
ERR_MSG("accept");
return -1;
}
printf("[%s:%d] newfd = %d\n", inet_ntoa(cin.sin_addr) ,ntohs(cin.sin_port),newfd);
//能运行到当前位置则代表了有客户端连接成功,需要创建一个子进程,用于与客户端进行交互;
pid = fork();
if(0 == pid)
{
//子进程运行该段代码,并且只运行与客户端交互的代码
close(sfd); //由于sfd用于连接,子进程用不到,可以直接关闭
deal_cli_msg(newfd, cin);
exit(0); //由于子进程只负责交互,所以当交互结束后,需要退出子进程
}
else if(pid > 0)
{
//父进程
close(newfd);
}
else
{
ERR_MSG("fork");
return -1;
}
}
close(sfd);
return 0;
}
int deal_cli_msg(int newfd, struct sockaddr_in cin)
{
char buf[128] = "";
ssize_t res = 0;
while(1)
{
bzero(buf, sizeof(buf));
//循环接收数据,阻塞方式接收
res = recv(newfd, buf, sizeof(buf), 0);
if(res < 0)
{
ERR_MSG("recv");
return -1;
}
else if(0 == res)
{
printf("[%s:%d] newfd = %d 客户端已退出\n", inet_ntoa(cin.sin_addr) ,ntohs(cin.sin_port),newfd);
break;
}
printf("[%s:%d] newfd = %d:%s\n", inet_ntoa(cin.sin_addr) ,ntohs(cin.sin_port),newfd, buf);
//发送数据,将对方发送过来的数据发送回去
strcat(buf, "*_*");
if(send(newfd, buf, sizeof(buf), 0) < 0)
{
ERR_MSG("send");
return -1;
}
printf("发送成功\n");
}
close(newfd);
return 0;
}
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#define ERR_MSG(msg) do{\
fprintf(stderr, "line: %d ", __LINE__);\
perror(msg);\
return -1;\
}while(0)
#define PORT 8888
#define IP "192.168.8.177"
struct msg
{
int newfd;
struct sockaddr_in cin;
};
void* deal_cli_msg(void* arg);
int main(int argc, const char *argv[])
{
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
ERR_MSG("socket");
int reuse = 1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
ERR_MSG("setsockopt");
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
ERR_MSG("bind");
if(listen(sfd,128)<0)
ERR_MSG("listen");
struct sockaddr_in cin;
socklen_t addrlen = sizeof(cin);
int newfd = 0;
pthread_t tid;
struct msg info;
while(1)
{
newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
if(newfd<0)
ERR_MSG("accept");
info.newfd = newfd;
info.cin = cin;
if(pthread_create(&tid,NULL,deal_cli_msg,(void*)&info)!=0)
ERR_MSG("pthread_create");
}
close(sfd);
return 0;
}
void* deal_cli_msg(void* arg)
{
pthread_detach(pthread_self());
struct msg info = *(struct msg*)arg;
int newfd = info.newfd;
struct sockaddr_in cin = info.cin;
char buf[128]="";
ssize_t res = 0;
while(1)
{
bzero(buf,sizeof(buf));
res = recv(newfd,buf,sizeof(buf),0);
if(res<0)
{
perror("send");
pthread_exit(NULL);
}
else if(0==res)
{
printf("客户端已退出\n");
break;
}
printf("%s\n",buf);
strcat(buf,"*_*");
if(send(newfd,buf,sizeof(buf),0)<0)
{
perror("send");
pthread_exit(NULL);
}
printf("发送成功\n");
}
close(newfd);
pthread_exit(NULL);
}