pub mod affected;
pub mod category;
pub mod date;
pub mod id;
pub mod informational;
pub mod keyword;
pub mod linter;
pub mod metadata;
pub mod parser;
pub mod versions;
pub use self::{
affected::Affected, category::Category, date::Date, id::Id, informational::Informational,
keyword::Keyword, linter::Linter, metadata::Metadata, 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())
}
}
impl FromStr for Advisory {
type Err = Error;
fn from_str(advisory_data: &str) -> Result<Self, Error> {
let parts = parser::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)?;
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)
}
}