Merge branch 'main' into feat/kms-vault-transit2

This commit is contained in:
cxymds
2026-04-07 11:11:43 +08:00
committed by GitHub
9 changed files with 346 additions and 35 deletions
+9 -4
View File
@@ -16,8 +16,9 @@ use http::HeaderMap;
use http::Uri;
use rustfs_credentials::{Credentials, get_global_action_cred};
use rustfs_iam::error::Error as IamError;
use rustfs_iam::sys::SESSION_POLICY_NAME;
use rustfs_iam::sys::get_claims_from_token_with_secret;
use rustfs_iam::sys::{
SESSION_POLICY_NAME, get_claims_from_token_with_secret, get_claims_from_token_with_secret_allow_missing_exp,
};
use rustfs_utils::http::ip::get_source_ip_raw;
use s3s::S3Error;
use s3s::S3ErrorCode;
@@ -353,8 +354,12 @@ pub fn check_claims_from_token(token: &str, cred: &Credentials) -> S3Result<Hash
};
if !token.is_empty() {
let claims: HashMap<String, Value> =
get_claims_from_token_with_secret(token, secret).map_err(|_e| s3_error!(InvalidRequest, "invalid token"))?;
let claims: HashMap<String, Value> = if cred.is_service_account() {
get_claims_from_token_with_secret_allow_missing_exp(token, secret)
.map_err(|_e| s3_error!(InvalidRequest, "invalid token"))?
} else {
get_claims_from_token_with_secret(token, secret).map_err(|_e| s3_error!(InvalidRequest, "invalid token"))?
};
return Ok(claims);
}
+89 -1
View File
@@ -1876,11 +1876,29 @@ impl S3Access for FS {
#[cfg(test)]
mod tests {
use super::*;
use http::{HeaderMap, Method, Uri};
use http::{Extensions, HeaderMap, Method, Uri};
use rustfs_policy::policy::{BucketPolicy, bucket_policy_uses_existing_object_tag_conditions};
use std::collections::HashMap;
use time::OffsetDateTime;
fn build_request<T>(input: T, method: Method) -> S3Request<T> {
S3Request {
input,
method,
uri: Uri::from_static("/"),
headers: HeaderMap::new(),
extensions: Extensions::new(),
credentials: None,
region: None,
service: None,
trailing_headers: None,
}
}
fn ensure_req_info<T>(req: &mut S3Request<T>) {
req.extensions.insert(ReqInfo::default());
}
#[test]
fn get_bucket_policy_uses_get_bucket_policy_action() {
assert_eq!(get_bucket_policy_authorize_action(), Action::S3Action(S3Action::GetBucketPolicyAction));
@@ -2315,4 +2333,74 @@ mod tests {
assert_eq!(req_info.object.as_deref(), Some("test-key"));
assert_eq!(req_info.version_id, None);
}
#[tokio::test]
async fn abort_multipart_upload_rejects_unauthorized_request() {
let fs = FS::new();
let mut req = build_request(
AbortMultipartUploadInput::builder()
.bucket("bucket".to_string())
.key("object".to_string())
.upload_id("upload-id".to_string())
.build()
.unwrap(),
Method::DELETE,
);
ensure_req_info(&mut req);
let err = fs
.abort_multipart_upload(&mut req)
.await
.expect_err("missing credentials should reject access");
assert_eq!(err.code(), &S3ErrorCode::AccessDenied);
}
#[tokio::test]
async fn complete_multipart_upload_rejects_unauthorized_request() {
let fs = FS::new();
let mut req = build_request(
CompleteMultipartUploadInput::builder()
.bucket("bucket".to_string())
.key("object".to_string())
.upload_id("upload-id".to_string())
.multipart_upload(Some(CompletedMultipartUpload::default()))
.build()
.unwrap(),
Method::POST,
);
ensure_req_info(&mut req);
let err = fs
.complete_multipart_upload(&mut req)
.await
.expect_err("missing credentials should reject access");
assert_eq!(err.code(), &S3ErrorCode::AccessDenied);
}
#[tokio::test]
async fn upload_part_copy_rejects_unauthorized_request() {
let fs = FS::new();
let mut req = build_request(
UploadPartCopyInput::builder()
.bucket("dst-bucket".to_string())
.key("dst-object".to_string())
.upload_id("upload-id".to_string())
.part_number(1)
.copy_source(CopySource::Bucket {
bucket: "src-bucket".into(),
key: "src-object".into(),
version_id: None,
})
.build()
.unwrap(),
Method::PUT,
);
ensure_req_info(&mut req);
let err = fs
.upload_part_copy(&mut req)
.await
.expect_err("missing credentials should reject access");
assert_eq!(err.code(), &S3ErrorCode::AccessDenied);
}
}