Very experimental commit but gonna dev on my pc for a bit

encode_image.rb

@@ -7,11 +7,11 @@ class EncodeImage
   @@utils = nil
 
 
-  def initialize(p, m, uri)
+  def initialize(p, m, i)
     @@password = p
     @@message  = m
 
-    @@utils = ImageUtils.new(@@password, @@message, uri)
+    @@utils = ImageUtils.new(@@password, @@message, i)
 
     @@table = @@utils.generateLookupTable
   end
@@ -32,4 +32,46 @@ class EncodeImage
 
     @@utils.getImage.saveImage
   end
+
+
+# Need to properly sever this at some point...
+
+  def decode
+puts "Hacked in decode..."
+    s = @@utils.getImage.getSize
+
+    msg_bits = 0
+
+    @@table.reverse.each do |loc|
+      pixel = @@utils.getImage.getPixel(loc[:x], loc[:y])
+      msg_bits = (msg_bits << 6) | pixel.decodeBits
+    end
+
+#this trash needs to move somewhere...
+puts "Try to convert the message..."
+    len = msg_bits & 0xffff
+    msg_bits = msg_bits >> 16
+
+    # check the padding bits....
+    if (msg_bits & 255) != 0 then
+puts "Padding failed scrutiny! "
+puts msg_bits.to_s(2)
+      return false
+    end
+
+    msg_bits = msg_bits >> 8
+
+puts "Padding passed scrutiny..."
+    msg_chars = []
+
+    while msg_bits > 0 do
+      byte = msg_bits & 0xff
+      msg_chars.push(byte.chr)
+      msg_bits = msg_bits >> 8
+    end
+
+    msg = msg_chars.reverse.join
+
+    return msg
+  end
 end

image_utils.rb

@@ -1,116 +1,99 @@
-require "rmagick"
+require "./pixel_image.rb"
 require "digest"
 
 class ImageUtils
   @@magickImg = nil
+  @@pixelImg = nil
 
   @@password = ""
   @@message = ""
-  @@salt       = 0
-
-  @@imagickImg = nil
-
-  @@width      = 0
-  @@height     = 0
-
-  @@pixel_arr  = nil
 
   @@hash = nil
 
-  @@rng        = nil
+  def initialize(pass, msg, i)
-
-  def initialize(pass, msg, uri)
     @@password = pass
     @@message  = msg
-    @@salt     = 0
 
-    @@magickImg = Magick::Image.read(uri).first
+    if msg.length > 1024 then
-
+      puts "Maximum length exceeded.  Exiting."
-    @@width = @@magickImg.columns
+      exit
-    @@height = @@magickImg.rows
-
-    @@pixel_arr = @@magickImg.get_pixels(0, 0, @@width, @@height)
-
-    @@hash = self.generateHash
-
-    # convert hex string into an int to store for making the lookup table
-    @@rng = Random.new(@@hash.to_i(16))
-
-    puts self.getLookupTable.inspect
     end
 
-  def getPixel(x, y)
+    @@magickImg = i
-    loc = (y*@@width) + x
 
-    return @@pixel_arr[loc]
+    @@pixelImg = PixelImage.new(@@magickImg)
+
+    self.generateHash
   end
 
-  def getSalt
+  def getImage
-    for x in 0..@@width-1
+    return @@pixelImg
-      for y in 0..@@height-1
-        pix = self.getPixel(x, y)
-
-        @@salt = @@salt + (pix.red/512)
-        @@salt = @@salt + (pix.green/512)
-        @@salt = @@salt + (pix.blue/512)
-      end
-    end
-
-    h = Digest::SHA512.hexdigest(@@salt.to_s(16))
-
-    return h
   end
 
   def generateHash
-    pass_h = Digest::SHA512.hexdigest(@@password)
+    h = Digest::SHA512.hexdigest(@@password)
+    s = @@pixelImg.getSalt
 
-    return Digest::SHA512.hexdigest(pass_h + @@salt.to_s)
+    @@hash = Digest::SHA512.hexdigest(self.interlaceHash(h, s))
   end
 
-  def getBits(o)
+  def interlaceHash(h1, h2)
-    msg_bytes = o.unpack("c*")
+    puts h1
+    puts h2
 
-    msg_bits = 0
+    h1 = self.getBits(h1)
+    h2 = self.getBits(h2)
     
-    for b in msg_bytes
+    interlaced = 0
-      msg_bits = (msg_bits << 8) | b
+
+    while h1.to_i > 0 and h2.to_i > 0 do
+      interlaced = (interlaced << 1) | (h1 & 1)
+      h1 = h1 >> 1
+      interlaced = (interlaced << 1) | (h2 & 1)
+      h2 = h2 >> 1
+
+#      puts interlaced
+#      puts h1
+#      puts h2
     end
 
-    return msg_bits
+    puts interlaced.to_s.inspect
+
+    return interlaced.to_s
   end
 
-  def getLookupTable
+  def generateLookupTable
     tbl = []
 
-    # track hash classes.  if too many occur during the interval, abaondon 
+    hash_bits = @@pixelImg.getBits(@@hash)
-    # encoding.
+    msg_bits = @@pixelImg.getBits(@@message)
-    clashes = 0
-    clash_limit = 100
 
-    msg_bits = self.getBits(@@message)
+    # gross, but pad in the 24 bits that will become the msg length
+    # and padding bits...
+    msg_bits = msg_bits << 24
 
     while msg_bits > 0 do
       loc = nil
 
       loop do
-        x = @@rng.rand(0..@@width)
+        x = (hash_bits.to_i % @@pixelImg.getSize[:width])
-        y = @@rng.rand(0..@@height)
+        y = (hash_bits.to_i % @@pixelImg.getSize[:height])
 
         loc = {"x": x, "y": y}
-        if !tbl.include?(loc) then
-          clashes = 0
-        else
-          clashes = clashes + 1
-        end
 
-        if clashes < clash_limit then
+      break if !tbl.include?(loc)
-          break
+        hash_bits = hash_bits >> 1
-        else
+
-          puts "wow clashed so much sorry for breaking"
+        if hash_bits <= 0 then
+          puts
+          puts "wow too much data"
+          puts "leftover bits: " + msg_bits.to_s(36)
           exit
         end
       end
 
+      hash_bits = hash_bits >> 1
+
       tbl.push(loc)
 
       msg_bits = msg_bits >> 6

runner.rb

@@ -1,10 +1,13 @@
+require "rmagick"
+
 require "./image_utils.rb"
 require "./encode_image.rb"
 
+img = Magick::ImageList.new "./images/original/arty.png"
+
 #srand(12345)
 
 pass = "my_password"
-
 message = "This is the message that I would like to save please " +
           "and thank you also I love you? AND apostrophe's, and " +
           "also some commas somewhere hiding around here,  It's " +
@@ -21,21 +24,17 @@ message = "This is the message that I would like to save please " +
           "are going even further.  further than anyone believed " +
           "`~!@\#\{$%^&*()-_=+\][}{';:'}]} Wooow. 123456789!1234" +
           "56789! so put 15 more??????400!"
+#message = "tests"
 
 puts message.length
 puts message
 puts
 
-start = Time.new.to_f
-
 puts "Initializing EncodeImage..."
 
-iu = ImageUtils.new(pass, message, "./images/original/arty.png")
+start = Time.new.to_f
 
-puts "Took " + (Time.new.to_f - start).to_s + " seconds to encode... 🐱"
+em = EncodeImage.new(pass, message, img)
-
-=begin
-em = EncodeImage.new(pass, message, "./images/original/arty.png")
 
 puts em.encode
 puts "Took " + (Time.new.to_f - start).to_s + " seconds to encode... 🐱"
@@ -47,12 +46,13 @@ result = em.decode
 
 puts result
 
-=end
-
 puts "Took " + (Time.new.to_f - start).to_s + " seconds to decode... 🐱"
 
 if message == result then
   puts "WOW THEY MATCH! 🐱🐱🐱🐱🐱🐱🐱🐱"
 end
 
+exit  
+puts
+
 puts "Done!"