mirror of
https://github.com/openai/codex.git
synced 2026-09-17 12:23:33 +00:00
b5ea64a203
## What changed - Export `read_sensitive_file_to_string` from `codex-exec-server`. - Require the opened path to be a regular disk file and avoid following its final symlink component on Unix or reparse point on Windows. - Read valid UTF-8 file contents asynchronously and return I/O errors for unsupported inputs. ## Testing Add tests covering regular files, directories, and symlinks. GitOrigin-RevId: 68809e94c0d3719e5685c064f9610a0455ffd8d7
84 lines
2.5 KiB
Rust
84 lines
2.5 KiB
Rust
use std::io;
|
|
use std::path::Path;
|
|
use tokio::io::AsyncReadExt;
|
|
|
|
pub(crate) async fn open(path: &Path) -> io::Result<tokio::fs::File> {
|
|
let mut options = tokio::fs::OpenOptions::new();
|
|
options.read(true);
|
|
configure_open(&mut options);
|
|
|
|
let file = options.open(path).await?;
|
|
if !is_disk_file(&file) || !file.metadata().await?.is_file() {
|
|
return Err(io::Error::new(
|
|
io::ErrorKind::InvalidInput,
|
|
format!("path `{}` is not a file", path.display()),
|
|
));
|
|
}
|
|
Ok(file)
|
|
}
|
|
|
|
/// Reads a regular UTF-8 file without following a symlink at its final path component.
|
|
pub async fn read_sensitive_file_to_string(path: &Path) -> io::Result<String> {
|
|
let mut options = tokio::fs::OpenOptions::new();
|
|
options.read(true);
|
|
configure_open(&mut options);
|
|
|
|
#[cfg(unix)]
|
|
options.custom_flags(libc::O_NONBLOCK | libc::O_NOFOLLOW);
|
|
|
|
#[cfg(windows)]
|
|
{
|
|
use windows_sys::Win32::Storage::FileSystem::FILE_FLAG_OPEN_REPARSE_POINT;
|
|
|
|
options.custom_flags(FILE_FLAG_OPEN_REPARSE_POINT);
|
|
}
|
|
|
|
let mut file = options.open(path).await?;
|
|
let metadata = file.metadata().await?;
|
|
if !is_disk_file(&file) || !metadata.is_file() || metadata.file_type().is_symlink() {
|
|
return Err(io::Error::new(
|
|
io::ErrorKind::InvalidInput,
|
|
format!("path `{}` is not a regular file", path.display()),
|
|
));
|
|
}
|
|
|
|
let mut contents = String::new();
|
|
file.read_to_string(&mut contents).await?;
|
|
Ok(contents)
|
|
}
|
|
|
|
#[cfg(unix)]
|
|
fn configure_open(options: &mut tokio::fs::OpenOptions) {
|
|
options.custom_flags(libc::O_NONBLOCK);
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
fn configure_open(options: &mut tokio::fs::OpenOptions) {
|
|
use windows_sys::Win32::Storage::FileSystem::SECURITY_IDENTIFICATION;
|
|
|
|
options.security_qos_flags(SECURITY_IDENTIFICATION);
|
|
}
|
|
|
|
#[cfg(not(any(unix, windows)))]
|
|
fn configure_open(_options: &mut tokio::fs::OpenOptions) {}
|
|
|
|
#[cfg(windows)]
|
|
fn is_disk_file(file: &tokio::fs::File) -> bool {
|
|
use std::os::windows::io::AsRawHandle;
|
|
use windows_sys::Win32::Foundation::HANDLE;
|
|
use windows_sys::Win32::Storage::FileSystem::FILE_TYPE_DISK;
|
|
use windows_sys::Win32::Storage::FileSystem::GetFileType;
|
|
|
|
// SAFETY: `file` owns this handle for the duration of the call.
|
|
unsafe { GetFileType(file.as_raw_handle() as HANDLE) == FILE_TYPE_DISK }
|
|
}
|
|
|
|
#[cfg(not(windows))]
|
|
fn is_disk_file(_file: &tokio::fs::File) -> bool {
|
|
true
|
|
}
|
|
|
|
#[cfg(test)]
|
|
#[path = "regular_file_tests.rs"]
|
|
mod tests;
|