Linux c实现服务端与客户端聊天

主要利用socket通信实现，具体代码如下

客户端：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#define MAXLINE 511


char *EXIT_STRING = "exit";
int input_and_send(int sd);
int recv_and_print(int sd);




int main(int argc, char *argv[]){
    pid_t pid;
     int s;
    static struct sockaddr_in servaddr;
    
    if(argc != 3){
        printf("사용법: %s server_ip port 
",argv[0]);
        exit(0);
    }
    if((s = socket(PF_INET, SOCK_STREAM, 0)) < 0){
        printf("Client: Can't open stream socket.
");
        exit(0);
    }
    
    bzero((char *)&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
    servaddr.sin_port = htons(atoi(argv[2]));
    
    if(connect(s,(struct sockaddr *)&servaddr,sizeof(servaddr)) < 0){
        printf("Client: can't connect to server.
");
        exit(0);
    }
    if( (pid=fork())>0 )
        input_and_sent1(s);
    else if(pid == 0)
        recv_and_print(s);
    close(s);
    return 0;
}

int input_and_sent1(int sd){

    char buf[MAXLINE+1];
    int nbyte;
    while(fgets(buf, sizeof(buf),stdin) != NULL){
        nbyte = strlen(buf);
        write(sd, buf, strlen(buf));
        if(strstr(buf,EXIT_STRING) != NULL){
            puts("Good Bye!");
            close(sd);
            exit(0);
        }
    }
     return 0;
}

int recv_and_print(int sd){
    char buf[MAXLINE+1];    
    int nbyte;
    while(1){
        if((nbyte=read(sd, buf,MAXLINE))<0){
            perror("read fail");
            close(sd);
            exit(0);
        }
    buf[nbyte] = 0;
    if(strstr(buf,EXIT_STRING) != NULL)
        break;
    printf("Girl: %s",buf);
    }
    return 0;
}
    
    

服务器端：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>

char *EXIT_STRING = "exit";
int recv_and_print(int sd);
int input_and_send(int sd);
#define MAXLINE 511

int main(int argc, char *argv[])
{
    struct sockaddr_in cliaddr, servaddr;
    int listen_sock,accp_sock,addrlen = sizeof(cliaddr);
    pid_t pid;
    if(argc != 2){
        printf("사용법: %s port
",argv[0]);
        exit(0);
    }
    
    if((listen_sock=socket(PF_INET, SOCK_STREAM, 0)) < 0){
        perror("socket fail");
        exit(0);
    }
    
    bzero((char *)&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(atoi(argv[1]));

    if(bind(listen_sock, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0){
        perror("bind fail");
        exit(0);
    }
    puts("等待连接......");
    listen(listen_sock, 1);

    if((accp_sock = accept(listen_sock,(struct sockaddr *)&cliaddr, &addrlen)) < 0){
        perror("accept fail");
        exit(0);
    }
    
    puts("连接成功.....");
    if((pid = fork()) > 0)
        input_and_send(accp_sock);
    else if(pid == 0)
        recv_and_print(accp_sock);
    close(listen_sock);
    close(accp_sock);
    return 0;
}

int input_and_send(int sd){
    char buf[MAXLINE+1];
    int nbyte;
    while(fgets(buf, sizeof(buf),stdin) != NULL){
        nbyte = strlen(buf);
        write(sd, buf ,strlen(buf));
        if(strstr(buf, EXIT_STRING) != NULL){
            puts("Good Bye.");
            close(sd);
            exit(0);
        }
    }
    return 0;
}

int recv_and_print(int sd){
    char buf[MAXLINE+1];
    int nbyte;
    while(1){
        if((nbyte=read(sd, buf, MAXLINE)) < 0){
            perror("read fail");
            close(sd);
            exit(0);
        }
    buf[nbyte] = 0;
    if(strstr(buf, EXIT_STRING) != NULL)
        break;
    printf("Harley Gwak: %s",buf);
    }
    return 0;
}

第一次写博客，一直都是伸手党，今天就把我之前写过的东西分享下，代码正常运行的，仅供参考。