多進程服務器

原創

2020-02-21 10:46

基於TCP實現多進程服務器
服務器端：
1、創建套接字：

 #include <sys/types.h>          
 #include <sys/socket.h>
 int socket(int domain, int type, int protocol);

domain：一個地址描述。目前僅支持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：等待連接隊列的最大長度，本程序設爲10

4、接收

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

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

5、創建進程
在子進程中進行讀寫數據處理，父進程再次fork後退出，並關閉sock，否則會一直產生新的sock。

源代碼：
#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<string.h>
#include<unistd.h>
int startup( 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 use(char* proc)
{
    printf("%s [server_ip][server_port]\n");
}
int main(int argc,char* argv[])
{
    if(argc!=3)
    {
        use(argv[0]);
        return 1;
    }
    int listen_sock = startup(argv[1],atoi(argv[2]));
    struct sockaddr_in remote;
    socklen_t len = sizeof(remote);
    while(1)
    {
    //接收
        int newsock = accept(listen_sock,(struct sockaddr*)&remote,&len);
            if(newsock<0)
            {
                perror("accept");
                return 5;
            }
    char buf[1024];
    //創建進程
    pid_t id = fork();
    if(id==0)
    {//child
        while(1)
        {
        ssize_t s = read(newsock,buf,sizeof(buf));
        if(s>0)
        {
            buf[s]=0;
            printf("Client say#%s\n",buf);
            fflush(stdout);
            write(newsock,buf,strlen(buf));
        }
        else if(s==0)
        {
            printf("client quit!/n");
            break;
        }
        }
        close(newsock);
    }
    else if(id>0)
    {//father
        pid_t id = fork();
        if(id>0)
        {
        close(newsock);
        exit(1);

        }
        else if(id<0)
        {
            perror("fork");
            return 5;
        }
    }
    else
    {
        perror("fork");
        return 4;
    }
}
    return 0;
}

客戶端：
1、創建套接字

 #include <sys/types.h>          
 #include <sys/socket.h>
 int socket(int domain, int type, int protocol);

2、連接

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

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

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<string.h>
#include<unistd.h>
static void use(char* proc)
{
    printf("%s [server_ip][server_port]\n");
}
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;
}