@mherod/get-cookie 1.0.6 → 1.1.2

package/index.js

@@ -53,44 +53,21 @@ async function getFirefoxCookie({name, domain}) {
     if (process.env.VERBOSE) {
         console.log(`Trying Firefox cookie ${name} for domain ${domain}`);
     }
-    return await new Promise((resolve, reject) => {
-        let sql;
-        sql = "SELECT value FROM moz_cookies";
-        if (typeof name === 'string' || typeof domain === 'string') {
-            sql += ` WHERE `;
-            if (typeof name === 'string') {
-                sql += `name = '${name}'`;
-                if (typeof domain === 'string') {
-                    sql += ` AND `;
-                }
-            }
+    let sql;
+    sql = "SELECT value FROM moz_cookies";
+    if (typeof name === 'string' || typeof domain === 'string') {
+        sql += ` WHERE `;
+        if (typeof name === 'string') {
+            sql += `name = '${name}'`;
             if (typeof domain === 'string') {
-                sql += `host LIKE '${domain}';`;
+                sql += ` AND `;
             }
         }
-        const command = `sqlite3 "${file}" "${sql}"`;
-        if (process.env.VERBOSE) {
-            console.log(command);
+        if (typeof domain === 'string') {
+            sql += `host LIKE '${domain}';`;
         }
-        exec(command, {encoding: 'binary', maxBuffer: 5 * 1024}, (error, stdout, stderr) => {
-            if (error) {
-                reject(error);
-                return;
-            }
-            if (stderr) {
-                reject(error);
-                return;
-            }
-            let stdoutAsBuffer = stdout;
-            if (typeof stdoutAsBuffer === 'string' && stdoutAsBuffer.length > 0) {
-                // noinspection JSCheckFunctionSignatures
-                stdoutAsBuffer = Buffer.from(stdoutAsBuffer, 'binary').slice(0, -1);
-            }
-            if (stdoutAsBuffer && stdoutAsBuffer.length > 0) {
-                resolve(stdoutAsBuffer);
-            }
-        });
-    });
+    }
+    return await doSqliteQuery1(file, sql);
 }
 
 async function getEncryptedChromeCookie(name, domain) {
@@ -100,33 +77,64 @@ async function getEncryptedChromeCookie(name, domain) {
     if (domain && typeof domain !== 'string') {
         throw new Error('domain must be a string');
     }
-    return await new Promise((resolve, reject) => {
-        const file = `${process.env.HOME}/Library/Application Support/Google/Chrome/Default/Cookies`;
-        if (!fs.existsSync(file)) {
-            reject(new Error(`File ${file} does not exist`));
-            return;
+    const file = `${process.env.HOME}/Library/Application Support/Google/Chrome/Default/Cookies`;
+    if (!fs.existsSync(file)) {
+        throw new Error(`File ${file} does not exist`);
+    }
+    if (process.env.VERBOSE) {
+        console.log(`Trying Chrome cookie ${name} for domain ${domain}`);
+    }
+    let sql;
+    sql = `SELECT encrypted_value FROM cookies`;
+    if (typeof name === 'string' || typeof domain === 'string') {
+        sql += ` WHERE `;
+        if (typeof name === 'string') {
+            sql += `name = '${name}'`;
+            if (typeof domain === 'string') {
+                sql += ` AND `;
+            }
         }
-        if (process.env.VERBOSE) {
-            console.log(`Trying Chrome cookie ${name} for domain ${domain}`);
+        if (typeof domain === 'string') {
+            sql += `host_key LIKE '${domain}';`;
         }
-        let sql;
-        sql = `SELECT encrypted_value FROM cookies`;
-        if (typeof name === 'string' || typeof domain === 'string') {
-            sql += ` WHERE `;
-            if (typeof name === 'string') {
-                sql += `name = '${name}'`;
-                if (typeof domain === 'string') {
-                    sql += ` AND `;
+    }
+    return await doSqliteQuery1(file, sql);
+}
+
+async function doSqliteQuery1(file, sql) {
+    const sqlite3 = require('sqlite3');
+    const db = new sqlite3.Database(file);
+    return new Promise((resolve, reject) => {
+        db.all(sql, (err, rows) => {
+            if (err) {
+                reject(err);
+                return;
+            }
+            const rows1 = rows;
+            if (rows1.length === 0) {
+                if (process.env.VERBOSE) {
+                    console.log(`No rows found`);
                 }
+                resolve(null);
+                return;
             }
-            if (typeof domain === 'string') {
-                sql += `host_key LIKE '${domain}';`;
+            if (Array.isArray(rows1)) {
+                // noinspection JSCheckFunctionSignatures
+                const [value] = rows1.flatMap(row => Object.values(row));
+                resolve(Buffer.from(value));
+                return;
             }
-        }
-        const command = `sqlite3 "${file}" "${sql}"`;
-        if (process.env.VERBOSE) {
-            console.log(command);
-        }
+            resolve(rows1);
+        });
+    });
+}
+
+async function doSqliteQuery(file, sql) {
+    const command = `sqlite3 "${file}" "${sql}"`;
+    if (process.env.VERBOSE) {
+        console.log(command);
+    }
+    return await new Promise((resolve, reject) => {
         exec(command, {encoding: 'binary', maxBuffer: 5 * 1024}, (error, stdout, stderr) => {
             if (error) {
                 reject(error);
@@ -156,63 +164,72 @@ async function getEncryptedChromeCookie(name, domain) {
  */
 async function decrypt(password, encryptedData) {
     if (typeof password !== 'string') {
-        throw new Error('password must be a string');
+        throw new Error('password must be a string: ' + password);
     }
-    if (typeof encryptedData !== 'object') {
-        throw new Error('encryptedData must be a object');
+    const encryptedData1 = encryptedData;
+    if (encryptedData1 == null || typeof encryptedData1 !== 'object') {
+        throw new Error('encryptedData must be a object: ' + encryptedData1);
     }
     if (process.env.VERBOSE) {
         console.log(`Trying to decrypt with password ${password}`);
     }
     return await new Promise((resolve, reject) => {
         crypto.pbkdf2(password, 'saltysalt', 1003, 16, 'sha1', (error, buffer) => {
-            if (error) {
-                if (process.env.VERBOSE) {
-                    console.log("Error doing pbkdf2", error);
+            try {
+                if (error) {
+                    if (process.env.VERBOSE) {
+                        console.log("Error doing pbkdf2", error);
+                    }
+                    reject(error);
+                    return;
                 }
-                reject(error);
-                return;
-            }
-            if (buffer.length !== 16) {
-                if (process.env.VERBOSE) {
-                    console.log("Error doing pbkdf2, buffer length is not 16", buffer.length);
+
+                if (buffer.length !== 16) {
+                    if (process.env.VERBOSE) {
+                        console.log("Error doing pbkdf2, buffer length is not 16", buffer.length);
+                    }
+                    reject(new Error('Buffer length is not 16'));
+                    return;
                 }
-                reject(new Error('Buffer length is not 16'));
-                return;
-            }
 
-            const iv = new Buffer.from(new Array(17).join(' '), 'binary');
-            const decipher = crypto.createDecipheriv('aes-128-cbc', buffer, iv);
-            decipher.setAutoPadding(false);
-            encryptedData = encryptedData.slice(3);
+                const iv = new Buffer.from(new Array(17).join(' '), 'binary');
+                const decipher = crypto.createDecipheriv('aes-128-cbc', buffer, iv);
+                decipher.setAutoPadding(false);
 
-            if (encryptedData.length % 16 !== 0) {
-                if (process.env.VERBOSE) {
-                    console.log("Error doing pbkdf2, encryptedData length is not a multiple of 16", encryptedData.length);
+                if (encryptedData1 && encryptedData1.slice) {
+                    encryptedData = encryptedData1.slice(3);
                 }
-                reject(new Error('encryptedData length is not a multiple of 16'));
-                return;
-            }
 
-            let decoded = decipher.update(encryptedData, 'binary', 'utf8');
-            // let decoded = decipher.update(encryptedData);
-            try {
-                decipher.final('utf-8');
-            } catch (e) {
-                if (process.env.VERBOSE) {
-                    console.log("Error doing decipher.final()", e);
+                if (encryptedData1.length % 16 !== 0) {
+                    if (process.env.VERBOSE) {
+                        console.log("Error doing pbkdf2, encryptedData length is not a multiple of 16", encryptedData1.length);
+                    }
+                    reject(new Error('encryptedData length is not a multiple of 16'));
+                    return;
                 }
-                reject(e);
-                return;
-            }
 
-            let padding = decoded[decoded.length - 1];
-            if (padding) {
-                decoded = decoded.slice(0, decoded.length - padding);
+                let decoded = decipher.update(encryptedData1, 'binary', 'utf8');
+                // let decoded = decipher.update(encryptedData);
+                try {
+                    decipher.final('utf-8');
+                } catch (e) {
+                    if (process.env.VERBOSE) {
+                        console.log("Error doing decipher.final()", e);
+                    }
+                    reject(e);
+                    return;
+                }
+
+                let padding = decoded[decoded.length - 1];
+                if (padding) {
+                    decoded = decoded.slice(0, 0 - padding);
+                }
+                // noinspection JSCheckFunctionSignatures
+                decoded = decoded.toString('utf8');
+                resolve(decoded);
+            } catch (e) {
+                reject(e);
             }
-            // noinspection JSCheckFunctionSignatures
-            decoded = decoded.toString('utf8');
-            resolve(decoded)
         });
     });
 }
@@ -230,8 +247,10 @@ async function getChromeCookie({name, domain}) {
     if (domain && typeof domain !== 'string') {
         throw new Error('domain must be a string');
     }
-    const password = await getChromePassword();
-    const encryptedData = await getEncryptedChromeCookie(name, domain);
+    const chromePasswordPromise = getChromePassword();
+    const encryptedChromeCookiePromise = getEncryptedChromeCookie(name, domain);
+    const password = await chromePasswordPromise;
+    const encryptedData = await encryptedChromeCookiePromise;
     if (process.env.VERBOSE) {
         console.log("Received encrypted", encryptedData);
     }
@@ -239,7 +258,9 @@ async function getChromeCookie({name, domain}) {
     try {
         s = await decrypt(password, encryptedData);
     } catch (e) {
-        console.error(e);
+        if (process.env.VERBOSE) {
+            console.error("Error decrypting", e);
+        }
         throw new Error('Failed to decrypt');
     }
     if (process.env.VERBOSE) {
@@ -293,7 +314,7 @@ module.exports = {
 if (process.argv) {
     if (process.argv.length > 2) {
         const name = process.argv[2];
-        const domain = process.argv[3];
+        const domain = process.argv[3] ?? '%';
 
         if (process.argv.includes('--verbose')) {
             process.env.VERBOSE = "true";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mherod/get-cookie",
-  "version": "1.0.6",
+  "version": "1.1.2",
   "description": "Node.js module for querying a local user's Chrome cookie",
   "main": "index.js",
   "bin": "index.js",
@@ -9,13 +9,12 @@
   },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",
-    "installForHomebrew": "pkg . --target host --out-path $HOMEBREW_FORMULA_PREFIX/bin/ --debug",
-    "installForMac": "pkg . --target node17-macos-x64 --out-path /usr/local/bin/ --debug",
-    "installForHost": "pkg . --target host --out-path /usr/local/bin/ --debug"
+    "installGlobal": "rm -rf /usr/local/bin/get-cookie ; npm install --force --global ."
   },
   "keywords": [],
   "author": "Matthew Herod",
   "license": "ISC",
   "dependencies": {
+    "sqlite3": "^5.0.8"
   }
 }