Exemple #1
0
        public static async Task <IActionResult> Run(
            [HttpTrigger(AuthorizationLevel.Function, "get", "post", Route = null)] HttpRequest req,
            ILogger log, Microsoft.Azure.WebJobs.ExecutionContext context)
        {
            // Log the start
            log.LogInformation($"C# Timer trigger function executed at: {DateTime.Now}");

            // Initialize the configuration object
            var config = new ConfigurationBuilder()
                         .SetBasePath(context.FunctionAppDirectory)
                         .AddJsonFile("local.settings.json", optional: true, reloadOnChange: true)
                         .AddEnvironmentVariables()
                         .Build();

            // Get the storage account information
            string storageAccountName = config["StorageAccountName"];
            string storageAccountKey  = config["StorageAccountKey"];

            // Get the parallelization configuration
            int maxThreads = 0;

            try
            {
                int.TryParse(config["Threads"], out maxThreads);
            }
            catch (Exception e)
            {
                maxThreads = 20;
            }

            // Get the values from the request
            string ip = req.Query["IP"];

            // Validate the input
            if (ip == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to get host. IP cannot be null."));
            }

            // Create the resources
            Azure az = new Azure(storageAccountName, storageAccountKey);

            // Check if the host exists
            if (!az.ExistsHost(ip))
            {
                return((ActionResult) new BadRequestObjectResult("The supplied host does not exist."));
            }

            // Read the RBLs from the configuration
            List <RBL> MyRBLs = az.GetRBLs().Where(x => x.Type.ToUpper().Equals("IP")).ToList();

            // Process each host
            log.LogInformation($"Processing host: {ip}");

            // The results
            List <RBLHostResult> results = new List <RBLHostResult>();

            // Create the tasks
            List <Task <RBLHostResult> > tasks = new List <Task <RBLHostResult> >();

            foreach (RBL r in MyRBLs)
            {
                tasks.Add(new Task <RBLHostResult>(() => r.QueryHost(ip)));
            }

            // Start the first batch of tasks
            for (int i = 0; i < maxThreads && i < tasks.Count; i++)
            {
                tasks[i].Start();
            }

            // While there are tasks that are not completed,
            while (tasks.Count > 0)
            {
                // Wait for a task to complete
                var t = await Task.WhenAny(tasks);

                tasks.Remove(t);

                // Process the result
                try
                {
                    results.Add(t.Result);
                }
                catch (OperationCanceledException) { }
                catch (Exception exc) { }

                // Start the next task
                if (tasks.Count > 0)
                {
                    foreach (Task <RBLHostResult> nextTask in tasks)
                    {
                        if (nextTask.Status == TaskStatus.Created)
                        {
                            nextTask.Start();
                            break;
                        }
                    }
                }
            }

            // Wait for all tasks (in case we missed one).
            Task.WaitAll(tasks.ToArray());

            // Log the end
            log.LogInformation($"C# Timer trigger function completed at: {DateTime.Now}");

            // Return the result
            var jsonToReturn = JsonConvert.SerializeObject(results);

            return((ActionResult) new OkObjectResult(jsonToReturn));
        }
Exemple #2
0
        public static async Task <IActionResult> Run(
            [HttpTrigger(AuthorizationLevel.Function, "get", "post", Route = null)] HttpRequest req,
            ILogger log, ExecutionContext context)
        {
            // Log the start
            log.LogInformation($"C# Timer trigger function executed at: {DateTime.Now}");

            // Initialize the configuration object
            var config = new ConfigurationBuilder()
                         .SetBasePath(context.FunctionAppDirectory)
                         .AddJsonFile("local.settings.json", optional: true, reloadOnChange: true)
                         .AddEnvironmentVariables()
                         .Build();

            // Get the storage account information
            string storageAccountName = config["StorageAccountName"];
            string storageAccountKey  = config["StorageAccountKey"];

            // Check the supplied values
            if (storageAccountKey == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to read the storage account key from the configuration."));
            }
            if (storageAccountName == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to read the storage account name from the configuration."));
            }

            // Create the resources
            Azure az;

            try
            {
                az = new Azure(storageAccountName, storageAccountKey);
            }
            catch (Exception e)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to initialize the Azure interface. " + e.Message));
            }

            // Get the values from the request
            string fqdn = req.Query["FQDN"];

            // Validate the input
            if (fqdn == null)
            {
                return((ActionResult) new BadRequestObjectResult("The RBL FQDN cannot be null."));
            }

            // Remove the host
            try
            {
                az.RemoveRBL(fqdn);
            }
            catch (Exception e)
            {
                return((ActionResult) new BadRequestObjectResult("Failed remove the RBL. " + e.Message));
            }

            // Return the result
            return((ActionResult) new OkObjectResult($"The RBL {fqdn} has been removed"));
        }
Exemple #3
0
        public static async Task <IActionResult> Run(
            [HttpTrigger(AuthorizationLevel.Function, "get", "post", Route = null)] HttpRequest req,
            ILogger log, ExecutionContext context)
        {
            // Log the start
            log.LogInformation($"C# Timer trigger function executed at: {DateTime.Now}");

            // Initialize the configuration object
            var config = new ConfigurationBuilder()
                         .SetBasePath(context.FunctionAppDirectory)
                         .AddJsonFile("local.settings.json", optional: true, reloadOnChange: true)
                         .AddEnvironmentVariables()
                         .Build();

            // Get the storage account information
            string storageAccountName = config["StorageAccountName"];
            string storageAccountKey  = config["StorageAccountKey"];

            // Check the supplied values
            if (storageAccountKey == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to read the storage account key from the configuration."));
            }
            if (storageAccountName == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to read the storage account name from the configuration."));
            }

            // Create the resources
            Azure az;

            try
            {
                az = new Azure(storageAccountName, storageAccountKey);
            }
            catch (Exception e)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to initialize the Azure interface. " + e.Message));
            }

            // Get the values from the request
            string hostname = req.Query["Hostname"];
            string ip       = req.Query["IP"];

            // Validate the input
            if (hostname == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to add the host. The supplied hostname is not valid."));
            }
            if (ip == null)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to add the host. The supplied IP is not valid."));
            }
            Regex ipRegex = new Regex(@"\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b");

            if (!ipRegex.IsMatch(ip))
            {
                return((ActionResult) new BadRequestObjectResult("Failed to add the host. The supplied IP is not valid."));
            }

            // Add the host
            try
            {
                az.AddHost(hostname, ip);
            }
            catch (Exception e)
            {
                return((ActionResult) new BadRequestObjectResult("Failed to add the host. " + e.Message));
            }

            // Return a success result
            return(new OkObjectResult($"The host {hostname} - {ip} has been added."));
        }