enquo-core 0.3.0-aarch64-linux → 0.4.0-aarch64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dbfd5f231408af0cfa36b7b2eaabe134356d9d2d4e429eedab0f67401bed6c0a
-  data.tar.gz: cc280385cc90257998b4b93a24e74f91c19f3e4cacae95aafe266609d582e565
+  metadata.gz: 14300d7e8d5a6a3122666cd3c7a83686d294373c5537defb60d8531d722cd4eb
+  data.tar.gz: db807a0c99d1aa1670251261e6918c55cac14b17831f1317607db91d18a19f98
 SHA512:
-  metadata.gz: f062a7c0e60f5df87c7e8c0f417064fa28d9a726b8292840a902161d5eeb555e6dac221d155fe46f4823b34d636350e8bb608b2a3cd379190b0937bb5536f427
-  data.tar.gz: a2ab46fb08130389c7d26c54d220a977f6964585eddfa6db3221b4b7d6a114cce873426423761adf31cf8db61a081093a611a14b6d98ea7ccbf0dcaee724f4eb
+  metadata.gz: 50ae282576040722cc4f2fdcb0940ff8728e6edf61ac8c43e02a9d0aaec355245ab7d0d71d137cc9628f698bc28388c4a693aa910f03522a80ecb960ea4053b8
+  data.tar.gz: 1adcd8c8bc6054d5e8733ac3bd483d675938c73037d91a77af79c7585833be6817b4870566e68b48ee58794db59734eda5558099feb7c8f3f4f5b632fac6705d

data/ext/enquo/src/lib.rs

@@ -1,8 +1,8 @@
 #[macro_use]
 extern crate rutie;
 
-use enquo_core::{Field, Root, I64};
-use rutie::{Class, Integer, Module, Object, RString, Symbol, VerifiedObject, VM};
+use enquo_core::{Date, Field, Root, I64};
+use rutie::{AnyObject, Class, Integer, Module, Object, RString, Symbol, VerifiedObject, VM};
 
 class!(EnquoRoot);
 class!(EnquoRootKeyStatic);
@@ -57,7 +57,8 @@ methods!(
     EnquoRootKeyStatic,
     _rbself,
     fn enquo_root_key_static_new(root_key: RString) -> EnquoRootKeyStatic {
-        let k = root_key.unwrap().to_vec_u8_unchecked();
+        #[allow(clippy::redundant_clone)]
+        let k = root_key.unwrap().to_vec_u8_unchecked().to_owned();
         let klass = Module::from_existing("Enquo")
             .get_nested_class("RootKey")
             .get_nested_class("Static");
@@ -78,6 +79,8 @@ impl VerifiedObject for EnquoRootKeyStatic {
     }
 }
 
+// rustfmt fucks this so it doesn't compile
+#[rustfmt::skip]
 methods!(
     EnquoField,
     rbself,
@@ -114,6 +117,58 @@ methods!(
             "Failed to decrypt i64 value",
         );
         Integer::from(value)
+    },
+    fn enquo_field_encrypt_date(
+        y_r: Integer,
+        m_r: Integer,
+        d_r: Integer,
+        context: RString,
+        mode: Symbol
+    ) -> RString {
+        let y = y_r.unwrap().to_i32() as i16;
+        let m = m_r.unwrap().to_i32() as u8;
+        let d = d_r.unwrap().to_i32() as u8;
+        let field = rbself.get_data(&*FIELD_WRAPPER);
+        let r_mode = mode.unwrap();
+        let s_mode = r_mode.to_str();
+
+        let mut res = maybe_raise(
+            if s_mode == "unsafe" {
+                Date::new_with_unsafe_parts(
+                    (y, m, d),
+                    &context.unwrap().to_vec_u8_unchecked(),
+                    field,
+                )
+            } else {
+                Date::new((y, m, d), &context.unwrap().to_vec_u8_unchecked(), field)
+            },
+            "Failed to create encrypted date",
+        );
+        if s_mode == "no_query" {
+            res.drop_ore_ciphertexts();
+        }
+
+        RString::new_utf8(&serde_json::to_string(&res).unwrap())
+    },
+    fn enquo_field_decrypt_date(ciphertext: RString, context: RString) -> AnyObject {
+        let ct_r = ciphertext.unwrap();
+        let ct = ct_r.to_str_unchecked();
+        let e_value: Date =
+            maybe_raise(serde_json::from_str(ct), "Failed to deserialize ciphertext");
+
+        let field = rbself.get_data(&*FIELD_WRAPPER);
+
+        let (y, m, d) = maybe_raise(
+            e_value.decrypt(&context.unwrap().to_vec_u8_unchecked(), field),
+            "Failed to decrypt date value",
+        );
+        let klass = Class::from_existing("Date");
+        let args: [AnyObject; 3] = [
+            Integer::from(y as i32).into(),
+            Integer::from(m as i32).into(),
+            Integer::from(d as i32).into(),
+        ];
+        klass.protect_send("new", &args).unwrap()
     }
 );
 
@@ -135,6 +190,8 @@ pub extern "C" fn Init_enquo() {
             .define(|fieldklass| {
                 fieldklass.def_private("_encrypt_i64", enquo_field_encrypt_i64);
                 fieldklass.def_private("_decrypt_i64", enquo_field_decrypt_i64);
+                fieldklass.def_private("_encrypt_date", enquo_field_encrypt_date);
+                fieldklass.def_private("_decrypt_date", enquo_field_decrypt_date);
             });
         topmod.define_nested_module("RootKey").define(|rkmod| {
             rkmod

data/lib/enquo/field.rb

@@ -1,3 +1,5 @@
+require "date"
+
 module Enquo
 	class Field
 		def self.new(*_)
@@ -35,5 +37,37 @@ module Enquo
 
 			_decrypt_i64(data, ctx)
 		end
+
+		def encrypt_date(d, ctx, safety: true, no_query: false)
+			unless d.is_a?(Date)
+				raise ArgumentError, "Enquo::Field#encrypt_date can only encrypt Dates"
+			end
+
+			unless d.year >= -2 ** 15 && d.year < 2 ** 15 - 1
+				raise RangeError, "Enquo::Field#encrypt_date can only encrypt dates where the year is between -32,768 and 32,767 (got #{d.year})"
+			end
+
+			unless ctx.is_a?(String)
+				raise ArgumentError, "Encryption context must be a string (got a #{ctx.class})"
+			end
+
+			_encrypt_date(d.year, d.month, d.day, ctx, no_query ? :no_query : safety == :unsafe ? :unsafe : :default)
+		end
+
+		def decrypt_date(data, ctx)
+			unless data.is_a?(String)
+				raise ArgumentError, "Enquo::Field#decrypt_date can only decrypt from a string (got #{data.class})"
+			end
+
+			unless data.encoding == Encoding::UTF_8 && data.valid_encoding?
+				raise ArgumentError, "Enquo::Field#decrypt_date can only decrypt validly-encoded UTF-8 strings (got #{data.encoding})"
+			end
+
+			unless ctx.is_a?(String)
+				raise ArgumentError, "Encryption context must be a string (got a #{ctx.class})"
+			end
+
+			_decrypt_date(data, ctx)
+		end
 	end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enquo-core
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: aarch64-linux
 authors:
 - Matt Palmer
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-10-03 00:00:00.000000000 Z
+date: 2022-10-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler