memflowup 0.1.0-beta2

setup tool for the memflow physical memory introspection framework
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

use crate::util;

use std::fs;

use log::{error, info};

use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
pub struct Release {
    pub url: String,
    pub assets_url: String,
    pub upload_url: String,
    pub html_url: String,
    pub id: u64,
    // author
    pub node_id: String,
    pub tag_name: String,
    pub name: String,
    pub draft: bool,
    pub prerelease: bool,
    pub created_at: String,
    pub published_at: String,
    pub assets: Vec<Asset>,
    pub tarball_url: String,
    pub zipball_url: String,
    pub body: String,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct Asset {
    pub url: String,
    pub id: u64,
    pub node_id: String,
    pub name: String,
    pub label: String,
    // uploader
    pub content_type: String,
    pub state: String,
    pub size: u64,
    pub download_count: u64,
    pub created_at: String,
    pub updated_at: String,
    pub browser_download_url: String,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct Branch {
    pub name: String,
    pub commit: Commit,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct Commit {
    pub sha: String,
}

// TODO: filter for archicture
#[cfg(not(any(target_os = "windows", target_os = "macos")))]
fn find_platform_asset<'a>(release: &'a Release) -> Option<&'a Asset> {
    release.assets.iter().find(|a| a.name.contains("linux"))
}

#[cfg(target_os = "windows")]
fn find_platform_asset<'a>(release: &'a Release) -> Option<&'a Asset> {
    release.assets.iter().find(|a| a.name.contains("windows"))
}

#[cfg(target_os = "macos")]
fn find_platform_asset<'a>(release: &'a Release) -> Option<&'a Asset> {
    release.assets.iter().find(|a| a.name.contains("macos"))
}

/// Downloads a release from the specified repository.
pub fn download_repository_release_latest(group: &str, repository: &str) -> Result<String, String> {
    let releases: Vec<Release> = util::http_get_json(&format!(
        "https://api.github.com/repos/{}/{}/releases",
        group, repository
    ))?;

    match releases.iter().find(|r| !r.draft && !r.prerelease) {
        Some(release) => {
            info!(
                "latest stable release: {} (tag: {})",
                release.name, release.tag_name
            );
            match find_platform_asset(release) {
                Some(asset) => {
                    info!("valid binary found for current platform: {}", asset.name);
                    println!("yolo: {:?}", asset);
                    let out_file = util::tmp_file(&asset.name);
                    download_file(&asset.browser_download_url, &out_file)?;
                    Ok(out_file)
                },
                None => {
                    Err(format!("unable to find appropiate binary for the current platform for release {}/{}/{}", group, repository, release.tag_name).into())
                }
            }
        }
        None => Err(format!("unable to find a release for {}/{}", group, repository).into()),
    }
}

pub fn get_branch(url: &str, branch: &str) -> Result<Branch, &'static str> {
    let url = format!(
        "{}/branches/{}",
        url.replace("github.com", "api.github.com/repos"),
        branch
    );

    info!("Getting branch from {}", url);

    util::http_get_json(&url)
}

/// Downloads the given url to the destination file
fn download_file(url: &str, file: &str) -> Result<(), String> {
    info!("download file from '{}' to '{}'", url.clone(), file.clone());
    let bytes = match util::http_download_file(url) {
        Ok(b) => b,
        Err(err) => {
            error!("{}", err);
            return Err(err.into());
        }
    };

    match fs::write(file, bytes) {
        Ok(()) => Ok(().into()),
        Err(err) => {
            error!("{}", err);
            Err(err.to_string().into())
        }
    }
}

pub fn download_raw(repo: &str, branch: &str, path: &str) -> Result<Vec<u8>, &'static str> {
    let url = format!(
        "{}/{}/{}",
        repo.replace("github.com", "raw.githubusercontent.com"),
        branch,
        path
    );
    info!("download raw from '{}'", url);

    util::http_download_file(&url)
}