Asynchronous Notification IOモデル エコーサーバ（WINDOWS）

#include <stdio.h>
#include <winsock2.h>
#include <string.h>

#define BUF_SIZE 100

void CompressSockets(SOCKET hSockArr[], int idx, int total);
void CompressEvents(WSAEVENT hEventArr[], int idx, int total);
void ErrorHandling(char *msg);

int main(int argc, char *argv[]) {
    WSADATA wsaData;
    SOCKET hServSock, hClntSock;
    SOCKADDR_IN ServAddr, ClntAddr;
    int AddrSize;

    SOCKET hSockArr[WSA_MAXIMUM_WAIT_EVENTS];
    WSAEVENT hEventArr[WSA_MAXIMUM_WAIT_EVENTS];
    WSAEVENT newEvent;
    WSANETWORKEVENTS netEvents;

    int numOfClntSock = 0;
    int strLen, i;
    int posInfo, startIdx;
    char msg[BUF_SIZE] = { 0, };

    if (argc != 2) {
        printf("Usage : %s <PORT> \n", argv[0]);
        exit(1);
    }
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        ErrorHandling("WSAStartup() error");
    }

    hServSock = socket(PF_INET, SOCK_STREAM, 0);
    memset(&ServAddr, 0, sizeof(ServAddr));
    ServAddr.sin_family = AF_INET;
    ServAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    ServAddr.sin_port = htons(atoi(argv[1]));

    if (bind(hServSock, (SOCKADDR*)&ServAddr, sizeof(ServAddr)) == SOCKET_ERROR) {
        ErrorHandling("bind() error");
    }
    if (listen(hServSock, 5) == SOCKET_ERROR) {
        ErrorHandling("listen() error");
    }

    newEvent = WSACreateEvent();
    if (WSAEventSelect(hServSock, newEvent, FD_ACCEPT) == SOCKET_ERROR) {
        ErrorHandling("WSAEventSelect() error");
    }

    hSockArr[numOfClntSock] = hServSock;
    hEventArr[numOfClntSock] = newEvent;
    numOfClntSock++;

    while (1) {
        posInfo = WSAWaitForMultipleEvents(numOfClntSock, hEventArr, FALSE, WSA_INFINITE, FALSE);
        startIdx = posInfo - WSA_WAIT_EVENT_0;
        for (i = startIdx; i < numOfClntSock; i++) {
            int sigEventIdx = WSAWaitForMultipleEvents(1, &hEventArr[i], TRUE, 0, FALSE);
            if ((sigEventIdx == WSA_WAIT_FAILED || sigEventIdx == WSA_WAIT_TIMEOUT)) {
                continue;
            }
            else {
                sigEventIdx = i;
                WSAEnumNetworkEvents(hSockArr[sigEventIdx], hEventArr[sigEventIdx], &netEvents);
                if (netEvents.lNetworkEvents & FD_ACCEPT) {
                    if (netEvents.iErrorCode[FD_ACCEPT_BIT] != 0) {
                        puts("Accept error");
                        break;
                    }
                    AddrSize = sizeof(ClntAddr);
                    hClntSock = accept(hSockArr[sigEventIdx], (SOCKADDR*)&ClntAddr, &AddrSize);
                    newEvent = WSACreateEvent();
                    WSAEventSelect(hClntSock, newEvent, FD_READ | FD_CLOSE);
                    hSockArr[numOfClntSock] = hClntSock;
                    hEventArr[numOfClntSock] = newEvent;
                    numOfClntSock++;
                    puts("Connected New Client...");
                }

                if (netEvents.lNetworkEvents & FD_READ) {
                    if (netEvents.iErrorCode[FD_READ_BIT] != 0) {
                        puts("Read Error");
                        break;
                    }
                    strLen = recv(hSockArr[sigEventIdx], msg, sizeof(msg), 0);
                    send(hSockArr[sigEventIdx], msg, strLen, 0);
                }

                if (netEvents.lNetworkEvents & FD_CLOSE) {
                    if (netEvents.iErrorCode[FD_CLOSE_BIT] != 0) {
                        puts("Close Error");
                        break;
                    }
                    WSACloseEvent(hEventArr[sigEventIdx]);
                    closesocket(hSockArr[sigEventIdx]);
                    numOfClntSock--;
                    CompressSockets(hSockArr, sigEventIdx, numOfClntSock);
                    CompressEvents(hEventArr, sigEventIdx, numOfClntSock);
                }
            }
        }
    }
    WSACleanup();
    return 0;
}

void CompressSockets(SOCKET hSockArr[], int idx, int total) {
    int i;
    for (i = idx; i < total; i++) {
        hSockArr[i] = hSockArr[i + 1];
    }
}

void CompressEvents(WSAEVENT hEventArr[], int idx, int total) {
    int i;
    for (i = idx; i < total; i++) {
        hEventArr[i] = hEventArr[i + 1];
    }
}

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

チャットクライアント_Windows版

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <windows.h>
#include <process.h>
#define BUF_SIZE 100
#define NAME_SIZE 20

unsigned WINAPI SendMsg(void *arg);
unsigned WINAPI RecvMsg(void *arg);
void ErrorHandling(char *msg);

char name[NAME_SIZE] = "[DEFAULT]";
char msg[BUF_SIZE] = { 0, };

int main(int argc, char *argv[]) {
    WSADATA wsaData;
    HANDLE hSndThread, hRcvThread;
    SOCKET sock;
    SOCKADDR_IN ServAddr;
    if (argc != 4) {
        printf("Usage : %s <IP> <PORT> <NAME>");
        exit(1);
    }
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        ErrorHandling("WSAStartup() error");
    }
    sprintf(name, "[%s]", argv[3]);
    sock = socket(PF_INET, SOCK_STREAM, 0);
    memset(&ServAddr, 0, sizeof(ServAddr));
    ServAddr.sin_family = AF_INET;
    ServAddr.sin_addr.s_addr = inet_addr(argv[1]);
    ServAddr.sin_port = htons(atoi(argv[2]));

    if (connect(sock, (SOCKADDR*)&ServAddr, sizeof(ServAddr)) == SOCKET_ERROR) {
        ErrorHandling("connect() error");
    }

    hSndThread = (HANDLE)_beginthreadex(NULL, 0, SendMsg, (void*)&sock, 0, NULL);
    hRcvThread = (HANDLE)_beginthreadex(NULL, 0, RecvMsg, (void*)&sock, 0, NULL);

    WaitForSingleObject(hSndThread, INFINITE);
    WaitForSingleObject(hRcvThread, INFINITE);

    closesocket(sock);
    WSACleanup();
    return 0;
}

unsigned WINAPI SendMsg(void *arg) {
    SOCKET sock = *((SOCKET*)arg);
    char nameMsg[BUF_SIZE + NAME_SIZE] = { 0, };
    while (1) {
        fgets(msg, BUF_SIZE, stdin);
        if (!strcmp(msg, "q\n") || !strcmp(msg, "Q\n")) {
            closesocket(sock);
            exit(0);
        }
        sprintf(nameMsg, "[%s] %s", name, msg);
        send(sock, nameMsg, strlen(nameMsg), 0);
    }
    return 0;
}

unsigned WINAPI RecvMsg(void *arg) {
    SOCKET sock = *((SOCKET*)arg);
    int strLen;
    char nameMsg[BUF_SIZE + NAME_SIZE] = { 0, };
    while (1) {
        strLen = recv(sock, nameMsg, BUF_SIZE + NAME_SIZE - 1, 0);
        if (strLen == -1) {
            return -1;
        }
        nameMsg[strLen] = 0;
        fputs(nameMsg, stdout);
    }
    return 0;
}

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

チャットサーバ_Windows版

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <windows.h>
#include <process.h>

#define BUF_SIZE 100
#define MAX_CLNT 256

unsigned WINAPI HandleClnt(void *arg);
void SendMsg(char *msg, int len);
void ErrorHandling(char *msg);

int clntCnt = 0;
SOCKET clntSocks[MAX_CLNT];
HANDLE hMutex;

int main(int argc, char *argv[]) {
    WSADATA wsaData;
    SOCKET ServSock, ClntSock;
    SOCKADDR_IN ServAddr, ClntAddr;
    int AddrSize;
    HANDLE hThread;

    if (argc != 2) {
        printf("Usage : %s <PORT> \n", argv[0]);
        exit(1);
    }
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        ErrorHandling("WSAStartup() error");
    }
    hMutex = CreateMutex(NULL, FALSE, NULL);
    ServSock = socket(PF_INET, SOCK_STREAM, 0);
    memset(&ServAddr, 0, sizeof(ServAddr));
    ServAddr.sin_family = AF_INET;
    ServAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    ServAddr.sin_port = htons(atoi(argv[1]));

    if (bind(ServSock, (SOCKADDR*)&ServAddr, sizeof(ServAddr)) == SOCKET_ERROR) {
        ErrorHandling("bind() error");
    }
    if (listen(ServSock, 5) == SOCKET_ERROR) {
        ErrorHandling("listen() error");
    }

    while (1)
    {
        AddrSize = sizeof(ClntAddr);
        ClntSock = accept(ServSock, (SOCKADDR*)&ClntAddr, &AddrSize);
        WaitForSingleObject(hMutex, INFINITE);
        clntSocks[clntCnt++] = ClntSock;
        ReleaseMutex(hMutex);
        hThread = (HANDLE)_beginthreadex(NULL, 0, HandleClnt, (void*)&ClntSock, 0, NULL);
        printf("Connected Client IP : %s \n", inet_ntoa(ClntAddr.sin_addr));
    }
    closesocket(ServSock);
    WSACleanup();
    return 0;
}

unsigned WINAPI HandleClnt(void *arg) {
    SOCKET ClntSock = *((SOCKET*)arg);
    int strLen = 0, i;
    char msg[BUF_SIZE];

    while ((strLen = recv(ClntSock, msg, sizeof(msg), 0)) != 0)
    {
        SendMsg(msg, strLen);
    }
    WaitForSingleObject(hMutex, INFINITE);
    for (i = 0; i<clntCnt; i++)
    {
        if (ClntSock == clntSocks[i]) {
            while (i++<clntCnt - 1)
            {
                clntSocks[i] = clntSocks[i + 1];
            }
            break;
        }
    }
    clntCnt--;
    ReleaseMutex(hMutex);
    closesocket(ClntSock);
    return 0;
}

void SendMsg(char *msg, int len) {
    int i;
    WaitForSingleObject(hMutex, INFINITE);
    for (i = 0; i<clntCnt; i++)
    {
        send(clntSocks[i], msg, len, 0);
    }
    ReleaseMutex(hMutex);
}

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

マルチスレッド基盤チャットクライアント

[root@localhost tcpip]# cat ./chat_client.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <pthread.h>

#define BUF_SIZE 100
#define NAME_SIZE 20
void *send_msg(void *arg);
void *recv_msg(void *arg);

char name[NAME_SIZE]="[DEFAULT]";
char msg[BUF_SIZE]={0,};

int main(int argc, char *argv[]){
        int sock;
        struct sockaddr_in serv_addr;
        pthread_t snd_thread, rcv_thread;
        void *thread_return;
        if(argc!=4){
                printf("Usage : %s <IP> <PORT> <NAME> \n",argv[0]);
                exit(1);
        }

        sprintf(name, "[%s]", argv[3]);
        sock=socket(PF_INET, SOCK_STREAM, 0);
        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){
                perror("connect");
        }

        pthread_create(&snd_thread, NULL, send_msg, (void*)&sock);
        pthread_create(&rcv_thread, NULL, recv_msg, (void*)&sock);
        pthread_join(snd_thread, &thread_return);
        pthread_join(rcv_thread, &thread_return);
        close(sock);
        return 0;
}

void *send_msg(void *arg){
        int sock=*((int*)arg);
        char name_msg[NAME_SIZE+BUF_SIZE]={0,};
        while(1)
        {
                fgets(msg, BUF_SIZE, stdin);
                if(!strcmp(msg,"q\n")||!strcmp(msg,"Q\n")){
                        close(sock);
                        exit(0);
                }
                sprintf(name_msg, "%s %s", name, msg);
                write(sock, name_msg, strlen(name_msg));
        }
        return NULL;
}

void *recv_msg(void *arg){
        int sock=*((int*)arg);
        char name_msg[BUF_SIZE+NAME_SIZE]={0,};
        int str_len;
        while(1)
        {
                str_len=read(sock, name_msg, NAME_SIZE+BUF_SIZE);
                if(str_len==-1){
                        return (void*)-1;
                }
                name_msg[str_len]=0;
                fputs(name_msg, stdout);
        }
        return NULL;
}

マルチスレッド基盤チャットサーバ

[root@localhost tcpip]# cat ./chat_server.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <unistd.h>
#include <pthread.h>

#define BUF_SIZE 100
#define MAX_CLNT 256

void *handle_clnt(void *arg);
void send_msg(char *msg, int len);
pthread_mutex_t mutx;
int clnt_cnt=0;
int clnt_socks[MAX_CLNT];

int main(int argc, char *argv[]){
        int serv_sock, clnt_sock;
        struct sockaddr_in serv_addr, clnt_addr;
        socklen_t addr_size;
        pthread_t t_id;

        if(argc!=2){
                printf("Usage : %s <PORT> \n", argv[0]);
                exit(1);
        }
        pthread_mutex_init(&mutx, NULL);
        serv_sock=socket(PF_INET, SOCK_STREAM, 0);
        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){
                perror("bind");
        }
        if(listen(serv_sock, 5)==-1){
                perror("listen");
        }

        while(1)
        {
                addr_size=sizeof(clnt_addr);
                clnt_sock=accept(serv_sock, (struct sockaddr*)&clnt_addr, &addr_size);
                pthread_mutex_lock(&mutx);
                clnt_socks[clnt_cnt++]=clnt_sock;
                pthread_mutex_unlock(&mutx);
                pthread_create(&t_id, NULL, handle_clnt, (void*)&clnt_sock);
                pthread_detach(t_id);
                printf("Connected client ip : %s \n", inet_ntoa(clnt_addr.sin_addr));
        }
        close(serv_sock);
        return 0;
}

void *handle_clnt(void *arg){
        int clnt_sock=*((int*)arg);
        int str_len=0, i;
        char msg[BUF_SIZE]={0,};
        while((str_len=read(clnt_sock, msg, sizeof(msg)))!=0)
        {
                send_msg(msg, str_len);
        }
        pthread_mutex_lock(&mutx);
        for(i=0; i<clnt_cnt; i++)
        {
                if(clnt_sock==clnt_socks[i]){
                        while(i++<clnt_cnt-1)
                        {
                                clnt_socks[i]=clnt_socks[i+1];
                        }
                        break;
                }
        }
        clnt_cnt--;
        pthread_mutex_unlock(&mutx);
        close(clnt_sock);
        return NULL;
}

void send_msg(char *msg, int len){
        int i;
        pthread_mutex_lock(&mutx);
        for(i=0; i<clnt_cnt; i++)
        {
                write(clnt_socks[i], msg, len);
        }
        pthread_mutex_unlock(&mutx);
}