<?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>dist/build/Network/Socket/SendFile/Linux.hs</title>
<link type='text/css' rel='stylesheet' href='hscolour.css' />
</head>
<body>
<pre><a name="line-1"></a><span class='hs-comment'>{-# LINE 1 "src/Network/Socket/SendFile/Linux.hsc" #-}</span>
<a name="line-2"></a><span class='hs-comment'>{-# LANGUAGE ForeignFunctionInterface #-}</span>
<a name="line-3"></a><span class='hs-comment'>{-# LINE 2 "src/Network/Socket/SendFile/Linux.hsc" #-}</span>
<a name="line-4"></a><span class='hs-comment'>-- | Linux system-dependent code for 'sendfile'.</span>
<a name="line-5"></a><span class='hs-keyword'>module</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'>SendFile</span><span class='hs-varop'>.</span><span class='hs-conid'>Linux</span> <span class='hs-layout'>(</span><span class='hs-sel'>_sendFile</span><span class='hs-layout'>,</span> <span class='hs-varid'>sendFileIter</span><span class='hs-layout'>,</span> <span class='hs-varid'>sendfile</span><span class='hs-layout'>)</span> <span class='hs-keyword'>where</span>
<a name="line-6"></a>
<a name="line-7"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Int</span> <span class='hs-layout'>(</span><span class='hs-conid'>Int32</span><span class='hs-layout'>,</span> <span class='hs-conid'>Int64</span><span class='hs-layout'>)</span> <span class='hs-comment'>-- Int64 is imported on 64-bit systems</span>
<a name="line-8"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Word</span> <span class='hs-layout'>(</span><span class='hs-conid'>Word32</span><span class='hs-layout'>,</span> <span class='hs-conid'>Word64</span><span class='hs-layout'>)</span> <span class='hs-comment'>-- Word64 is imported on 64-bit systems</span>
<a name="line-9"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Foreign</span><span class='hs-varop'>.</span><span class='hs-conid'>C</span> <span class='hs-layout'>(</span><span class='hs-conid'>CInt</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-10"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Foreign</span><span class='hs-varop'>.</span><span class='hs-conid'>C</span><span class='hs-varop'>.</span><span class='hs-conid'>Error</span> <span class='hs-layout'>(</span><span class='hs-varid'>eAGAIN</span><span class='hs-layout'>,</span> <span class='hs-varid'>getErrno</span><span class='hs-layout'>,</span> <span class='hs-varid'>throwErrno</span><span class='hs-layout'>)</span>
<a name="line-11"></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-layout'>(</span><span class='hs-varid'>alloca</span><span class='hs-layout'>)</span>
<a name="line-12"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Foreign</span><span class='hs-varop'>.</span><span class='hs-conid'>Ptr</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ptr</span><span class='hs-layout'>)</span>
<a name="line-13"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Foreign</span><span class='hs-varop'>.</span><span class='hs-conid'>Storable</span><span class='hs-layout'>(</span><span class='hs-varid'>poke</span><span class='hs-layout'>)</span>
<a name="line-14"></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-varop'>.</span><span class='hs-conid'>SendFile</span><span class='hs-varop'>.</span><span class='hs-conid'>Iter</span> <span class='hs-layout'>(</span><span class='hs-conid'>Iter</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'>runIter</span><span class='hs-layout'>)</span>
<a name="line-15"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>Posix</span><span class='hs-varop'>.</span><span class='hs-conid'>Types</span> <span class='hs-layout'>(</span><span class='hs-conid'>Fd</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-16"></a>
<a name="line-17"></a><span class='hs-comment'>{-# LINE 15 "src/Network/Socket/SendFile/Linux.hsc" #-}</span>
<a name="line-18"></a>
<a name="line-19"></a><span class='hs-comment'>{-# LINE 16 "src/Network/Socket/SendFile/Linux.hsc" #-}</span>
<a name="line-20"></a>
<a name="line-21"></a><span class='hs-comment'>-- | automatically loop and send everything</span>
<a name="line-22"></a><span class='hs-sel'>_sendFile</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-conid'>()</span>
<a name="line-23"></a><span class='hs-sel'>_sendFile</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>off</span> <span class='hs-varid'>count</span> <span class='hs-keyglyph'>=</span>
<a name="line-24"></a> <span class='hs-keyword'>do</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>runIter</span> <span class='hs-layout'>(</span><span class='hs-varid'>sendFileIter</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>count</span> <span class='hs-varid'>off</span> <span class='hs-varid'>count</span><span class='hs-layout'>)</span> <span class='hs-comment'>-- set blockSize == count. ie. send it all if we can.</span>
<a name="line-25"></a> <span class='hs-varid'>return</span> <span class='hs-conid'>()</span>
<a name="line-26"></a>
<a name="line-27"></a><a name="sendFileIter"></a><span class='hs-comment'>-- | a way to send things in chunks</span>
<a name="line-28"></a><span class='hs-definition'>sendFileIter</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Fd</span> <span class='hs-comment'>-- ^ file descriptor corresponding to network socket</span>
<a name="line-29"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Fd</span> <span class='hs-comment'>-- ^ file descriptor corresponding to file</span>
<a name="line-30"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-comment'>-- ^ maximum number of bytes to send at once</span>
<a name="line-31"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-comment'>-- ^ offset into file</span>
<a name="line-32"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-comment'>-- ^ total number of bytes to send</span>
<a name="line-33"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-conid'>Iter</span>
<a name="line-34"></a><span class='hs-definition'>sendFileIter</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>blockSize</span> <span class='hs-varid'>off</span> <span class='hs-varid'>remaining</span> <span class='hs-keyglyph'>=</span>
<a name="line-35"></a><span class='hs-comment'>-- alloca $ \poff -> </span>
<a name="line-36"></a><span class='hs-comment'>-- do poke poff off</span>
<a name="line-37"></a> <span class='hs-varid'>sendFileIterI</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-layout'>(</span><span class='hs-varid'>min</span> <span class='hs-varid'>blockSize</span> <span class='hs-varid'>maxBytes</span><span class='hs-layout'>)</span> <span class='hs-varid'>off</span> <span class='hs-varid'>remaining</span>
<a name="line-38"></a>
<a name="line-39"></a><a name="sendFileIterI"></a><span class='hs-definition'>sendFileIterI</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Fd</span> <span class='hs-comment'>-- ^ file descriptor corresponding to network socket</span>
<a name="line-40"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Fd</span> <span class='hs-comment'>-- ^ file descriptor corresponding to file</span>
<a name="line-41"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-comment'>-- ^ maximum number of bytes to send at once</span>
<a name="line-42"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-comment'>-- ^ offset into file</span>
<a name="line-43"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-comment'>-- ^ total number of bytes to send</span>
<a name="line-44"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-conid'>Iter</span>
<a name="line-45"></a><span class='hs-definition'>sendFileIterI</span> <span class='hs-sel'>_out_fd</span> <span class='hs-sel'>_in_fd</span> <span class='hs-sel'>_blockSize</span> <span class='hs-sel'>_off</span> <span class='hs-num'>0</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Done</span> <span class='hs-num'>0</span><span class='hs-layout'>)</span>
<a name="line-46"></a><span class='hs-definition'>sendFileIterI</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>blockSize</span> <span class='hs-varid'>off</span> <span class='hs-varid'>remaining</span> <span class='hs-keyglyph'>=</span>
<a name="line-47"></a> <span class='hs-keyword'>do</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>bytes</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>min</span> <span class='hs-varid'>remaining</span> <span class='hs-varid'>blockSize</span>
<a name="line-48"></a> <span class='hs-layout'>(</span><span class='hs-varid'>wouldBlock</span><span class='hs-layout'>,</span> <span class='hs-varid'>sbytes</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>sendfile</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>off</span> <span class='hs-varid'>bytes</span>
<a name="line-49"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>cont</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>sendFileIterI</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>blockSize</span> <span class='hs-layout'>(</span><span class='hs-varid'>off</span> <span class='hs-varop'>+</span> <span class='hs-varid'>sbytes</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>remaining</span> <span class='hs-varop'>`safeMinus`</span> <span class='hs-varid'>sbytes</span><span class='hs-layout'>)</span>
<a name="line-50"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>wouldBlock</span> <span class='hs-keyword'>of</span>
<a name="line-51"></a> <span class='hs-conid'>True</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>WouldBlock</span> <span class='hs-varid'>sbytes</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>cont</span><span class='hs-layout'>)</span>
<a name="line-52"></a> <span class='hs-conid'>False</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>Sent</span> <span class='hs-varid'>sbytes</span> <span class='hs-varid'>cont</span><span class='hs-layout'>)</span>
<a name="line-53"></a>
<a name="line-54"></a><a name="sendfile"></a><span class='hs-comment'>-- | low-level wrapper around sendfile</span>
<a name="line-55"></a><span class='hs-comment'>-- non-blocking</span>
<a name="line-56"></a><span class='hs-comment'>-- returns number of bytes written and whether the fd would block (aka, EAGAIN)</span>
<a name="line-57"></a><span class='hs-comment'>-- does not call 'threadWaitWrite'</span>
<a name="line-58"></a><span class='hs-definition'>sendfile</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>Bool</span><span class='hs-layout'>,</span> <span class='hs-conid'>Int64</span><span class='hs-layout'>)</span>
<a name="line-59"></a><span class='hs-definition'>sendfile</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>off</span> <span class='hs-varid'>bytes</span> <span class='hs-keyglyph'>=</span>
<a name="line-60"></a> <span class='hs-varid'>alloca</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>poff</span> <span class='hs-keyglyph'>-></span>
<a name="line-61"></a> <span class='hs-keyword'>do</span> <span class='hs-varid'>poke</span> <span class='hs-varid'>poff</span> <span class='hs-varid'>off</span>
<a name="line-62"></a> <span class='hs-varid'>sendfileI</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>poff</span> <span class='hs-varid'>bytes</span>
<a name="line-63"></a>
<a name="line-64"></a><a name="sendfileI"></a><span class='hs-comment'>-- low-level wrapper around linux sendfile</span>
<a name="line-65"></a><span class='hs-definition'>sendfileI</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Ptr</span> <span class='hs-conid'>Int64</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int64</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>Bool</span><span class='hs-layout'>,</span> <span class='hs-conid'>Int64</span><span class='hs-layout'>)</span>
<a name="line-66"></a><span class='hs-definition'>sendfileI</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>poff</span> <span class='hs-varid'>bytes</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-67"></a> <span class='hs-varid'>sbytes</span> <span class='hs-keyglyph'><-</span> <span class='hs-comment'>{-# SCC "c_sendfile" #-}</span> <span class='hs-varid'>c_sendfile</span> <span class='hs-varid'>out_fd</span> <span class='hs-varid'>in_fd</span> <span class='hs-varid'>poff</span> <span class='hs-layout'>(</span><span class='hs-varid'>fromIntegral</span> <span class='hs-varid'>bytes</span><span class='hs-layout'>)</span>
<a name="line-68"></a> <span class='hs-keyword'>if</span> <span class='hs-varid'>sbytes</span> <span class='hs-varop'><=</span> <span class='hs-comment'>-</span><span class='hs-num'>1</span>
<a name="line-69"></a> <span class='hs-keyword'>then</span> <span class='hs-keyword'>do</span> <span class='hs-varid'>errno</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getErrno</span>
<a name="line-70"></a> <span class='hs-keyword'>if</span> <span class='hs-varid'>errno</span> <span class='hs-varop'>==</span> <span class='hs-varid'>eAGAIN</span>
<a name="line-71"></a> <span class='hs-keyword'>then</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>True</span><span class='hs-layout'>,</span> <span class='hs-num'>0</span><span class='hs-layout'>)</span>
<a name="line-72"></a> <span class='hs-keyword'>else</span> <span class='hs-varid'>throwErrno</span> <span class='hs-str'>"Network.Socket.SendFile.Linux.sendfileI"</span>
<a name="line-73"></a> <span class='hs-keyword'>else</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-conid'>False</span><span class='hs-layout'>,</span> <span class='hs-varid'>fromIntegral</span> <span class='hs-varid'>sbytes</span><span class='hs-layout'>)</span>
<a name="line-74"></a>
<a name="line-75"></a><a name="safeMinus"></a><span class='hs-definition'>safeMinus</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ord</span> <span class='hs-varid'>a</span><span class='hs-layout'>,</span> <span class='hs-conid'>Num</span> <span class='hs-varid'>a</span><span class='hs-layout'>,</span> <span class='hs-conid'>Show</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=></span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>a</span>
<a name="line-76"></a><span class='hs-definition'>safeMinus</span> <span class='hs-varid'>x</span> <span class='hs-varid'>y</span>
<a name="line-77"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>y</span> <span class='hs-varop'>></span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>error</span> <span class='hs-varop'>$</span> <span class='hs-str'>"y > x "</span> <span class='hs-varop'>++</span> <span class='hs-varid'>show</span> <span class='hs-layout'>(</span><span class='hs-varid'>y</span><span class='hs-layout'>,</span><span class='hs-varid'>x</span><span class='hs-layout'>)</span>
<a name="line-78"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>x</span> <span class='hs-comment'>-</span> <span class='hs-varid'>y</span>
<a name="line-79"></a>
<a name="line-80"></a>
<a name="line-81"></a><a name="maxBytes"></a><span class='hs-comment'>-- max num of bytes in one send</span>
<a name="line-82"></a><span class='hs-definition'>maxBytes</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int64</span>
<a name="line-83"></a><span class='hs-definition'>maxBytes</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fromIntegral</span> <span class='hs-layout'>(</span><span class='hs-varid'>maxBound</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Int32</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-84"></a><span class='hs-comment'>{-# LINE 80 "src/Network/Socket/SendFile/Linux.hsc" #-}</span>
<a name="line-85"></a>
<a name="line-86"></a><span class='hs-comment'>-- sendfile64 gives LFS support</span>
<a name="line-87"></a><span class='hs-keyword'>foreign</span> <span class='hs-keyword'>import</span> <span class='hs-keyword'>ccall</span> <span class='hs-keyword'>unsafe</span> <span class='hs-str'>"sendfile64"</span> <span class='hs-varid'>c_sendfile</span>
<a name="line-88"></a> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Fd</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Ptr</span> <span class='hs-layout'>(</span><span class='hs-conid'>Int64</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-conid'>Word32</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>Int32</span><span class='hs-layout'>)</span>
<a name="line-89"></a><span class='hs-comment'>{-# LINE 84 "src/Network/Socket/SendFile/Linux.hsc" #-}</span>
<a name="line-90"></a>
</pre></body>
</html>
