linkstore 1.0.2

Library for embedding, manipulating and retrieving data embedded in binaries using linker sections
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
52
53
54
55
56
57
58
59
60

use std::{
	fs::File,
	io::{Cursor, Read, Seek, SeekFrom, Write},
};

use crate::Error;

#[doc(hidden)]
pub trait BinaryHandle<'a>: Read + Write + Seek {
	fn get_memory_ref(&self) -> Option<&[u8]>;
	fn get_memory(&mut self) -> Result<Option<Vec<u8>>, Error>;
	fn size_hint(&mut self) -> Option<u64>;
}
impl<'a> BinaryHandle<'a> for Cursor<&'a mut [u8]> {
	#[inline]
	fn get_memory_ref(&self) -> Option<&[u8]> {
		Some(&**self.get_ref())
	}

	#[inline]
	fn get_memory(&mut self) -> Result<Option<Vec<u8>>, Error> {
		Ok(None)
	}

	#[inline]
	fn size_hint(&mut self) -> Option<u64> {
		Some(self.get_ref().len() as _)
	}
}
impl<'a> BinaryHandle<'a> for File {
	#[inline]
	fn get_memory_ref(&self) -> Option<&[u8]> {
		None
	}

	fn get_memory(&mut self) -> Result<Option<Vec<u8>>, Error> {
		let pos = self.stream_position()?;
		self.seek(SeekFrom::Start(0))?;

		let mut buf = if let Some(len) = self.size_hint() {
			Vec::with_capacity(len as usize)
		} else {
			Vec::new()
		};

		self.read_to_end(&mut buf)?;
		self.seek(SeekFrom::Start(pos))?;

		Ok(Some(buf))
	}

	fn size_hint(&mut self) -> Option<u64> {
		self.metadata().ok().map(|metadata| metadata.len()).or_else(|| {
			let current_pos = self.stream_position().ok()?;
			let len = self.seek(SeekFrom::End(0)).ok()?;
			self.seek(SeekFrom::Start(current_pos)).ok()?;
			Some(len)
		})
	}
}