| <!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `/root/.cargo/registry/src/github.com-1ecc6299db9ec823/h2-0.3.19/src/proto/connection.rs`."><meta name="keywords" content="rust, rustlang, rust-lang"><title>connection.rs - source</title><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../SourceSerif4-Regular.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../FiraSans-Regular.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../FiraSans-Medium.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../SourceCodePro-Regular.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../SourceSerif4-Bold.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../SourceCodePro-Semibold.ttf.woff2"><link rel="stylesheet" href="../../../normalize.css"><link rel="stylesheet" href="../../../rustdoc.css" id="mainThemeStyle"><link rel="stylesheet" href="../../../ayu.css" disabled><link rel="stylesheet" href="../../../dark.css" disabled><link rel="stylesheet" href="../../../light.css" id="themeStyle"><script id="default-settings" ></script><script src="../../../storage.js"></script><script defer src="../../../source-script.js"></script><script defer src="../../../source-files.js"></script><script defer src="../../../main.js"></script><noscript><link rel="stylesheet" href="../../../noscript.css"></noscript><link rel="alternate icon" type="image/png" href="../../../favicon-16x16.png"><link rel="alternate icon" type="image/png" href="../../../favicon-32x32.png"><link rel="icon" type="image/svg+xml" href="../../../favicon.svg"></head><body class="rustdoc source"><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><a class="sidebar-logo" href="../../../h2/index.html"><div class="logo-container"><img class="rust-logo" src="../../../rust-logo.svg" alt="logo"></div></a></nav><main><div class="width-limiter"><nav class="sub"><a class="sub-logo-container" href="../../../h2/index.html"><img class="rust-logo" src="../../../rust-logo.svg" alt="logo"></a><form class="search-form"><div class="search-container"><span></span><input class="search-input" name="search" autocomplete="off" spellcheck="false" placeholder="Click or press ‘S’ to search, ‘?’ for more options…" type="search"><div id="help-button" title="help" tabindex="-1"><a href="../../../help.html">?</a></div><div id="settings-menu" tabindex="-1"><a href="../../../settings.html" title="settings"><img width="22" height="22" alt="Change settings" src="../../../wheel.svg"></a></div></div></form></nav><section id="main-content" class="content"><div class="example-wrap"><pre class="src-line-numbers"><span id="1">1</span> |
| <span id="2">2</span> |
| <span id="3">3</span> |
| <span id="4">4</span> |
| <span id="5">5</span> |
| <span id="6">6</span> |
| <span id="7">7</span> |
| <span id="8">8</span> |
| <span id="9">9</span> |
| <span id="10">10</span> |
| <span id="11">11</span> |
| <span id="12">12</span> |
| <span id="13">13</span> |
| <span id="14">14</span> |
| <span id="15">15</span> |
| <span id="16">16</span> |
| <span id="17">17</span> |
| <span id="18">18</span> |
| <span id="19">19</span> |
| <span id="20">20</span> |
| <span id="21">21</span> |
| <span id="22">22</span> |
| <span id="23">23</span> |
| <span id="24">24</span> |
| <span id="25">25</span> |
| <span id="26">26</span> |
| <span id="27">27</span> |
| <span id="28">28</span> |
| <span id="29">29</span> |
| <span id="30">30</span> |
| <span id="31">31</span> |
| <span id="32">32</span> |
| <span id="33">33</span> |
| <span id="34">34</span> |
| <span id="35">35</span> |
| <span id="36">36</span> |
| <span id="37">37</span> |
| <span id="38">38</span> |
| <span id="39">39</span> |
| <span id="40">40</span> |
| <span id="41">41</span> |
| <span id="42">42</span> |
| <span id="43">43</span> |
| <span id="44">44</span> |
| <span id="45">45</span> |
| <span id="46">46</span> |
| <span id="47">47</span> |
| <span id="48">48</span> |
| <span id="49">49</span> |
| <span id="50">50</span> |
| <span id="51">51</span> |
| <span id="52">52</span> |
| <span id="53">53</span> |
| <span id="54">54</span> |
| <span id="55">55</span> |
| <span id="56">56</span> |
| <span id="57">57</span> |
| <span id="58">58</span> |
| <span id="59">59</span> |
| <span id="60">60</span> |
| <span id="61">61</span> |
| <span id="62">62</span> |
| <span id="63">63</span> |
| <span id="64">64</span> |
| <span id="65">65</span> |
| <span id="66">66</span> |
| <span id="67">67</span> |
| <span id="68">68</span> |
| <span id="69">69</span> |
| <span id="70">70</span> |
| <span id="71">71</span> |
| <span id="72">72</span> |
| <span id="73">73</span> |
| <span id="74">74</span> |
| <span id="75">75</span> |
| <span id="76">76</span> |
| <span id="77">77</span> |
| <span id="78">78</span> |
| <span id="79">79</span> |
| <span id="80">80</span> |
| <span id="81">81</span> |
| <span id="82">82</span> |
| <span id="83">83</span> |
| <span id="84">84</span> |
| <span id="85">85</span> |
| <span id="86">86</span> |
| <span id="87">87</span> |
| <span id="88">88</span> |
| <span id="89">89</span> |
| <span id="90">90</span> |
| <span id="91">91</span> |
| <span id="92">92</span> |
| <span id="93">93</span> |
| <span id="94">94</span> |
| <span id="95">95</span> |
| <span id="96">96</span> |
| <span id="97">97</span> |
| <span id="98">98</span> |
| <span id="99">99</span> |
| <span id="100">100</span> |
| <span id="101">101</span> |
| <span id="102">102</span> |
| <span id="103">103</span> |
| <span id="104">104</span> |
| <span id="105">105</span> |
| <span id="106">106</span> |
| <span id="107">107</span> |
| <span id="108">108</span> |
| <span id="109">109</span> |
| <span id="110">110</span> |
| <span id="111">111</span> |
| <span id="112">112</span> |
| <span id="113">113</span> |
| <span id="114">114</span> |
| <span id="115">115</span> |
| <span id="116">116</span> |
| <span id="117">117</span> |
| <span id="118">118</span> |
| <span id="119">119</span> |
| <span id="120">120</span> |
| <span id="121">121</span> |
| <span id="122">122</span> |
| <span id="123">123</span> |
| <span id="124">124</span> |
| <span id="125">125</span> |
| <span id="126">126</span> |
| <span id="127">127</span> |
| <span id="128">128</span> |
| <span id="129">129</span> |
| <span id="130">130</span> |
| <span id="131">131</span> |
| <span id="132">132</span> |
| <span id="133">133</span> |
| <span id="134">134</span> |
| <span id="135">135</span> |
| <span id="136">136</span> |
| <span id="137">137</span> |
| <span id="138">138</span> |
| <span id="139">139</span> |
| <span id="140">140</span> |
| <span id="141">141</span> |
| <span id="142">142</span> |
| <span id="143">143</span> |
| <span id="144">144</span> |
| <span id="145">145</span> |
| <span id="146">146</span> |
| <span id="147">147</span> |
| <span id="148">148</span> |
| <span id="149">149</span> |
| <span id="150">150</span> |
| <span id="151">151</span> |
| <span id="152">152</span> |
| <span id="153">153</span> |
| <span id="154">154</span> |
| <span id="155">155</span> |
| <span id="156">156</span> |
| <span id="157">157</span> |
| <span id="158">158</span> |
| <span id="159">159</span> |
| <span id="160">160</span> |
| <span id="161">161</span> |
| <span id="162">162</span> |
| <span id="163">163</span> |
| <span id="164">164</span> |
| <span id="165">165</span> |
| <span id="166">166</span> |
| <span id="167">167</span> |
| <span id="168">168</span> |
| <span id="169">169</span> |
| <span id="170">170</span> |
| <span id="171">171</span> |
| <span id="172">172</span> |
| <span id="173">173</span> |
| <span id="174">174</span> |
| <span id="175">175</span> |
| <span id="176">176</span> |
| <span id="177">177</span> |
| <span id="178">178</span> |
| <span id="179">179</span> |
| <span id="180">180</span> |
| <span id="181">181</span> |
| <span id="182">182</span> |
| <span id="183">183</span> |
| <span id="184">184</span> |
| <span id="185">185</span> |
| <span id="186">186</span> |
| <span id="187">187</span> |
| <span id="188">188</span> |
| <span id="189">189</span> |
| <span id="190">190</span> |
| <span id="191">191</span> |
| <span id="192">192</span> |
| <span id="193">193</span> |
| <span id="194">194</span> |
| <span id="195">195</span> |
| <span id="196">196</span> |
| <span id="197">197</span> |
| <span id="198">198</span> |
| <span id="199">199</span> |
| <span id="200">200</span> |
| <span id="201">201</span> |
| <span id="202">202</span> |
| <span id="203">203</span> |
| <span id="204">204</span> |
| <span id="205">205</span> |
| <span id="206">206</span> |
| <span id="207">207</span> |
| <span id="208">208</span> |
| <span id="209">209</span> |
| <span id="210">210</span> |
| <span id="211">211</span> |
| <span id="212">212</span> |
| <span id="213">213</span> |
| <span id="214">214</span> |
| <span id="215">215</span> |
| <span id="216">216</span> |
| <span id="217">217</span> |
| <span id="218">218</span> |
| <span id="219">219</span> |
| <span id="220">220</span> |
| <span id="221">221</span> |
| <span id="222">222</span> |
| <span id="223">223</span> |
| <span id="224">224</span> |
| <span id="225">225</span> |
| <span id="226">226</span> |
| <span id="227">227</span> |
| <span id="228">228</span> |
| <span id="229">229</span> |
| <span id="230">230</span> |
| <span id="231">231</span> |
| <span id="232">232</span> |
| <span id="233">233</span> |
| <span id="234">234</span> |
| <span id="235">235</span> |
| <span id="236">236</span> |
| <span id="237">237</span> |
| <span id="238">238</span> |
| <span id="239">239</span> |
| <span id="240">240</span> |
| <span id="241">241</span> |
| <span id="242">242</span> |
| <span id="243">243</span> |
| <span id="244">244</span> |
| <span id="245">245</span> |
| <span id="246">246</span> |
| <span id="247">247</span> |
| <span id="248">248</span> |
| <span id="249">249</span> |
| <span id="250">250</span> |
| <span id="251">251</span> |
| <span id="252">252</span> |
| <span id="253">253</span> |
| <span id="254">254</span> |
| <span id="255">255</span> |
| <span id="256">256</span> |
| <span id="257">257</span> |
| <span id="258">258</span> |
| <span id="259">259</span> |
| <span id="260">260</span> |
| <span id="261">261</span> |
| <span id="262">262</span> |
| <span id="263">263</span> |
| <span id="264">264</span> |
| <span id="265">265</span> |
| <span id="266">266</span> |
| <span id="267">267</span> |
| <span id="268">268</span> |
| <span id="269">269</span> |
| <span id="270">270</span> |
| <span id="271">271</span> |
| <span id="272">272</span> |
| <span id="273">273</span> |
| <span id="274">274</span> |
| <span id="275">275</span> |
| <span id="276">276</span> |
| <span id="277">277</span> |
| <span id="278">278</span> |
| <span id="279">279</span> |
| <span id="280">280</span> |
| <span id="281">281</span> |
| <span id="282">282</span> |
| <span id="283">283</span> |
| <span id="284">284</span> |
| <span id="285">285</span> |
| <span id="286">286</span> |
| <span id="287">287</span> |
| <span id="288">288</span> |
| <span id="289">289</span> |
| <span id="290">290</span> |
| <span id="291">291</span> |
| <span id="292">292</span> |
| <span id="293">293</span> |
| <span id="294">294</span> |
| <span id="295">295</span> |
| <span id="296">296</span> |
| <span id="297">297</span> |
| <span id="298">298</span> |
| <span id="299">299</span> |
| <span id="300">300</span> |
| <span id="301">301</span> |
| <span id="302">302</span> |
| <span id="303">303</span> |
| <span id="304">304</span> |
| <span id="305">305</span> |
| <span id="306">306</span> |
| <span id="307">307</span> |
| <span id="308">308</span> |
| <span id="309">309</span> |
| <span id="310">310</span> |
| <span id="311">311</span> |
| <span id="312">312</span> |
| <span id="313">313</span> |
| <span id="314">314</span> |
| <span id="315">315</span> |
| <span id="316">316</span> |
| <span id="317">317</span> |
| <span id="318">318</span> |
| <span id="319">319</span> |
| <span id="320">320</span> |
| <span id="321">321</span> |
| <span id="322">322</span> |
| <span id="323">323</span> |
| <span id="324">324</span> |
| <span id="325">325</span> |
| <span id="326">326</span> |
| <span id="327">327</span> |
| <span id="328">328</span> |
| <span id="329">329</span> |
| <span id="330">330</span> |
| <span id="331">331</span> |
| <span id="332">332</span> |
| <span id="333">333</span> |
| <span id="334">334</span> |
| <span id="335">335</span> |
| <span id="336">336</span> |
| <span id="337">337</span> |
| <span id="338">338</span> |
| <span id="339">339</span> |
| <span id="340">340</span> |
| <span id="341">341</span> |
| <span id="342">342</span> |
| <span id="343">343</span> |
| <span id="344">344</span> |
| <span id="345">345</span> |
| <span id="346">346</span> |
| <span id="347">347</span> |
| <span id="348">348</span> |
| <span id="349">349</span> |
| <span id="350">350</span> |
| <span id="351">351</span> |
| <span id="352">352</span> |
| <span id="353">353</span> |
| <span id="354">354</span> |
| <span id="355">355</span> |
| <span id="356">356</span> |
| <span id="357">357</span> |
| <span id="358">358</span> |
| <span id="359">359</span> |
| <span id="360">360</span> |
| <span id="361">361</span> |
| <span id="362">362</span> |
| <span id="363">363</span> |
| <span id="364">364</span> |
| <span id="365">365</span> |
| <span id="366">366</span> |
| <span id="367">367</span> |
| <span id="368">368</span> |
| <span id="369">369</span> |
| <span id="370">370</span> |
| <span id="371">371</span> |
| <span id="372">372</span> |
| <span id="373">373</span> |
| <span id="374">374</span> |
| <span id="375">375</span> |
| <span id="376">376</span> |
| <span id="377">377</span> |
| <span id="378">378</span> |
| <span id="379">379</span> |
| <span id="380">380</span> |
| <span id="381">381</span> |
| <span id="382">382</span> |
| <span id="383">383</span> |
| <span id="384">384</span> |
| <span id="385">385</span> |
| <span id="386">386</span> |
| <span id="387">387</span> |
| <span id="388">388</span> |
| <span id="389">389</span> |
| <span id="390">390</span> |
| <span id="391">391</span> |
| <span id="392">392</span> |
| <span id="393">393</span> |
| <span id="394">394</span> |
| <span id="395">395</span> |
| <span id="396">396</span> |
| <span id="397">397</span> |
| <span id="398">398</span> |
| <span id="399">399</span> |
| <span id="400">400</span> |
| <span id="401">401</span> |
| <span id="402">402</span> |
| <span id="403">403</span> |
| <span id="404">404</span> |
| <span id="405">405</span> |
| <span id="406">406</span> |
| <span id="407">407</span> |
| <span id="408">408</span> |
| <span id="409">409</span> |
| <span id="410">410</span> |
| <span id="411">411</span> |
| <span id="412">412</span> |
| <span id="413">413</span> |
| <span id="414">414</span> |
| <span id="415">415</span> |
| <span id="416">416</span> |
| <span id="417">417</span> |
| <span id="418">418</span> |
| <span id="419">419</span> |
| <span id="420">420</span> |
| <span id="421">421</span> |
| <span id="422">422</span> |
| <span id="423">423</span> |
| <span id="424">424</span> |
| <span id="425">425</span> |
| <span id="426">426</span> |
| <span id="427">427</span> |
| <span id="428">428</span> |
| <span id="429">429</span> |
| <span id="430">430</span> |
| <span id="431">431</span> |
| <span id="432">432</span> |
| <span id="433">433</span> |
| <span id="434">434</span> |
| <span id="435">435</span> |
| <span id="436">436</span> |
| <span id="437">437</span> |
| <span id="438">438</span> |
| <span id="439">439</span> |
| <span id="440">440</span> |
| <span id="441">441</span> |
| <span id="442">442</span> |
| <span id="443">443</span> |
| <span id="444">444</span> |
| <span id="445">445</span> |
| <span id="446">446</span> |
| <span id="447">447</span> |
| <span id="448">448</span> |
| <span id="449">449</span> |
| <span id="450">450</span> |
| <span id="451">451</span> |
| <span id="452">452</span> |
| <span id="453">453</span> |
| <span id="454">454</span> |
| <span id="455">455</span> |
| <span id="456">456</span> |
| <span id="457">457</span> |
| <span id="458">458</span> |
| <span id="459">459</span> |
| <span id="460">460</span> |
| <span id="461">461</span> |
| <span id="462">462</span> |
| <span id="463">463</span> |
| <span id="464">464</span> |
| <span id="465">465</span> |
| <span id="466">466</span> |
| <span id="467">467</span> |
| <span id="468">468</span> |
| <span id="469">469</span> |
| <span id="470">470</span> |
| <span id="471">471</span> |
| <span id="472">472</span> |
| <span id="473">473</span> |
| <span id="474">474</span> |
| <span id="475">475</span> |
| <span id="476">476</span> |
| <span id="477">477</span> |
| <span id="478">478</span> |
| <span id="479">479</span> |
| <span id="480">480</span> |
| <span id="481">481</span> |
| <span id="482">482</span> |
| <span id="483">483</span> |
| <span id="484">484</span> |
| <span id="485">485</span> |
| <span id="486">486</span> |
| <span id="487">487</span> |
| <span id="488">488</span> |
| <span id="489">489</span> |
| <span id="490">490</span> |
| <span id="491">491</span> |
| <span id="492">492</span> |
| <span id="493">493</span> |
| <span id="494">494</span> |
| <span id="495">495</span> |
| <span id="496">496</span> |
| <span id="497">497</span> |
| <span id="498">498</span> |
| <span id="499">499</span> |
| <span id="500">500</span> |
| <span id="501">501</span> |
| <span id="502">502</span> |
| <span id="503">503</span> |
| <span id="504">504</span> |
| <span id="505">505</span> |
| <span id="506">506</span> |
| <span id="507">507</span> |
| <span id="508">508</span> |
| <span id="509">509</span> |
| <span id="510">510</span> |
| <span id="511">511</span> |
| <span id="512">512</span> |
| <span id="513">513</span> |
| <span id="514">514</span> |
| <span id="515">515</span> |
| <span id="516">516</span> |
| <span id="517">517</span> |
| <span id="518">518</span> |
| <span id="519">519</span> |
| <span id="520">520</span> |
| <span id="521">521</span> |
| <span id="522">522</span> |
| <span id="523">523</span> |
| <span id="524">524</span> |
| <span id="525">525</span> |
| <span id="526">526</span> |
| <span id="527">527</span> |
| <span id="528">528</span> |
| <span id="529">529</span> |
| <span id="530">530</span> |
| <span id="531">531</span> |
| <span id="532">532</span> |
| <span id="533">533</span> |
| <span id="534">534</span> |
| <span id="535">535</span> |
| <span id="536">536</span> |
| <span id="537">537</span> |
| <span id="538">538</span> |
| <span id="539">539</span> |
| <span id="540">540</span> |
| <span id="541">541</span> |
| <span id="542">542</span> |
| <span id="543">543</span> |
| <span id="544">544</span> |
| <span id="545">545</span> |
| <span id="546">546</span> |
| <span id="547">547</span> |
| <span id="548">548</span> |
| <span id="549">549</span> |
| <span id="550">550</span> |
| <span id="551">551</span> |
| <span id="552">552</span> |
| <span id="553">553</span> |
| <span id="554">554</span> |
| <span id="555">555</span> |
| <span id="556">556</span> |
| <span id="557">557</span> |
| <span id="558">558</span> |
| <span id="559">559</span> |
| <span id="560">560</span> |
| <span id="561">561</span> |
| <span id="562">562</span> |
| <span id="563">563</span> |
| <span id="564">564</span> |
| <span id="565">565</span> |
| <span id="566">566</span> |
| <span id="567">567</span> |
| <span id="568">568</span> |
| <span id="569">569</span> |
| <span id="570">570</span> |
| <span id="571">571</span> |
| <span id="572">572</span> |
| <span id="573">573</span> |
| <span id="574">574</span> |
| <span id="575">575</span> |
| <span id="576">576</span> |
| <span id="577">577</span> |
| <span id="578">578</span> |
| <span id="579">579</span> |
| <span id="580">580</span> |
| <span id="581">581</span> |
| <span id="582">582</span> |
| <span id="583">583</span> |
| <span id="584">584</span> |
| <span id="585">585</span> |
| <span id="586">586</span> |
| <span id="587">587</span> |
| <span id="588">588</span> |
| <span id="589">589</span> |
| <span id="590">590</span> |
| <span id="591">591</span> |
| <span id="592">592</span> |
| <span id="593">593</span> |
| <span id="594">594</span> |
| <span id="595">595</span> |
| <span id="596">596</span> |
| </pre><pre class="rust"><code><span class="kw">use </span><span class="kw">crate</span>::codec::UserError; |
| <span class="kw">use </span><span class="kw">crate</span>::frame::{Reason, StreamId}; |
| <span class="kw">use crate</span>::{client, frame, server}; |
| |
| <span class="kw">use </span><span class="kw">crate</span>::frame::DEFAULT_INITIAL_WINDOW_SIZE; |
| <span class="kw">use </span><span class="kw">crate</span>::proto::<span class="kw-2">*</span>; |
| |
| <span class="kw">use </span>bytes::{Buf, Bytes}; |
| <span class="kw">use </span>futures_core::Stream; |
| <span class="kw">use </span>std::io; |
| <span class="kw">use </span>std::marker::PhantomData; |
| <span class="kw">use </span>std::pin::Pin; |
| <span class="kw">use </span>std::task::{Context, Poll}; |
| <span class="kw">use </span>std::time::Duration; |
| <span class="kw">use </span>tokio::io::{AsyncRead, AsyncWrite}; |
| |
| <span class="doccomment">/// An H2 connection |
| </span><span class="attribute">#[derive(Debug)] |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">struct </span>Connection<T, P, B: Buf = Bytes> |
| <span class="kw">where |
| </span>P: Peer, |
| { |
| <span class="doccomment">/// Read / write frame values |
| </span>codec: Codec<T, Prioritized<B>>, |
| |
| inner: ConnectionInner<P, B>, |
| } |
| |
| <span class="comment">// Extracted part of `Connection` which does not depend on `T`. Reduces the amount of duplicated |
| // method instantiations. |
| </span><span class="attribute">#[derive(Debug)] |
| </span><span class="kw">struct </span>ConnectionInner<P, B: Buf = Bytes> |
| <span class="kw">where |
| </span>P: Peer, |
| { |
| <span class="doccomment">/// Tracks the connection level state transitions. |
| </span>state: State, |
| |
| <span class="doccomment">/// An error to report back once complete. |
| /// |
| /// This exists separately from State in order to support |
| /// graceful shutdown. |
| </span>error: <span class="prelude-ty">Option</span><frame::GoAway>, |
| |
| <span class="doccomment">/// Pending GOAWAY frames to write. |
| </span>go_away: GoAway, |
| |
| <span class="doccomment">/// Ping/pong handler |
| </span>ping_pong: PingPong, |
| |
| <span class="doccomment">/// Connection settings |
| </span>settings: Settings, |
| |
| <span class="doccomment">/// Stream state handler |
| </span>streams: Streams<B, P>, |
| |
| <span class="doccomment">/// A `tracing` span tracking the lifetime of the connection. |
| </span>span: tracing::Span, |
| |
| <span class="doccomment">/// Client or server |
| </span>_phantom: PhantomData<P>, |
| } |
| |
| <span class="kw">struct </span>DynConnection<<span class="lifetime">'a</span>, B: Buf = Bytes> { |
| state: <span class="kw-2">&</span><span class="lifetime">'a </span><span class="kw-2">mut </span>State, |
| |
| go_away: <span class="kw-2">&</span><span class="lifetime">'a </span><span class="kw-2">mut </span>GoAway, |
| |
| streams: DynStreams<<span class="lifetime">'a</span>, B>, |
| |
| error: <span class="kw-2">&</span><span class="lifetime">'a </span><span class="kw-2">mut </span><span class="prelude-ty">Option</span><frame::GoAway>, |
| |
| ping_pong: <span class="kw-2">&</span><span class="lifetime">'a </span><span class="kw-2">mut </span>PingPong, |
| } |
| |
| <span class="attribute">#[derive(Debug, Clone)] |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">struct </span>Config { |
| <span class="kw">pub </span>next_stream_id: StreamId, |
| <span class="kw">pub </span>initial_max_send_streams: usize, |
| <span class="kw">pub </span>max_send_buffer_size: usize, |
| <span class="kw">pub </span>reset_stream_duration: Duration, |
| <span class="kw">pub </span>reset_stream_max: usize, |
| <span class="kw">pub </span>remote_reset_stream_max: usize, |
| <span class="kw">pub </span>settings: frame::Settings, |
| } |
| |
| <span class="attribute">#[derive(Debug)] |
| </span><span class="kw">enum </span>State { |
| <span class="doccomment">/// Currently open in a sane state |
| </span>Open, |
| |
| <span class="doccomment">/// The codec must be flushed |
| </span>Closing(Reason, Initiator), |
| |
| <span class="doccomment">/// In a closed state |
| </span>Closed(Reason, Initiator), |
| } |
| |
| <span class="kw">impl</span><T, P, B> Connection<T, P, B> |
| <span class="kw">where |
| </span>T: AsyncRead + AsyncWrite + Unpin, |
| P: Peer, |
| B: Buf, |
| { |
| <span class="kw">pub fn </span>new(codec: Codec<T, Prioritized<B>>, config: Config) -> Connection<T, P, B> { |
| <span class="kw">fn </span>streams_config(config: <span class="kw-2">&</span>Config) -> streams::Config { |
| streams::Config { |
| local_init_window_sz: config |
| .settings |
| .initial_window_size() |
| .unwrap_or(DEFAULT_INITIAL_WINDOW_SIZE), |
| initial_max_send_streams: config.initial_max_send_streams, |
| local_max_buffer_size: config.max_send_buffer_size, |
| local_next_stream_id: config.next_stream_id, |
| local_push_enabled: config.settings.is_push_enabled().unwrap_or(<span class="bool-val">true</span>), |
| extended_connect_protocol_enabled: config |
| .settings |
| .is_extended_connect_protocol_enabled() |
| .unwrap_or(<span class="bool-val">false</span>), |
| local_reset_duration: config.reset_stream_duration, |
| local_reset_max: config.reset_stream_max, |
| remote_reset_max: config.remote_reset_stream_max, |
| remote_init_window_sz: DEFAULT_INITIAL_WINDOW_SIZE, |
| remote_max_initiated: config |
| .settings |
| .max_concurrent_streams() |
| .map(|max| max <span class="kw">as </span>usize), |
| } |
| } |
| <span class="kw">let </span>streams = Streams::new(streams_config(<span class="kw-2">&</span>config)); |
| Connection { |
| codec, |
| inner: ConnectionInner { |
| state: State::Open, |
| error: <span class="prelude-val">None</span>, |
| go_away: GoAway::new(), |
| ping_pong: PingPong::new(), |
| settings: Settings::new(config.settings), |
| streams, |
| span: <span class="macro">tracing::debug_span!</span>(<span class="string">"Connection"</span>, peer = %P::NAME), |
| _phantom: PhantomData, |
| }, |
| } |
| } |
| |
| <span class="doccomment">/// connection flow control |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>set_target_window_size(<span class="kw-2">&mut </span><span class="self">self</span>, size: WindowSize) { |
| <span class="self">self</span>.inner.streams.set_target_connection_window_size(size); |
| } |
| |
| <span class="doccomment">/// Send a new SETTINGS frame with an updated initial window size. |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>set_initial_window_size(<span class="kw-2">&mut </span><span class="self">self</span>, size: WindowSize) -> <span class="prelude-ty">Result</span><(), UserError> { |
| <span class="kw">let </span><span class="kw-2">mut </span>settings = frame::Settings::default(); |
| settings.set_initial_window_size(<span class="prelude-val">Some</span>(size)); |
| <span class="self">self</span>.inner.settings.send_settings(settings) |
| } |
| |
| <span class="doccomment">/// Send a new SETTINGS frame with extended CONNECT protocol enabled. |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>set_enable_connect_protocol(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Result</span><(), UserError> { |
| <span class="kw">let </span><span class="kw-2">mut </span>settings = frame::Settings::default(); |
| settings.set_enable_connect_protocol(<span class="prelude-val">Some</span>(<span class="number">1</span>)); |
| <span class="self">self</span>.inner.settings.send_settings(settings) |
| } |
| |
| <span class="doccomment">/// Returns the maximum number of concurrent streams that may be initiated |
| /// by this peer. |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>max_send_streams(<span class="kw-2">&</span><span class="self">self</span>) -> usize { |
| <span class="self">self</span>.inner.streams.max_send_streams() |
| } |
| |
| <span class="doccomment">/// Returns the maximum number of concurrent streams that may be initiated |
| /// by the remote peer. |
| </span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>max_recv_streams(<span class="kw-2">&</span><span class="self">self</span>) -> usize { |
| <span class="self">self</span>.inner.streams.max_recv_streams() |
| } |
| |
| <span class="attribute">#[cfg(feature = <span class="string">"unstable"</span>)] |
| </span><span class="kw">pub fn </span>num_wired_streams(<span class="kw-2">&</span><span class="self">self</span>) -> usize { |
| <span class="self">self</span>.inner.streams.num_wired_streams() |
| } |
| |
| <span class="doccomment">/// Returns `Ready` when the connection is ready to receive a frame. |
| /// |
| /// Returns `Error` as this may raise errors that are caused by delayed |
| /// processing of received frames. |
| </span><span class="kw">fn </span>poll_ready(<span class="kw-2">&mut </span><span class="self">self</span>, cx: <span class="kw-2">&mut </span>Context) -> Poll<<span class="prelude-ty">Result</span><(), Error>> { |
| <span class="kw">let </span>_e = <span class="self">self</span>.inner.span.enter(); |
| <span class="kw">let </span>span = <span class="macro">tracing::trace_span!</span>(<span class="string">"poll_ready"</span>); |
| <span class="kw">let </span>_e = span.enter(); |
| <span class="comment">// The order of these calls don't really matter too much |
| </span><span class="macro">ready!</span>(<span class="self">self</span>.inner.ping_pong.send_pending_pong(cx, <span class="kw-2">&mut </span><span class="self">self</span>.codec))<span class="question-mark">?</span>; |
| <span class="macro">ready!</span>(<span class="self">self</span>.inner.ping_pong.send_pending_ping(cx, <span class="kw-2">&mut </span><span class="self">self</span>.codec))<span class="question-mark">?</span>; |
| <span class="macro">ready!</span>(<span class="self">self |
| </span>.inner |
| .settings |
| .poll_send(cx, <span class="kw-2">&mut </span><span class="self">self</span>.codec, <span class="kw-2">&mut </span><span class="self">self</span>.inner.streams))<span class="question-mark">?</span>; |
| <span class="macro">ready!</span>(<span class="self">self</span>.inner.streams.send_pending_refusal(cx, <span class="kw-2">&mut </span><span class="self">self</span>.codec))<span class="question-mark">?</span>; |
| |
| Poll::Ready(<span class="prelude-val">Ok</span>(())) |
| } |
| |
| <span class="doccomment">/// Send any pending GOAWAY frames. |
| /// |
| /// This will return `Some(reason)` if the connection should be closed |
| /// afterwards. If this is a graceful shutdown, this returns `None`. |
| </span><span class="kw">fn </span>poll_go_away(<span class="kw-2">&mut </span><span class="self">self</span>, cx: <span class="kw-2">&mut </span>Context) -> Poll<<span class="prelude-ty">Option</span><io::Result<Reason>>> { |
| <span class="self">self</span>.inner.go_away.send_pending_go_away(cx, <span class="kw-2">&mut </span><span class="self">self</span>.codec) |
| } |
| |
| <span class="kw">pub fn </span>go_away_from_user(<span class="kw-2">&mut </span><span class="self">self</span>, e: Reason) { |
| <span class="self">self</span>.inner.as_dyn().go_away_from_user(e) |
| } |
| |
| <span class="kw">fn </span>take_error(<span class="kw-2">&mut </span><span class="self">self</span>, ours: Reason, initiator: Initiator) -> <span class="prelude-ty">Result</span><(), Error> { |
| <span class="kw">let </span>(debug_data, theirs) = <span class="self">self |
| </span>.inner |
| .error |
| .take() |
| .as_ref() |
| .map_or((Bytes::new(), Reason::NO_ERROR), |frame| { |
| (frame.debug_data().clone(), frame.reason()) |
| }); |
| |
| <span class="kw">match </span>(ours, theirs) { |
| (Reason::NO_ERROR, Reason::NO_ERROR) => <span class="prelude-val">Ok</span>(()), |
| (ours, Reason::NO_ERROR) => <span class="prelude-val">Err</span>(Error::GoAway(Bytes::new(), ours, initiator)), |
| <span class="comment">// If both sides reported an error, give their |
| // error back to th user. We assume our error |
| // was a consequence of their error, and less |
| // important. |
| </span>(<span class="kw">_</span>, theirs) => <span class="prelude-val">Err</span>(Error::remote_go_away(debug_data, theirs)), |
| } |
| } |
| |
| <span class="doccomment">/// Closes the connection by transitioning to a GOAWAY state |
| /// iff there are no streams or references |
| </span><span class="kw">pub fn </span>maybe_close_connection_if_no_streams(<span class="kw-2">&mut </span><span class="self">self</span>) { |
| <span class="comment">// If we poll() and realize that there are no streams or references |
| // then we can close the connection by transitioning to GOAWAY |
| </span><span class="kw">if </span>!<span class="self">self</span>.inner.streams.has_streams_or_other_references() { |
| <span class="self">self</span>.inner.as_dyn().go_away_now(Reason::NO_ERROR); |
| } |
| } |
| |
| <span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>take_user_pings(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Option</span><UserPings> { |
| <span class="self">self</span>.inner.ping_pong.take_user_pings() |
| } |
| |
| <span class="doccomment">/// Advances the internal state of the connection. |
| </span><span class="kw">pub fn </span>poll(<span class="kw-2">&mut </span><span class="self">self</span>, cx: <span class="kw-2">&mut </span>Context) -> Poll<<span class="prelude-ty">Result</span><(), Error>> { |
| <span class="comment">// XXX(eliza): cloning the span is unfortunately necessary here in |
| // order to placate the borrow checker — `self` is mutably borrowed by |
| // `poll2`, which means that we can't borrow `self.span` to enter it. |
| // The clone is just an atomic ref bump. |
| </span><span class="kw">let </span>span = <span class="self">self</span>.inner.span.clone(); |
| <span class="kw">let </span>_e = span.enter(); |
| <span class="kw">let </span>span = <span class="macro">tracing::trace_span!</span>(<span class="string">"poll"</span>); |
| <span class="kw">let </span>_e = span.enter(); |
| |
| <span class="kw">loop </span>{ |
| <span class="macro">tracing::trace!</span>(connection.state = <span class="question-mark">?</span><span class="self">self</span>.inner.state); |
| <span class="comment">// TODO: probably clean up this glob of code |
| </span><span class="kw">match </span><span class="self">self</span>.inner.state { |
| <span class="comment">// When open, continue to poll a frame |
| </span>State::Open => { |
| <span class="kw">let </span>result = <span class="kw">match </span><span class="self">self</span>.poll2(cx) { |
| Poll::Ready(result) => result, |
| <span class="comment">// The connection is not ready to make progress |
| </span>Poll::Pending => { |
| <span class="comment">// Ensure all window updates have been sent. |
| // |
| // This will also handle flushing `self.codec` |
| </span><span class="macro">ready!</span>(<span class="self">self</span>.inner.streams.poll_complete(cx, <span class="kw-2">&mut </span><span class="self">self</span>.codec))<span class="question-mark">?</span>; |
| |
| <span class="kw">if </span>(<span class="self">self</span>.inner.error.is_some() |
| || <span class="self">self</span>.inner.go_away.should_close_on_idle()) |
| && !<span class="self">self</span>.inner.streams.has_streams() |
| { |
| <span class="self">self</span>.inner.as_dyn().go_away_now(Reason::NO_ERROR); |
| <span class="kw">continue</span>; |
| } |
| |
| <span class="kw">return </span>Poll::Pending; |
| } |
| }; |
| |
| <span class="self">self</span>.inner.as_dyn().handle_poll2_result(result)<span class="question-mark">? |
| </span>} |
| State::Closing(reason, initiator) => { |
| <span class="macro">tracing::trace!</span>(<span class="string">"connection closing after flush"</span>); |
| <span class="comment">// Flush/shutdown the codec |
| </span><span class="macro">ready!</span>(<span class="self">self</span>.codec.shutdown(cx))<span class="question-mark">?</span>; |
| |
| <span class="comment">// Transition the state to error |
| </span><span class="self">self</span>.inner.state = State::Closed(reason, initiator); |
| } |
| State::Closed(reason, initiator) => { |
| <span class="kw">return </span>Poll::Ready(<span class="self">self</span>.take_error(reason, initiator)); |
| } |
| } |
| } |
| } |
| |
| <span class="kw">fn </span>poll2(<span class="kw-2">&mut </span><span class="self">self</span>, cx: <span class="kw-2">&mut </span>Context) -> Poll<<span class="prelude-ty">Result</span><(), Error>> { |
| <span class="comment">// This happens outside of the loop to prevent needing to do a clock |
| // check and then comparison of the queue possibly multiple times a |
| // second (and thus, the clock wouldn't have changed enough to matter). |
| </span><span class="self">self</span>.clear_expired_reset_streams(); |
| |
| <span class="kw">loop </span>{ |
| <span class="comment">// First, ensure that the `Connection` is able to receive a frame |
| // |
| // The order here matters: |
| // - poll_go_away may buffer a graceful shutdown GOAWAY frame |
| // - If it has, we've also added a PING to be sent in poll_ready |
| </span><span class="kw">if let </span><span class="prelude-val">Some</span>(reason) = <span class="macro">ready!</span>(<span class="self">self</span>.poll_go_away(cx)<span class="question-mark">?</span>) { |
| <span class="kw">if </span><span class="self">self</span>.inner.go_away.should_close_now() { |
| <span class="kw">if </span><span class="self">self</span>.inner.go_away.is_user_initiated() { |
| <span class="comment">// A user initiated abrupt shutdown shouldn't return |
| // the same error back to the user. |
| </span><span class="kw">return </span>Poll::Ready(<span class="prelude-val">Ok</span>(())); |
| } <span class="kw">else </span>{ |
| <span class="kw">return </span>Poll::Ready(<span class="prelude-val">Err</span>(Error::library_go_away(reason))); |
| } |
| } |
| <span class="comment">// Only NO_ERROR should be waiting for idle |
| </span><span class="macro">debug_assert_eq!</span>( |
| reason, |
| Reason::NO_ERROR, |
| <span class="string">"graceful GOAWAY should be NO_ERROR" |
| </span>); |
| } |
| <span class="macro">ready!</span>(<span class="self">self</span>.poll_ready(cx))<span class="question-mark">?</span>; |
| |
| <span class="kw">match </span><span class="self">self |
| </span>.inner |
| .as_dyn() |
| .recv_frame(<span class="macro">ready!</span>(Pin::new(<span class="kw-2">&mut </span><span class="self">self</span>.codec).poll_next(cx)<span class="question-mark">?</span>))<span class="question-mark">? |
| </span>{ |
| ReceivedFrame::Settings(frame) => { |
| <span class="self">self</span>.inner.settings.recv_settings( |
| frame, |
| <span class="kw-2">&mut </span><span class="self">self</span>.codec, |
| <span class="kw-2">&mut </span><span class="self">self</span>.inner.streams, |
| )<span class="question-mark">?</span>; |
| } |
| ReceivedFrame::Continue => (), |
| ReceivedFrame::Done => { |
| <span class="kw">return </span>Poll::Ready(<span class="prelude-val">Ok</span>(())); |
| } |
| } |
| } |
| } |
| |
| <span class="kw">fn </span>clear_expired_reset_streams(<span class="kw-2">&mut </span><span class="self">self</span>) { |
| <span class="self">self</span>.inner.streams.clear_expired_reset_streams(); |
| } |
| } |
| |
| <span class="kw">impl</span><P, B> ConnectionInner<P, B> |
| <span class="kw">where |
| </span>P: Peer, |
| B: Buf, |
| { |
| <span class="kw">fn </span>as_dyn(<span class="kw-2">&mut </span><span class="self">self</span>) -> DynConnection<<span class="lifetime">'_</span>, B> { |
| <span class="kw">let </span>ConnectionInner { |
| state, |
| go_away, |
| streams, |
| error, |
| ping_pong, |
| .. |
| } = <span class="self">self</span>; |
| <span class="kw">let </span>streams = streams.as_dyn(); |
| DynConnection { |
| state, |
| go_away, |
| streams, |
| error, |
| ping_pong, |
| } |
| } |
| } |
| |
| <span class="kw">impl</span><B> DynConnection<<span class="lifetime">'_</span>, B> |
| <span class="kw">where |
| </span>B: Buf, |
| { |
| <span class="kw">fn </span>go_away(<span class="kw-2">&mut </span><span class="self">self</span>, id: StreamId, e: Reason) { |
| <span class="kw">let </span>frame = frame::GoAway::new(id, e); |
| <span class="self">self</span>.streams.send_go_away(id); |
| <span class="self">self</span>.go_away.go_away(frame); |
| } |
| |
| <span class="kw">fn </span>go_away_now(<span class="kw-2">&mut </span><span class="self">self</span>, e: Reason) { |
| <span class="kw">let </span>last_processed_id = <span class="self">self</span>.streams.last_processed_id(); |
| <span class="kw">let </span>frame = frame::GoAway::new(last_processed_id, e); |
| <span class="self">self</span>.go_away.go_away_now(frame); |
| } |
| |
| <span class="kw">fn </span>go_away_now_data(<span class="kw-2">&mut </span><span class="self">self</span>, e: Reason, data: Bytes) { |
| <span class="kw">let </span>last_processed_id = <span class="self">self</span>.streams.last_processed_id(); |
| <span class="kw">let </span>frame = frame::GoAway::with_debug_data(last_processed_id, e, data); |
| <span class="self">self</span>.go_away.go_away_now(frame); |
| } |
| |
| <span class="kw">fn </span>go_away_from_user(<span class="kw-2">&mut </span><span class="self">self</span>, e: Reason) { |
| <span class="kw">let </span>last_processed_id = <span class="self">self</span>.streams.last_processed_id(); |
| <span class="kw">let </span>frame = frame::GoAway::new(last_processed_id, e); |
| <span class="self">self</span>.go_away.go_away_from_user(frame); |
| |
| <span class="comment">// Notify all streams of reason we're abruptly closing. |
| </span><span class="self">self</span>.streams.handle_error(Error::user_go_away(e)); |
| } |
| |
| <span class="kw">fn </span>handle_poll2_result(<span class="kw-2">&mut </span><span class="self">self</span>, result: <span class="prelude-ty">Result</span><(), Error>) -> <span class="prelude-ty">Result</span><(), Error> { |
| <span class="kw">match </span>result { |
| <span class="comment">// The connection has shutdown normally |
| </span><span class="prelude-val">Ok</span>(()) => { |
| <span class="kw-2">*</span><span class="self">self</span>.state = State::Closing(Reason::NO_ERROR, Initiator::Library); |
| <span class="prelude-val">Ok</span>(()) |
| } |
| <span class="comment">// Attempting to read a frame resulted in a connection level |
| // error. This is handled by setting a GOAWAY frame followed by |
| // terminating the connection. |
| </span><span class="prelude-val">Err</span>(Error::GoAway(debug_data, reason, initiator)) => { |
| <span class="kw">let </span>e = Error::GoAway(debug_data.clone(), reason, initiator); |
| <span class="macro">tracing::debug!</span>(error = <span class="question-mark">?</span>e, <span class="string">"Connection::poll; connection error"</span>); |
| |
| <span class="comment">// We may have already sent a GOAWAY for this error, |
| // if so, don't send another, just flush and close up. |
| </span><span class="kw">if </span><span class="self">self |
| </span>.go_away |
| .going_away() |
| .map_or(<span class="bool-val">false</span>, |frame| frame.reason() == reason) |
| { |
| <span class="macro">tracing::trace!</span>(<span class="string">" -> already going away"</span>); |
| <span class="kw-2">*</span><span class="self">self</span>.state = State::Closing(reason, initiator); |
| <span class="kw">return </span><span class="prelude-val">Ok</span>(()); |
| } |
| |
| <span class="comment">// Reset all active streams |
| </span><span class="self">self</span>.streams.handle_error(e); |
| <span class="self">self</span>.go_away_now_data(reason, debug_data); |
| <span class="prelude-val">Ok</span>(()) |
| } |
| <span class="comment">// Attempting to read a frame resulted in a stream level error. |
| // This is handled by resetting the frame then trying to read |
| // another frame. |
| </span><span class="prelude-val">Err</span>(Error::Reset(id, reason, initiator)) => { |
| <span class="macro">debug_assert_eq!</span>(initiator, Initiator::Library); |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>id, <span class="question-mark">?</span>reason, <span class="string">"stream error"</span>); |
| <span class="self">self</span>.streams.send_reset(id, reason); |
| <span class="prelude-val">Ok</span>(()) |
| } |
| <span class="comment">// Attempting to read a frame resulted in an I/O error. All |
| // active streams must be reset. |
| // |
| // TODO: Are I/O errors recoverable? |
| </span><span class="prelude-val">Err</span>(Error::Io(e, inner)) => { |
| <span class="macro">tracing::debug!</span>(error = <span class="question-mark">?</span>e, <span class="string">"Connection::poll; IO error"</span>); |
| <span class="kw">let </span>e = Error::Io(e, inner); |
| |
| <span class="comment">// Reset all active streams |
| </span><span class="self">self</span>.streams.handle_error(e.clone()); |
| |
| <span class="comment">// Return the error |
| </span><span class="prelude-val">Err</span>(e) |
| } |
| } |
| } |
| |
| <span class="kw">fn </span>recv_frame(<span class="kw-2">&mut </span><span class="self">self</span>, frame: <span class="prelude-ty">Option</span><Frame>) -> <span class="prelude-ty">Result</span><ReceivedFrame, Error> { |
| <span class="kw">use </span><span class="kw">crate</span>::frame::Frame::<span class="kw-2">*</span>; |
| <span class="kw">match </span>frame { |
| <span class="prelude-val">Some</span>(Headers(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv HEADERS"</span>); |
| <span class="self">self</span>.streams.recv_headers(frame)<span class="question-mark">?</span>; |
| } |
| <span class="prelude-val">Some</span>(Data(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv DATA"</span>); |
| <span class="self">self</span>.streams.recv_data(frame)<span class="question-mark">?</span>; |
| } |
| <span class="prelude-val">Some</span>(Reset(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv RST_STREAM"</span>); |
| <span class="self">self</span>.streams.recv_reset(frame)<span class="question-mark">?</span>; |
| } |
| <span class="prelude-val">Some</span>(PushPromise(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv PUSH_PROMISE"</span>); |
| <span class="self">self</span>.streams.recv_push_promise(frame)<span class="question-mark">?</span>; |
| } |
| <span class="prelude-val">Some</span>(Settings(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv SETTINGS"</span>); |
| <span class="kw">return </span><span class="prelude-val">Ok</span>(ReceivedFrame::Settings(frame)); |
| } |
| <span class="prelude-val">Some</span>(GoAway(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv GOAWAY"</span>); |
| <span class="comment">// This should prevent starting new streams, |
| // but should allow continuing to process current streams |
| // until they are all EOS. Once they are, State should |
| // transition to GoAway. |
| </span><span class="self">self</span>.streams.recv_go_away(<span class="kw-2">&</span>frame)<span class="question-mark">?</span>; |
| <span class="kw-2">*</span><span class="self">self</span>.error = <span class="prelude-val">Some</span>(frame); |
| } |
| <span class="prelude-val">Some</span>(Ping(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv PING"</span>); |
| <span class="kw">let </span>status = <span class="self">self</span>.ping_pong.recv_ping(frame); |
| <span class="kw">if </span>status.is_shutdown() { |
| <span class="macro">assert!</span>( |
| <span class="self">self</span>.go_away.is_going_away(), |
| <span class="string">"received unexpected shutdown ping" |
| </span>); |
| |
| <span class="kw">let </span>last_processed_id = <span class="self">self</span>.streams.last_processed_id(); |
| <span class="self">self</span>.go_away(last_processed_id, Reason::NO_ERROR); |
| } |
| } |
| <span class="prelude-val">Some</span>(WindowUpdate(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv WINDOW_UPDATE"</span>); |
| <span class="self">self</span>.streams.recv_window_update(frame)<span class="question-mark">?</span>; |
| } |
| <span class="prelude-val">Some</span>(Priority(frame)) => { |
| <span class="macro">tracing::trace!</span>(<span class="question-mark">?</span>frame, <span class="string">"recv PRIORITY"</span>); |
| <span class="comment">// TODO: handle |
| </span>} |
| <span class="prelude-val">None </span>=> { |
| <span class="macro">tracing::trace!</span>(<span class="string">"codec closed"</span>); |
| <span class="self">self</span>.streams.recv_eof(<span class="bool-val">false</span>).expect(<span class="string">"mutex poisoned"</span>); |
| <span class="kw">return </span><span class="prelude-val">Ok</span>(ReceivedFrame::Done); |
| } |
| } |
| <span class="prelude-val">Ok</span>(ReceivedFrame::Continue) |
| } |
| } |
| |
| <span class="kw">enum </span>ReceivedFrame { |
| Settings(frame::Settings), |
| Continue, |
| Done, |
| } |
| |
| <span class="kw">impl</span><T, B> Connection<T, client::Peer, B> |
| <span class="kw">where |
| </span>T: AsyncRead + AsyncWrite, |
| B: Buf, |
| { |
| <span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>streams(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="kw-2">&</span>Streams<B, client::Peer> { |
| <span class="kw-2">&</span><span class="self">self</span>.inner.streams |
| } |
| } |
| |
| <span class="kw">impl</span><T, B> Connection<T, server::Peer, B> |
| <span class="kw">where |
| </span>T: AsyncRead + AsyncWrite + Unpin, |
| B: Buf, |
| { |
| <span class="kw">pub fn </span>next_incoming(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Option</span><StreamRef<B>> { |
| <span class="self">self</span>.inner.streams.next_incoming() |
| } |
| |
| <span class="comment">// Graceful shutdown only makes sense for server peers. |
| </span><span class="kw">pub fn </span>go_away_gracefully(<span class="kw-2">&mut </span><span class="self">self</span>) { |
| <span class="kw">if </span><span class="self">self</span>.inner.go_away.is_going_away() { |
| <span class="comment">// No reason to start a new one. |
| </span><span class="kw">return</span>; |
| } |
| |
| <span class="comment">// According to http://httpwg.org/specs/rfc7540.html#GOAWAY: |
| // |
| // > A server that is attempting to gracefully shut down a connection |
| // > SHOULD send an initial GOAWAY frame with the last stream |
| // > identifier set to 2^31-1 and a NO_ERROR code. This signals to the |
| // > client that a shutdown is imminent and that initiating further |
| // > requests is prohibited. After allowing time for any in-flight |
| // > stream creation (at least one round-trip time), the server can |
| // > send another GOAWAY frame with an updated last stream identifier. |
| // > This ensures that a connection can be cleanly shut down without |
| // > losing requests. |
| </span><span class="self">self</span>.inner.as_dyn().go_away(StreamId::MAX, Reason::NO_ERROR); |
| |
| <span class="comment">// We take the advice of waiting 1 RTT literally, and wait |
| // for a pong before proceeding. |
| </span><span class="self">self</span>.inner.ping_pong.ping_shutdown(); |
| } |
| } |
| |
| <span class="kw">impl</span><T, P, B> Drop <span class="kw">for </span>Connection<T, P, B> |
| <span class="kw">where |
| </span>P: Peer, |
| B: Buf, |
| { |
| <span class="kw">fn </span>drop(<span class="kw-2">&mut </span><span class="self">self</span>) { |
| <span class="comment">// Ignore errors as this indicates that the mutex is poisoned. |
| </span><span class="kw">let _ </span>= <span class="self">self</span>.inner.streams.recv_eof(<span class="bool-val">true</span>); |
| } |
| } |
| </code></pre></div> |
| </section></div></main><div id="rustdoc-vars" data-root-path="../../../" data-current-crate="h2" data-themes="ayu,dark,light" data-resource-suffix="" data-rustdoc-version="1.66.0-nightly (5c8bff74b 2022-10-21)" ></div></body></html> |