initial commit

Author: gucki

.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ceph-ruby.gemspec
+gemspec

LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Corin Langosch
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

README.md

@@ -0,0 +1,29 @@
+# Ceph::Ruby
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'ceph-ruby'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ceph-ruby
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

ceph-ruby.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ceph-ruby/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "ceph-ruby"
+  gem.version       = CephRuby::VERSION
+  gem.authors       = ["Corin Langosch"]
+  gem.email         = ["info@corinlangosch.com"]
+  gem.description   = %q{TODO: Write a gem description}
+  gem.summary       = %q{TODO: Write a gem summary}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency('ffi', '~> 1.1.5')
+end

lib/ceph-ruby.rb

@@ -0,0 +1,8 @@
+require "ffi"
+
+require "ceph-ruby/version"
+require "ceph-ruby/rados"
+require "ceph-ruby/rbd"
+
+module CephRuby
+end

lib/ceph-ruby/rados.rb

@@ -0,0 +1,87 @@
+require "ceph-ruby/rados/lib"
+require "ceph-ruby/rados/pool"
+
+module CephRuby
+  class Rados
+    attr_accessor :state, :cluster
+
+    def self.version
+      major = FFI::MemoryPointer.new(:int)
+      minor= FFI::MemoryPointer.new(:int)
+      extra = FFI::MemoryPointer.new(:int)
+      Lib.rados_version(major, minor, extra)
+      {
+        :major => major.get_int(0),
+        :minor => minor.get_int(0),
+        :extra => extra.get_int(0),
+      }
+    end
+
+    def initialize(configuration_path = "/etc/ceph/ceph.conf")
+      cluster_p = FFI::MemoryPointer.new(:pointer)
+      ret = Lib.rados_create(cluster_p, nil)
+      raise "could not initialize rados cluster: #{ret}" if ret < 0
+      self.cluster = cluster_p.get_pointer(0)
+      self.state = :initialized
+      read_configuration_file(configuration_path)
+
+      if block_given?
+        connect
+        yield self
+        shutdown
+      end
+    end
+
+    def read_configuration_file(path = "/etc/ceph/ceph.conf")
+      require_state(:initialized, :connected)
+      raise ArgumentError, "path must be a string" unless path.is_a?(String)
+      ret = Lib.rados_conf_read_file(cluster, path)
+      raise "error reading configuration file '#{path}': #{ret}" if ret < 0
+    end
+
+    def shutdown
+      return unless cluster
+      Lib.rados_shutdown(cluster)
+      self.cluster = nil
+      self.state = :shutdown
+    end
+
+    def connect
+      require_state(:initialized)
+      ret = Lib.rados_connect(cluster)
+      raise "could not connect to cluster: #{ret}" if ret < 0
+      self.state = :connected
+    end
+
+    def exists?(name)
+      require_state(:connected)
+      raise ArgumentError, "name must be a string" unless name.is_a?(String)
+      ret = Lib.rados_pool_lookup(cluster, name)
+      return true if ret >= 0
+      return false if ret == -Errno::ENOENT::Errno
+      raise "error looking up pool '#{name}': #{ret}"
+    end
+
+    def pool(name, &block)
+      require_state(:connected)
+      raise ArgumentError, "name must be a string" unless name.is_a?(String)
+      ioctx_p = FFI::MemoryPointer.new(:pointer)
+      ret = Lib.rados_ioctx_create(cluster, name, ioctx_p)
+      raise "error creating io context for '#{name}': #{ret}" if ret < 0
+      Pool.new(self, name, ioctx_p.get_pointer(0), &block)
+    end
+
+    def rbd(name, &block)
+      pool(name) do |pool|
+        Rbd.new(pool, &block)
+      end
+    end
+
+    private
+
+    def require_state(*states)
+      return if states.include?(state)
+      raise "current state must be one of: #{states*", "}"
+    end
+  end
+end

lib/ceph-ruby/rados/lib.rb

@@ -0,0 +1,26 @@
+require "ffi"
+
+module CephRuby
+  class Rados
+    module Lib
+      extend FFI::Library
+
+      ffi_lib ['rados', 'librados.so.2']
+
+      attach_function 'rados_version', [:pointer, :pointer, :pointer], :void
+
+      attach_function 'rados_create', [:pointer, :string], :int
+      attach_function 'rados_connect', [:pointer], :int
+      attach_function 'rados_conf_read_file', [:pointer, :string], :int
+      attach_function 'rados_shutdown', [:pointer], :void
+
+      attach_function 'rados_pool_lookup', [:pointer, :string], :int
+
+      attach_function 'rados_ioctx_create', [:pointer, :string, :pointer], :int
+      attach_function 'rados_ioctx_destroy', [:pointer], :void
+
+      attach_function 'rados_write', [:pointer, :string, :buffer_in, :size_t, :off_t], :int
+      attach_function 'rados_read', [:pointer, :string, :buffer_out, :size_t, :off_t], :int
+    end
+  end
+end

lib/ceph-ruby/rados/pool.rb

@@ -0,0 +1,32 @@
+module CephRuby
+  class Rados
+    class Pool
+      attr_accessor :state, :cluster, :name, :ioctx
+
+      def initialize(cluster, name, ioctx)
+        self.cluster = cluster
+        self.name = name
+        self.ioctx = ioctx
+        self.state = :initialized
+
+        if block_given?
+          yield self
+          close
+        end
+      end
+
+      def close
+        require_state(:initialized)
+        Lib.rados_ioctx_destroy(ioctx)
+        self.state = :closed
+      end
+
+      private
+
+      def require_state(*states)
+        return if states.include?(state)
+        raise "current state must be one of: #{states*", "}"
+      end
+    end
+  end
+end

lib/ceph-ruby/rbd.rb

@@ -0,0 +1,54 @@
+require "ceph-ruby/rbd/lib"
+require "ceph-ruby/rbd/image"
+
+module CephRuby
+  class Rbd
+    attr_accessor :pool
+
+    def self.version
+      major = FFI::MemoryPointer.new(:int)
+      minor= FFI::MemoryPointer.new(:int)
+      extra = FFI::MemoryPointer.new(:int)
+      Lib.rbd_version(major, minor, extra)
+      {
+        :major => major.get_int(0),
+        :minor => minor.get_int(0),
+        :extra => extra.get_int(0),
+      }
+    end
+
+    def initialize(pool)
+      self.pool = pool
+      if block_given?
+        yield self
+        close
+      end
+    end
+
+    def close
+    end
+
+    def create(name, size, order = 0, &block)
+      raise ArgumentError, "name must be a string" unless name.is_a?(String)
+      raise ArgumentError, "size must be an integer" unless size.is_a?(Integer)
+      raise ArgumentError, "order must be an integer" unless order.is_a?(Integer)
+
+      order_p = FFI::MemoryPointer.new(:int)
+      order_p.put_int(0, order)
+      ret = Lib.rbd_create(pool.ioctx, name, size, order_p)
+      raise "error creating rbd image '#{name}': #{ret}" if ret < 0
+
+      Image.new(self, name, &block) if block
+    end
+
+    def open(name, &block)
+      Image.new(self, name, &block)
+    end
+
+    def destroy(name)
+      raise ArgumentError, "name must be a string" unless name.is_a?(String)
+      ret = Lib.rbd_remove(pool.ioctx, name)
+      raise "error destroying rbd image '#{name}': #{ret}" if ret < 0
+    end
+  end
+end

lib/ceph-ruby/rbd/image.rb

@@ -0,0 +1,74 @@
+module CephRuby
+  class Rbd
+    class Image
+      attr_accessor :state, :rbd, :name, :image
+
+      def initialize(rbd, name, snapshot = nil)
+        raise ArgumentError, "name must be a string" unless name.is_a?(String)
+        raise ArgumentError, "snapshot must be a string" if snapshot && !snapshot.is_a?(String)
+
+        image_p = FFI::MemoryPointer.new(:pointer)
+        ret = Lib.rbd_open(rbd.pool.ioctx, name, image_p, snapshot)
+        raise "could not open rbd image '#{name}': #{ret}" if ret < 0
+
+        self.image = image_p.get_pointer(0)
+        self.rbd = rbd
+        self.name = name
+        self.state = :initialized
+
+        if block_given?
+          yield self
+          close
+        end
+      end
+
+      def close
+        require_state(:initialized)
+        Lib.rbd_close(image)
+        self.state = :closed
+      end
+
+      def write(offset, data)
+        require_state(:initialized)
+        raise ArgumentError, "offset must be an integer" unless offset.is_a?(Integer)
+        size = data.bytesize
+        ret = Lib.rbd_write(image, offset, size, data)
+        raise "error writing #{size} bytes to image '#{name}': #{ret}" if ret < 0
+        raise "wrote only #{ret} of #{size} bytes to image '#{name}' @ #{offset}" if ret < size
+      end
+
+      def read(offset, size)
+        require_state(:initialized)
+        raise ArgumentError, "offset must be an integer" unless offset.is_a?(Integer)
+        raise ArgumentError, "size must be an integer" unless size.is_a?(Integer)
+        data_p = FFI::MemoryPointer.new(:char, size)
+        ret = Lib.rbd_read(image, offset, size, data_p)
+        raise "error reading #{size} bytes from image '#{name}' @ #{offset}: #{ret}" if ret < 0
+        data_p.get_bytes(0, ret)
+      end
+
+      def stat
+        stat = Lib::StatStruct.new
+        ret = Lib.rbd_stat(image, stat, stat.size)
+        raise "error getting stat of image '#{name}'" if ret < 0
+        stat
+      end
+
+      def resize(size)
+        ret = Lib.rbd_resize(image, size)
+        raise "error resizing image '#{name}' to #{size} bytes" if ret < 0
+      end
+
+      def size
+        stat[:size]
+      end
+
+      private
+
+      def require_state(*states)
+        return if states.include?(state)
+        raise "current state must be one of: #{states*", "}"
+      end
+    end
+  end
+end

lib/ceph-ruby/rbd/lib.rb

@@ -0,0 +1,34 @@
+require "ffi"
+
+module CephRuby
+  class Rbd
+    module Lib
+      extend FFI::Library
+
+      ffi_lib ['rbd', 'librbd.so.1']
+
+      attach_function 'rbd_version', [:pointer, :pointer, :pointer], :void
+
+      attach_function 'rbd_create', [:pointer, :string, :size_t, :pointer], :int
+      attach_function 'rbd_remove', [:pointer, :string], :int
+
+      attach_function 'rbd_open', [:pointer, :string, :pointer, :string], :int
+      attach_function 'rbd_close', [:pointer], :void
+
+      attach_function 'rbd_write', [:pointer, :off_t, :size_t, :buffer_in], :int
+      attach_function 'rbd_read', [:pointer, :off_t, :size_t, :buffer_out], :int
+      attach_function 'rbd_stat', [:pointer, :pointer, :size_t], :int
+      attach_function 'rbd_resize', [:pointer, :size_t], :int
+
+      class StatStruct < FFI::Struct
+        layout :size, :int,
+          :obj_size, :int,
+          :num_objs, :int,
+          :order, :int,
+          :block_name_prefix, [:char, 24],
+          :parent_pool, :int,
+          :parent_name, [:char, 96]
+      end
+    end
+  end
+end

lib/ceph-ruby/version.rb

@@ -0,0 +1,3 @@
+module CephRuby
+  VERSION = "0.0.1"
+end