summaryrefslogtreecommitdiffstats
path: root/browser/components/migration/ESEDBReader.jsm
diff options
context:
space:
mode:
authorwolfbeast <mcwerewolf@gmail.com>2018-06-04 13:17:38 +0200
committerwolfbeast <mcwerewolf@gmail.com>2018-06-04 13:17:38 +0200
commita1be17c1cea81ebb1e8b131a662c698d78f3f7f2 (patch)
treea92f7de513be600cc07bac458183e9af40e00c06 /browser/components/migration/ESEDBReader.jsm
parentbf11fdd304898ac675e39b01b280d39550e419d0 (diff)
downloadUXP-a1be17c1cea81ebb1e8b131a662c698d78f3f7f2.tar
UXP-a1be17c1cea81ebb1e8b131a662c698d78f3f7f2.tar.gz
UXP-a1be17c1cea81ebb1e8b131a662c698d78f3f7f2.tar.lz
UXP-a1be17c1cea81ebb1e8b131a662c698d78f3f7f2.tar.xz
UXP-a1be17c1cea81ebb1e8b131a662c698d78f3f7f2.zip
Issue #303 Part 1: Move basilisk files from /browser to /application/basilisk
Diffstat (limited to 'browser/components/migration/ESEDBReader.jsm')
-rw-r--r--browser/components/migration/ESEDBReader.jsm590
1 files changed, 0 insertions, 590 deletions
diff --git a/browser/components/migration/ESEDBReader.jsm b/browser/components/migration/ESEDBReader.jsm
deleted file mode 100644
index 0768c65aa..000000000
--- a/browser/components/migration/ESEDBReader.jsm
+++ /dev/null
@@ -1,590 +0,0 @@
-/* This Source Code Form is subject to the terms of the Mozilla Public
- * License, v. 2.0. If a copy of the MPL was not distributed with this file,
- * You can obtain one at http://mozilla.org/MPL/2.0/. */
-
-"use strict";
-
-this.EXPORTED_SYMBOLS = ["ESEDBReader"]; /* exported ESEDBReader */
-
-const { classes: Cc, interfaces: Ci, utils: Cu, results: Cr } = Components;
-
-Cu.import("resource://gre/modules/ctypes.jsm");
-Cu.import("resource://gre/modules/XPCOMUtils.jsm");
-Cu.import("resource://gre/modules/Services.jsm");
-XPCOMUtils.defineLazyGetter(this, "log", () => {
- let ConsoleAPI = Cu.import("resource://gre/modules/Console.jsm", {}).ConsoleAPI;
- let consoleOptions = {
- maxLogLevelPref: "browser.esedbreader.loglevel",
- prefix: "ESEDBReader",
- };
- return new ConsoleAPI(consoleOptions);
-});
-
-// We have a globally unique identifier for ESE instances. A new one
-// is used for each different database opened.
-let gESEInstanceCounter = 0;
-
-// We limit the length of strings that we read from databases.
-const MAX_STR_LENGTH = 64 * 1024;
-
-// Kernel-related types:
-const KERNEL = {};
-KERNEL.FILETIME = new ctypes.StructType("FILETIME", [
- {dwLowDateTime: ctypes.uint32_t},
- {dwHighDateTime: ctypes.uint32_t}
-]);
-KERNEL.SYSTEMTIME = new ctypes.StructType("SYSTEMTIME", [
- {wYear: ctypes.uint16_t},
- {wMonth: ctypes.uint16_t},
- {wDayOfWeek: ctypes.uint16_t},
- {wDay: ctypes.uint16_t},
- {wHour: ctypes.uint16_t},
- {wMinute: ctypes.uint16_t},
- {wSecond: ctypes.uint16_t},
- {wMilliseconds: ctypes.uint16_t}
-]);
-
-// DB column types, cribbed from the ESE header
-var COLUMN_TYPES = {
- JET_coltypBit: 1, /* True, False, or NULL */
- JET_coltypUnsignedByte: 2, /* 1-byte integer, unsigned */
- JET_coltypShort: 3, /* 2-byte integer, signed */
- JET_coltypLong: 4, /* 4-byte integer, signed */
- JET_coltypCurrency: 5, /* 8 byte integer, signed */
- JET_coltypIEEESingle: 6, /* 4-byte IEEE single precision */
- JET_coltypIEEEDouble: 7, /* 8-byte IEEE double precision */
- JET_coltypDateTime: 8, /* Integral date, fractional time */
- JET_coltypBinary: 9, /* Binary data, < 255 bytes */
- JET_coltypText: 10, /* ANSI text, case insensitive, < 255 bytes */
- JET_coltypLongBinary: 11, /* Binary data, long value */
- JET_coltypLongText: 12, /* ANSI text, long value */
-
- JET_coltypUnsignedLong: 14, /* 4-byte unsigned integer */
- JET_coltypLongLong: 15, /* 8-byte signed integer */
- JET_coltypGUID: 16, /* 16-byte globally unique identifier */
-};
-
-// Not very efficient, but only used for error messages
-function getColTypeName(numericValue) {
- return Object.keys(COLUMN_TYPES).find(t => COLUMN_TYPES[t] == numericValue) || "unknown";
-}
-
-// All type constants and method wrappers go on this object:
-const ESE = {};
-ESE.JET_ERR = ctypes.long;
-ESE.JET_PCWSTR = ctypes.char16_t.ptr;
-// The ESE header calls this JET_API_PTR, but because it isn't ever used as a
-// pointer and because OS.File code implies that the name you give a type
-// matters, I opted for a different name.
-// Note that this is defined differently on 32 vs. 64-bit in the header.
-ESE.JET_API_ITEM = ctypes.voidptr_t.size == 4 ? ctypes.unsigned_long : ctypes.uint64_t;
-ESE.JET_INSTANCE = ESE.JET_API_ITEM;
-ESE.JET_SESID = ESE.JET_API_ITEM;
-ESE.JET_TABLEID = ESE.JET_API_ITEM;
-ESE.JET_COLUMNID = ctypes.unsigned_long;
-ESE.JET_GRBIT = ctypes.unsigned_long;
-ESE.JET_COLTYP = ctypes.unsigned_long;
-ESE.JET_DBID = ctypes.unsigned_long;
-
-ESE.JET_COLUMNDEF = new ctypes.StructType("JET_COLUMNDEF", [
- {"cbStruct": ctypes.unsigned_long},
- {"columnid": ESE.JET_COLUMNID },
- {"coltyp": ESE.JET_COLTYP },
- {"wCountry": ctypes.unsigned_short }, // sepcifies the country/region for the column definition
- {"langid": ctypes.unsigned_short },
- {"cp": ctypes.unsigned_short },
- {"wCollate": ctypes.unsigned_short }, /* Must be 0 */
- {"cbMax": ctypes.unsigned_long },
- {"grbit": ESE.JET_GRBIT }
-]);
-
-// Track open databases
-let gOpenDBs = new Map();
-
-// Track open libraries
-let gLibs = {};
-this.ESE = ESE; // Required for tests.
-this.KERNEL = KERNEL; // ditto
-this.gLibs = gLibs; // ditto
-
-function convertESEError(errorCode) {
- switch (errorCode) {
- case -1213 /* JET_errPageSizeMismatch */:
- case -1002 /* JET_errInvalidName*/:
- case -1507 /* JET_errColumnNotFound */:
- // The DB format has changed and we haven't updated this migration code:
- return "The database format has changed, error code: " + errorCode;
- case -1207 /* JET_errDatabaseLocked */:
- case -1302 /* JET_errTableLocked */:
- return "The database or table is locked, error code: " + errorCode;
- case -1809 /* JET_errPermissionDenied*/:
- case -1907 /* JET_errAccessDenied */:
- return "Access or permission denied, error code: " + errorCode;
- case -1044 /* JET_errInvalidFilename */:
- return "Invalid file name";
- case -1811 /* JET_errFileNotFound */:
- return "File not found";
- case -550 /* JET_errDatabaseDirtyShutdown */:
- return "Database in dirty shutdown state (without the requisite logs?)";
- case -514 /* JET_errBadLogVersion */:
- return "Database log version does not match the version of ESE in use.";
- default:
- return "Unknown error: " + errorCode;
- }
-}
-
-function handleESEError(method, methodName, shouldThrow = true, errorLog = true) {
- return function () {
- let rv;
- try {
- rv = method.apply(null, arguments);
- } catch (ex) {
- log.error("Error calling into ctypes method", methodName, ex);
- throw ex;
- }
- let resultCode = parseInt(rv.toString(10), 10);
- if (resultCode < 0) {
- if (errorLog) {
- log.error("Got error " + resultCode + " calling " + methodName);
- }
- if (shouldThrow) {
- throw new Error(convertESEError(rv));
- }
- } else if (resultCode > 0 && errorLog) {
- log.warn("Got warning " + resultCode + " calling " + methodName);
- }
- return resultCode;
- };
-}
-
-
-function declareESEFunction(methodName, ...args) {
- let declaration = ["Jet" + methodName, ctypes.winapi_abi, ESE.JET_ERR].concat(args);
- let ctypeMethod = gLibs.ese.declare.apply(gLibs.ese, declaration);
- ESE[methodName] = handleESEError(ctypeMethod, methodName);
- ESE["FailSafe" + methodName] = handleESEError(ctypeMethod, methodName, false);
- ESE["Manual" + methodName] = handleESEError(ctypeMethod, methodName, false, false);
-}
-
-function declareESEFunctions() {
- declareESEFunction("GetDatabaseFileInfoW", ESE.JET_PCWSTR, ctypes.voidptr_t,
- ctypes.unsigned_long, ctypes.unsigned_long);
-
- declareESEFunction("GetSystemParameterW", ESE.JET_INSTANCE, ESE.JET_SESID,
- ctypes.unsigned_long, ESE.JET_API_ITEM.ptr,
- ESE.JET_PCWSTR, ctypes.unsigned_long);
- declareESEFunction("SetSystemParameterW", ESE.JET_INSTANCE.ptr,
- ESE.JET_SESID, ctypes.unsigned_long, ESE.JET_API_ITEM,
- ESE.JET_PCWSTR);
- declareESEFunction("CreateInstanceW", ESE.JET_INSTANCE.ptr, ESE.JET_PCWSTR);
- declareESEFunction("Init", ESE.JET_INSTANCE.ptr);
-
- declareESEFunction("BeginSessionW", ESE.JET_INSTANCE, ESE.JET_SESID.ptr,
- ESE.JET_PCWSTR, ESE.JET_PCWSTR);
- declareESEFunction("AttachDatabaseW", ESE.JET_SESID, ESE.JET_PCWSTR,
- ESE.JET_GRBIT);
- declareESEFunction("DetachDatabaseW", ESE.JET_SESID, ESE.JET_PCWSTR);
- declareESEFunction("OpenDatabaseW", ESE.JET_SESID, ESE.JET_PCWSTR,
- ESE.JET_PCWSTR, ESE.JET_DBID.ptr, ESE.JET_GRBIT);
- declareESEFunction("OpenTableW", ESE.JET_SESID, ESE.JET_DBID, ESE.JET_PCWSTR,
- ctypes.voidptr_t, ctypes.unsigned_long, ESE.JET_GRBIT,
- ESE.JET_TABLEID.ptr);
-
- declareESEFunction("GetColumnInfoW", ESE.JET_SESID, ESE.JET_DBID,
- ESE.JET_PCWSTR, ESE.JET_PCWSTR, ctypes.voidptr_t,
- ctypes.unsigned_long, ctypes.unsigned_long);
-
- declareESEFunction("Move", ESE.JET_SESID, ESE.JET_TABLEID, ctypes.long,
- ESE.JET_GRBIT);
-
- declareESEFunction("RetrieveColumn", ESE.JET_SESID, ESE.JET_TABLEID,
- ESE.JET_COLUMNID, ctypes.voidptr_t, ctypes.unsigned_long,
- ctypes.unsigned_long.ptr, ESE.JET_GRBIT, ctypes.voidptr_t);
-
- declareESEFunction("CloseTable", ESE.JET_SESID, ESE.JET_TABLEID);
- declareESEFunction("CloseDatabase", ESE.JET_SESID, ESE.JET_DBID,
- ESE.JET_GRBIT);
-
- declareESEFunction("EndSession", ESE.JET_SESID, ESE.JET_GRBIT);
-
- declareESEFunction("Term", ESE.JET_INSTANCE);
-}
-
-function unloadLibraries() {
- log.debug("Unloading");
- if (gOpenDBs.size) {
- log.error("Shouldn't unload libraries before DBs are closed!");
- for (let db of gOpenDBs.values()) {
- db._close();
- }
- }
- for (let k of Object.keys(ESE)) {
- delete ESE[k];
- }
- gLibs.ese.close();
- gLibs.kernel.close();
- delete gLibs.ese;
- delete gLibs.kernel;
-}
-
-function loadLibraries() {
- Services.obs.addObserver(unloadLibraries, "xpcom-shutdown", false);
- gLibs.ese = ctypes.open("esent.dll");
- gLibs.kernel = ctypes.open("kernel32.dll");
- KERNEL.FileTimeToSystemTime = gLibs.kernel.declare("FileTimeToSystemTime",
- ctypes.default_abi, ctypes.int, KERNEL.FILETIME.ptr, KERNEL.SYSTEMTIME.ptr);
-
- declareESEFunctions();
-}
-
-function ESEDB(rootPath, dbPath, logPath) {
- log.info("Created db");
- this.rootPath = rootPath;
- this.dbPath = dbPath;
- this.logPath = logPath;
- this._references = 0;
- this._init();
-}
-
-ESEDB.prototype = {
- rootPath: null,
- dbPath: null,
- logPath: null,
- _opened: false,
- _attached: false,
- _sessionCreated: false,
- _instanceCreated: false,
- _dbId: null,
- _sessionId: null,
- _instanceId: null,
-
- _init() {
- if (!gLibs.ese) {
- loadLibraries();
- }
- this.incrementReferenceCounter();
- this._internalOpen();
- },
-
- _internalOpen() {
- try {
- let dbinfo = new ctypes.unsigned_long();
- ESE.GetDatabaseFileInfoW(this.dbPath, dbinfo.address(),
- ctypes.unsigned_long.size, 17);
-
- let pageSize = ctypes.UInt64.lo(dbinfo.value);
- ESE.SetSystemParameterW(null, 0, 64 /* JET_paramDatabasePageSize*/,
- pageSize, null);
-
- this._instanceId = new ESE.JET_INSTANCE();
- ESE.CreateInstanceW(this._instanceId.address(),
- "firefox-dbreader-" + (gESEInstanceCounter++));
- this._instanceCreated = true;
-
- ESE.SetSystemParameterW(this._instanceId.address(), 0,
- 0 /* JET_paramSystemPath*/, 0, this.rootPath);
- ESE.SetSystemParameterW(this._instanceId.address(), 0,
- 1 /* JET_paramTempPath */, 0, this.rootPath);
- ESE.SetSystemParameterW(this._instanceId.address(), 0,
- 2 /* JET_paramLogFilePath*/, 0, this.logPath);
-
- // Shouldn't try to call JetTerm if the following call fails.
- this._instanceCreated = false;
- ESE.Init(this._instanceId.address());
- this._instanceCreated = true;
- this._sessionId = new ESE.JET_SESID();
- ESE.BeginSessionW(this._instanceId, this._sessionId.address(), null,
- null);
- this._sessionCreated = true;
-
- const JET_bitDbReadOnly = 1;
- ESE.AttachDatabaseW(this._sessionId, this.dbPath, JET_bitDbReadOnly);
- this._attached = true;
- this._dbId = new ESE.JET_DBID();
- ESE.OpenDatabaseW(this._sessionId, this.dbPath, null,
- this._dbId.address(), JET_bitDbReadOnly);
- this._opened = true;
- } catch (ex) {
- try {
- this._close();
- } catch (innerException) {
- Cu.reportError(innerException);
- }
- // Make sure caller knows we failed.
- throw ex;
- }
- gOpenDBs.set(this.dbPath, this);
- },
-
- checkForColumn(tableName, columnName) {
- if (!this._opened) {
- throw new Error("The database was closed!");
- }
-
- let columnInfo;
- try {
- columnInfo = this._getColumnInfo(tableName, [{name: columnName}]);
- } catch (ex) {
- return null;
- }
- return columnInfo[0];
- },
-
- tableExists(tableName) {
- if (!this._opened) {
- throw new Error("The database was closed!");
- }
-
- let tableId = new ESE.JET_TABLEID();
- let rv = ESE.ManualOpenTableW(this._sessionId, this._dbId, tableName, null,
- 0, 4 /* JET_bitTableReadOnly */,
- tableId.address());
- if (rv == -1305 /* JET_errObjectNotFound */) {
- return false;
- }
- if (rv < 0) {
- log.error("Got error " + rv + " calling OpenTableW");
- throw new Error(convertESEError(rv));
- }
-
- if (rv > 0) {
- log.error("Got warning " + rv + " calling OpenTableW");
- }
- ESE.FailSafeCloseTable(this._sessionId, tableId);
- return true;
- },
-
- tableItems: function*(tableName, columns) {
- if (!this._opened) {
- throw new Error("The database was closed!");
- }
-
- let tableOpened = false;
- let tableId;
- try {
- tableId = this._openTable(tableName);
- tableOpened = true;
-
- let columnInfo = this._getColumnInfo(tableName, columns);
-
- let rv = ESE.ManualMove(this._sessionId, tableId,
- -2147483648 /* JET_MoveFirst */, 0);
- if (rv == -1603 /* JET_errNoCurrentRecord */) {
- // There are no rows in the table.
- this._closeTable(tableId);
- return;
- }
- if (rv != 0) {
- throw new Error(convertESEError(rv));
- }
-
- do {
- let rowContents = {};
- for (let column of columnInfo) {
- let [buffer, bufferSize] = this._getBufferForColumn(column);
- // We handle errors manually so we accurately deal with NULL values.
- let err = ESE.ManualRetrieveColumn(this._sessionId, tableId,
- column.id, buffer.address(),
- bufferSize, null, 0, null);
- rowContents[column.name] = this._convertResult(column, buffer, err);
- }
- yield rowContents;
- } while (ESE.ManualMove(this._sessionId, tableId, 1 /* JET_MoveNext */, 0) === 0);
- } catch (ex) {
- if (tableOpened) {
- this._closeTable(tableId);
- }
- throw ex;
- }
- this._closeTable(tableId);
- },
-
- _openTable(tableName) {
- let tableId = new ESE.JET_TABLEID();
- ESE.OpenTableW(this._sessionId, this._dbId, tableName, null,
- 0, 4 /* JET_bitTableReadOnly */, tableId.address());
- return tableId;
- },
-
- _getBufferForColumn(column) {
- let buffer;
- if (column.type == "string") {
- let wchar_tArray = ctypes.ArrayType(ctypes.char16_t);
- // size on the column is in bytes, 2 bytes to a wchar, so:
- let charCount = column.dbSize >> 1;
- buffer = new wchar_tArray(charCount);
- } else if (column.type == "boolean") {
- buffer = new ctypes.uint8_t();
- } else if (column.type == "date") {
- buffer = new KERNEL.FILETIME();
- } else if (column.type == "guid") {
- let byteArray = ctypes.ArrayType(ctypes.uint8_t);
- buffer = new byteArray(column.dbSize);
- } else {
- throw new Error("Unknown type " + column.type);
- }
- return [buffer, buffer.constructor.size];
- },
-
- _convertResult(column, buffer, err) {
- if (err != 0) {
- if (err == 1004) {
- // Deal with null values:
- buffer = null;
- } else {
- Cu.reportError("Unexpected JET error: " + err + ";" + " retrieving value for column " + column.name);
- throw new Error(convertESEError(err));
- }
- }
- if (column.type == "string") {
- return buffer ? buffer.readString() : "";
- }
- if (column.type == "boolean") {
- return buffer ? (buffer.value == 255) : false;
- }
- if (column.type == "guid") {
- if (buffer.length != 16) {
- Cu.reportError("Buffer size for guid field " + column.id + " should have been 16!");
- return "";
- }
- let rv = "{";
- for (let i = 0; i < 16; i++) {
- if (i == 4 || i == 6 || i == 8 || i == 10) {
- rv += "-";
- }
- let byteValue = buffer.addressOfElement(i).contents;
- // Ensure there's a leading 0
- rv += ("0" + byteValue.toString(16)).substr(-2);
- }
- return rv + "}";
- }
- if (column.type == "date") {
- if (!buffer) {
- return null;
- }
- let systemTime = new KERNEL.SYSTEMTIME();
- let result = KERNEL.FileTimeToSystemTime(buffer.address(), systemTime.address());
- if (result == 0) {
- throw new Error(ctypes.winLastError);
- }
-
- // System time is in UTC, so we use Date.UTC to get milliseconds from epoch,
- // then divide by 1000 to get seconds, and round down:
- return new Date(Date.UTC(systemTime.wYear,
- systemTime.wMonth - 1,
- systemTime.wDay,
- systemTime.wHour,
- systemTime.wMinute,
- systemTime.wSecond,
- systemTime.wMilliseconds));
- }
- return undefined;
- },
-
- _getColumnInfo(tableName, columns) {
- let rv = [];
- for (let column of columns) {
- let columnInfoFromDB = new ESE.JET_COLUMNDEF();
- ESE.GetColumnInfoW(this._sessionId, this._dbId, tableName, column.name,
- columnInfoFromDB.address(), ESE.JET_COLUMNDEF.size, 0 /* JET_ColInfo */);
- let dbType = parseInt(columnInfoFromDB.coltyp.toString(10), 10);
- let dbSize = parseInt(columnInfoFromDB.cbMax.toString(10), 10);
- if (column.type == "string") {
- if (dbType != COLUMN_TYPES.JET_coltypLongText &&
- dbType != COLUMN_TYPES.JET_coltypText) {
- throw new Error("Invalid column type for column " + column.name +
- "; expected text type, got type " + getColTypeName(dbType));
- }
- if (dbSize > MAX_STR_LENGTH) {
- throw new Error("Column " + column.name + " has more than 64k data in it. This API is not designed to handle data that large.");
- }
- } else if (column.type == "boolean") {
- if (dbType != COLUMN_TYPES.JET_coltypBit) {
- throw new Error("Invalid column type for column " + column.name +
- "; expected bit type, got type " + getColTypeName(dbType));
- }
- } else if (column.type == "date") {
- if (dbType != COLUMN_TYPES.JET_coltypLongLong) {
- throw new Error("Invalid column type for column " + column.name +
- "; expected long long type, got type " + getColTypeName(dbType));
- }
- } else if (column.type == "guid") {
- if (dbType != COLUMN_TYPES.JET_coltypGUID) {
- throw new Error("Invalid column type for column " + column.name +
- "; expected guid type, got type " + getColTypeName(dbType));
- }
- } else if (column.type) {
- throw new Error("Unknown column type " + column.type + " requested for column " +
- column.name + ", don't know what to do.");
- }
-
- rv.push({name: column.name, id: columnInfoFromDB.columnid, type: column.type, dbSize, dbType});
- }
- return rv;
- },
-
- _closeTable(tableId) {
- ESE.FailSafeCloseTable(this._sessionId, tableId);
- },
-
- _close() {
- this._internalClose();
- gOpenDBs.delete(this.dbPath);
- },
-
- _internalClose() {
- if (this._opened) {
- log.debug("close db");
- ESE.FailSafeCloseDatabase(this._sessionId, this._dbId, 0);
- log.debug("finished close db");
- this._opened = false;
- }
- if (this._attached) {
- log.debug("detach db");
- ESE.FailSafeDetachDatabaseW(this._sessionId, this.dbPath);
- this._attached = false;
- }
- if (this._sessionCreated) {
- log.debug("end session");
- ESE.FailSafeEndSession(this._sessionId, 0);
- this._sessionCreated = false;
- }
- if (this._instanceCreated) {
- log.debug("term");
- ESE.FailSafeTerm(this._instanceId);
- this._instanceCreated = false;
- }
- },
-
- incrementReferenceCounter() {
- this._references++;
- },
-
- decrementReferenceCounter() {
- this._references--;
- if (this._references <= 0) {
- this._close();
- }
- },
-};
-
-let ESEDBReader = {
- openDB(rootDir, dbFile, logDir) {
- let dbFilePath = dbFile.path;
- if (gOpenDBs.has(dbFilePath)) {
- let db = gOpenDBs.get(dbFilePath);
- db.incrementReferenceCounter();
- return db;
- }
- // ESE is really picky about the trailing slashes according to the docs,
- // so we do as we're told and ensure those are there:
- return new ESEDB(rootDir.path + "\\", dbFilePath, logDir.path + "\\");
- },
-
- closeDB(db) {
- db.decrementReferenceCounter();
- },
-
- COLUMN_TYPES,
-};
-