[LINUX] TCP/IP Simple Chat

어디서나 볼 수 있는 TCP/IP 예제

---

Server.c

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

void error_handling(char *message);

int main(int argc, char **argv)
{
        int serv_sock;
        int clnt_sock;
        struct sockaddr_in serv_addr;
        struct sockaddr_in clnt_addr;
        int clnt_addr_size;
        char message[] = "Hello sp0ngee!\n";

        if(argc != 2)
        {
                printf("Usage : %s <port>\n", argv[0]);
                exit(1);
        }

        /* 서버 소켓 생성 */
        serv_sock = socket(PF_INET, SOCK_STREAM, 0);
        if(serv_sock == -1)
                error_handling("socket() error");

        memset(&serv_addr, 0, sizeof(serv_addr));

        serv_addr.sin_family = AF_INET;
        serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
        serv_addr.sin_port = htons(atoi(argv[1]));

        /* 소켓에 주소 할당 */
        if(bind(serv_sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)
                error_handling("bind() error");

        /* 연결 요청 대기 상태로 진입 */
        if(listen(serv_sock, 5) == -1)
                error_handling("listen() error");

        clnt_addr_size = sizeof(clnt_addr);
        /* 연결 요청 수락 */
        clnt_sock = accept(serv_sock, (struct sockaddr*) &clnt_addr, &clnt_addr_size);

        if(clnt_sock == -1)
                error_handling("accept() error");

        /* 데이터 전송 */
        write(clnt_sock, message, sizeof(message));
        /* 연결 종료 */
        close(clnt_sock);

        return 0;
}

void error_handling(char *message)
{
        fputs(message, stderr);
        fputc('\n', stderr);
        exit(1);
}

Client.c

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

void error_handling(char *message);

int main(int argc, char **argv)
{
        int sock;
        struct sockaddr_in serv_addr;
        char message[30];
        int str_len;

        if(argc != 3)
        {
                printf("Usage : %s <IP> <port>\n", argv[0]);
                exit(1);
        }

        /* 서버 접속을 위한 소켓 생성 */
        sock = socket(PF_INET, SOCK_STREAM, 0);

        if(sock == -1)
                error_handling("socket() error");

        memset(&serv_addr, 0, sizeof(serv_addr));

        serv_addr.sin_family = AF_INET;
        serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
        serv_addr.sin_port = htons(atoi(argv[2]));

        /* 서버로 연결 요청 */
        if(connect(sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)
                error_handling("connect() error");

        /* 데이터 수신 */
        str_len = read(sock, message, sizeof(message)-1);

        if(str_len == -1)
                error_handling("read() error");
        message[str_len] = 0;

        printf("Message from server: %s\n", message);
        /* 연결 종료 */
        close(sock);

        return 0;
}

void error_handling(char *message)
{
        fputs(message, stderr);
        fputc('\n', stderr);
        exit(1);
}
출처 : 
http://sp0ngee.tistory.com