LINUX下实现QQ对话

LINUX下实现QQ对话

1.tcp_unblock_client.c

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <resolv.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/wait.h>
#include <netdb.h>
#include <fcntl.h>

#define  MAXBUF 128

int main(int argc,char **argv)
{
    int sockfd,ret,i;
    struct sockaddr_in dest,mine;

    char buffer[MAXBUF+1];
    if((sockfd = socket(AF_INET,SOCK_STREAM,0))<0)
    {
        perror("Socket");
        exit(EXIT_FAILURE);
    }

    bzero(&dest,sizeof(dest));
    dest.sin_family=AF_INET;
    dest.sin_port = htons(7838);

    if(inet_aton(argv[1],(struct in_addr *)&dest.sin_addr.s_addr)==0)
    {
        perror(argv[1]);
        exit(EXIT_FAILURE);
    }

    bzero(&mine,sizeof(mine));
    mine.sin_family=AF_INET;
    mine.sin_port = htons(7839);

    if(inet_aton(argv[2],(struct in_addr *)&dest.sin_addr.s_addr)==0)
    {
        perror(argv[2]);
        exit(EXIT_FAILURE);
    }

    if(bind(sockfd,(struct sockaddr *)&mine,sizeof(struct sockaddr))==-1)
    {
        perror(argv[3]);
        exit(EXIT_FAILURE);
    }

    if(connect(sockfd,(struct sockaddr *)&dest,sizeof(dest))!=0)
    {
        perror("connect");
        exit(EXIT_FAILURE);
    }

    if(fcntl(sockfd,F_SETFL,O_NONBLOCK)==-1)
    {
        perror("fcntl");
        exit(EXIT_FAILURE);
    }


    while(1)
    {
        bzero(buffer,MAXBUF+1);
        ret = recv(sockfd,buffer,MAXBUF,0);
        if(ret>0)
        {
            printf("get %d message:%s",ret,buffer);
            ret=0;
        }

        else if(ret<0)
        {
            if(errno == EAGAIN)
            {
                errno=0;
                continue;
            }
            else
            {
                perror("recv");
                exit(EXIT_FAILURE);
            }
        }
        memset(buffer,'\0',MAXBUF+1);
        printf("input message to send:");

        fgets(buffer,MAXBUF,stdin);
        if((ret=send(sockfd,buffer,strlen(buffer),0))==-1)
        {
            perror("send");
            exit(EXIT_FAILURE);
        }
    }
    close(sockfd);
    return 0;
}


2. tcp_unblock_server.c
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <unistd.h>
#include <sys/wait.h>
#include <netdb.h>
#include <fcntl.h>
#include <arpa/inet.h>

#define BUFSIZE 128

int main(int argc,char *argv[])
{

    int server_sockfd,client_sockfd;
    int server_len,client_len;
    struct sockaddr_in server_address;
    struct sockaddr_in client_address;
    int i,byte;
    char char_send[BUFSIZE];

    server_sockfd = socket(AF_INET,SOCK_STREAM,0);
    server_address.sin_family=AF_INET;

    if(inet_aton(argv[1],(struct in_addr *)&server_address.sin_addr.s_addr)==0)
    {
        perror(argv[1]);
        exit(EXIT_FAILURE);
    }

    server_address.sin_port=htons(7838);
    server_len=sizeof(server_address);

    bind(server_sockfd,(struct sockaddr *)&server_address,server_len);
    listen(server_sockfd,5);
  
    printf("server waiting for connect\n");

    client_len=sizeof(client_address);

    client_sockfd=accept(server_sockfd,(struct sockaddr *)&client_address,(socklen_t *)&client_len);


    for(i=0;i<5;i++)
    {
        memset(char_send,'\0',BUFSIZE);

        printf("input message to send:");
        fgets(char_send,BUFSIZE,stdin);
        if((byte=send(client_sockfd,char_send,strlen(char_send),0))==-1)
        {
            perror("send");
            exit(EXIT_FAILURE);
        }
        memset(char_send,'\0',BUFSIZE);


        byte = recv(client_sockfd, char_send, BUFSIZE, MSG_DONTWAIT);
        if(byte>0)
        {

            printf("get %d message:%s",byte,char_send);
        }
        else if(byte<0)
        {
            if(errno==EAGAIN)
            {
                errno=0;
                continue;
            }
            else
            {

                perror("recv");
                exit(EXIT_FAILURE);
            }
        }
    }
    shutdown(client_sockfd,2);
    shutdown(server_sockfd,2);
}

说明： 把两个文件编译
   1.  /forlinx/root/socket_test_ok/nonblock_socket# gcc -o tcp_unblock_server tcp_unblock_server.c
   2. root@ubuntu:/forlinx/root/socket_test_ok/nonblock_socket# gcc -o tcp_unblock_client tcp_unblock_client.c

   运行软件
       打开一个窗口：
        root@ubuntu:/forlinx/root/socket_test_ok/nonblock_socket# ./tcp_unblock_client 192.168.3.106 192.168.3.106
get 6 message:hello
input message to send:hello
get 3 message:ok
input message to send:ok
get 5 message:test
input message to send:test
get 3 message:go
input message to send:go
get 5 message:read
input message to send:end
input message to send:
   打开另一个窗口
  root@ubuntu:/forlinx/root/socket_test_ok/nonblock_socket# ./tcp_unblock_server 192.168.3.106
server waiting for connect
input message to send:hello
input message to send:ok
get 6 message:hello
input message to send:test
get 3 message:ok
input message to send:go
get 5 message:test
input message to send:read
get 3 message:go