mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2025-02-18 12:30:36 +03:00
Adjustments
This commit is contained in:
parent
53e735ea9d
commit
72a901347d
|
@ -25,7 +25,7 @@ import socket
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
def checkImpacketLibs():
|
def checkLibs():
|
||||||
try:
|
try:
|
||||||
from impacket import ImpactDecoder
|
from impacket import ImpactDecoder
|
||||||
from impacket import ImpactPacket
|
from impacket import ImpactPacket
|
||||||
|
@ -46,10 +46,10 @@ def setNonBlocking(fd):
|
||||||
|
|
||||||
def main(src, dst):
|
def main(src, dst):
|
||||||
if subprocess.mswindows:
|
if subprocess.mswindows:
|
||||||
sys.stderr.write('ERROR: icmpsh master can only be run on Posix systems\n')
|
sys.stderr.write('icmpsh master can only run on Posix systems\n')
|
||||||
sys.exit(255)
|
sys.exit(255)
|
||||||
|
|
||||||
checkImpacketLibs()
|
checkLibs()
|
||||||
|
|
||||||
# Make standard input a non-blocking file
|
# Make standard input a non-blocking file
|
||||||
stdin_fd = sys.stdin.fileno()
|
stdin_fd = sys.stdin.fileno()
|
||||||
|
@ -61,7 +61,7 @@ def main(src, dst):
|
||||||
try:
|
try:
|
||||||
sock = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_ICMP)
|
sock = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_ICMP)
|
||||||
except socket.error, e:
|
except socket.error, e:
|
||||||
print 'You need to run icmpsh master with administrator privileges'
|
sys.stderr.write('You need to run icmpsh master with administrator privileges\n')
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
sock.setblocking(0)
|
sock.setblocking(0)
|
||||||
|
@ -132,13 +132,10 @@ def main(src, dst):
|
||||||
sock.sendto(ip.get_packet(), (dst, 0))
|
sock.sendto(ip.get_packet(), (dst, 0))
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
if subprocess.mswindows:
|
|
||||||
print 'ERROR: this icmpsh master can only be run on Posix systems'
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
if len(sys.argv) < 3:
|
if len(sys.argv) < 3:
|
||||||
print 'ERROR: missing mandatory options. Execute as root:'
|
msg = 'missing mandatory options. Execute as root:\n'
|
||||||
print './icmpsh-m.py <source IP address> <destination IP address>'
|
msg += './icmpsh-m.py <source IP address> <destination IP address>\n'
|
||||||
|
sys.stderr.write(msg)
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
main(sys.argv[1], sys.argv[2])
|
main(sys.argv[1], sys.argv[2])
|
||||||
|
|
Loading…
Reference in New Issue
Block a user