rustfs/crates/ecstore/src/services/tier/warm_backend_s3.rs

// Copyright 2024 RustFS Team
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#![allow(unused_imports)]
#![allow(unused_variables)]
#![allow(unused_mut)]
#![allow(unused_assignments)]
#![allow(unused_must_use)]
#![allow(clippy::all)]

use std::collections::HashMap;
use std::sync::Arc;
use url::Url;

use crate::client::{
    api_get_options::GetObjectOptions,
    api_list::ListObjectsOptions,
    api_put_object::PutObjectOptions,
    api_remove::{RemoveObjectOptions, RemoveObjectResult},
    api_s3_datatypes::ListVersionsResult,
    credentials::{Credentials, SignatureType, Static, Value},
    provider_versions::validate_remote_version_id,
    transition_api::{BucketLookupType, Options, TransitionClient, TransitionCore},
    transition_api::{ReadCloser, ReaderImpl},
};
use crate::error::ErrorResponse;
use crate::error::error_resp_to_object_err;
use crate::services::tier::{
    tier_config::TierS3,
    warm_backend::{TransitionCandidateProbe, WarmBackend, WarmBackendGetOpts, build_transition_put_options},
};
use http::HeaderMap;
use rustfs_utils::egress::validate_outbound_url;
use rustfs_utils::path::SLASH_SEPARATOR;
use s3s::dto::BucketVersioningStatus;

pub struct WarmBackendS3 {
    pub client: Arc<TransitionClient>,
    pub core: TransitionCore,
    pub bucket: String,
    pub prefix: String,
    pub storage_class: String,
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum RemoteBucketVersioning {
    Disabled,
    Suspended,
    Enabled,
}

fn remote_bucket_versioning_from_status(status: Option<&str>) -> Result<RemoteBucketVersioning, std::io::Error> {
    Ok(match status {
        Some(BucketVersioningStatus::ENABLED) => RemoteBucketVersioning::Enabled,
        Some(BucketVersioningStatus::SUSPENDED) => RemoteBucketVersioning::Suspended,
        Some(status) => {
            return Err(std::io::Error::new(
                std::io::ErrorKind::InvalidData,
                format!("remote tier bucket returned unsupported versioning status {status}"),
            ));
        }
        None => RemoteBucketVersioning::Disabled,
    })
}

impl WarmBackendS3 {
    pub async fn new(conf: &TierS3, _tier: &str) -> Result<Self, std::io::Error> {
        Self::new_with_bucket_lookup(conf, BucketLookupType::BucketLookupAuto, "s3").await
    }

    pub(crate) async fn new_with_bucket_lookup(
        conf: &TierS3,
        bucket_lookup: BucketLookupType,
        tier_type: &str,
    ) -> Result<Self, std::io::Error> {
        let u = match Url::parse(&conf.endpoint) {
            Ok(u) => u,
            Err(err) => {
                return Err(std::io::Error::other(err.to_string()));
            }
        };
        validate_outbound_url(&u).map_err(|err| std::io::Error::other(format!("tier endpoint is not allowed: {err}")))?;

        if conf.aws_role_web_identity_token_file == "" && conf.aws_role_arn != ""
            || conf.aws_role_web_identity_token_file != "" && conf.aws_role_arn == ""
        {
            return Err(std::io::Error::other("both the token file and the role ARN are required"));
        } else if conf.access_key == "" && conf.secret_key != "" || conf.access_key != "" && conf.secret_key == "" {
            return Err(std::io::Error::other("both the access and secret keys are required"));
        } else if conf.aws_role
            && (conf.aws_role_web_identity_token_file != ""
                || conf.aws_role_arn != ""
                || conf.access_key != ""
                || conf.secret_key != "")
        {
            return Err(std::io::Error::other(
                "AWS Role cannot be activated with static credentials or the web identity token file",
            ));
        } else if conf.bucket == "" {
            return Err(std::io::Error::other("no bucket name was provided"));
        }

        let creds: Credentials<Static>;

        if conf.access_key != "" && conf.secret_key != "" {
            //creds = Credentials::new_static_v4(conf.access_key, conf.secret_key, "");
            creds = Credentials::new(Static(Value {
                access_key_id: conf.access_key.clone(),
                secret_access_key: conf.secret_key.clone(),
                session_token: "".to_string(),
                signer_type: SignatureType::SignatureV4,
                ..Default::default()
            }));
        } else {
            return Err(std::io::Error::other("insufficient parameters for S3 backend authentication"));
        }
        let opts = Options {
            creds,
            secure: u.scheme() == "https",
            region: conf.region.clone(),
            bucket_lookup,
            ..Default::default()
        };
        let host = u
            .host()
            .ok_or_else(|| std::io::Error::other("Invalid endpoint URL: missing host"))?;
        let client = TransitionClient::new(&host.to_string(), opts, tier_type).await?;

        let client = Arc::new(client);
        let core = TransitionCore(Arc::clone(&client));
        Ok(Self {
            client,
            core,
            bucket: conf.bucket.clone(),
            prefix: conf.prefix.clone().trim_matches('/').to_string(),
            storage_class: conf.storage_class.clone(),
        })
    }

    pub fn get_dest(&self, object: &str) -> String {
        let mut dest_obj = object.to_string();
        if self.prefix != "" {
            dest_obj = format!("{}/{}", &self.prefix, object);
        }
        return dest_obj;
    }

    pub(crate) async fn remove_with_result(&self, object: &str, rv: &str) -> Result<RemoveObjectResult, std::io::Error> {
        let mut opts = RemoveObjectOptions::default();
        if !rv.is_empty() {
            opts.version_id = rv.to_string();
        }
        self.client
            .remove_object_inner(&self.bucket, &self.get_dest(object), opts)
            .await
    }

    pub(crate) async fn get_with_headers(
        &self,
        object: &str,
        rv: &str,
        opts: WarmBackendGetOpts,
    ) -> Result<(HeaderMap, ReadCloser), std::io::Error> {
        let mut gopts = GetObjectOptions::default();

        if !rv.is_empty() {
            gopts.version_id = rv.to_string();
        }
        if opts.start_offset >= 0 && opts.length > 0 {
            gopts
                .set_range(opts.start_offset, opts.start_offset + opts.length - 1)
                .map_err(std::io::Error::other)?;
        }
        let (_, headers, reader) = self.core.get_object(&self.bucket, &self.get_dest(object), &gopts).await?;
        Ok((headers, reader))
    }

    async fn remote_bucket_versioning(&self) -> Result<RemoteBucketVersioning, std::io::Error> {
        let config = self.client.get_bucket_versioning(&self.bucket).await?;
        remote_bucket_versioning_from_status(config.status.as_ref().map(|status| status.as_str()))
    }

    async fn probe_transition_candidate_versions(
        &self,
        object: &str,
        bucket_versioning: RemoteBucketVersioning,
    ) -> Result<TransitionCandidateProbe, std::io::Error> {
        let remote_object = self.get_dest(object);
        let mut opts = ListObjectsOptions::default();
        opts.set("prefix", &remote_object);
        opts.set("max-keys", "1000");

        let mut key_marker = String::new();
        let mut version_id_marker = String::new();
        let mut candidates = TransitionCandidateVersions::default();
        loop {
            let versions = self
                .client
                .list_object_versions_query(&self.bucket, &opts, &key_marker, &version_id_marker, "")
                .await?;
            candidates.extend(&remote_object, &versions);
            if candidates.is_ambiguous() {
                return Ok(TransitionCandidateProbe::Ambiguous);
            }
            if !versions.is_truncated {
                return classify_transition_candidates(candidates, bucket_versioning);
            }

            advance_version_markers(&mut key_marker, &mut version_id_marker, &versions)?;
        }
    }
}

fn classify_transition_candidates(
    candidates: TransitionCandidateVersions,
    bucket_versioning: RemoteBucketVersioning,
) -> Result<TransitionCandidateProbe, std::io::Error> {
    let probe = candidates.classify(bucket_versioning);
    if let TransitionCandidateProbe::VersionedPresent(version_id) = &probe {
        validate_remote_version_id(version_id)?;
    }
    Ok(probe)
}

fn advance_version_markers(
    key_marker: &mut String,
    version_id_marker: &mut String,
    versions: &ListVersionsResult,
) -> Result<(), std::io::Error> {
    let next_markers = (&versions.next_key_marker, &versions.next_version_id_marker);
    if next_markers == (&*key_marker, &*version_id_marker) {
        return Err(std::io::Error::new(
            std::io::ErrorKind::InvalidData,
            "ListObjectVersions pagination markers did not advance",
        ));
    }
    key_marker.clone_from(&versions.next_key_marker);
    version_id_marker.clone_from(&versions.next_version_id_marker);
    Ok(())
}

#[derive(Default)]
struct TransitionCandidateVersions {
    version_id: Option<String>,
    ambiguous: bool,
}

impl TransitionCandidateVersions {
    fn extend(&mut self, remote_object: &str, versions: &ListVersionsResult) {
        for version in versions.versions.iter().filter(|version| version.key == remote_object) {
            if self.version_id.is_some() {
                self.ambiguous = true;
                return;
            }
            self.version_id = Some(version.version_id.clone());
        }
    }

    fn is_ambiguous(&self) -> bool {
        self.ambiguous
    }

    fn classify(self, bucket_versioning: RemoteBucketVersioning) -> TransitionCandidateProbe {
        if self.ambiguous {
            return TransitionCandidateProbe::Ambiguous;
        }
        let Some(version_id) = self.version_id else {
            return TransitionCandidateProbe::Missing;
        };

        match bucket_versioning {
            RemoteBucketVersioning::Disabled => TransitionCandidateProbe::UnversionedPresent,
            RemoteBucketVersioning::Suspended if version_id == "null" => TransitionCandidateProbe::VersionedPresent(version_id),
            RemoteBucketVersioning::Suspended | RemoteBucketVersioning::Enabled if !version_id.is_empty() => {
                TransitionCandidateProbe::VersionedPresent(version_id)
            }
            RemoteBucketVersioning::Suspended | RemoteBucketVersioning::Enabled => TransitionCandidateProbe::Ambiguous,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::client::api_s3_datatypes::{ListVersionsResult, Version};

    #[tokio::test]
    async fn new_rejects_loopback_endpoint_before_network_setup() {
        let conf = TierS3 {
            endpoint: "https://127.0.0.1:9000".to_string(),
            bucket: "tier-bucket".to_string(),
            access_key: "access".to_string(),
            secret_key: "secret".to_string(),
            region: "us-east-1".to_string(),
            ..Default::default()
        };

        match WarmBackendS3::new(&conf, "tier").await {
            Ok(_) => panic!("loopback endpoint should be rejected"),
            Err(err) => assert!(err.to_string().contains("not allowed")),
        }
    }

    fn list_versions(versions: &[(&str, &str)], delete_markers: &[(&str, &str)], is_truncated: bool) -> ListVersionsResult {
        ListVersionsResult {
            versions: versions
                .iter()
                .map(|(key, version_id)| Version {
                    key: (*key).to_string(),
                    version_id: (*version_id).to_string(),
                    ..Default::default()
                })
                .collect(),
            delete_markers: delete_markers
                .iter()
                .map(|(key, version_id)| Version {
                    key: (*key).to_string(),
                    version_id: (*version_id).to_string(),
                    ..Default::default()
                })
                .collect(),
            is_truncated,
            ..Default::default()
        }
    }

    fn classify_pages(bucket_versioning: RemoteBucketVersioning, pages: &[ListVersionsResult]) -> TransitionCandidateProbe {
        let mut candidates = TransitionCandidateVersions::default();
        for page in pages {
            candidates.extend("archive/object", page);
        }
        candidates.classify(bucket_versioning)
    }

    #[test]
    fn transition_candidate_probe_classifier_is_fail_closed() {
        assert_eq!(
            classify_pages(RemoteBucketVersioning::Disabled, &[list_versions(&[], &[], false)],),
            TransitionCandidateProbe::Missing
        );
        assert_eq!(
            classify_pages(RemoteBucketVersioning::Disabled, &[list_versions(&[("archive/object", "")], &[], false)],),
            TransitionCandidateProbe::UnversionedPresent
        );
        assert_eq!(
            classify_pages(
                RemoteBucketVersioning::Enabled,
                &[list_versions(&[("archive/object", "version-a")], &[], false)],
            ),
            TransitionCandidateProbe::VersionedPresent("version-a".to_string())
        );
        assert_eq!(
            classify_pages(
                RemoteBucketVersioning::Suspended,
                &[list_versions(&[("archive/object", "null")], &[], false)],
            ),
            TransitionCandidateProbe::VersionedPresent("null".to_string())
        );
        assert_eq!(
            classify_pages(RemoteBucketVersioning::Enabled, &[list_versions(&[("archive/object", "")], &[], false)],),
            TransitionCandidateProbe::Ambiguous
        );
        assert_eq!(
            classify_pages(
                RemoteBucketVersioning::Enabled,
                &[list_versions(
                    &[("archive/object", "version-a"), ("archive/object", "version-b")],
                    &[],
                    false,
                )],
            ),
            TransitionCandidateProbe::Ambiguous
        );
    }

    #[test]
    fn transition_candidate_probe_reconciles_all_pages_and_ignores_delete_markers() {
        assert_eq!(
            classify_pages(
                RemoteBucketVersioning::Enabled,
                &[
                    list_versions(&[], &[("archive/object", "marker-a")], true),
                    list_versions(&[("archive/object", "version-a"), ("archive/object-adjacent", "unrelated"),], &[], false,),
                ],
            ),
            TransitionCandidateProbe::VersionedPresent("version-a".to_string())
        );
        assert_eq!(
            classify_pages(
                RemoteBucketVersioning::Enabled,
                &[
                    list_versions(&[("archive/object", "version-a")], &[], true),
                    list_versions(&[("archive/object", "version-b")], &[], false),
                ],
            ),
            TransitionCandidateProbe::Ambiguous
        );
    }

    #[test]
    fn transition_candidate_pagination_advances_both_markers() {
        let mut key_marker = "old-key".to_string();
        let mut version_id_marker = "old-version".to_string();
        let page = ListVersionsResult {
            next_key_marker: "next-key".to_string(),
            next_version_id_marker: "next-version".to_string(),
            ..Default::default()
        };

        advance_version_markers(&mut key_marker, &mut version_id_marker, &page)
            .expect("new ListObjectVersions markers should advance pagination");
        assert_eq!(key_marker, "next-key");
        assert_eq!(version_id_marker, "next-version");

        let err = advance_version_markers(&mut key_marker, &mut version_id_marker, &page)
            .expect_err("repeated ListObjectVersions markers must fail closed");
        assert_eq!(err.kind(), std::io::ErrorKind::InvalidData);
    }

    #[test]
    fn transition_candidate_probe_rejects_untrusted_version_ids() {
        let mut candidates = TransitionCandidateVersions::default();
        candidates.extend("archive/object", &list_versions(&[("archive/object", "version\ninjection")], &[], false));

        let err = classify_transition_candidates(candidates, RemoteBucketVersioning::Enabled)
            .expect_err("control characters in listed version IDs must fail closed");
        assert_eq!(err.kind(), std::io::ErrorKind::InvalidData);
    }

    #[test]
    fn remote_bucket_versioning_status_parser_fails_closed() {
        assert_eq!(
            remote_bucket_versioning_from_status(None).expect("absent status means disabled"),
            RemoteBucketVersioning::Disabled
        );
        assert_eq!(
            remote_bucket_versioning_from_status(Some(BucketVersioningStatus::ENABLED)).expect("enabled status should parse"),
            RemoteBucketVersioning::Enabled
        );
        assert_eq!(
            remote_bucket_versioning_from_status(Some(BucketVersioningStatus::SUSPENDED)).expect("suspended status should parse"),
            RemoteBucketVersioning::Suspended
        );
        let err = remote_bucket_versioning_from_status(Some("UnexpectedStatus"))
            .expect_err("unknown versioning status must fail closed");
        assert_eq!(err.kind(), std::io::ErrorKind::InvalidData);
    }
}

#[async_trait::async_trait]
impl WarmBackend for WarmBackendS3 {
    async fn put_with_meta(
        &self,
        object: &str,
        r: ReaderImpl,
        length: i64,
        meta: HashMap<String, String>,
    ) -> Result<String, std::io::Error> {
        let client = self.client.clone();
        let res = client
            .put_object(&self.bucket, &self.get_dest(object), r, length, &{
                let mut opts = build_transition_put_options(self.storage_class.clone(), meta);
                opts.send_content_md5 = true;
                opts
            })
            .await?;
        Ok(res.version_id)
    }

    async fn put(&self, object: &str, r: ReaderImpl, length: i64) -> Result<String, std::io::Error> {
        self.put_with_meta(object, r, length, HashMap::new()).await
    }

    async fn get(&self, object: &str, rv: &str, opts: WarmBackendGetOpts) -> Result<ReadCloser, std::io::Error> {
        self.get_with_headers(object, rv, opts).await.map(|(_, reader)| reader)
    }

    async fn remove(&self, object: &str, rv: &str) -> Result<(), std::io::Error> {
        self.remove_with_result(object, rv).await.map(|_| ())
    }

    async fn probe_transition_candidate(&self, object: &str) -> Result<TransitionCandidateProbe, std::io::Error> {
        let bucket_versioning = self.remote_bucket_versioning().await?;
        self.probe_transition_candidate_versions(object, bucket_versioning).await
    }

    async fn in_use(&self) -> Result<bool, std::io::Error> {
        let result = self
            .core
            .list_objects_v2(&self.bucket, &self.prefix, "", "", SLASH_SEPARATOR, 1)
            .await?;

        Ok(result.common_prefixes.len() > 0 || result.contents.len() > 0)
    }
}