<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html>
<head>
<!-- Generated by HsColour, http://code.haskell.org/~malcolm/hscolour/ -->
<title>src/System/IO/Streams/Network.hs</title>
<link type='text/css' rel='stylesheet' href='hscolour.css' />
</head>
<body>
<pre><a name="line-1"></a><span class='hs-comment'>{-# LANGUAGE CPP #-}</span>
<a name="line-2"></a>
<a name="line-3"></a><span class='hs-comment'>-- | Converting network 'Socket's to streams.</span>
<a name="line-4"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>IO</span><span class='hs-varop'>.</span><span class='hs-conid'>Streams</span><span class='hs-varop'>.</span><span class='hs-conid'>Network</span>
<a name="line-5"></a> <span class='hs-layout'>(</span> <span class='hs-comment'>-- * Sockets to Streams</span>
<a name="line-6"></a> <span class='hs-varid'>socketToStreams</span>
<a name="line-7"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>socketToStreamsWithBufferSize</span>
<a name="line-8"></a> <span class='hs-layout'>)</span> <span class='hs-keyword'>where</span>
<a name="line-9"></a>
<a name="line-10"></a><span class='hs-comment'>------------------------------------------------------------------------------</span>
<a name="line-11"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Exception</span> <span class='hs-layout'>(</span><span class='hs-varid'>catch</span><span class='hs-layout'>)</span>
<a name="line-12"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span><span class='hs-varop'>.</span><span class='hs-conid'>Char8</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>S</span>
<a name="line-13"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span><span class='hs-varop'>.</span><span class='hs-conid'>Internal</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>S</span>
<a name="line-14"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Foreign</span><span class='hs-varop'>.</span><span class='hs-conid'>ForeignPtr</span> <span class='hs-layout'>(</span><span class='hs-varid'>newForeignPtr</span><span class='hs-layout'>,</span> <span class='hs-varid'>withForeignPtr</span><span class='hs-layout'>)</span>
<a name="line-15"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Foreign</span><span class='hs-varop'>.</span><span class='hs-conid'>Marshal</span><span class='hs-varop'>.</span><span class='hs-conid'>Alloc</span> <span class='hs-layout'>(</span><span class='hs-varid'>finalizerFree</span><span class='hs-layout'>,</span> <span class='hs-varid'>mallocBytes</span><span class='hs-layout'>)</span>
<a name="line-16"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Network</span><span class='hs-varop'>.</span><span class='hs-conid'>Socket</span> <span class='hs-layout'>(</span><span class='hs-conid'>Socket</span><span class='hs-layout'>)</span>
<a name="line-17"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Network</span><span class='hs-varop'>.</span><span class='hs-conid'>Socket</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>N</span>
<a name="line-18"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Network</span><span class='hs-varop'>.</span><span class='hs-conid'>Socket</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>NB</span>
<a name="line-19"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Prelude</span> <span class='hs-layout'>(</span><span class='hs-conid'>IO</span><span class='hs-layout'>,</span> <span class='hs-conid'>Int</span><span class='hs-layout'>,</span> <span class='hs-conid'>Maybe</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-varid'>return</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>$!</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'><=</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>>>=</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-20"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>IO</span><span class='hs-varop'>.</span><span class='hs-conid'>Error</span> <span class='hs-layout'>(</span><span class='hs-varid'>ioError</span><span class='hs-layout'>,</span> <span class='hs-varid'>isEOFError</span><span class='hs-layout'>)</span>
<a name="line-21"></a><span class='hs-comment'>------------------------------------------------------------------------------</span>
<a name="line-22"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>IO</span><span class='hs-varop'>.</span><span class='hs-conid'>Streams</span><span class='hs-varop'>.</span><span class='hs-conid'>Internal</span> <span class='hs-layout'>(</span><span class='hs-conid'>InputStream</span><span class='hs-layout'>,</span> <span class='hs-conid'>OutputStream</span><span class='hs-layout'>)</span>
<a name="line-23"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>IO</span><span class='hs-varop'>.</span><span class='hs-conid'>Streams</span><span class='hs-varop'>.</span><span class='hs-conid'>Internal</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>Streams</span>
<a name="line-24"></a>
<a name="line-25"></a>
<a name="line-26"></a><a name="bUFSIZ"></a><span class='hs-comment'>------------------------------------------------------------------------------</span>
<a name="line-27"></a><span class='hs-definition'>bUFSIZ</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span>
<a name="line-28"></a><span class='hs-definition'>bUFSIZ</span> <span class='hs-keyglyph'>=</span> <span class='hs-num'>4096</span>
<a name="line-29"></a>
<a name="line-30"></a>
<a name="line-31"></a><a name="socketToStreams"></a><span class='hs-comment'>------------------------------------------------------------------------------</span>
<a name="line-32"></a><span class='hs-comment'>-- | Converts a 'Socket' to an 'InputStream' \/ 'OutputStream' pair. Note that,</span>
<a name="line-33"></a><span class='hs-comment'>-- as is usually the case in @io-streams@, writing a 'Nothing' to the generated</span>
<a name="line-34"></a><span class='hs-comment'>-- 'OutputStream' does not cause the underlying 'Socket' to be closed.</span>
<a name="line-35"></a><span class='hs-definition'>socketToStreams</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Socket</span>
<a name="line-36"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>InputStream</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span><span class='hs-layout'>,</span> <span class='hs-conid'>OutputStream</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span><span class='hs-layout'>)</span>
<a name="line-37"></a><span class='hs-definition'>socketToStreams</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>socketToStreamsWithBufferSize</span> <span class='hs-varid'>bUFSIZ</span>
<a name="line-38"></a>
<a name="line-39"></a>
<a name="line-40"></a><a name="socketToStreamsWithBufferSize"></a><span class='hs-comment'>------------------------------------------------------------------------------</span>
<a name="line-41"></a><span class='hs-comment'>-- | Converts a 'Socket' to an 'InputStream' \/ 'OutputStream' pair, with</span>
<a name="line-42"></a><span class='hs-comment'>-- control over the size of the receive buffers. Note that, as is usually the</span>
<a name="line-43"></a><span class='hs-comment'>-- case in @io-streams@, writing a 'Nothing' to the generated 'OutputStream'</span>
<a name="line-44"></a><span class='hs-comment'>-- does not cause the underlying 'Socket' to be closed.</span>
<a name="line-45"></a><span class='hs-definition'>socketToStreamsWithBufferSize</span>
<a name="line-46"></a> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span> <span class='hs-comment'>-- ^ how large the receive buffer should be</span>
<a name="line-47"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Socket</span> <span class='hs-comment'>-- ^ network socket</span>
<a name="line-48"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>InputStream</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span><span class='hs-layout'>,</span> <span class='hs-conid'>OutputStream</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span><span class='hs-layout'>)</span>
<a name="line-49"></a><span class='hs-definition'>socketToStreamsWithBufferSize</span> <span class='hs-varid'>bufsiz</span> <span class='hs-varid'>socket</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-50"></a> <span class='hs-varid'>is</span> <span class='hs-keyglyph'><-</span> <span class='hs-conid'>Streams</span><span class='hs-varop'>.</span><span class='hs-varid'>makeInputStream</span> <span class='hs-varid'>input</span>
<a name="line-51"></a> <span class='hs-varid'>os</span> <span class='hs-keyglyph'><-</span> <span class='hs-conid'>Streams</span><span class='hs-varop'>.</span><span class='hs-varid'>makeOutputStream</span> <span class='hs-varid'>output</span>
<a name="line-52"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$!</span> <span class='hs-layout'>(</span><span class='hs-varid'>is</span><span class='hs-layout'>,</span> <span class='hs-varid'>os</span><span class='hs-layout'>)</span>
<a name="line-53"></a>
<a name="line-54"></a> <span class='hs-keyword'>where</span>
<a name="line-55"></a><span class='hs-cpp'>#if MIN_VERSION_network(2,4,0)</span>
<a name="line-56"></a> <span class='hs-varid'>recv</span> <span class='hs-varid'>buf</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>N</span><span class='hs-varop'>.</span><span class='hs-varid'>recvBuf</span> <span class='hs-varid'>socket</span> <span class='hs-varid'>buf</span> <span class='hs-varid'>bufsiz</span> <span class='hs-varop'>`catch`</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>ioe</span> <span class='hs-keyglyph'>-></span>
<a name="line-57"></a> <span class='hs-keyword'>if</span> <span class='hs-varid'>isEOFError</span> <span class='hs-varid'>ioe</span> <span class='hs-keyword'>then</span> <span class='hs-varid'>return</span> <span class='hs-num'>0</span> <span class='hs-keyword'>else</span> <span class='hs-varid'>ioError</span> <span class='hs-varid'>ioe</span>
<a name="line-58"></a>
<a name="line-59"></a> <span class='hs-varid'>mkFp</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mallocBytes</span> <span class='hs-varid'>bufsiz</span> <span class='hs-varop'>>>=</span> <span class='hs-varid'>newForeignPtr</span> <span class='hs-varid'>finalizerFree</span>
<a name="line-60"></a>
<a name="line-61"></a> <span class='hs-varid'>input</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-62"></a> <span class='hs-varid'>fp</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>mkFp</span>
<a name="line-63"></a> <span class='hs-varid'>n</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>withForeignPtr</span> <span class='hs-varid'>fp</span> <span class='hs-varid'>recv</span>
<a name="line-64"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$!</span> <span class='hs-keyword'>if</span> <span class='hs-varid'>n</span> <span class='hs-varop'><=</span> <span class='hs-num'>0</span>
<a name="line-65"></a> <span class='hs-keyword'>then</span> <span class='hs-conid'>Nothing</span>
<a name="line-66"></a> <span class='hs-keyword'>else</span> <span class='hs-conid'>Just</span> <span class='hs-varop'>$!</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-varid'>fromForeignPtr</span> <span class='hs-varid'>fp</span> <span class='hs-num'>0</span> <span class='hs-varid'>n</span>
<a name="line-67"></a><span class='hs-cpp'>#else</span>
<a name="line-68"></a> <span class='hs-varid'>input</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-69"></a> <span class='hs-varid'>s</span> <span class='hs-keyglyph'><-</span> <span class='hs-conid'>NB</span><span class='hs-varop'>.</span><span class='hs-varid'>recv</span> <span class='hs-varid'>socket</span> <span class='hs-varid'>bufsiz</span>
<a name="line-70"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$!</span> <span class='hs-keyword'>if</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>then</span> <span class='hs-conid'>Nothing</span> <span class='hs-keyword'>else</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>s</span>
<a name="line-71"></a><span class='hs-cpp'>#endif</span>
<a name="line-72"></a>
<a name="line-73"></a> <span class='hs-varid'>output</span> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-varop'>$!</span> <span class='hs-conid'>()</span>
<a name="line-74"></a> <span class='hs-varid'>output</span> <span class='hs-layout'>(</span><span class='hs-conid'>Just</span> <span class='hs-varid'>s</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>if</span> <span class='hs-conid'>S</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>then</span> <span class='hs-varid'>return</span> <span class='hs-varop'>$!</span> <span class='hs-conid'>()</span> <span class='hs-keyword'>else</span> <span class='hs-conid'>NB</span><span class='hs-varop'>.</span><span class='hs-varid'>sendAll</span> <span class='hs-varid'>socket</span> <span class='hs-varid'>s</span>
</pre></body>
</html>
