server.c
#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.77" //填本机IP ifconfig查看
#define PORT 6666 //1024 - 49151
int sfd;
int newfd = 0;
struct sockaddr_in cin;
char buf[128] = "";
ssize_t res = 0;
void *funcA(void *arg)
{
while(1)
{
bzero(buf, sizeof(buf));
//接收
res = recv(newfd, buf, sizeof(buf), 0);
if(res < 0)
{
ERR_MSG("recv");
return NULL;
}
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);
}
}
void *funcB(void *arg)
{
while(1)
{
//发送
bzero(buf, sizeof(buf));
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf)-1] = '\0';
res = send(newfd, buf, sizeof(buf), 0);
if(res < 0)
{
ERR_MSG("send");
return NULL;
}
printf("send success\n");
}
}
int main(int argc, const char *argv[])
{
//创建流式套接字
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地址的网络字节序
//更换网络环境后IP地址可能改变
//绑定服务器的IP和端口
if(bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为被动监听状态
if(listen(sfd, 5) < 0)
{
ERR_MSG("listen");
return -1;
}
printf("listen success\n");
socklen_t addrlen = sizeof(cin);
//从已经完成连接的队列获取一个客户端的信息,生成一个新的文件描述符
//生成的新的文件描述符才是与客户端通信的文件描述符
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);
pthread_t A, B;
//创建一个线程用于接受消息
if(pthread_create(&A, NULL, funcA, NULL) != 0)
{
fprintf(stderr, "pthread_create failed\n");
return -1;
}
//创建一个线程用于发送消息
if(pthread_create(&B, NULL, funcB, NULL) != 0)
{
fprintf(stderr, "pthread_create failed\n");
return -1;
}
pthread_join(A,NULL);
//关闭文件描述符
close(sfd);
close(newfd);
return 0;
}
client.c
#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.77" //填本机IP ifconfig查看
#define PORT 6666 //1024 - 49151
int sfd = 0;
char buf[128] = "";
ssize_t res = 0;
void *funcA(void *arg)
{
while(1)
{
bzero(buf, sizeof(buf));
//发送
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf)-1] = '\0';
write(sfd, buf, sizeof(buf));
if(strcmp(buf,"quit") == 0)
break;
}
}
void *funcB(void *arg)
{
while(1)
{
//接收
read(sfd, buf, sizeof(buf));
printf("应答消息[%s]\n", buf);
}
}
int main(int argc, const char *argv[])
{
//创建流式套接字
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地址的网络字节序
//更换网络环境后IP地址可能改变
//与服务器建立连接
if(connect(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
{
ERR_MSG("connect");
return -1;
}
printf("与服务器连接成功\n");
pthread_t A, B;
//创建一个线程用于发送消息
if(pthread_create(&A, NULL, funcA, NULL) != 0)
{
fprintf(stderr, "pthread_create failed\n");
return -1;
}
//创建一个线程用于接受消息
if(pthread_create(&B, NULL, funcB, NULL) != 0)
{
fprintf(stderr, "pthread_create failed\n");
return -1;
}
pthread_join(A, NULL);
//关闭文件描述符
close(sfd);
return 0;
}
运行结果如下: