Ejemplo n.º 1
0
 /**
  * Execute command
  *
  * @param  InputInterface  $input  Input instance
  * @param  OutputInterface $output Output instance
  *
  * @return int|null|void
  */
 public function execute(InputInterface $input, OutputInterface $output)
 {
     $this->elevateProcess($input, $output);
     $procList = array();
     $openFilesTotal = 0;
     $command = new CommandBuilder('lsof', '-n');
     $command->addPipeCommand(new CommandBuilder('grep', '-oE \'^[a-z]+\''))->addPipeCommand(new CommandBuilder('sort'))->addPipeCommand(new CommandBuilder('uniq', '-c'))->addPipeCommand(new CommandBuilder('sort', '-n'))->setOutputRedirect(CommandBuilder::OUTPUT_REDIRECT_NO_STDERR);
     $execOutput = $command->execute()->getOutput();
     foreach ($execOutput as $execOutputLine) {
         // get open files and proc name from output
         list($procOpenFiles, $procName) = explode(' ', trim($execOutputLine), 2);
         // add to total stats
         $openFilesTotal += $procOpenFiles;
         $procList[] = array('name' => $procName, 'open_files' => $procOpenFiles);
     }
     // ########################
     // Output
     // ########################
     /** @var \Symfony\Component\Console\Helper\Table $table */
     $table = new Table($output);
     $table->setHeaders(array('Process', 'Open Files'));
     foreach ($procList as $procRow) {
         $procRow['open_files'] = FormatUtility::number($procRow['open_files']);
         $table->addRow(array_values($procRow));
     }
     // Stats: average
     $table->addRow(new TableSeparator());
     $statsRow = array();
     $statsRow['name'] = 'Total';
     $statsRow['open_files'] = FormatUtility::number($openFilesTotal);
     $table->addRow(array_values($statsRow));
     $table->render();
     return 0;
 }
Ejemplo n.º 2
0
 /**
  * Execute command
  *
  * @param  InputInterface  $input  Input instance
  * @param  OutputInterface $output Output instance
  *
  * @return int|null|void
  */
 public function execute(InputInterface $input, OutputInterface $output)
 {
     $database = $input->getArgument('db');
     $dumpFile = $input->getArgument('file');
     if (!is_file($dumpFile) || !is_readable($dumpFile)) {
         $output->writeln('<p-error>File is not readable</p-error>');
         return 1;
     }
     $dumpFileType = PhpUtility::getMimeType($dumpFile);
     $output->writeln('<h2>Restoring dump "' . $dumpFile . '" into database "' . $database . '"</h2>');
     if (DatabaseConnection::databaseExists($database)) {
         // Dropping
         $output->writeln('<p>Dropping database</p>');
         $query = 'DROP DATABASE IF EXISTS ' . DatabaseConnection::sanitizeSqlDatabase($database);
         DatabaseConnection::exec($query);
     }
     // Creating
     $output->writeln('<p>Creating database</p>');
     $query = 'CREATE DATABASE ' . DatabaseConnection::sanitizeSqlDatabase($database);
     DatabaseConnection::exec($query);
     // Inserting
     putenv('USER='******'MYSQL_PWD=' . DatabaseConnection::getDbPassword());
     $commandMysql = new CommandBuilder('mysql', '--user=%s %s --one-database', array(DatabaseConnection::getDbUsername(), $database));
     // Set server connection details
     if ($input->getOption('host')) {
         $commandMysql->addArgumentTemplate('-h %s', $input->getOption('host'));
     }
     if ($input->getOption('port')) {
         $commandMysql->addArgumentTemplate('-P %s', $input->getOption('port'));
     }
     $commandFile = new CommandBuilder();
     $commandFile->addArgument($dumpFile);
     $commandFile->addPipeCommand($commandMysql);
     switch ($dumpFileType) {
         case 'application/x-bzip2':
             $output->writeln('<p>Using BZIP2 decompression</p>');
             $commandFile->setCommand('bzcat');
             break;
         case 'application/gzip':
         case 'application/x-gzip':
             $output->writeln('<p>Using GZIP decompression</p>');
             $commandFile->setCommand('gzcat');
             break;
         case 'application/x-lzma':
         case 'application/x-xz':
             $output->writeln('<p>Using LZMA decompression</p>');
             $commandFile->setCommand('xzcat');
             break;
         default:
             $output->writeln('<p>Using plaintext (no decompression)</p>');
             $commandFile->setCommand('cat');
             break;
     }
     $output->writeln('<p>Reading dump</p>');
     $commandFile->executeInteractive();
     $output->writeln('<h2>Database "' . $database . '" restored</h2>');
     return 0;
 }
Ejemplo n.º 3
0
 /**
  * Execute command
  *
  * @param  InputInterface  $input  Input instance
  * @param  OutputInterface $output Output instance
  *
  * @return int|null|void
  */
 public function execute(InputInterface $input, OutputInterface $output)
 {
     $this->elevateProcess($input, $output);
     $dockerInterface = $this->getApplication()->getConfigValue('docker', 'interface');
     $output->writeln('<h2>Starting network sniffing</h2>');
     $protocol = $this->getProtocol();
     $command = new CommandBuilder();
     switch ($protocol) {
         // ############################################
         // OSI LEVEL 2
         // ############################################
         // ##############
         // ARP
         // ##############
         case 'arp':
             $output->writeln('<p>Using protocol "arp"</p>');
             $command->setCommand('tshark');
             $command->addArgument('arp');
             break;
             // ############################################
             // OSI LEVEL 3
             // ############################################
             // ##############
             // ICMP
             // ##############
         // ############################################
         // OSI LEVEL 3
         // ############################################
         // ##############
         // ICMP
         // ##############
         case 'icmp':
             $output->writeln('<p>Using protocol "icmp"</p>');
             $command->setCommand('tshark');
             $command->addArgument('icmp');
             break;
             // ############################################
             // OSI LEVEL 4
             // ############################################
             // ##############
             // TCP connections
             // ##############
         // ############################################
         // OSI LEVEL 4
         // ############################################
         // ##############
         // TCP connections
         // ##############
         case 'con':
         case 'tcp':
             $output->writeln('<p>Using protocol "tcp"</p>');
             $command->setCommand('tshark');
             $command->addArgumentRaw('-R "tcp.flags.syn==1 && tcp.flags.ack==0"');
             break;
             // ############################################
             // OSI LEVEL 5-7
             // ############################################
             // ##############
             // HTTP
             // ##############
         // ############################################
         // OSI LEVEL 5-7
         // ############################################
         // ##############
         // HTTP
         // ##############
         case 'http':
             $output->writeln('<p>Using protocol "http"</p>');
             $command->setCommand('tshark');
             $command->addArgumentRaw('tcp port 80 or tcp port 443 -2 -V -R "http.request" -Tfields -e ip.dst -e http.request.method -e http.request.full_uri');
             break;
             // ##############
             // HTTP (full)
             // ##############
         // ##############
         // HTTP (full)
         // ##############
         case 'http-full':
             $output->writeln('<p>Using protocol "http" (full mode)</p>');
             $command->setCommand('tshark');
             $command->addArgumentRaw('tcp port 80 or tcp port 443 -2 -V -R "http.request || http.response"');
             break;
             // ##############
             // SOLR
             // ##############
         // ##############
         // SOLR
         // ##############
         case 'solr':
             $output->writeln('<p>Using protocol "solr"</p>');
             $command->setCommand('tcpdump');
             $command->addArgumentRaw('-nl -s0 -w- port 8983');
             $pipeCommand = new CommandBuilder('strings', '-n -8');
             $command->addPipeCommand($pipeCommand);
             break;
             // ##############
             // ELASTICSEARCH
             // ##############
         // ##############
         // ELASTICSEARCH
         // ##############
         case 'elasticsearch':
             $output->writeln('<p>Using protocol "elasticsearch"</p>');
             $command->setCommand('tcpdump');
             $command->addArgumentRaw('-A -nn -s 0 \'tcp dst port 9200 and (((ip[2:2] - ((ip[0]&0xf)<<2)) - ((tcp[12]&0xf0)>>2)) != 0)\'');
             break;
             // ##############
             // MEMCACHE
             // ##############
         // ##############
         // MEMCACHE
         // ##############
         case 'memcache':
         case 'memcached':
             $output->writeln('<p>Using protocol "memcache"</p>');
             $command->setCommand('tcpdump');
             $command->addArgumentRaw('-s 65535 -A -ttt port 11211| cut -c 9- | grep -i \'^get\\|set\'');
             break;
             // ##############
             // REDIS
             // ##############
         // ##############
         // REDIS
         // ##############
         case 'redis':
             $output->writeln('<p>Using protocol "redis"</p>');
             $command->setCommand('tcpdump');
             $command->addArgumentRaw('-s 65535 tcp port 6379');
             break;
             // ##############
             // SMTP
             // ##############
         // ##############
         // SMTP
         // ##############
         case 'smtp':
         case 'mail':
             $output->writeln('<p>Using protocol "smtp"</p>');
             $command->setCommand('tshark');
             $command->addArgumentRaw('tcp -f "port 25" -R "smtp"');
             break;
             // ##############
             // MYSQL
             // ##############
         // ##############
         // MYSQL
         // ##############
         case 'mysql':
             $output->writeln('<p>Using protocol "mysql"</p>');
             $command->setCommand('tshark');
             $command->addArgumentRaw('tcp -d tcp.port==3306,mysql -T fields -e mysql.query "port 3306"');
             break;
             // ##############
             // DNS
             // ##############
         // ##############
         // DNS
         // ##############
         case 'dns':
             $output->writeln('<p>Using protocol "dns"</p>');
             $command->setCommand('tshark');
             $command->addArgumentRaw('-nn -e ip.src -e dns.qry.name -E separator=" " -T fields port 53');
             break;
             // ##############
             // HELP
             // ##############
         // ##############
         // HELP
         // ##############
         default:
             $output->writeln('<p-error>Protocol not supported:</p-error>');
             $output->writeln('<p-error>  OSI layer 7: http, solr, elasticsearch, memcache, redis, smtp, mysql, dns</p-error>');
             $output->writeln('<p-error>  OSI layer 4: tcp</p-error>');
             $output->writeln('<p-error>  OSI layer 3: icmp</p-error>');
             $output->writeln('<p-error>  OSI layer 2: arp</p-error>');
             return 1;
             break;
     }
     switch ($command->getCommand()) {
         case 'tshark':
             $output->writeln('<p>Using sniffer "tshark"</p>');
             $command->addArgumentTemplate('-i %s', $dockerInterface);
             break;
         case 'tcpdump':
             $output->writeln('<p>Using sniffer "tcpdump"</p>');
             $command->addArgumentTemplate('-i %s', $dockerInterface);
             break;
         case 'ngrep':
             $output->writeln('<p>Using sniffer "ngrep"</p>');
             $command->addArgumentTemplate('-d %s', $dockerInterface);
             break;
     }
     $this->setTerminalTitle('sniffer', $protocol, '(' . $command->getCommand() . ')');
     $command->executeInteractive();
     return 0;
 }
Ejemplo n.º 4
0
 /**
  * Reload tty
  */
 public static function reloadTtyBanner($ttyName)
 {
     // Check if we can reload tty
     try {
         $who = new CommandBuilder('who');
         $who->addPipeCommand(new CommandBuilder('grep', '%s', array($ttyName)));
         $who->execute();
         // if there is no exception -> there is a logged in user
     } catch (\Exception $e) {
         // if there is an exception -> there is NO logged in user
         try {
             $ps = new CommandBuilder('ps', 'h -o pid,comm,args -C getty');
             $ps->addPipeCommand(new CommandBuilder('grep', '%s', array($ttyName)));
             $output = $ps->execute()->getOutput();
             if (!empty($output)) {
                 $outputLine = trim(reset($output));
                 $outputLineParts = preg_split('/[\\s]+/', $outputLine);
                 list($pid) = $outputLineParts;
                 posix_kill($pid, SIGHUP);
             }
         } catch (\Exception $e) {
         }
     }
 }
Ejemplo n.º 5
0
 /**
  * Execute command
  *
  * @param  InputInterface  $input  Input instance
  * @param  OutputInterface $output Output instance
  *
  * @return int|null|void
  */
 public function execute(InputInterface $input, OutputInterface $output)
 {
     $this->elevateProcess($input, $output);
     $pid = null;
     $grep = $input->getArgument('grep');
     $command = new CommandBuilder('strace', '-f');
     $command->setOutputRedirect(CommandBuilder::OUTPUT_REDIRECT_ALL_STDOUT);
     $output->writeln('<h2>Starting process stracing</h2>');
     if (empty($pid)) {
         list($pidList, $processList) = $this->buildProcessList();
         if ($input->getOption('all')) {
             $pid = 'all';
         } else {
             try {
                 $question = new ChoiceQuestion('Please choose process for tracing', $processList);
                 $question->setMaxAttempts(1);
                 $questionDialog = new QuestionHelper();
                 $pid = $questionDialog->ask($input, $output, $question);
             } catch (\InvalidArgumentException $e) {
                 // Invalid value, just stop here
                 throw new \CliTools\Exception\StopException(1);
             }
         }
     }
     if (!empty($pid)) {
         switch ($pid) {
             case 'all':
                 $command->addArgumentTemplate('-p %s', implode(',', $pidList));
                 break;
             default:
                 $command->addArgumentTemplate('-p %s', $pid);
                 break;
         }
         // Stats
         if ($input->getOption('c')) {
             $command->addArgument('-c');
         }
         // Relative time
         if ($input->getOption('r')) {
             $command->addArgument('-r');
         } else {
             $command->addArgument('-tt');
         }
         // System trace filter
         if ($input->getOption('e')) {
             $command->addArgumentTemplate('-e %s', $input->getOption('e'));
         }
         // Add grep
         if (!empty($grep)) {
             $grepCommand = new CommandBuilder('grep');
             $grepCommand->addArgument('--color=auto')->addArgument($grep);
             $command->addPipeCommand($grepCommand);
         }
         $command->executeInteractive();
     }
     return 0;
 }