// /*************************************************************************** // Aaru Data Preservation Suite // ---------------------------------------------------------------------------- // // Filename : UploadStatsController.cs // Author(s) : Natalia Portillo // // Component : Aaru Server. // // --[ Description ] ---------------------------------------------------------- // // Handles statistics uploads. // // --[ License ] -------------------------------------------------------------- // // This library is free software; you can redistribute it and/or modify // it under the terms of the GNU Lesser General Public License as // published by the Free Software Foundation; either version 2.1 of the // License, or (at your option) any later version. // // This library is distributed in the hope that it will be useful, but // WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU // Lesser General Public License for more details. // // You should have received a copy of the GNU Lesser General Public // License along with this library; if not, see . // // ---------------------------------------------------------------------------- // Copyright © 2011-2024 Natalia Portillo // ****************************************************************************/ using System.Diagnostics; using System.IO; using System.Net; using System.Xml.Serialization; using Aaru.CommonTypes.Metadata; using Aaru.Server.Old.Database.Models; using Aaru.Server.Old.Core; using Microsoft.AspNetCore.Http; using Newtonsoft.Json; using DbContext = Aaru.Server.Old.Database.DbContext; using OperatingSystem = Aaru.Server.Old.Database.Models.OperatingSystem; using Version = Aaru.Server.Old.Database.Models.Version; namespace Aaru.Server.Old.Controllers; public sealed class UploadStatsController : Controller { readonly DbContext _ctx; public UploadStatsController(DbContext ctx) => _ctx = ctx; /// Receives statistics from Aaru.Core, processes them and adds them to a server-side global statistics XML /// HTTP response [Route("api/uploadstats")] [HttpPost] public async Task UploadStats() { var response = new ContentResult { StatusCode = (int)HttpStatusCode.OK, ContentType = "text/plain" }; try { var newStats = new Stats(); HttpRequest request = HttpContext.Request; var xs = new XmlSerializer(newStats.GetType()); newStats = (Stats)xs.Deserialize(new StringReader(await new StreamReader(request.Body).ReadToEndAsync())); if(newStats == null) { response.Content = "notstats"; return response; } StatsConverter.Convert(newStats); response.Content = "ok"; return response; } catch(Exception) { #if DEBUG if(Debugger.IsAttached) throw; #endif response.Content = "error"; return response; } } /// Receives a report from Aaru.Core, verifies it's in the correct format and stores it on the server /// HTTP response [Route("api/uploadstatsv2")] [HttpPost] public async Task UploadStatsV2() { var response = new ContentResult { StatusCode = (int)HttpStatusCode.OK, ContentType = "text/plain" }; try { HttpRequest request = HttpContext.Request; var sr = new StreamReader(request.Body); string statsString = await sr.ReadToEndAsync(); StatsDto newstats = JsonConvert.DeserializeObject(statsString); if(newstats == null) { response.Content = "notstats"; return response; } if(newstats.Commands != null) { foreach(NameValueStats nvs in newstats.Commands) { if(nvs.name == "analyze") nvs.name = "fs-info"; Command existing = _ctx.Commands.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.Commands.AddAsync(new Command { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.Versions != null) { foreach(NameValueStats nvs in newstats.Versions) { Version existing = _ctx.Versions.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.Versions.AddAsync(new Version { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.Filesystems != null) { foreach(NameValueStats nvs in newstats.Filesystems) { Filesystem existing = _ctx.Filesystems.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.Filesystems.AddAsync(new Filesystem { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.Partitions != null) { foreach(NameValueStats nvs in newstats.Partitions) { Partition existing = _ctx.Partitions.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.Partitions.AddAsync(new Partition { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.MediaFormats != null) { foreach(NameValueStats nvs in newstats.MediaFormats) { MediaFormat existing = _ctx.MediaFormats.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.MediaFormats.AddAsync(new MediaFormat { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.Filters != null) { foreach(NameValueStats nvs in newstats.Filters) { Filter existing = _ctx.Filters.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.Filters.AddAsync(new Filter { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.OperatingSystems != null) { foreach(OsStats operatingSystem in newstats.OperatingSystems) { OperatingSystem existing = _ctx.OperatingSystems.FirstOrDefault(c => c.Name == operatingSystem.name && c.Version == operatingSystem.version); if(existing == null) { await _ctx.OperatingSystems.AddAsync(new OperatingSystem { Name = operatingSystem.name, Version = operatingSystem.version, Count = operatingSystem.Value }); } else existing.Count += operatingSystem.Value; } } if(newstats.Medias != null) { foreach(MediaStats media in newstats.Medias) { Media existing = _ctx.Medias.FirstOrDefault(c => c.Type == media.type && c.Real == media.real); if(existing == null) { await _ctx.Medias.AddAsync(new Media { Type = media.type, Real = media.real, Count = media.Value }); } else existing.Count += media.Value; } } if(newstats.Devices != null) { foreach(DeviceStats device in from device in newstats.Devices let existing = _ctx.DeviceStats.FirstOrDefault(c => c.Bus == device.Bus && c.Manufacturer == device.Manufacturer && c.Model == device.Model && c.Revision == device.Revision) where existing == null select device) { await _ctx.DeviceStats.AddAsync(new DeviceStat { Bus = device.Bus, Manufacturer = device.Manufacturer, Model = device.Model, Revision = device.Revision }); } } if(newstats.RemoteApplications != null) { foreach(OsStats application in newstats.RemoteApplications) { RemoteApplication existing = _ctx.RemoteApplications.FirstOrDefault(c => c.Name == application.name && c.Version == application.version); if(existing == null) { await _ctx.RemoteApplications.AddAsync(new RemoteApplication { Name = application.name, Version = application.version, Count = application.Value }); } else existing.Count += application.Value; } } if(newstats.RemoteArchitectures != null) { foreach(NameValueStats nvs in newstats.RemoteArchitectures) { RemoteArchitecture existing = _ctx.RemoteArchitectures.FirstOrDefault(c => c.Name == nvs.name); if(existing == null) { await _ctx.RemoteArchitectures.AddAsync(new RemoteArchitecture { Name = nvs.name, Count = nvs.Value }); } else existing.Count += nvs.Value; } } if(newstats.RemoteOperatingSystems != null) { foreach(OsStats remoteOperatingSystem in newstats.RemoteOperatingSystems) { RemoteOperatingSystem existing = _ctx.RemoteOperatingSystems.FirstOrDefault(c => c.Name == remoteOperatingSystem.name && c.Version == remoteOperatingSystem.version); if(existing == null) { await _ctx.RemoteOperatingSystems.AddAsync(new RemoteOperatingSystem { Name = remoteOperatingSystem.name, Version = remoteOperatingSystem.version, Count = remoteOperatingSystem.Value }); } else existing.Count += remoteOperatingSystem.Value; } } await _ctx.SaveChangesAsync(); response.Content = "ok"; return response; } // ReSharper disable once RedundantCatchClause catch { #if DEBUG if(Debugger.IsAttached) throw; #endif response.Content = "error"; return response; } } }