AleoHQ/leo
1
1
Fork 0
mirror of https://github.com/AleoHQ/leo.git synced 2024-11-29 03:35:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

Restrict included package files

Browse Source
This commit is contained in:
raychu86 2020-08-20 00:20:39 -07:00
parent fce7435eae
commit 7110b5b67b
2 changed files with 39 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
package/src/root/zip.rs
View File

@ -28,6 +28,8 @@ use crate::{
PROVING_KEY_FILE_EXTENSION, PROVING_KEY_FILE_EXTENSION,
VERIFICATION_KEY_FILE_EXTENSION, VERIFICATION_KEY_FILE_EXTENSION,
}, },
root::{MANIFEST_FILE_NAME, README_FILE_NAME},
source::{SOURCE_DIRECTORY_NAME, SOURCE_FILE_EXTENSION},
}; };
use serde::Deserialize; use serde::Deserialize;
@ -89,12 +91,15 @@ impl ZipFile {
let path = entry.path(); let path = entry.path();
let name = path.strip_prefix(src_dir.as_path()).unwrap(); let name = path.strip_prefix(src_dir.as_path()).unwrap();
// filter excluded paths // Add file/directory exclusion
if is_excluded(name) {
let included = is_included(name);
log::debug!("Checking if {:?} is included - {}", name, included);
if !included {
continue; continue;
} }
// write file or directory // Write file or directory
if path.is_file() { if path.is_file() {
log::info!("Adding file {:?} as {:?}", path, name); log::info!("Adding file {:?} as {:?}", path, name);
zip.start_file_from_path(name, options)?; zip.start_file_from_path(name, options)?;
@ -142,28 +147,43 @@ impl ZipFile {
} }
} }
fn is_excluded(path: &Path) -> bool { /// Check if the file path should be included in the package zip file.
log::debug!("Checking if {:?} is excluded", path); fn is_included(path: &Path) -> bool {
// excluded directories: `input`, `output`, `imports` // excluded directories: `input`, `output`, `imports`
if path.ends_with(INPUTS_DIRECTORY_NAME.trim_end_matches("/")) if path.ends_with(INPUTS_DIRECTORY_NAME.trim_end_matches("/"))
| path.ends_with(OUTPUTS_DIRECTORY_NAME.trim_end_matches("/")) | path.ends_with(OUTPUTS_DIRECTORY_NAME.trim_end_matches("/"))
| path.ends_with(IMPORTS_DIRECTORY_NAME.trim_end_matches("/")) | path.ends_with(IMPORTS_DIRECTORY_NAME.trim_end_matches("/"))
{ {
return true; return false;
} }
// excluded extensions: `.in`, `.bytes`, `lpk`, `lvk`, `.proof`, `.sum`, `.zip`, `.bytes` // excluded extensions: `.in`, `.bytes`, `lpk`, `lvk`, `.proof`, `.sum`, `.zip`, `.bytes`
if let Some(true) = path.extension().map(|ext| {
ext.eq(INPUT_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(ZIP_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(PROVING_KEY_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(VERIFICATION_KEY_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(PROOF_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(CHECKSUM_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(ZIP_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(CIRCUIT_FILE_EXTENSION.trim_start_matches("."))
}) {
return false;
}
// Allow the README.md and Leo.toml files in the root directory
if (path.ends_with(README_FILE_NAME) | path.ends_with(MANIFEST_FILE_NAME)) & (path.parent() == Some(Path::new("")))
{
return true;
}
// Only allow additional files in the `src/` directory
if !path.starts_with(SOURCE_DIRECTORY_NAME.trim_end_matches("/")) {
return false;
}
// Allow the `.leo` files in the `src/` directory
path.extension() path.extension()
.map(|ext| { .map(|ext| ext.eq(SOURCE_FILE_EXTENSION.trim_start_matches(".")))
ext.eq(INPUT_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(ZIP_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(PROVING_KEY_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(VERIFICATION_KEY_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(PROOF_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(CHECKSUM_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(ZIP_FILE_EXTENSION.trim_start_matches("."))
| ext.eq(CIRCUIT_FILE_EXTENSION.trim_start_matches("."))
})
.unwrap_or(false) .unwrap_or(false)
} }

4
package/src/source/directory.rs
View File

@ -20,7 +20,7 @@ use std::{fs, path::PathBuf};
pub static SOURCE_DIRECTORY_NAME: &str = "src/"; pub static SOURCE_DIRECTORY_NAME: &str = "src/";
static SOURCE_FILE_EXTENSION: &str = "leo"; pub static SOURCE_FILE_EXTENSION: &str = ".leo";
pub struct SourceDirectory; pub struct SourceDirectory;
@ -61,7 +61,7 @@ impl SourceDirectory {
let file_extension = file_path let file_extension = file_path
.extension() .extension()
.ok_or_else(|| SourceDirectoryError::GettingFileExtension(file_path.as_os_str().to_owned()))?; .ok_or_else(|| SourceDirectoryError::GettingFileExtension(file_path.as_os_str().to_owned()))?;
if file_extension != SOURCE_FILE_EXTENSION { if file_extension != SOURCE_FILE_EXTENSION.trim_start_matches(".") {
return Err(SourceDirectoryError::InvalidFileExtension( return Err(SourceDirectoryError::InvalidFileExtension(
file_path.as_os_str().to_owned(), file_path.as_os_str().to_owned(),
file_extension.to_owned(), file_extension.to_owned(),