LibCrypto: Add UnsignedBigInteger and implement addition

UnsignedBigInteger stores an unsigned ainteger of arbitrary length.

A big integer is represented  as a vector of word. Each
word is an unsigned int.

3 files changed, 152 insertions, 1 deletions

diff --git a/Libraries/LibCrypto/BigInt/UnsignedBigInteger.cpp b/Libraries/LibCrypto/BigInt/UnsignedBigInteger.cpp
new file mode 100644
index 0000000000..d22ce57a19
--- /dev/null
+++ b/Libraries/LibCrypto/BigInt/UnsignedBigInteger.cpp
@@ -0,0 +1,89 @@
+/*
+ * Copyright (c) 2020, Itamar S. <itamar8910@gmail.com>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "UnsignedBigInteger.h"
+
+namespace Crypto {
+
+UnsignedBigInteger UnsignedBigInteger::add(const UnsignedBigInteger& other)
+{
+    const UnsignedBigInteger* const longer = (length() > other.length()) ? this : &other;
+    const UnsignedBigInteger* const shorter = (longer == &other) ? this : &other;
+    UnsignedBigInteger result;
+
+    u8 carry = 0;
+    for (size_t i = 0; i < shorter->length(); ++i) {
+        u32 word_addition_result = shorter->m_words[i] + longer->m_words[i];
+        u8 carry_out = 0;
+        // if there was a carry, the result will be smaller than any of the operands
+        if (word_addition_result < shorter->m_words[i]) {
+            carry_out = 1;
+        }
+        if (carry) {
+            word_addition_result++;
+        }
+        carry = carry_out;
+        result.m_words.append(word_addition_result);
+    }
+
+    for (size_t i = shorter->length(); i < longer->length(); ++i) {
+        u32 word_addition_result = longer->m_words[i] + carry;
+
+        carry = 0;
+        if (word_addition_result < longer->m_words[i]) {
+            carry = 1;
+        }
+        result.m_words.append(word_addition_result);
+    }
+    if (carry) {
+        result.m_words.append(carry);
+    }
+    return result;
+}
+
+bool UnsignedBigInteger::operator==(const UnsignedBigInteger& other) const
+{
+    if (trimmed_length() != other.trimmed_length()) {
+        return false;
+    }
+    for (size_t i = 0; i < trimmed_length(); ++i) {
+        if (m_words[i] != other.words()[i])
+            return false;
+    }
+    return true;
+}
+
+size_t UnsignedBigInteger::trimmed_length() const
+{
+    size_t num_leading_zeroes = 0;
+    for (int i = length() - 1; i >= 0; --i, ++num_leading_zeroes) {
+        if (m_words[i] != 0)
+            break;
+    }
+    return length() - num_leading_zeroes;
+}
+
+}
diff --git a/Libraries/LibCrypto/BigInt/UnsignedBigInteger.h b/Libraries/LibCrypto/BigInt/UnsignedBigInteger.h
new file mode 100644
index 0000000000..cf3b280c8c
--- /dev/null
+++ b/Libraries/LibCrypto/BigInt/UnsignedBigInteger.h
@@ -0,0 +1,61 @@
+/*
+ * Copyright (c) 2020, Itamar S. <itamar8910@gmail.com>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#pragma once
+#include <AK/LogStream.h>
+#include <AK/Types.h>
+#include <AK/Vector.h>
+
+namespace Crypto {
+class UnsignedBigInteger {
+public:
+    UnsignedBigInteger(u32 x) { m_words.append(x); }
+    UnsignedBigInteger() {}
+
+    const AK::Vector<u32>& words() const { return m_words; }
+
+    UnsignedBigInteger add(const UnsignedBigInteger& other);
+
+    size_t length() const { return m_words.size(); }
+
+    // The "trimmed length" is the number of words after trimming leading zeroed words
+    size_t trimmed_length() const;
+
+    bool operator==(const UnsignedBigInteger& other) const;
+
+private:
+    AK::Vector<u32> m_words;
+};
+
+}
+
+inline const LogStream& operator<<(const LogStream& stream, const Crypto::UnsignedBigInteger value)
+{
+    for (int i = value.length() - 1; i >= 0; --i) {
+        stream << value.words()[i] << "|";
+    }
+    return stream;
+}
diff --git a/Libraries/LibCrypto/Makefile b/Libraries/LibCrypto/Makefile
index 80f75f5130..a9c249dde9 100644
--- a/Libraries/LibCrypto/Makefile
+++ b/Libraries/LibCrypto/Makefile
@@ -1,7 +1,8 @@
 LIBCRYPTO_OBJS = \
 	   	Cipher/AES.o \
 		Hash/MD5.o \
-		Hash/SHA2.o
+		Hash/SHA2.o \
+		BigInt/UnsignedBigInteger.o
 
 OBJS = $(LIBCRYPTO_OBJS)