C语言再学习 -- C语言搭建TCP服务器/客户端

TCP/UDP讲过~
参看：UNIX再学习 – TCP/UDP 客户机/服务器

这里记录一下可用的TCP服务器和客户端代码。
参看：用C语言搭建TCP服务器/客户端

一、TCP服务器

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <string.h>
#include <errno.h>//分别打印错误信息，函数名和行号到标准错误流中
#define ERR_MSG(msg) do{\fprintf(stderr,"__%s__ __%s__ __%d__",__FILE__,__func__,__LINE__);\perror(msg);\
}while(0)//循环只是为了去括号，无任何意义#define IP "000.000.000.000"  //ifconfig，ip根据自己的去设置int main(int argc, char const *argv[])
{//创建流式套接字int sfd = socket(AF_INET,SOCK_STREAM,0);if(sfd < 0){ERR_MSG("socket");return -1;}printf("sock success sfd=%d\n",sfd);//允许端口快速被覆盖重用。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 IPstruct sockaddr_in sin;//定义结构体变量，引出成员并填充地址sin.sin_family      = AF_INET;  //这里填ipv4协议sin.sin_port        = htons(6666);//端口号的网络字节序1024～49151,将端口号转换成网络字节序sin.sin_addr.s_addr = inet_addr(IP);//将ip地址的点分十进制转换成网络字节序，并存到sin下的sin_addr下的s_addr中//绑定服务器的地址信息//bind函数可以绑定地址信息到套接字文件中，参数分别是//指定要绑定到那个套接字上，填对应的文件描述符//通用地址结构体，真实的地址信息结构体根据地址族制定，绑定IP和端口号到服务器套接字上//真实的地址信息结构体的大小if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0){//将sin中得地址信息绑定到sfd套接字文件中(因为sin的变量类型不一样//需要强制类型转换),最后可以使用sizeof直接算出sin结构体变量的大小ERR_MSG("bind");return -1;}printf("bind success __%d__\n",__LINE__);//输出当前行号//将套接字设置为被动监听状态if(listen(sfd,128) < 0){//将sfd套接字设置为被监听状态，允许同时有128个客户端未完成链接ERR_MSG("listen");return -1;}printf("listen success __%d__\n",__LINE__);//链接成功后输出行号struct sockaddr_in cin;//存储客户端信息，定义存储地址信息结构体变量socklen_t addrlen = sizeof(cin);//定义结构体类型存取结构体长度//超时检测//struct timeval tm = {60,0};//if(setsockopt(sfd,SOL_SOCKET,SO_RCVTIMEO,&tm,sizeof(tm)));//生成新的文件描述符与客户端通信int newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);//定义整形变量，接收accept函数的返回值，从以完成的链接的sfd队列中获取一个客户端的//信息，生成一个新的cin文件描述符，该文件描述符是通信使用的文件描述副if(newfd < 0){if(11 == errno){printf("time out...\n");return -1;}ERR_MSG("accept");return -1;}printf("[%s:%d] newfd=%d 客户端链接成功 __%d__\n",\inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);//客户端链接成功后输出，转换后的ip地址和转换后的主机字节序，和行号char buf[128] = "";//定义字符型数组存取数据ssize_t res = 0;//定义结构提变量接收返回值while(1){bzero(buf,sizeof(buf));//将buf数组清零//接收res = recv(newfd,buf,sizeof(buf),0);//从得到accept返回值的newfd套接字中//接收缓冲区的数据，并存储在数组中，用sizeof算出指定读取字节最后用0表示以阻塞方式//读取if(res <0)//返回值小于0,读取失败{ERR_MSG("recv");return -1;}else if(0 == res)//返回值等于0,客户端下线{printf("[%s:%d] newfd=%d 客户端下线__%d__\n",\inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);//输出转换后的ip地址和主机字节序，和新的文件描述符break;}printf("[%s:%d] newfd=%d : %s __%d__\n",\inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf,__LINE__);//输出ip地址和主机字节序，新的文件描述符和buf中的内容//发送strcat(buf,"*_*");//在buf数组的末尾追加*_*，证明服务器以收到数据if(send(newfd,buf,sizeof(buf),0) < 0)//如果返回值小于0,发送失败{//send函数可以将but数组中的数据发送给accept函数中得到的文件描述符，//第三项是大小，第四项0是阻塞发送ERR_MSG("send");return -1;}printf("send success __%d__\n",__LINE__);//发送成功后显示数据}//关闭if(close(newfd) < 0)//关闭newfd套接字{ERR_MSG("close");//关闭失败返回-1,并输出错误内容return -1;}if(close(sfd) < 0)//关闭sfd套接字{ERR_MSG("close");return -1;}return 0;
}

二、TCP客户端

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <string.h>
#define ERR_MSG(msg) do{\fprintf(stderr,"__%s__ __%s__ __%d__",__FILE__,__func__,__LINE__);\perror(msg);\
}while(0)#define IP "000.000.000.000"  //ip地址根据自己的去设置
#define PORT 6666 //端口号 1024～49151int main(int argc, char const *argv[])
{//创建流式套接字int cfd = socket(AF_INET,SOCK_STREAM,0);//定义整形变量接收socket函数返回的文件描述副if(cfd < 0){ERR_MSG("socket");return -1;}printf("create socket success cfd=%d __%d__\n",cfd,__LINE__);//绑定客户端自身的地址信息结构体---》非必须绑定//若客户端没有绑定地址信息，//测操作系统会自动帮客户端绑定本机IP，以及49151～65535内的随机端口//填充服务器的地址信息结构体，给connect函数使用//要链接哪个服务器，就填哪个服务器的地址信息struct sockaddr_in sin;sin.sin_family      = AF_INET;  //这里填ipv4协议sin.sin_port        = htons(PORT);//端口号的网络字节序1024～49151,将端口号转换成网络字节序sin.sin_addr.s_addr = inet_addr(IP);//将ip地址的点分十进制转换成网络字节序，并存到sin下的sin_addr下的s_addr中//链接服务器if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)//当返回值小于0，失败{//通过connect函数链接服务器：将指定要链接的文件描述符绑定sin的结构体变量链接服务器，ERR_MSG("connect");return -1;}printf("connect success __%d__\n",__LINE__);//成功后返回行号char buf[128] = "";//定义字符型数组ssize_t res = 0;//定义结构体变量，接收返回值while(1){//从终端获取数据printf("请输入>>> ");fgets(buf,sizeof(buf),stdin);//从标准输入流中读取数据，存入到buf数组中buf[strlen(buf)-1] = 0;//将buf的最后一位‘\n’改为0//发送if(send(cfd,buf,sizeof(buf),0) < 0)//返回值小于0,发送失败{//通过cfd套接字文件描述符向缓冲区发送buf中的数据ERR_MSG("send");return -1;}printf("send success __%d__\n",__LINE__);//成功输出行号//接收bzero(buf,sizeof(buf));//将buf数组清零res = recv(cfd,buf,sizeof(buf),0);//通过该函数从该套接字文件描述符中接收数据，并存入buf数组中if(res < 0)//如果返回值小于0{ERR_MSG("recv");//接收失败return -1;}else if(0 == res)//返回之等于0，断线{printf("服务器离线\n");break;}printf("%s __%d__\n",buf,__LINE__);//接收成功输出buf数组中的内容和行号}//关闭if(close(cfd) < 0)//关闭cfd套接字文件描述符{ERR_MSG("close");return -1;}return 0;
}