nvd_cve 0.1.0

Search for CVEs against a local cached copy of NIST National Vulnerability Database (NVD).
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use clap::ArgMatches;
use nvd_cve::cache::{search_by_id, CacheConfig};
use nvd_cve::cache::{search_description, sync_blocking};
use nvd_cve::client::{BlockingHttpClient, ReqwestBlockingClient};

pub fn sync(matches: &ArgMatches) {
    let mut config = CacheConfig::new();

    if matches.is_present("show") {
        println!("Default Config Values:\n{}", config);
        return;
    }

    if let Some(url) = matches.value_of("url") {
        config.url = String::from(url);
    }

    if let Some(feeds) = matches.value_of("feeds") {
        config.feeds = feeds.split(',').map(|feed| feed.to_string()).collect();
    }

    if let Some(db) = matches.value_of("db") {
        config.db = String::from(db);
    }

    if matches.is_present("no_progress") {
        config.show_progress = false;
    }

    if matches.is_present("force") {
        config.force_update = true;
    }

    if matches.is_present("verbose") {
        env_logger::init();
    }

    let client = ReqwestBlockingClient::new(&config.url, None, None, None);

    if let Err(error) = sync_blocking(&config, client) {
        eprintln!("Fatal Error: {:?}", error);
        std::process::exit(1);
    }
}

pub fn search(matches: &ArgMatches) {
    let mut config = CacheConfig::new();

    if let Some(db) = matches.value_of("db") {
        config.db = String::from(db);
    }

    if let Some(text) = matches.value_of("text") {
        match search_description(&config, text) {
            Ok(cves) => {
                if cves.is_empty() {
                    eprintln!("No results found");
                    std::process::exit(1);
                } else {
                    for cve in cves {
                        println!("{}", cve);
                    }
                }
            }
            Err(error) => {
                eprintln!("Fatal Error: {:?}", error);
                std::process::exit(2);
            }
        }
    } else if let Some(cve) = matches.value_of("CVE") {
        match search_by_id(&config, cve) {
            Ok(cve_result) => println!("{}", serde_json::to_string_pretty(&cve_result).unwrap()),
            Err(error) => {
                eprintln!("Fatal Error: {:?}", error);
                std::process::exit(3);
            }
        }
    }
}