MFC 多线程局域网扫描工具

Plain Text
// 线程参数结构
struct ThreadData
{
CString ip; // 待扫描IP地址
int thread_id; // 线程唯一ID
BOOL thread_result; // Ping结果 TRUE=在线 FALSE=离线
MPingDlg* pDlg; // 对话框指针，用于发送消息
HANDLE hEvent; // 线程完成事件句柄
};

Plain Text
UINT ThreadFunction(LPVOID pParam)
{
ThreadData* pData = (ThreadData*)pParam;
MyPing ping;
const int SCAN_CNT = 4; // 4次重试
BOOL bPingSuccess = FALSE;

// 多重Ping重试，避免内网丢包误判
for (int i = 0; i < SCAN_CNT; i++)
{
PingResult result;
ping.Ping(pData->ip, 300, 128, result); // 超时300ms，数据包128字节
if (result.bSuccess)
{
bPingSuccess = TRUE;
break;
}
Sleep(20);
}

pData->thread_result = bPingSuccess;
// 发送自定义消息，更新主线程UI
if (pData->pDlg != NULL)
{
pData->pDlg->PostMessage(WM_THREAD_FINISH, pData->thread_id, bPingSuccess ? 1 : 0);
}

SetEvent(pData->hEvent);
return 0;
}

Plain Text
// 批量创建所有扫描线程
for(int i = 0; i < NUM_THREADS; i++)
{
threadData[i].ip.Format("%s.%d", m_strComboIP.GetBuffer(0), i);
threadData[i].thread_id = i;
threadData[i].thread_result = FALSE;
threadData[i].pDlg = this;
hEvents[i] = CreateEvent(NULL, TRUE, FALSE, NULL);
threadData[i].hEvent = hEvents[i];
pThreads[i] = AfxBeginThread(ThreadFunction, &threadData[i]);
}
m_strComboIP.ReleaseBuffer();

// 分批等待所有线程完成，防止界面卡死
while (1)
{
bool allFinished = true;
for (int i = 0; i < NUM_THREADS; i += MAXIMUM_WAIT_OBJECTS)
{
int batchCount = min(MAXIMUM_WAIT_OBJECTS, NUM_THREADS - i);
DWORD ret = WaitForMultipleObjects(
batchCount,
&hEvents[i],
TRUE,
10
);
if (ret != WAIT_OBJECT_0)
{
allFinished = false;
}
}
if (allFinished)
break;

// 消息循环，刷新界面
MSG msg;
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}

Plain Text
LRESULT MPingDlg::OnThreadFinish(WPARAM wParam, LPARAM lParam)
{
int thread_id = (int)wParam;
int result = (int)lParam;
if (result)
{
// 在线主机标记绿色
SetCellColor(thread_id / 16, thread_id % 16, RGB(0, 255, 0));
m_iIPCount++;
UpdateWindow();
UpdateCount();
}
return 0;
}

Plain Text
void MPingDlg::OnBnClickedBtnExport()
{
CString sFilePath;
CFileDialog dlg(FALSE, ".txt", "IP列表",
OFN_HIDEREADONLY, "列表文件(*.txt)|*.txt||");
if(dlg.DoModal() != IDOK) return;
sFilePath = dlg.GetPathName();

// 清理文件属性
DWORD FileAttribute = GetFileAttributes(sFilePath);
if(FileAttribute & FILE_ATTRIBUTE_READONLY)
SetFileAttributes(sFilePath, FileAttribute & ~FILE_ATTRIBUTE_READONLY);

// 写入在线IP数据
FILE* fp = fopen(sFilePath.GetBuffer(0), "wb");
int cnt = 0;
if(fp)
{
fprintf(fp, "序号 IP地址\r\n");
for(int i = 0; i < 255; i++)
{
if(m_cellColors[i / 16][i % 16] == RGB(0, 255, 0))
{
CString sIP;
sIP.Format("%s.%d", m_strComboIP, i);
fprintf(fp, "%03d %s\r\n", cnt + 1, sIP.GetBuffer(0));
cnt++;
}
}
}
fclose(fp);
}

Plain Text
char szHostName[128] = {0};
const char* pszAddr;
struct hostent* pHost;
if(gethostname(szHostName, 128) == 0)
{
pHost = gethostbyname(szHostName);
for(int i = 0; pHost != NULL && pHost->h_addr_list[i] != NULL; i++)
{
pszAddr = inet_ntoa(*(struct in_addr*)pHost->h_addr_list[i]);
CString sIP(pszAddr);
CStringArray strArray;
theApp.GetSplitArrayByString(sIP, strArray, ".");
if(strArray.GetSize() == 4)
{
CString sIPHead;
sIPHead.Format("%s.%s.%s", strArray[0], strArray[1], strArray[2]);
m_ctrlComboIP.AddString(sIPHead);
}
}
}
m_ctrlComboIP.SetCurSel(0);