1
0
mirror of https://github.com/Melon-Bread/EagleJab synced 2024-11-24 20:28:19 -05:00
EagleJab/FalconPunch2.py
2015-05-01 17:48:03 -04:00

52 lines
1.7 KiB
Python
Executable File

#!/usr/bin/env python2
import os, argparse, socket, sys, struct
import tkMessageBox
from io import open
import tkSimpleDialog, SimpleDialog
import tkFileDialog, FileDialog
# New & improved args parsing
parser = argparse.ArgumentParser(description=u'Sends .CIA files to the 3DS via FBI')
parser.add_argument(u'-c', u'--cia', help=u'.CIA rom file', metavar=u'FILE', required=False, nargs=1)
parser.add_argument(u'-i', u'--ip', help=u'The IP address of the target 3DS (e.g. 192.168.1.123)', metavar=u'STRING',
required=False, nargs=1)
args = parser.parse_args()
# Ask for the desired .CIA file if none is given
if args.cia:
cia = args.cia[0]
else:
cia = tkFileDialog.askopenfilename(title=u"Falcon Punch - Choose CIA to Send", initialdir=u"/",
defaultextension=u".cia", filetypes=[(u"CIA File", u"*.cia")], multiple=False)
statinfo = os.stat(cia)
fbiinfo = struct.pack(u'!q', statinfo.st_size)
# Asks for the IP address of the 3DS if none is given
if args.ip:
dsip = args.ip[0]
else:
dsip = tkSimpleDialog.askstring(u"Falcon Punch", u"Enter 3Ds' IP:")
file = open(cia, u"rb")
sock = socket.socket()
sock.connect((dsip, 5000))
sock.send(fbiinfo)
# Sends the each chunk of the .CIA till there is nothing left
while True:
chunk = file.read(16384)
if not chunk:
# Prints or displays a confirmation based on how the program is given info
confirmation = u"Sent: \n" + cia + u"\n to the 3DS (" + dsip + u")"
if args.cia and args.ip:
print confirmation
else:
tkMessageBox.showinfo(u"Falcon Punch", confirmation)
break # EOF
sock.sendall(chunk)
sock.close()
sys.exit()