第23章 尝试互联网（1）

while(TRUE)
{
    SOCKET sc=accept(hListenSocket,NULL,0);

    if (sc==INVALID_SOCKET)  break;

    //在这里创建一个新线程，对新套接字进行通信，以便马上能处理新连接。
    //但这里直接对新连接进行数据收发，因为其他客户的连接请求可能没办法及时处理。
    //新套接字可以通过lParam参数传递线程函数。
    //可以用其他线程中closesocket这个监听套接字,表示不再进行监听。这样accept
    //会返回INVALID_SOCKET，这样程序就可以退出循环。（注意关闭的是监听套接字）
}

/*-----------------------------------------------------------------
   TCPECHO.C —— 一个简易的TCP服务器程序(将收到的字符发回给客户端)
                 (c)浅墨浓香，2015.6.24
-----------------------------------------------------------------*/
#include <Windows.h>
#include "resource.h"

#pragma comment(lib,"Ws2_32.lib")
//typedef struct
//{
//    HWND hwnd;
//    int iTcp_Port;
//    SOCKET hSocket;
//}SOCKETPARAM;

#define TCP_PORT  9999  //监听端口
#define F_STOP    1


TCHAR szAppName[] = TEXT("TcpEcho");
int g_iThreadCount = 0;
HWND g_hwnd = NULL; //对话框句柄
int g_dwFlag=0; //退出标志

BOOL CALLBACK DlgProc(HWND, UINT, WPARAM, LPARAM);

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)
{
    DialogBox(hInstance, TEXT("TCPECHO"), NULL, DlgProc);
    return 0;
}

//通信服务线程，每个客户端登录的连接将产生一个线程
DWORD WINAPI ServiceThread(PVOID pVoid)
{
    SOCKET SrvSocket = (SOCKET)pVoid;
    char szBuffer[512];
    FD_SET fds;
    TIMEVAL tv;
    int iRet;
    
    ++g_iThreadCount;
    SetDlgItemInt(g_hwnd, IDC_COUNT, g_iThreadCount, FALSE);


    while (!(g_dwFlag & F_STOP))
    {
        /*
         *select函数用于判断套接字是否有可读、可读或是否有异常.
         *如果检测的就绪时，会返回就数据套接字的数量。
         *如果因超时而返回时，返回值是0,
         *如果因异常（如连接断开）时，返回SOCKET_ERROR
        */
        fds.fd_count = 1;
        fds.fd_array[0] = SrvSocket;
        tv.tv_usec = 200 * 1000; //200ms
        tv.tv_sec = 0;

        iRet = select(0, &fds, NULL, NULL, &tv); //因默认recv和send是阻塞的，用该函数来检测
                                                 //当套接字可读或可写时再去recv或send以防止
                                                 //该线程被阻塞，因为一旦被阻塞，当用户关闭
                                                 //应用程序时，g_dwFlag退出标志就无法被检测
                                                 //到，而如果客户端没发送数据时，会使该线程
                                                 //无法退出。所以用select加以检测。（很重要！）

        if (SOCKET_ERROR ==iRet)
             break;

        if (iRet)
        {
            if (SOCKET_ERROR == recv(SrvSocket, szBuffer, sizeof(szBuffer), 0))
                break;

            if (SOCKET_ERROR == send(SrvSocket, szBuffer, iRet, 0))
                break;
        }
    }
    closesocket(SrvSocket);
    --g_iThreadCount;
    SetDlgItemInt(g_hwnd, IDC_COUNT, g_iThreadCount, FALSE);
    return TRUE;
}

//监听线程
DWORD WINAPI ListenThread(PVOID pVoid)
{
    SOCKET ServiceSocket,ListenSocket;
    SOCKADDR_IN sa;
    HANDLE  hThread;
    
    TCHAR szErrorBind[] = TEXT("无法绑定到TCP端口9999，请检查是否有其它程序在使用!");

    //创建socket
    ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    *(SOCKET*)pVoid = ListenSocket;

    //绑定socket
    memset(&sa, 0, sizeof(SOCKADDR_IN));
    sa.sin_port = htons(TCP_PORT);
    sa.sin_family = AF_INET;
    sa.sin_addr.S_un.S_addr = INADDR_ANY;
    
    if (bind(ListenSocket, (PSOCKADDR)&sa, sizeof(SOCKADDR_IN))) //返回0表示无错误，是成功的。
    {
        MessageBox(g_hwnd, szErrorBind, szAppName, MB_OK | MB_ICONSTOP);
        closesocket(ListenSocket);
        return FALSE;
        //ExitProcess(0);
    }

    //开始监听，等待连接并为每个连接创建一个新的服务线程
    listen(ListenSocket, 5);
    
    while (TRUE)
    {
        ServiceSocket = accept(ListenSocket, NULL, 0);

        if (ServiceSocket == INVALID_SOCKET)
            break;

        hThread = CreateThread(NULL, 0, ServiceThread, (LPVOID)ServiceSocket, 0, 0);
        CloseHandle(hThread);//线程是内核对象，关闭表示不需用操作了（如唤醒、挂机）。
    }

    closesocket(ListenSocket);
    return TRUE;
}

BOOL CALLBACK DlgProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    WSADATA WSAData;
    static SOCKET ListenSocket;
    static HANDLE hListenThread;

    switch (message)
    {
    case WM_INITDIALOG:
        g_hwnd = hwnd;

        //载入WS2_32.DLL动态链0x0002:MAKEWORD(2,0)
        WSAStartup(MAKEWORD(2, 0), &WSAData); //动态库的信息返回到WSAdata变量中
        
        //创建监听线程
        hListenThread = CreateThread(NULL, 0, ListenThread, (LPVOID)&ListenSocket, 0, 0);
        CloseHandle(hListenThread);  //只是关闭了一个线程句柄对象，表示我不再使用该句柄,即不对这个句柄对
                                     //应的线程做任何干预了(如挂起或唤醒)。并没有结束线程。
        return TRUE;

    case WM_CLOSE:
        closesocket(ListenSocket); //当未有客户端连接时，该socket在线程中创建，且未退出线程。
                                    //所以要在这里监听socket，此时会将accept返回失败，监听线程退出。
        g_dwFlag |= F_STOP;         //设置退出标志，以便让服务线程中止
        while (g_iThreadCount > 0); //等待服务线程关闭
        WSACleanup();
        EndDialog(hwnd, 0);
        return TRUE;
    }
    return FALSE;
}

//{{NO_DEPENDENCIES}}
// Microsoft Visual C++ 生成的包含文件。
// 供 TcpEcho.rc 使用
//
#define IDC_COUNT                       1001

// Next default values for new objects
// 
#ifdef APSTUDIO_INVOKED
#ifndef APSTUDIO_READONLY_SYMBOLS
#define _APS_NEXT_RESOURCE_VALUE        102
#define _APS_NEXT_COMMAND_VALUE         40001
#define _APS_NEXT_CONTROL_VALUE         1002
#define _APS_NEXT_SYMED_VALUE           101
#endif
#endif

// Microsoft Visual C++ generated resource script.
//
#include "resource.h"

#define APSTUDIO_READONLY_SYMBOLS
/////////////////////////////////////////////////////////////////////////////
//
// Generated from the TEXTINCLUDE 2 resource.
//
#include "winres.h"

/////////////////////////////////////////////////////////////////////////////
#undef APSTUDIO_READONLY_SYMBOLS

/////////////////////////////////////////////////////////////////////////////
// 中文(简体，中国) resources

#if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_CHS)
LANGUAGE LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED

#ifdef APSTUDIO_INVOKED
/////////////////////////////////////////////////////////////////////////////
//
// TEXTINCLUDE
//

1 TEXTINCLUDE 
BEGIN
    "resource.h\0"
END

2 TEXTINCLUDE 
BEGIN
    "#include ""winres.h""\r\n"
    "\0"
END

3 TEXTINCLUDE 
BEGIN
    "\r\n"
    "\0"
END

#endif    // APSTUDIO_INVOKED


/////////////////////////////////////////////////////////////////////////////
//
// Dialog
//

TCPECHO DIALOGEX 0, 0, 165, 36
STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | WS_POPUP | WS_CAPTION | WS_SYSMENU
CAPTION "TcpEcho服务器"
FONT 8, "MS Shell Dlg", 400, 0, 0x1
BEGIN
    LTEXT           "当前连线的客户端数量：",IDC_STATIC,15,15,89,8
    LTEXT           "0",IDC_COUNT,109,15,44,8
END


/////////////////////////////////////////////////////////////////////////////
//
// DESIGNINFO
//

#ifdef APSTUDIO_INVOKED
GUIDELINES DESIGNINFO
BEGIN
    "TCPECHO", DIALOG
    BEGIN
        LEFTMARGIN, 7
        RIGHTMARGIN, 158
        TOPMARGIN, 7
        BOTTOMMARGIN, 29
    END
END
#endif    // APSTUDIO_INVOKED

#endif    // 中文(简体，中国) resources
/////////////////////////////////////////////////////////////////////////////



#ifndef APSTUDIO_INVOKED
/////////////////////////////////////////////////////////////////////////////
//
// Generated from the TEXTINCLUDE 3 resource.
//


/////////////////////////////////////////////////////////////////////////////
#endif    // not APSTUDIO_INVOKED