<!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++: files.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>files.cpp</h1><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">// files.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 <a name="l00005"></a>00005 <span class="preprocessor">#ifndef CRYPTOPP_IMPORTS</span> <a name="l00006"></a>00006 <span class="preprocessor"></span> <a name="l00007"></a>00007 <span class="preprocessor">#include "files.h"</span> <a name="l00008"></a>00008 <a name="l00009"></a>00009 <span class="preprocessor">#include <limits></span> <a name="l00010"></a>00010 <a name="l00011"></a>00011 NAMESPACE_BEGIN(CryptoPP) <a name="l00012"></a>00012 <a name="l00013"></a>00013 using namespace std; <a name="l00014"></a>00014 <a name="l00015"></a>00015 <span class="preprocessor">#ifndef NDEBUG</span> <a name="l00016"></a>00016 <span class="preprocessor"></span><span class="keywordtype">void</span> Files_TestInstantiations() <a name="l00017"></a>00017 { <a name="l00018"></a>00018 <a class="code" href="class_file_store.html" title="file-based implementation of Store interface">FileStore</a> f0; <a name="l00019"></a>00019 <a class="code" href="class_file_source.html" title="file-based implementation of Source interface">FileSource</a> f1; <a name="l00020"></a>00020 <a class="code" href="class_file_sink.html" title="file-based implementation of Sink interface">FileSink</a> f2; <a name="l00021"></a>00021 } <a name="l00022"></a>00022 <span class="preprocessor">#endif</span> <a name="l00023"></a>00023 <span class="preprocessor"></span> <a name="l00024"></a>00024 <span class="keywordtype">void</span> FileStore::StoreInitialize(<span class="keyword">const</span> <a class="code" href="class_name_value_pairs.html" title="interface for retrieving values given their names">NameValuePairs</a> &parameters) <a name="l00025"></a>00025 { <a name="l00026"></a>00026 m_waiting = <span class="keyword">false</span>; <a name="l00027"></a>00027 m_stream = NULL; <a name="l00028"></a>00028 m_file.release(); <a name="l00029"></a>00029 <a name="l00030"></a>00030 <span class="keyword">const</span> <span class="keywordtype">char</span> *fileName = NULL; <a name="l00031"></a>00031 <span class="preprocessor">#if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400</span> <a name="l00032"></a>00032 <span class="preprocessor"></span> <span class="keyword">const</span> <span class="keywordtype">wchar_t</span> *fileNameWide = NULL; <a name="l00033"></a>00033 <span class="keywordflow">if</span> (!parameters.<a class="code" href="class_name_value_pairs.html#a96686e9f8d6ce3ab870e516fb72b608e" title="get a named value, returns true if the name exists">GetValue</a>(Name::InputFileNameWide(), fileNameWide)) <a name="l00034"></a>00034 #endif <a name="l00035"></a>00035 <span class="keywordflow">if</span> (!parameters.<a class="code" href="class_name_value_pairs.html#a96686e9f8d6ce3ab870e516fb72b608e" title="get a named value, returns true if the name exists">GetValue</a>(Name::InputFileName(), fileName)) <a name="l00036"></a>00036 { <a name="l00037"></a>00037 parameters.<a class="code" href="class_name_value_pairs.html#a96686e9f8d6ce3ab870e516fb72b608e" title="get a named value, returns true if the name exists">GetValue</a>(Name::InputStreamPointer(), m_stream); <a name="l00038"></a>00038 <span class="keywordflow">return</span>; <a name="l00039"></a>00039 } <a name="l00040"></a>00040 <a name="l00041"></a>00041 ios::openmode binary = parameters.<a class="code" href="class_name_value_pairs.html#a943b2009297783f1c35bae46efc3b5f7" title="get a named value, returns the default if the name doesn&#39;t exist">GetValueWithDefault</a>(Name::InputBinaryMode(), <span class="keyword">true</span>) ? ios::binary : ios::openmode(0); <a name="l00042"></a>00042 m_file.reset(<span class="keyword">new</span> std::ifstream); <a name="l00043"></a>00043 <span class="preprocessor">#ifdef CRYPTOPP_UNIX_AVAILABLE</span> <a name="l00044"></a>00044 <span class="preprocessor"></span> std::string narrowed; <a name="l00045"></a>00045 <span class="keywordflow">if</span> (fileNameWide) <a name="l00046"></a>00046 fileName = (narrowed = StringNarrow(fileNameWide)).c_str(); <a name="l00047"></a>00047 <span class="preprocessor">#endif</span> <a name="l00048"></a>00048 <span class="preprocessor"></span><span class="preprocessor">#if _MSC_VER >= 1400</span> <a name="l00049"></a>00049 <span class="preprocessor"></span> <span class="keywordflow">if</span> (fileNameWide) <a name="l00050"></a>00050 { <a name="l00051"></a>00051 m_file->open(fileNameWide, ios::in | binary); <a name="l00052"></a>00052 <span class="keywordflow">if</span> (!*m_file) <a name="l00053"></a>00053 <span class="keywordflow">throw</span> OpenErr(StringNarrow(fileNameWide, <span class="keyword">false</span>)); <a name="l00054"></a>00054 } <a name="l00055"></a>00055 <span class="preprocessor">#endif</span> <a name="l00056"></a>00056 <span class="preprocessor"></span> <span class="keywordflow">if</span> (fileName) <a name="l00057"></a>00057 { <a name="l00058"></a>00058 m_file->open(fileName, ios::in | binary); <a name="l00059"></a>00059 <span class="keywordflow">if</span> (!*m_file) <a name="l00060"></a>00060 <span class="keywordflow">throw</span> OpenErr(fileName); <a name="l00061"></a>00061 } <a name="l00062"></a>00062 m_stream = m_file.get(); <a name="l00063"></a>00063 } <a name="l00064"></a>00064 <a name="l00065"></a><a class="code" href="class_file_store.html#ac003753c1e1a312cdaa6149e5f53bfb9">00065</a> lword <a class="code" href="class_file_store.html#ac003753c1e1a312cdaa6149e5f53bfb9" title="returns number of bytes that is currently ready for retrieval">FileStore::MaxRetrievable</a>()<span class="keyword"> const</span> <a name="l00066"></a>00066 <span class="keyword"></span>{ <a name="l00067"></a>00067 <span class="keywordflow">if</span> (!m_stream) <a name="l00068"></a>00068 <span class="keywordflow">return</span> 0; <a name="l00069"></a>00069 <a name="l00070"></a>00070 streampos current = m_stream->tellg(); <a name="l00071"></a>00071 streampos end = m_stream->seekg(0, ios::end).tellg(); <a name="l00072"></a>00072 m_stream->seekg(current); <a name="l00073"></a>00073 <span class="keywordflow">return</span> end-current; <a name="l00074"></a>00074 } <a name="l00075"></a>00075 <a name="l00076"></a>00076 <span class="keywordtype">size_t</span> FileStore::TransferTo2(<a class="code" href="class_buffered_transformation.html" title="interface for buffered transformations">BufferedTransformation</a> &target, lword &transferBytes, <span class="keyword">const</span> std::string &channel, <span class="keywordtype">bool</span> blocking) <a name="l00077"></a>00077 { <a name="l00078"></a>00078 <span class="keywordflow">if</span> (!m_stream) <a name="l00079"></a>00079 { <a name="l00080"></a>00080 transferBytes = 0; <a name="l00081"></a>00081 <span class="keywordflow">return</span> 0; <a name="l00082"></a>00082 } <a name="l00083"></a>00083 <a name="l00084"></a>00084 lword size=transferBytes; <a name="l00085"></a>00085 transferBytes = 0; <a name="l00086"></a>00086 <a name="l00087"></a>00087 <span class="keywordflow">if</span> (m_waiting) <a name="l00088"></a>00088 <span class="keywordflow">goto</span> output; <a name="l00089"></a>00089 <a name="l00090"></a>00090 <span class="keywordflow">while</span> (size && m_stream->good()) <a name="l00091"></a>00091 { <a name="l00092"></a>00092 { <a name="l00093"></a>00093 <span class="keywordtype">size_t</span> spaceSize = 1024; <a name="l00094"></a>00094 m_space = HelpCreatePutSpace(target, channel, 1, UnsignedMin(<span class="keywordtype">size_t</span>(0)-1, size), spaceSize); <a name="l00095"></a>00095 <a name="l00096"></a>00096 m_stream->read((<span class="keywordtype">char</span> *)m_space, (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>)STDMIN(size, (lword)spaceSize)); <a name="l00097"></a>00097 } <a name="l00098"></a>00098 m_len = m_stream->gcount(); <a name="l00099"></a>00099 <span class="keywordtype">size_t</span> blockedBytes; <a name="l00100"></a>00100 output: <a name="l00101"></a>00101 blockedBytes = target.ChannelPutModifiable2(channel, m_space, m_len, 0, blocking); <a name="l00102"></a>00102 m_waiting = blockedBytes > 0; <a name="l00103"></a>00103 <span class="keywordflow">if</span> (m_waiting) <a name="l00104"></a>00104 <span class="keywordflow">return</span> blockedBytes; <a name="l00105"></a>00105 size -= m_len; <a name="l00106"></a>00106 transferBytes += m_len; <a name="l00107"></a>00107 } <a name="l00108"></a>00108 <a name="l00109"></a>00109 <span class="keywordflow">if</span> (!m_stream->good() && !m_stream->eof()) <a name="l00110"></a>00110 <span class="keywordflow">throw</span> ReadErr(); <a name="l00111"></a>00111 <a name="l00112"></a>00112 <span class="keywordflow">return</span> 0; <a name="l00113"></a>00113 } <a name="l00114"></a>00114 <a name="l00115"></a>00115 <span class="keywordtype">size_t</span> FileStore::CopyRangeTo2(<a class="code" href="class_buffered_transformation.html" title="interface for buffered transformations">BufferedTransformation</a> &target, lword &begin, lword end, <span class="keyword">const</span> std::string &channel, <span class="keywordtype">bool</span> blocking)<span class="keyword"> const</span> <a name="l00116"></a>00116 <span class="keyword"></span>{ <a name="l00117"></a>00117 <span class="keywordflow">if</span> (!m_stream) <a name="l00118"></a>00118 <span class="keywordflow">return</span> 0; <a name="l00119"></a>00119 <a name="l00120"></a>00120 <span class="keywordflow">if</span> (begin == 0 && end == 1) <a name="l00121"></a>00121 { <a name="l00122"></a>00122 <span class="keywordtype">int</span> result = m_stream->peek(); <a name="l00123"></a>00123 <span class="keywordflow">if</span> (result == char_traits<char>::eof()) <a name="l00124"></a>00124 <span class="keywordflow">return</span> 0; <a name="l00125"></a>00125 <span class="keywordflow">else</span> <a name="l00126"></a>00126 { <a name="l00127"></a>00127 <span class="keywordtype">size_t</span> blockedBytes = target.ChannelPut(channel, byte(result), blocking); <a name="l00128"></a>00128 begin += 1-blockedBytes; <a name="l00129"></a>00129 <span class="keywordflow">return</span> blockedBytes; <a name="l00130"></a>00130 } <a name="l00131"></a>00131 } <a name="l00132"></a>00132 <a name="l00133"></a>00133 <span class="comment">// TODO: figure out what happens on cin</span> <a name="l00134"></a>00134 streampos current = m_stream->tellg(); <a name="l00135"></a>00135 streampos endPosition = m_stream->seekg(0, ios::end).tellg(); <a name="l00136"></a>00136 streampos newPosition = current + (streamoff)begin; <a name="l00137"></a>00137 <a name="l00138"></a>00138 <span class="keywordflow">if</span> (newPosition >= endPosition) <a name="l00139"></a>00139 { <a name="l00140"></a>00140 m_stream->seekg(current); <a name="l00141"></a>00141 <span class="keywordflow">return</span> 0; <span class="comment">// don't try to seek beyond the end of file</span> <a name="l00142"></a>00142 } <a name="l00143"></a>00143 m_stream->seekg(newPosition); <a name="l00144"></a>00144 <span class="keywordflow">try</span> <a name="l00145"></a>00145 { <a name="l00146"></a>00146 assert(!m_waiting); <a name="l00147"></a>00147 lword copyMax = end-begin; <a name="l00148"></a>00148 <span class="keywordtype">size_t</span> blockedBytes = <span class="keyword">const_cast<</span><a class="code" href="class_file_store.html" title="file-based implementation of Store interface">FileStore</a> *<span class="keyword">></span>(<span class="keyword">this</span>)->TransferTo2(target, copyMax, channel, blocking); <a name="l00149"></a>00149 begin += copyMax; <a name="l00150"></a>00150 <span class="keywordflow">if</span> (blockedBytes) <a name="l00151"></a>00151 { <a name="l00152"></a>00152 <span class="keyword">const_cast<</span><a class="code" href="class_file_store.html" title="file-based implementation of Store interface">FileStore</a> *<span class="keyword">></span>(<span class="keyword">this</span>)->m_waiting = <span class="keyword">false</span>; <a name="l00153"></a>00153 <span class="keywordflow">return</span> blockedBytes; <a name="l00154"></a>00154 } <a name="l00155"></a>00155 } <a name="l00156"></a>00156 <span class="keywordflow">catch</span>(...) <a name="l00157"></a>00157 { <a name="l00158"></a>00158 m_stream->clear(); <a name="l00159"></a>00159 m_stream->seekg(current); <a name="l00160"></a>00160 <span class="keywordflow">throw</span>; <a name="l00161"></a>00161 } <a name="l00162"></a>00162 m_stream->clear(); <a name="l00163"></a>00163 m_stream->seekg(current); <a name="l00164"></a>00164 <a name="l00165"></a>00165 <span class="keywordflow">return</span> 0; <a name="l00166"></a>00166 } <a name="l00167"></a>00167 <a name="l00168"></a><a class="code" href="class_file_store.html#a224d31ea858a3c890fec604a030db7cf">00168</a> lword <a class="code" href="class_file_store.html#a224d31ea858a3c890fec604a030db7cf" title="discard skipMax bytes from the output buffer">FileStore::Skip</a>(lword skipMax) <a name="l00169"></a>00169 { <a name="l00170"></a>00170 <span class="keywordflow">if</span> (!m_stream) <a name="l00171"></a>00171 <span class="keywordflow">return</span> 0; <a name="l00172"></a>00172 <a name="l00173"></a>00173 lword oldPos = m_stream->tellg(); <a name="l00174"></a>00174 std::istream::off_type offset; <a name="l00175"></a>00175 <span class="keywordflow">if</span> (!SafeConvert(skipMax, offset)) <a name="l00176"></a>00176 <span class="keywordflow">throw</span> <a class="code" href="class_invalid_argument.html" title="exception thrown when an invalid argument is detected">InvalidArgument</a>(<span class="stringliteral">"FileStore: maximum seek offset exceeded"</span>); <a name="l00177"></a>00177 m_stream->seekg(offset, ios::cur); <a name="l00178"></a>00178 <span class="keywordflow">return</span> (lword)m_stream->tellg() - oldPos; <a name="l00179"></a>00179 } <a name="l00180"></a>00180 <a name="l00181"></a>00181 <span class="keywordtype">void</span> FileSink::IsolatedInitialize(<span class="keyword">const</span> <a class="code" href="class_name_value_pairs.html" title="interface for retrieving values given their names">NameValuePairs</a> &parameters) <a name="l00182"></a>00182 { <a name="l00183"></a>00183 m_stream = NULL; <a name="l00184"></a>00184 m_file.release(); <a name="l00185"></a>00185 <a name="l00186"></a>00186 <span class="keyword">const</span> <span class="keywordtype">char</span> *fileName = NULL; <a name="l00187"></a>00187 <span class="preprocessor">#if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400</span> <a name="l00188"></a>00188 <span class="preprocessor"></span> <span class="keyword">const</span> <span class="keywordtype">wchar_t</span> *fileNameWide = NULL; <a name="l00189"></a>00189 <span class="keywordflow">if</span> (!parameters.<a class="code" href="class_name_value_pairs.html#a96686e9f8d6ce3ab870e516fb72b608e" title="get a named value, returns true if the name exists">GetValue</a>(Name::OutputFileNameWide(), fileNameWide)) <a name="l00190"></a>00190 #endif <a name="l00191"></a>00191 <span class="keywordflow">if</span> (!parameters.<a class="code" href="class_name_value_pairs.html#a96686e9f8d6ce3ab870e516fb72b608e" title="get a named value, returns true if the name exists">GetValue</a>(Name::OutputFileName(), fileName)) <a name="l00192"></a>00192 { <a name="l00193"></a>00193 parameters.<a class="code" href="class_name_value_pairs.html#a96686e9f8d6ce3ab870e516fb72b608e" title="get a named value, returns true if the name exists">GetValue</a>(Name::OutputStreamPointer(), m_stream); <a name="l00194"></a>00194 <span class="keywordflow">return</span>; <a name="l00195"></a>00195 } <a name="l00196"></a>00196 <a name="l00197"></a>00197 ios::openmode binary = parameters.<a class="code" href="class_name_value_pairs.html#a943b2009297783f1c35bae46efc3b5f7" title="get a named value, returns the default if the name doesn&#39;t exist">GetValueWithDefault</a>(Name::OutputBinaryMode(), <span class="keyword">true</span>) ? ios::binary : ios::openmode(0); <a name="l00198"></a>00198 m_file.reset(<span class="keyword">new</span> std::ofstream); <a name="l00199"></a>00199 <span class="preprocessor">#ifdef CRYPTOPP_UNIX_AVAILABLE</span> <a name="l00200"></a>00200 <span class="preprocessor"></span> std::string narrowed; <a name="l00201"></a>00201 <span class="keywordflow">if</span> (fileNameWide) <a name="l00202"></a>00202 fileName = (narrowed = StringNarrow(fileNameWide)).c_str(); <a name="l00203"></a>00203 <span class="preprocessor">#endif</span> <a name="l00204"></a>00204 <span class="preprocessor"></span><span class="preprocessor">#if _MSC_VER >= 1400</span> <a name="l00205"></a>00205 <span class="preprocessor"></span> <span class="keywordflow">if</span> (fileNameWide) <a name="l00206"></a>00206 { <a name="l00207"></a>00207 m_file->open(fileNameWide, ios::out | ios::trunc | binary); <a name="l00208"></a>00208 <span class="keywordflow">if</span> (!*m_file) <a name="l00209"></a>00209 <span class="keywordflow">throw</span> OpenErr(StringNarrow(fileNameWide, <span class="keyword">false</span>)); <a name="l00210"></a>00210 } <a name="l00211"></a>00211 <span class="preprocessor">#endif</span> <a name="l00212"></a>00212 <span class="preprocessor"></span> <span class="keywordflow">if</span> (fileName) <a name="l00213"></a>00213 { <a name="l00214"></a>00214 m_file->open(fileName, ios::out | ios::trunc | binary); <a name="l00215"></a>00215 <span class="keywordflow">if</span> (!*m_file) <a name="l00216"></a>00216 <span class="keywordflow">throw</span> OpenErr(fileName); <a name="l00217"></a>00217 } <a name="l00218"></a>00218 m_stream = m_file.get(); <a name="l00219"></a>00219 } <a name="l00220"></a>00220 <a name="l00221"></a>00221 <span class="keywordtype">bool</span> FileSink::IsolatedFlush(<span class="keywordtype">bool</span> hardFlush, <span class="keywordtype">bool</span> blocking) <a name="l00222"></a>00222 { <a name="l00223"></a>00223 <span class="keywordflow">if</span> (!m_stream) <a name="l00224"></a>00224 <span class="keywordflow">throw</span> Err(<span class="stringliteral">"FileSink: output stream not opened"</span>); <a name="l00225"></a>00225 <a name="l00226"></a>00226 m_stream->flush(); <a name="l00227"></a>00227 <span class="keywordflow">if</span> (!m_stream->good()) <a name="l00228"></a>00228 <span class="keywordflow">throw</span> WriteErr(); <a name="l00229"></a>00229 <a name="l00230"></a>00230 <span class="keywordflow">return</span> <span class="keyword">false</span>; <a name="l00231"></a>00231 } <a name="l00232"></a>00232 <a name="l00233"></a><a class="code" href="class_file_sink.html#a7d9f9457b3a9d7db252e8ba75487020a">00233</a> <span class="keywordtype">size_t</span> <a class="code" href="class_file_sink.html#a7d9f9457b3a9d7db252e8ba75487020a" title="input multiple bytes for blocking or non-blocking processing">FileSink::Put2</a>(<span class="keyword">const</span> byte *inString, <span class="keywordtype">size_t</span> length, <span class="keywordtype">int</span> messageEnd, <span class="keywordtype">bool</span> blocking) <a name="l00234"></a>00234 { <a name="l00235"></a>00235 <span class="keywordflow">if</span> (!m_stream) <a name="l00236"></a>00236 <span class="keywordflow">throw</span> <a class="code" href="class_file_sink_1_1_err.html">Err</a>(<span class="stringliteral">"FileSink: output stream not opened"</span>); <a name="l00237"></a>00237 <a name="l00238"></a>00238 <span class="keywordflow">while</span> (length > 0) <a name="l00239"></a>00239 { <a name="l00240"></a>00240 std::streamsize size; <a name="l00241"></a>00241 <span class="keywordflow">if</span> (!SafeConvert(length, size)) <a name="l00242"></a>00242 size = numeric_limits<std::streamsize>::max(); <a name="l00243"></a>00243 m_stream->write((<span class="keyword">const</span> <span class="keywordtype">char</span> *)inString, size); <a name="l00244"></a>00244 inString += size; <a name="l00245"></a>00245 length -= size; <a name="l00246"></a>00246 } <a name="l00247"></a>00247 <a name="l00248"></a>00248 <span class="keywordflow">if</span> (messageEnd) <a name="l00249"></a>00249 m_stream->flush(); <a name="l00250"></a>00250 <a name="l00251"></a>00251 <span class="keywordflow">if</span> (!m_stream->good()) <a name="l00252"></a>00252 <span class="keywordflow">throw</span> <a class="code" href="class_file_sink_1_1_write_err.html">WriteErr</a>(); <a name="l00253"></a>00253 <a name="l00254"></a>00254 <span class="keywordflow">return</span> 0; <a name="l00255"></a>00255 } <a name="l00256"></a>00256 <a name="l00257"></a>00257 NAMESPACE_END <a name="l00258"></a>00258 <a name="l00259"></a>00259 <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>