Minor bug fix

This commit is contained in:
Bernardo Damele 2010-10-29 12:36:37 +00:00
parent f7d42af046
commit 963fcb57b6

View File

@ -25,14 +25,6 @@ import socket
import subprocess import subprocess
import sys import sys
def checkLibs():
try:
from impacket import ImpactDecoder
from impacket import ImpactPacket
except ImportError:
sys.stderr.write('You need to install Python Impacket library first\n')
sys.exit(255)
def setNonBlocking(fd): def setNonBlocking(fd):
""" """
Make a file descriptor non-blocking Make a file descriptor non-blocking
@ -49,7 +41,12 @@ def main(src, dst):
sys.stderr.write('icmpsh master can only run on Posix systems\n') sys.stderr.write('icmpsh master can only run on Posix systems\n')
sys.exit(255) sys.exit(255)
checkLibs() try:
from impacket import ImpactDecoder
from impacket import ImpactPacket
except ImportError:
sys.stderr.write('You need to install Python Impacket library first\n')
sys.exit(255)
# Make standard input a non-blocking file # Make standard input a non-blocking file
stdin_fd = sys.stdin.fileno() stdin_fd = sys.stdin.fileno()