ruby-macho 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6903e361104a718205c145e401ee484e38289217
-  data.tar.gz: 1bcea8a215a1e19b2cde870818206152582162be
+  metadata.gz: 2b104665e169f8988913d6032d3755032d814a2f
+  data.tar.gz: fc3f95ed644f84e913a78709227e178f33e9e995
 SHA512:
-  metadata.gz: fc14af02958fa369eac1e5fe6a4cce3fff48d21797d601fe4d4778a80336253fae58864994504a2612989ab718bcdb0c475edb797792dbbdfb6d09eb1840657d
-  data.tar.gz: a0d438e785176bbc06556889440aaf013181898ceec4012a837c181c785d5a6be508fc830b0027e456107682cfd5845040a2c1bef6338e05cfdb4e6212c1f57a
+  metadata.gz: 986523ca44a54d65a2d1624b31cbd612f520bab44ffaf405971052174224e9fedde4aabf1aa5d91fe451ee58cde5d892d95030bf6dad8f4688389fe8fc3f18dd
+  data.tar.gz: b5e6ffe8212660e32ed089d7dba2294818257aeb523b8eac8a3d4d66b9a549a210b40037109c26ff2cd31e47c7921fbe3389153b86dff30816efab83a061cbd4

data/lib/macho/fat_file.rb

@@ -48,12 +48,23 @@ module MachO
 		end
 
 		def linked_dylibs
-			dylibs = machos.map(&:linked_dylibs)
+			dylibs = machos.map(&:linked_dylibs).flatten
 
 			# can machos inside fat binaries have different dylibs?
 			dylibs.uniq!
 		end
 
+		# stub
+		def change_dylib(old_path, new_path)
+			machos.each do |macho|
+				macho.change_install_name(old_path, new_path)
+			end
+
+			synchronize_raw_data
+		end
+
+		alias :change_install_name :change_dylib
+
 		def write(filename)
 			File.open(filename, "wb") { |f| f.write(@raw_data) }
 		end

data/lib/macho/load_commands.rb

@@ -163,6 +163,12 @@ module MachO
 	SEG_UNIXSTACK = "__UNIXSTACK"
 	SEG_IMPORT = "__IMPORT"
 
+	# constant bits for flags in SegmentCommand, SegmentCommand64
+	SG_HIGHVM = 0x1
+	SG_FVMLIB = 0x2
+	SG_NORELOC = 0x4
+	SG_PROTECTED_VERSION_1 = 0x8
+
 	# Mach-O load command structure
 	# this is the most generic load command - only cmd ID and size are
 	# represented, and no actual data. used when a more specific class

data/lib/macho/macho_file.rb

@@ -217,11 +217,74 @@ module MachO
 			dylibs
 		end
 
-		# stub
-		def change_dylib!(old_path, new_path)
-			raise DylibUnknownError.new(old_path) unless linked_dylibs.include?(old_path)
+		# TODO: genericize change_dylib and dylib_id= for DRYness
+		def change_dylib(old_path, new_path)
+			idx = linked_dylibs.index(old_path)
+			raise DylibUnknownError.new(old_path) if idx.nil?
+
+			# this is a bit of a hack - since there is a 1-1 ordered association
+			# between linked_dylibs and command('LC_LOAD_DYLIB'), we can use
+			# their indices interchangeably to avoid having to loop.
+			dylib_cmd = command('LC_LOAD_DYLIB')[idx]
+
+			if magic32?
+				cmd_round = 4
+			else
+				cmd_round = 8
+			end
+
+			new_sizeofcmds = header[:sizeofcmds]
+			old_install_name = old_path.dup
+			new_install_name = new_path.dup
+
+			old_pad = MachO.round(old_install_name.size, cmd_round) - old_install_name.size
+			new_pad = MachO.round(new_install_name.size, cmd_round) - new_install_name.size
+
+			old_install_name << "\x00" * old_pad
+			new_install_name << "\x00" * new_pad
+
+			new_size = DylibCommand.bytesize + new_install_name.size
+			new_sizeofcmds += new_size - dylib_cmd.cmdsize
+
+			low_fileoff = 2**64
+
+			segments.each do |seg|
+				sections(seg).each do |sect|
+					if sect.size != 0 && !sect.flag?(S_ZEROFILL) &&
+							!sect.flag?(S_THREAD_LOCAL_ZEROFILL) &&
+							sect.offset < low_fileoff
+
+						low_fileoff = sect.offset
+					end
+				end
+			end
+
+			if new_sizeofcmds + header.bytesize > low_fileoff
+				raise HeaderPadError.new(@filename)
+			end
+
+			set_sizeofcmds(new_sizeofcmds)
+
+			@raw_data[dylib_cmd.offset + 4, 4] = [new_size].pack("V")
+
+			@raw_data.slice!(dylib_cmd.offset + dylib_cmd.name...dylib_cmd.offset + dylib_cmd.class.bytesize + old_install_name.size)
+
+			@raw_data.insert(dylib_cmd.offset + dylib_cmd.name, new_install_name)
+
+			null_pad = old_install_name.size - new_install_name.size
+
+			if null_pad < 0
+				@raw_data.slice!(new_sizeofcmds + header.bytesize, null_pad.abs)
+			else
+				@raw_data.insert(new_sizeofcmds + header.bytesize, "\x00" * null_pad)
+			end
+
+			header = get_mach_header
+			load_commands = get_load_commands
 		end
 
+		alias :change_install_name :change_dylib
+
 		# get all sections in a segment by name
 		def sections(segment)
 			sections = []

metadata

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