index.js

/**
 * @author anchao01
 * @description 根据地址获取chrome相关cookie
 * @date 2018年6月17日
 */

var fs = require( 'fs' )
var sqlite3 = require( 'sqlite3' ).verbose()
var url = require('url')
var crypto = require('crypto')
const path = require('path')

var KEY_LENGTH = 16
var SALT = 'saltysalt'
var IV = new Buffer(new Array(KEY_LENGTH + 1).join(' '))

var DBIns = getDBIns()

function getCookieInfo(cookiePath) {
    var info = {
        path: '',
        key: ''
    }

    var password = null
    var iterations = 0

    return new Promise(function(resolve) {
        switch( process.platform ) {
            case 'darwin':
                info.path = cookiePath['darwin']
                password = require('keytar').getPassword('Chrome Safe Storage', 'Chrome')
                iterations = 1003
                break;
            case 'linux':
                info.path = cookiePath['linux']
                password = 'peanuts'
                iterations = 1
                break;
            default:
                console.error( 'Currently your OS is not supported!' )
                process.exit(1);
        }

        if (password instanceof Promise) {
            return password
            .then(function(psd) {
                info.key = getDerivedKey(psd, iterations)
                resolve(info)
            })
        }

        info.key = getDerivedKey(password, iterations)
        resolve(info)
    })
}

function getDerivedKey(password, iterations) {
    return crypto.pbkdf2Sync(password, SALT, iterations, KEY_LENGTH, 'sha1')
}

function decryptorCookie(key, iv, encryptedCookie) {
    decipher = crypto.createDecipheriv('AES-128-CBC', key, iv)
    decryptedCookie = decipher.update(encryptedCookie.slice(3))

    return decryptedCookie.toString() + decipher.final('utf8')
}

function getDBIns() {
    var db = null
    var dbClose = true

    return {
        getDB: function(path) {
            if (dbClose) {
                if (!fs.existsSync(path)) {
                    throw(new Error('Ensure Chrome is installed on this deveice?'))
                }
                db = new sqlite3.Database(path)
                dbClose = false
            }

            return db
        },
        closeDB: function(callback) {
            if (dbClose) {
                return
            }

            db.close(function(err) {
                dbClose = true

                if (err) {
                    return excuteCallback(callback, err)
                }

                excuteCallback(callback, null, 'closed successfully')
            })
        }
    }
}

function ifMatchHost(rawUrl, hostKey) {
    return rawUrl.indexOf(hostKey) > -1
}

function getDomain(hostname) {
    hostname = hostname.toLowerCase()
    var res = /(\.[a-z]+\.[a-z]+$)/.exec(hostname)

    return res && res[1]
}

function excuteCallback(callback) {
    if (typeof callback !== 'function') {
        return
    }

    var args = Array.prototype.slice.call(arguments, 1)
    return callback.apply(null, args)
}

function findMacCookiePath() {
    const chromePath = process.env.HOME + '/Library/Application Support/Google/Chrome';

    if (!fs.existsSync(chromePath)) {
        throw new Error('Chrome is required to be installed on your Machine')
    }

    const dirs = fs.readdirSync(chromePath).filter(item => item.startsWith('Default'));

    for (let i = 0; i < dirs.length; i++) {
        const cookiePath = path.join(chromePath, dirs[i], 'Cookies')

        if (!fs.existsSync(cookiePath)) {
            continue;
        }

        return cookiePath
    }
}

module.exports = function (urlVal, callback) {
    let cookiePath = {
        'darwin': findMacCookiePath(),
        'linux': process.env.HOME + '/.config/google-chrome/Default'
    };

    if (typeof urlVal === 'object') {
        Object.assign(cookiePath, urlVal.cookiePath);
        urlVal = urlVal.url;
    }

    var urlInfo = url.parse(urlVal)
    var cookies = {}
    var fullCookiesInfo = []
    var hasCallback = typeof callback === 'function'
    var domain = getDomain(urlInfo.hostname)

    return getCookieInfo(cookiePath)
    .then(function(cookieInfo) {
        return new Promise(function (resolve, reject) {
            db = DBIns.getDB(cookieInfo.path)
            db.serialize(function() {
                db.each( "SELECT * FROM cookies where host_key like '%" + domain + "%'", function( err, cookie ) {
                    if (err) {
                        if (hasCallback) {
                            return callback(err)
                        }
                        reject(err);
                        return;
                    }

                    if (cookie.value === '' && cookie.encrypted_value.length > 0) {
                        cookie.value = decryptorCookie(cookieInfo.key, IV, cookie.encrypted_value)
                    }

                    if (ifMatchHost(urlVal, cookie.host_key)) {
                        if (cookies[cookie.name]) {
                            cookies[cookie.name] += '; ' + cookie.value
                        }

                        cookies[cookie.name] = cookie.value
                    }

                    fullCookiesInfo.push(cookie)
                }, function() {
                    DBIns.closeDB(function(err) {
                        if (err) {
                            excuteCallback(callback, err)
                        }
                    })
                    resolve({ cookies, fullCookiesInfo })
                    return excuteCallback(callback, null, cookies, fullCookiesInfo)
                });

            });
        })
    })
}