Merge pull request #9 from g0tmi1k/check-path

SSH: Check path before trying to brute force

lib/main.py

@@ -206,7 +206,7 @@ def openvpn(self):
  port = 443
 
  if not os.path.exists(self.openvpn_path):
- mess = "openvpn: %s path doesn't exists on the system !!!"% (self.openvpn_path)
+ mess = "openvpn: %s path doesn't exists on the system !!!" % os.path.abspath(self.openvpn_path)
  raise CrowbarExceptions(mess)
 
  if self.args.port is not None:
@@ -236,15 +236,15 @@ def openvpn(self):
  try:
  userfile = open(self.args.username_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists !!!"% self.args.username_file
+ mess = "File: %s doesn't exists !!!" % os.path.abspath(self.args.username_file)
  raise CrowbarExceptions(mess)
 
  for user in userfile:
  if self.args.passwd_file:
  try:
  passwdfile = open(self.args.passwd_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists !!!"% self.args.passwd_file
+ mess = "File: %s doesn't exists !!!" % os.path.abspath(self.args.passwd_file)
  raise CrowbarExceptions(mess) 
 
  for password in passwdfile:
@@ -262,7 +262,7 @@ def openvpn(self):
  try:
  passwdfile = open(self.args.passwd_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists !!!"% self.args.passwd_file
+ mess = "File: %s doesn't exists !!!" % os.path.abspath(self.args.passwd_file)
  raise CrowbarExceptions(mess) 
 
  for password in passwdfile:
@@ -301,7 +301,7 @@ def vnckey(self, *options):
  port = 5901
 
  if not os.path.exists(self.vncviewer_path):
- mess = "vncviewer: %s path doesn't exists on the system !!!"% (self.vncviewer_path)
+ mess = "vncviewer: %s path doesn't exists on the system !!!" % os.path.abspath(self.vncviewer_path)
  raise CrowbarExceptions(mess)
 
  if self.args.port is not None:
@@ -312,7 +312,7 @@ def vnckey(self, *options):
 
 
  if not os.path.isfile(self.args.key_file):
- mess = "Key file: \"%s\" doesn't exists."% self.args.key_file 
+ mess = "Key file: \"%s\" doesn't exists." % os.path.abspath(self.args.key_file)
  raise CrowbarExceptions(mess) 
 
 
@@ -355,7 +355,7 @@ def rdp(self):
  port = 3389
 
  if not os.path.exists(self.xfreerdp_path):
- mess = "xfreerdp: %s path doesn't exists on the system !!!"% (self.xfreerdp_path)
+ mess = "xfreerdp: %s path doesn't exists on the system !!!" % os.path.abspath(self.xfreerdp_path)
  raise CrowbarExceptions(mess)
 
  if self.args.port is not None:
@@ -378,7 +378,7 @@ def rdp(self):
  try:
  userfile = open(self.args.username_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists !!!"% self.args.username_file
+ mess = "File: %s doesn't exists !!!" % os.path.abspath(self.args.username_file)
  raise CrowbarExceptions(mess)
 
  for user in userfile:
@@ -389,7 +389,7 @@ def rdp(self):
  try:
  passwdfile = open(self.args.passwd_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists"% self.args.passwd_file 
+ mess = "File: %s doesn't exists" % os.path.abspath(self.args.passwd_file)
  raise CrowbarExceptions(mess)
 
  for password in passwdfile:
@@ -401,7 +401,7 @@ def rdp(self):
  try:
  passwdfile = open(self.args.passwd_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists"% self.args.passwd_file 
+ mess = "File: %s doesn't exists" % os.path.abspath(self.args.passwd_file)
  raise CrowbarExceptions(mess)
 
  for password in passwdfile:
@@ -448,7 +448,10 @@ def sshkey(self):
  pool = ThreadPool(self.args.thread)
  except Exception, err:
  raise CrowbarExceptions(str(err))
-
+
+ if not os.path.exists(self.args.key_file):
+ mess = "Key file/folder: \"%s\" doesn't exists." % os.path.abspath(self.args.key_file)
+ raise CrowbarExceptions(mess)
 
  for ip in self.ip_list:
  if not self.args.quiet:
@@ -458,7 +461,7 @@ def sshkey(self):
  try:
  userfile = open(self.args.username_file, "r").read().splitlines()
  except:
- mess = "File: %s doesn't exists !!!"% self.args.username_file
+ mess = "File: %s doesn't exists !!!" % os.path.abspath(self.args.username_file)
  raise CrowbarExceptions(mess)
 
  for user in userfile: