TCP套接字编程

对于socket不理解的可以查看： C语言通过socket编程实现TCP通信， Linux socket编程/socket详解

先看看程序效果图

提示如果服务器，直接通过ctrl+z关闭，这样程序占用的地址不会释放，需要以下操作

1.ps    //查看进程

2.kill -9 程序pid    //强制杀死进程

服务端：

客户端：

接下来是代码

服务端：

#include <sys/types.h>
#include <sys/socket.h> //包含套接字的函数库
#include <netinet/in.h> //包含AF_INET的相关结构
#include <arpa/inet.h>  //包含AF_INET的操作函数
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h> 

/* 
 监听后，一直处于accept阻塞状态， 直到有客户端连接， 
输入EOF后，断开与客户端的连接,输入quit关闭服务器
读取EOF后，断开和此客户端连接,读取quit关闭服务器
 */  

#define PORT 3333  
void main()
{
    printf("程序开始\n");
     
    int s_fd,c_fd;                  //服务器和客户端的套接字标识符
    int s_len,c_len;                //服务器和客户端的消息长度
    struct sockaddr_in s_addr;    //服务器套接字地址
    struct sockaddr_in c_addr;    //客户端套接字地
    int dataBytes=0;//读取消息长度
    //消息
    char sendbuf[BUFSIZ];  
    char recvbuf[BUFSIZ];  
    
    /***************************创建套接字**************************/
    //socket函数，失败返回-1  
    //int socket(int domain, int type, int protocol);  
    //第一个参数表示使用的地址类型，一般都是ipv4即AF_INET  
    //第二个参数表示套接字类型：tcp一般用SOCK_STREAM数据流传输
    //第三个参数设置为0 
    s_fd = socket(AF_INET,SOCK_STREAM,0);
    if(s_fd < 0){
        perror("创建套接字失败");
        return;
    }
    printf("创建套接字成功\n");
    /****************************开启地址复用*******************/
    int optval = 1;//1允许地址重用0禁止
    int optlen = sizeof(optval);
    setsockopt(s_fd,SOL_SOCKET,SO_REUSEADDR,&optval,optlen);
    /********************初始化服务器套接字*****************/
    //htons和htonl将端口和地址转成网络字节序
    //定义服务器中地址中的域，AF_INET指IPv4
    s_addr.sin_family = AF_INET;
    //定义服务端的套接字端口
    s_addr.sin_port = htons(PORT);
    //定义套接字地址
    //ip可是本服务器的ip,用宏INADDR_ANY代替，代表0.0.0.0，表明所有地址 
    s_addr.sin_addr.s_addr = htonl(INADDR_ANY);

    /****************绑定套接字设置的端口号和IP***********************/
    //对于bind，accept之类的函数，里面套接字参数都是需要强制转换成(struct sockaddr *)  
    //bind三个参数：服务器端的套接字的文件描述符
    s_len = sizeof(s_addr);//设置发送消息长度
    if( bind(s_fd,(struct sockaddr*)&s_addr,s_len) < 0){
        perror("绑定套接字失败");
        return;
    }
    printf("绑定套接字成功\n");
    /*****************设置服务器上的socket为监听状态*******************/
    //监听,最大连接数10 
    if(listen(s_fd,10) < 0){
        perror("监听失败");
        return;
    }
    printf("监听成功端口: %d\n", PORT);

    while(1){
        
        printf("等待连接...\n");
        //fflush(stdin)刷新标准输入缓冲区，fflush(stdout)刷新标准输出缓冲区
        fflush(stdout);
        //设置接收消息长度
        c_len = sizeof(c_addr);
        /********************接收客户端连接请求*************************/
        //调用accept函数后，会进入阻塞状态  
        //accept返回一个套接字的文件描述符，这样服务器端便有两个套接字的文件描述符，  
        //s_fd和c_fd 
        //s_fd仍然继续在监听状态，c_fd则负责接收和发送数据  
        //c_addr是一个传出参数，accept返回时，传出客户端的地址和端口号  
        //c_len是一个传入-传出参数，传入的是调用者提供的缓冲区的c_addr的长度，以避免缓冲区溢出。  
        //传出的是客户端地址结构体的实际长度。  
        //出错返回-1 
        c_fd = accept(s_fd,(struct sockaddr*)&c_addr,(socklen_t*)&c_len);
        if(c_fd < 0){
            perror("accept失败");
            continue;
        }

        printf("新连接：\n");
        //inet_ntoa   ip地址转换函数，将网络字节序IP转换为点分十进制IP  
        //表达式：char *inet_ntoa (struct in_addr); 
        printf("\tIP is %s\n", inet_ntoa(c_addr.sin_addr));  
        printf("\tPort is %d\n", htons(c_addr.sin_port));  
        printf("等待消息...\n");
        while(1){
            dataBytes=0;
            /*************************接收数据***********************/
            printf("--------------------读取:"); 
            fflush(stdout);
            dataBytes = recv(c_fd,recvbuf,BUFSIZ,0);
            if( dataBytes < 0){
                perror("读取失败");
                continue;
            }else if(dataBytes == 0){
                printf("无消息\n");
            }else
                printf("%s\n",recvbuf);

            //判断退出，quit，断开连接，关闭客户端
            if(strncmp(recvbuf,"quit",4) == 0){
                sprintf(recvbuf,"%s","EOF");
                send(c_fd,recvbuf,sizeof(recvbuf)+1,0);
                //关闭连接
                close(s_fd);
                printf("关闭服务器\n");
                printf("程序结束\n"); 
                return;
            }
            //EOF，断开连接
            if(strncmp(recvbuf, "EOF",3) == 0){
                break;
            }  
                
            /************************发送数据*************************/
            printf("--------------------发送:");     
            scanf("%s",sendbuf);
            send(c_fd,sendbuf,strlen(sendbuf)+1,0);
            
            //判断退出，quit，断开连接，关闭客户端
            if(strncmp(sendbuf,"quit",4) == 0){
                sprintf(sendbuf,"%s","EOF");
                send(c_fd,sendbuf,sizeof(sendbuf)+1,0);
                //关闭连接
                close(s_fd);
                printf("程序结束\n"); 
                return;
            }
            //EOF，断开连接
            if(strncmp(sendbuf,"EOF",3) == 0){
                break;
            }
        }//while 收发消息
        printf("断开连接\n\n");
    }//while accept

    //关闭连接
    close(s_fd);
    printf("程序结束\n"); 
    return;
}

客户端：

#include <sys/types.h>
#include <sys/socket.h> //包含套接字的函数库
#include <netinet/in.h> //包含AF_INET的相关结构
#include <arpa/inet.h>  //包含AF_INET的操作函数
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h> 

/* 
 连接到服务器后，会不停循环，等待输入， 
输入EOF后，断开连接,输入quit，断开连接并关闭服务器
读取EOF后，断开连接,读取quit，断开连接并关闭服务器
 */  
#define PORT 3333  
void main()
{
    printf("程序开始\n");

    //客户端套接字标识符，只需要一个套接字文件描述符，用于和客户端通信
    int sockfd; 
    //描述客户端的socket
    struct sockaddr_in addr;
    //消息长度
    int dataBytes = 0;
    //消息
    char sendbuf[BUFSIZ];  
    char recvbuf[BUFSIZ];  
    
    //创建套接字
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0){
        perror("创建套接字失败");
        return;
    }
    printf("创建套接字成功\n");
    /********************初始化客户端套接字*****************/
    //htons和htonl将端口和地址转成网络字节序
    //定义客户端中地址中的域，AF_INET指IPv4
    addr.sin_family = AF_INET;
    //定义服务端的套接字端口
    addr.sin_port = htons(PORT);
    //指定服务器端的ip，本地测试：127.0.0.1  
    //inet_addr()函数，将点分十进制IP转换成网络字节序IP 
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    /**********************连接服务器**********************/
    if(connect(sockfd, (struct sockaddr *)&addr, sizeof(addr)) < 0)  
    {  
        perror("连接服务器失败");  
        return;  
    }
    printf("连接到服务器...\n");
    while(1){
        /*******************发送消息************************/
        printf("--------------------发送:");
        scanf("%s", sendbuf);   
        send(sockfd,sendbuf,strlen(sendbuf)+1,0);

        if(strcmp(sendbuf, "quit") == 0){
            //关闭服务器
            //关闭客户端
            break;
        }
        if(strcmp(sendbuf, "EOF") == 0){
            //关闭客户端
            break;
        }
        
        //读取消息
        printf("--------------------读取:");
        fflush(stdout);
        dataBytes = recv(sockfd, recvbuf, 200, 0);  
        printf("%s\n", recvbuf);
        if(strncmp(recvbuf, "quit",4) == 0){
            //关闭服务器
            //关闭客户端
            break;
        }
        if(strncmp(recvbuf, "EOF",3) == 0){
            //关闭客户端
            break;
        }
    }

    //关闭连接
    close(sockfd);
    printf("程序结束\n"); 
    return;
}