projectmoon
/
rocket_cors
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
rocket_cors/src/rocket_cors/headers.rs.html

719 lines
55 KiB
HTML
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

<!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 to the Rust file `src/headers.rs`."><meta name="keywords" content="rust, rustlang, rust-lang"><title>headers.rs.html -- source</title><link rel="stylesheet" type="text/css" href="../../normalize.css"><link rel="stylesheet" type="text/css" href="../../rustdoc.css" id="mainThemeStyle"><link rel="stylesheet" type="text/css" href="../../dark.css"><link rel="stylesheet" type="text/css" href="../../light.css" id="themeStyle"><script src="../../storage.js"></script><noscript><link rel="stylesheet" href="../../noscript.css"></noscript><link rel="shortcut icon" href="../../favicon.ico"><style type="text/css">#crate-search{background-image:url("../../down-arrow.svg");}</style></head><body class="rustdoc source"><!--[if lte IE 8]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="sidebar-menu">&#9776;</div><a href='../../rocket_cors/index.html'><div class='logo-container'><img src='../../rust-logo.png' alt='logo'></div></a></nav><div class="theme-picker"><button id="theme-picker" aria-label="Pick another theme!"><img src="../../brush.svg" width="18" alt="Pick another theme!"></button><div id="theme-choices"></div></div><script src="../../theme.js"></script><nav class="sub"><form class="search-form js-only"><div class="search-container"><div><select id="crate-search"><option value="All crates">All crates</option></select><input class="search-input" name="search" autocomplete="off" spellcheck="false" placeholder="Click or press ‘S’ to search, ‘?’ for more options…" type="search"></div><a id="settings-menu" href="../../settings.html"><img src="../../wheel.svg" width="18" alt="Change settings"></a></div></form></nav><section id="main" class="content"><pre class="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>
</pre><div class="example-wrap"><pre class="rust ">
<span class="doccomment">//! CORS specific Request Headers</span>
<span class="kw">use</span> <span class="ident">std</span>::<span class="ident">collections</span>::<span class="ident">HashSet</span>;
<span class="kw">use</span> <span class="ident">std</span>::<span class="ident">fmt</span>;
<span class="kw">use</span> <span class="ident">std</span>::<span class="ident">ops</span>::<span class="ident">Deref</span>;
<span class="kw">use</span> <span class="ident">std</span>::<span class="ident">str</span>::<span class="ident">FromStr</span>;
<span class="kw">use</span> <span class="ident">rocket</span>::<span class="ident">http</span>::<span class="ident">Status</span>;
<span class="kw">use</span> <span class="ident">rocket</span>::<span class="ident">request</span>::{<span class="self">self</span>, <span class="ident">FromRequest</span>};
<span class="kw">use</span> <span class="ident">rocket</span>::{<span class="self">self</span>, <span class="ident">Outcome</span>};
<span class="attribute">#[<span class="ident">cfg</span>(<span class="ident">feature</span> <span class="op">=</span> <span class="string">&quot;serialization&quot;</span>)]</span>
<span class="kw">use</span> <span class="ident">serde_derive</span>::{<span class="ident">Deserialize</span>, <span class="ident">Serialize</span>};
<span class="kw">use</span> <span class="ident">unicase</span>::<span class="ident">UniCase</span>;
<span class="attribute">#[<span class="ident">cfg</span>(<span class="ident">feature</span> <span class="op">=</span> <span class="string">&quot;serialization&quot;</span>)]</span>
<span class="kw">use</span> <span class="ident">unicase_serde</span>;
<span class="doccomment">/// A case insensitive header name</span>
<span class="attribute">#[<span class="ident">derive</span>(<span class="ident">Eq</span>, <span class="ident">PartialEq</span>, <span class="ident">Clone</span>, <span class="ident">Debug</span>, <span class="ident">Hash</span>)]</span>
<span class="attribute">#[<span class="ident">cfg_attr</span>(<span class="ident">feature</span> <span class="op">=</span> <span class="string">&quot;serialization&quot;</span>, <span class="ident">derive</span>(<span class="ident">Serialize</span>, <span class="ident">Deserialize</span>))]</span>
<span class="kw">pub</span> <span class="kw">struct</span> <span class="ident">HeaderFieldName</span>(
<span class="attribute">#[<span class="ident">cfg_attr</span>(<span class="ident">feature</span> <span class="op">=</span> <span class="string">&quot;serialization&quot;</span>, <span class="ident">serde</span>(<span class="ident">with</span> <span class="op">=</span> <span class="string">&quot;unicase_serde::unicase&quot;</span>))]</span> <span class="ident">UniCase</span><span class="op">&lt;</span><span class="ident">String</span><span class="op">&gt;</span>,
);
<span class="kw">impl</span> <span class="ident">Deref</span> <span class="kw">for</span> <span class="ident">HeaderFieldName</span> {
<span class="kw">type</span> <span class="ident">Target</span> <span class="op">=</span> <span class="ident">String</span>;
<span class="kw">fn</span> <span class="ident">deref</span>(<span class="kw-2">&amp;</span><span class="self">self</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="kw-2">&amp;</span><span class="self">Self</span>::<span class="ident">Target</span> {
<span class="self">self</span>.<span class="number">0</span>.<span class="ident">deref</span>()
}
}
<span class="kw">impl</span> <span class="ident">fmt</span>::<span class="ident">Display</span> <span class="kw">for</span> <span class="ident">HeaderFieldName</span> {
<span class="kw">fn</span> <span class="ident">fmt</span>(<span class="kw-2">&amp;</span><span class="self">self</span>, <span class="ident">f</span>: <span class="kw-2">&amp;</span><span class="kw-2">mut</span> <span class="ident">fmt</span>::<span class="ident">Formatter</span><span class="op">&lt;</span><span class="lifetime">&#39;_</span><span class="op">&gt;</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">fmt</span>::<span class="prelude-ty">Result</span> {
<span class="self">self</span>.<span class="number">0</span>.<span class="ident">fmt</span>(<span class="ident">f</span>)
}
}
<span class="kw">impl</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span><span class="op">&gt;</span> <span class="ident">From</span><span class="op">&lt;</span><span class="kw-2">&amp;</span><span class="lifetime">&#39;a</span> <span class="ident">str</span><span class="op">&gt;</span> <span class="kw">for</span> <span class="ident">HeaderFieldName</span> {
<span class="kw">fn</span> <span class="ident">from</span>(<span class="ident">s</span>: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a</span> <span class="ident">str</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="self">Self</span> {
<span class="ident">HeaderFieldName</span>(<span class="ident">From</span>::<span class="ident">from</span>(<span class="ident">s</span>))
}
}
<span class="kw">impl</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span><span class="op">&gt;</span> <span class="ident">From</span><span class="op">&lt;</span><span class="ident">String</span><span class="op">&gt;</span> <span class="kw">for</span> <span class="ident">HeaderFieldName</span> {
<span class="kw">fn</span> <span class="ident">from</span>(<span class="ident">s</span>: <span class="ident">String</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="self">Self</span> {
<span class="ident">HeaderFieldName</span>(<span class="ident">From</span>::<span class="ident">from</span>(<span class="ident">s</span>))
}
}
<span class="kw">impl</span> <span class="ident">FromStr</span> <span class="kw">for</span> <span class="ident">HeaderFieldName</span> {
<span class="kw">type</span> <span class="prelude-val">Err</span> <span class="op">=</span> <span class="op">&lt;</span><span class="ident">String</span> <span class="kw">as</span> <span class="ident">FromStr</span><span class="op">&gt;</span>::<span class="prelude-val">Err</span>;
<span class="kw">fn</span> <span class="ident">from_str</span>(<span class="ident">s</span>: <span class="kw-2">&amp;</span><span class="ident">str</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="prelude-ty">Result</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="self">Self</span>::<span class="prelude-val">Err</span><span class="op">&gt;</span> {
<span class="prelude-val">Ok</span>(<span class="ident">HeaderFieldName</span>(<span class="ident">FromStr</span>::<span class="ident">from_str</span>(<span class="ident">s</span>)<span class="question-mark">?</span>))
}
}
<span class="doccomment">/// A set of case insensitive header names</span>
<span class="kw">pub</span> <span class="kw">type</span> <span class="ident">HeaderFieldNamesSet</span> <span class="op">=</span> <span class="ident">HashSet</span><span class="op">&lt;</span><span class="ident">HeaderFieldName</span><span class="op">&gt;</span>;
<span class="doccomment">/// The `Origin` request header used in CORS</span>
<span class="doccomment">///</span>
<span class="doccomment">/// You can use this as a rocket [Request Guard](https://rocket.rs/guide/requests/#request-guards)</span>
<span class="doccomment">/// to ensure that `Origin` is passed in correctly.</span>
<span class="doccomment">///</span>
<span class="doccomment">/// Reference: [Mozilla](https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Origin)</span>
<span class="attribute">#[<span class="ident">derive</span>(<span class="ident">Eq</span>, <span class="ident">PartialEq</span>, <span class="ident">Clone</span>, <span class="ident">Hash</span>, <span class="ident">Debug</span>)]</span>
<span class="kw">pub</span> <span class="kw">enum</span> <span class="ident">Origin</span> {
<span class="doccomment">/// A `null` Origin</span>
<span class="ident">Null</span>,
<span class="doccomment">/// A well-formed origin that was parsed by [`url::Url::origin`]</span>
<span class="ident">Parsed</span>(<span class="ident">url</span>::<span class="ident">Origin</span>),
<span class="doccomment">/// An unknown &quot;opaque&quot; origin that could not be parsed</span>
<span class="ident">Opaque</span>(<span class="ident">String</span>),
}
<span class="kw">impl</span> <span class="ident">Origin</span> {
<span class="doccomment">/// Perform an</span>
<span class="doccomment">/// [ASCII serialization](https://html.spec.whatwg.org/multipage/#ascii-serialisation-of-an-origin)</span>
<span class="doccomment">/// of this origin.</span>
<span class="kw">pub</span> <span class="kw">fn</span> <span class="ident">ascii_serialization</span>(<span class="kw-2">&amp;</span><span class="self">self</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">String</span> {
<span class="self">self</span>.<span class="ident">to_string</span>()
}
<span class="doccomment">/// Returns whether the origin was parsed as non-opaque</span>
<span class="kw">pub</span> <span class="kw">fn</span> <span class="ident">is_tuple</span>(<span class="kw-2">&amp;</span><span class="self">self</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">bool</span> {
<span class="kw">match</span> <span class="self">self</span> {
<span class="ident">Origin</span>::<span class="ident">Null</span> <span class="op">=</span><span class="op">&gt;</span> <span class="bool-val">false</span>,
<span class="ident">Origin</span>::<span class="ident">Parsed</span>(<span class="kw-2">ref</span> <span class="ident">parsed</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="ident">parsed</span>.<span class="ident">is_tuple</span>(),
<span class="ident">Origin</span>::<span class="ident">Opaque</span>(<span class="kw">_</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="bool-val">false</span>,
}
}
}
<span class="kw">impl</span> <span class="ident">FromStr</span> <span class="kw">for</span> <span class="ident">Origin</span> {
<span class="kw">type</span> <span class="prelude-val">Err</span> <span class="op">=</span> <span class="kw">crate</span>::<span class="ident">Error</span>;
<span class="kw">fn</span> <span class="ident">from_str</span>(<span class="ident">input</span>: <span class="kw-2">&amp;</span><span class="ident">str</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="prelude-ty">Result</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="self">Self</span>::<span class="prelude-val">Err</span><span class="op">&gt;</span> {
<span class="kw">if</span> <span class="ident">input</span>.<span class="ident">to_lowercase</span>() <span class="op">=</span><span class="op">=</span> <span class="string">&quot;null&quot;</span> {
<span class="prelude-val">Ok</span>(<span class="ident">Origin</span>::<span class="ident">Null</span>)
} <span class="kw">else</span> {
<span class="kw">match</span> <span class="kw">crate</span>::<span class="ident">to_origin</span>(<span class="ident">input</span>)<span class="question-mark">?</span> {
<span class="ident">url</span>::<span class="ident">Origin</span>::<span class="ident">Opaque</span>(<span class="kw">_</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="prelude-val">Ok</span>(<span class="ident">Origin</span>::<span class="ident">Opaque</span>(<span class="ident">input</span>.<span class="ident">to_string</span>())),
<span class="ident">parsed</span> @ <span class="ident">url</span>::<span class="ident">Origin</span>::<span class="ident">Tuple</span>(..) <span class="op">=</span><span class="op">&gt;</span> <span class="prelude-val">Ok</span>(<span class="ident">Origin</span>::<span class="ident">Parsed</span>(<span class="ident">parsed</span>)),
}
}
}
}
<span class="kw">impl</span> <span class="ident">fmt</span>::<span class="ident">Display</span> <span class="kw">for</span> <span class="ident">Origin</span> {
<span class="kw">fn</span> <span class="ident">fmt</span>(<span class="kw-2">&amp;</span><span class="self">self</span>, <span class="ident">f</span>: <span class="kw-2">&amp;</span><span class="kw-2">mut</span> <span class="ident">fmt</span>::<span class="ident">Formatter</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">fmt</span>::<span class="prelude-ty">Result</span> {
<span class="kw">match</span> <span class="self">self</span> {
<span class="ident">Origin</span>::<span class="ident">Null</span> <span class="op">=</span><span class="op">&gt;</span> <span class="macro">write</span><span class="macro">!</span>(<span class="ident">f</span>, <span class="string">&quot;null&quot;</span>),
<span class="ident">Origin</span>::<span class="ident">Parsed</span>(<span class="kw-2">ref</span> <span class="ident">parsed</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="macro">write</span><span class="macro">!</span>(<span class="ident">f</span>, <span class="string">&quot;{}&quot;</span>, <span class="ident">parsed</span>.<span class="ident">ascii_serialization</span>()),
<span class="ident">Origin</span>::<span class="ident">Opaque</span>(<span class="kw-2">ref</span> <span class="ident">opaque</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="macro">write</span><span class="macro">!</span>(<span class="ident">f</span>, <span class="string">&quot;{}&quot;</span>, <span class="ident">opaque</span>),
}
}
}
<span class="kw">impl</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span>, <span class="lifetime">&#39;r</span><span class="op">&gt;</span> <span class="ident">FromRequest</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span>, <span class="lifetime">&#39;r</span><span class="op">&gt;</span> <span class="kw">for</span> <span class="ident">Origin</span> {
<span class="kw">type</span> <span class="ident">Error</span> <span class="op">=</span> <span class="kw">crate</span>::<span class="ident">Error</span>;
<span class="kw">fn</span> <span class="ident">from_request</span>(<span class="ident">request</span>: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a</span> <span class="ident">rocket</span>::<span class="ident">Request</span><span class="op">&lt;</span><span class="lifetime">&#39;r</span><span class="op">&gt;</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">request</span>::<span class="ident">Outcome</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="kw">crate</span>::<span class="ident">Error</span><span class="op">&gt;</span> {
<span class="kw">match</span> <span class="ident">request</span>.<span class="ident">headers</span>().<span class="ident">get_one</span>(<span class="string">&quot;Origin&quot;</span>) {
<span class="prelude-val">Some</span>(<span class="ident">origin</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="kw">match</span> <span class="self">Self</span>::<span class="ident">from_str</span>(<span class="ident">origin</span>) {
<span class="prelude-val">Ok</span>(<span class="ident">origin</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Success</span>(<span class="ident">origin</span>),
<span class="prelude-val">Err</span>(<span class="ident">e</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Failure</span>((<span class="ident">Status</span>::<span class="ident">BadRequest</span>, <span class="ident">e</span>)),
},
<span class="prelude-val">None</span> <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Forward</span>(()),
}
}
}
<span class="doccomment">/// The `Access-Control-Request-Method` request header</span>
<span class="doccomment">///</span>
<span class="doccomment">/// You can use this as a rocket [Request Guard](https://rocket.rs/guide/requests/#request-guards)</span>
<span class="doccomment">/// to ensure that the header is passed in correctly.</span>
<span class="attribute">#[<span class="ident">derive</span>(<span class="ident">Debug</span>)]</span>
<span class="kw">pub</span> <span class="kw">struct</span> <span class="ident">AccessControlRequestMethod</span>(<span class="kw">pub</span> <span class="kw">crate</span>::<span class="ident">Method</span>);
<span class="kw">impl</span> <span class="ident">FromStr</span> <span class="kw">for</span> <span class="ident">AccessControlRequestMethod</span> {
<span class="kw">type</span> <span class="prelude-val">Err</span> <span class="op">=</span> ();
<span class="kw">fn</span> <span class="ident">from_str</span>(<span class="ident">method</span>: <span class="kw-2">&amp;</span><span class="ident">str</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="prelude-ty">Result</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="self">Self</span>::<span class="prelude-val">Err</span><span class="op">&gt;</span> {
<span class="prelude-val">Ok</span>(<span class="ident">AccessControlRequestMethod</span>(<span class="kw">crate</span>::<span class="ident">Method</span>::<span class="ident">from_str</span>(<span class="ident">method</span>)<span class="question-mark">?</span>))
}
}
<span class="kw">impl</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span>, <span class="lifetime">&#39;r</span><span class="op">&gt;</span> <span class="ident">FromRequest</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span>, <span class="lifetime">&#39;r</span><span class="op">&gt;</span> <span class="kw">for</span> <span class="ident">AccessControlRequestMethod</span> {
<span class="kw">type</span> <span class="ident">Error</span> <span class="op">=</span> <span class="kw">crate</span>::<span class="ident">Error</span>;
<span class="kw">fn</span> <span class="ident">from_request</span>(<span class="ident">request</span>: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a</span> <span class="ident">rocket</span>::<span class="ident">Request</span><span class="op">&lt;</span><span class="lifetime">&#39;r</span><span class="op">&gt;</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">request</span>::<span class="ident">Outcome</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="kw">crate</span>::<span class="ident">Error</span><span class="op">&gt;</span> {
<span class="kw">match</span> <span class="ident">request</span>.<span class="ident">headers</span>().<span class="ident">get_one</span>(<span class="string">&quot;Access-Control-Request-Method&quot;</span>) {
<span class="prelude-val">Some</span>(<span class="ident">request_method</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="kw">match</span> <span class="self">Self</span>::<span class="ident">from_str</span>(<span class="ident">request_method</span>) {
<span class="prelude-val">Ok</span>(<span class="ident">request_method</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Success</span>(<span class="ident">request_method</span>),
<span class="prelude-val">Err</span>(<span class="kw">_</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Failure</span>((<span class="ident">Status</span>::<span class="ident">BadRequest</span>, <span class="kw">crate</span>::<span class="ident">Error</span>::<span class="ident">BadRequestMethod</span>)),
},
<span class="prelude-val">None</span> <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Forward</span>(()),
}
}
}
<span class="doccomment">/// The `Access-Control-Request-Headers` request header</span>
<span class="doccomment">///</span>
<span class="doccomment">/// You can use this as a rocket [Request Guard](https://rocket.rs/guide/requests/#request-guards)</span>
<span class="doccomment">/// to ensure that the header is passed in correctly.</span>
<span class="attribute">#[<span class="ident">derive</span>(<span class="ident">Eq</span>, <span class="ident">PartialEq</span>, <span class="ident">Debug</span>)]</span>
<span class="kw">pub</span> <span class="kw">struct</span> <span class="ident">AccessControlRequestHeaders</span>(<span class="kw">pub</span> <span class="ident">HeaderFieldNamesSet</span>);
<span class="doccomment">/// Will never fail</span>
<span class="kw">impl</span> <span class="ident">FromStr</span> <span class="kw">for</span> <span class="ident">AccessControlRequestHeaders</span> {
<span class="kw">type</span> <span class="prelude-val">Err</span> <span class="op">=</span> ();
<span class="doccomment">/// Will never fail</span>
<span class="kw">fn</span> <span class="ident">from_str</span>(<span class="ident">headers</span>: <span class="kw-2">&amp;</span><span class="ident">str</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="prelude-ty">Result</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="self">Self</span>::<span class="prelude-val">Err</span><span class="op">&gt;</span> {
<span class="kw">if</span> <span class="ident">headers</span>.<span class="ident">trim</span>().<span class="ident">is_empty</span>() {
<span class="kw">return</span> <span class="prelude-val">Ok</span>(<span class="ident">AccessControlRequestHeaders</span>(<span class="ident">HashSet</span>::<span class="ident">new</span>()));
}
<span class="kw">let</span> <span class="ident">set</span>: <span class="ident">HeaderFieldNamesSet</span> <span class="op">=</span> <span class="ident">headers</span>
.<span class="ident">split</span>(<span class="string">&#39;,&#39;</span>)
.<span class="ident">map</span>(<span class="op">|</span><span class="ident">header</span><span class="op">|</span> <span class="ident">From</span>::<span class="ident">from</span>(<span class="ident">header</span>.<span class="ident">trim</span>().<span class="ident">to_string</span>()))
.<span class="ident">collect</span>();
<span class="prelude-val">Ok</span>(<span class="ident">AccessControlRequestHeaders</span>(<span class="ident">set</span>))
}
}
<span class="kw">impl</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span>, <span class="lifetime">&#39;r</span><span class="op">&gt;</span> <span class="ident">FromRequest</span><span class="op">&lt;</span><span class="lifetime">&#39;a</span>, <span class="lifetime">&#39;r</span><span class="op">&gt;</span> <span class="kw">for</span> <span class="ident">AccessControlRequestHeaders</span> {
<span class="kw">type</span> <span class="ident">Error</span> <span class="op">=</span> <span class="kw">crate</span>::<span class="ident">Error</span>;
<span class="kw">fn</span> <span class="ident">from_request</span>(<span class="ident">request</span>: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a</span> <span class="ident">rocket</span>::<span class="ident">Request</span><span class="op">&lt;</span><span class="lifetime">&#39;r</span><span class="op">&gt;</span>) <span class="op">-</span><span class="op">&gt;</span> <span class="ident">request</span>::<span class="ident">Outcome</span><span class="op">&lt;</span><span class="self">Self</span>, <span class="kw">crate</span>::<span class="ident">Error</span><span class="op">&gt;</span> {
<span class="kw">match</span> <span class="ident">request</span>.<span class="ident">headers</span>().<span class="ident">get_one</span>(<span class="string">&quot;Access-Control-Request-Headers&quot;</span>) {
<span class="prelude-val">Some</span>(<span class="ident">request_headers</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="kw">match</span> <span class="self">Self</span>::<span class="ident">from_str</span>(<span class="ident">request_headers</span>) {
<span class="prelude-val">Ok</span>(<span class="ident">request_headers</span>) <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Success</span>(<span class="ident">request_headers</span>),
<span class="prelude-val">Err</span>(()) <span class="op">=</span><span class="op">&gt;</span> {
<span class="macro">unreachable</span><span class="macro">!</span>(<span class="string">&quot;`AccessControlRequestHeaders::from_str` should never fail&quot;</span>)
}
},
<span class="prelude-val">None</span> <span class="op">=</span><span class="op">&gt;</span> <span class="ident">Outcome</span>::<span class="ident">Forward</span>(()),
}
}
}
<span class="attribute">#[<span class="ident">cfg</span>(<span class="ident">test</span>)]</span>
<span class="kw">mod</span> <span class="ident">tests</span> {
<span class="kw">use</span> <span class="ident">std</span>::<span class="ident">str</span>::<span class="ident">FromStr</span>;
<span class="kw">use</span> <span class="ident">hyper</span>;
<span class="kw">use</span> <span class="ident">rocket</span>;
<span class="kw">use</span> <span class="ident">rocket</span>::<span class="ident">local</span>::<span class="ident">Client</span>;
<span class="kw">use</span> <span class="kw">super</span>::<span class="kw-2">*</span>;
<span class="doccomment">/// Make a client with no routes for unit testing</span>
<span class="kw">fn</span> <span class="ident">make_client</span>() <span class="op">-</span><span class="op">&gt;</span> <span class="ident">Client</span> {
<span class="kw">let</span> <span class="ident">rocket</span> <span class="op">=</span> <span class="ident">rocket</span>::<span class="ident">ignite</span>();
<span class="ident">Client</span>::<span class="ident">new</span>(<span class="ident">rocket</span>).<span class="ident">expect</span>(<span class="string">&quot;valid rocket instance&quot;</span>)
}
<span class="comment">// `Origin::from_str` tests</span>
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">origin_is_parsed_properly</span>() {
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="ident">parsed</span>.<span class="ident">ascii_serialization</span>(), <span class="ident">url</span>);
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">origin_parsing_strips_paths</span>() {
<span class="comment">// this should never really be sent by a compliant user agent</span>
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz/path/somewhere&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
<span class="kw">let</span> <span class="ident">expected</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz&quot;</span>;
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="ident">parsed</span>.<span class="ident">ascii_serialization</span>(), <span class="ident">expected</span>);
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="attribute">#[<span class="ident">should_panic</span>(<span class="ident">expected</span> <span class="op">=</span> <span class="string">&quot;BadOrigin&quot;</span>)]</span>
<span class="kw">fn</span> <span class="ident">origin_parsing_disallows_invalid_origins</span>() {
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;invalid_url&quot;</span>;
<span class="kw">let</span> <span class="kw">_</span> <span class="op">=</span> <span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>).<span class="ident">unwrap</span>();
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">origin_parses_opaque_origins</span>() {
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;blob://foobar&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
<span class="macro">assert</span><span class="macro">!</span>(<span class="op">!</span><span class="ident">parsed</span>.<span class="ident">is_tuple</span>());
}
<span class="comment">// The following tests check that CORS Request headers are parsed correctly</span>
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">origin_header_conversion</span>() {
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="ident">parsed</span>.<span class="ident">ascii_serialization</span>(), <span class="ident">url</span>);
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz:1234&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="ident">parsed</span>.<span class="ident">ascii_serialization</span>(), <span class="ident">url</span>);
<span class="comment">// this should never really be sent by a compliant user agent</span>
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz/path/somewhere&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
<span class="kw">let</span> <span class="ident">expected</span> <span class="op">=</span> <span class="string">&quot;https://foo.bar.xyz&quot;</span>;
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="ident">parsed</span>.<span class="ident">ascii_serialization</span>(), <span class="ident">expected</span>);
<span class="kw">let</span> <span class="ident">url</span> <span class="op">=</span> <span class="string">&quot;invalid_url&quot;</span>;
<span class="kw">let</span> <span class="kw">_</span> <span class="op">=</span> <span class="macro">is_err</span><span class="macro">!</span>(<span class="ident">Origin</span>::<span class="ident">from_str</span>(<span class="ident">url</span>));
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">origin_header_parsing</span>() {
<span class="kw">let</span> <span class="ident">client</span> <span class="op">=</span> <span class="ident">make_client</span>();
<span class="kw">let</span> <span class="kw-2">mut</span> <span class="ident">request</span> <span class="op">=</span> <span class="ident">client</span>.<span class="ident">get</span>(<span class="string">&quot;/&quot;</span>);
<span class="kw">let</span> <span class="ident">origin</span> <span class="op">=</span> <span class="ident">hyper</span>::<span class="ident">header</span>::<span class="ident">Origin</span>::<span class="ident">new</span>(<span class="string">&quot;https&quot;</span>, <span class="string">&quot;www.example.com&quot;</span>, <span class="prelude-val">None</span>);
<span class="ident">request</span>.<span class="ident">add_header</span>(<span class="ident">origin</span>);
<span class="kw">let</span> <span class="ident">outcome</span>: <span class="ident">request</span>::<span class="ident">Outcome</span><span class="op">&lt;</span><span class="ident">Origin</span>, <span class="kw">crate</span>::<span class="ident">Error</span><span class="op">&gt;</span> <span class="op">=</span>
<span class="ident">FromRequest</span>::<span class="ident">from_request</span>(<span class="ident">request</span>.<span class="ident">inner</span>());
<span class="kw">let</span> <span class="ident">parsed_header</span> <span class="op">=</span> <span class="macro">assert_matches</span><span class="macro">!</span>(<span class="ident">outcome</span>, <span class="ident">Outcome</span>::<span class="ident">Success</span>(<span class="ident">s</span>), <span class="ident">s</span>);
<span class="macro">assert_eq</span><span class="macro">!</span>(
<span class="string">&quot;https://www.example.com&quot;</span>,
<span class="ident">parsed_header</span>.<span class="ident">ascii_serialization</span>()
);
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">request_method_conversion</span>() {
<span class="kw">let</span> <span class="ident">method</span> <span class="op">=</span> <span class="string">&quot;POST&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed_method</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">AccessControlRequestMethod</span>::<span class="ident">from_str</span>(<span class="ident">method</span>));
<span class="macro">assert_matches</span><span class="macro">!</span>(
<span class="ident">parsed_method</span>,
<span class="ident">AccessControlRequestMethod</span>(<span class="kw">crate</span>::<span class="ident">Method</span>(<span class="ident">rocket</span>::<span class="ident">http</span>::<span class="ident">Method</span>::<span class="ident">Post</span>))
);
<span class="kw">let</span> <span class="ident">method</span> <span class="op">=</span> <span class="string">&quot;options&quot;</span>;
<span class="kw">let</span> <span class="ident">parsed_method</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">AccessControlRequestMethod</span>::<span class="ident">from_str</span>(<span class="ident">method</span>));
<span class="macro">assert_matches</span><span class="macro">!</span>(
<span class="ident">parsed_method</span>,
<span class="ident">AccessControlRequestMethod</span>(<span class="kw">crate</span>::<span class="ident">Method</span>(<span class="ident">rocket</span>::<span class="ident">http</span>::<span class="ident">Method</span>::<span class="ident">Options</span>))
);
<span class="kw">let</span> <span class="ident">method</span> <span class="op">=</span> <span class="string">&quot;INVALID&quot;</span>;
<span class="macro">is_err</span><span class="macro">!</span>(<span class="ident">AccessControlRequestMethod</span>::<span class="ident">from_str</span>(<span class="ident">method</span>));
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">request_method_parsing</span>() {
<span class="kw">let</span> <span class="ident">client</span> <span class="op">=</span> <span class="ident">make_client</span>();
<span class="kw">let</span> <span class="kw-2">mut</span> <span class="ident">request</span> <span class="op">=</span> <span class="ident">client</span>.<span class="ident">get</span>(<span class="string">&quot;/&quot;</span>);
<span class="kw">let</span> <span class="ident">method</span> <span class="op">=</span> <span class="ident">hyper</span>::<span class="ident">header</span>::<span class="ident">AccessControlRequestMethod</span>(<span class="ident">hyper</span>::<span class="ident">method</span>::<span class="ident">Method</span>::<span class="ident">Get</span>);
<span class="ident">request</span>.<span class="ident">add_header</span>(<span class="ident">method</span>);
<span class="kw">let</span> <span class="ident">outcome</span>: <span class="ident">request</span>::<span class="ident">Outcome</span><span class="op">&lt;</span><span class="ident">AccessControlRequestMethod</span>, <span class="kw">crate</span>::<span class="ident">Error</span><span class="op">&gt;</span> <span class="op">=</span>
<span class="ident">FromRequest</span>::<span class="ident">from_request</span>(<span class="ident">request</span>.<span class="ident">inner</span>());
<span class="kw">let</span> <span class="ident">parsed_header</span> <span class="op">=</span> <span class="macro">assert_matches</span><span class="macro">!</span>(<span class="ident">outcome</span>, <span class="ident">Outcome</span>::<span class="ident">Success</span>(<span class="ident">s</span>), <span class="ident">s</span>);
<span class="kw">let</span> <span class="ident">AccessControlRequestMethod</span>(<span class="ident">parsed_method</span>) <span class="op">=</span> <span class="ident">parsed_header</span>;
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="string">&quot;GET&quot;</span>, <span class="ident">parsed_method</span>.<span class="ident">as_str</span>());
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">request_headers_conversion</span>() {
<span class="kw">let</span> <span class="ident">headers</span> <span class="op">=</span> [<span class="string">&quot;foo&quot;</span>, <span class="string">&quot;bar&quot;</span>, <span class="string">&quot;baz&quot;</span>];
<span class="kw">let</span> <span class="ident">parsed_headers</span> <span class="op">=</span> <span class="macro">not_err</span><span class="macro">!</span>(<span class="ident">AccessControlRequestHeaders</span>::<span class="ident">from_str</span>(<span class="kw-2">&amp;</span><span class="ident">headers</span>.<span class="ident">join</span>(<span class="string">&quot;, &quot;</span>)));
<span class="kw">let</span> <span class="ident">expected_headers</span>: <span class="ident">HeaderFieldNamesSet</span> <span class="op">=</span>
<span class="ident">headers</span>.<span class="ident">iter</span>().<span class="ident">map</span>(<span class="op">|</span><span class="ident">s</span><span class="op">|</span> (<span class="kw-2">*</span><span class="ident">s</span>).<span class="ident">to_string</span>().<span class="ident">into</span>()).<span class="ident">collect</span>();
<span class="kw">let</span> <span class="ident">AccessControlRequestHeaders</span>(<span class="ident">actual_headers</span>) <span class="op">=</span> <span class="ident">parsed_headers</span>;
<span class="macro">assert_eq</span><span class="macro">!</span>(<span class="ident">actual_headers</span>, <span class="ident">expected_headers</span>);
}
<span class="attribute">#[<span class="ident">test</span>]</span>
<span class="kw">fn</span> <span class="ident">request_headers_parsing</span>() {
<span class="kw">let</span> <span class="ident">client</span> <span class="op">=</span> <span class="ident">make_client</span>();
<span class="kw">let</span> <span class="kw-2">mut</span> <span class="ident">request</span> <span class="op">=</span> <span class="ident">client</span>.<span class="ident">get</span>(<span class="string">&quot;/&quot;</span>);
<span class="kw">let</span> <span class="ident">headers</span> <span class="op">=</span> <span class="ident">hyper</span>::<span class="ident">header</span>::<span class="ident">AccessControlRequestHeaders</span>(<span class="macro">vec</span><span class="macro">!</span>[
<span class="ident">FromStr</span>::<span class="ident">from_str</span>(<span class="string">&quot;accept-language&quot;</span>).<span class="ident">unwrap</span>(),
<span class="ident">FromStr</span>::<span class="ident">from_str</span>(<span class="string">&quot;date&quot;</span>).<span class="ident">unwrap</span>(),
]);
<span class="ident">request</span>.<span class="ident">add_header</span>(<span class="ident">headers</span>);
<span class="kw">let</span> <span class="ident">outcome</span>: <span class="ident">request</span>::<span class="ident">Outcome</span><span class="op">&lt;</span><span class="ident">AccessControlRequestHeaders</span>, <span class="kw">crate</span>::<span class="ident">Error</span><span class="op">&gt;</span> <span class="op">=</span>
<span class="ident">FromRequest</span>::<span class="ident">from_request</span>(<span class="ident">request</span>.<span class="ident">inner</span>());
<span class="kw">let</span> <span class="ident">parsed_header</span> <span class="op">=</span> <span class="macro">assert_matches</span><span class="macro">!</span>(<span class="ident">outcome</span>, <span class="ident">Outcome</span>::<span class="ident">Success</span>(<span class="ident">s</span>), <span class="ident">s</span>);
<span class="kw">let</span> <span class="ident">AccessControlRequestHeaders</span>(<span class="ident">parsed_headers</span>) <span class="op">=</span> <span class="ident">parsed_header</span>;
<span class="kw">let</span> <span class="kw-2">mut</span> <span class="ident">parsed_headers</span>: <span class="ident">Vec</span><span class="op">&lt;</span><span class="ident">String</span><span class="op">&gt;</span> <span class="op">=</span>
<span class="ident">parsed_headers</span>.<span class="ident">iter</span>().<span class="ident">map</span>(<span class="ident">ToString</span>::<span class="ident">to_string</span>).<span class="ident">collect</span>();
<span class="ident">parsed_headers</span>.<span class="ident">sort</span>();
<span class="macro">assert_eq</span><span class="macro">!</span>(
<span class="macro">vec</span><span class="macro">!</span>[<span class="string">&quot;accept-language&quot;</span>.<span class="ident">to_string</span>(), <span class="string">&quot;date&quot;</span>.<span class="ident">to_string</span>()],
<span class="ident">parsed_headers</span>
);
}
}
</pre></div>
</section><section id="search" class="content hidden"></section><section class="footer"></section><script>window.rootPath = "../../";window.currentCrate = "rocket_cors";</script><script src="../../aliases.js"></script><script src="../../main.js"></script><script src="../../source-script.js"></script><script src="../../source-files.js"></script><script defer src="../../search-index.js"></script></body></html>
Reference in New Issue View Git Blame Copy Permalink