diff --git a/Ciphers/AffineCipher.js b/Ciphers/AffineCipher.js new file mode 100644 index 0000000000..367cb18597 --- /dev/null +++ b/Ciphers/AffineCipher.js @@ -0,0 +1,95 @@ +/** + * @description - The affine cipher is a type of monoalphabetic substitution cipher, where each letter in an alphabet is mapped to its numeric equivalent, encrypted using a simple mathematical function, and converted back to a letter + * @see - [wiki](https://en.wikipedia.org/wiki/Affine_cipher) + */ + +// Default key for Affine Cipher +const key = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' + +/** + * Fix result for negative value in modulas equation + * @param {Number} n - Constant number + * @param {Number} m - Modulos value + * @return {Number} Return n mod m + */ +function mod (n, m) { + return ((n % m) + m) % m +} + +/** + * Modular multiplicative inverse + * @param {Number} a - A coefficient + * @param {Number} m - Modulos value + * @return {Number} Return modular multiplicative inverse of coefficient a and modulos m + */ +function inverseMod (a, m) { + for (let x = 1; x < m; x++) { + if (mod(a * x, m) === 1) return x + } +} + +/** + * Argument validation + * @param {String} str - String to be checked + * @param {Number} a - A coefficient to be checked + * @param {Number} b - B coefficient to be checked + * @return {Boolean} Result of the checking + */ +function isCorrectFormat (str, a, b) { + if (typeof a !== 'number' || typeof b !== 'number') { + throw new TypeError('Coefficient a, b should be number') + } + + if (typeof str !== 'string') { + throw new TypeError('Argument str should be String') + } + + return true +} +/** + * Find character index based on ASCII order + * @param {String} char - Character index to be found + * @return {Boolean} Character index + */ +function findCharIndex (char) { + return char.toUpperCase().charCodeAt(0) - ('A').charCodeAt(0) +} + +/** + * Encrypt a Affine Cipher + * @param {String} str - String to be encrypted + * @param {Number} a - A coefficient + * @param {Number} b - B coefficient + * @return {String} result - Encrypted string + */ +function encrypt (str, a, b) { + let result = '' + if (isCorrectFormat(str, a, b)) { + for (let x = 0; x < str.length; x++) { + const charIndex = findCharIndex(str[x]) + if (charIndex < 0) result += '-1' + ' ' + else result += mod(a * charIndex + b, 26) + ' ' + } + } + return result.trim() +} +/** + * Decrypt a Affine Cipher + * @param {String} str - String to be decrypted + * @param {Number} a - A coefficient + * @param {Number} b - B coefficient + * @return {String} result - Decrypted string + */ +function decrypt (str, a, b) { + let result = '' + if (isCorrectFormat(str, a, b)) { + str = str.split(' ') + for (let x = 0; x < str.length; x++) { + if (str[x] === '-1') result += ' ' + else result += key[mod(inverseMod(a, 26) * (parseInt(str[x]) - b), 26)] + } + } + return result +} + +export { encrypt, decrypt } diff --git a/Ciphers/test/AffineCipher.test.js b/Ciphers/test/AffineCipher.test.js new file mode 100644 index 0000000000..632704bce2 --- /dev/null +++ b/Ciphers/test/AffineCipher.test.js @@ -0,0 +1,27 @@ +import { encrypt, decrypt } from '../AffineCipher' + +describe('Test Affine Cipher', () => { + it('Test - 1, Pass invalid input to encrypt function', () => { + expect(() => encrypt(null, null, null)).toThrow() + expect(() => encrypt('null', null, null)).toThrow() + expect(() => encrypt('null', 1, null)).toThrow() + expect(() => encrypt('null', null, 1)).toThrow() + }) + + it('Test - 2, Pass invalid input to decrypt function', () => { + expect(() => decrypt(null, null, null)).toThrow() + expect(() => decrypt('null', null, null)).toThrow() + expect(() => decrypt('null', 1, null)).toThrow() + expect(() => decrypt('null', null, 1)).toThrow() + }) + + it('Test - 3 Pass string value to encrypt and ecrypt function', () => { + const a = 5 + const b = 8 + expect(decrypt(encrypt('HELLO WORLD', a, b), a, b)).toBe('HELLO WORLD') + expect(decrypt(encrypt('ABC DEF', a, b), a, b)).toBe('ABC DEF') + expect(decrypt(encrypt('Brown fox jump over the fence', a, b), a, b)).toBe( + 'BROWN FOX JUMP OVER THE FENCE' + ) + }) +})