keepass2android/src/Kp2aBusinessLogic/database/edit/SetPassword.cs at da8ce7dc4860cdfca7bbd03f96fbb52808ce982c · PhilippC/keepass2android · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/*
This file is part of Keepass2Android, Copyright 2013 Philipp Crocoll. This file is based on Keepassdroid, Copyright Brian Pellin.

  Keepass2Android is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 2 of the License, or
  (at your option) any later version.

  Keepass2Android 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with Keepass2Android.  If not, see <http://www.gnu.org/licenses/>.
  */
using System;
using Android.App;
using Android.Content;
using KeePassLib;
using KeePassLib.Keys;

namespace keepass2android
{
  public class SetPassword : OperationWithFinishHandler
  {

    private readonly String _password;
    private readonly String _keyfile;
    private readonly IKp2aApp _app;
    private readonly bool _dontSave;

    public SetPassword(IKp2aApp app, String password, String keyfile, OnOperationFinishedHandler operationFinishedHandler) : base(app, operationFinishedHandler)
    {

      _app = app;
      _password = password;
      _keyfile = keyfile;
      _dontSave = false;
    }

    public SetPassword(IKp2aApp app, String password, String keyfile, OnOperationFinishedHandler operationFinishedHandler, bool dontSave)
        : base(app, operationFinishedHandler)
    {
      _app = app;
      _password = password;
      _keyfile = keyfile;
      _dontSave = dontSave;
    }


    public override void Run()
    {
      StatusLogger.UpdateMessage(UiStringKey.SettingPassword);
      PwDatabase pm = _app.CurrentDb.KpDatabase;
      CompositeKey newKey = new CompositeKey();
      if (String.IsNullOrEmpty(_password) == false)
      {
        newKey.AddUserKey(new KcpPassword(_password));
      }
      if (String.IsNullOrEmpty(_keyfile) == false)
      {
        try
        {
          newKey.AddUserKey(new KcpKeyFile(_keyfile));
        }
        catch (Exception)
        {
          //TODO MessageService.ShowWarning (strKeyFile, KPRes.KeyFileError, exKF);
          return;
        }
      }

      DateTime previousMasterKeyChanged = pm.MasterKeyChanged;
      CompositeKey previousKey = pm.MasterKey;

      pm.MasterKeyChanged = DateTime.UtcNow;
      pm.MasterKey = newKey;

      // Save Database
      _operationFinishedHandler = new AfterSave(_app, previousKey, previousMasterKeyChanged, pm, operationFinishedHandler);
      SaveDb save = new SaveDb(_app, _app.CurrentDb, operationFinishedHandler, _dontSave, null);
      save.SetStatusLogger(StatusLogger);
      save.Run();
    }

    private class AfterSave : OnOperationFinishedHandler
    {
      private readonly CompositeKey _backup;
      private readonly DateTime _previousKeyChanged;
      private readonly PwDatabase _db;

      public AfterSave(IActiveContextProvider activeContextProvider, CompositeKey backup, DateTime previousKeyChanged, PwDatabase db, OnOperationFinishedHandler operationFinishedHandler) : base(activeContextProvider, operationFinishedHandler)
      {
        _previousKeyChanged = previousKeyChanged;
        _backup = backup;
        _db = db;
      }

      public override void Run()
      {
        if (!Success)
        {
          _db.MasterKey = _backup;
          _db.MasterKeyChanged = _previousKeyChanged;
        }

        base.Run();
      }

    }


  }

}