#coding=utf-8

import sys


##
# Première ébauche de code
# afin de modifier la chaine hexadecimale correspondant a une instruction
# ASM, (code à modifier, afin de gérer les programmes plus gros où peuvent
# se trouver plusieurs itérations de la même séquence Hexa'.
#
# (Pour cette version, il est conseillé de donner une serie de opcodes plus large
# que celle de l'instruction à proprement parlé.)
##




##
# generation de opcodes à partir du String entré par l'utilisateur
##
def opcode_gen(x):
 
        buf = ""
        finaly_str = ""
        i = 0
       
        for lettre in x:
                buf += lettre
                if (i % 2) != 0:
			buf = chr(int(buf,16))
                        finaly_str += buf 			
                        buf = ""			   
                i +=1
 
        
        return finaly_str



##
# Recherches modifications & copie du code
##
def make_crack(prog, initial, final):

	f_read = open(prog,"rb")
	data = f_read.read()
	
	if (opcode_gen(initial) in data):
		print "[*] Instruction found"

		data = data.replace(opcode_gen(initial),opcode_gen(final),1)

		name = raw_input(" Name of the new cracked binary : ")
		f = open(name,"wb")
		f.write(data)
		f.close()
		f_read.close() 

		print "[*] Succes !!!"
	else:
		print "[!] Error : Instruction not found"



##
# Fonction main()
##
def main(prog):
	print "[*] Binary name : " + prog
	i = raw_input(" Find (exmpl : 7404) : ")
	f = raw_input(" Replace by : ")
	if (i != "" and f != ""):
		make_crack(prog, i, f)
	else:
		print "[!] Error : NULL ?"

##
# Header
##
def header():
	print """
---------------------------------
-App : PatchR
-Powered by HWC-CREW
-Butterfly Project
-Desc : Replace opcodes in a binary (to change instructions)
---------------------------------
	"""



if (len(sys.argv) < 2):
	print "[!] Error : please specify a binary file"
	print "[*] Use : ./patchR [binary_name]"

else:
	header()
	main(sys.argv[1])