From 08a31523b26f0a5020ef4d1a1ded66b45d980088 Mon Sep 17 00:00:00 2001
From: bwhitn <brian.m.whitney@outlook.com>
Date: Mon, 18 Dec 2017 05:04:11 -0800
Subject: [PATCH] changed the function comment

---
 src/core/config/Categories.js      |  1 +
 src/core/config/OperationConfig.js |  7 +++++++
 src/core/config/modules/Default.js |  1 +
 src/core/operations/Entropy.js     | 22 ++++++++++++++++++++++
 4 files changed, 31 insertions(+)

diff --git a/src/core/config/Categories.js b/src/core/config/Categories.js
index 3bc672b7..7a66582e 100755
--- a/src/core/config/Categories.js
+++ b/src/core/config/Categories.js
@@ -302,6 +302,7 @@ const Categories = [
         ops: [
             "Entropy",
             "Frequency distribution",
+            "Chi Square",
             "Detect File Type",
             "Scan for Embedded Files",
             "Disassemble x86",
diff --git a/src/core/config/OperationConfig.js b/src/core/config/OperationConfig.js
index 43072558..b914e10d 100755
--- a/src/core/config/OperationConfig.js
+++ b/src/core/config/OperationConfig.js
@@ -3186,6 +3186,13 @@ const OperationConfig = {
             }
         ]
     },
+    "Chi Square": {
+        module: "Default",
+        description: "Calculates the Chi Square distribution of values.",
+        inputType: "byteArray",
+        outputType: "",
+        args: []
+    }
     "Numberwang": {
         module: "Default",
         description: "Based on the popular gameshow by Mitchell and Webb.",
diff --git a/src/core/config/modules/Default.js b/src/core/config/modules/Default.js
index dec015a5..f2deb543 100644
--- a/src/core/config/modules/Default.js
+++ b/src/core/config/modules/Default.js
@@ -142,6 +142,7 @@ OpModules.Default = {
     "Microsoft Script Decoder": MS.runDecodeScript,
     "Entropy":              Entropy.runEntropy,
     "Frequency distribution": Entropy.runFreqDistrib,
+    "Chi Square":           Entropy.calcChiSq,
     "Detect File Type":     FileType.runDetect,
     "Scan for Embedded Files": FileType.runScanForEmbeddedFiles,
     "Generate UUID":        UUID.runGenerateV4,
diff --git a/src/core/operations/Entropy.js b/src/core/operations/Entropy.js
index 3451914d..7ca87bb3 100755
--- a/src/core/operations/Entropy.js
+++ b/src/core/operations/Entropy.js
@@ -163,6 +163,28 @@ const Entropy = {
         return -entropy;
     },
 
+
+    /**
+     * Calculates the Chi Square distribution of values.
+     *
+     * @private
+     * @param {byteArray} data
+     * @param {Object[]} args
+     * @returns {number}
+     */
+    calcChiSq: function(input, args) {
+        let distArray = new Array(256).fill(0),
+            total = 0;
+        for (let i = 0; i < input.length; i++) {
+            distArray[data[i]]++;
+        }
+        for (let i = 0; i < distArray.length; i++) {
+            if (distArray[i] > 0) {
+                total += Math.pow(distArray[i] - input.length / 256, 2) / (input.length / 256);
+            }
+        }
+        return total;
+    }
 };
 
 export default Entropy;