Socket網絡編程--聊天程序(2)

　　上一節簡單如何通過Socket創建一個連接，然后進行通信。只是每個人只能說一句話。而且還是必須說完才會接收到信息，總之是很不方便的事情。所以這一小節我們將對上一次的程序進行修改，修改成每個人可以多說話，主要是通過Linux下多進程fork實現的。

　　一對一，server和client是每個人都可以多說幾句話

　　由於控制台度數據的函數fgets是阻塞函數，要每個人都可以多說話，這里我使用的是讀取和發送都在不同的進程里面，使之互相沒有影響。

　　fork函數

　　#include <unistd.h>

　　pid_t fork(void);    //返回值：子進程中返回0，父進程中返回子進程ID，出錯返回-1

　　client.c

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netdb.h> //for gethostbyname
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define MAX_BUF 4096
#define SERVER_PORT 12138


int main(int argc,char *argv[])
{
    int sockfd;//socket
    char sendBuf[MAX_BUF],recvBuf[MAX_BUF];
    int sendSize,recvSize;//用於記錄記錄發送和接收到數據的大小
    struct hostent * host;
    struct sockaddr_in servAddr;
    char username[32];
    char * p;
    int pid;

    if(argc != 3)
    {
        perror("use: ./client [hostname] [username]");
        exit(-1);
    }
    p=username;
    strcpy(p,argv[2]);
    printf("username:%s\n",username);
    host=gethostbyname(argv[1]);
    if(host==NULL)
    {
        perror("fail to get host by name.");
        exit(-1);
    }
    printf("Success to get host by name ...\n");

    //創建socket
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("fail to establish a socket");
        exit(1);
    }
    printf("Success to establish a socket...\n");

    /*init sockaddr_in*/
    servAddr.sin_family=AF_INET;
    servAddr.sin_port=htons(SERVER_PORT);
    servAddr.sin_addr=*((struct in_addr *)host->h_addr);
    bzero(&(servAddr.sin_zero),8);

    /*connect the socket*/
    if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)
    {
        perror("fail to connect the socket");
        exit(1);
    }
    printf("Success to connect the socket...\n");

    //send-recv 一些返回指沒有判斷，具體可以看server.c
    if((pid=fork())<0)
    {
        perror("fork error\n");
    }
    else if(pid>0)/*child*/
    {
        while(1)
        {
            fgets(sendBuf,MAX_BUF,stdin);
            send(sockfd,sendBuf,strlen(sendBuf),0);
            memset(sendBuf,0,sizeof(sendBuf));
        }
    }
    else
    {
        while(1)
        {
            recv(sockfd,recvBuf,MAX_BUF,0);
            printf("Server:%s\n",recvBuf);
            memset(recvBuf,0,sizeof(recvBuf));
        }
    }

    return 0;
}

　　server.c

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/un.h>
#include <sys/ioctl.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include <arpa/inet.h>


#define SERVER_PORT 12138
#define BACKLOG 20
#define MAX_CON_NO 10
#define MAX_DATA_SIZE 4096

int main(int argc,char *argv[])
{
    struct sockaddr_in serverSockaddr,clientSockaddr;
    char sendBuf[MAX_DATA_SIZE],recvBuf[MAX_DATA_SIZE];
    int sendSize,recvSize;
    int sockfd,clientfd;
    int on=1;
    int sinSize=0;
    char username[32];
    int pid;

    if(argc != 2)
    {
        printf("usage: ./server [username]\n");
        exit(1);
    }
    strcpy(username,argv[1]);
    printf("username:%s\n",username);

    /*establish a socket*/
    if((sockfd = socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("fail to establish a socket");
        exit(1);
    }
    printf("Success to establish a socket...\n");

    /*init sockaddr_in*/
    serverSockaddr.sin_family=AF_INET;
    serverSockaddr.sin_port=htons(SERVER_PORT);
    serverSockaddr.sin_addr.s_addr=htonl(INADDR_ANY);
    bzero(&(serverSockaddr.sin_zero),8);

    setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on));

    /*bind socket*/
    if(bind(sockfd,(struct sockaddr *)&serverSockaddr,sizeof(struct sockaddr))==-1)
    {
        perror("fail to bind");
        exit(1);
    }
    printf("Success to bind the socket...\n");

    /*listen on the socket*/
    if(listen(sockfd,BACKLOG)==-1)
    {
        perror("fail to listen");
        exit(1);
    }

    /*accept a client's request*/  //這里的sinSize還沒有處理，具體看下一節
    if((clientfd=accept(sockfd,(struct sockaddr  *)&clientSockaddr, &sinSize))==-1) 
    {
        perror("fail to accept");
        exit(1);
    }
    printf("Success to accpet a connection request...\n");
    printf(" %s join in!\n",inet_ntoa(clientSockaddr.sin_addr));

    if((pid=fork())<0)
    {
        perror("fork error\n");
    }
    else if(pid>0)/*child*/
    {
        /*send datas to client*/
        while(1)
        {
            fgets(sendBuf,MAX_DATA_SIZE,stdin);
            if((sendSize=send(clientfd,sendBuf,strlen(sendBuf),0))!=strlen(sendBuf))
            {
                perror("fail to send datas");
                exit(1);
            }
            printf("Success to send datas\n");
            memset(sendBuf,0,MAX_DATA_SIZE);
        }
    }
    else
    {
        while(1)
        {
            /*receive datas from client*/
            if((recvSize=recv(clientfd,recvBuf,MAX_DATA_SIZE,0))==-1)
            {
                perror("fail to receive datas");
                exit(1);
            }
            printf("Client:%s\n",recvBuf);
            memset(recvBuf,0,MAX_DATA_SIZE);
        }

    }

    return 0;
}

 

 

　　本文地址: http://www.cnblogs.com/wunaozai/p/3870194.html