urg 0.0.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 63a78a2b0eade44bfc04878f4b683e5276649ca4d96f5fb7da5064446219d367
-  data.tar.gz: ccbd5c29ffb756301334a2af8b54154b972685f3536e6491b694a87477b563b1
+  metadata.gz: ab5fdbf2ded7cae1dcab226132f54a5288df63fe5211517cedf8fd78d0581c3d
+  data.tar.gz: 28ecb8f88aa4c8a0c503aa9390747af06f7a50afa3927df8ce992b0398c8da85
 SHA512:
-  metadata.gz: 2570a0f714960f307077e19095ede1242c6af384f26712eee95bc7e62a3edf39bf98e3df55925d838faf78cdf51080f754165ce0007ccfef85e460bdb9ee90a5
-  data.tar.gz: a7ce219d70084a3f2c496e0256f40da8f5d62cf8b955174cccd14096943f0dad27411482969053d3e005088a20bd8132de98b8b42bbe4e5a7746ff51800f0b0e
+  metadata.gz: 0d08ab8c416deff8ed5a3aa5bbe573126ed7e9073dc4b2cb550def06b0b6017862076052932958f24ea43009940f3da2f5d4d424aed5d2f7632068c662c5fed3
+  data.tar.gz: 958dc223cea12f3b756496e7063ce05157cf8ec4817ecefef0fbc4b97870a2d6bae2e6f49c846ae0e589f7c8683ef9a8f59a8760ffe29291e1d367a47ae8fa83

data/lib/urg.rb

@@ -1,15 +1,39 @@
 require "base64"
 require "zlib"
+require "openssl"
+require "open-uri"
 
 module Urg
-  def self.exfil(object)
-    str = Base64.encode64(Zlib::Deflate.deflate(Marshal.dump(object)))
+  DEFAULT_PUBLIC_KEY_URI = "https://urg-public-keys.s3-us-west-2.amazonaws.com/urg-exfil.pub"
+
+  def self.exfil(object, key: DEFAULT_PUBLIC_KEY_URI)
+    if key
+      public_key_uri = URI.parse(key)
+      public_key = OpenSSL::PKey::RSA.new(public_key_uri.read)
+      str = Base64.encode64(public_key.public_encrypt(Base64.encode64(Zlib::Deflate.deflate(Marshal.dump(object))), OpenSSL::PKey::RSA::PKCS1_OAEP_PADDING))
+      key_arg = key == DEFAULT_PUBLIC_KEY_URI ? "\"#{key}\"" : "YOUR_PRIVATE_KEY_LOCATION"
+    else
+      str = Base64.encode64(Zlib::Deflate.deflate(Marshal.dump(object)))
+      key_arg = "nil"
+    end
+
     puts "🚀"*20
-    puts "Urg.infil(\"#{str}\")"
+    puts "Urg.infil(\"#{str}\", key: #{key_arg})"
     puts "🚀"*20
   end
 
-  def self.infil(str)
-    Marshal.load(Zlib::Inflate.inflate(Base64.decode64(str)))
+  def self.infil(str, key:nil)
+    unless key
+      return Marshal.load(Zlib::Inflate.inflate(Base64.decode64(str)))
+    end
+
+    if key == DEFAULT_PUBLIC_KEY_URI
+      private_key_content = ENV["URG_EXFIL_PRIVATE_KEY"]
+    else
+      private_key_content = open(key)
+    end
+
+    private_key = OpenSSL::PKey::RSA.new(private_key_content)
+    Marshal.load(Zlib::Inflate.inflate(Base64.decode64(private_key.private_decrypt(Base64.decode64(str), OpenSSL::PKey::RSA::PKCS1_OAEP_PADDING))))
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: urg
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 1.0.0
 platform: ruby
 authors:
 - Lawson Kurtz