Remove MBLicenseFile
This commit is contained in:
hawken
Bond-009
parent
d707e969fa
commit
19319e8eb6
|
|
@ -1,204 +0,0 @@
|
|||
using System;
|
||||
using System.Collections.Concurrent;
|
||||
using System.Collections.Generic;
|
||||
using System.Globalization;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using MediaBrowser.Common.Configuration;
|
||||
using MediaBrowser.Model.Cryptography;
|
||||
using MediaBrowser.Model.IO;
|
||||
|
||||
namespace Emby.Server.Implementations.Security
|
||||
{
|
||||
internal class MBLicenseFile
|
||||
{
|
||||
private readonly IApplicationPaths _appPaths;
|
||||
private readonly IFileSystem _fileSystem;
|
||||
private readonly ICryptoProvider _cryptographyProvider;
|
||||
|
||||
public string RegKey
|
||||
{
|
||||
get => _regKey;
|
||||
set
|
||||
{
|
||||
_updateRecords.Clear();
|
||||
_regKey = value;
|
||||
}
|
||||
}
|
||||
|
||||
private string Filename => Path.Combine(_appPaths.ConfigurationDirectoryPath, "mb.lic");
|
||||
|
||||
private readonly ConcurrentDictionary<Guid, FeatureRegInfo> _updateRecords = new ConcurrentDictionary<Guid, FeatureRegInfo>();
|
||||
private readonly object _fileLock = new object();
|
||||
private string _regKey;
|
||||
|
||||
public MBLicenseFile(IApplicationPaths appPaths, IFileSystem fileSystem, ICryptoProvider cryptographyProvider)
|
||||
{
|
||||
_appPaths = appPaths;
|
||||
_fileSystem = fileSystem;
|
||||
_cryptographyProvider = cryptographyProvider;
|
||||
|
||||
Load();
|
||||
}
|
||||
|
||||
private void SetUpdateRecord(Guid key, FeatureRegInfo value)
|
||||
{
|
||||
_updateRecords.AddOrUpdate(key, value, (k, v) => value);
|
||||
}
|
||||
|
||||
private Guid GetKey(string featureId)
|
||||
{
|
||||
return new Guid(_cryptographyProvider.ComputeMD5(Encoding.Unicode.GetBytes(featureId)));
|
||||
}
|
||||
|
||||
public void AddRegCheck(string featureId, DateTime expirationDate)
|
||||
{
|
||||
var key = GetKey(featureId);
|
||||
var value = new FeatureRegInfo
|
||||
{
|
||||
ExpirationDate = expirationDate,
|
||||
LastChecked = DateTime.UtcNow
|
||||
};
|
||||
|
||||
SetUpdateRecord(key, value);
|
||||
Save();
|
||||
}
|
||||
|
||||
public void RemoveRegCheck(string featureId)
|
||||
{
|
||||
var key = GetKey(featureId);
|
||||
FeatureRegInfo val;
|
||||
|
||||
_updateRecords.TryRemove(key, out val);
|
||||
|
||||
Save();
|
||||
}
|
||||
|
||||
public FeatureRegInfo GetRegInfo(string featureId)
|
||||
{
|
||||
var key = GetKey(featureId);
|
||||
FeatureRegInfo info = null;
|
||||
_updateRecords.TryGetValue(key, out info);
|
||||
|
||||
if (info == null)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
// guard agains people just putting a large number in the file
|
||||
return info.LastChecked < DateTime.UtcNow ? info : null;
|
||||
}
|
||||
|
||||
private void Load()
|
||||
{
|
||||
string[] contents = null;
|
||||
var licenseFile = Filename;
|
||||
lock (_fileLock)
|
||||
{
|
||||
try
|
||||
{
|
||||
contents = _fileSystem.ReadAllLines(licenseFile);
|
||||
}
|
||||
catch (FileNotFoundException)
|
||||
{
|
||||
lock (_fileLock)
|
||||
{
|
||||
_fileSystem.WriteAllBytes(licenseFile, Array.Empty<byte>());
|
||||
}
|
||||
}
|
||||
catch (IOException)
|
||||
{
|
||||
lock (_fileLock)
|
||||
{
|
||||
_fileSystem.WriteAllBytes(licenseFile, Array.Empty<byte>());
|
||||
}
|
||||
}
|
||||
}
|
||||
if (contents != null && contents.Length > 0)
|
||||
{
|
||||
//first line is reg key
|
||||
RegKey = contents[0];
|
||||
|
||||
//next is legacy key
|
||||
if (contents.Length > 1)
|
||||
{
|
||||
// Don't need this anymore
|
||||
}
|
||||
|
||||
//the rest of the lines should be pairs of features and timestamps
|
||||
for (var i = 2; i < contents.Length; i = i + 2)
|
||||
{
|
||||
var line = contents[i];
|
||||
if (string.IsNullOrWhiteSpace(line))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
Guid feat;
|
||||
if (Guid.TryParse(line, out feat))
|
||||
{
|
||||
var lineParts = contents[i + 1].Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries);
|
||||
|
||||
long ticks;
|
||||
if (long.TryParse(lineParts[0], out ticks))
|
||||
{
|
||||
var info = new FeatureRegInfo
|
||||
{
|
||||
LastChecked = new DateTime(ticks)
|
||||
};
|
||||
|
||||
if (lineParts.Length > 1 && long.TryParse(lineParts[1], out ticks))
|
||||
{
|
||||
info.ExpirationDate = new DateTime(ticks);
|
||||
}
|
||||
|
||||
SetUpdateRecord(feat, info);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void Save()
|
||||
{
|
||||
//build our array
|
||||
var lines = new List<string>
|
||||
{
|
||||
RegKey,
|
||||
|
||||
// Legacy key
|
||||
string.Empty
|
||||
};
|
||||
|
||||
foreach (var pair in _updateRecords
|
||||
.ToList())
|
||||
{
|
||||
lines.Add(pair.Key.ToString());
|
||||
|
||||
var dateLine = pair.Value.LastChecked.Ticks.ToString(CultureInfo.InvariantCulture) + "|" +
|
||||
pair.Value.ExpirationDate.Ticks.ToString(CultureInfo.InvariantCulture);
|
||||
|
||||
lines.Add(dateLine);
|
||||
}
|
||||
|
||||
var licenseFile = Filename;
|
||||
_fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(licenseFile));
|
||||
lock (_fileLock)
|
||||
{
|
||||
_fileSystem.WriteAllLines(licenseFile, lines);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
internal class FeatureRegInfo
|
||||
{
|
||||
public DateTime ExpirationDate { get; set; }
|
||||
public DateTime LastChecked { get; set; }
|
||||
|
||||
public FeatureRegInfo()
|
||||
{
|
||||
ExpirationDate = DateTime.MinValue;
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user