<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head> <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> <title>Crypto++: winpipes.cpp Source File</title> <link href="tabs.css" rel="stylesheet" type="text/css"/> <link href="doxygen.css" rel="stylesheet" type="text/css"/> </head> <body> <!-- Generated by Doxygen 1.6.1 --> <div class="navigation" id="top"> <div class="tabs"> <ul> <li><a href="index.html"><span>Main Page</span></a></li> <li><a href="namespaces.html"><span>Namespaces</span></a></li> <li><a href="annotated.html"><span>Classes</span></a></li> <li class="current"><a href="files.html"><span>Files</span></a></li> </ul> </div> <div class="tabs"> <ul> <li><a href="files.html"><span>File List</span></a></li> <li><a href="globals.html"><span>File Members</span></a></li> </ul> </div> <h1>winpipes.cpp</h1><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">// winpipes.cpp - written and placed in the public domain by Wei Dai</span> <a name="l00002"></a>00002 <a name="l00003"></a>00003 <span class="preprocessor">#include "pch.h"</span> <a name="l00004"></a>00004 <span class="preprocessor">#include "winpipes.h"</span> <a name="l00005"></a>00005 <a name="l00006"></a>00006 <span class="preprocessor">#ifdef WINDOWS_PIPES_AVAILABLE</span> <a name="l00007"></a>00007 <span class="preprocessor"></span> <a name="l00008"></a>00008 <span class="preprocessor">#include "wait.h"</span> <a name="l00009"></a>00009 <a name="l00010"></a>00010 NAMESPACE_BEGIN(CryptoPP) <a name="l00011"></a>00011 <a name="l00012"></a>00012 <a class="code" href="class_windows_handle.html" title="Windows Handle.">WindowsHandle</a>::<a class="code" href="class_windows_handle.html" title="Windows Handle.">WindowsHandle</a>(HANDLE h, <span class="keywordtype">bool</span> own) <a name="l00013"></a>00013 : m_h(h), m_own(own) <a name="l00014"></a>00014 { <a name="l00015"></a>00015 } <a name="l00016"></a>00016 <a name="l00017"></a>00017 WindowsHandle::~WindowsHandle() <a name="l00018"></a>00018 { <a name="l00019"></a>00019 <span class="keywordflow">if</span> (m_own) <a name="l00020"></a>00020 { <a name="l00021"></a>00021 <span class="keywordflow">try</span> <a name="l00022"></a>00022 { <a name="l00023"></a>00023 CloseHandle(); <a name="l00024"></a>00024 } <a name="l00025"></a>00025 <span class="keywordflow">catch</span> (...) <a name="l00026"></a>00026 { <a name="l00027"></a>00027 } <a name="l00028"></a>00028 } <a name="l00029"></a>00029 } <a name="l00030"></a>00030 <a name="l00031"></a>00031 <span class="keywordtype">bool</span> WindowsHandle::HandleValid()<span class="keyword"> const</span> <a name="l00032"></a>00032 <span class="keyword"></span>{ <a name="l00033"></a>00033 <span class="keywordflow">return</span> m_h && m_h != INVALID_HANDLE_VALUE; <a name="l00034"></a>00034 } <a name="l00035"></a>00035 <a name="l00036"></a>00036 <span class="keywordtype">void</span> WindowsHandle::AttachHandle(HANDLE h, <span class="keywordtype">bool</span> own) <a name="l00037"></a>00037 { <a name="l00038"></a>00038 <span class="keywordflow">if</span> (m_own) <a name="l00039"></a>00039 CloseHandle(); <a name="l00040"></a>00040 <a name="l00041"></a>00041 m_h = h; <a name="l00042"></a>00042 m_own = own; <a name="l00043"></a>00043 HandleChanged(); <a name="l00044"></a>00044 } <a name="l00045"></a>00045 <a name="l00046"></a>00046 HANDLE WindowsHandle::DetachHandle() <a name="l00047"></a>00047 { <a name="l00048"></a>00048 HANDLE h = m_h; <a name="l00049"></a>00049 m_h = INVALID_HANDLE_VALUE; <a name="l00050"></a>00050 HandleChanged(); <a name="l00051"></a>00051 <span class="keywordflow">return</span> h; <a name="l00052"></a>00052 } <a name="l00053"></a>00053 <a name="l00054"></a>00054 <span class="keywordtype">void</span> WindowsHandle::CloseHandle() <a name="l00055"></a>00055 { <a name="l00056"></a>00056 <span class="keywordflow">if</span> (m_h != INVALID_HANDLE_VALUE) <a name="l00057"></a>00057 { <a name="l00058"></a>00058 ::CloseHandle(m_h); <a name="l00059"></a>00059 m_h = INVALID_HANDLE_VALUE; <a name="l00060"></a>00060 HandleChanged(); <a name="l00061"></a>00061 } <a name="l00062"></a>00062 } <a name="l00063"></a>00063 <a name="l00064"></a>00064 <span class="comment">// ********************************************************</span> <a name="l00065"></a>00065 <a name="l00066"></a>00066 <span class="keywordtype">void</span> WindowsPipe::HandleError(<span class="keyword">const</span> <span class="keywordtype">char</span> *operation)<span class="keyword"> const</span> <a name="l00067"></a>00067 <span class="keyword"></span>{ <a name="l00068"></a>00068 DWORD err = GetLastError(); <a name="l00069"></a>00069 <span class="keywordflow">throw</span> Err(GetHandle(), operation, err); <a name="l00070"></a>00070 } <a name="l00071"></a>00071 <a name="l00072"></a>00072 WindowsPipe::Err::Err(HANDLE s, <span class="keyword">const</span> std::string& operation, <span class="keywordtype">int</span> error) <a name="l00073"></a>00073 : <a class="code" href="class_o_s___error.html" title="error reported by the operating system">OS_Error</a>(IO_ERROR, <span class="stringliteral">"WindowsPipe: "</span> + operation + <span class="stringliteral">" operation failed with error 0x"</span> + IntToString(error, 16), operation, error) <a name="l00074"></a>00074 , m_h(s) <a name="l00075"></a>00075 { <a name="l00076"></a>00076 } <a name="l00077"></a>00077 <a name="l00078"></a>00078 <span class="comment">// *************************************************************</span> <a name="l00079"></a>00079 <a name="l00080"></a>00080 WindowsPipeReceiver::WindowsPipeReceiver() <a name="l00081"></a>00081 : m_resultPending(false), m_eofReceived(false) <a name="l00082"></a>00082 { <a name="l00083"></a>00083 m_event.AttachHandle(CreateEvent(NULL, <span class="keyword">true</span>, <span class="keyword">false</span>, NULL), <span class="keyword">true</span>); <a name="l00084"></a>00084 CheckAndHandleError(<span class="stringliteral">"CreateEvent"</span>, m_event.HandleValid()); <a name="l00085"></a>00085 memset(&m_overlapped, 0, <span class="keyword">sizeof</span>(m_overlapped)); <a name="l00086"></a>00086 m_overlapped.hEvent = m_event; <a name="l00087"></a>00087 } <a name="l00088"></a>00088 <a name="l00089"></a><a class="code" href="class_windows_pipe_receiver.html#a97f6f63da2411395a55e416e0ea8b11b">00089</a> <span class="keywordtype">bool</span> <a class="code" href="class_windows_pipe_receiver.html#a97f6f63da2411395a55e416e0ea8b11b" title="receive data from network source, returns whether result is immediately available...">WindowsPipeReceiver::Receive</a>(byte* buf, <span class="keywordtype">size_t</span> bufLen) <a name="l00090"></a>00090 { <a name="l00091"></a>00091 assert(!m_resultPending && !m_eofReceived); <a name="l00092"></a>00092 <a name="l00093"></a>00093 HANDLE h = GetHandle(); <a name="l00094"></a>00094 <span class="comment">// don't queue too much at once, or we might use up non-paged memory</span> <a name="l00095"></a>00095 <span class="keywordflow">if</span> (ReadFile(h, buf, UnsignedMin((DWORD)128*1024, bufLen), &m_lastResult, &m_overlapped)) <a name="l00096"></a>00096 { <a name="l00097"></a>00097 <span class="keywordflow">if</span> (m_lastResult == 0) <a name="l00098"></a>00098 m_eofReceived = <span class="keyword">true</span>; <a name="l00099"></a>00099 } <a name="l00100"></a>00100 <span class="keywordflow">else</span> <a name="l00101"></a>00101 { <a name="l00102"></a>00102 <span class="keywordflow">switch</span> (GetLastError()) <a name="l00103"></a>00103 { <a name="l00104"></a>00104 <span class="keywordflow">default</span>: <a name="l00105"></a>00105 CheckAndHandleError(<span class="stringliteral">"ReadFile"</span>, <span class="keyword">false</span>); <a name="l00106"></a>00106 <span class="keywordflow">case</span> ERROR_BROKEN_PIPE: <a name="l00107"></a>00107 <span class="keywordflow">case</span> ERROR_HANDLE_EOF: <a name="l00108"></a>00108 m_lastResult = 0; <a name="l00109"></a>00109 m_eofReceived = <span class="keyword">true</span>; <a name="l00110"></a>00110 <span class="keywordflow">break</span>; <a name="l00111"></a>00111 <span class="keywordflow">case</span> ERROR_IO_PENDING: <a name="l00112"></a>00112 m_resultPending = <span class="keyword">true</span>; <a name="l00113"></a>00113 } <a name="l00114"></a>00114 } <a name="l00115"></a>00115 <span class="keywordflow">return</span> !m_resultPending; <a name="l00116"></a>00116 } <a name="l00117"></a>00117 <a name="l00118"></a><a class="code" href="class_windows_pipe_receiver.html#a44a3dbb7ef3161d84731c3c320eaa33a">00118</a> <span class="keywordtype">void</span> <a class="code" href="class_windows_pipe_receiver.html#a44a3dbb7ef3161d84731c3c320eaa33a" title="put wait objects into container">WindowsPipeReceiver::GetWaitObjects</a>(<a class="code" href="class_wait_object_container.html" title="container of wait objects">WaitObjectContainer</a> &container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>& callStack) <a name="l00119"></a>00119 { <a name="l00120"></a>00120 <span class="keywordflow">if</span> (m_resultPending) <a name="l00121"></a>00121 container.AddHandle(m_event, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"WindowsPipeReceiver::GetWaitObjects() - result pending"</span>, &callStack)); <a name="l00122"></a>00122 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (!m_eofReceived) <a name="l00123"></a>00123 container.SetNoWait(<a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"WindowsPipeReceiver::GetWaitObjects() - result ready"</span>, &callStack)); <a name="l00124"></a>00124 } <a name="l00125"></a>00125 <a name="l00126"></a>00126 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> WindowsPipeReceiver::GetReceiveResult() <a name="l00127"></a>00127 { <a name="l00128"></a>00128 <span class="keywordflow">if</span> (m_resultPending) <a name="l00129"></a>00129 { <a name="l00130"></a>00130 HANDLE h = GetHandle(); <a name="l00131"></a>00131 <span class="keywordflow">if</span> (GetOverlappedResult(h, &m_overlapped, &m_lastResult, <span class="keyword">false</span>)) <a name="l00132"></a>00132 { <a name="l00133"></a>00133 <span class="keywordflow">if</span> (m_lastResult == 0) <a name="l00134"></a>00134 m_eofReceived = <span class="keyword">true</span>; <a name="l00135"></a>00135 } <a name="l00136"></a>00136 <span class="keywordflow">else</span> <a name="l00137"></a>00137 { <a name="l00138"></a>00138 <span class="keywordflow">switch</span> (GetLastError()) <a name="l00139"></a>00139 { <a name="l00140"></a>00140 <span class="keywordflow">default</span>: <a name="l00141"></a>00141 CheckAndHandleError(<span class="stringliteral">"GetOverlappedResult"</span>, <span class="keyword">false</span>); <a name="l00142"></a>00142 <span class="keywordflow">case</span> ERROR_BROKEN_PIPE: <a name="l00143"></a>00143 <span class="keywordflow">case</span> ERROR_HANDLE_EOF: <a name="l00144"></a>00144 m_lastResult = 0; <a name="l00145"></a>00145 m_eofReceived = <span class="keyword">true</span>; <a name="l00146"></a>00146 } <a name="l00147"></a>00147 } <a name="l00148"></a>00148 m_resultPending = <span class="keyword">false</span>; <a name="l00149"></a>00149 } <a name="l00150"></a>00150 <span class="keywordflow">return</span> m_lastResult; <a name="l00151"></a>00151 } <a name="l00152"></a>00152 <a name="l00153"></a>00153 <span class="comment">// *************************************************************</span> <a name="l00154"></a>00154 <a name="l00155"></a>00155 WindowsPipeSender::WindowsPipeSender() <a name="l00156"></a>00156 : m_resultPending(false), m_lastResult(0) <a name="l00157"></a>00157 { <a name="l00158"></a>00158 m_event.AttachHandle(CreateEvent(NULL, <span class="keyword">true</span>, <span class="keyword">false</span>, NULL), <span class="keyword">true</span>); <a name="l00159"></a>00159 CheckAndHandleError(<span class="stringliteral">"CreateEvent"</span>, m_event.HandleValid()); <a name="l00160"></a>00160 memset(&m_overlapped, 0, <span class="keyword">sizeof</span>(m_overlapped)); <a name="l00161"></a>00161 m_overlapped.hEvent = m_event; <a name="l00162"></a>00162 } <a name="l00163"></a>00163 <a name="l00164"></a>00164 <span class="keywordtype">void</span> WindowsPipeSender::Send(<span class="keyword">const</span> byte* buf, <span class="keywordtype">size_t</span> bufLen) <a name="l00165"></a>00165 { <a name="l00166"></a>00166 DWORD written = 0; <a name="l00167"></a>00167 HANDLE h = GetHandle(); <a name="l00168"></a>00168 <span class="comment">// don't queue too much at once, or we might use up non-paged memory</span> <a name="l00169"></a>00169 <span class="keywordflow">if</span> (WriteFile(h, buf, UnsignedMin((DWORD)128*1024, bufLen), &written, &m_overlapped)) <a name="l00170"></a>00170 { <a name="l00171"></a>00171 m_resultPending = <span class="keyword">false</span>; <a name="l00172"></a>00172 m_lastResult = written; <a name="l00173"></a>00173 } <a name="l00174"></a>00174 <span class="keywordflow">else</span> <a name="l00175"></a>00175 { <a name="l00176"></a>00176 <span class="keywordflow">if</span> (GetLastError() != ERROR_IO_PENDING) <a name="l00177"></a>00177 CheckAndHandleError(<span class="stringliteral">"WriteFile"</span>, <span class="keyword">false</span>); <a name="l00178"></a>00178 <a name="l00179"></a>00179 m_resultPending = <span class="keyword">true</span>; <a name="l00180"></a>00180 } <a name="l00181"></a>00181 } <a name="l00182"></a>00182 <a name="l00183"></a><a class="code" href="class_windows_pipe_sender.html#a26b98ea160dc37248cb9934edb8a27ef">00183</a> <span class="keywordtype">void</span> <a class="code" href="class_windows_pipe_sender.html#a26b98ea160dc37248cb9934edb8a27ef" title="put wait objects into container">WindowsPipeSender::GetWaitObjects</a>(<a class="code" href="class_wait_object_container.html" title="container of wait objects">WaitObjectContainer</a> &container, <a class="code" href="class_call_stack.html">CallStack</a> <span class="keyword">const</span>& callStack) <a name="l00184"></a>00184 { <a name="l00185"></a>00185 <span class="keywordflow">if</span> (m_resultPending) <a name="l00186"></a>00186 container.AddHandle(m_event, <a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"WindowsPipeSender::GetWaitObjects() - result pending"</span>, &callStack)); <a name="l00187"></a>00187 <span class="keywordflow">else</span> <a name="l00188"></a>00188 container.SetNoWait(<a class="code" href="class_call_stack.html">CallStack</a>(<span class="stringliteral">"WindowsPipeSender::GetWaitObjects() - result ready"</span>, &callStack)); <a name="l00189"></a>00189 } <a name="l00190"></a>00190 <a name="l00191"></a>00191 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> WindowsPipeSender::GetSendResult() <a name="l00192"></a>00192 { <a name="l00193"></a>00193 <span class="keywordflow">if</span> (m_resultPending) <a name="l00194"></a>00194 { <a name="l00195"></a>00195 HANDLE h = GetHandle(); <a name="l00196"></a>00196 BOOL result = GetOverlappedResult(h, &m_overlapped, &m_lastResult, <span class="keyword">false</span>); <a name="l00197"></a>00197 CheckAndHandleError(<span class="stringliteral">"GetOverlappedResult"</span>, result); <a name="l00198"></a>00198 m_resultPending = <span class="keyword">false</span>; <a name="l00199"></a>00199 } <a name="l00200"></a>00200 <span class="keywordflow">return</span> m_lastResult; <a name="l00201"></a>00201 } <a name="l00202"></a>00202 <a name="l00203"></a>00203 NAMESPACE_END <a name="l00204"></a>00204 <a name="l00205"></a>00205 <span class="preprocessor">#endif</span> </pre></div></div> <hr size="1"/><address style="text-align: right;"><small>Generated on 9 Dec 2009 for Crypto++ by <a href="http://www.doxygen.org/index.html"> <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.6.1 </small></address> </body> </html>