DocSpace-buildtools/common/ASC.Data.Storage/Encryption/EncryptionSettings.cs

/*
 *
 * (c) Copyright Ascensio System Limited 2010-2020
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/


using System;
using System.Security.Cryptography;

using ASC.Common;
using ASC.Common.Caching;
using ASC.Core;

namespace ASC.Data.Storage.Encryption
{
    public class EncryptionSettings
    {

        internal string password;

        public string Password
        {
            get { return password; }
            set { password = (value ?? string.Empty).Replace('#', '_'); }
        }

        public EncryprtionStatus Status { get; set; }

        public bool NotifyUsers { get; set; }

        public EncryptionSettings(EncryptionSettingsProto encryptionSettingsProto)
        {
            Password = encryptionSettingsProto.Password;
            Status = encryptionSettingsProto.Status;
            NotifyUsers = encryptionSettingsProto.NotifyUsers;
        }

        public EncryptionSettings()
        {
            Password = string.Empty;
            Status = EncryprtionStatus.Decrypted;
            NotifyUsers = true;
        }
    }

    public class EncryptionSettingsHelper 
    {
        private const string key = "EncryptionSettings";

        private CoreConfiguration CoreConfiguration { get; }
        private AscCacheNotify AscCacheNotify { get; }

        public EncryptionSettingsHelper(CoreConfiguration coreConfiguration, AscCacheNotify ascCacheNotify)
        {
            CoreConfiguration = coreConfiguration;
            AscCacheNotify = ascCacheNotify;
        }

        public void Save(EncryptionSettings encryptionSettings)
        {
            var settings = Serialize(encryptionSettings);
            CoreConfiguration.SaveSetting(key, settings);

            AscCacheNotify.ClearCache();
        }

        public EncryptionSettings Load()
        {
            var settings = CoreConfiguration.GetSetting(key);

            return Deserialize(settings);
        }

        public string Serialize(EncryptionSettings encryptionSettings)
        {
            return string.Join("#",
                string.IsNullOrEmpty(encryptionSettings.password) ? string.Empty : Crypto.GetV(encryptionSettings.password, 1, true),
                (int)encryptionSettings.Status,
                encryptionSettings.NotifyUsers
            );
        }

        public EncryptionSettings Deserialize(string value)
        {
            if (string.IsNullOrEmpty(value))
            {
                return new EncryptionSettings();
            }

            var parts = value.Split(new[] { '#' }, StringSplitOptions.None);

            var password = string.IsNullOrEmpty(parts[0]) ? string.Empty : Crypto.GetV(parts[0], 1, false);
            var status = int.Parse(parts[1]);
            var notifyUsers = bool.Parse(parts[2]);

            return new EncryptionSettings
            {
                Password = password,
                Status = (EncryprtionStatus)status,
                NotifyUsers = notifyUsers
            };
        }

        // source System.Web.Security.Membership.GeneratePassword
        public string GeneratePassword(int length, int numberOfNonAlphanumericCharacters)
        {
            var punctuations = "!@#$%^&*()_-+=[{]};:>|./?".ToCharArray();

            if (length < 1 || length > 128)
            {
                throw new ArgumentException("password_length_incorrect", "length");
            }

            if (numberOfNonAlphanumericCharacters > length || numberOfNonAlphanumericCharacters < 0)
            {
                throw new ArgumentException("min_required_non_alphanumeric_characters_incorrect", "numberOfNonAlphanumericCharacters");
            }

            byte[] array = new byte[length];
            char[] array2 = new char[length];
            int num = 0;

            using (var rng = new RNGCryptoServiceProvider())
            {
                rng.GetBytes(array);
            }

            for (int i = 0; i < length; i++)
            {
                int num2 = (int)array[i] % 87;
                if (num2 < 10)
                {
                    array2[i] = (char)(48 + num2);
                    continue;
                }

                if (num2 < 36)
                {
                    array2[i] = (char)(65 + num2 - 10);
                    continue;
                }

                if (num2 < 62)
                {
                    array2[i] = (char)(97 + num2 - 36);
                    continue;
                }

                array2[i] = punctuations[num2 - 62];
                num++;
            }

            if (num < numberOfNonAlphanumericCharacters)
            {
                Random random = new Random();
                for (int j = 0; j < numberOfNonAlphanumericCharacters - num; j++)
                {
                    int num3;
                    do
                    {
                        num3 = random.Next(0, length);
                    }
                    while (!char.IsLetterOrDigit(array2[num3]));
                    array2[num3] = punctuations[random.Next(0, punctuations.Length)];
                }
            }

            return new string(array2);
        }
    }

    public static class EncryptionSettingsHelperExtension
    {
        public static DIHelper AddEncryptionSettingsHelperService(this DIHelper services)
        {
            services.TryAddScoped<EncryptionSettingsHelper>();
            services.TryAddScoped<AscCacheNotify>();
            return services
                .AddCoreConfigurationService();
        }
    }
}