Fix lifetime of leveldb::MojoEnv instances.

components/leveldb/leveldb_database_impl.h

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef COMPONENTS_LEVELDB_LEVELDB_DATABASE_IMPL_H_
 #define COMPONENTS_LEVELDB_LEVELDB_DATABASE_IMPL_H_
 
 #include <memory>
 
 #include "base/unguessable_token.h"
+#include "components/leveldb/env_mojo.h"
 #include "components/leveldb/public/interfaces/leveldb.mojom.h"
 #include "mojo/public/cpp/bindings/interface_request.h"
 #include "third_party/leveldatabase/src/include/leveldb/db.h"
 
 namespace leveldb {
 
 // The backing to a database object that we pass to our called.
 class LevelDBDatabaseImpl : public mojom::LevelDBDatabase {
  public:
-  LevelDBDatabaseImpl(std::unique_ptr<leveldb::Env> environment,
-                      std::unique_ptr<leveldb::DB> db);
+  LevelDBDatabaseImpl(std::unique_ptr<Env> owned_environment,
+                      std::unique_ptr<DB> db);
+  LevelDBDatabaseImpl(std::unique_ptr<MojoEnv::Directory> directory,
+                      std::unique_ptr<DB> db);
   ~LevelDBDatabaseImpl() override;
 
   // Overridden from LevelDBDatabase:
   void Put(const std::vector<uint8_t>& key,
            const std::vector<uint8_t>& value,
            const PutCallback& callback) override;
   void Delete(const std::vector<uint8_t>& key,
               const DeleteCallback& callback) override;
   void DeletePrefixed(const std::vector<uint8_t>& key_prefix,
                       const DeletePrefixedCallback& callback) override;
@@ skipping 29 matching lines @@
  private:
   // Returns the state of |it| to a caller. Note: This assumes that all the
   // iterator movement methods have the same callback signature. We don't
   // directly reference the underlying type in case of bindings change.
   void ReplyToIteratorMessage(leveldb::Iterator* it,
                               const IteratorSeekToFirstCallback& callback);
 
   leveldb::Status DeletePrefixedHelper(const leveldb::Slice& key_prefix,
                                        leveldb::WriteBatch* batch);
 
-  std::unique_ptr<leveldb::Env> environment_;
-  std::unique_ptr<leveldb::DB> db_;
+  std::unique_ptr<Env> owned_environment_;
+  std::unique_ptr<MojoEnv::Directory> directory_;
+  std::unique_ptr<DB> db_;
 
   std::map<base::UnguessableToken, const Snapshot*> snapshot_map_;
 
   // TODO(erg): If we have an existing iterator which depends on a snapshot,
   // and delete the snapshot from the client side, that shouldn't delete the
   // snapshot maybe? At worse it's a DDoS if there's multiple users of the
   // system, but this maybe should be fixed...
 
   std::map<base::UnguessableToken, Iterator*> iterator_map_;
 
   DISALLOW_COPY_AND_ASSIGN(LevelDBDatabaseImpl);
 };
 
 }  // namespace leveldb
 
 #endif  // COMPONENTS_LEVELDB_LEVELDB_DATABASE_IMPL_H_