ruby-macho 0.1.9 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 939f8d565fc3d98bd633414e797b2f8cd70c0df6
-  data.tar.gz: 3ab244e120eba39d1f84a8b495de6c221f4fc8f0
+  metadata.gz: 513849997d23e143113243138d5924dc483a2a0b
+  data.tar.gz: 2e2d1256e5b42be47cdcb81c8981e35021c5fe6a
 SHA512:
-  metadata.gz: bb2d31798f5d4b6fefd00a2c3e5c562595331015f9a44b9d3b5bcce84b43db3c473236222412bc1b26e358161df2451e15c8ea292c1c0fc433292afc9871772c
-  data.tar.gz: 39bd050a693d85cd48e7ce3594d5cfd199732e171df7026474af03723bdd3c1a58108df08c078333fc0304ed84691c14a76bb15b6a168f8411f993e4e7b0b5d2
+  metadata.gz: 3e1bb350afe1cbbbfff4a50197886c4c7c6f4675f0de8e5480deafce7ee4986febd3b7d59c067753cea95ed4dc4a92b8e38d8c486344e3c7b2937bcaff22b800
+  data.tar.gz: dba7611dadf4672c0929bc5f3c8f70c60051b01c278cb0383209304526a780008a2b19c0e50848a42d962b78c43c11174dd7c0913f123bff88b8392c30acc9fe

data/lib/macho/exceptions.rb

@@ -74,4 +74,12 @@ module MachO
 			super "No such dylib name: #{dylib}"
 		end
 	end
+
+	# Raised when attempting to change an rpath that doesn't exist.
+	class RpathUnknownError < MachOError
+		# @param path [String] the unknown runtime path
+		def initialize(path)
+			super "No such runtime path: #{path}"
+		end
+	end
 end

data/lib/macho/macho_file.rb

@@ -228,6 +228,27 @@ module MachO
 
 		alias :change_dylib :change_install_name
 
+		# All runtime paths searched by the dynamic linker for the Mach-O.
+		# @return [Array<String>] an array of all runtime paths
+		def rpaths
+			command(:LC_RPATH).map(&:path).map(&:to_s)
+		end
+
+		# Changes the runtime path `old_path` to `new_path`
+		# @example
+		#  file.change_rpath("/usr/lib", "/usr/local/lib")
+		# @param old_path [String] the old runtime path
+		# @param new_path [String] the new runtime path
+		# @return [void]
+		# @raise [MachO::RpathUnknownError] if no such old runtime path exists
+		# @api private
+		def change_rpath(old_path, new_path)
+			rpath_cmd = command(:LC_RPATH).find { |r| r.path.to_s == old_path }
+			raise RpathUnknownError.new(old_path) if rpath_cmd.nil?
+
+			set_path_in_rpath(rpath_cmd, old_path, new_path)
+		end
+
 		# All sections of the segment `segment`.
 		# @param segment [MachO::SegmentCommand, MachO::SegmentCommand64] the segment being inspected
 		# @return [Array<MachO::Section>] if the Mach-O is 32-bit
@@ -408,14 +429,34 @@ module MachO
 			@raw_data[20..23] = new_size
 		end
 
-		# Updates the `name` field in a DylibCommand, regardless of load command type
+		# Updates the `name` field in a DylibCommand.
 		# @param dylib_cmd [MachO::DylibCommand] the dylib command
 		# @param old_name [String] the old dylib name
 		# @param new_name [String] the new dylib name
 		# @return [void]
-		# @raise [MachO::HeaderPadError] if the new name exceeds the header pad buffer
 		# @private
 		def set_name_in_dylib(dylib_cmd, old_name, new_name)
+			set_lc_str_in_cmd(dylib_cmd, dylib_cmd.name, old_name, new_name)
+		end
+
+		# Updates the `path` field in an RpathCommand.
+		# @param rpath_cmd [MachO::RpathCommand] the rpath command
+		# @param old_path [String] the old runtime name
+		# @param new_path [String] the new runtime name
+		# @return [void]
+		# @private
+		def set_path_in_rpath(rpath_cmd, old_path, new_path)
+			set_lc_str_in_cmd(rpath_cmd, rpath_cmd.path, old_path, new_path)
+		end
+
+		# Updates a generic LCStr field in any LoadCommand.
+		# @param cmd [MachO::LoadCommand] the load command
+		# @param lc_str [MachO::LoadCommand::LCStr] the load command string
+		# @param old_str [String] the old string
+		# @param new_str [String] the new string
+		# @raise [MachO::HeaderPadError] if the new name exceeds the header pad buffer
+		# @private
+		def set_lc_str_in_cmd(cmd, lc_str, old_str, new_str)
 			if magic32?
 				cmd_round = 4
 			else
@@ -423,19 +464,19 @@ module MachO
 			end
 
 			new_sizeofcmds = header.sizeofcmds
-			old_name = old_name.dup
-			new_name = new_name.dup
+			old_str = old_str.dup
+			new_str = new_str.dup
 
-			old_pad = MachO.round(old_name.size, cmd_round) - old_name.size
-			new_pad = MachO.round(new_name.size, cmd_round) - new_name.size
+			old_pad = MachO.round(old_str.size + 1, cmd_round) - old_str.size
+			new_pad = MachO.round(new_str.size + 1, cmd_round) - new_str.size
 
 			# pad the old and new IDs with null bytes to meet command bounds
-			old_name << "\x00" * old_pad
-			new_name << "\x00" * new_pad
+			old_str << "\x00" * old_pad
+			new_str << "\x00" * new_pad
 
-			# calculate the new size of the DylibCommand and sizeofcmds in MH
-			new_size = DylibCommand.bytesize + new_name.size
-			new_sizeofcmds += new_size - dylib_cmd.cmdsize
+			# calculate the new size of the cmd and sizeofcmds in MH
+			new_size = cmd.class.bytesize + new_str.size
+			new_sizeofcmds += new_size - cmd.cmdsize
 
 			low_fileoff = 2**64 # ULLONGMAX
 
@@ -458,17 +499,17 @@ module MachO
 			# update sizeofcmds in mach_header
 			set_sizeofcmds(new_sizeofcmds)
 
-			# update cmdsize in the dylib_command
-			@raw_data[dylib_cmd.offset + 4, 4] = [new_size].pack("V")
+			# update cmdsize in the cmd
+			@raw_data[cmd.offset + 4, 4] = [new_size].pack("V")
 
-			# delete the old name
-			@raw_data.slice!(dylib_cmd.offset + dylib_cmd.name.to_i...dylib_cmd.offset + dylib_cmd.class.bytesize + old_name.size)
+			# delete the old str
+			@raw_data.slice!(cmd.offset + lc_str.to_i...cmd.offset + cmd.class.bytesize + old_str.size)
 
-			# insert the new id
-			@raw_data.insert(dylib_cmd.offset + dylib_cmd.name.to_i, new_name)
+			# insert the new str
+			@raw_data.insert(cmd.offset + lc_str.to_i, new_str)
 
 			# pad/unpad after new_sizeofcmds until offsets are corrected
-			null_pad = old_name.size - new_name.size
+			null_pad = old_str.size - new_str.size
 
 			if null_pad < 0
 				@raw_data.slice!(new_sizeofcmds + header.class.bytesize, null_pad.abs)

data/lib/macho/tools.rb

@@ -17,12 +17,8 @@ module MachO
 		def self.change_dylib_id(filename, new_id)
 			file = MachO.open(filename)
 
-			if File.is_a? MachO::MachOFile
-				file.dylib_id = new_id
-				file.write!
-			else
-				raise MachOError.new("changing dylib ids for fat binaries is incomplete")
-			end
+			file.dylib_id = new_id
+			file.write!
 		end
 
 		# Changes a shared library install name in a Mach-O or Fat binary, overwriting the source file.
@@ -34,11 +30,8 @@ module MachO
 		def self.change_install_name(filename, old_name, new_name)
 			file = MachO.open(filename)
 
-			if File.is_a? MachO::MachOFile
-				file.change_install_name(old_name, new_name)
-			else
-				raise MachOError.new("changing install names for fat binaries is incomplete")
-			end
+			file.change_install_name(old_name, new_name)
+			file.write!
 		end
 
 		# Changes a runtime path in a Mach-O or Fat binary, overwriting the source file.

data/lib/macho.rb

@@ -12,5 +12,5 @@ require "#{File.dirname(__FILE__)}/macho/tools"
 # The primary namespace for ruby-macho.
 module MachO
 	# release version
-	VERSION = "0.1.9".freeze
+	VERSION = "0.2.0".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-macho
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.2.0
 platform: ruby
 authors:
 - William Woodruff
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-11-07 00:00:00.000000000 Z
+date: 2015-11-11 00:00:00.000000000 Z
 dependencies: []
 description: A library for viewing and manipulating Mach-O files in Ruby.
 email: william@tuffbizz.com