View difference between Paste ID: RrkbXYZu and X0D4TVH5

 # This ruby program will patch the linker executable (link.exe)
 # so that linker warning LNK4099 is ignorable.
 #
 # Reference: http://www.bottledlight.com/docs/lnk4099.html
 
 require "fileutils"
 
 def link_exes()
 	res = []
 	res << File.join(ENV["VS90COMNTOOLS"], "../../VC/bin/link.exe") if ENV["VS90COMNTOOLS"]
 	res << File.join(ENV["VS100COMNTOOLS"], "../../VC/bin/link.exe") if ENV["VS100COMNTOOLS"]
 	res << File.join(ENV["XEDK"], "bin/win32/link.exe") if ENV["XEDK"]
 	return res
 end
 
 def patch_link_exe(exe)
 	data = nil
 	File.open(exe, "rb") {|f| data = f.read}
-	pattern = [4099].pack("I")
+	unpatched = [4088, 4099, 4105].pack("III")
-	offset = 0
+	patched = [4088, 65535, 4105].pack("III")
-	i = 0
+
-	while true
+	if data.scan(patched).size > 0
-		i = data.index(pattern, offset)
+		puts "* Already patched #{exe}"
-		return false unless i
+		return
-		offset = i+1
+
-		next unless i%4 == 0
+
-		int_before = data[i-4,4].unpack("I")[0]
+	num_unpatched = data.scan(unpatched).size
-		int_after = data[i+4,4].unpack("I")[0]
+	raise "Multiple patch locations in #{exe}" if num_unpatched > 1
-		break if int_before == 4088 && int_after == 4105
+	raise "Patch location not found in #{exe}" if num_unpatched == 0
 
-	puts "* Found patch location #{exe}:#{i}"
+	offset = data.index(unpatched)
 	puts "* Found patch location #{exe}:#{offset}"
 	bak = exe + "-" + Time.now.strftime("%y%m%d-%H%M%S") + ".bak"
 	puts "  Creating backup #{bak}"
 	FileUtils.cp(exe, bak)
-	data[i,4] = [65535].pack("I")
+
 	data[offset,unpatched.size] = patched
 	File.open(exe, "wb") {|f| f.write(data)}
 	return true
 end
 
 link_exes.each do |exe|
 	patch_link_exe(exe)
 end