聊天工具服务器

最新推荐文章于 2020-01-19 17:00:17 发布

loyuehui

最新推荐文章于 2020-01-19 17:00:17 发布

阅读量337

点赞数

文章标签： socket 聊天工具

本文链接：https://blog.csdn.net/loyuehui/article/details/46318633

版权

#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 <sys/wait.h>
#include <unistd.h>
#include <arpa/inet.h>

#define MAXBUF 1024

int main(int argc, char *argv[])
{
    int pid;
    int sockfd, new_fd;
    socklen_t len;
    struct sockaddr_in my_addr,their_addr;
    unsigned int myport,lisum;
    char buf[MAXBUF+1];
    if(argv[2])
    {
        myport=atoi(argv[2]);
    }
    else
    {
        myport =7575;
    }

    if ( argv[3] )
    {
        lisum =atoi(argv[3]);
    }
    else
    {
        lisum = 5;
    }

    if ( (sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1 )
    {
        perror("socket");
        exit(EXIT_FAILURE);
    }

    bzero(&my_addr, sizeof(my_addr));
    my_addr.sin_family = AF_INET;
    my_addr.sin_port   = htons(myport);
    if(argv[1])
    {
        my_addr.sin_addr.s_addr = inet_addr(argv[1]);
    }
    else
    {
        my_addr.sin_addr.s_addr= INADDR_ANY;
    }

    if ( bind(sockfd, (struct sockaddr *)&my_addr,sizeof(struct sockaddr)) == -1 )
    {
        perror("bind");
        exit(EXIT_FAILURE);
    }

    printf("wait for connect \n");
    len = sizeof(struct sockaddr);
    if ( (new_fd=accept(sockfd, (struct sockaddr*)&their_addr),&len)) == -1)
    {
        perror("accept");
        exit(EXIT_FAILURE);
    }
    else
    {
        printf("server :got connection from :%s, port %d, socket %d\n",inet_ntoa(their_addr.sin_addr),ntohs(their_addr.sin_port),new_fd);
    }

    while ( 1 )
    {
        printf("newfd= %d\n",new_fd);
        bzero(buf, MAXBUF+1);
        printf("input the message to send :");
        fgets(buf, MAXBUF, stdin);
        if ( !strncasecmp(buf, "quit",4) )
        {
            printf("i will close the connect \n");
            break;
        }
        len = send(new_fd, buf, strlen(buf)-1,0);
        if ( len >0 )
        {
            printf("message :%s \t send sucessful ,send %d byte\n",buf,len);
        }
        else
        {
            printf("message : %s send failure! errono code is %d merro msg is '%s' \n",buf,errno,strerror(errno));
            break;
        }
        bzero(buf, MAXBUF+1);
        len=recv(new_fd, buf, MAXBUF, 0);
        if ( len>0 )
        {
            printf("message recv successful : '%s'   %d byte recv \n",buf,len);
        }
        else
        {
            if ( len <0 )
            {
                printf("recv failure! errno code is %d, errno msg is '%s'\n",errno,strerror(errno));
            }
            else
            {
                printf("the other one close quit \n");
                break;
            }
        }
        close(new_fd);
        close(sock_fd);
        return 0;
    }

}