YubiRuby 0.1.0 → 1.0.0

data/CHANGELOG

@@ -1,3 +1,9 @@
+v1.0.0. broken api compatibility. TestCase. fixed some bugs in (mod)hex decoding. Exceptions. FakeYubikey generator. yubikey app for personal key management.
+
+v0.1.2. backported fixes from 1.0.0
+
+v0.1.1. fixed some bugs in hex decoding.
+
 v0.1.0. modhex installed as binary.
 
 v0.0.1. First release

data/Manifest

@@ -3,8 +3,8 @@ LICENSE
 Manifest
 README
 Rakefile
-YubiRuby.gemspec
 bin/modhex
+bin/yubikey
 ext/libyubikey/extconf.rb
 ext/libyubikey/libyubikey.c
 ext/libyubikey/ykaes.c
@@ -13,7 +13,17 @@ ext/libyubikey/ykhex.c
 ext/libyubikey/ykmodhex.c
 ext/libyubikey/ykparse.c
 ext/libyubikey/yubikey.h
+lib/exceptions.rb
+lib/fake_yubikey.rb
 lib/hex.rb
 lib/yubiruby.rb
-tests/tc_modhex.rb
-tests/ts_yubiruby.rb
+tasks/yard_doc.rake
+test/fake_yubikey_spec.rb
+test/hex_spec.rb
+test/modhex_spec.rb
+test/path_loader.rb
+test/tc_crc16.rb
+test/tc_fake_yubikey.rb
+test/tc_hex.rb
+test/tc_modhex.rb
+test/test_helper.rb

data/README

@@ -0,0 +1,44 @@
+= YubyRuby
+
+Yubico Server wrapper in Ruby
+
+
+== Installation
+
+ $ gem install YubiRuby
+
+== Some examples
+
+Checking an OTP
+
+ require 'rubygems'
+ require 'YubiRuby'
+ 
+ key = "6df89690b5f51bd9ac912c5004781e86" #use your AES key
+ y = YubiRuby::Yubikey.new(key);
+ puts y.key
+ otp = gets().strip
+ puts y.parse(otp)
+ puts "Ouput: #{y}"
+ puts "uid: #{y.uid}"
+ puts "session counter: #{y.session_counter}"
+ puts "capslock: #{y.triggered_by_capslock?}"
+ puts "timestamp low/high: #{y.timestamp_low}/#{y.timestamp_high}"
+ puts "session use: #{y.session_use}"
+ puts "random: #{y.random}"
+ puts "crc: #{y.crc}"
+ puts "crc residue: #{y.crc_residue}"
+ puts "crc residue ok?: #{y.crc?} (#{y.crc_residue} == #{YubiRuby::Yubikey::CRC_OK_RESIDUE})"
+
+Generating a valid yubikey
+
+ key = YubiRuby::FakeYubikey.new
+ key.set_random_key
+ key.set_random_uid
+ key.public_id = "vvefeeccebek"
+ puts "Yubikey generated"
+ puts "public id:\t#{key.public_id}"
+ puts "key:\t\t#{key.key}"
+ puts "uid:\t\t#{key.uid}"
+ puts "And now 3 otp"
+ 3.times { puts key.otp }

data/Rakefile

@@ -3,6 +3,8 @@ require 'rubygems'
 require 'rake'
 require 'echoe'
 
+ENV['SPEC_OPTS'] ||= '-c'
+
 Echoe.new('YubiRuby') do |p|
   p.description    = <<-EOF
 Yubikey integration - 
@@ -40,8 +42,12 @@ EOF
   p.author         = "Alessio Caiazza"
   p.email          = "nolith@abisso.org"
   p.platform       = Gem::Platform::RUBY
-  p.ignore_pattern = ["tmp/**/*", "script/*"]
-  p.development_dependencies = []
+  p.ignore_pattern = ["tmp/**/*", "script/*", "meta/**/*"]
+  p.development_dependencies = ['yard', 'rspec']
+  p.runtime_dependencies << ['ruby-aes-normal', '~> 1.0']
+  p.runtime_dependencies << ['bit-struct'] #, '~> 0.13.6']
+  p.spec_pattern = ['test/**/*_spec.rb']
+  p.has_rdoc = true
 end
 
 Dir["#{File.dirname(__FILE__)}/tasks/*.rake"].sort.each { |ext| load ext }

data/YubiRuby.gemspec

@@ -2,12 +2,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{YubiRuby}
-  s.version = "0.1.0"
+  s.version = "1.0.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
   s.authors = ["Alessio Caiazza"]
-  s.date = %q{2010-02-03}
-  s.default_executable = %q{modhex}
+  s.date = %q{2010-03-21}
   s.description = %q{Yubikey integration - 
 
 Includes Prototypes for low-level Yubikey OTP functions 
@@ -40,24 +39,37 @@ THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 }
   s.email = %q{nolith@abisso.org}
-  s.executables = ["modhex"]
+  s.executables = ["modhex", "yubikey"]
   s.extensions = ["ext/libyubikey/extconf.rb"]
-  s.extra_rdoc_files = ["CHANGELOG", "LICENSE", "README", "bin/modhex", "ext/libyubikey/extconf.rb", "ext/libyubikey/libyubikey.c", "ext/libyubikey/ykaes.c", "ext/libyubikey/ykcrc.c", "ext/libyubikey/ykhex.c", "ext/libyubikey/ykmodhex.c", "ext/libyubikey/ykparse.c", "ext/libyubikey/yubikey.h", "lib/hex.rb", "lib/yubiruby.rb"]
-  s.files = ["CHANGELOG", "LICENSE", "Manifest", "README", "Rakefile", "YubiRuby.gemspec", "bin/modhex", "ext/libyubikey/extconf.rb", "ext/libyubikey/libyubikey.c", "ext/libyubikey/ykaes.c", "ext/libyubikey/ykcrc.c", "ext/libyubikey/ykhex.c", "ext/libyubikey/ykmodhex.c", "ext/libyubikey/ykparse.c", "ext/libyubikey/yubikey.h", "lib/hex.rb", "lib/yubiruby.rb", "tests/tc_modhex.rb", "tests/ts_yubiruby.rb"]
+  s.extra_rdoc_files = ["CHANGELOG", "LICENSE", "README", "bin/modhex", "bin/yubikey", "ext/libyubikey/extconf.rb", "ext/libyubikey/libyubikey.c", "ext/libyubikey/ykaes.c", "ext/libyubikey/ykcrc.c", "ext/libyubikey/ykhex.c", "ext/libyubikey/ykmodhex.c", "ext/libyubikey/ykparse.c", "ext/libyubikey/yubikey.h", "lib/exceptions.rb", "lib/fake_yubikey.rb", "lib/hex.rb", "lib/yubiruby.rb", "tasks/yard_doc.rake"]
+  s.files = ["CHANGELOG", "LICENSE", "Manifest", "README", "Rakefile", "bin/modhex", "bin/yubikey", "ext/libyubikey/extconf.rb", "ext/libyubikey/libyubikey.c", "ext/libyubikey/ykaes.c", "ext/libyubikey/ykcrc.c", "ext/libyubikey/ykhex.c", "ext/libyubikey/ykmodhex.c", "ext/libyubikey/ykparse.c", "ext/libyubikey/yubikey.h", "lib/exceptions.rb", "lib/fake_yubikey.rb", "lib/hex.rb", "lib/yubiruby.rb", "tasks/yard_doc.rake", "test/fake_yubikey_spec.rb", "test/hex_spec.rb", "test/modhex_spec.rb", "test/path_loader.rb", "test/tc_crc16.rb", "test/tc_fake_yubikey.rb", "test/tc_hex.rb", "test/tc_modhex.rb", "test/test_helper.rb", "YubiRuby.gemspec"]
   s.homepage = %q{http://bitbucket.org/nolith/yubiruby}
   s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "YubiRuby", "--main", "README"]
   s.require_paths = ["lib", "ext"]
   s.rubyforge_project = %q{yubiruby}
-  s.rubygems_version = %q{1.3.5}
+  s.rubygems_version = %q{1.3.6}
   s.summary = %q{Yubikey integration -   Includes Prototypes for low-level Yubikey OTP functions  witten by Simon Josefsson <simon@josefsson.org>. Copyright (c) 2006, 2007, 2008, 2009 Yubico AB All rights reserved.  Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:  * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.  * 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 OWNER 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.}
+  s.test_files = ["test/test_helper.rb"]
 
   if s.respond_to? :specification_version then
     current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
     s.specification_version = 3
 
     if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<ruby-aes-normal>, ["~> 1.0"])
+      s.add_runtime_dependency(%q<bit-struct>, [">= 0"])
+      s.add_development_dependency(%q<yard>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, [">= 0"])
     else
+      s.add_dependency(%q<ruby-aes-normal>, ["~> 1.0"])
+      s.add_dependency(%q<bit-struct>, [">= 0"])
+      s.add_dependency(%q<yard>, [">= 0"])
+      s.add_dependency(%q<rspec>, [">= 0"])
     end
   else
+    s.add_dependency(%q<ruby-aes-normal>, ["~> 1.0"])
+    s.add_dependency(%q<bit-struct>, [">= 0"])
+    s.add_dependency(%q<yard>, [">= 0"])
+    s.add_dependency(%q<rspec>, [">= 0"])
   end
 end

data/bin/yubikey

@@ -0,0 +1,39 @@
+#!/usr/bin/env ruby -wKU
+
+require 'yubiruby'
+
+
+if ARGV.length > 1
+  msg=<<-EOF
+Usage: #{$PROGRAM_NAME} [-v]
+  
+Produce an OTP data
+
+-v for verbose output
+EOF
+  abort msg
+end
+
+verbose = ARGV.include? "-h"
+
+key = nil
+begin
+  File.open(File.expand_path("~/.yubiruby")) do |f|  
+    key = Marshal.load(f)  
+  end
+rescue Errno::ENOENT
+  puts "Key not found. Generating a new one."
+  key = YubiRuby::FakeYubikey.new
+  key.set_random_key
+  key.set_random_uid
+  tmp = " "*3
+  0.upto(2) { |i| tmp[i] = rand(2**8)}
+  key.public_id = "vv" + YubiRuby::HEX.encode(tmp).modhex_encode[2..-1]
+  puts "Yubikey generated"
+  puts "public id:\t#{key.public_id}"
+  puts "key:\t\t#{key.key}"
+  puts "uid:\t\t#{key.uid}"
+end
+
+puts key.otp
+File.open(File.expand_path("~/.yubiruby"), 'w+') { |f| Marshal.dump(key, f) }

data/ext/libyubikey/libyubikey.c

@@ -33,6 +33,8 @@
 #include <stdlib.h>
 #include "yubikey.h"
 
+//#undef NDEBUG
+
 #define MODHEX_SING_ENCODE	"encode"
 #define MODHEX_SING_DECODE	"decode"
 #define MODHEX_ENCODE	"modhex_encode"
@@ -42,6 +44,7 @@
 static VALUE mYubiRuby;
 static VALUE mModHex;
 static VALUE cYubikey;
+static VALUE mCRC16;
 
 
 /*  ModHex     */
@@ -92,13 +95,15 @@ static VALUE modhex_sing_decode(VALUE self, VALUE obj)
 {
 	VALUE str = StringValue(obj);
 	int src_size = RSTRING(str)->len;
+	if (rb_funcall(str, rb_intern(MODHEX_VALID), 0) == Qfalse)
+		rb_raise(rb_eval_string("YubiRuby::NoModHexEncodedError"), "");
 	char *dst = (char*) malloc(((src_size/2)+1)*sizeof(char));
 	/* ModHex decode input string SRC of length DSTSIZE/2 into output
 	   string DST.  The output string DST is always DSTSIZE/2 large plus
 	   the terminating zero.  */
 	yubikey_modhex_decode(dst, RSTRING(str)->ptr, src_size);
 	
-	return rb_str_new2(dst);
+	return rb_str_new(dst, src_size/2);
 }
 
 /* 
@@ -371,9 +376,13 @@ static VALUE yubikey_to_str(VALUE self)
 	
 	VALUE ary = rb_ary_new2(YUBIKEY_BLOCK_SIZE);
 	for (i = 0; i < YUBIKEY_BLOCK_SIZE; i++) {
-		VALUE num = INT2FIX(((uint8_t*)data->token)[i]);
+		uint8_t c_num = ((uint8_t*)data->token)[i];
+		VALUE num = INT2FIX(c_num);
 		//invoke .to_s 16 on each number
-		rb_ary_store(ary, i, rb_funcall(num, rb_intern("to_s"), 1, INT2FIX(16) ));
+		VALUE hexed = rb_funcall(num,rb_intern("to_s"), 1, INT2FIX(16));
+		if(c_num < 16) //if num < 16 prepend a '0'
+			hexed = rb_funcall(hexed,rb_intern("insert"), 2, INT2FIX(0),rb_str_new2("0"));			
+		rb_ary_store(ary, i, hexed );
 	}
 
 	return rb_funcall(ary, rb_intern("join"), 0); 
@@ -393,9 +402,9 @@ static VALUE yubikey_get_uid(VALUE self)
 }
 
 /* call-seq:
- * 	yubikey.counter -> Fixnum
+ * 	yubikey.session_counter -> Fixnum
  *
- * Gets the decoded token counter field.
+ * Gets the decoded token session counter field.
  */
 static VALUE yubikey_get_counter(VALUE self) 
 {
@@ -480,6 +489,13 @@ static VALUE yubikey_check_crc(VALUE self)
 }
 /*   Yubikey END */
 
+/* CRC 16 */
+static VALUE crc16_sing_calc(VALUE self, VALUE data) 
+{
+	VALUE str = StringValue(data);
+	return INT2NUM(yubikey_crc16(RSTRING(str)->ptr, RSTRING(str)->len));
+}
+
 /* Implementation of Yubikey OTP functions.
  * This Module may be used to interact with a Yubikey
  */
@@ -523,13 +539,16 @@ void Init_libyubikey() {
 	rb_define_method(cYubikey, "to_str", yubikey_to_str, 0);
 	rb_define_alias(cYubikey, "to_s", "to_str");
 	rb_define_method(cYubikey, "uid", yubikey_get_uid, 0);
-	rb_define_method(cYubikey, "counter", yubikey_get_counter, 0);
+	rb_define_method(cYubikey, "session_counter", yubikey_get_counter, 0);
 	rb_define_method(cYubikey, "triggered_by_capslock?", yubikey_get_triggered_by_capslock, 0);
 	rb_define_method(cYubikey, "timestamp_low", yubikey_get_tsl, 0);
 	rb_define_method(cYubikey, "timestamp_high", yubikey_get_tsh, 0);
-	rb_define_method(cYubikey, "session", yubikey_get_session, 0);
+	rb_define_method(cYubikey, "session_use", yubikey_get_session, 0);
 	rb_define_method(cYubikey, "random", yubikey_get_random, 0);
 	rb_define_method(cYubikey, "crc", yubikey_get_crc, 0);
 	rb_define_method(cYubikey, "crc_residue", yubikey_calc_crc, 0);
 	rb_define_method(cYubikey, "crc?", yubikey_check_crc, 0);
+	
+	mCRC16 = rb_define_module_under(mYubiRuby, "CRC16");
+	rb_define_singleton_method(mCRC16, "calculate", crc16_sing_calc, 1);
 }

data/lib/exceptions.rb

@@ -0,0 +1,52 @@
+# Author::    Alessio Caiazza  (mailto:nolith@abisso.org)
+# Copyright:: Copyright (c) 2010 Alessio Caiazza
+# License::   New BSD License - http://www.opensource.org/licenses/bsd-license.php
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+# * Redistributions of source code must retain the above copyright
+#   notice, this list of conditions and the following disclaimer.
+#
+# * 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
+# OWNER 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.
+
+
+module YubiRuby
+  class NoHexEncodedError < Exception
+    def initialize
+      super('The string provided isn\'t hex encoded')
+    end
+  end
+  
+  class NoModHexEncodedError < Exception
+    def inizialize(msg=nil)
+      if msg.nil? || msg.empty?
+        msg = 'The string provided isn\'t modhex encoded'
+      end
+      super(msg)
+    end
+  end
+  
+  class YubikeySessionCounterOverflow < Exception
+    def initialize
+      super('Session Counter Overflow')
+    end
+  end
+end

data/lib/fake_yubikey.rb

@@ -0,0 +1,209 @@
+# Author::    Alessio Caiazza  (mailto:nolith@abisso.org)
+# Copyright:: Copyright (c) 2010 Alessio Caiazza
+# License::   New BSD License - http://www.opensource.org/licenses/bsd-license.php
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+# * Redistributions of source code must retain the above copyright
+#   notice, this list of conditions and the following disclaimer.
+#
+# * 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
+# OWNER 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.
+
+require 'rubygems'
+require 'ruby-aes'
+require 'bit-struct'
+
+# 1-complement is uses to computer crc field
+#
+#
+class Numeric
+    def ones_complement(bits)
+        self ^ ((1 << bits) - 1)
+    end
+end
+
+module YubiRuby
+  #
+  # This class simulate a yubikey allowing you to make some tests
+  #
+  class FakeYubikey 
+    # Maximum allowed length for a modhex encoded public id
+    PUBLIC_ID_MAX_LEN = 32
+    
+    # 
+    # Yubikey internal data 
+    #
+    class Data < BitStruct
+      YUBIKEY_BLOCK_SIZE = 16
+      YUBIKEY_KEY_SIZE = 16
+      YUBIKEY_UID_SIZE = 6
+      SESSION_COUNTER_OVERFLOW = 0x7FFF
+    
+      hex_octets  :uid,     YUBIKEY_UID_SIZE*8,     "User ID"
+      unsigned    :session_counter,    16, {:endian=>:little, :display_name=>"Session Counter"}  
+      unsigned    :timestamp_low,   16, {:endian=>:little, :display_name=>"Timestamp low"}
+      unsigned    :timestamp_high,  8, "Timestamp high"
+      unsigned    :session_use,  8, "Session Use"
+      unsigned    :random,   16,  {:endian=>:little, :display_name=>"random"}
+      unsigned    :crc,  16, {:endian=>:little, :display_name=>"crc"}
+      
+    end
+
+    attr_accessor :key
+    #attr_reader :public_id
+    
+    def initialize
+      @public_id = ""
+      @data = YubiRuby::FakeYubikey::Data.new
+      @data.session_counter = 0
+      init
+    end
+    
+    # 
+    # Sets the public id of the key. 
+    # 
+    # public_id must be a modhex valid string no longer than
+    # 32 char, if longer it will be truncated.
+    #
+    # @param [String, #read] value the new public_id 
+    # @return [String] the new public_id
+    def public_id=(value)
+      raise NoModHexEncodedError unless value.modhex?
+      
+      value = value[0, PUBLIC_ID_MAX_LEN] if value.length > PUBLIC_ID_MAX_LEN
+      raise NoModHexEncodedError unless value.modhex?
+      
+      @public_id = value
+    end
+    
+    # Returns the value of attribute public_id.
+    def public_id
+      @public_id
+    end
+    
+    # Simaluates the power on sequence of a YubuKey.
+    # 
+    # This will increase session_counter
+    def init
+      @data.session_counter += 1
+      if (@data.session_counter == Data::SESSION_COUNTER_OVERFLOW+1)
+        raise YubikeySessionCounterOverflow.new 
+      end
+      @data.session_use = 0
+      @data.timestamp_low = rand(2**16)
+      @data.timestamp_high = rand(2**8)
+    end
+    
+    %w{session_use timestamp_low timestamp_high session_counter random crc}.each do |name|
+      define_method name do 
+        @data.send name
+      end
+    end
+    
+    # Generates a random key attribute
+    def set_random_key
+      @key = produce_random_bytes_in_hex Data::YUBIKEY_KEY_SIZE
+    end
+    
+    # Generates a random uid attribute
+    def set_random_uid
+      self.uid= produce_random_bytes_in_hex(Data::YUBIKEY_UID_SIZE)
+    end
+    
+    # Sets the attribute uid
+    def uid=(value)
+      if YubiRuby::HEX.decode(value).size == Data::YUBIKEY_UID_SIZE
+        @data.uid = "#{value[0,2]}:#{value[2,2]}:#{value[4,2]}:#{value[6,2]}:#{value[8,2]}:#{value[10,2]}"
+      else
+        raise NoHexEncodedError.new
+      end
+    end
+    
+    # Returns the value of attribute uid.
+    def uid
+      @data.uid.gsub(':','')
+    end
+    
+    # Simulates the pression of the yubikey's button
+    #
+    # session_use will increase and if it overflows also
+    # session_counter will be increased
+    def otp
+      @data.random = rand(2**16)
+      @data.session_use += 1
+      @data.session_counter +=1 if @data.session_use == 0
+      generate_otp
+    end
+    
+    def to_s
+      @data.to_s
+    end
+    
+    def self.generate_specific_otp_and_fake_yubikey(key, uid, session_counter, 
+        timestamp_low, timestamp_high, session_use, random)
+      y_key = FakeYubikey.new
+      y_key.key = key
+      y_key.uid = uid
+      y_key.data.session_use = session_use
+      y_key.data.timestamp_low = timestamp_low
+      y_key.data.timestamp_high = timestamp_high
+      y_key.data.session_counter = session_counter
+      y_key.data.random = random
+      [y_key.generate_otp, y_key]
+    end
+    
+    def data
+      @data
+    end
+    
+    # Generates an OTP without altering any internal data.
+    def generate_otp
+      @data.crc = YubiRuby::CRC16.calculate(@data.to_s[0...-2]).ones_complement 16
+      data = Aes.encrypt_block(128, 'ECB', YubiRuby::HEX.decode(@key), nil, 
+        @data.to_s).modhex_encode
+      "#{@public_id}#{data}"
+    end
+    
+    #serialization support 
+    def marshal_dump
+        {'key' => key, 'uid' => uid, 'session_counter' => @data.session_counter,
+          'public_id' => public_id }
+    end
+    
+    #serialization support 
+    def marshal_load(data)
+      @data = YubiRuby::FakeYubikey::Data.new
+      self.key = data['key']
+      self.uid = data['uid']
+      self.data.session_counter = data['session_counter']
+      self.public_id = data['public_id']
+      init
+    end
+    
+    
+    private
+    def produce_random_bytes_in_hex(size)
+      tmp = " "*size
+      0.upto(size-1) { |i| tmp[i] = rand(2**8)}
+      YubiRuby::HEX.encode(tmp)
+    end
+  end
+end