2021-08-19

accept函式（TCP）

#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

功能：阻塞等待客戶端的連接請求

sockfd：檔案描述符，socket函式的回傳值(也就是listen函式中設定為監聽狀態的套接字)
addr：接收到的客戶端的資訊結構體（自動填充，定義變數即可）（包括客戶端IP和埠資訊等）
addrlen：addr的長度

回傳值：
成功：新的檔案描述符（只要有客戶端連接，就會產生新的檔案描述符，這個新的檔案描述符專門與指定的客戶端進行通信的，原始套接字其實只是起到一個連接作用）
失敗：‐1

在TCP服務器與客戶端通信時，要使用accept回傳值，不能再使用socket回傳值，意味著在TCP中socket回傳值只是起到一個連接作用，真正通信是accept回傳值

代碼示例：
服務器：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>

#define N 128

int main(int argc,char const *argv[])
{
    if(argc < 3)
    {
        fprintf(stderr,"Usage: %s [ip] [port]\n",argv[0]);
        exit(1);
    }
    //第一步：創建套接字
    int sockfd;
    if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1)
    {
        perror("fail to socket");
        exit(1);
    }

    //第二步：由于服務器要先執行且客戶端要找到服務器，所以要將套接字與服務器網路資訊結構體系結
    struct sockaddr_in serveraddr;
    socklen_t addrlen = sizeof(serveraddr);

    serveraddr.sin_family = AF_INET;
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
    //先把字串轉換成整型數，再將主機位元組序轉換成網路位元組序
    serveraddr.sin_port = htons(atoi(argv[2]));

    if(bind(sockfd,(struct sockaddr *)&serveraddr,addrlen) == -1)
    {
        perror("fail to bind");
        exit(1);
    }

    //第三步：將套接字設定為被動監聽狀態
    if(listen(sockfd,10) == -1)
    {
        perror("fail to listen");
        exit(1);
    }

    //第四步：阻塞等待客戶端的鏈接請求
    int acceptfd;
    struct sockaddr_in clientaddr;
    if((acceptfd = accept(sockfd,(struct sockaddr *)&clientaddr,&addrlen)) == -1)
    {
        perror("fail to accept");
        exit(1);
    }

    //列印連接客戶端的資訊
    printf("ip: %s port: %d\n",inet_ntoa(clientaddr.sin_addr),ntohs(clientaddr.sin_port));
    
    //第五步：進行通信
    //tcp服務器與客戶端通信時，需要使用accept函式的回傳值
    char buf[N] = "";
    if(recv(acceptfd,buf,N,0) == -1)
    {
        perror("fail to recv");
    }
    printf("from client: %s\n",buf);
    strcat(buf,"++");
    if(send(acceptfd,buf,N,0) == -1)
    {
        perror("fail to send");
        exit(1);
    }

    //關閉套接字與檔案描述符
    close(acceptfd);
    close(sockfd);
    
    return 0;
}

客戶端：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>

#define N 128

//默認argv是一個指標陣列，所以每一個成員都是一個字串
int main(int argc,char const *argv[])
{
	if(argc < 3)
	{
		fprintf(stderr,"Usage: %s [ip] [port]\n",argv[0]);
		exit(1);
	}

	//第一步：創建套接字
	int sockfd;
	if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1)
	{
		perror("fail to socket");
		exit(1);
	}

	//第二步：發送客戶端連接請求
	struct sockaddr_in serveraddr;
	socklen_t addrlen = sizeof(serveraddr);

	serveraddr.sin_family = AF_INET;
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
	//先將字串轉換成整型資料，再將主機位元組序轉換成16位網路位元組序
	serveraddr.sin_port = htons(atoi(argv[2]));

	if(connect(sockfd,(struct sockaddr *)&serveraddr,
											addrlen) == -1)
	{
		perror("fail to connect");
		exit(1);
	}

	//第三步：進行通信
	//發送資料
	char buf[N] = "";
	fgets(buf,N,stdin);
	buf[strlen(buf) - 1] = '\0';
	if(send(sockfd,buf,N,0) == -1)
	{
		perror("fail to send");
		exit(1);
	}

	//接收資料
	char text[N] = "";
	if(recv(sockfd,text,N,0) == -1)
	{
		perror("fail to recv");
		exit(1);
	}
	printf("from server:%s\n",text);

	//第四步：關閉套接字檔案描述符
	close(sockfd);

	return 0;
}

運行結果：