recordingoutputstream.html

来自「网络爬虫开源代码」· HTML 代码 · 共 638 行 · 第 1/4 页

<a name="312" href="#312">312</a> <em>     */</em>
<a name="313" href="#313">313</a>     <strong>private</strong> <strong>void</strong> record(byte[] b, <strong>int</strong> off, <strong>int</strong> len) throws IOException {
<a name="314" href="#314">314</a>         <strong>if</strong>(<strong>this</strong>.shouldDigest) {
<a name="315" href="#315">315</a>             assert <strong>this</strong>.digest != <strong>null</strong>: <span class="string">"Digest is null."</span>;
<a name="316" href="#316">316</a>             <strong>this</strong>.digest.update(b, off, len);
<a name="317" href="#317">317</a>         }
<a name="318" href="#318">318</a>         tailRecord(b, off, len);
<a name="319" href="#319">319</a>     }
<a name="320" href="#320">320</a> 
<a name="321" href="#321">321</a>     <em>/**<em>*</em></em>
<a name="322" href="#322">322</a> <em>     * Record without digesting.</em>
<a name="323" href="#323">323</a> <em>     * </em>
<a name="324" href="#324">324</a> <em>     * @param b Buffer to record.</em>
<a name="325" href="#325">325</a> <em>     * @param off Offset into buffer at which to start recording.</em>
<a name="326" href="#326">326</a> <em>     * @param len Length of buffer to record.</em>
<a name="327" href="#327">327</a> <em>     *</em>
<a name="328" href="#328">328</a> <em>     * @exception IOException Failed write to backing file.</em>
<a name="329" href="#329">329</a> <em>     */</em>
<a name="330" href="#330">330</a>     <strong>private</strong> <strong>void</strong> tailRecord(byte[] b, <strong>int</strong> off, <strong>int</strong> len) throws IOException {
<a name="331" href="#331">331</a>         <strong>if</strong>(<strong>this</strong>.position >= <strong>this</strong>.buffer.length){
<a name="332" href="#332">332</a>             <em class="comment">// TODO: Its possible to call write w/o having first opened a</em>
<a name="333" href="#333">333</a>             <em class="comment">// stream.  Lets protect ourselves against this.</em>
<a name="334" href="#334">334</a>             <strong>if</strong> (<strong>this</strong>.diskStream == <strong>null</strong>) {
<a name="335" href="#335">335</a>                 <strong>throw</strong> <strong>new</strong> IOException(<span class="string">"diskstream is null"</span>);
<a name="336" href="#336">336</a>             }
<a name="337" href="#337">337</a>             <strong>this</strong>.diskStream.write(b, off, len);
<a name="338" href="#338">338</a>             <strong>this</strong>.position += len;
<a name="339" href="#339">339</a>         } <strong>else</strong> {
<a name="340" href="#340">340</a>             assert <strong>this</strong>.buffer != <strong>null</strong>: <span class="string">"Buffer is null"</span>;
<a name="341" href="#341">341</a>             <strong>int</strong> toCopy = (<strong>int</strong>)Math.min(<strong>this</strong>.buffer.length - <strong>this</strong>.position, len);
<a name="342" href="#342">342</a>             assert b != <strong>null</strong>: <span class="string">"Passed buffer is null"</span>;
<a name="343" href="#343">343</a>             System.arraycopy(b, off, <strong>this</strong>.buffer, (<strong>int</strong>)<strong>this</strong>.position, toCopy);
<a name="344" href="#344">344</a>             <strong>this</strong>.position += toCopy;
<a name="345" href="#345">345</a>             <em class="comment">// TODO verify these are +1 -1 right</em>
<a name="346" href="#346">346</a>             <strong>if</strong> (toCopy &lt; len) {
<a name="347" href="#347">347</a>                 tailRecord(b, off + toCopy, len - toCopy);
<a name="348" href="#348">348</a>             }
<a name="349" href="#349">349</a>         }
<a name="350" href="#350">350</a>     }
<a name="351" href="#351">351</a> 
<a name="352" href="#352">352</a>     <strong>public</strong> <strong>void</strong> close() throws IOException {
<a name="353" href="#353">353</a>         <strong>if</strong>(contentBeginMark&lt;0) {
<a name="354" href="#354">354</a>             <em class="comment">// if unset, consider 0 posn as content-start</em>
<a name="355" href="#355">355</a>             <em class="comment">// (so that a -1 never survives to replay step)</em>
<a name="356" href="#356">356</a>             contentBeginMark = 0;
<a name="357" href="#357">357</a>         }
<a name="358" href="#358">358</a>         <strong>if</strong> (<strong>this</strong>.out != <strong>null</strong>) {
<a name="359" href="#359">359</a>             <strong>this</strong>.out.close();
<a name="360" href="#360">360</a>             <strong>this</strong>.out = <strong>null</strong>;
<a name="361" href="#361">361</a>         }
<a name="362" href="#362">362</a>         closeRecorder();
<a name="363" href="#363">363</a>     }
<a name="364" href="#364">364</a>     
<a name="365" href="#365">365</a>     <strong>protected</strong> <strong>synchronized</strong> <strong>void</strong> closeDiskStream()
<a name="366" href="#366">366</a>     throws IOException {
<a name="367" href="#367">367</a>         <strong>if</strong> (<strong>this</strong>.diskStream != <strong>null</strong>) {
<a name="368" href="#368">368</a>             <strong>this</strong>.diskStream.close();
<a name="369" href="#369">369</a>             <strong>this</strong>.diskStream = <strong>null</strong>;
<a name="370" href="#370">370</a>         }
<a name="371" href="#371">371</a>     }
<a name="372" href="#372">372</a> 
<a name="373" href="#373">373</a>     <strong>public</strong> <strong>void</strong> closeRecorder() throws IOException {
<a name="374" href="#374">374</a>         recording = false;
<a name="375" href="#375">375</a>         closeDiskStream(); <em class="comment">// if any</em>
<a name="376" href="#376">376</a>         <em class="comment">// This setting of size is important.  Its passed to ReplayInputStream</em>
<a name="377" href="#377">377</a>         <em class="comment">// on creation.  It uses it to know EOS.</em>
<a name="378" href="#378">378</a>         <strong>if</strong> (<strong>this</strong>.size == 0) {
<a name="379" href="#379">379</a>             <strong>this</strong>.size = <strong>this</strong>.position;
<a name="380" href="#380">380</a>         }
<a name="381" href="#381">381</a>     }
<a name="382" href="#382">382</a> 
<a name="383" href="#383">383</a>     <em class="comment">/*<em class="comment"> (non-Javadoc)</em></em>
<a name="384" href="#384">384</a> <em class="comment">     * @see java.io.OutputStream#flush()</em>
<a name="385" href="#385">385</a> <em class="comment">     */</em>
<a name="386" href="#386">386</a>     <strong>public</strong> <strong>void</strong> flush() throws IOException {
<a name="387" href="#387">387</a>         <strong>if</strong> (<strong>this</strong>.out != <strong>null</strong>) {
<a name="388" href="#388">388</a>             <strong>this</strong>.out.flush();
<a name="389" href="#389">389</a>         }
<a name="390" href="#390">390</a>         <strong>if</strong> (<strong>this</strong>.diskStream != <strong>null</strong>) {
<a name="391" href="#391">391</a>             <strong>this</strong>.diskStream.flush();
<a name="392" href="#392">392</a>         }
<a name="393" href="#393">393</a>     }
<a name="394" href="#394">394</a> 
<a name="395" href="#395">395</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> getReplayInputStream() throws IOException {
<a name="396" href="#396">396</a>         <strong>return</strong> getReplayInputStream(0);
<a name="397" href="#397">397</a>     }
<a name="398" href="#398">398</a>     
<a name="399" href="#399">399</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> getReplayInputStream(<strong>long</strong> skip) throws IOException {
<a name="400" href="#400">400</a>         <em class="comment">// If this method is being called, then assumption must be that the</em>
<a name="401" href="#401">401</a>         <em class="comment">// stream is closed. If it ain't, then the stream gotten won't work</em>
<a name="402" href="#402">402</a>         <em class="comment">// -- the size will zero so any attempt at a read will get back EOF.</em>
<a name="403" href="#403">403</a>         assert <strong>this</strong>.out == <strong>null</strong>: <span class="string">"Stream is still open."</span>;
<a name="404" href="#404">404</a>         <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> replay = <strong>new</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a>(<strong>this</strong>.buffer, 
<a name="405" href="#405">405</a>                 <strong>this</strong>.size, <strong>this</strong>.contentBeginMark, <strong>this</strong>.backingFilename);
<a name="406" href="#406">406</a>         replay.skip(skip);
<a name="407" href="#407">407</a>         <strong>return</strong> replay; 
<a name="408" href="#408">408</a>     }
<a name="409" href="#409">409</a> 
<a name="410" href="#410">410</a>     <em>/**<em>*</em></em>
<a name="411" href="#411">411</a> <em>     * Return a replay stream, cued up to begining of content</em>
<a name="412" href="#412">412</a> <em>     *</em>
<a name="413" href="#413">413</a> <em>     * @throws IOException</em>
<a name="414" href="#414">414</a> <em>     * @return An RIS.</em>
<a name="415" href="#415">415</a> <em>     */</em>
<a name="416" href="#416">416</a>     <strong>public</strong> <a href="../../../org/archive/io/ReplayInputStream.html">ReplayInputStream</a> getContentReplayInputStream() throws IOException {
<a name="417" href="#417">417</a>         <strong>return</strong> getReplayInputStream(<strong>this</strong>.contentBeginMark);
<a name="418" href="#418">418</a>     }
<a name="419" href="#419">419</a> 
<a name="420" href="#420">420</a>     <strong>public</strong> <strong>long</strong> getSize() {
<a name="421" href="#421">421</a>         <strong>return</strong> <strong>this</strong>.size;
<a name="422" href="#422">422</a>     }
<a name="423" href="#423">423</a> 
<a name="424" href="#424">424</a>     <em>/**<em>*</em></em>
<a name="425" href="#425">425</a> <em>     * Remember the current position as the start of the "response</em>
<a name="426" href="#426">426</a> <em>     * body". Useful when recording HTTP traffic as a way to start</em>
<a name="427" href="#427">427</a> <em>     * replays after the headers.</em>
<a name="428" href="#428">428</a> <em>     */</em>
<a name="429" href="#429">429</a>     <strong>public</strong> <strong>void</strong> markContentBegin() {
<a name="430" href="#430">430</a>         <strong>this</strong>.contentBeginMark = <strong>this</strong>.position;
<a name="431" href="#431">431</a>         startDigest();
<a name="432" href="#432">432</a>     }
<a name="433" href="#433">433</a> 
<a name="434" href="#434">434</a>     <em>/**<em>*</em></em>
<a name="435" href="#435">435</a> <em>     * Return stored content-begin-mark (which is also end-of-headers)</em>
<a name="436" href="#436">436</a> <em>     */</em>
<a name="437" href="#437">437</a>     <strong>public</strong> <strong>long</strong> getContentBegin() {
<a name="438" href="#438">438</a>         <strong>return</strong> <strong>this</strong>.contentBeginMark;
<a name="439" href="#439">439</a>     }
<a name="440" href="#440">440</a>     
<a name="441" href="#441">441</a>     <em>/**<em>*</em></em>
<a name="442" href="#442">442</a> <em>     * Starts digesting recorded data, if a MessageDigest has been</em>
<a name="443" href="#443">443</a> <em>     * set.</em>
<a name="444" href="#444">444</a> <em>     */</em>
<a name="445" href="#445">445</a>     <strong>public</strong> <strong>void</strong> startDigest() {
<a name="446" href="#446">446</a>         <strong>if</strong> (<strong>this</strong>.digest != <strong>null</strong>) {
<a name="447" href="#447">447</a>             <strong>this</strong>.digest.reset();
<a name="448" href="#448">448</a>             <strong>this</strong>.shouldDigest = <strong>true</strong>;
<a name="449" href="#449">449</a>         }
<a name="450" href="#450">450</a>     }
<a name="451" href="#451">451</a> 
<a name="452" href="#452">452</a>     <em>/**<em>*</em></em>
<a name="453" href="#453">453</a> <em>     * Convenience method for setting SHA1 digest.</em>
<a name="454" href="#454">454</a> <em>     * @see #setDigest(String)</em>
<a name="455" href="#455">455</a> <em>     */</em>
<a name="456" href="#456">456</a>     <strong>public</strong> <strong>void</strong> setSha1Digest() {
<a name="457" href="#457">457</a>         setDigest(SHA1);
<a name="458" href="#458">458</a>     }
<a name="459" href="#459">459</a>     
<a name="460" href="#460">460</a> 
<a name="461" href="#461">461</a>     <em>/**<em>*</em></em>
<a name="462" href="#462">462</a> <em>     * Sets a digest function which may be applied to recorded data.</em>
<a name="463" href="#463">463</a> <em>     * The difference between calling this method and {@link #setDigest(MessageDigest)}</em>
<a name="464" href="#464">464</a> <em>     * is that this method tries to reuse MethodDigest instance if already allocated</em>
<a name="465" href="#465">465</a> <em>     * and of appropriate algorithm.</em>
<a name="466" href="#466">466</a> <em>     * @param algorithm Message digest algorithm to use.</em>
<a name="467" href="#467">467</a> <em>     * @see #setDigest(MessageDigest)</em>
<a name="468" href="#468">468</a> <em>     */</em>
<a name="469" href="#469">469</a>     <strong>public</strong> <strong>void</strong> setDigest(String algorithm) {
<a name="470" href="#470">470</a>         <strong>try</strong> {
<a name="471" href="#471">471</a>             <em class="comment">// Reuse extant digest if its sha1 algorithm.</em>