http_header 0.7.0

A simple, dependency-less HTTP-header builder and parser
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use std::io::{ self, Read, Write };


/// An extension of the `Write` trait
pub trait ReadExt {
	/// Reads until either `pat` is matched or `buf` is filled completely
	///
	/// Returns either `Some(bytes_read)` if the pattern has been matched or `None` if `buf` has
	/// been filled completely without matching the pattern.
	fn read_until(&mut self, buf: &mut[u8], pat: impl AsRef<[u8]>)
		-> Result<Option<usize>, io::Error>;
}
impl<T: Read> ReadExt for T {
	fn read_until(&mut self, buf: &mut[u8], pat: impl AsRef<[u8]>)
		-> Result<Option<usize>, io::Error>
	{
		let pat = pat.as_ref();

		// Read the input byte-per-byte and check for the pattern
		let mut pos = 0;
		while pos < buf.len() {
			// Read the next byte and adjust `pos`
			match self.read_exact(&mut buf[pos .. pos + 1]) {
				Ok(_) => pos += 1,
				Err(e) => match e.kind() {
					io::ErrorKind::UnexpectedEof => return Ok(None),
					e => Err(e)?
				}
			}
			
			// Check for pattern
			if pos >= pat.len() && &buf[pos - pat.len() .. pos] == pat {
				return Ok(Some(pos))
			}
		}
		Ok(None)
	}
}


/// An extension of the `Write` trait
pub trait WriteExt {
	/// Writes all of `data`
	fn write(&mut self, data: impl AsRef<[u8]>) -> Result<&mut Self, io::Error>;
}
impl<T: Write> WriteExt for T {
	fn write(&mut self, data: impl AsRef<[u8]>) -> Result<&mut Self, io::Error> {
		self.write_all(data.as_ref())?;
		Ok(self)
	}
}