Ejemplo n.º 1
0
        public static IActionResult Run(
            [HttpTrigger(AuthorizationLevel.Function, "get", Route = "FileTracker/{filename}")] HttpRequest req,
            string filename,
            ILogger log)
        {
            log.LogInformation($"Retrieving File Tracker for {filename}");

            return(new OkObjectResult(FileTrackerRepository.Get(filename)));
        }
Ejemplo n.º 2
0
        public static async Task <string> Run(
            [ServiceBusTrigger("scan-for-virus")] string myQueueItem,
            [Blob("%AzureStorage:Container%", FileAccess.Read, Connection = "AzureStorage:ConnectionString")] CloudBlobContainer cloudBlobContainer,
            ILogger log)
        {
            log.LogInformation($"Scanning {myQueueItem}");

            var config = new ConfigurationBuilder()
                         .AddEnvironmentVariables()
                         .Build();

            // Download Blob Content
            var cloudBlockBlob = cloudBlobContainer.GetBlockBlobReference(myQueueItem);
            var memoryStream   = new MemoryStream();
            await cloudBlockBlob.DownloadToStreamAsync(memoryStream);

            // Scan for Viruses
            var virusScanner = new CloudmersiveVirusScanner();

            virusScanner.SetConfiguration(config);

            var stopWatch = new Stopwatch();

            stopWatch.Start();
            FileTrackerRepository.AddOperationResult(myQueueItem, "Virus Scan");

            var scanResult = await virusScanner.ScanStreamAsync(memoryStream);

            stopWatch.Stop();

            FileTrackerRepository.UpdateOperationResult(myQueueItem, "Virus Scan", stopWatch.ElapsedMilliseconds, false, scanResult.Message);

            log.LogInformation($"Scan Results for {myQueueItem}, Safe: {scanResult.IsSafe}, Message: {scanResult.Message}");

            return(myQueueItem);
        }
Ejemplo n.º 3
0
        public static async Task <string> Run(
            [ServiceBusTrigger("parse-and-validate")] string myQueueItem,
            [Blob("%AzureStorage:Container%", FileAccess.Read, Connection = "AzureStorage:ConnectionString")] CloudBlobContainer cloudBlobContainer,
            ILogger log)
        {
            log.LogInformation($"Parsing and validating: {myQueueItem}");

            // Download Blob Content
            var cloudBlockBlob = cloudBlobContainer.GetBlockBlobReference(myQueueItem);
            var memoryStream   = new MemoryStream();
            await cloudBlockBlob.DownloadToStreamAsync(memoryStream);

            // Parse and Validate File
            var stopWatch = new Stopwatch();

            stopWatch.Start();
            FileTrackerRepository.AddOperationResult(myQueueItem, "Parse and Validate");

            var parseErrors = CsvFile.Validate(memoryStream);

            stopWatch.Stop();

            FileTrackerRepository.UpdateOperationResult(myQueueItem, "Parse and Validate", stopWatch.ElapsedMilliseconds, true);

            if (parseErrors.Count > 0)
            {
                log.LogInformation($"Parsing and validating results for: {myQueueItem}, {string.Join(",", parseErrors)}");
                throw new ApplicationException("File is not valid");
            }
            else
            {
                log.LogInformation($"Parsing and validating results for: {myQueueItem}, looks good for next stage");
            }

            return(myQueueItem);
        }
        public static async Task <IActionResult> Run(
            [HttpTrigger(AuthorizationLevel.Function, "post", Route = null)] HttpRequest req,
            ILogger log,
            [Blob("%AzureStorage:Container%", FileAccess.Write, Connection = "AzureStorage:ConnectionString")] CloudBlobContainer cloudBlobContainer)
        {
            if (!(await cloudBlobContainer.ExistsAsync()))
            {
                await cloudBlobContainer.CreateAsync();
            }

            var content = await req.ReadFormAsync();

            var blobNames = new List <string>();

            if (content.Files.Count > 0)
            {
                log.LogInformation("File Upload in progress!");

                foreach (var file in content.Files)
                {
                    var blobName       = $"{Guid.NewGuid()}{Path.GetExtension(file.FileName)}".Replace("\"", "");
                    var cloudBlockBlob = cloudBlobContainer.GetBlockBlobReference(blobName);
                    var xff            = req.Headers.FirstOrDefault(x => x.Key == "X-Forwarded-For").Value.FirstOrDefault();

                    blobNames.Add(blobName);
                    cloudBlockBlob.Properties.ContentDisposition = file.ContentDisposition;
                    cloudBlockBlob.Properties.ContentType        = file.ContentType;
                    cloudBlockBlob.Metadata.Add("originalName", file.FileName);
                    if (!string.IsNullOrEmpty(xff))
                    {
                        cloudBlockBlob.Metadata.Add("sourceIp", xff);
                    }

                    log.LogInformation($"Uploading {file.FileName} as {blobName}");

                    var stopWatch = new Stopwatch();
                    stopWatch.Start();
                    using (var fileStream = file.OpenReadStream())
                    {
                        await cloudBlockBlob.UploadFromStreamAsync(fileStream);
                    }
                    stopWatch.Stop();

                    FileTrackerRepository.AddNew(file.FileName, blobName, stopWatch.ElapsedMilliseconds);

                    // Add message in queue for next step
                    var config = new ConfigurationBuilder()
                                 .AddEnvironmentVariables()
                                 .Build();
                    var serviceBusConnection = config["AzureWebJobsServiceBus"];
                    var queueName            = "scan-for-virus";

                    var messageSender = new MessageSender(serviceBusConnection, queueName);
                    var message       = new Message(Encoding.UTF8.GetBytes(blobName));

                    await messageSender.SendAsync(message).ConfigureAwait(false);

                    log.LogInformation($"Added message in {queueName}!");
                }

                return(new OkObjectResult(new { blobs = blobNames }));
            }
            else
            {
                log.LogError("No Files to Upload!");

                return(new BadRequestObjectResult("No files found in request."));
            }
        }