#DOSBox Identifier of Executables and Dispatcher (DIED)
#Version 1.1
#Yushatak 2010
dosboxpath="C:\Program Files (x86)\DOSBox-0.74\dosbox.exe"
from sys import argv,exit
from subprocess import Popen
from os import sep
import pythoncom
from win32com.shell import shell, shellcon
import os.path
import struct
exe=argv[2]
mode=argv[1]
if mode=="dispatch" or mode=="analyze":
	f=open(exe,"rb")
	data=f.read(60) #read dos header info
	print "Location: "+str(f.tell())
	reloc=f.read(4) #read PE header location
	print "Location: "+str(f.tell())
	offset=struct.unpack('<I',reloc)[0]
	if mode=="analyze":
		print "Analyzing..."
		print "File: "+exe
		if not offset==0:
			print "PE Offset: "+str(offset)
	if (data[0]=="N" and data[1]=="E"): 
		if mode=="analyze":
			print "Type: NE Executable"
	if (data[0]=="L" and data[1]=="E"):
		if mode=="analyze":
			print "Type: LE Executable"
	if (data[0]=="M" and data[1]=="Z"):
		f.seek(offset,0) #seek to absolute position in file described by offset
		pe=f.read(4) #read 4 bytes
		pe=pe.encode('hex')
		if pe=="50450000": #<-- "PE00" in hex
			if mode=="analyze":
				print "Type: PE Executable"
				if not (pe.find("UPX")==-1):
					print "Compression: UPX"
				exit()
			Popen([exe]+argv[3:])
			exit()
		else:
			#execute dosbox
			if mode=="analyze":
				print "Type: MZ Executable"
				exit()
			print "Launching DOSBox"
			Popen([dosboxpath,exe," -exit -noconsole","-fullscreen"])
			exit()
elif mode=="shortcut":
	shortcut=pythoncom.CoCreateInstance(shell.CLSID_ShellLink,None,pythoncom.CLSCTX_INPROC_SERVER,shell.IID_IShellLink)
	shortcut.SetPath('"'+dosboxpath+'"')
	shortcut.SetIconLocation(exe[:-4]+".ico", 0)
	print exe[:-4]
	shortcut.SetArguments(exe+' -exit -noconsole -fullscreen')
	file=shortcut.QueryInterface(pythoncom.IID_IPersistFile)
	file.Save(os.path.join(shell.SHGetFolderPath(0,shellcon.CSIDL_DESKTOP,0,0),os.path.split(exe)[1][:-4]+".lnk"),0)

else:
	print "Mode Error!"