Spaces:
Sleeping
Sleeping
| var MODES = require('./modes') | |
| var AuthCipher = require('./authCipher') | |
| var Buffer = require('safe-buffer').Buffer | |
| var StreamCipher = require('./streamCipher') | |
| var Transform = require('cipher-base') | |
| var aes = require('./aes') | |
| var ebtk = require('evp_bytestokey') | |
| var inherits = require('inherits') | |
| function Cipher (mode, key, iv) { | |
| Transform.call(this) | |
| this._cache = new Splitter() | |
| this._cipher = new aes.AES(key) | |
| this._prev = Buffer.from(iv) | |
| this._mode = mode | |
| this._autopadding = true | |
| } | |
| inherits(Cipher, Transform) | |
| Cipher.prototype._update = function (data) { | |
| this._cache.add(data) | |
| var chunk | |
| var thing | |
| var out = [] | |
| while ((chunk = this._cache.get())) { | |
| thing = this._mode.encrypt(this, chunk) | |
| out.push(thing) | |
| } | |
| return Buffer.concat(out) | |
| } | |
| var PADDING = Buffer.alloc(16, 0x10) | |
| Cipher.prototype._final = function () { | |
| var chunk = this._cache.flush() | |
| if (this._autopadding) { | |
| chunk = this._mode.encrypt(this, chunk) | |
| this._cipher.scrub() | |
| return chunk | |
| } | |
| if (!chunk.equals(PADDING)) { | |
| this._cipher.scrub() | |
| throw new Error('data not multiple of block length') | |
| } | |
| } | |
| Cipher.prototype.setAutoPadding = function (setTo) { | |
| this._autopadding = !!setTo | |
| return this | |
| } | |
| function Splitter () { | |
| this.cache = Buffer.allocUnsafe(0) | |
| } | |
| Splitter.prototype.add = function (data) { | |
| this.cache = Buffer.concat([this.cache, data]) | |
| } | |
| Splitter.prototype.get = function () { | |
| if (this.cache.length > 15) { | |
| var out = this.cache.slice(0, 16) | |
| this.cache = this.cache.slice(16) | |
| return out | |
| } | |
| return null | |
| } | |
| Splitter.prototype.flush = function () { | |
| var len = 16 - this.cache.length | |
| var padBuff = Buffer.allocUnsafe(len) | |
| var i = -1 | |
| while (++i < len) { | |
| padBuff.writeUInt8(len, i) | |
| } | |
| return Buffer.concat([this.cache, padBuff]) | |
| } | |
| function createCipheriv (suite, password, iv) { | |
| var config = MODES[suite.toLowerCase()] | |
| if (!config) throw new TypeError('invalid suite type') | |
| if (typeof password === 'string') password = Buffer.from(password) | |
| if (password.length !== config.key / 8) throw new TypeError('invalid key length ' + password.length) | |
| if (typeof iv === 'string') iv = Buffer.from(iv) | |
| if (config.mode !== 'GCM' && iv.length !== config.iv) throw new TypeError('invalid iv length ' + iv.length) | |
| if (config.type === 'stream') { | |
| return new StreamCipher(config.module, password, iv) | |
| } else if (config.type === 'auth') { | |
| return new AuthCipher(config.module, password, iv) | |
| } | |
| return new Cipher(config.module, password, iv) | |
| } | |
| function createCipher (suite, password) { | |
| var config = MODES[suite.toLowerCase()] | |
| if (!config) throw new TypeError('invalid suite type') | |
| var keys = ebtk(password, false, config.key, config.iv) | |
| return createCipheriv(suite, keys.key, keys.iv) | |
| } | |
| exports.createCipheriv = createCipheriv | |
| exports.createCipher = createCipher | |