internal static string ToSerializedValue(this VulnerabilityAssessmentPolicyBaselineName value)
{
switch (value)
{
case VulnerabilityAssessmentPolicyBaselineName.Master:
return("master");
case VulnerabilityAssessmentPolicyBaselineName.Default:
return("default");
}
return(null);
}
public virtual Response <bool> Exists(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(ruleId, nameof(ruleId));
using var scope = _serverDatabaseVulnerabilityAssessmentRuleBaselineDatabaseVulnerabilityAssessmentRuleBaselinesClientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.Exists");
scope.Start();
try
{
var response = GetIfExists(ruleId, baselineName, cancellationToken: cancellationToken);
return(Response.FromValue(response.Value != null, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual Response <bool> Exists(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(ruleId, nameof(ruleId));
using var scope = _managedInstanceDatabaseVulnerabilityAssessmentRuleBaselineManagedDatabaseVulnerabilityAssessmentRuleBaselinesClientDiagnostics.CreateScope("ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineCollection.Exists");
scope.Start();
try
{
var response = _managedInstanceDatabaseVulnerabilityAssessmentRuleBaselineManagedDatabaseVulnerabilityAssessmentRuleBaselinesRestClient.Get(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, cancellationToken: cancellationToken);
return(Response.FromValue(response.Value != null, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual async Task <Response <bool> > ExistsAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(ruleId, nameof(ruleId));
using var scope = _managedInstanceDatabaseVulnerabilityAssessmentRuleBaselineManagedDatabaseVulnerabilityAssessmentRuleBaselinesClientDiagnostics.CreateScope("ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineCollection.Exists");
scope.Start();
try
{
var response = await GetIfExistsAsync(ruleId, baselineName, cancellationToken : cancellationToken).ConfigureAwait(false);
return(Response.FromValue(response.Value != null, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual Response <bool> CheckIfExists(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var scope = _clientDiagnostics.CreateScope("ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineCollection.CheckIfExists");
scope.Start();
try
{
var response = GetIfExists(ruleId, baselineName, cancellationToken: cancellationToken);
return(Response.FromValue(response.Value != null, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
private void ValidateBaselineRule(SqlManagementClient sqlClient, ResourceGroup resourceGroup, ManagedInstance managedInstance, string dbName, bool isServerLevelRule)
{
string testRuleId = "VA2031";
var ruleBaselineToSet = new DatabaseVulnerabilityAssessmentRuleBaseline(
new List <DatabaseVulnerabilityAssessmentRuleBaselineItem>()
{
new DatabaseVulnerabilityAssessmentRuleBaselineItem()
{
Result = new string[] { "userA", "SELECT" }
},
new DatabaseVulnerabilityAssessmentRuleBaselineItem()
{
Result = new string[] { "userB", "SELECT" }
}
});
VulnerabilityAssessmentPolicyBaselineName baselineName = isServerLevelRule
? VulnerabilityAssessmentPolicyBaselineName.Master
: VulnerabilityAssessmentPolicyBaselineName.Default;
sqlClient.ManagedDatabaseVulnerabilityAssessmentRuleBaselines.CreateOrUpdate(resourceGroup.Name, managedInstance.Name, dbName, testRuleId, baselineName, ruleBaselineToSet);
// Get baseline
DatabaseVulnerabilityAssessmentRuleBaseline actualBaseline = sqlClient.ManagedDatabaseVulnerabilityAssessmentRuleBaselines.Get(resourceGroup.Name, managedInstance.Name, dbName, testRuleId, baselineName);
Assert.Equal(ruleBaselineToSet.BaselineResults.Count(), actualBaseline.BaselineResults.Count());
SqlManagementTestUtilities.AssertCollection(ruleBaselineToSet.BaselineResults[0].Result, actualBaseline.BaselineResults[0].Result);
SqlManagementTestUtilities.AssertCollection(ruleBaselineToSet.BaselineResults[1].Result, actualBaseline.BaselineResults[1].Result);
// Delete baseline
sqlClient.ManagedDatabaseVulnerabilityAssessmentRuleBaselines.Delete(resourceGroup.Name, managedInstance.Name, dbName, testRuleId, baselineName);
// Get baseline should fail after deleting the baseline
Assert.Throws <CloudException>(() =>
{
sqlClient.ManagedDatabaseVulnerabilityAssessmentRuleBaselines.Get(resourceGroup.Name, managedInstance.Name, dbName, testRuleId, baselineName);
});
}
/// <summary>
/// Removes the database's vulnerability assessment rule baseline.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group. The name is case insensitive.
/// </param>
/// <param name='workspaceName'>
/// The name of the workspace.
/// </param>
/// <param name='sqlPoolName'>
/// SQL pool name
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a Sql pool level rule and master for workspace level rule).
/// Possible values include: 'master', 'default'
/// </param>
public static void Delete(this ISqlPoolVulnerabilityAssessmentRuleBaselinesOperations operations, string resourceGroupName, string workspaceName, string sqlPoolName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName)
{
operations.DeleteAsync(resourceGroupName, workspaceName, sqlPoolName, ruleId, baselineName).GetAwaiter().GetResult();
}
internal HttpMessage CreateGetRequest(string resourceGroupName, string serverName, string databaseName, VulnerabilityAssessmentName vulnerabilityAssessmentName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName)
{
var message = _pipeline.CreateMessage();
var request = message.Request;
request.Method = RequestMethod.Get;
var uri = new RawRequestUriBuilder();
uri.Reset(endpoint);
uri.AppendPath("/subscriptions/", false);
uri.AppendPath(subscriptionId, true);
uri.AppendPath("/resourceGroups/", false);
uri.AppendPath(resourceGroupName, true);
uri.AppendPath("/providers/Microsoft.Sql/servers/", false);
uri.AppendPath(serverName, true);
uri.AppendPath("/databases/", false);
uri.AppendPath(databaseName, true);
uri.AppendPath("/vulnerabilityAssessments/", false);
uri.AppendPath(vulnerabilityAssessmentName.ToString(), true);
uri.AppendPath("/rules/", false);
uri.AppendPath(ruleId, true);
uri.AppendPath("/baselines/", false);
uri.AppendPath(baselineName.ToSerialString(), true);
uri.AppendQuery("api-version", "2017-03-01-preview", true);
request.Uri = uri;
return(message);
}
public virtual ServerDatabaseVulnerabilityAssessmentRuleBaselineCreateOrUpdateOperation CreateOrUpdate(bool waitForCompletion, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, DatabaseVulnerabilityAssessmentRuleBaselineData parameters, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(ruleId, nameof(ruleId));
if (parameters == null)
{
throw new ArgumentNullException(nameof(parameters));
}
using var scope = _serverDatabaseVulnerabilityAssessmentRuleBaselineDatabaseVulnerabilityAssessmentRuleBaselinesClientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.CreateOrUpdate");
scope.Start();
try
{
var response = _serverDatabaseVulnerabilityAssessmentRuleBaselineDatabaseVulnerabilityAssessmentRuleBaselinesRestClient.CreateOrUpdate(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, parameters, cancellationToken);
var operation = new ServerDatabaseVulnerabilityAssessmentRuleBaselineCreateOrUpdateOperation(ArmClient, response);
if (waitForCompletion)
{
operation.WaitForCompletion(cancellationToken);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
internal HttpMessage CreateGetRequest(string subscriptionId, string resourceGroupName, string managedInstanceName, string databaseName, VulnerabilityAssessmentName vulnerabilityAssessmentName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName)
{
var message = _pipeline.CreateMessage();
var request = message.Request;
request.Method = RequestMethod.Get;
var uri = new RawRequestUriBuilder();
uri.Reset(endpoint);
uri.AppendPath("/subscriptions/", false);
uri.AppendPath(subscriptionId, true);
uri.AppendPath("/resourceGroups/", false);
uri.AppendPath(resourceGroupName, true);
uri.AppendPath("/providers/Microsoft.Sql/managedInstances/", false);
uri.AppendPath(managedInstanceName, true);
uri.AppendPath("/databases/", false);
uri.AppendPath(databaseName, true);
uri.AppendPath("/vulnerabilityAssessments/", false);
uri.AppendPath(vulnerabilityAssessmentName.ToString(), true);
uri.AppendPath("/rules/", false);
uri.AppendPath(ruleId, true);
uri.AppendPath("/baselines/", false);
uri.AppendPath(baselineName.ToSerialString(), true);
uri.AppendQuery("api-version", "2020-11-01-preview", true);
request.Uri = uri;
request.Headers.Add("Accept", "application/json");
message.SetProperty("UserAgentOverride", _userAgent);
return(message);
}
public virtual async Task <Response <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineResource> > GetManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
return(await GetManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselines().GetAsync(ruleId, baselineName, cancellationToken).ConfigureAwait(false));
}
/// <summary>
/// Gets a database's vulnerability assessment rule baseline.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group that contains the resource. You can obtain
/// this value from the Azure Resource Manager API or the portal.
/// </param>
/// <param name='serverName'>
/// The name of the server.
/// </param>
/// <param name='databaseName'>
/// The name of the database for which the vulnerability assessment rule
/// baseline is defined.
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a database level rule and master for server level rule).
/// Possible values include: 'master', 'default'
/// </param>
public static DatabaseVulnerabilityAssessmentRuleBaseline Get(this IDatabaseVulnerabilityAssessmentRuleBaselinesOperations operations, string resourceGroupName, string serverName, string databaseName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName)
{
return(operations.GetAsync(resourceGroupName, serverName, databaseName, ruleId, baselineName).GetAwaiter().GetResult());
}
public virtual Response <DatabaseVulnerabilityAssessmentRuleBaseline> Get(string resourceGroupName, string managedInstanceName, string databaseName, VulnerabilityAssessmentName vulnerabilityAssessmentName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
using var scope = _clientDiagnostics.CreateScope("ManagedDatabaseVulnerabilityAssessmentRuleBaselinesOperations.Get");
scope.Start();
try
{
return(RestClient.Get(resourceGroupName, managedInstanceName, databaseName, vulnerabilityAssessmentName, ruleId, baselineName, cancellationToken));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public async virtual Task <Response <bool> > CheckIfExistsAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var scope = _clientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.CheckIfExistsAsync");
scope.Start();
try
{
var response = await GetIfExistsAsync(ruleId, baselineName, cancellationToken : cancellationToken).ConfigureAwait(false);
return(Response.FromValue(response.Value != null, response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual Response <ServerDatabaseVulnerabilityAssessmentRuleBaseline> Get(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var scope = _clientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.Get");
scope.Start();
try
{
var response = _databaseVulnerabilityAssessmentRuleBaselinesRestClient.Get(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, cancellationToken);
if (response.Value == null)
{
throw _clientDiagnostics.CreateRequestFailedException(response.GetRawResponse());
}
return(Response.FromValue(new ServerDatabaseVulnerabilityAssessmentRuleBaseline(Parent, response.Value), response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual async Task <ArmOperation <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineResource> > CreateOrUpdateAsync(WaitUntil waitUntil, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, DatabaseVulnerabilityAssessmentRuleBaselineData data, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(ruleId, nameof(ruleId));
Argument.AssertNotNull(data, nameof(data));
using var scope = _managedInstanceDatabaseVulnerabilityAssessmentRuleBaselineManagedDatabaseVulnerabilityAssessmentRuleBaselinesClientDiagnostics.CreateScope("ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineCollection.CreateOrUpdate");
scope.Start();
try
{
var response = await _managedInstanceDatabaseVulnerabilityAssessmentRuleBaselineManagedDatabaseVulnerabilityAssessmentRuleBaselinesRestClient.CreateOrUpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, data, cancellationToken).ConfigureAwait(false);
var operation = new SqlArmOperation <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineResource>(Response.FromValue(new ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineResource(Client, response), response.GetRawResponse()));
if (waitUntil == WaitUntil.Completed)
{
await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
/// <summary>
/// Gets a database's vulnerability assessment rule baseline.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group that contains the resource. You can obtain
/// this value from the Azure Resource Manager API or the portal.
/// </param>
/// <param name='serverName'>
/// The name of the server.
/// </param>
/// <param name='databaseName'>
/// The name of the database for which the vulnerability assessment rule
/// baseline is defined.
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a database level rule and master for server level rule).
/// Possible values include: 'master', 'default'
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <DatabaseVulnerabilityAssessmentRuleBaseline> GetAsync(this IDatabaseVulnerabilityAssessmentRuleBaselinesOperations operations, string resourceGroupName, string serverName, string databaseName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.GetWithHttpMessagesAsync(resourceGroupName, serverName, databaseName, ruleId, baselineName, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
public virtual async Task <Response <ServerDatabaseVulnerabilityAssessmentRuleBaselineResource> > GetAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(ruleId, nameof(ruleId));
using var scope = _serverDatabaseVulnerabilityAssessmentRuleBaselineDatabaseVulnerabilityAssessmentRuleBaselinesClientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.Get");
scope.Start();
try
{
var response = await _serverDatabaseVulnerabilityAssessmentRuleBaselineDatabaseVulnerabilityAssessmentRuleBaselinesRestClient.GetAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, cancellationToken).ConfigureAwait(false);
if (response.Value == null)
{
throw new RequestFailedException(response.GetRawResponse());
}
return(Response.FromValue(new ServerDatabaseVulnerabilityAssessmentRuleBaselineResource(Client, response.Value), response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public async virtual Task <Response <ServerDatabaseVulnerabilityAssessmentRuleBaseline> > GetIfExistsAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var scope = _clientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.GetIfExistsAsync");
scope.Start();
try
{
var response = await _databaseVulnerabilityAssessmentRuleBaselinesRestClient.GetAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, cancellationToken : cancellationToken).ConfigureAwait(false);
return(response.Value == null
? Response.FromValue <ServerDatabaseVulnerabilityAssessmentRuleBaseline>(null, response.GetRawResponse())
: Response.FromValue(new ServerDatabaseVulnerabilityAssessmentRuleBaseline(this, response.Value), response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
/// <summary>
/// Removes the database's vulnerability assessment rule baseline.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group. The name is case insensitive.
/// </param>
/// <param name='workspaceName'>
/// The name of the workspace.
/// </param>
/// <param name='sqlPoolName'>
/// SQL pool name
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a Sql pool level rule and master for workspace level rule).
/// Possible values include: 'master', 'default'
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task DeleteAsync(this ISqlPoolVulnerabilityAssessmentRuleBaselinesOperations operations, string resourceGroupName, string workspaceName, string sqlPoolName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default(CancellationToken))
{
(await operations.DeleteWithHttpMessagesAsync(resourceGroupName, workspaceName, sqlPoolName, ruleId, baselineName, null, cancellationToken).ConfigureAwait(false)).Dispose();
}
public async virtual Task <DatabaseVulnerabilityAssessmentRuleBaselineCreateOrUpdateOperation> CreateOrUpdateAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, DatabaseVulnerabilityAssessmentRuleBaselineData parameters, bool waitForCompletion = true, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
if (parameters == null)
{
throw new ArgumentNullException(nameof(parameters));
}
using var scope = _clientDiagnostics.CreateScope("ServerDatabaseVulnerabilityAssessmentRuleBaselineCollection.CreateOrUpdate");
scope.Start();
try
{
var response = await _databaseVulnerabilityAssessmentRuleBaselinesRestClient.CreateOrUpdateAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, parameters, cancellationToken).ConfigureAwait(false);
var operation = new DatabaseVulnerabilityAssessmentRuleBaselineCreateOrUpdateOperation(Parent, response);
if (waitForCompletion)
{
await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
/// <summary>
/// Creates or updates a Sql pool's vulnerability assessment rule baseline.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group. The name is case insensitive.
/// </param>
/// <param name='workspaceName'>
/// The name of the workspace.
/// </param>
/// <param name='sqlPoolName'>
/// SQL pool name
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a Sql pool level rule and master for workspace level rule).
/// Possible values include: 'master', 'default'
/// </param>
/// <param name='parameters'>
/// The requested rule baseline resource.
/// </param>
public static SqlPoolVulnerabilityAssessmentRuleBaseline CreateOrUpdate(this ISqlPoolVulnerabilityAssessmentRuleBaselinesOperations operations, string resourceGroupName, string workspaceName, string sqlPoolName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, SqlPoolVulnerabilityAssessmentRuleBaseline parameters)
{
return(operations.CreateOrUpdateAsync(resourceGroupName, workspaceName, sqlPoolName, ruleId, baselineName, parameters).GetAwaiter().GetResult());
}
/// <summary>
/// Gets a database's vulnerability assessment rule baseline.
/// </summary>
/// <param name='resourceGroupName'>
/// The name of the resource group that contains the resource. You can obtain
/// this value from the Azure Resource Manager API or the portal.
/// </param>
/// <param name='serverName'>
/// The name of the server.
/// </param>
/// <param name='databaseName'>
/// The name of the database for which the vulnerability assessment rule
/// baseline is defined.
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a database level rule and master for server level rule).
/// Possible values include: 'master', 'default'
/// </param>
/// <param name='customHeaders'>
/// Headers that will be added to request.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
/// <exception cref="CloudException">
/// Thrown when the operation returned an invalid status code
/// </exception>
/// <exception cref="SerializationException">
/// Thrown when unable to deserialize the response
/// </exception>
/// <exception cref="ValidationException">
/// Thrown when a required parameter is null
/// </exception>
/// <exception cref="System.ArgumentNullException">
/// Thrown when a required parameter is null
/// </exception>
/// <return>
/// A response object containing the response body and response headers.
/// </return>
public async Task <AzureOperationResponse <DatabaseVulnerabilityAssessmentRuleBaselineInner> > GetWithHttpMessagesAsync(string resourceGroupName, string serverName, string databaseName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, Dictionary <string, List <string> > customHeaders = null, CancellationToken cancellationToken = default(CancellationToken))
{
if (resourceGroupName == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "resourceGroupName");
}
if (serverName == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "serverName");
}
if (databaseName == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "databaseName");
}
if (ruleId == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "ruleId");
}
if (Client.SubscriptionId == null)
{
throw new ValidationException(ValidationRules.CannotBeNull, "this.Client.SubscriptionId");
}
string vulnerabilityAssessmentName = "default";
string apiVersion = "2017-03-01-preview";
// Tracing
bool _shouldTrace = ServiceClientTracing.IsEnabled;
string _invocationId = null;
if (_shouldTrace)
{
_invocationId = ServiceClientTracing.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("resourceGroupName", resourceGroupName);
tracingParameters.Add("serverName", serverName);
tracingParameters.Add("databaseName", databaseName);
tracingParameters.Add("vulnerabilityAssessmentName", vulnerabilityAssessmentName);
tracingParameters.Add("ruleId", ruleId);
tracingParameters.Add("baselineName", baselineName);
tracingParameters.Add("apiVersion", apiVersion);
tracingParameters.Add("cancellationToken", cancellationToken);
ServiceClientTracing.Enter(_invocationId, this, "Get", tracingParameters);
}
// Construct URL
var _baseUrl = Client.BaseUri.AbsoluteUri;
var _url = new System.Uri(new System.Uri(_baseUrl + (_baseUrl.EndsWith("/") ? "" : "/")), "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}/databases/{databaseName}/vulnerabilityAssessments/{vulnerabilityAssessmentName}/rules/{ruleId}/baselines/{baselineName}").ToString();
_url = _url.Replace("{resourceGroupName}", System.Uri.EscapeDataString(resourceGroupName));
_url = _url.Replace("{serverName}", System.Uri.EscapeDataString(serverName));
_url = _url.Replace("{databaseName}", System.Uri.EscapeDataString(databaseName));
_url = _url.Replace("{vulnerabilityAssessmentName}", System.Uri.EscapeDataString(vulnerabilityAssessmentName));
_url = _url.Replace("{ruleId}", System.Uri.EscapeDataString(ruleId));
_url = _url.Replace("{baselineName}", System.Uri.EscapeDataString(Rest.Serialization.SafeJsonConvert.SerializeObject(baselineName, Client.SerializationSettings).Trim('"')));
_url = _url.Replace("{subscriptionId}", System.Uri.EscapeDataString(Client.SubscriptionId));
List <string> _queryParameters = new List <string>();
if (apiVersion != null)
{
_queryParameters.Add(string.Format("api-version={0}", System.Uri.EscapeDataString(apiVersion)));
}
if (_queryParameters.Count > 0)
{
_url += (_url.Contains("?") ? "&" : "?") + string.Join("&", _queryParameters);
}
// Create HTTP transport objects
var _httpRequest = new HttpRequestMessage();
HttpResponseMessage _httpResponse = null;
_httpRequest.Method = new HttpMethod("GET");
_httpRequest.RequestUri = new System.Uri(_url);
// Set Headers
if (Client.GenerateClientRequestId != null && Client.GenerateClientRequestId.Value)
{
_httpRequest.Headers.TryAddWithoutValidation("x-ms-client-request-id", System.Guid.NewGuid().ToString());
}
if (Client.AcceptLanguage != null)
{
if (_httpRequest.Headers.Contains("accept-language"))
{
_httpRequest.Headers.Remove("accept-language");
}
_httpRequest.Headers.TryAddWithoutValidation("accept-language", Client.AcceptLanguage);
}
if (customHeaders != null)
{
foreach (var _header in customHeaders)
{
if (_httpRequest.Headers.Contains(_header.Key))
{
_httpRequest.Headers.Remove(_header.Key);
}
_httpRequest.Headers.TryAddWithoutValidation(_header.Key, _header.Value);
}
}
// Serialize Request
string _requestContent = null;
// Set Credentials
if (Client.Credentials != null)
{
cancellationToken.ThrowIfCancellationRequested();
await Client.Credentials.ProcessHttpRequestAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
}
// Send Request
if (_shouldTrace)
{
ServiceClientTracing.SendRequest(_invocationId, _httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
_httpResponse = await Client.HttpClient.SendAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
if (_shouldTrace)
{
ServiceClientTracing.ReceiveResponse(_invocationId, _httpResponse);
}
HttpStatusCode _statusCode = _httpResponse.StatusCode;
cancellationToken.ThrowIfCancellationRequested();
string _responseContent = null;
if ((int)_statusCode != 200)
{
var ex = new CloudException(string.Format("Operation returned an invalid status code '{0}'", _statusCode));
try
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
CloudError _errorBody = Rest.Serialization.SafeJsonConvert.DeserializeObject <CloudError>(_responseContent, Client.DeserializationSettings);
if (_errorBody != null)
{
ex = new CloudException(_errorBody.Message);
ex.Body = _errorBody;
}
}
catch (JsonException)
{
// Ignore the exception
}
ex.Request = new HttpRequestMessageWrapper(_httpRequest, _requestContent);
ex.Response = new HttpResponseMessageWrapper(_httpResponse, _responseContent);
if (_httpResponse.Headers.Contains("x-ms-request-id"))
{
ex.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
if (_shouldTrace)
{
ServiceClientTracing.Error(_invocationId, ex);
}
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw ex;
}
// Create Result
var _result = new AzureOperationResponse <DatabaseVulnerabilityAssessmentRuleBaselineInner>();
_result.Request = _httpRequest;
_result.Response = _httpResponse;
if (_httpResponse.Headers.Contains("x-ms-request-id"))
{
_result.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
// Deserialize Response
if ((int)_statusCode == 200)
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
try
{
_result.Body = Rest.Serialization.SafeJsonConvert.DeserializeObject <DatabaseVulnerabilityAssessmentRuleBaselineInner>(_responseContent, Client.DeserializationSettings);
}
catch (JsonException ex)
{
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw new SerializationException("Unable to deserialize the response.", _responseContent, ex);
}
}
if (_shouldTrace)
{
ServiceClientTracing.Exit(_invocationId, _result);
}
return(_result);
}
/// <summary>
/// Creates or updates a Sql pool's vulnerability assessment rule baseline.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group. The name is case insensitive.
/// </param>
/// <param name='workspaceName'>
/// The name of the workspace.
/// </param>
/// <param name='sqlPoolName'>
/// SQL pool name
/// </param>
/// <param name='ruleId'>
/// The vulnerability assessment rule ID.
/// </param>
/// <param name='baselineName'>
/// The name of the vulnerability assessment rule baseline (default implies a
/// baseline on a Sql pool level rule and master for workspace level rule).
/// Possible values include: 'master', 'default'
/// </param>
/// <param name='parameters'>
/// The requested rule baseline resource.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <SqlPoolVulnerabilityAssessmentRuleBaseline> CreateOrUpdateAsync(this ISqlPoolVulnerabilityAssessmentRuleBaselinesOperations operations, string resourceGroupName, string workspaceName, string sqlPoolName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, SqlPoolVulnerabilityAssessmentRuleBaseline parameters, CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.CreateOrUpdateWithHttpMessagesAsync(resourceGroupName, workspaceName, sqlPoolName, ruleId, baselineName, parameters, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
public virtual async Task <Response <DatabaseVulnerabilityAssessmentRuleBaseline> > CreateOrUpdateAsync(string resourceGroupName, string managedInstanceName, string databaseName, VulnerabilityAssessmentName vulnerabilityAssessmentName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, DatabaseVulnerabilityAssessmentRuleBaseline parameters, CancellationToken cancellationToken = default)
{
using var scope = _clientDiagnostics.CreateScope("ManagedDatabaseVulnerabilityAssessmentRuleBaselinesOperations.CreateOrUpdate");
scope.Start();
try
{
return(await RestClient.CreateOrUpdateAsync(resourceGroupName, managedInstanceName, databaseName, vulnerabilityAssessmentName, ruleId, baselineName, parameters, cancellationToken).ConfigureAwait(false));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public static ResourceIdentifier CreateResourceIdentifier(string subscriptionId, string resourceGroupName, string serverName, string databaseName, VulnerabilityAssessmentName vulnerabilityAssessmentName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName)
{
var resourceId = $"/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}/databases/{databaseName}/vulnerabilityAssessments/{vulnerabilityAssessmentName}/rules/{ruleId}/baselines/{baselineName}";
return(new ResourceIdentifier(resourceId));
}
public async Task <Response <DatabaseVulnerabilityAssessmentRuleBaselineData> > GetAsync(string subscriptionId, string resourceGroupName, string managedInstanceName, string databaseName, VulnerabilityAssessmentName vulnerabilityAssessmentName, string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (subscriptionId == null)
{
throw new ArgumentNullException(nameof(subscriptionId));
}
if (resourceGroupName == null)
{
throw new ArgumentNullException(nameof(resourceGroupName));
}
if (managedInstanceName == null)
{
throw new ArgumentNullException(nameof(managedInstanceName));
}
if (databaseName == null)
{
throw new ArgumentNullException(nameof(databaseName));
}
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var message = CreateGetRequest(subscriptionId, resourceGroupName, managedInstanceName, databaseName, vulnerabilityAssessmentName, ruleId, baselineName);
await _pipeline.SendAsync(message, cancellationToken).ConfigureAwait(false);
switch (message.Response.Status)
{
case 200:
{
DatabaseVulnerabilityAssessmentRuleBaselineData value = default;
using var document = await JsonDocument.ParseAsync(message.Response.ContentStream, default, cancellationToken).ConfigureAwait(false);
value = DatabaseVulnerabilityAssessmentRuleBaselineData.DeserializeDatabaseVulnerabilityAssessmentRuleBaselineData(document.RootElement);
return(Response.FromValue(value, message.Response));
}
public async virtual Task <Response <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaseline> > GetAsync(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var scope = _clientDiagnostics.CreateScope("ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineCollection.Get");
scope.Start();
try
{
var response = await _managedDatabaseVulnerabilityAssessmentRuleBaselinesRestClient.GetAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, cancellationToken).ConfigureAwait(false);
if (response.Value == null)
{
throw await _clientDiagnostics.CreateRequestFailedExceptionAsync(response.GetRawResponse()).ConfigureAwait(false);
}
return(Response.FromValue(new ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaseline(Parent, response.Value), response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual Response <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineResource> GetManagedInstanceDatabaseVulnerabilityAssessmentRuleBaseline(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
return(GetManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselines().Get(ruleId, baselineName, cancellationToken));
}
public virtual Response <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaseline> GetIfExists(string ruleId, VulnerabilityAssessmentPolicyBaselineName baselineName, CancellationToken cancellationToken = default)
{
if (ruleId == null)
{
throw new ArgumentNullException(nameof(ruleId));
}
using var scope = _clientDiagnostics.CreateScope("ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaselineCollection.GetIfExists");
scope.Start();
try
{
var response = _managedDatabaseVulnerabilityAssessmentRuleBaselinesRestClient.Get(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Parent.Name, Id.Parent.Name, Id.Name, ruleId, baselineName, cancellationToken: cancellationToken);
return(response.Value == null
? Response.FromValue <ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaseline>(null, response.GetRawResponse())
: Response.FromValue(new ManagedInstanceDatabaseVulnerabilityAssessmentRuleBaseline(this, response.Value), response.GetRawResponse()));
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
