ec2_amitools 1.0.2

data/lib/ec2/platform/linux/mtab.rb

@@ -0,0 +1,74 @@
+# Copyright 2008-2014 Amazon.com, Inc. or its affiliates.  All Rights
+# Reserved.  Licensed under the Amazon Software License (the
+# "License").  You may not use this file except in compliance with the
+# License. A copy of the License is located at
+# http://aws.amazon.com/asl or in the "license" file accompanying this
+# file.  This file is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
+# the License for the specific language governing permissions and
+# limitations under the License.
+
+module EC2
+  module Platform
+    module Linux
+      LOCAL_FS_TYPES = ['ext2', 'ext3', 'ext4', 'xfs', 'jfs', 'reiserfs', 'tmpfs']
+      class Mtab
+        class Entry
+          REGEX = /^(\S+)\s+(\S+)\s+(\S+)\s+(\S+).*$/
+          attr_reader :device   # mounted device.
+          attr_reader :mpoint   # mount point.
+          attr_reader :fstype   # file system type.
+          attr_reader :options  # options
+          attr_reader :value    # entire line
+          
+          
+          def initialize(dev, mnt_point, fs_type, opts, line)
+            @device = dev
+            @mpoint = mnt_point
+            @fstype = fs_type
+            @options= opts
+            @value  = line
+          end
+          
+          def self.parse(line)
+            return nil if line[0,1] == '#'
+            if (m = REGEX.match(line))
+              parts = m.captures
+              return Entry.new(parts[0], parts[1], parts[2], parts[3], line.strip)
+            else
+              return nil
+            end
+          end
+          
+          def to_s
+            value
+          end
+          
+          def print
+            puts(to_s)
+          end
+        end
+        
+        attr_reader :entries
+        LOCATION = '/etc/mtab'
+        
+        def initialize(filename = LOCATION)
+          begin
+            f = File.new(filename, File::RDONLY)
+          rescue SystemCallError => e
+            raise FileError(filename, "could not open #{filename} to read mount table", e)
+          end
+          @entries = Hash.new
+          f.readlines.each do |line|
+            entry = Entry.parse(line)
+            @entries[entry.mpoint] = entry unless entry.nil?
+          end          
+        end
+        
+        def self.load
+          self.new()
+        end
+      end
+    end
+  end
+end

data/lib/ec2/platform/linux/pipeline.rb

@@ -0,0 +1,40 @@
+# Copyright 2008-2014 Amazon.com, Inc. or its affiliates.  All Rights
+# Reserved.  Licensed under the Amazon Software License (the
+# "License").  You may not use this file except in compliance with the
+# License. A copy of the License is located at
+# http://aws.amazon.com/asl or in the "license" file accompanying this
+# file.  This file is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
+# the License for the specific language governing permissions and
+# limitations under the License.
+
+#------------------------------------------------------------------------------
+require 'ec2/platform/base/pipeline'
+
+#------------------------------------------------------------------------------
+module EC2    
+  #----------------------------------------------------------------------------
+  module Platform
+    #--------------------------------------------------------------------------
+    module Linux
+      #------------------------------------------------------------------------    
+      class Pipeline < EC2::Platform::Base::Pipeline
+        
+        #----------------------------------------------------------------------
+        # Given a pipeline of commands, modify it so that we can obtain
+        # the exit status of each pipeline stage by reading the tempfile
+        # associated with that stage. 
+        def pipestatus(cmd)
+          command = cmd
+          command << ';' unless cmd.rstrip[-1,1] == ';'
+          command << ' ' unless cmd[-1,1] == ' '
+          list = []
+          @tempfiles.each_with_index do |file, index| 
+            list << "echo ${PIPESTATUS[#{index}]} > #{file.path}"
+          end
+          command + list.join(' & ')
+        end
+      end
+    end
+  end
+end

data/lib/ec2/platform/linux/rsync.rb

@@ -0,0 +1,114 @@
+# Copyright 2008-2014 Amazon.com, Inc. or its affiliates.  All Rights
+# Reserved.  Licensed under the Amazon Software License (the
+# "License").  You may not use this file except in compliance with the
+# License. A copy of the License is located at
+# http://aws.amazon.com/asl or in the "license" file accompanying this
+# file.  This file is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
+# the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'ec2/amitools/fileutil'
+module EC2
+  module Platform
+    module Linux
+      class Rsync
+        class Command
+          EXECUTABLE='rsync'
+          def initialize(e = EXECUTABLE)
+            @src   = nil
+            @dst   = nil
+            @options = []
+            @executable = e
+            @quiet = false
+          end
+          
+          def archive;        @options << '-rlpgoD';     self; end
+          def times;          @options << '-t';          self; end
+          def recursive;      @options << '-r';          self; end
+          def sparse;         @options << '-S';          self; end
+          def links;          @options << '-l';          self; end
+          def dereference;    @options << '-L';          self; end
+          def xattributes;    @options << '-X';          self; end
+          def version;        @options << '--version';   self; end
+          def src(path)       @src = path;               self; end
+          def dst(path)       @dst = path;               self; end
+          def quietly;        @quiet = true;             self; end
+          
+          alias :source :src
+          alias :from :src
+          alias :destination :dst
+          alias :to :dst
+          
+          def exclude(files)
+            if files.is_a? Array
+              files.each {|file| exclude file }
+            else
+              @options << "--exclude '#{files}'" unless files.nil?
+            end
+            self
+          end
+
+          def include(files)
+            if files.is_a? Array
+              files.each {|file| include file }
+            else
+              @options << "--include '#{files}'" unless files.nil?
+            end
+            self
+          end
+          
+          def expand
+            "#{@executable} #{@options.join(' ')} #{@src} #{@dst} #{'2>&1 > /dev/null' if @quiet}".strip
+          end
+        end
+
+        def self.symlinking?
+          begin
+            src = FileUtil.tempdir('ec2-ami-tools-rsync-test-src')
+            dst = FileUtil.tempdir('ec2-ami-tools-rsync-test-dst')
+            FileUtils.mkdir(src)
+            FileUtils.touch("#{src}/foo")
+            FileUtils.symlink("#{src}/foo", "#{src}/bar")
+            FileUtils.mkdir("#{src}/baz")
+            File.open("#{src}/baz/food", 'w+'){|io| io << IO.read(__FILE__) }
+            FileUtils.symlink("#{src}/baz/food", "#{src}/baz/bard")
+            FileUtils.mkdir(dst)
+            incantation = Command.new.archive.recursive.sparse.links.src("#{src}/").dst("#{dst}")
+            `#{incantation.expand} 2>&1`
+            rc = $?.exitstatus
+            return true if rc == 0
+            if rc == 23
+              #check that the structure was copied reasonably anyway
+              slist = Dir["#{src}/**/**"]
+              dlist = Dir["#{dst}/**/**"]
+              return false unless dlist == dlist
+              slist.each do |sitem|
+                ditem = item.gsub(src, dst)
+                return false unless dlist.include? ditem 
+                if File.file?(sitem) or File.symlink?(sitem)
+                  @out.print "comparing #{sitem} to #{ditem}" if @out
+                  return false unless IO.read(ditem) == IO.read(sitem)
+                end
+                if ['food', 'bard'].include? File.basename(ditem)
+                  return false unless IO.read(sitem) == IO.read(__FILE__)
+                end
+              end
+              return true
+            end
+            return false
+          rescue Exception
+            return false
+          ensure
+            FileUtils.rm_rf src
+            FileUtils.rm_rf dst
+          end      
+        end
+        
+        def self.usable?()
+          @@usable ||= self.symlinking?
+        end
+      end
+    end
+  end
+end

data/lib/ec2/platform/linux/tar.rb

@@ -0,0 +1,124 @@
+# Copyright 2008-2014 Amazon.com, Inc. or its affiliates.  All Rights
+# Reserved.  Licensed under the Amazon Software License (the
+# "License").  You may not use this file except in compliance with the
+# License. A copy of the License is located at
+# http://aws.amazon.com/asl or in the "license" file accompanying this
+# file.  This file is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
+# the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'ec2/platform/linux/constants'
+module EC2
+  module Platform
+    module Linux
+      class Tar
+        class Command
+          EXECUTABLE=EC2::Platform::Linux::Constants::Utility::TAR
+          def initialize(e = EXECUTABLE)
+            @files   = []
+            @options = []
+            @executable = e
+          end
+          
+          def version;        @options << '--version';   self; end
+          def verbose;        @options << '-v';          self; end
+          def create;         @options << '-c';          self; end
+          def bzip2;          @options << '-j';          self; end
+          def diff;           @options << '-d';          self; end
+          def gzip;           @options << '-z';          self; end
+          def extract;        @options << '-x';          self; end
+          def update;         @options << '-u';          self; end
+          def sparse;         @options << '-S';          self; end
+          def dereference;    @options << '-h';          self; end
+          
+          def archive(filename)
+            filename = '-' if filename.nil? 
+            @options << "-f #{filename}"
+            self
+          end
+
+          def owner(user)
+            @options << "--owner #{user}"
+            self
+          end
+
+          def group(grp)
+            @options << "--group #{grp}"
+            self
+          end
+          
+          def chdir(dir)
+            @options << "-C #{dir}" unless dir.nil?
+            self
+          end
+          
+          def add(filename, dir = nil)
+            item = dir.nil? ? filename : "-C #{dir} #{filename}"
+            @files << item
+            self
+          end
+          def expand
+            "#{@executable} #{@options.join(' ')} #{@files.join(' ')}".strip            
+          end
+        end
+        class Version
+          RECOMMENDED  = 'tar 1.15'
+          REGEX = /(?:tar).*?(\d+)\.(\d+)\.?(\d*)/
+          attr_reader :values
+          attr_reader :string
+          
+          def initialize(str=nil)
+            @string = str
+            @string = default if str.nil? or str.empty?
+            @values = Version.parse @string
+          end
+          def default
+            s = `#{Command.new.version.expand}`.strip
+            s = nil unless $? == 0
+            s
+          end
+          def string= (str)
+            @string = str
+            @values = Version.parse @string
+          end
+          def >= (other)
+            return nil if @values.nil?
+            if other.nil? or not other.is_a? Version
+              raise ArgumentError, "Cannot compare with invalid version #{other}"
+            end
+            @values.zip(other.values).each do |mine, others|
+              return false if mine < others
+              return true if mine > others
+            end
+            return true
+          end          
+          def usable?
+             self >= Version.new(Version::RECOMMENDED)
+          end
+          def self.parse(str)
+            match = REGEX.match(str)
+            return nil if match.nil?
+            begin
+              items = match.captures.collect do |cap|
+                cap.sub!(/^0*/, "")
+                case cap
+                when ""
+                  num = 0
+                else
+                  num = Integer(cap)
+                end
+              end
+            rescue ArgumentError
+              return nil
+            end
+            items
+          end          
+          def self.current
+            Version.new
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ec2/platform/linux/uname.rb

@@ -0,0 +1,50 @@
+# Copyright 2008-2014 Amazon.com, Inc. or its affiliates.  All Rights
+# Reserved.  Licensed under the Amazon Software License (the
+# "License").  You may not use this file except in compliance with the
+# License. A copy of the License is located at
+# http://aws.amazon.com/asl or in the "license" file accompanying this
+# file.  This file is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
+# the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'ostruct'
+require 'ec2/platform'
+
+module EC2
+  module Platform
+    module Linux
+      class Uname
+        @@uname ||= OpenStruct.new
+        def self.all
+          @@uname.all ||= `uname -a`.strip
+        end
+        def self.platform
+          @@uname.platform ||= `uname -i`.strip
+        end
+        def self.nodename
+          @@uname.nodename ||= `uname -n`.strip
+        end
+        def self.processor
+          @@uname.processor ||= `uname -p`.strip
+        end
+        def self.release
+          @@uname.release ||= `uname -r`.strip
+        end
+        def self.os
+          @@uname.os ||= `uname -s`.strip
+        end
+        def self.machine
+          @@uname.machine ||= `uname -m`.strip
+        end
+        def self.uname
+          @@uname
+        end        
+      end
+    end
+  end
+end
+if __FILE__ == $0
+   include EC2::Platform::Linux
+   puts "Uname = #{Uname.all.inspect}"
+end

data/lib/ec2/platform/linux.rb

@@ -0,0 +1,83 @@
+# Copyright 2008-2014 Amazon.com, Inc. or its affiliates.  All Rights
+# Reserved.  Licensed under the Amazon Software License (the
+# "License").  You may not use this file except in compliance with the
+# License. A copy of the License is located at
+# http://aws.amazon.com/asl or in the "license" file accompanying this
+# file.  This file is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See
+# the License for the specific language governing permissions and
+# limitations under the License.
+
+#------------------------------------------------------------------------------
+require 'ec2/platform/base'
+require 'ec2/platform/linux/identity'
+require 'ec2/platform/linux/architecture'
+require 'ec2/platform/linux/fstab'
+require 'ec2/platform/linux/mtab'
+require 'ec2/platform/linux/image'
+require 'ec2/platform/linux/rsync'
+require 'ec2/platform/linux/tar'
+require 'ec2/platform/linux/uname'
+require 'ec2/platform/linux/pipeline'
+require 'ec2/platform/linux/constants'
+
+module EC2
+  module Platform    
+    module Linux
+      module Distribution
+        include EC2::Platform::Base::Distribution
+        AMAZON    = 'Amazon Linux'
+        REDHAT    = 'Red Hat Linux'
+        GENTOO    = 'Gentoo'
+        DEBIAN    = 'Debian'
+        UBUNTU    = 'Ubuntu'
+        FEDORA    = 'Fedora'
+        SLACKWARE = 'Slackware'
+        SUSE      = 'SuSE Linux'
+        MANDRAKE  = 'Mandrake'
+        CAOS      = 'Caos Linux'
+        
+        IDENTITIES= [
+          # file                     distro                    regex
+          ['/etc/system-release-cpe', Distribution::AMAZON, /amazon/],
+          ['/etc/caos-release',       Distribution::CAOS,        nil],
+          ['/etc/debian-release',     Distribution::DEBIAN,      nil],
+          ['/etc/debian_version',     Distribution::DEBIAN,      nil],
+          ['/etc/fedora-release',     Distribution::FEDORA,      nil],
+          ['/etc/gentoo-release',     Distribution::GENTOO,      nil],
+          ['/etc/redhat-release',     Distribution::REDHAT,      nil],
+          ['/etc/slackware-version',  Distribution::SLACKWARE,   nil],
+          ['/etc/slackware-release',  Distribution::SLACKWARE,   nil],
+          ['/etc/SuSE-release',       Distribution::SUSE,        nil],
+          ['/etc/ubuntu-release',     Distribution::UBUNTU,      nil],
+          ['/etc/ubuntu-version',     Distribution::UBUNTU,      nil],
+          ['/etc/mandrake-release',   Distribution::MANDRAKE,    nil],
+        ]
+      end
+      
+      class System < EC2::Platform::Base::System
+        
+        BUNDLING_ARCHITECTURE = EC2::Platform::Linux::Architecture.bundling    
+        
+        #---------------------------------------------------------------------#
+        def self.distribution
+          Distribution::IDENTITIES.each do |file, distro, regex|
+            if File.exists? file 
+              if regex.is_a? Regexp
+                return distro if regex.match((IO.read file rescue nil))
+              else              
+                return distro
+              end
+            end
+          end
+          return Distribution::UNKNOWN
+        end
+        
+        #---------------------------------------------------------------------#              
+        def self.superuser?()
+          return `id -u`.strip == '0'
+        end
+      end
+    end
+  end
+end