基於TCP的服務器/客戶端通信

原創

2020-02-21 10:46

簡易TCP服務器/客戶端
服務器端：
1、創建套接字
int socket( int af, int type, int protocol);
af：一個地址描述。目前僅支持AF_INET格式，也就是說ARPA Internet地址格式。
type：指定socket類型。新套接口的類型描述類型，
TCP（SOCK_STREAM）和UDP（SOCK_DGRAM）。常用的socket類型有，SOCK_STREAM、SOCK_DGRAM、SOCK_RAW、SOCK_PACKET、SOCK_SEQPACKET等等。
protocol：顧名思義，就是指定協議。套接口所用的協議。如調用者不想指定，可用0。常用的協議有，IPPROTO_TCP、IPPROTO_UDP、IPPROTO_STCP、IPPROTO_TIPC等，它們分別對應TCP傳輸協議、UDP傳輸協議、STCP傳輸協議、TIPC傳輸協議。
2、綁定
將一本地地址與一套接口捆綁

#include <sys/types.h>
#include <sys/socket.h>
int bind( int sockfd , const struct sockaddr * my_addr, socklen_t addrlen);

參數列表中，sockfd 表示已經建立的socket編號（描述符）；
my_addr 是一個指向sockaddr結構體類型的指針；
參數addrlen表示my_addr結構的長度，可以用sizeof函數獲得。
3、監聽
創建一個套接口並監聽申請的連接.

#include <sys/socket.h>
int listen( int sockfd, int backlog);

sockfd：用於標識一個已捆綁未連接套接口的描述字。
backlog：等待連接隊列的最大長度
4、接收
在一個套接口接受一個連接

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

參數
sockfd：套接字描述符，該套接口在listen()後監聽連接。
addr：（可選）指針，指向一緩衝區，其中接收爲通訊層所知的連接實體的地址。Addr參數的實際格式由套接口創建時所產生的地址族確定。
addrlen：（可選）指針，輸入參數，配合addr一起使用，指向存有addr地址長度的整型數。
代碼：

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include<stdlib.h>
#include<string.h>
int startup(const char* ip,int port)
{
     //創建套接字
    int sock = socket(AF_INET,SOCK_STREAM,0);
    if(sock<0)
    {
        perror("socket");
        return 2;
    }
    struct sockaddr_in local;
    local.sin_family=AF_INET;
    local.sin_port = htons(port);
    local.sin_addr.s_addr = inet_addr(ip);
     //綁定
    if(bind(sock,(struct sockaddr*)&local,sizeof(local))<0)
    {
        perror("bind");
        return 3;
    }
    //監聽
    if(listen(sock,10)<0)
    {
        perror("listen");
        return 4;
    }
    return sock;
}
static void Usage(char* proc)
{
    printf("%s [server_ip] [server_port]\n",proc);
}
int main(int argc,char* argv[])
{
    if(argc!=3)
    {
        Usage(argv[0]);
        return 1;
    }
    int listen_sock = startup(argv[1],atoi(argv[2]));
    struct sockaddr_in remote;
    socklen_t len = sizeof(remote);
    char buf[1024];
    while(1)
    {
    //接受連接
        int sock = accept(listen_sock,(struct sockaddr*)&remote,&len);
        if(sock<0)
        {
            perror("accept");
            continue;
        }
    while(1)
    {
        ssize_t s = read(sock,buf,sizeof(buf));
        buf[s] = 0;
        if(s>0)
        {
            printf("Client say#%s\n",buf);
            write(sock,buf,strlen(buf));
        }
    }
    close(sock);
}
    return 0;
}

客戶端：
1、創建套接字（同服務器）
2、連接
connect()用於建立與指定socket的連接

 #include <sys/socket.h>
 int connect(int s, const struct sockaddr * name, int namelen);

s：標識一個未連接socket
name：指向要連接套接字的sockaddr結構體的指針
namelen：sockaddr結構體的字節長度

代碼：

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<string.h>
#include<stdlib.h>
static void Use(const char* proc)
{
    printf("%s [server_ip] [server_port]\n",proc);
}
int main(int argc,char* argv[])
{
    if(argc!=3)
    {
        Use(argv[0]);
        return 1;
    }
    //創建套接字
    int sock = socket(AF_INET,SOCK_STREAM,0);
    if(sock<0)
    {
        perror("socket");
        return 2;
    }
    struct sockaddr_in peer;
    peer.sin_family = AF_INET;
    peer.sin_port = htons(atoi(argv[2]));
    peer.sin_addr.s_addr = inet_addr(argv[1]);
    //連接
    if(connect(sock,(struct sockaddr*)&peer,sizeof(peer))<0)
    {
        perror("connect");
        return 3;
    }
    char buf[1024];
    while(1)
    {
        printf("please enter#");
        fflush(stdout);
        ssize_t s = read(0,buf,sizeof(buf)-1);
        if(s>0)
        {
            buf[s-1]=0;
            write(sock,buf,strlen(buf));
            ssize_t _s = read(sock,buf,sizeof(buf)-1);
            if(_s>0)
            {
                buf[_s]=0;
                printf("server echo#%s\n",buf);
            }
        }
    }
    close(sock);
    return 0;
}