pub mod affected;
mod category;
mod date;
mod id;
mod informational;
mod keyword;
mod license;
pub mod linter;
mod metadata;
mod parts;
pub(crate) mod versions;
pub use self::{
affected::Affected,
category::Category,
date::Date,
id::{Id, IdKind},
informational::Informational,
keyword::Keyword,
license::License,
linter::Linter,
metadata::Metadata,
parts::Parts,
versions::Versions,
};
pub use cvss::Severity;
use crate::{
error::{Error, ErrorKind},
fs,
};
use serde::{Deserialize, Serialize};
use std::{path::Path, str::FromStr};
#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
pub struct Advisory {
#[serde(rename = "advisory")]
pub metadata: Metadata,
pub affected: Option<Affected>,
pub versions: Versions,
}
impl Advisory {
pub fn load_file(path: impl AsRef<Path>) -> Result<Self, Error> {
let path = path.as_ref();
let advisory_data = fs::read_to_string(path)
.map_err(|e| format_err!(ErrorKind::Io, "couldn't open {}: {}", path.display(), e))?;
advisory_data
.parse()
.map_err(|e| format_err!(ErrorKind::Parse, "error parsing {}: {}", path.display(), e))
}
pub fn id(&self) -> &Id {
&self.metadata.id
}
pub fn title(&self) -> &str {
self.metadata.title.as_ref()
}
pub fn description(&self) -> &str {
self.metadata.description.as_ref()
}
pub fn date(&self) -> &Date {
&self.metadata.date
}
pub fn severity(&self) -> Option<Severity> {
self.metadata.cvss.as_ref().map(|cvss| cvss.severity())
}
pub fn withdrawn(&self) -> bool {
self.metadata.withdrawn.is_some()
}
}
impl FromStr for Advisory {
type Err = Error;
fn from_str(advisory_data: &str) -> Result<Self, Error> {
let parts = Parts::parse(advisory_data)?;
let front_matter = if parts.front_matter.starts_with("[advisory]") {
parts.front_matter.to_owned()
} else {
String::from("[advisory]\n") + parts.front_matter
};
let mut advisory: Self = toml::from_str(&front_matter).map_err(Error::from_toml)?;
if !advisory.metadata.title.is_empty() {
fail!(
ErrorKind::Parse,
"invalid `title` attribute in advisory TOML"
);
}
if !advisory.metadata.description.is_empty() {
fail!(
ErrorKind::Parse,
"invalid `description` attribute in advisory TOML"
);
}
advisory.metadata.title = parts.title.to_owned();
advisory.metadata.description = parts.description.to_owned();
Ok(advisory)
}
}