Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
65 changes: 63 additions & 2 deletions src/config.rs
Original file line number Diff line number Diff line change
Expand Up @@ -230,7 +230,7 @@ impl ProjectPathsConfig {
if component == Component::CurDir || component == Component::ParentDir {
// if the import is relative we assume it's already part of the processed input
// file set
utils::canonicalize(cwd.join(import)).map_err(|err| {
utils::normalize_solidity_import_path(cwd, import).map_err(|err| {
SolcError::msg(format!("failed to resolve relative import \"{err:?}\""))
})
} else {
Expand All @@ -253,7 +253,7 @@ impl ProjectPathsConfig {
// also try to resolve absolute imports from the project paths
for path in [&self.root, &self.sources, &self.tests, &self.scripts] {
if cwd.starts_with(path) {
if let Ok(import) = utils::canonicalize(path.join(import)) {
if let Ok(import) = utils::normalize_solidity_import_path(path, import) {
return Ok(import);
}
}
Expand Down Expand Up @@ -1026,4 +1026,65 @@ mod tests {
Path::new("/root/test/")
);
}

#[test]
fn can_resolve_import() {
let dir = tempfile::tempdir().unwrap();
let config = ProjectPathsConfig::builder().root(dir.path()).build().unwrap();
config.create_all().unwrap();

fs::write(config.sources.join("A.sol"), r"pragma solidity ^0.8.0; contract A {}").unwrap();

// relative import
assert_eq!(
config
.resolve_import_and_include_paths(
&config.sources,
Path::new("./A.sol"),
&mut Default::default(),
)
.unwrap(),
config.sources.join("A.sol")
);

// direct import
assert_eq!(
config
.resolve_import_and_include_paths(
&config.sources,
Path::new("src/A.sol"),
&mut Default::default(),
)
.unwrap(),
config.sources.join("A.sol")
);
}

#[test]
fn can_resolve_remapped_import() {
let dir = tempfile::tempdir().unwrap();
let mut config = ProjectPathsConfig::builder().root(dir.path()).build().unwrap();
config.create_all().unwrap();

let dependency = config.root.join("dependency");
fs::create_dir(&dependency).unwrap();
fs::write(dependency.join("A.sol"), r"pragma solidity ^0.8.0; contract A {}").unwrap();

config.remappings.push(Remapping {
context: None,
name: "@dependency/".into(),
path: "dependency/".into(),
});

assert_eq!(
config
.resolve_import_and_include_paths(
&config.sources,
Path::new("@dependency/A.sol"),
&mut Default::default(),
)
.unwrap(),
dependency.join("A.sol")
);
}
}
142 changes: 140 additions & 2 deletions src/utils.rs
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,52 @@ pub fn canonicalize(path: impl AsRef<Path>) -> Result<PathBuf, SolcIoError> {
res.map_err(|err| SolcIoError::new(err, path))
}

/// Returns a normalized Solidity file path for the given import path based on the specified
/// directory.
///
/// This function resolves `./` and `../`, but, unlike [`canonicalize`], it does not resolve
/// symbolic links.
///
/// The function returns an error if the normalized path does not exist in the file system.
///
/// See also: <https://docs.soliditylang.org/en/v0.8.23/path-resolution.html>
pub fn normalize_solidity_import_path(
directory: impl AsRef<Path>,
import_path: impl AsRef<Path>,
) -> Result<PathBuf, SolcIoError> {
let original = directory.as_ref().join(import_path);
let cleaned = clean_solidity_path(&original);

// this is to align the behavior with `canonicalize`
use path_slash::PathExt;
let normalized = PathBuf::from(dunce::simplified(&cleaned).to_slash_lossy().as_ref());

fs::metadata(&normalized).map(|_| normalized).map_err(|err| SolcIoError::new(err, original))
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

what's this call for?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The function normalize_solidity_import_path checks if the returned path exists in the file system. The use of fs::metadata(&normalized) is to obtain an io::Error for non-existing files without reading their entire contents. To return the same type of error as utils::canonicalize, an io::Error is necessary for SolcIoError.

While there is a specific function, Path::is_file(), that confirms if a path exists in the file system, it only returns a bool.

https://doc.rust-lang.org/std/path/struct.Path.html#method.is_file

This is a convenience function that coerces errors to false. If you want to check errors, call fs::metadata and handle its Result.

However, I have now realized that there's an alias Path::metadata() for paths. Therefore, I will switch to using this function and add a comment.

https://doc.rust-lang.org/std/path/struct.Path.html#method.metadata

}

fn clean_solidity_path(original_path: impl AsRef<Path>) -> PathBuf {
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Rust doesn't have this normalization function (resolves . and .. but doesn't resolve symlinks), so I wrote the function myself. The function is simple, but I found several crates that perform similar tasks, so I can switch to one of them if preferred.
e.g.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

gotcha—I'll defer to @mattsse here, but the function is so small I don't mind keeping it here instead of pulling in an additional dep, as we also have tests.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could we add some docs here? I know it's an internal function, but having docs is nice and useful to revisit this later whenever needed

let mut new_path = Vec::new();

for component in original_path.as_ref().components() {
match component {
Component::Prefix(..) | Component::RootDir | Component::Normal(..) => {
new_path.push(component);
}
Component::CurDir => {}
Component::ParentDir => match new_path.last() {
Some(Component::Normal(..)) => {
new_path.pop();
}
_ => {
new_path.push(component);
}
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we can use if let Some(Component::Normal) = new_path.last() {pop} else {...} here

which is a bit nicer imo

},
}
}

new_path.iter().collect()
}

/// Returns the same path config but with canonicalized paths.
///
/// This will take care of potential symbolic linked directories.
Expand Down Expand Up @@ -228,7 +274,7 @@ pub fn resolve_library(libs: &[impl AsRef<Path>], source: impl AsRef<Path>) -> O
/// until the `root` is reached.
///
/// If an existing file under `root` is found, this returns the path up to the `import` path and the
/// canonicalized `import` path itself:
/// normalized `import` path itself:
///
/// For example for following layout:
///
Expand All @@ -247,7 +293,7 @@ pub fn resolve_absolute_library(
) -> Option<(PathBuf, PathBuf)> {
let mut parent = cwd.parent()?;
while parent != root {
if let Ok(import) = canonicalize(parent.join(import)) {
if let Ok(import) = normalize_solidity_import_path(parent, import) {
return Some((parent.to_path_buf(), import));
}
parent = parent.parent()?;
Expand Down Expand Up @@ -654,6 +700,98 @@ pragma solidity ^0.8.0;
assert_eq!(Some("^0.8.0"), find_version_pragma(s).map(|s| s.as_str()));
}

#[test]
fn can_normalize_solidity_import_path() {
let dir = tempfile::tempdir().unwrap();
let dir_path = dir.path();

// File structure:
//
// `dir_path`
// └── src (`cwd`)
// ├── Token.sol
// └── common
// └── Burnable.sol

fs::create_dir_all(dir_path.join("src/common")).unwrap();
fs::write(dir_path.join("src/Token.sol"), "").unwrap();
fs::write(dir_path.join("src/common/Burnable.sol"), "").unwrap();

// assume that the import path is specified in Token.sol
let cwd = dir_path.join("src");

assert_eq!(
normalize_solidity_import_path(&cwd, "./common/Burnable.sol").unwrap(),
dir_path.join("src/common/Burnable.sol"),
);

assert!(normalize_solidity_import_path(&cwd, "./common/Pausable.sol").is_err());
}

// This test is exclusive to unix because creating a symlink is a privileged action on Windows.
// https://doc.rust-lang.org/std/os/windows/fs/fn.symlink_dir.html#limitations
#[test]
#[cfg(unix)]
fn can_normalize_solidity_import_path_symlink() {
let dir = tempfile::tempdir().unwrap();
let dir_path = dir.path();

// File structure:
//
// `dir_path`
// ├── dependency
// │   └── Math.sol
// └── project
// ├── node_modules
// │   └── dependency -> symlink to actual 'dependency' directory
// └── src (`cwd`)
// └── Token.sol

fs::create_dir_all(dir_path.join("project/src")).unwrap();
fs::write(dir_path.join("project/src/Token.sol"), "").unwrap();
fs::create_dir(dir_path.join("project/node_modules")).unwrap();

fs::create_dir(dir_path.join("dependency")).unwrap();
fs::write(dir_path.join("dependency/Math.sol"), "").unwrap();

std::os::unix::fs::symlink(
dir_path.join("dependency"),
dir_path.join("project/node_modules/dependency"),
)
.unwrap();

// assume that the import path is specified in Token.sol
let cwd = dir_path.join("project/src");

assert_eq!(
normalize_solidity_import_path(cwd, "../node_modules/dependency/Math.sol").unwrap(),
dir_path.join("project/node_modules/dependency/Math.sol"),
);
}

#[test]
fn can_clean_solidity_path() {
assert_eq!(clean_solidity_path("a"), PathBuf::from("a"));
assert_eq!(clean_solidity_path("./a"), PathBuf::from("a"));
assert_eq!(clean_solidity_path("../a"), PathBuf::from("../a"));
assert_eq!(clean_solidity_path("/a/"), PathBuf::from("/a"));
assert_eq!(clean_solidity_path("//a"), PathBuf::from("/a"));
assert_eq!(clean_solidity_path("a/b"), PathBuf::from("a/b"));
assert_eq!(clean_solidity_path("a//b"), PathBuf::from("a/b"));
assert_eq!(clean_solidity_path("/a/b"), PathBuf::from("/a/b"));
assert_eq!(clean_solidity_path("a/./b"), PathBuf::from("a/b"));
assert_eq!(clean_solidity_path("a/././b"), PathBuf::from("a/b"));
assert_eq!(clean_solidity_path("/a/../b"), PathBuf::from("/b"));
assert_eq!(clean_solidity_path("a/b/c"), PathBuf::from("a/b/c"));
assert_eq!(clean_solidity_path("a/b/../c"), PathBuf::from("a/c"));
assert_eq!(clean_solidity_path("a/b/../../c"), PathBuf::from("c"));
assert_eq!(clean_solidity_path("a/b/../../../c"), PathBuf::from("../c"));
assert_eq!(
clean_solidity_path("a/../b/../../c/./Token.sol"),
PathBuf::from("../c/Token.sol")
);
}

#[test]
fn can_find_ancestor() {
let a = Path::new("/foo/bar/bar/test.txt");
Expand Down
68 changes: 68 additions & 0 deletions tests/project.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1619,6 +1619,74 @@ fn can_compile_std_json_input() {
}
}

// This test is exclusive to unix because creating a symlink is a privileged action on windows.
// https://doc.rust-lang.org/std/os/windows/fs/fn.symlink_dir.html#limitations
#[test]
#[cfg(unix)]
fn can_create_standard_json_input_with_symlink() {
let mut project = TempProject::dapptools().unwrap();
let dependency = TempProject::dapptools().unwrap();

// File structure:
//
// project
// ├── node_modules
// │   └── dependency -> symlink to actual 'dependency' directory
// └── src
// └── A.sol
//
// dependency
// └── src
// ├── B.sol
// └── C.sol

fs::create_dir(project.root().join("node_modules")).unwrap();

std::os::unix::fs::symlink(dependency.root(), project.root().join("node_modules/dependency"))
.unwrap();
project.project_mut().paths.remappings.push(Remapping {
context: None,
name: "@dependency/".into(),
path: "node_modules/dependency/".into(),
});

project
.add_source(
"A",
r"pragma solidity >=0.8.0; import '@dependency/src/B.sol'; contract A is B {}",
)
.unwrap();
dependency
.add_source("B", r"pragma solidity >=0.8.0; import './C.sol'; contract B is C {}")
.unwrap();
dependency.add_source("C", r"pragma solidity >=0.8.0; contract C {}").unwrap();

// solc compiles using the host file system; therefore, this setup is considered valid
project.assert_no_errors();

// can create project root based paths
let std_json =
project.project().standard_json_input(project.sources_path().join("A.sol")).unwrap();
assert_eq!(
std_json.sources.iter().map(|(path, _)| path.clone()).collect::<Vec<_>>(),
vec![
PathBuf::from("src/A.sol"),
PathBuf::from("node_modules/dependency/src/B.sol"),
PathBuf::from("node_modules/dependency/src/C.sol")
]
);

// can compile using the created json
let compiler_errors = Solc::default()
.compile(&std_json)
.unwrap()
.errors
.into_iter()
.filter_map(|e| if e.severity.is_error() { Some(e.message) } else { None })
.collect::<Vec<_>>();
assert!(compiler_errors.is_empty(), "{:?}", compiler_errors);
}

#[test]
fn can_compile_model_checker_sample() {
let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/model-checker-sample");
Expand Down