vhd 0.0.2 → 0.0.3

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    MWQ2NzlhZDJhOThjZTgyMDZkOTU5ODIzYzczNmQ4N2FlMTJkYjBhMg==
+    NTBlMjI2MjFlMGFkNWZlODViODg5ZGJmYjIxYmMwN2JjYzg5ZTRlNQ==
   data.tar.gz: !binary |-
-    MzRmMjJiODJhZTZkNzg5M2FjNWYzMDRiN2IwMWVhOTJiY2NhNjNjZg==
+    NWU1N2VmNGU0Mjc4MDc4NmMyYWNlZjliYWE1ODQ2ZmQ5ZjUyMGU1ZQ==
 !binary "U0hBNTEy":
   metadata.gz: !binary |-
-    ODhiZTdmYTg1MmM2NjQwMDU1YTRlYTUxZjBjY2U3MDIyMTU5NDFjNjkwMjM0
-    NmVlNzg2YTgwYzhiNTMzMjE5NGZmYzVjMDVmM2Y1YjA5MzhiYTFmZmFmYzhl
-    YTYxOWEyNmY3YWRjZmQ1NjQzOWNjMjJiMTA4ZDRhNzcxMjhmNmU=
+    YTMzOTdiNWZkMThjYmRhZGExODZhMmE0OTA1MTM4Yzg3YTFjMzk2OGRmYTc4
+    MWY5YTcxMDI1N2MzZGUyODYyNGU5YzUyNjg4NzhhNmZmNjgzMGU2YjhiNGYx
+    MzdkOGU4NzI1MjY3MmQ3ZTdkYzkzZjAwODkwY2I0YmE5ODA4MTY=
   data.tar.gz: !binary |-
-    ZDhlZWNmMTc0MWU3OGFmYjZhY2UzNTc2OTZmMGMyMDdjYTM3NDg0ZTk0ZDQ4
-    MWMwMDEwNjkwYjUxNzJhN2Q2YTViY2Q1MzZjMTVlMjRkMDQ5OWU4YjE4Y2Yw
-    ZjIzYTZiODdlMWFhZDFiNDA2MWM1ZmIyNTE3MWY2ZDE2NzY0ZmQ=
+    YjY0YTRkMzAzNzMyYzkyMzM4NjBkNWFhZDNlNzU1OTk3ZTNmN2Q0MGJlNmMw
+    ZTUyMTZlMzIyYjFlN2IyNGEyZjYwZGQzNzU3MjQwMzkxODgwMDI0MmUwMjA2
+    MDkwN2U4N2U0MjY5MjUxMzkwOWQ5YjRjOThhMWI0NmZkMDVjY2M=

data/Gemfile

@@ -1,3 +1,5 @@
+source 'https://rubygems.org'
+
 gemspec
 
 group :test do

data/Gemfile.lock

@@ -1,32 +1,33 @@
 PATH
   remote: .
   specs:
-    vhd-ruby (0.0.1)
-      ffi
+    vhd (0.0.2)
+      bit-struct
 
 GEM
+  remote: https://rubygems.org/
   specs:
-    awesome_print (1.1.0)
-    coderay (1.0.9)
-    diff-lcs (1.2.1)
-    ffaker (1.15.0)
-    ffi (1.9.3)
-    method_source (0.8.1)
-    pry (0.9.12)
-      coderay (~> 1.0.5)
+    awesome_print (1.2.0)
+    bit-struct (0.15.0)
+    coderay (1.1.0)
+    diff-lcs (1.2.5)
+    ffaker (1.23.0)
+    method_source (0.8.2)
+    pry (0.9.12.4)
+      coderay (~> 1.0)
       method_source (~> 0.8)
       slop (~> 3.4)
     pry-nav (0.2.3)
       pry (~> 0.9.10)
-    rspec (2.13.0)
-      rspec-core (~> 2.13.0)
-      rspec-expectations (~> 2.13.0)
-      rspec-mocks (~> 2.13.0)
-    rspec-core (2.13.1)
-    rspec-expectations (2.13.0)
+    rspec (2.14.1)
+      rspec-core (~> 2.14.0)
+      rspec-expectations (~> 2.14.0)
+      rspec-mocks (~> 2.14.0)
+    rspec-core (2.14.7)
+    rspec-expectations (2.14.4)
       diff-lcs (>= 1.1.3, < 2.0)
-    rspec-mocks (2.13.0)
-    slop (3.4.4)
+    rspec-mocks (2.14.4)
+    slop (3.4.7)
 
 PLATFORMS
   ruby
@@ -37,4 +38,4 @@ DEPENDENCIES
   pry
   pry-nav
   rspec
-  vhd-ruby!
+  vhd!

data/lib/vhd.rb

@@ -1,5 +1,8 @@
-require 'ffi'
+require 'time'
+require 'bit-struct'
+require 'securerandom'
 
 module Vhd
   autoload :Library, 'vhd/library'
+  autoload :Footer, 'vhd/footer'
 end

data/lib/vhd/footer.rb

@@ -0,0 +1,31 @@
+class Vhd::Footer < BitStruct
+  unsigned :cookie        , 8   , "Cookie"
+  unsigned :features      , 4   , "Features"
+  unsigned :ff            , 4   , "File Format"
+  unsigned :data_offset   , 8   , "Data Offset"
+  unsigned :timestamp     , 4   , "Timestamp"
+  unsigned :creator_app   , 4   , "Creator Application"
+  unsigned :creator_ver   , 4   , "Creator Version"
+  unsigned :creator_host  , 4   , "Creator Host OS"
+  unsigned :original_size , 8   , "Original Size"
+  unsigned :current_size  , 8   , "Current Size"
+  unsigned :geometry      , 4   , "Disk Geometry"
+  unsigned :type          , 4   , "Disk Type"
+  unsigned :checksum      , 4   , "Footer Checksum"
+  unsigned :uuid          , 16  , "Unique Id"
+  unsigned :saved_state   , 1   , "Saved State"
+  unsigned :reserved      , 427 , "Reserved Space"
+
+  initial_value.cookie       = "conectix"
+  initial_value.features     = 2
+  initial_value.ff           = 65536
+  initial_value.offset       = 18446744073709551615
+  initial_value.timestamp    = (Time.now - Time.parse("Jan 1, 2000 12:00:00 AM GMT")).to_i
+  initial_value.creator_app  = "rvhd"
+  initial_value.creator_ver  = 0
+  initial_value.creator_host = "Wi2k"
+  initial_value.type         = 2
+  initial_value.uuid         = SecureRandom.uuid.delete('-')
+  initial_value.saved_state  = 0
+  initial_value.reserved     = 0
+end

data/lib/vhd/library.rb

@@ -1,24 +1,88 @@
-module Vhd::Library
-  extend FFI::Library
-  ffi_lib "libvhd.so.0.1.1"
-  attach_function :vhd_create, [ :string, :uint64, :int, :uint64, :uint ], :int
+class Vhd::Library
+  VALID_TYPES = [:fixed]
 
-  HD_TYPE_FIXED   = 2
-  HD_TYPE_DYNAMIC = 3
+  attr_reader :name, :footer, :size
 
-  def self.create_dynamic_disk(name, size_in_gb)
-    size_in_bytes = self.size_in_bytes(size_in_gb)
+  def initialize(options={})
+    raise "Invalid vhd type" unless VALID_TYPES.include?(options[:type])
+    raise "Name is required" unless options[:name]
+    raise "Size is required" unless options[:size]
+    @name   = options[:name]
+    @footer = {}
+    @size   = options[:size]
 
-    vhd_create(name, size_in_bytes, HD_TYPE_DYNAMIC, 0, 0)
+    self.generate_footer
   end
 
-  def self.create_fixed_disk(name, size_in_gb)
-    size_in_bytes = self.size_in_bytes(size_in_gb)
+  def create
+    File.open(@name, "wb") { |f| f.print @footer.values.join }
+  end
+
+  def generate_footer(options={})
+    @footer[:cookie]       = "conectix".force_encoding("BINARY")
+    @footer[:features]     = ["00000002"].pack("H*")
+    @footer[:ff]           = ["00010000"].pack("H*")
+    @footer[:offset]       = ["FFFFFFFFFFFFFFFF"].pack("H*")
+    @footer[:time]         = [(Time.now - Time.parse("Jan 1, 2000 12:00:00 AM GMT")).to_i.to_s(16)].pack("H*")
+    @footer[:creator_app]  = "rvhd".force_encoding("UTF-8")
+    @footer[:creator_ver]  = ["00060002"].pack("H*")
+    @footer[:creator_host] = "Wi2k".force_encoding("UTF-8")
+    @footer[:orig_size]    = size_in_hex
+    @footer[:curr_size]    = size_in_hex
+    @footer[:geometry]     = nil
+    @footer[:disk_type]    = ["00000002"].pack("H*")
+    @footer[:checksum]     = nil
+    @footer[:uuid]         = SecureRandom.hex.scan(/../).map { |c| c.hex.chr.force_encoding("BINARY") }.join
+    @footer[:state]        = ["0"].pack("H*")
+    @footer[:reserved]     = Array("00"*427).pack("H*")
+
+    self.geometry
+    self.checksum
+  end
 
-    vhd_create(name, size_in_bytes, HD_TYPE_FIXED, 0, 2)
+  def size_in_bytes
+    (((@size * 1024) * 1024) * 1024).to_i
   end
 
-  def self.size_in_bytes(size_in_gb)
-    size_in_gb * 1024 * 1024 * 1024
+  def size_in_hex
+    hex_size = size_in_bytes.to_s(16)
+    hex_size = "0" + hex_size until hex_size.length == 16
+    [hex_size].pack("H*")
+  end
+
+  def geometry
+    max_size      = 65535 * 16 * 255 * 512
+    capacity      = size_in_bytes > max_size ? max_size : size_in_bytes
+    total_sectors = capacity / 512
+
+    if total_sectors > (65535 * 16 * 63)
+      sectors_per_track  = 255
+      heads_per_cylinder = 16
+    else
+      sectors_per_track     = 17
+      cylinders_times_heads = total_sectors / sectors_per_track
+      heads_per_cylinder    = (cylinders_times_heads + 1023) / 1024
+      heads_per_cylinder    = 4 if heads_per_cylinder < 4
+
+      cylinders_times_heads >= ((heads_per_cylinder * 1024) or (heads_per_cylinder > 16))
+      sectors_per_track  = 63
+      heads_per_cylinder = 16
+    end
+
+    cylinders = (total_sectors / sectors_per_track) / heads_per_cylinder
+
+    @footer[:geometry] = [cylinders, heads_per_cylinder, sectors_per_track].pack("SCC")
+  end
+
+  def checksum
+    checksum = 0
+
+    @footer.each do |k,v|
+      next if k == :checksum
+
+      checksum += v.codepoints.inject(0) { |r,c| r += c }
+    end
+
+    @footer[:checksum] = ["%08x" % ((~checksum).abs ^ 0xFFFFFFFF)].pack("H*")
   end
 end

data/lib/vhd/version.rb

@@ -1,3 +1,3 @@
 module Vhd
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/vhd_tests.rb

@@ -2,22 +2,17 @@ require 'spec_helper'
 
 describe "vhds" do
   before(:each) do
-    @file_name = "/tmp/#{Faker::Name.first_name}"
+    @file = Tempfile.new(Faker::Name.first_name.downcase)
   end
 
-  let(:file_name) { @file_name }
-
-  it "should create a dynamic vhd" do
-    Vhd::Library.create_dynamic_disk(file_name, 0.1)
-    File.size(file_name).should == 3584
-  end
+  let(:file_name) { @file.path }
 
   it "should create a fixed vhd" do
-    Vhd::Library.create_fixed_disk(file_name, 0.1)
-    File.size(file_name).should == 109052416
+    disk = Vhd::Library.new(type: :fixed, size: 1, name: file_name)
+    disk.create
   end
 
   after(:each) do
-    FileUtils.rm_f(@file_name)
+    @file.close
   end
 end

data/{vhd-ruby.gemspec → vhd.gemspec}

@@ -1,6 +1,5 @@
 # -*- encoding: utf-8 -*-
 require File.expand_path('../lib/vhd/version', __FILE__)
-require File.expand_path('../lib/vhd.rb', __FILE__)
 
 Gem::Specification.new do |gem|
   gem.authors       = ["Eugene Howe"]
@@ -16,5 +15,5 @@ Gem::Specification.new do |gem|
   gem.require_paths = ["lib"]
   gem.version       = Vhd::VERSION
 
-  gem.add_dependency "ffi"
+  gem.add_dependency "bit-struct"
 end

metadata

@@ -1,17 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: vhd
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Eugene Howe
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-20 00:00:00.000000000 Z
+date: 2014-01-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: ffi
+  name: bit-struct
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ! '>='
@@ -34,13 +34,14 @@ files:
 - Gemfile
 - Gemfile.lock
 - README.md
-- lib/libvhd.so.0.1.1
+- fixed_footer.bin
 - lib/vhd.rb
+- lib/vhd/footer.rb
 - lib/vhd/library.rb
 - lib/vhd/version.rb
 - spec/spec_helper.rb
 - spec/vhd_tests.rb
-- vhd-ruby.gemspec
+- vhd.gemspec
 homepage: ''
 licenses: []
 metadata: {}