EnterCriticalSection

本文介绍如何使用临界区(CRITICAL_SECTION)来确保多线程环境下数据的一致性和完整性,通过示例代码详细解释了EnterCriticalSection和LeaveCriticalSection函数的作用及使用方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

多个线程操作相同的数据时,一般是需要按顺序访问的,否则会引导数据错乱,无法控制数据,变成随机变量。为解决这个问题,就需要引入互斥 变量,让每个线程都按顺序地访问变量。这样就需要使用EnterCriticalSection和LeaveCriticalSection函数。
函数 EnterCriticalSection 和 LeaveCriticalSection 声明如下:
?
1
2
3
4
5
6
WINBASEAPI
VOID
WINAPI
EnterCriticalSection(
     __inout LPCRITICAL_SECTION lpCriticalSection
);
是多线程中用来确保同一时刻只有一个线程操作被保护的数据的操作函数,相关的多线程数据操作函数还有:
?
1
2
3
4
5
6
InitializeCriticalSection(&cs); //初始化临界区
EnterCriticalSection(&cs); //进入临界区
//操作数据
MyMoney*=10; //所有访问MyMoney变量的程序都需要这样写Enter.. Leave...
LeaveCriticalSection(&cs); //离开临界区
DeleteCriticalSection(&cs); //删除临界区

访问机制

比如说我们定义了一个共享资源dwTime[100],两个线程ThreadFuncA和ThreadFuncB都对它进行读写操作。当我们想要保证 dwTime[100]的操作完整性,即不希望写到一半的数据被另一个线程读取,那么用CRITICAL_SECTION来进行线程同步如下:
第一个线程函数:
DWORD WINAPI ThreadFuncA(LPVOID lp)
{
EnterCriticalSection(&cs);
...
// 操作dwTime
...
LeaveCriticalSection(&cs);
return 0;
}
写出这个函数之后,很多初学者都会错误地以为,此时cs对dwTime进行了锁定操作,dwTime处于cs的保护之中。一个“自然而然”的想法就是——cs和dwTime一一对应上了。
这么想,就大错特错了。dwTime并没有和任何东西对应,它仍然是任何其它线程都可以访问的。如果你像如下的方式来写第二个线程,那么就会有问题:
DWORD WINAPI ThreadFuncB(LPVOID lp)
{
...
// 操作dwTime
...
return 0;
}
当线程ThreadFuncA执行了EnterCriticalSection(&cs),并开始操作dwTime[100]的时候,线程 ThreadFuncB可能随时醒过来,也开始操作dwTime[100],这样,dwTime[100]中的数据就被破坏了。
为了让CRITICAL_SECTION发挥作用,我们必须在访问dwTime的任何一个地方都加上 EnterCriticalSection(&cs)和LeaveCriticalSection(&cs)语句。所以,必须按照下面的 方式来写第二个线程函数:
DWORD WINAPI ThreadFuncB(LPVOID lp)
{
EnterCriticalSection(&cs);
...
// 操作dwTime
...
LeaveCriticalSection(&cs);
return 0;
}
这样,当线程ThreadFuncB醒过来时,它遇到的第一个语句是EnterCriticalSection(&cs),这个语句将对cs变量 进行访问。如果这个时候第一个线程仍然在操作dwTime[100],cs变量中包含的值将告诉第二个线程,已有其它线程占用了cs。因此,第二个线程的 EnterCriticalSection(&cs)语句将不会返回,而处于挂起等待状态。直到第一个线程执行了 LeaveCriticalSection(&cs),第二个线程的EnterCriticalSection(&cs)语句才会返回, 并且继续执行下面的操作。
这个过程实际上是通过限制有且只有一个函数进入CriticalSection变量来实现代码段同步的。简单地说,对于同一个 CRITICAL_SECTION,当一个线程执行了EnterCriticalSection而没有执行LeaveCriticalSection的时 候,其它任何一个线程都无法完全执行EnterCriticalSection而不得不处于等待状态。
再次强调一次,没有任何资源被“锁定”,CRITICAL_SECTION这个东东不是针对于资源的,而是针对于不同线程间的代码段的!我们能够用它来进 行所谓资源的“锁定”,其实是因为我们在任何访问共享资源的地方都加入了EnterCriticalSection和 LeaveCriticalSection语句,使得同一时间只能够有一个线程的代码段访问到该共享资源而已(其它想访问该资源的代码段不得不等待)。
这就是使用一个CRITICAL_SECTION时的情况。你应该要知道,它并没有什么可以同步的资源的“集合”。这个概念不正确。如果是两个CRITICAL_SECTION,就以此类推。 [1]  
线程锁的概念函数EnterCriticalSection和LeaveCriticalSection的用法
注:使用结构CRITICAL_SECTION 需加入头文件#include “afxmt.h”
定义一个全局的锁 CRITICAL_SECTION的实例
  和一个静态全局变量
CRITICAL_SECTION cs;//可以理解为锁定一个资源
static int n_AddValue = 0;//定义一个静态的全部变量n_AddValue
创建两个线程函数,代码实现如下:
//第一个线程
UINT FirstThread(LPVOID lParam)
{
EnterCriticalSection(&cs);//加锁 接下来的代码处理过程中不允许其他线程进行操作,除非遇到LeaveCriticalSection
for(int i = 0; i<10; i++){
n_AddValue ++;
cout << "n_AddValue in FirstThread is "<<n_AddValue <<endl;
}
LeaveCriticalSection(&cs);//解锁 到EnterCriticalSection之间代码资源已经释放了,其他线程可以进行操作
return 0;
}
//第二个线程
UINT SecondThread(LPVOID lParam)
{
EnterCriticalSection(&cs);//加锁
for(int i = 0; i<10; i++){
n_AddValue ++;
cout << "n_AddValue in SecondThread is "<<n_AddValue <<endl;
}
LeaveCriticalSection(&cs);//解锁
return 0;
}
在主函数添加以下代码
int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
int nRetCode = 0;
// 初始化 MFC 并在失败时显示错误
if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
{
// TODO: 更改错误代码以符合您的需要
_tprintf(_T("错误: MFC 初始化失败/n"));
nRetCode = 1;
}
else
{
InitializeCriticalSection(&cs);//初始化结构CRITICAL_SECTION
CWinThread *pFirstThread,*pSecondThread;//存储函数AfxBeginThread返回的CWinThread指针
pFirstThread = AfxBeginThread(FirstThread,LPVOID(NULL));//启动第一个线程
pSecondThread = AfxBeginThread(SecondThread,LPVOID(NULL));//启动第二个线程
HANDLE hThreadHandle[2];//
hThreadHandle[0] = pFirstThread->m_hThread;
hThreadHandle[1] = pSecondThread->m_hThread;
//等待线程返回
WaitForMultipleObjects(2,hThreadHandle,TRUE,INFINITE);
}
DeleteCriticalSection(&cs);//删除临界区
return nRetCode;
}
输出:
n_AddValue in FirstThread is 1
  n_AddValue in FirstThread is 2
  n_AddValue in FirstThread is 3
  n_AddValue in FirstThread is 4
  n_AddValue in FirstThread is 5
  n_AddValue in FirstThread is 6
  n_AddValue in FirstThread is 7
  n_AddValue in FirstThread is 8
  n_AddValue in FirstThread is 9
  n_AddValue in FirstThread is 10
  n_AddValue in SecondThread is 11
  n_AddValue in SecondThread is 12
  n_AddValue in SecondThread is 13
  n_AddValue in SecondThread is 14
  n_AddValue in SecondThread is 15
  n_AddValue in SecondThread is 16
  n_AddValue in SecondThread is 17
  n_AddValue in SecondThread is 18
  n_AddValue in SecondThread is 19
  n_AddValue in SecondThread is 20
如果把两个线程函数中的EnterCriticalSection和LeaveCriticalSection位置移到for循环中去,线程的执行顺序将会改变
  输出也就跟着改变,如:
//第一个线程
UINT FirstThread(LPVOID lParam)
{
for(int i = 0; i<10; i++){
EnterCriticalSection(&cs);//加锁 锁移到for循环内部里
n_AddValue ++;
cout << "n_AddValue in FirstThread is "<<n_AddValue <<endl;
LeaveCriticalSection(&cs);//解锁
}
return 0;
}
//第二个线程
UINT SecondThread(LPVOID lParam)
{
for(int i = 0; i<10; i++){
EnterCriticalSection(&cs);//加锁
n_AddValue ++;
cout << "n_AddValue in SecondThread is "<<n_AddValue <<endl;
LeaveCriticalSection(&cs);//解锁
}
return 0;
}
其他代码不变,输出的结果如下:
n_AddValue in FirstThread is 1
  n_AddValue in SecondThread is 2
  n_AddValue in FirstThread is 3
  n_AddValue in SecondThread is 4
  n_AddValue in FirstThread is 5
  n_AddValue in SecondThread is 6
  n_AddValue in FirstThread is 7
  n_AddValue in SecondThread is 8
  n_AddValue in FirstThread is 9
  n_AddValue in SecondThread is 10
  n_AddValue in FirstThread is 11
  n_AddValue in SecondThread is 12
  n_AddValue in FirstThread is 13
  n_AddValue in SecondThread is 14
  n_AddValue in FirstThread is 15
  n_AddValue in SecondThread is 16
  n_AddValue in FirstThread is 17
  n_AddValue in SecondThread is 18
  n_AddValue in FirstThread is 19
  n_AddValue in SecondThread is 20
个人认为在函数EnterCriticalSection和LeaveCriticalSection中间的代码执行过程不会被其他线程干拢或者这么讲不允许其他线程中
  的代码执行。这样可以有效防止一个全局变量在两个线程中同时被操作的可能性 [2
#include "Communication.h" #include <iostream> CommunicationBase::CommunicationBase() : m_socket(INVALID_SOCKET), m_running(false), m_receiveThread(NULL), m_queueEvent(NULL) { InitializeCriticalSection(&m_queueCS); m_queueEvent = CreateEvent(NULL, FALSE, FALSE, NULL); } CommunicationBase::~CommunicationBase() { CloseSocket(); DeleteCriticalSection(&m_queueCS); if (m_queueEvent) CloseHandle(m_queueEvent); } void CommunicationBase::CloseSocket() { if (m_socket != INVALID_SOCKET) { m_running = false; // 关闭套接字会终止接收线程 shutdown(m_socket, SD_BOTH); closesocket(m_socket); m_socket = INVALID_SOCKET; if (m_receiveThread) { WaitForSingleObject(m_receiveThread, INFINITE); CloseHandle(m_receiveThread); m_receiveThread = NULL; } } } bool CommunicationBase::SendMessageTcp(const std::string& message) { if (m_socket == INVALID_SOCKET || !m_running) return false; const char* data = message.c_str(); int length = static_cast<int>(message.size()); int totalSent = 0; while (totalSent < length) { int sent = send(m_socket, data + totalSent, length - totalSent, 0); if (sent == SOCKET_ERROR) { std::cerr << "发送失败: " << WSAGetLastError() << std::endl; return false; } totalSent += sent; } return true; } //bool SendMessageTcp(SOCKET socket, const std::string& message) { // const char* data = message.data(); // int total = static_cast<int>(message.size()); // int sent = 0; // // while (sent < total && m_running) { // int bytes = send(socket, data + sent, total - sent, 0); // // if (bytes == SOCKET_ERROR) { // int error = WSAGetLastError(); // if (error == WSAEWOULDBLOCK) { // // 等待套接字可写 // fd_set writeSet; // FD_ZERO(&writeSet); // FD_SET(socket, &writeSet); // // timeval timeout{ 0, 10000 }; // 10ms超时 // if (select(0, NULL, &writeSet, NULL, &timeout) <= 0) { // continue; // 超时或错误 // } // } // else { // // 处理其他错误 // closesocket(socket); // return false; // } // } // else { // sent += bytes; // } // } // return sent == total; //} std::string CommunicationBase::ReceiveMessage(DWORD timeout_ms) { EnterCriticalSection(&m_queueCS); if (m_messageQueue.empty()) { LeaveCriticalSection(&m_queueCS); WaitForSingleObject(m_queueEvent, timeout_ms); EnterCriticalSection(&m_queueCS); } if (m_messageQueue.empty()) { LeaveCriticalSection(&m_queueCS); return ""; } std::string msg = m_messageQueue.front(); m_messageQueue.pop(); LeaveCriticalSection(&m_queueCS); return msg; } DWORD WINAPI CommunicationBase::ReceiveThreadProc(LPVOID lpParam) { CommunicationBase* pThis = reinterpret_cast<CommunicationBase*>(lpParam); char buffer[4096]; int bytesReceived; while (pThis->m_running) { bytesReceived = recv(pThis->m_socket, buffer, sizeof(buffer) - 1, 0); if (bytesReceived == SOCKET_ERROR) { int error = WSAGetLastError(); if (error == WSAEWOULDBLOCK) { Sleep(10); continue; } break; } if (bytesReceived <= 0) { break; } buffer[bytesReceived] = '\0'; std::string msg(buffer); EnterCriticalSection(&pThis->m_queueCS); pThis->m_messageQueue.push(msg); SetEvent(pThis->m_queueEvent); LeaveCriticalSection(&pThis->m_queueCS); } pThis->m_running = false; return 0; } Server::Server(int port) : m_port(port), m_acceptThread(NULL) { InitializeCriticalSection(&m_clientsCS); m_socket = INVALID_SOCKET; } Server::~Server() { Stop(); DeleteCriticalSection(&m_clientsCS); } bool Server::Start() { WSADATA wsaData; if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) { std::cerr << "WSAStartup失败" << std::endl; return false; } m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (m_socket == INVALID_SOCKET) { std::cerr << "创建套接字失败" << std::endl; WSACleanup(); return false; } sockaddr_in serverAddr; serverAddr.sin_family = AF_INET; serverAddr.sin_addr.s_addr = INADDR_ANY; serverAddr.sin_port = htons(m_port); if (bind(m_socket, (sockaddr*)& serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) { std::cerr << "绑定失败: " << WSAGetLastError() << std::endl; closesocket(m_socket); WSACleanup(); return false; } if (listen(m_socket, SOMAXCONN) == SOCKET_ERROR) { std::cerr << "监听失败" << std::endl; closesocket(m_socket); WSACleanup(); return false; } // 设置非阻塞模式 u_long mode = 1; ioctlsocket(m_socket, FIONBIO, &mode); m_running = true; // 启动接受客户端线程 m_acceptThread = CreateThread(NULL, 0, AcceptThreadProc, this, 0, NULL); if (m_acceptThread == NULL) { std::cerr << "创建接受线程失败" << std::endl; closesocket(m_socket); WSACleanup(); return false; } std::cout << "服务器已启动,监听端口: " << m_port << std::endl; return true; } void Server::Stop() { m_running = false; // 关闭所有客户端套接字 EnterCriticalSection(&m_clientsCS); for (size_t i = 0; i < m_clientSockets.size(); ++i) { if (m_clientSockets[i] != INVALID_SOCKET) { shutdown(m_clientSockets[i], SD_BOTH); closesocket(m_clientSockets[i]); } } m_clientSockets.clear(); LeaveCriticalSection(&m_clientsCS); CloseSocket(); if (m_acceptThread) { WaitForSingleObject(m_acceptThread, INFINITE); CloseHandle(m_acceptThread); m_acceptThread = NULL; } WSACleanup(); std::cout << "服务器已停止" << std::endl; } DWORD WINAPI Server::AcceptThreadProc(LPVOID lpParam) { Server* pServer = reinterpret_cast<Server*>(lpParam); pServer->AcceptClients(); return 0; } DWORD WINAPI Server::ClientThreadProc(LPVOID lpParam) { ClientThreadParam* param = reinterpret_cast<ClientThreadParam*>(lpParam); param->server->HandleClient(param->clientSocket); delete param; return 0; } void Server::AcceptClients() { while (m_running) { sockaddr_in clientAddr; int addrLen = sizeof(clientAddr); SOCKET clientSocket = accept(m_socket, (sockaddr*)& clientAddr, &addrLen); if (clientSocket == INVALID_SOCKET) { int error = WSAGetLastError(); if (error == WSAEWOULDBLOCK) { Sleep(100); continue; } break; } // 设置非阻塞模式a u_long mode = 1; ioctlsocket(clientSocket, FIONBIO, &mode); EnterCriticalSection(&m_clientsCS); m_clientSockets.push_back(clientSocket); LeaveCriticalSection(&m_clientsCS); char ipStr[INET_ADDRSTRLEN]; inet_ntop(AF_INET, &clientAddr.sin_addr, ipStr, INET_ADDRSTRLEN); std::cout << "新客户端连接: " << ipStr << ":" << ntohs(clientAddr.sin_port) << std::endl; // 创建线程参数 ClientThreadParam* param = new ClientThreadParam; param->server = this; param->clientSocket = clientSocket; // 创建客户端处理线程 HANDLE hThread = CreateThread( NULL, 0, ClientThreadProc, param, 0, NULL ); if (hThread == NULL) { std::cerr << "创建客户端线程失败: " << GetLastError() << std::endl; delete param; closesocket(clientSocket); } else { CloseHandle(hThread); // 立即关闭线程句柄,不保留 } } } void Server::HandleClient(SOCKET clientSocket) { char buffer[4096]; int bytesReceived; while (m_running) { bytesReceived = recv(clientSocket, buffer, sizeof(buffer) - 1, 0); if (bytesReceived == SOCKET_ERROR) { int error = WSAGetLastError(); if (error == WSAEWOULDBLOCK) { Sleep(10); continue; } break; } if (bytesReceived <= 0) { break; } buffer[bytesReceived] = '\0'; std::string msg(buffer); // 添加到消息队列 EnterCriticalSection(&m_queueCS); m_messageQueue.push(msg); SetEvent(m_queueEvent); LeaveCriticalSection(&m_queueCS); std::cout << "收到消息: " << msg << std::endl; // 广播给其他客户端 Broadcast(msg, clientSocket); } // 移除断开连接的客户端 EnterCriticalSection(&m_clientsCS); for (std::vector<SOCKET>::iterator it = m_clientSockets.begin(); it != m_clientSockets.end(); ++it) { if (*it == clientSocket) { closesocket(clientSocket); m_clientSockets.erase(it); break; } } LeaveCriticalSection(&m_clientsCS); std::cout << "客户端断开连接" << std::endl; } void Server::StartReceiver() { // 服务器使用每个客户端单独的线程处理接收 } void Server::Broadcast(const std::string& message, SOCKET excludeSocket) { EnterCriticalSection(&m_clientsCS); for (size_t i = 0; i < m_clientSockets.size(); ++i) { if (m_clientSockets[i] != excludeSocket && m_clientSockets[i] != INVALID_SOCKET) { send(m_clientSockets[i], message.c_str(), static_cast<int>(message.size()), 0); } } LeaveCriticalSection(&m_clientsCS); } Client::Client(const std::string& ip, int port) : m_serverIP(ip), m_port(port) { m_socket = INVALID_SOCKET; } Client::~Client() { Disconnect(); } bool Client::Connect() { WSADATA wsaData; if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) { std::cerr << "WSAStartup failed" << std::endl; return false; } m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (m_socket == INVALID_SOCKET) { std::cerr << "Socket creation failed" << std::endl; WSACleanup(); return false; } sockaddr_in serverAddr; serverAddr.sin_family = AF_INET; serverAddr.sin_port = htons(m_port); inet_pton(AF_INET, m_serverIP.c_str(), &serverAddr.sin_addr); if (connect(m_socket, (sockaddr*)& serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) { std::cerr << "Connection failed: " << WSAGetLastError() << std::endl; closesocket(m_socket); WSACleanup(); return false; } // 设置非阻塞模式 u_long mode = 1; ioctlsocket(m_socket, FIONBIO, &mode); m_running = true; StartReceiver(); std::cout << "Connected to server: " << m_serverIP << ":" << m_port << std::endl; return true; } void Client::Disconnect() { m_running = false; CloseSocket(); WSACleanup(); std::cout << "Disconnected from server" << std::endl; } void Client::StartReceiver() { m_receiveThread = CreateThread(NULL, 0, ReceiveThreadProc, this, 0, NULL); if (m_receiveThread == NULL) { std::cerr << "Failed to create receive thread" << std::endl; m_running = false; } } // 明确的发送接口 bool Client::Send(const std::string& message) { if (m_socket == INVALID_SOCKET || !m_running) { return false; } return SendMessageTcp(message); } // 阻塞式接收 bool Client::Receive(std::string& outMessage, DWORD timeout_ms) { outMessage = ReceiveMessage(timeout_ms); return !outMessage.empty(); } // 非阻塞式接收 bool Client::TryReceive(std::string& outMessage) { EnterCriticalSection(&m_queueCS); if (m_messageQueue.empty()) { LeaveCriticalSection(&m_queueCS); return false; } outMessage = m_messageQueue.front(); m_messageQueue.pop(); LeaveCriticalSection(&m_queueCS); return true; } 仔细阅读代码
最新发布
06-25
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值