public PageService(IDiskService disk, AesEncryption crypto, CacheService cache, Logger log)
{
_disk = disk;
_crypto = crypto;
_cache = cache;
_log = log;
}
internal TransactionService(IDiskService disk, AesEncryption crypto, PageService pager, LockService locker, CacheService cache, int cacheSize, Logger log)
{
_disk = disk;
_crypto = crypto;
_cache = cache;
_locker = locker;
_pager = pager;
_cacheSize = cacheSize;
_log = log;
}
/// <summary>
/// Initialize LiteEngine using custom disk service implementation and full engine options
/// </summary>
public LiteEngine(IDiskService disk, string password = null, TimeSpan?timeout = null, int cacheSize = 5000, Logger log = null, bool utcDate = false)
{
_timeout = timeout ?? TimeSpan.FromMinutes(1);
_cacheSize = cacheSize;
_disk = disk ?? throw new ArgumentNullException(nameof(disk));
_log = log ?? new Logger();
_bsonReader = new BsonReader(utcDate);
try
{
// initialize datafile (create) and set log instance
_disk.Initialize(_log, password);
// lock disk (read mode) before read header
var position = _disk.Lock(LockState.Read, _timeout);
var buffer = _disk.ReadPage(0);
_disk.Unlock(LockState.Read, position);
// create header instance from array bytes
var header = BasePage.ReadPage(buffer) as HeaderPage;
// hash password with sha1 or keep as empty byte[20]
var sha1 = password == null ? new byte[20] : AesEncryption.HashSHA1(password);
// compare header password with user password even if not passed password (datafile can have password)
if (sha1.BinaryCompareTo(header.Password) != 0)
{
throw LiteException.DatabaseWrongPassword();
}
// initialize AES encryptor
if (password != null)
{
_crypto = new AesEncryption(password, header.Salt);
}
// initialize all services
InitializeServices();
// if header are marked with recovery, do it now
if (header.Recovery)
{
_trans.Recovery();
}
}
catch (Exception)
{
// explicit dispose
Dispose();
throw;
}
}
//public void Dispose()
//{
// // dispose datafile and journal file
// _disk.Dispose();
// // dispose crypto
// if (_crypto != null) _crypto.Dispose();
//}
/// <summary>
/// Initialize new datafile with header page + lock reserved area zone
/// </summary>
public static void CreateDatabase(Stream stream, string password = null, long initialSize = 0)
{
// calculate how many empty pages will be added on disk
var emptyPages = initialSize == 0 ? 0 : (initialSize - (2 * BasePage.PAGE_SIZE)) / BasePage.PAGE_SIZE;
// if too small size (less than 2 pages), assume no initial size
if (emptyPages < 0)
{
emptyPages = 0;
}
// create a new header page in bytes (keep second page empty)
var header = new HeaderPage
{
LastPageID = initialSize == 0 ? 1 : (uint)emptyPages + 1,
FreeEmptyPageID = initialSize == 0 ? uint.MaxValue : 2
};
if (password != null)
{
header.Password = AesEncryption.HashSHA1(password);
header.Salt = AesEncryption.Salt();
}
// point to begin file
stream.Seek(0, SeekOrigin.Begin);
// get header page in bytes
var buffer = header.WritePage();
stream.Write(buffer, 0, BasePage.PAGE_SIZE);
// write second page as an empty AREA (it's not a page) just to use as lock control
stream.Write(new byte[BasePage.PAGE_SIZE], 0, BasePage.PAGE_SIZE);
// create crypto class if has password
var crypto = password != null ? new AesEncryption(password, header.Salt) : null;
// if initial size is defined, lets create empty pages in a linked list
if (emptyPages > 0)
{
stream.SetLength(initialSize);
var pageID = 1u;
while (++pageID < (emptyPages + 2))
{
var empty = new EmptyPage(pageID)
{
PrevPageID = pageID == 2 ? 0 : pageID - 1,
NextPageID = pageID == emptyPages + 1 ? uint.MaxValue : pageID + 1
};
var bytes = empty.WritePage();
if (password != null)
{
bytes = crypto.Encrypt(bytes);
}
stream.Write(bytes, 0, BasePage.PAGE_SIZE);
}
}
if (crypto != null)
{
crypto.Dispose();
}
}
