public override IEnumerable<UInt160> GetAddresses()
{
using (WalletDataContext ctx = new WalletDataContext(connectionString))
{
return ctx.Accounts.Select(p => p.ScriptHash).ToArray().Select(p => new UInt160(p));
}
}
public static UserWallet CreateDatabase(string path, string password)
{
SqlConnectionStringBuilder sb = new SqlConnectionStringBuilder();
sb.AttachDBFilename = path;
sb.DataSource = @"(LocalDB)\v11.0";
sb.IntegratedSecurity = true;
using (WalletDataContext ctx = new WalletDataContext(sb.ToString()))
using (RNGCryptoServiceProvider rng = new RNGCryptoServiceProvider())
{
byte[] passwordKey = password.ToAesKey();
byte[] masterKey = new byte[32];
byte[] iv = new byte[16];
rng.GetNonZeroBytes(masterKey);
rng.GetNonZeroBytes(iv);
masterKey.AesEncrypt(passwordKey, iv);
Array.Clear(passwordKey, 0, passwordKey.Length);
ctx.Database.Delete();
ctx.Database.Create();
ctx.Keys.Add(new Key
{
Name = Key.MasterKey,
Value = masterKey
});
ctx.Keys.Add(new Key
{
Name = Key.IV,
Value = iv
});
ctx.SaveChanges();
}
UserWallet wallet = OpenDatabase(path, password);
wallet.CreateEntry();
return wallet;
}
public static UserWallet OpenDatabase(string path, string password)
{
SqlConnectionStringBuilder sb = new SqlConnectionStringBuilder();
sb.AttachDBFilename = path;
sb.DataSource = @"(LocalDB)\v11.0";
sb.IntegratedSecurity = true;
string connectionString = sb.ToString();
using (WalletDataContext ctx = new WalletDataContext(connectionString))
{
byte[] masterKey = ctx.Keys.First(p => p.Name == Key.MasterKey).Value;
byte[] passwordKey = password.ToAesKey();
byte[] iv = ctx.Keys.First(p => p.Name == Key.IV).Value;
masterKey.AesDecrypt(passwordKey, iv);
Array.Clear(passwordKey, 0, passwordKey.Length);
return new UserWallet(connectionString, masterKey, iv);
}
}
protected override void SaveEncryptedEntry(UInt160 scriptHash, byte[] redeemScript, byte[] encryptedPrivateKey)
{
using (WalletDataContext ctx = new WalletDataContext(connectionString))
{
byte[] scriptHashBytes = scriptHash.ToArray();
Account account = ctx.Accounts.FirstOrDefault(p => p.ScriptHash == scriptHashBytes);
if (account == null)
{
account = ctx.Accounts.Add(new Account
{
ScriptHash = scriptHash.ToArray(),
RedeemScript = redeemScript,
PrivateKeyEncrypted = encryptedPrivateKey
});
}
else
{
account.RedeemScript = redeemScript;
account.PrivateKeyEncrypted = encryptedPrivateKey;
}
ctx.SaveChanges();
}
}
protected override void GetEncryptedEntry(UInt160 scriptHash, out byte[] redeemScript, out byte[] encryptedPrivateKey)
{
using (WalletDataContext ctx = new WalletDataContext(connectionString))
{
//Account account = ctx.Accounts.FirstOrDefault(p => p.ScriptHash == scriptHash.ToArray());
//It throws a NotSupportedException:
//LINQ to Entities does not recognize the method 'Byte[] ToArray()' method, and this method cannot be translated into a store expression.
//I don't know why.
//So,
byte[] temp = scriptHash.ToArray();
Account account = ctx.Accounts.FirstOrDefault(p => p.ScriptHash == temp);
//It works!
if (account == null)
{
redeemScript = null;
encryptedPrivateKey = null;
return;
}
redeemScript = account.RedeemScript;
encryptedPrivateKey = account.PrivateKeyEncrypted;
}
}
protected override void DeleteEntry(UInt160 scriptHash)
{
using (WalletDataContext ctx = new WalletDataContext(connectionString))
{
Account account = ctx.Accounts.FirstOrDefault(p => p.ScriptHash == scriptHash.ToArray());
if (account != null)
{
ctx.Accounts.Remove(account);
ctx.SaveChanges();
}
}
}
