virustotal3 3.0.3

The library for VirusTotal Public API 3.0
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
80
81
82

use super::{FileRescanResponse, FileScanResponse, VtClient};
use anyhow::{Context, Result};
use reqwest::Client;
use serde_json::{from_str, Value};

impl<'a> VtClient<'a> {
    /// Upload and scan a file
    /// <https://developers.virustotal.com/reference/files-scan>
    ///
    /// # Example
    ///
    /// ```
    /// use virustotal3::VtClient;
    ///
    /// // Not async function to send file
    /// let vt = VtClient::new("Your API Key");
    /// vt.scan_file("eicar.txt")?;
    /// ```
    pub fn scan_file(self, filename: &str) -> Result<FileScanResponse> {
        let form = reqwest::blocking::multipart::Form::new()
            .file("file", filename)
            .context("File not found")?;

        let url = format!("{}/files", self.endpoint);
        let resp = reqwest::blocking::Client::new()
            .post(url)
            .header("x-apikey", self.api_key)
            .header("accept", "application/json")
            .header("content-type", "multipart/form-data")
            .multipart(form)
            .send()
            .context("Error! Probably maximum request limit achieved!")?;
        let text: &str = &resp.text()?;
        Ok(from_str(text)?)
    }

    /// Retrieve file scan reports
    /// <https://developers.virustotal.com/reference/file-info>
    ///
    /// # Example
    ///
    /// ```
    /// use virustotal3::VtClient;
    ///
    /// let vt = VtClient::new("Your API Key");
    /// vt.get_report_file("SHA-256, SHA-1 or MD5 identifying the file").await?;
    /// ```
    pub async fn get_report_file(self, resource: &str) -> Result<Value> {
        let url = &[self.endpoint, "/files/", resource].join("");
        let resp = Client::new()
            .get(url)
            .header("x-apikey", self.api_key)
            .send()
            .await
            .context("Error! Probably maximum request limit achieved!")?;
        let text: &str = &resp.text().await?;
        Ok(from_str(text)?)
    }

    /// Rescanning already submitted files
    /// <https://developers.virustotal.com/reference/files-analyse>
    ///
    /// # Example
    ///
    /// ```
    /// use virustotal3::VtClient;
    ///
    /// let vt = VtClient::new("Your API Key");
    /// vt.rescan_file("SHA-256, SHA-1 or MD5 identifying the file").await?;
    /// ```
    pub async fn rescan_file(self, resource: &str) -> Result<FileRescanResponse> {
        let url = format!("{}/files/{resource}/analyse", self.endpoint);
        let resp = Client::new()
            .post(&url)
            .header("x-apikey", self.api_key)
            .send()
            .await
            .context("Error! Probably maximum request limit achieved!")?;
        let text: &str = &resp.text().await?;
        Ok(from_str(text)?)
    }
}