#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>
typedef void (*sighandler_t)(int);
#define ERR_MSG(msg) do{\
fprintf(stderr, "line:%d\n", __LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.8.61" //填你们自己的本机IP ifconfig查看
#define PORT 6666 //1024~49151
int deal_cli_msg(int newfd, struct sockaddr_in cin);
void handler(int sig)
{
while(waitpid(-1, NULL, WNOHANG) > 0);
}
int main(int argc, const char *argv[])
{
//捕获17号信号
sighandler_t s = signal(SIGCHLD, 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;
}
//允许端口快速重用--->允许程序退出后,端口号能被新的进程快速覆盖
//必须放在bind之前
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) <0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("允许端口快速重用成功\n");
//填充地址信息结构体
//根据地址族的不同,地址信息结构体不一致,AF_INET: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地址的网络字节序,ifconfig
//更换网络环境后,IP地址会改变
//绑定服务器的IP和端口
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为被动监听状态
if(listen(sfd, 128) < 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) //子进程中为真
{
//子进程中负责与客户端交互,recv send
close(sfd);
deal_cli_msg(newfd, cin);
//退出子进程
close(newfd);
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)
{
fprintf(stderr, "[%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("send success\n");
}
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\n", __LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.8.61" //填你们自己的本机IP ifconfig查看
#define PORT 6666 //1024~49151
//需要传入给分支线程的数据类型
struct msg
{
int newfd;
struct sockaddr_in cin;
};
void* deal_cli_msg(void* arg); //void* arg = &cli_info
int main(int argc, const char *argv[])
{
//创建流式套接字
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
//允许端口快速重用--->允许程序退出后,端口号能被新的进程快速覆盖
//必须放在bind之前
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) <0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("允许端口快速重用成功\n");
//填充地址信息结构体
//根据地址族的不同,地址信息结构体不一致,AF_INET: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地址的网络字节序,ifconfig
//更换网络环境后,IP地址会改变
//绑定服务器的IP和端口
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为被动监听状态
if(listen(sfd, 128) < 0)
{
ERR_MSG("listen");
return -1;
}
printf("listen success\n");
struct sockaddr_in cin; //存储获取到的客户端的地址信息
socklen_t addrlen = sizeof(cin); //真实地址信息结构体的大小
//从已经完成连接的队列头中获取一个客户端的信息,生成一个新的文件描述符
//生成的新的文件描述符才是与客户端通信的文件描述符
int newfd = 0;
pthread_t tid;
struct msg cli_info;
while(1)
{
//主线程只负责连接
//accept函数会先预选一个文件描述符,
//如果解除阻塞的时候该文件描述符没有被使用,则直接占用该预选的文件描述符
//否则重新遍历查找。
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);
cli_info.newfd = newfd;
cli_info.cin = cin;
//有客户端连接成功
//则需要创建一个分支线程,用于与客户端交互
if(pthread_create(&tid, NULL, deal_cli_msg, &cli_info) != 0)
{
fprintf(stderr, "line:%d pthread_create error\n", __LINE__);
break;
}
//分离线程
//pthread_detach(tid);
}
close(sfd);
return 0;
}
//线程处理函数
void* deal_cli_msg(void* arg) //void* arg = &cli_info
{
//分离线程
pthread_detach(pthread_self());
//必须另存,如果不另存,会让多个线程访问一块空间,倒置数据可能被覆盖修改
int newfd = ((struct msg*)arg)->newfd;
struct sockaddr_in cin = ((struct msg*)arg)->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");
break;
}
else if(0 == res)
{
fprintf(stderr, "[%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");
break;
}
printf("send success\n");
}
close(newfd);
pthread_exit(NULL);
}