ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-22 14:19:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

DoNotRunExtensionsOnOverprovisionedVms must be false if Overprovision is false (#2375)

Browse Source
This commit is contained in:
George Pollard
2022-09-13 10:00:47 +12:00
committed by GitHub
parent 9eaf92ea35
commit 44f74f622a
5 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/ApiService/ApiService/onefuzzlib/NodeOperations.cs
View File

@ -235,15 +235,15 @@ public class NodeOperations : StatefulOrm<Node, NodeState, NodeOperations>, INod
List<string> queryParts = new();
if (poolId is not null) {
queryParts.Add($"(pool_id eq '{poolId}')");
queryParts.Add(TableClient.CreateQueryFilter($"(pool_id eq {poolId})"));
}
if (poolName is not null) {
queryParts.Add($"(pool_name eq '{poolName}')");
queryParts.Add(TableClient.CreateQueryFilter($"(pool_name eq {poolName.String})"));
}
if (scalesetId is not null) {
queryParts.Add($"(scaleset_id eq '{scalesetId}')");
queryParts.Add(TableClient.CreateQueryFilter($"(scaleset_id eq {scalesetId})"));
}
if (states is not null) {
@ -469,7 +469,7 @@ public class NodeOperations : StatefulOrm<Node, NodeState, NodeOperations>, INod
}
if (poolName is not null) {
queryParts.Add($"(PartitionKey eq '{poolName}')");
queryParts.Add($"(PartitionKey eq '{poolName.String}')");
}
if (scaleSetId is not null) {
@ -513,7 +513,7 @@ public class NodeOperations : StatefulOrm<Node, NodeState, NodeOperations>, INod
}
public IAsyncEnumerable<Node> SearchByPoolName(PoolName poolName) {
return QueryAsync($"(pool_name eq '{poolName}')");
return QueryAsync(TableClient.CreateQueryFilter($"(pool_name eq {poolName.String})"));
}

3
src/ApiService/ApiService/onefuzzlib/NotificationOperations.cs
View File

@ -1,5 +1,6 @@
using System.Text.Json;
using ApiService.OneFuzzLib.Orm;
using Azure.Data.Tables;
using Azure.Storage.Sas;
namespace Microsoft.OneFuzz.Service;
@ -77,7 +78,7 @@ public class NotificationOperations : Orm<Notification>, INotificationOperations
}
public IAsyncEnumerable<Notification> GetNotifications(Container container) {
return QueryAsync(filter: $"container eq '{container}'");
return QueryAsync(filter: TableClient.CreateQueryFilter($"container eq {container.String}"));
}
public IAsyncEnumerable<(Task, IEnumerable<Container>)> GetQueueTasks() {

11
src/ApiService/ApiService/onefuzzlib/ProxyForwardOperations.cs
View File

@ -1,5 +1,6 @@
using System.Threading.Tasks;
using ApiService.OneFuzzLib.Orm;
using Azure.Data.Tables;
namespace Microsoft.OneFuzz.Service;
@ -24,11 +25,11 @@ public class ProxyForwardOperations : Orm<ProxyForward>, IProxyForwardOperations
var conditions =
new[] {
scalesetId != null ? $"scaleset_id eq '{scalesetId}'" : null,
region != null ? $"PartitionKey eq '{region}'" : null ,
machineId != null ? $"machine_id eq '{machineId}'" : null ,
proxyId != null ? $"proxy_id eq '{proxyId}'" : null ,
dstPort != null ? $"dst_port eq {dstPort}" : null ,
scalesetId is not null ? TableClient.CreateQueryFilter($"scaleset_id eq {scalesetId}") : null,
region is not null ? TableClient.CreateQueryFilter($"PartitionKey eq {region.String}") : null ,
machineId is not null ? TableClient.CreateQueryFilter($"machine_id eq {machineId}") : null ,
proxyId is not null ? TableClient.CreateQueryFilter($"proxy_id eq {proxyId}") : null ,
dstPort is not null ? TableClient.CreateQueryFilter($"dst_port eq {dstPort}") : null ,
}.Where(x => x != null);
var filter = Query.And(conditions!);

3
src/ApiService/ApiService/onefuzzlib/ProxyOperations.cs
View File

@ -1,5 +1,6 @@
using System.Threading.Tasks;
using ApiService.OneFuzzLib.Orm;
using Azure.Data.Tables;
using Azure.ResourceManager.Compute;
using Azure.ResourceManager.Compute.Models;
using Azure.Storage.Sas;
@ -42,7 +43,7 @@ public class ProxyOperations : StatefulOrm<Proxy, VmState, ProxyOperations>, IPr
}
public async Async.Task<Proxy?> GetOrCreate(Region region) {
var proxyList = QueryAsync(filter: $"region eq '{region}' and outdated eq false");
var proxyList = QueryAsync(filter: TableClient.CreateQueryFilter($"region eq {region.String} and outdated eq false"));
await foreach (var proxy in proxyList) {
if (IsOutdated(proxy)) {

4
src/ApiService/ApiService/onefuzzlib/VmssOperations.cs
View File

@ -271,7 +271,7 @@ public class VmssOperations : IVmssOperations {
}
var vmssData = new VirtualMachineScaleSetData(location) {
DoNotRunExtensionsOnOverprovisionedVms = true,
DoNotRunExtensionsOnOverprovisionedVms = false,
UpgradePolicy = new() {
Mode = UpgradeMode.Manual,
},
@ -399,7 +399,7 @@ public class VmssOperations : IVmssOperations {
entry.SetAbsoluteExpiration(TimeSpan.FromMinutes(10));
var sub = _creds.GetSubscriptionResource();
var skus = sub.GetResourceSkusAsync(filter: TableClient.CreateQueryFilter($"location eq '{region}'"));
var skus = sub.GetResourceSkusAsync(filter: TableClient.CreateQueryFilter($"location eq {region.String}"));
var skuNames = new List<string>();
await foreach (var sku in skus) {