wait_8cpp-source.html

著名的密码库Crypto++的文档 C++语言的杰作。程序员必备。

<a name="l00122"></a>00122                                 thread.terminate = <span class="keyword">true</span>;
<a name="l00123"></a>00123                                 threadHandles[i] = thread.threadHandle;
<a name="l00124"></a>00124                         }
<a name="l00125"></a>00125                         PulseEvent(m_startWaiting);
<a name="l00126"></a>00126                         ::WaitForMultipleObjects((DWORD)m_threads.size(), threadHandles, TRUE, INFINITE);
<a name="l00127"></a>00127                         <span class="keywordflow">for</span> (i=0; i&lt;m_threads.size(); i++)
<a name="l00128"></a>00128                                 CloseHandle(threadHandles[i]);
<a name="l00129"></a>00129                         CloseHandle(m_startWaiting);
<a name="l00130"></a>00130                         CloseHandle(m_stopWaiting);
<a name="l00131"></a>00131                 }
<a name="l00132"></a>00132         }
<a name="l00133"></a>00133         <span class="keywordflow">catch</span> (...)
<a name="l00134"></a>00134         {
<a name="l00135"></a>00135         }
<a name="l00136"></a>00136 }
<a name="l00137"></a>00137 
<a name="l00138"></a>00138 
<a name="l00139"></a><a class="code" href="class_wait_object_container.html#9ce9ba96948923609861455027da7830">00139</a> <span class="keywordtype">void</span> <a class="code" href="class_wait_object_container.html#9ce9ba96948923609861455027da7830">WaitObjectContainer::AddHandle</a>(HANDLE handle, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>&amp; callStack)
<a name="l00140"></a>00140 {
<a name="l00141"></a>00141         DetectNoWait(m_handles.size(), <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"WaitObjectContainer::AddHandle()"</span>, &amp;callStack));
<a name="l00142"></a>00142         m_handles.push_back(handle);
<a name="l00143"></a>00143 }
<a name="l00144"></a>00144 
<a name="l00145"></a>00145 DWORD WINAPI WaitingThread(LPVOID lParam)
<a name="l00146"></a>00146 {
<a name="l00147"></a>00147         std::auto_ptr&lt;WaitingThreadData&gt; pThread((<a class="code" href="struct_waiting_thread_data.html">WaitingThreadData</a> *)lParam);
<a name="l00148"></a>00148         <a class="code" href="struct_waiting_thread_data.html">WaitingThreadData</a> &amp;thread = *pThread;
<a name="l00149"></a>00149         std::vector&lt;HANDLE&gt; handles;
<a name="l00150"></a>00150 
<a name="l00151"></a>00151         <span class="keywordflow">while</span> (<span class="keyword">true</span>)
<a name="l00152"></a>00152         {
<a name="l00153"></a>00153                 thread.<a class="code" href="struct_waiting_thread_data.html#986b97f33225dfe57313b7d9cda0fe2f">waitingToWait</a> = <span class="keyword">true</span>;
<a name="l00154"></a>00154                 ::WaitForSingleObject(thread.<a class="code" href="struct_waiting_thread_data.html#eda135a718184d03b6e0f7e40d80d845">startWaiting</a>, INFINITE);
<a name="l00155"></a>00155                 thread.<a class="code" href="struct_waiting_thread_data.html#986b97f33225dfe57313b7d9cda0fe2f">waitingToWait</a> = <span class="keyword">false</span>;
<a name="l00156"></a>00156 
<a name="l00157"></a>00157                 <span class="keywordflow">if</span> (thread.<a class="code" href="struct_waiting_thread_data.html#80b572f7ff1b55eedc59c36e0126eab8">terminate</a>)
<a name="l00158"></a>00158                         <span class="keywordflow">break</span>;
<a name="l00159"></a>00159                 <span class="keywordflow">if</span> (!thread.<a class="code" href="struct_waiting_thread_data.html#66baa8d0d2e0c1f75be9f8d686d3dbb8">count</a>)
<a name="l00160"></a>00160                         <span class="keywordflow">continue</span>;
<a name="l00161"></a>00161 
<a name="l00162"></a>00162                 handles.resize(thread.<a class="code" href="struct_waiting_thread_data.html#66baa8d0d2e0c1f75be9f8d686d3dbb8">count</a> + 1);
<a name="l00163"></a>00163                 handles[0] = thread.<a class="code" href="struct_waiting_thread_data.html#ba4a150f351400411734fe6490c7fc30">stopWaiting</a>;
<a name="l00164"></a>00164                 std::copy(thread.<a class="code" href="struct_waiting_thread_data.html#4e1ed9d51b8dc85a5c446f4c52855e14">waitHandles</a>, thread.<a class="code" href="struct_waiting_thread_data.html#4e1ed9d51b8dc85a5c446f4c52855e14">waitHandles</a>+thread.<a class="code" href="struct_waiting_thread_data.html#66baa8d0d2e0c1f75be9f8d686d3dbb8">count</a>, handles.begin()+1);
<a name="l00165"></a>00165 
<a name="l00166"></a>00166                 DWORD result = ::WaitForMultipleObjects((DWORD)handles.size(), &amp;handles[0], FALSE, INFINITE);
<a name="l00167"></a>00167 
<a name="l00168"></a>00168                 <span class="keywordflow">if</span> (result == WAIT_OBJECT_0)
<a name="l00169"></a>00169                         <span class="keywordflow">continue</span>;       <span class="comment">// another thread finished waiting first, so do nothing</span>
<a name="l00170"></a>00170                 SetEvent(thread.<a class="code" href="struct_waiting_thread_data.html#ba4a150f351400411734fe6490c7fc30">stopWaiting</a>);
<a name="l00171"></a>00171                 <span class="keywordflow">if</span> (!(result &gt; WAIT_OBJECT_0 &amp;&amp; result &lt; WAIT_OBJECT_0 + handles.size()))
<a name="l00172"></a>00172                 {
<a name="l00173"></a>00173                         assert(!<span class="stringliteral">"error in WaitingThread"</span>);      <span class="comment">// break here so we can see which thread has an error</span>
<a name="l00174"></a>00174                         *thread.<a class="code" href="struct_waiting_thread_data.html#4998f48f7817e5d2d61646f39dc2a3f9">error</a> = ::GetLastError();
<a name="l00175"></a>00175                 }
<a name="l00176"></a>00176         }
<a name="l00177"></a>00177 
<a name="l00178"></a>00178         <span class="keywordflow">return</span> S_OK;    <span class="comment">// return a value here to avoid compiler warning</span>
<a name="l00179"></a>00179 }
<a name="l00180"></a>00180 
<a name="l00181"></a>00181 <span class="keywordtype">void</span> WaitObjectContainer::CreateThreads(<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> count)
<a name="l00182"></a>00182 {
<a name="l00183"></a>00183         <span class="keywordtype">size_t</span> currentCount = m_threads.size();
<a name="l00184"></a>00184         <span class="keywordflow">if</span> (currentCount == 0)
<a name="l00185"></a>00185         {
<a name="l00186"></a>00186                 m_startWaiting = ::CreateEvent(NULL, TRUE, FALSE, NULL);
<a name="l00187"></a>00187                 m_stopWaiting = ::CreateEvent(NULL, TRUE, FALSE, NULL);
<a name="l00188"></a>00188         }
<a name="l00189"></a>00189 
<a name="l00190"></a>00190         <span class="keywordflow">if</span> (currentCount &lt; count)
<a name="l00191"></a>00191         {
<a name="l00192"></a>00192                 m_threads.resize(count);
<a name="l00193"></a>00193                 <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=currentCount; i&lt;count; i++)
<a name="l00194"></a>00194                 {
<a name="l00195"></a>00195                         m_threads[i] = <span class="keyword">new</span> <a class="code" href="struct_waiting_thread_data.html">WaitingThreadData</a>;
<a name="l00196"></a>00196                         <a class="code" href="struct_waiting_thread_data.html">WaitingThreadData</a> &amp;thread = *m_threads[i];
<a name="l00197"></a>00197                         thread.<a class="code" href="struct_waiting_thread_data.html#80b572f7ff1b55eedc59c36e0126eab8">terminate</a> = <span class="keyword">false</span>;
<a name="l00198"></a>00198                         thread.<a class="code" href="struct_waiting_thread_data.html#eda135a718184d03b6e0f7e40d80d845">startWaiting</a> = m_startWaiting;
<a name="l00199"></a>00199                         thread.<a class="code" href="struct_waiting_thread_data.html#ba4a150f351400411734fe6490c7fc30">stopWaiting</a> = m_stopWaiting;
<a name="l00200"></a>00200                         thread.<a class="code" href="struct_waiting_thread_data.html#986b97f33225dfe57313b7d9cda0fe2f">waitingToWait</a> = <span class="keyword">false</span>;
<a name="l00201"></a>00201                         thread.<a class="code" href="struct_waiting_thread_data.html#c6eb16bc993de39439c8e81d8c2e24d9">threadHandle</a> = CreateThread(NULL, 0, &amp;WaitingThread, &amp;thread, 0, &amp;thread.<a class="code" href="struct_waiting_thread_data.html#c0d84cfe066c1a33431bfe5ec5d1eb66">threadId</a>);
<a name="l00202"></a>00202                 }
<a name="l00203"></a>00203         }
<a name="l00204"></a>00204 }
<a name="l00205"></a>00205 
<a name="l00206"></a><a class="code" href="class_wait_object_container.html#15356e5e6f3329d8056ed7ba53290be6">00206</a> <span class="keywordtype">bool</span> <a class="code" href="class_wait_object_container.html#15356e5e6f3329d8056ed7ba53290be6">WaitObjectContainer::Wait</a>(<span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> milliseconds)
<a name="l00207"></a>00207 {
<a name="l00208"></a>00208         <span class="keywordflow">if</span> (m_noWait || (m_handles.empty() &amp;&amp; !m_firstEventTime))
<a name="l00209"></a>00209         {
<a name="l00210"></a>00210                 SetLastResult(LASTRESULT_NOWAIT);
<a name="l00211"></a>00211                 <span class="keywordflow">return</span> <span class="keyword">true</span>;
<a name="l00212"></a>00212         }
<a name="l00213"></a>00213 
<a name="l00214"></a>00214         <span class="keywordtype">bool</span> timeoutIsScheduledEvent = <span class="keyword">false</span>;
<a name="l00215"></a>00215 
<a name="l00216"></a>00216         <span class="keywordflow">if</span> (m_firstEventTime)
<a name="l00217"></a>00217         {
<a name="l00218"></a>00218                 <span class="keywordtype">double</span> timeToFirstEvent = SaturatingSubtract(m_firstEventTime, m_eventTimer.<a class="code" href="class_timer_base.html#3262a9d0815f1899701ee83c3ef8cf43">ElapsedTimeAsDouble</a>());
<a name="l00219"></a>00219 
<a name="l00220"></a>00220                 <span class="keywordflow">if</span> (timeToFirstEvent &lt;= milliseconds)
<a name="l00221"></a>00221                 {
<a name="l00222"></a>00222                         milliseconds = (<span class="keywordtype">unsigned</span> long)timeToFirstEvent;
<a name="l00223"></a>00223                         timeoutIsScheduledEvent = <span class="keyword">true</span>;
<a name="l00224"></a>00224                 }
<a name="l00225"></a>00225 
<a name="l00226"></a>00226                 <span class="keywordflow">if</span> (m_handles.empty() || !milliseconds)
<a name="l00227"></a>00227                 {
<a name="l00228"></a>00228                         <span class="keywordflow">if</span> (milliseconds)
<a name="l00229"></a>00229                                 Sleep(milliseconds);
<a name="l00230"></a>00230                         SetLastResult(timeoutIsScheduledEvent ? LASTRESULT_SCHEDULED : LASTRESULT_TIMEOUT);
<a name="l00231"></a>00231                         <span class="keywordflow">return</span> timeoutIsScheduledEvent;
<a name="l00232"></a>00232                 }
<a name="l00233"></a>00233         }
<a name="l00234"></a>00234 
<a name="l00235"></a>00235         <span class="keywordflow">if</span> (m_handles.size() &gt; MAXIMUM_WAIT_OBJECTS)
<a name="l00236"></a>00236         {
<a name="l00237"></a>00237                 <span class="comment">// too many wait objects for a single WaitForMultipleObjects call, so use multiple threads</span>
<a name="l00238"></a>00238                 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> WAIT_OBJECTS_PER_THREAD = MAXIMUM_WAIT_OBJECTS-1;
<a name="l00239"></a>00239                 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> nThreads = (<span class="keywordtype">unsigned</span> int)((m_handles.size() + WAIT_OBJECTS_PER_THREAD - 1) / WAIT_OBJECTS_PER_THREAD);
<a name="l00240"></a>00240                 <span class="keywordflow">if</span> (nThreads &gt; MAXIMUM_WAIT_OBJECTS)    <span class="comment">// still too many wait objects, maybe implement recursive threading later?</span>
<a name="l00241"></a>00241                         <span class="keywordflow">throw</span> <a class="code" href="class_wait_object_container_1_1_err.html" title="exception thrown by WaitObjectContainer">Err</a>(<span class="stringliteral">"WaitObjectContainer: number of wait objects exceeds limit"</span>);
<a name="l00242"></a>00242                 CreateThreads(nThreads);
<a name="l00243"></a>00243                 DWORD error = S_OK;
<a name="l00244"></a>00244                 
<a name="l00245"></a>00245                 <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i=0; i&lt;m_threads.size(); i++)
<a name="l00246"></a>00246                 {
<a name="l00247"></a>00247                         <a class="code" href="struct_waiting_thread_data.html">WaitingThreadData</a> &amp;thread = *m_threads[i];
<a name="l00248"></a>00248                         <span class="keywordflow">while</span> (!thread.<a class="code" href="struct_waiting_thread_data.html#986b97f33225dfe57313b7d9cda0fe2f">waitingToWait</a>)   <span class="comment">// spin until thread is in the initial "waiting to wait" state</span>
<a name="l00249"></a>00249                                 Sleep(0);
<a name="l00250"></a>00250                         <span class="keywordflow">if</span> (i&lt;nThreads)
<a name="l00251"></a>00251                         {
<a name="l00252"></a>00252                                 thread.<a class="code" href="struct_waiting_thread_data.html#4e1ed9d51b8dc85a5c446f4c52855e14">waitHandles</a> = &amp;m_handles[i*WAIT_OBJECTS_PER_THREAD];
<a name="l00253"></a>00253                                 thread.<a class="code" href="struct_waiting_thread_data.html#66baa8d0d2e0c1f75be9f8d686d3dbb8">count</a> = UnsignedMin(WAIT_OBJECTS_PER_THREAD, m_handles.size() - i*WAIT_OBJECTS_PER_THREAD);
<a name="l00254"></a>00254                                 thread.<a class="code" href="struct_waiting_thread_data.html#4998f48f7817e5d2d61646f39dc2a3f9">error</a> = &amp;error;
<a name="l00255"></a>00255                         }
<a name="l00256"></a>00256                         <span class="keywordflow">else</span>
<a name="l00257"></a>00257                                 thread.<a class="code" href="struct_waiting_thread_data.html#66baa8d0d2e0c1f75be9f8d686d3dbb8">count</a> = 0;
<a name="l00258"></a>00258                 }
<a name="l00259"></a>00259 
<a name="l00260"></a>00260                 ResetEvent(m_stopWaiting);
<a name="l00261"></a>00261                 PulseEvent(m_startWaiting);
<a name="l00262"></a>00262 
<a name="l00263"></a>00263                 DWORD result = ::WaitForSingleObject(m_stopWaiting, milliseconds);