예제 #1
0
 protected function checkVirusTotalForHash($hash)
 {
     try {
         $virusTotalFile = new File(config('virustotal.api_key'));
         $virusReport = $virusTotalFile->getReport($hash);
     } catch (\Exception $exception) {
         // Swallow any exceptions raised while querying the API
         return;
     }
     if (is_array($virusReport) && $virusReport['response_code'] === 1) {
         // Make sure we have scan stats
         if (isset($virusReport['total']) && isset($virusReport['positives'])) {
             $detectionRatio = $virusReport['positives'] / $virusReport['total'];
             if ($detectionRatio >= config('virustotal.detection_threshold')) {
                 throw new FileInfectedException();
             }
         }
     }
 }
예제 #2
0
파일: ScanFiles.php 프로젝트: BePsvPT/CCU
 /**
  * Execute the console command.
  *
  * @return mixed
  */
 public function handle()
 {
     // 確認是否有 api key
     if (null === $this->apiPublicKey) {
         $this->error('Invalid VirusTotal api public key.');
         return;
     }
     if (!$this->filesystem->isFile($file = $this->argument('file'))) {
         $this->error('File not exists.');
         return;
     }
     // 取得欲儲存的 Model
     if (null === ($model = $this->option('model'))) {
         $model = $this->ask('The Eloquent ORM Model');
     }
     // 取得欲儲存的欄位
     if (null === ($column = $this->option('column'))) {
         $column = $this->ask('The table\'s column to store the result');
     }
     // 取得欲儲存的欄位
     if (null === ($index = $this->option('index'))) {
         $index = $this->ask('The primary key\'s value to specific row');
     }
     // 檢查 Model 是否存在
     if (!class_exists($model)) {
         $this->error('Model not exists.');
         return;
     }
     $model = (new $model())->find($index);
     // 檢查該比資料是否存在
     if (null === $model) {
         $this->error('Model not exists');
         return;
     }
     // 檢查欄位是否存在
     if (!Schema::hasColumn($model->getTableName(), $column)) {
         $this->error('Column not exists.');
         return;
     }
     // 檢查是否有替代檔名
     if (null !== ($fakeName = $this->option('fakeName')) && strlen($fakeName) > 0) {
         $fakePath = temp_path($fakeName);
         $this->filesystem->copy($file, $fakePath);
         $file = $fakePath;
     }
     $virusTotal = new File($this->apiPublicKey);
     $report = $virusTotal->scan($file);
     $model->{$column} = $report['permalink'];
     $model->save();
     if (isset($fakePath)) {
         $this->filesystem->delete($fakePath);
     }
     $this->info('File scan successfully!');
 }