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

TCP/UDP讲过~
参看：UNIX再学习 – TCP/UDP 客户机/服务器
参看：UNIX再学习 – 网络IPC：套接字

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

一、TCP服务器

#include 
#include 
#include 
#include 
#include 
#include 
#include 
 
//分别打印错误信息，函数名和行号到标准错误流中
#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 IP
    struct 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 
#include 
#include 
#include 
#include 
#include 
#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～49151
 
int 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;
}