把握命運，追逐夢想

對自己所做的事要有興趣，同時還要能夠堅持不懈

隨筆檔案

統計

隨筆 - 49
文章 - 0
評論 - 6
引用 - 0

留言簿(1)

閱讀排行榜

評論排行榜

一個簡單的網絡小程序

//tcpsever.cpp

#include"targetver.h"

#define WIN32_LEAN_AND_MEAN

#include<windows.h>

#include<commctrl.h>

#include<tchar.h>

#include<winsock2.h>

#include"resource.h"

#include<string>

HINSTANCE g_hInst = NULL; // 當前實例

HWND g_hWndDLG = NULL;

HANDLE g_hIconSm= NULL;

HANDLE g_hIconBg = NULL;

HANDLE g_hEnInfo = NULL;

SOCKET g_sServer = INVALID_SOCKET;

SOCKET g_sClient1 = INVALID_SOCKET;

SOCKET g_sClient2 = INVALID_SOCKET;

std::wstring wstrInfo;

TCHAR DataBuffer[256];

// 此代碼模塊中包含的函數的前向聲明:

BOOL InitWindow();

BOOL InitSock();

void AcceptClient();

void RecvInfo();

void CleanupSock();

INT_PTR CALLBACK About(HWND, UINT, WPARAM, LPARAM);

int APIENTRY _tWinMain(HINSTANCE hInstance,

HINSTANCE hPrevInstance,

LPTSTR lpCmdLine,

int nCmdShow)

{

UNREFERENCED_PARAMETER(hPrevInstance);

UNREFERENCED_PARAMETER(lpCmdLine);

g_hInst = hInstance;

if(!InitSock())

{

return FALSE;

}

if(!InitWindow())

{

return FALSE;

}

MSG msg = {0};

// 主消息循環:

while (msg.message != WM_QUIT)

{

if(PeekMessage(&msg, g_hWndDLG, 0, 0,PM_REMOVE))

{

if(IsDialogMessage(g_hWndDLG,&msg))

{

wstrInfo.append(L"i");

}

else

{

TranslateMessage(&msg);

DispatchMessage(&msg);

}

else

{

/*AcceptClient();

RecvInfo();*/

}

CleanupSock();

return (int) msg.wParam;

}

// “關于”框的消息處理程序。

INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)

{

switch (message)

{

//對于WM_INITDIALOG消息,必須要返回TRUE,無論是否處理這個消息,系統都會處理這個消息的.

case WM_INITDIALOG:

//todo:設置初始化時對話框,以及控件的初始狀態值

//這里沒有辦法seticon，因為此時對話框句柄還沒得到，無法指明窗口,似乎需要窗口句柄的都無法在這里初始化

//暈，犯糊涂，上面說錯了，消息發送的時候就帶著窗口句柄呢！

SendMessage(hDlg,WM_SETICON,ICON_SMALL,(LPARAM)g_hIconSm);

SendMessage(hDlg,WM_SETICON,ICON_BIG,(LPARAM)g_hIconBg);

return (INT_PTR)TRUE;

case WM_COMMAND:

switch (LOWORD(wParam))

{

case IDCANCEL:

DestroyWindow(g_hWndDLG);

return (INT_PTR)TRUE;

break;

}

break;

case WM_CLOSE:

MessageBox(hDlg,L"X按鈕阻止使用",L"無效",MB_ICONINFORMATION);

return (INT_PTR)TRUE;

break;

case WM_DESTROY:

PostQuitMessage(0);

return (INT_PTR)TRUE;

break;

}

return (INT_PTR)FALSE;

}

BOOL InitWindow()

{

//使用LoadImage有很多好處，它可以根據參數中的寬高來從具有多個image的文件中選擇合適大小的image

//盡量避免調用LoadIcon，幾次試驗發現，多次調用LoadIcon導致默認圖標被改變的問題

g_hIconSm = LoadImage(g_hInst,MAKEINTRESOURCE(IDI_ICON1),IMAGE_ICON,16,

16,LR_DEFAULTCOLOR);

g_hIconBg = LoadImage(g_hInst,MAKEINTRESOURCE(IDI_ICON1),IMAGE_ICON,32,

32,LR_DEFAULTCOLOR);

g_hWndDLG = CreateDialog(g_hInst, MAKEINTRESOURCE(IDD_DIALOGSERVER), NULL, About);

if(g_hWndDLG == NULL)

{

return FALSE;

}

ShowWindow(g_hWndDLG,SW_SHOW);

return TRUE;

}

BOOL InitSock()

{

WSADATA wsaData;

if((WSAStartup(MAKEWORD(2,2),&wsaData)) != 0)

{

MessageBox(g_hWndDLG,L"網絡模塊加載失敗,程序馬上退出",L"失敗",MB_ICONINFORMATION);

return FALSE;

}

if((g_sServer = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP))== INVALID_SOCKET)

{

MessageBox(g_hWndDLG,L"套接字初始化失敗,程序馬上退出",L"失敗",MB_ICONINFORMATION);

WSACleanup();

return FALSE;

}

SOCKADDR_IN serverAddr;

int port = 5150;

serverAddr.sin_family = AF_INET;

serverAddr.sin_port = htons(port);

serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);

if (bind(g_sServer, (SOCKADDR *)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR)

{

MessageBox(g_hWndDLG,L"套接字綁定失敗,程序馬上退出",L"失敗",MB_ICONINFORMATION);

closesocket(g_sServer);

WSACleanup();

return FALSE;

}

if(listen(g_sServer, 2) == SOCKET_ERROR)

{

MessageBox(g_hWndDLG,L"監聽請求失敗,程序馬上退出",L"失敗",MB_ICONINFORMATION);

closesocket(g_sServer);

WSACleanup();

return FALSE;

}

u_long iMode = 0;

ioctlsocket(g_sServer, FIONBIO, &iMode);

return TRUE;

}

void AcceptClient()

{

SOCKADDR_IN clientAddr;

int clientAddrLen;

if(g_sClient1 == INVALID_SOCKET)

{

if ((g_sClient1 = accept(g_sServer, (SOCKADDR *) &clientAddr, &clientAddrLen)) == INVALID_SOCKET)

{

if( WSAEWOULDBLOCK != WSAGetLastError() )

{

wstrInfo.append(L"\n接受請求失敗");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

else

{

return;

}

else

{

wstrInfo.append(L"\n接受請求成功");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

if(g_sClient2 == INVALID_SOCKET)

{

if ((g_sClient2 = accept(g_sServer, (SOCKADDR *) &clientAddr, &clientAddrLen)) == INVALID_SOCKET)

{

if( WSAEWOULDBLOCK != WSAGetLastError() )

{

wstrInfo.append(L"\n接受請求失敗");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

else

{

return;

}

else

{

wstrInfo.append(L"\n接受請求成功");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

void RecvInfo()

{

int ret = 0;

if(g_sClient1 != SOCKET_ERROR)

{

if (ret = recv(g_sClient1, (char*)DataBuffer, sizeof(DataBuffer), 0) == SOCKET_ERROR)

{

wstrInfo.append(L"\nrecv失敗");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

else if(ret == 0)

{

closesocket(g_sClient1);

g_sClient1 = INVALID_SOCKET;

wstrInfo.append(L"\n連接斷開");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

else

{

wstrInfo.append(L"\n");

wstrInfo.append(DataBuffer,ret/2);

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

if(g_sClient2 != SOCKET_ERROR)

{

if (ret = recv(g_sClient2, (char*)DataBuffer, sizeof(DataBuffer), 0) == SOCKET_ERROR)

{

wstrInfo.append(L"\nrecv失敗");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

else if(ret == 0)

{

closesocket(g_sClient2);

g_sClient2 = INVALID_SOCKET;

wstrInfo.append(L"\n連接斷開");

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

else

{

wstrInfo.append(L"\n");

wstrInfo.append(DataBuffer,ret/2);

SetDlgItemText(g_hWndDLG,IDC_ENINFO,wstrInfo.c_str());

}

void CleanupSock()

{

if(g_sClient1 != INVALID_SOCKET)

{

closesocket(g_sClient1);

g_sClient1 = INVALID_SOCKET;

}

if(g_sClient2 != INVALID_SOCKET)

{

closesocket(g_sClient2);

g_sClient2 = INVALID_SOCKET;

}

if(g_sServer != INVALID_SOCKET )

{

closesocket(g_sServer);

g_sServer = INVALID_SOCKET;

}

WSACleanup();

}

posted on 2009-07-30 07:22 把握命運閱讀(175) 評論(0) 編輯收藏引用

只有注冊用戶登錄后才能發表評論。
【推薦】100%開源！大型工業跨平臺軟件C++源碼提供，建模，組態！



網站導航: 博客園 IT新聞 BlogJava 博問 Chat2DB 管理

青青草原综合久久大伊人导航_色综合久久天天综合_日日噜噜夜夜狠狠久久丁香五月_热久久这里只有精品

把握命運，追逐夢想

導航

隨筆檔案

統計

留言簿(1)

閱讀排行榜

評論排行榜

一個簡單的網絡小程序