Start adding support for FabricClientSettings and FabricSecurityCredentials

crates/libs/core/src/client/mod.rs

@@ -3,12 +3,16 @@
 // Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
 // ------------------------------------------------------------
 
-use crate::Interface;
+use crate::{
+    types::{FabricClientSettings, FabricSecurityCredentials},
+    Interface,
+};
 use connection::{ClientConnectionEventHandlerBridge, LambdaClientConnectionNotificationHandler};
 use health_client::HealthClient;
 use mssf_com::FabricClient::{
-    IFabricClientConnectionEventHandler, IFabricHealthClient4, IFabricPropertyManagementClient2,
-    IFabricQueryClient10, IFabricServiceManagementClient6, IFabricServiceNotificationEventHandler,
+    IFabricClientConnectionEventHandler, IFabricClientSettings2, IFabricHealthClient4,
+    IFabricPropertyManagementClient2, IFabricQueryClient10, IFabricServiceManagementClient6,
+    IFabricServiceNotificationEventHandler,
 };
 use notification::{
     LambdaServiceNotificationHandler, ServiceNotificationEventHandler,
@@ -34,13 +38,22 @@ pub use notification::ServiceNotification;
 #[cfg(test)]
 mod tests;
 
+#[non_exhaustive]
+#[derive(Debug)]
+pub enum FabricClientCreationError {
+    InvalidFabricClientSettings(crate::Error),
+    InvalidFabricSecurityCredentials(crate::Error),
+}
+
 /// Creates FabricClient com object using SF com API.
 fn create_local_client_internal<T: Interface>(
     connection_strings: Option<&Vec<crate::WString>>,
     service_notification_handler: Option<&IFabricServiceNotificationEventHandler>,
     client_connection_handler: Option<&IFabricClientConnectionEventHandler>,
     client_role: Option<ClientRole>,
-) -> T {
+    client_settings: Option<FabricClientSettings>,
+    client_credentials: Option<FabricSecurityCredentials>,
+) -> Result<T, FabricClientCreationError> {
     let role = client_role.unwrap_or(ClientRole::Unknown);
 
     // create raw conn str ptrs.
@@ -51,7 +64,7 @@ fn create_local_client_internal<T: Interface>(
             .collect::<Vec<_>>()
     });
 
-    match connection_strings_ptrs {
+    let client = match connection_strings_ptrs {
         Some(addrs) => {
             assert!(
                 role == ClientRole::Unknown,
@@ -80,7 +93,24 @@ fn create_local_client_internal<T: Interface>(
         }
     }
     // if params are right, client should be created. There is no network call involved during obj creation.
-    .expect("failed to create fabric client")
+    .expect("failed to create fabric client");
+    if client_settings.is_some() || client_credentials.is_some() {
+        let setting_interface = client
+            .clone()
+            .cast::<IFabricClientSettings2>()
+            .expect("failed to cast fabric client to IFabricClientSettings2");
+        if let Some(desired_settings) = client_settings {
+            desired_settings
+                .apply(&setting_interface)
+                .map_err(FabricClientCreationError::InvalidFabricClientSettings)?;
+        }
+        if let Some(desired_credentials) = client_credentials {
+            desired_credentials
+                .apply(&setting_interface)
+                .map_err(FabricClientCreationError::InvalidFabricSecurityCredentials)?;
+        }
+    };
+    Ok(client)
 }
 
 // Builder for FabricClient
@@ -89,6 +119,8 @@ pub struct FabricClientBuilder {
     cc_handler: Option<LambdaClientConnectionNotificationHandler>,
     client_role: ClientRole,
     connection_strings: Option<Vec<crate::WString>>,
+    client_settings: Option<FabricClientSettings>,
+    client_credentials: Option<FabricSecurityCredentials>,
 }
 
 impl Default for FabricClientBuilder {
@@ -105,6 +137,8 @@ impl FabricClientBuilder {
             cc_handler: None,
             client_role: ClientRole::Unknown,
             connection_strings: None,
+            client_settings: None,
+            client_credentials: None,
         }
     }
 
@@ -174,17 +208,29 @@ impl FabricClientBuilder {
         self
     }
 
+    /// Sets the client settings
+    pub fn with_client_settings(mut self, client_settings: FabricClientSettings) -> Self {
+        self.client_settings = Some(client_settings);
+        self
+    }
+
+    // Sets the client credentials
+    pub fn with_credentials(mut self, client_credentials: FabricSecurityCredentials) -> Self {
+        self.client_credentials = Some(client_credentials);
+        self
+    }
+
     /// Build the fabricclient
     /// Remarks: FabricClient connect to SF cluster when
     /// the first API call is triggered. Build/create of the object does not
     /// establish connection.
-    pub fn build(self) -> FabricClient {
-        let c = Self::build_interface(self);
-        FabricClient::from_com(c)
+    pub fn build(self) -> Result<FabricClient, FabricClientCreationError> {
+        let c = Self::build_interface(self)?;
+        Ok(FabricClient::from_com(c))
     }
 
     /// Build the specific com interface of the fabric client.
-    pub fn build_interface<T: Interface>(self) -> T {
+    pub fn build_interface<T: Interface>(self) -> Result<T, FabricClientCreationError> {
         let cc_handler = self
             .cc_handler
             .map(ClientConnectionEventHandlerBridge::new_com);
@@ -193,6 +239,8 @@ impl FabricClientBuilder {
             self.sn_handler.as_ref(),
             cc_handler.as_ref(),
             Some(self.client_role),
+            self.client_settings,
+            self.client_credentials,
         )
     }
 }

crates/libs/core/src/client/tests.rs

@@ -19,7 +19,7 @@ use crate::{
 
 #[tokio::test]
 async fn test_fabric_client() {
-    let c = FabricClient::builder().build();
+    let c = FabricClient::builder().build().unwrap();
     let qc = c.get_query_manager();
     let timeout = Duration::from_secs(1);
     let paging_status;

crates/libs/core/src/sync/mod.rs

@@ -154,7 +154,7 @@ mod async_tests {
                 pub fn new() -> $name {
                     return $name {
                         com: paste::item! {
-                            crate::client::FabricClientBuilder::new().build_interface::<mssf_com::FabricClient::[<I $name>]>()
+                            crate::client::FabricClientBuilder::new().build_interface::<mssf_com::FabricClient::[<I $name>]>().unwrap()
                         },
                     };
                 }
@@ -242,7 +242,9 @@ mod async_tests {
     impl FabricQueryClient {
         pub fn new() -> FabricQueryClient {
             FabricQueryClient {
-                com: crate::client::FabricClient::builder().build_interface::<IFabricQueryClient>(),
+                com: crate::client::FabricClient::builder()
+                    .build_interface::<IFabricQueryClient>()
+                    .expect("default settings and credentials must be valid"),
             }
         }
 

crates/libs/core/src/types/client/mod.rs

@@ -23,6 +23,8 @@ mod metrics;
 pub use metrics::*;
 mod health;
 pub use health::*;
+mod settings;
+pub use settings::*;
 
 // FABRIC_SERVICE_NOTIFICATION_FILTER_FLAGS
 bitflags::bitflags! {

crates/libs/core/src/types/client/settings.rs

@@ -0,0 +1,31 @@
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+
+use mssf_com::FabricClient::IFabricClientSettings2;
+
+/// A idiomatic Rust version of FABRIC_CLIENT_SETTINGS
+///
+/// Note: we may choose to add additional optional fields in future without considering that a SemVer breaking change.
+/// You should default fields you're not interested in like so:
+/// ```
+/// # use std::num::NonZeroU32;
+/// # use mssf_core::types::FabricClientSettings;
+/// let my_settings = FabricClientSettings {
+///  // TODO: uncomment in next PR
+///  // PartitionLocationCacheLimit: Some(NonZeroU32::new(1).expect("Non-zero value")),
+///  // Any other hypothetical settings you're interested in here,
+///  ..Default::default()
+/// };
+/// ```
+#[derive(Default)]
+pub struct FabricClientSettings {}
+
+impl FabricClientSettings {
+    /// Note: only overrides non-default settings; leaves any settings set previously that don't explicitly have new values alone
+    pub(crate) fn apply(&self, _settings_interface: &IFabricClientSettings2) -> crate::Result<()> {
+        // Placeholder
+        Ok(())
+    }
+}

crates/libs/core/src/types/common/mod.rs

@@ -6,6 +6,8 @@
 // This mod contains common types shared between FabricRuntime and FabricClient.
 mod partition;
 pub use partition::*;
+mod security_credentials;
+pub use security_credentials::*;
 mod stateful;
 pub use stateful::*;
 mod metrics;

crates/libs/core/src/types/common/security_credentials/claims_credentials.rs

@@ -0,0 +1,7 @@
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+#![deny(unsafe_op_in_unsafe_fn)]
+#![deny(clippy::undocumented_unsafe_blocks)]
+// TODO: implement

crates/libs/core/src/types/common/security_credentials/fabric_protection_level.rs

@@ -0,0 +1,45 @@
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+use mssf_com::FabricTypes::{
+    FABRIC_PROTECTION_LEVEL, FABRIC_PROTECTION_LEVEL_ENCRYPTANDSIGN, FABRIC_PROTECTION_LEVEL_NONE,
+    FABRIC_PROTECTION_LEVEL_SIGN,
+};
+
+/// The Fabric Protection Level
+/// See https://learn.microsoft.com/en-us/dotnet/api/system.fabric.protectionlevel?view=azure-dotnet
+#[non_exhaustive]
+#[derive(Copy, Clone, PartialEq, Eq)]
+pub enum FabricProtectionLevel {
+    None,
+    Sign,
+    EncryptAndSign,
+}
+
+#[derive(Debug)]
+#[allow(dead_code, reason = "For error handling")]
+pub struct FabricProtectionLevelUnknownValueError(pub FABRIC_PROTECTION_LEVEL);
+
+impl TryFrom<FABRIC_PROTECTION_LEVEL> for FabricProtectionLevel {
+    type Error = FabricProtectionLevelUnknownValueError;
+
+    fn try_from(value: FABRIC_PROTECTION_LEVEL) -> Result<Self, Self::Error> {
+        match value {
+            FABRIC_PROTECTION_LEVEL_NONE => Ok(FabricProtectionLevel::None),
+            FABRIC_PROTECTION_LEVEL_SIGN => Ok(FabricProtectionLevel::Sign),
+            FABRIC_PROTECTION_LEVEL_ENCRYPTANDSIGN => Ok(FabricProtectionLevel::EncryptAndSign),
+            x => Err(FabricProtectionLevelUnknownValueError(x)),
+        }
+    }
+}
+
+impl From<FabricProtectionLevel> for FABRIC_PROTECTION_LEVEL {
+    fn from(value: FabricProtectionLevel) -> Self {
+        match value {
+            FabricProtectionLevel::None => FABRIC_PROTECTION_LEVEL_NONE,
+            FabricProtectionLevel::Sign => FABRIC_PROTECTION_LEVEL_SIGN,
+            FabricProtectionLevel::EncryptAndSign => FABRIC_PROTECTION_LEVEL_ENCRYPTANDSIGN,
+        }
+    }
+}

crates/libs/core/src/types/common/security_credentials/mod.rs

@@ -0,0 +1,27 @@
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+#![deny(unsafe_op_in_unsafe_fn)]
+#![deny(clippy::undocumented_unsafe_blocks)]
+use mssf_com::FabricClient::IFabricClientSettings2;
+
+mod claims_credentials;
+mod fabric_protection_level;
+pub use fabric_protection_level::*;
+mod windows_credentials;
+mod x509_credentials;
+pub use x509_credentials::*;
+
+/// Idiomatic FABRIC_SECURITY_CREDENTIALS wrapper
+/// Currently, just a placeholder
+#[non_exhaustive]
+pub enum FabricSecurityCredentials {}
+
+impl FabricSecurityCredentials {
+    /// Note: only overrides non-default settings; leaves any settings set previously that don't explicitly have new values alone
+    pub(crate) fn apply(&self, _settings_interface: &IFabricClientSettings2) -> crate::Result<()> {
+        // Placeholder
+        Ok(())
+    }
+}

crates/libs/core/src/types/common/security_credentials/windows_credentials.rs

@@ -0,0 +1,7 @@
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+#![deny(unsafe_op_in_unsafe_fn)]
+#![deny(clippy::undocumented_unsafe_blocks)]
+// TODO: implement

crates/libs/core/src/types/common/security_credentials/x509_credentials.rs

@@ -0,0 +1,81 @@
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+// ------------------------------------------------------------
+// Copyright (c) Microsoft Corporation.  All rights reserved.
+// Licensed under the MIT License (MIT). See License.txt in the repo root for license information.
+// ------------------------------------------------------------
+#![deny(unsafe_op_in_unsafe_fn)]
+#![deny(clippy::undocumented_unsafe_blocks)]
+
+use mssf_com::FabricTypes::{
+    FABRIC_X509_FIND_TYPE, FABRIC_X509_FIND_TYPE_FINDBYEXTENSION,
+    FABRIC_X509_FIND_TYPE_FINDBYSUBJECTNAME, FABRIC_X509_FIND_TYPE_FINDBYTHUMBPRINT,
+    FABRIC_X509_STORE_LOCATION, FABRIC_X509_STORE_LOCATION_CURRENTUSER,
+    FABRIC_X509_STORE_LOCATION_INVALID, FABRIC_X509_STORE_LOCATION_LOCALMACHINE,
+};
+use windows_core::WString;
+
+/// How to find the X509 certificate.
+#[non_exhaustive]
+pub enum FabricX509FindType {
+    FindByExtension { extension: WString },
+    FindBySubjectName { subject_name: WString },
+    FindByThumbprint { thumbprint: WString },
+}
+
+impl From<&FabricX509FindType> for FABRIC_X509_FIND_TYPE {
+    fn from(value: &FabricX509FindType) -> Self {
+        match value {
+            FabricX509FindType::FindByExtension { extension: _ } => {
+                FABRIC_X509_FIND_TYPE_FINDBYEXTENSION
+            }
+            FabricX509FindType::FindBySubjectName { subject_name: _ } => {
+                FABRIC_X509_FIND_TYPE_FINDBYSUBJECTNAME
+            }
+            FabricX509FindType::FindByThumbprint { thumbprint: _ } => {
+                FABRIC_X509_FIND_TYPE_FINDBYTHUMBPRINT
+            }
+        }
+    }
+}
+
+/// What store location the certificate will be found in
+#[non_exhaustive]
+#[derive(Copy, Clone)]
+pub enum FabricX509StoreLocation {
+    CurrentUser,
+    LocalMachine,
+}
+
+#[non_exhaustive]
+#[derive(Copy, Clone)]
+pub enum FabricX509StoreLocationConversionError {
+    InvalidValue,
+    UnknownValue(FABRIC_X509_STORE_LOCATION),
+}
+
+impl TryFrom<FABRIC_X509_STORE_LOCATION> for FabricX509StoreLocation {
+    type Error = FabricX509StoreLocationConversionError;
+
+    fn try_from(value: FABRIC_X509_STORE_LOCATION) -> Result<Self, Self::Error> {
+        match value {
+            FABRIC_X509_STORE_LOCATION_CURRENTUSER => Ok(FabricX509StoreLocation::CurrentUser),
+            FABRIC_X509_STORE_LOCATION_LOCALMACHINE => Ok(FabricX509StoreLocation::LocalMachine),
+            FABRIC_X509_STORE_LOCATION_INVALID => {
+                Err(FabricX509StoreLocationConversionError::InvalidValue)
+            }
+            x => Err(FabricX509StoreLocationConversionError::UnknownValue(x)),
+        }
+    }
+}
+
+impl From<FabricX509StoreLocation> for FABRIC_X509_STORE_LOCATION {
+    fn from(value: FabricX509StoreLocation) -> Self {
+        match value {
+            FabricX509StoreLocation::CurrentUser => FABRIC_X509_STORE_LOCATION_CURRENTUSER,
+            FabricX509StoreLocation::LocalMachine => FABRIC_X509_STORE_LOCATION_LOCALMACHINE,
+        }
+    }
+}

crates/samples/echomain-stateful2/src/test.rs

@@ -250,7 +250,7 @@ impl TestClient {
 // Uses fabric client to perform various actions for this service.
 #[tokio::test]
 async fn test_partition_info() {
-    let fc = FabricClient::builder().build();
+    let fc = FabricClient::builder().build().unwrap();
     let tc = TestClient::new(fc.clone());
     let timeout = Duration::from_secs(1);