Example #1
0
class Auto(object):
   def __init__(self, **kwargs):
      self.data = Data(**kwargs)

      self.run()

   def __setitem__(self, key, value):
      self.__dict__[key] = value
   def __getitem__(self, key):
      return self.__dict__[key]

   def run(self):
      pool = []
      self.data.log("If you want to exit hit \"CTRL + c\"")
      for key, value in self.data.get_all_servers().items():
         ip = key
         interface = self.data.find_interface(ip)
         tcp = TcpDump(self.data, ip, interface)
         pool.append(tcp)
         
      try: 
         while True:
            count = 0
            for i in pool:
               if i.process.is_alive() == False and i.process_check.is_alive() == False:
                  count += 1

            if count == len(pool):
               self.data.log("\n*** Exiting ***")
               sys.exit()

            continue
      except KeyboardInterrupt:
         self.data.log("\n*** Exiting ***")
         for i in pool:
            i.kill_tcpdump()
         self.data.log("\n*** Done ***")
         sys.exit()
Example #2
0
class Manual(object):
   def __init__(self, **kwargs):
      self.data = Data(**kwargs)
      if "server_ip" in kwargs:
         self.server_ip = kwargs['server_ip']

      if "interface" in kwargs:
         self.interface = kwargs['interface']

      if self.server_ip == False and self.interface == False:
         self.server_ip = self.data.find_ip()
         self.interface = []

         for i in self.server_ip: 
            self.interface.append(self.data.find_interface(i))

         self.interactive(self.data)

      if self.server_ip != False and self.interface == False:
         test_ip = self.data.find_ip()

         self.data.test_ip(self.server_ip)
         if self.server_ip not in test_ip:
            self.data.log("No nfs server mounted with specified ip (" + self.server_ip + ")")
            sys.exit()

         interface = self.data.find_interface(self.server_ip)
         self.run(self.server_ip, interface)

      else:
         test_ip = self.data.find_ip()

         self.data.test_ip(self.server_ip)
         if self.server_ip not in test_ip:
            self.data.log("No nfs server mounted with specified ip (" + self.server_ip + ")")
            sys.exit()

         self.run(self.server_ip, self.interface)

   def run(self, ip, interface):
      try:
         output = "Starting Manual Mode"
         self.data.log(output)

         tcp = TcpDump(self.data, ip, interface)
         if tcp.error == False:
            self.data.log("\nIf you want to exit hit \"CTRL + c\"")

            while True:
               time.sleep(0.5)
               if tcp.process_check.is_alive() == False and tcp.process.is_alive() == False:
                  self.data.log("*** Exiting ***")
                  self.data.log("*** Done ***")
                  sys.exit()
               continue

      except KeyboardInterrupt:
         self.data.log("\n*** Exiting ***")
         tcp.kill_tcpdump()
         self.data.log("*** Done ***")
         sys.exit()

      
   def interactive(self, data):
      output = "Starting Manual Mode\n"
      servers = self.data.get_all_servers()

      for (key, value), i in zip(servers.items(), range(0, len(servers))):
         output += "\t" + str(i) +") " + value['server_ip'] + " on " + value['client_mount'] + "\n"
      output += "Which server would you like to test on " + str(range(0, len(servers))) + ": "
     
      options = range(0, len(servers))
  
      user = self.data.log(output, log_input=True)

      while True:
         try:
            user = int(user)
         except:
            self.data.log("\n*** Invalid selection ***\n")
            user = self.data.log(str("Which server would you like to test on " + str(range(0, len(servers))) + ": "), log_input=True)
            continue

         if user in options:
            for key, value in servers.items():
               if value['mapping'] == user:
                  tcp = TcpDump(self.data, value['server_ip'], value['interface'])

                  try:
                     
                     if tcp.error == False: 
                        self.data.log("\nIf you want to exit hit \"CTRL + c\"")

                     while True:
                        time.sleep(0.5)
                        if tcp.process_check.is_alive() == False and tcp.process.is_alive() == False:
                           self.data.log("*** Exiting ***")
                           self.data.log("*** Done ***")
                           sys.exit(1)
                        continue
                  

                  except KeyboardInterrupt:
                     self.data.log("\n*** Exiting ***")
                     tcp.kill_tcpdump()
                     self.data.log("*** Done ***")
                     sys.exit()
         else:
            self.data.log("\n*** Invalid selection ***\n")
            user = self.data.log(str("Which server would you like to test on " + str(range(0, len(servers))) + ": "), log_input=True)