start.py

import os
import inspect
import argparse

snippet = '''
# paste these codes into your exp.py
# https://github.com/matrix1001/welpwn # reserve this link :)
import sys
sys.path.insert(0,'{dir}')
from PwnContext.core import *
'''

template = '''
#https://github.com/matrix1001/welpwn
from PwnContext import *

try:
    from IPython import embed as ipy
except ImportError:
    print ('IPython not installed.')

if __name__ == '__main__':        
    # context.terminal = ['tmux', 'splitw', '-h'] # uncomment this if you use tmux
    context.log_level = 'debug'
    # functions for quick script
    s       = lambda data               :ctx.send(str(data))        #in case that data is an int
    sa      = lambda delim,data         :ctx.sendafter(str(delim), str(data)) 
    sl      = lambda data               :ctx.sendline(str(data)) 
    sla     = lambda delim,data         :ctx.sendlineafter(str(delim), str(data)) 
    r       = lambda numb=4096          :ctx.recv(numb)
    ru      = lambda delims, drop=True  :ctx.recvuntil(delims, drop)
    irt     = lambda                    :ctx.interactive()
    rs      = lambda *args, **kwargs    :ctx.start(*args, **kwargs)
    dbg     = lambda gs='', **kwargs    :ctx.debug(gdbscript=gs, **kwargs)
    # misc functions
    uu32    = lambda data   :u32(data.ljust(4, '\\0'))
    uu64    = lambda data   :u64(data.ljust(8, '\\0'))
    leak    = lambda name,addr :log.success('{} = {:#x}'.format(name, addr))

    ctx.binary = './pwn'
    ctx.remote_libc = './libc.so'
    ctx.remote = ('1.1.1.1', 1111)
    ctx.debug_remote_libc = False # True for debugging remote libc, false for local.

    rs()
    # rs('remote') # uncomment this for exploiting remote target

    libc = ctx.libc # ELF object of the corresponding libc.
'''

if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        prog='start.py',
        description='''Excellent pwn framework.
    Author:matrix1001
    Github:https://github.com/matrix1001/welpwn''')
    parser.add_argument(
        '--template',
        action='store_true',
        help='Print a template'
    )
    args = parser.parse_args()
    if args.template:
        print(template)
    else:
        currentdir = os.path.dirname(
            os.path.abspath(inspect.getfile(inspect.currentframe()))
            )
        print(snippet.format(dir=currentdir))