mirror of
https://github.com/aaru-dps/Aaru.Server.git
synced 2025-12-16 19:24:27 +00:00
277 lines
10 KiB
C#
277 lines
10 KiB
C#
// /***************************************************************************
|
|
// Aaru Data Preservation Suite
|
|
// ----------------------------------------------------------------------------
|
|
//
|
|
// Filename : UploadReportController.cs
|
|
// Author(s) : Natalia Portillo <claunia@claunia.com>
|
|
//
|
|
// Component : Aaru Server.
|
|
//
|
|
// --[ Description ] ----------------------------------------------------------
|
|
//
|
|
// Handles report 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 <http://www.gnu.org/licenses/>.
|
|
//
|
|
// ----------------------------------------------------------------------------
|
|
// Copyright © 2011-2024 Natalia Portillo
|
|
// ****************************************************************************/
|
|
|
|
using System.Diagnostics;
|
|
using System.Net;
|
|
using System.Text;
|
|
using System.Xml.Serialization;
|
|
using Aaru.CommonTypes.Metadata;
|
|
using Aaru.Server.Database.Models;
|
|
using Cinchoo.PGP;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using MimeKit;
|
|
using Newtonsoft.Json;
|
|
using DbContext = Aaru.Server.Database.DbContext;
|
|
using SmtpClient = MailKit.Net.Smtp.SmtpClient;
|
|
|
|
namespace Aaru.Server.New.Controllers;
|
|
|
|
public sealed class UploadReportController : ControllerBase
|
|
{
|
|
readonly DbContext _ctx;
|
|
readonly IWebHostEnvironment _environment;
|
|
|
|
public UploadReportController(IWebHostEnvironment environment, DbContext ctx)
|
|
{
|
|
_environment = environment;
|
|
_ctx = ctx;
|
|
}
|
|
|
|
/// <summary>Receives a report from Aaru.Core, verifies it's in the correct format and stores it on the server</summary>
|
|
/// <returns>HTTP response</returns>
|
|
[Route("api/uploadreport")]
|
|
[HttpPost]
|
|
public async Task<IActionResult> UploadReport()
|
|
{
|
|
var response = new ContentResult
|
|
{
|
|
StatusCode = (int)HttpStatusCode.OK,
|
|
ContentType = "text/plain"
|
|
};
|
|
|
|
try
|
|
{
|
|
var newReport = new DeviceReport();
|
|
HttpRequest request = HttpContext.Request;
|
|
|
|
var xs = new XmlSerializer(newReport.GetType());
|
|
|
|
newReport =
|
|
(DeviceReport)xs.Deserialize(new StringReader(await new StreamReader(request.Body).ReadToEndAsync()));
|
|
|
|
if(newReport == null)
|
|
{
|
|
response.Content = "notstats";
|
|
|
|
return response;
|
|
}
|
|
|
|
var reportV2 = new DeviceReportV2(newReport);
|
|
var jsonSw = new StringWriter();
|
|
|
|
await jsonSw.WriteAsync(JsonConvert.SerializeObject(reportV2,
|
|
Formatting.Indented,
|
|
new JsonSerializerSettings
|
|
{
|
|
NullValueHandling = NullValueHandling.Ignore
|
|
}));
|
|
|
|
var reportV2String = jsonSw.ToString();
|
|
jsonSw.Close();
|
|
|
|
await SaveReport(reportV2, reportV2String);
|
|
|
|
response.Content = "ok";
|
|
|
|
return response;
|
|
}
|
|
|
|
// ReSharper disable once RedundantCatchClause
|
|
catch
|
|
{
|
|
#if DEBUG
|
|
if(Debugger.IsAttached) throw;
|
|
#endif
|
|
response.Content = "error";
|
|
|
|
return response;
|
|
}
|
|
}
|
|
|
|
/// <summary>Receives a report from Aaru.Core, verifies it's in the correct format and stores it on the server</summary>
|
|
/// <returns>HTTP response</returns>
|
|
[Route("api/uploadreportv2")]
|
|
[HttpPost]
|
|
public async Task<IActionResult> UploadReportV2()
|
|
{
|
|
var response = new ContentResult
|
|
{
|
|
StatusCode = (int)HttpStatusCode.OK,
|
|
ContentType = "text/plain"
|
|
};
|
|
|
|
try
|
|
{
|
|
HttpRequest request = HttpContext.Request;
|
|
|
|
var sr = new StreamReader(request.Body);
|
|
string reportJson = await sr.ReadToEndAsync();
|
|
DeviceReportV2 newReport = JsonConvert.DeserializeObject<DeviceReportV2>(reportJson);
|
|
|
|
if(newReport == null)
|
|
{
|
|
response.Content = "notstats";
|
|
|
|
return response;
|
|
}
|
|
|
|
await SaveReport(newReport, reportJson);
|
|
|
|
response.Content = "ok";
|
|
|
|
return response;
|
|
}
|
|
|
|
// ReSharper disable once RedundantCatchClause
|
|
catch
|
|
{
|
|
#if DEBUG
|
|
if(Debugger.IsAttached) throw;
|
|
#endif
|
|
response.Content = "error";
|
|
|
|
return response;
|
|
}
|
|
}
|
|
|
|
async Task SaveReport(DeviceReportV2 newReport, string reportJson)
|
|
{
|
|
var newUploadedReport = new UploadedReport(newReport);
|
|
|
|
// Ensure CHS and CurrentCHS are not duplicates
|
|
if(newUploadedReport.ATA?.ReadCapabilities is { CHS: not null, CurrentCHS: not null } &&
|
|
newUploadedReport.ATA.ReadCapabilities.CHS.Cylinders ==
|
|
newUploadedReport.ATA.ReadCapabilities.CurrentCHS.Cylinders &&
|
|
newUploadedReport.ATA.ReadCapabilities.CHS.Heads ==
|
|
newUploadedReport.ATA.ReadCapabilities.CurrentCHS.Heads &&
|
|
newUploadedReport.ATA.ReadCapabilities.CHS.Sectors ==
|
|
newUploadedReport.ATA.ReadCapabilities.CurrentCHS.Sectors)
|
|
newUploadedReport.ATA.ReadCapabilities.CHS = newUploadedReport.ATA.ReadCapabilities.CurrentCHS;
|
|
|
|
Chs? existingChs;
|
|
|
|
// Check if the CHS or CurrentCHS of this report already exist in the database
|
|
if(newUploadedReport.ATA?.ReadCapabilities?.CHS != null)
|
|
{
|
|
existingChs =
|
|
await _ctx.Chs.FirstOrDefaultAsync(c =>
|
|
c.Cylinders ==
|
|
newUploadedReport.ATA.ReadCapabilities.CHS.Cylinders &&
|
|
c.Heads == newUploadedReport.ATA.ReadCapabilities.CHS.Heads &&
|
|
c.Sectors == newUploadedReport.ATA.ReadCapabilities.CHS.Sectors);
|
|
|
|
if(existingChs != null) newUploadedReport.ATA.ReadCapabilities.CHS = existingChs;
|
|
}
|
|
|
|
if(newUploadedReport.ATA?.ReadCapabilities?.CurrentCHS != null)
|
|
{
|
|
existingChs =
|
|
await _ctx.Chs.FirstOrDefaultAsync(c =>
|
|
c.Cylinders ==
|
|
newUploadedReport.ATA.ReadCapabilities.CurrentCHS.Cylinders &&
|
|
c.Heads ==
|
|
newUploadedReport.ATA.ReadCapabilities.CurrentCHS.Heads &&
|
|
c.Sectors ==
|
|
newUploadedReport.ATA.ReadCapabilities.CurrentCHS.Sectors);
|
|
|
|
if(existingChs != null) newUploadedReport.ATA.ReadCapabilities.CurrentCHS = existingChs;
|
|
}
|
|
|
|
if(newUploadedReport.ATA?.RemovableMedias != null)
|
|
{
|
|
foreach(TestedMedia media in newUploadedReport.ATA.RemovableMedias)
|
|
{
|
|
if(media.CHS != null &&
|
|
media.CurrentCHS != null &&
|
|
media.CHS.Cylinders == media.CurrentCHS.Cylinders &&
|
|
media.CHS.Heads == media.CurrentCHS.Heads &&
|
|
media.CHS.Sectors == media.CurrentCHS.Sectors)
|
|
media.CHS = media.CurrentCHS;
|
|
|
|
if(media.CHS != null)
|
|
{
|
|
existingChs =
|
|
await _ctx.Chs.FirstOrDefaultAsync(c => c.Cylinders == media.CHS.Cylinders &&
|
|
c.Heads == media.CHS.Heads &&
|
|
c.Sectors == media.CHS.Sectors);
|
|
|
|
if(existingChs != null) media.CHS = existingChs;
|
|
}
|
|
|
|
if(media.CHS == null) continue;
|
|
|
|
|
|
existingChs =
|
|
await _ctx.Chs.FirstOrDefaultAsync(c => media.CurrentCHS != null &&
|
|
c.Cylinders == media.CurrentCHS.Cylinders &&
|
|
c.Heads == media.CurrentCHS.Heads &&
|
|
c.Sectors == media.CurrentCHS.Sectors);
|
|
|
|
if(existingChs != null) media.CurrentCHS = existingChs;
|
|
}
|
|
}
|
|
|
|
_ctx.Reports.Add(newUploadedReport);
|
|
await _ctx.SaveChangesAsync();
|
|
|
|
var pgpIn = new MemoryStream(Encoding.UTF8.GetBytes(reportJson));
|
|
var pgpOut = new MemoryStream();
|
|
var pgp = new ChoPGPEncryptDecrypt();
|
|
|
|
await pgp.EncryptAsync(pgpIn,
|
|
pgpOut,
|
|
Path.Combine(_environment.ContentRootPath ?? throw new InvalidOperationException(),
|
|
"public.asc"));
|
|
|
|
pgpOut.Position = 0;
|
|
reportJson = Encoding.UTF8.GetString(pgpOut.ToArray());
|
|
|
|
var message = new MimeMessage
|
|
{
|
|
Subject = "New device report",
|
|
Body = new TextPart("plain")
|
|
{
|
|
Text = reportJson
|
|
}
|
|
};
|
|
|
|
message.From.Add(new MailboxAddress("Aaru Server", "aaru@claunia.com"));
|
|
message.To.Add(new MailboxAddress("Natalia Portillo", "claunia@claunia.com"));
|
|
|
|
using var client = new SmtpClient();
|
|
|
|
await client.ConnectAsync("mail.claunia.com", 25, false);
|
|
await client.SendAsync(message);
|
|
await client.DisconnectAsync(true);
|
|
}
|
|
} |