Форум программистов [Powered by Invision Power Board]

Код

#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <winsock2.h>

HINSTANCE hInst;
HWND hWnd;

LRESULT CALLBACK    WndProc(HWND, UINT, WPARAM, LPARAM);

char szAppName[]="Example";

int WINAPI WinMain(HINSTANCE hInstance,HINSTANCE,LPTSTR lpCmdLine,int nCmdShow)
{
    MSG msg;
    WNDCLASSEX    wndclass;

    wndclass.cbSize = sizeof(wndclass);
    wndclass.style    = CS_HREDRAW | CS_VREDRAW;
    wndclass.lpfnWndProc = WndProc;
    wndclass.cbClsExtra    = 0;
    wndclass.cbWndExtra    = 0;
    wndclass.hInstance    =    hInstance;
    wndclass.hIcon        =    LoadIcon(NULL, IDI_APPLICATION);
    wndclass.hCursor    =    LoadCursor(NULL, IDC_ARROW);
    wndclass.hbrBackground    =    (HBRUSH) GetStockObject(WHITE_BRUSH);
    wndclass.lpszMenuName    =    NULL;
    wndclass.lpszClassName    =    szAppName;
    wndclass.hIconSm    =    LoadIcon(NULL, IDI_APPLICATION);

    RegisterClassEx(&wndclass);
    hWnd = CreateWindow(szAppName,"WSAAsyncSelect Example", WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);
    if (!hWnd)return FALSE;

    ShowWindow(hWnd, nCmdShow);
    UpdateWindow(hWnd);
    
    WSADATA       wsd;

    if (WSAStartup(MAKEWORD(2,2), &wsd) != 0)
    {
        MessageBox(0, "Can't load WinSock", "Error", 0);
        return 0;
    }

    SOCKET        sServerListen,
                  sClient;
    struct sockaddr_in localaddr,
                       clientaddr;
    HANDLE        hThread;
    DWORD         dwThreadId;
    int           iSize;

    sServerListen = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
    if (sServerListen == SOCKET_ERROR)
    {
        MessageBox(0, "Can't load WinSock", "Error", 0);
        return 0;
    }

    ULONG ulBlock;
    ulBlock = 1;
    if (ioctlsocket(sServerListen, FIONBIO, &ulBlock) == SOCKET_ERROR)
    {
        return 0;
    }

    localaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    localaddr.sin_family = AF_INET;
    localaddr.sin_port = htons(7777);

    if (bind(sServerListen, (struct sockaddr *)&localaddr, 
            sizeof(localaddr)) == SOCKET_ERROR)
    {
        MessageBox(0, "Can't bind", "Error", 0);
        return 1;
    }
    
    WSAAsyncSelect(sServerListen, hWnd, WM_USER+1, FD_ACCEPT);
    listen(sServerListen, 4);

    while (GetMessage(&msg, NULL, 0, 0)) 
    {
            TranslateMessage(&msg);
            DispatchMessage(&msg);
    }

    closesocket(sServerListen);
    WSACleanup();
    return (int) msg.wParam;
}

LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    int wmId, wmEvent;
    PAINTSTRUCT ps;
    HDC hdc;

    SOCKET ClientSocket;
    int ret;
    char szRecvBuff[1024], szSendBuff[1024];
    

    switch (message) 
    {
    case WM_USER+1:
        switch (WSAGETSELECTEVENT(lParam))
        {
        case FD_ACCEPT:
            ClientSocket = accept(wParam, 0, 0);
            WSAAsyncSelect(ClientSocket, hWnd, WM_USER+1, FD_READ | FD_WRITE | FD_CLOSE);
            break;

        case FD_READ:
            ZeroMemory(szRecvBuff,1024);
            ZeroMemory(szSendBuff,1024);
            ret = recv(wParam, szRecvBuff, 1024, 0);
            if (ret == 0)
                break;
            else if (ret == SOCKET_ERROR)
            {
                MessageBox(0, "Recive data filed", "Error", 0);
                break;
            }
            szRecvBuff[ret] = '\0';

            if(szRecvBuff[0]==0x03)closesocket(wParam);
            if(szRecvBuff[0]==0x0D){szRecvBuff[0]='~';szRecvBuff[1]='\0';}

            strncpy(szSendBuff,szRecvBuff,ret);
            szRecvBuff[ret] = '\0';
            ret=send(wParam, szSendBuff,ret, 0);
            break;

        case FD_WRITE://SOCKET READY  to send data
            break;

        case FD_CLOSE:
            closesocket(wParam);
            break;
    }
    case WM_COMMAND:
        wmId    = LOWORD(wParam); 
        wmEvent = HIWORD(wParam); 

    case WM_PAINT:
        hdc = BeginPaint(hWnd, &ps);

        EndPaint(hWnd, &ps);
        break;
    case WM_DESTROY:
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hWnd, message, wParam, lParam);
    }
    return 0;
}