Python Benign.execute 예제들

예제 #1

파일: tester.py 프로젝트: stacklikemind/ictf-2020-challs-public

def test_benign(benign_fp):
    try:
        benign = imp.load_source('benign', benign_fp)
        from benign import Benign
        print 'Benign..'
        b = Benign()
        b.execute(HOST, PORT, '', '')
        res = b.result()
        print 'Result: %s' % str(res)
        assert res['ERROR'] == 0
    except AssertionError:
        print 'AssertionError while executing Benign'
        print traceback.format_exc()
        print '----------------------'
        print RED + 'ERROR' + ENDC
        sys.exit(2)
    except Exception:
        print 'Exception while executing Benign'
        print traceback.format_exc()
        print '----------------------'
        print RED + 'ERROR' + ENDC
        sys.exit(2)

예제 #2

FLAG = 'xxxflagxxx'

print 'Setflag..'
sf = SetFlag()
sf.execute(HOST, PORT, FLAG)
res = sf.result()
print 'Result: %s' % str(res)
flag_id = res['FLAG_ID']
cookie = res['TOKEN']

print 'Exploit..'
e = Exploit()
e.execute(HOST, PORT, flag_id)
res = e.result()
print 'Result: %s' % str(res)
assert res['FLAG'] == FLAG

print 'Getflag..'
cf = GetFlag()
cf.execute(HOST, PORT, flag_id, cookie)
res = cf.result()
print 'Result: %s' % str(res)
assert res['FLAG'] == FLAG

print 'Benign..'
b = Benign()
b.execute(HOST, PORT)
res = b.result()
print 'Result: %s' % str(res)
assert res['ERROR'] == 0

예제 #3

FLAG = 'xxxflagxxx'

print 'Setflag..'
sf = SetFlag()
sf.execute(HOST, PORT, FLAG)
res = sf.result()
print 'Result: %s' % str(res)
flag_id = res['FLAG_ID']
token = res['TOKEN']

print 'Exploit..'
e = Exploit()
e.execute(HOST, PORT, flag_id)
res = e.result()
print 'Result: %s' % str(res)
assert res['FLAG'] == FLAG

print 'Getflag..'
cf = GetFlag()
cf.execute(HOST, PORT, flag_id, token)
res = cf.result()
print 'Result: %s' % str(res)
assert res['FLAG'] == FLAG

print 'Benign..'
b = Benign()
b.execute(HOST, PORT, FLAG_ID, TOKEN)
res = b.result()
print 'Result: %s' % str(res)
assert res['ERROR'] == 0