WinPayloads.py

#!/usr/bin/python
from lib.main import *
from lib.payloadextras import *
from lib.startmetasploit import *
from lib.menu import *


try:
    from lib.psexecspray import *
except:
    print t.bold_red + "[!] Rerun the setup.sh" + t.normal

if not re.search('winpayloads', os.getcwd().lower()):
    print t.bold_red + "[!!] Please Run From Winpayloads Dir" + t.normal
    sys.exit(1)

DIR = os.path.expanduser('~') + '/winpayloads'
if not os.path.isdir(DIR):
    os.mkdir(DIR)


try:
    print t.bold_green + "Checking if up-to-date || ctr + c to cancel" + t.normal
    gitrev = subprocess.check_output(['git', 'rev-parse', 'HEAD']).rstrip()
    gitlsremote = subprocess.check_output(['git', 'ls-remote', 'origin', 'master']).split()[0]
    if gitrev != gitlsremote:
        updateornah = raw_input(t.bold_red + "Do you want to update WinPayloads? y/[n]: " + t.normal)
        if updateornah.lower() == "y":
            p = subprocess.Popen(['git','pull'])
            p.wait()
            print t.bold_yellow + "Reload Winpayloads..." + t.normal
            sys.exit()
except subprocess.CalledProcessError:
    print  t.bold_red + "[!] No Connection to Github" + t.normal
except KeyboardInterrupt:
    pass


from lib.listener import StartAsync
async = StartAsync()
async.start()

try:
    getAndRunMainMenu()
except KeyboardInterrupt:
    print t.bold_green + '\n[*] Cleaning Up\n' + t.normal
    subprocess.call(['rm *.rc'], shell=True,
                    stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    subprocess.call(['rm *.ps1'], shell=True,
                    stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    subprocess.call(['rm logdict*'], shell=True,
                    stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    sys.exit()