126 lines
3.8 KiB
Rust
126 lines
3.8 KiB
Rust
use crate::Error;
|
|
use byteorder::{LittleEndian, ReadBytesExt};
|
|
use std::convert::TryFrom;
|
|
use std::io::Read;
|
|
|
|
pub fn parse<R: Read>(input: &mut R) -> Result<(), Error> {
|
|
if String::read(input)? != NAR_VERSION_MAGIC {
|
|
return Err(Error::BadNarVersionMagic);
|
|
}
|
|
|
|
parse_file(input)
|
|
}
|
|
|
|
const NAR_VERSION_MAGIC: &str = "nix-archive-1";
|
|
|
|
fn parse_file<R: Read>(input: &mut R) -> Result<(), Error> {
|
|
if String::read(input)? != "(" {
|
|
return Err(Error::MissingNarOpenTag);
|
|
}
|
|
|
|
if String::read(input)? != "type" {
|
|
return Err(Error::MissingNarField);
|
|
}
|
|
|
|
match String::read(input)?.as_ref() {
|
|
"regular" => {
|
|
let mut _executable = false;
|
|
let mut tag = String::read(input)?;
|
|
if tag == "executable" {
|
|
_executable = true;
|
|
if String::read(input)? != "" {
|
|
return Err(Error::BadExecutableField);
|
|
}
|
|
tag = String::read(input)?;
|
|
}
|
|
if tag != "contents" {
|
|
return Err(Error::MissingNarField);
|
|
}
|
|
let _contents = Vec::<u8>::read(input)?;
|
|
if String::read(input)? != ")" {
|
|
return Err(Error::MissingNarCloseTag);
|
|
}
|
|
}
|
|
"directory" => loop {
|
|
match String::read(input)?.as_ref() {
|
|
"entry" => {
|
|
if String::read(input)? != "(" {
|
|
return Err(Error::MissingNarOpenTag);
|
|
}
|
|
if String::read(input)? != "name" {
|
|
return Err(Error::MissingNarField);
|
|
}
|
|
let _name = String::read(input)?;
|
|
if String::read(input)? != "node" {
|
|
return Err(Error::MissingNarField);
|
|
}
|
|
parse_file(input)?;
|
|
let tag = String::read(input)?;
|
|
if tag != ")" {
|
|
return Err(Error::MissingNarCloseTag);
|
|
}
|
|
}
|
|
")" => break,
|
|
s => return Err(Error::BadNarField(s.into())),
|
|
}
|
|
},
|
|
"symlink" => {
|
|
if String::read(input)? != "target" {
|
|
return Err(Error::MissingNarField);
|
|
}
|
|
let _target = String::read(input)?;
|
|
if String::read(input)? != ")" {
|
|
return Err(Error::MissingNarCloseTag);
|
|
}
|
|
}
|
|
s => return Err(Error::BadNarField(s.into())),
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
trait Deserialize: Sized {
|
|
fn read<R: Read>(input: &mut R) -> Result<Self, Error>;
|
|
}
|
|
|
|
impl Deserialize for String {
|
|
fn read<R: Read>(input: &mut R) -> Result<Self, Error> {
|
|
let buf = Deserialize::read(input)?;
|
|
Ok(String::from_utf8(buf).map_err(|_| Error::BadNarString)?)
|
|
}
|
|
}
|
|
|
|
impl Deserialize for Vec<u8> {
|
|
fn read<R: Read>(input: &mut R) -> Result<Self, Error> {
|
|
let n: usize = Deserialize::read(input)?;
|
|
let mut buf = vec![0; n];
|
|
input.read_exact(&mut buf)?;
|
|
skip_padding(input, n)?;
|
|
Ok(buf)
|
|
}
|
|
}
|
|
|
|
fn skip_padding<R: Read>(input: &mut R, len: usize) -> Result<(), Error> {
|
|
if len % 8 != 0 {
|
|
let mut buf = [0; 8];
|
|
let buf = &mut buf[0..8 - (len % 8)];
|
|
input.read_exact(buf)?;
|
|
if !buf.iter().all(|b| *b == 0) {
|
|
return Err(Error::BadNarPadding);
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
impl Deserialize for u64 {
|
|
fn read<R: Read>(input: &mut R) -> Result<Self, Error> {
|
|
Ok(input.read_u64::<LittleEndian>()?)
|
|
}
|
|
}
|
|
|
|
impl Deserialize for usize {
|
|
fn read<R: Read>(input: &mut R) -> Result<Self, Error> {
|
|
let n: u64 = Deserialize::read(input)?;
|
|
Ok(usize::try_from(n).map_err(|_| Error::NarSizeFieldTooBig)?)
|
|
}
|
|
}
|