fixes the duplicate key bug when finding the same fil multiple times

Author: ata-no-one

Infrastructure/Database/FindingsQuery.php

@@ -29,6 +29,8 @@ public function create(): void {
             detection VARCHAR(128) NOT NULL,
             sha256 VARCHAR(64) NOT NULL,
             request_id VARCHAR(256) NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
             UNIQUE KEY file_path (file_path)
         )' . $charset_collate . ';';
 
@@ -68,14 +70,38 @@ public function table_exists(): bool {
         return false;
     }
 
+    private function exits( string $file_path ): bool {
+        global $wpdb;
+
+        if (!$this->table_exists()) {
+            return false;
+        }
+        return $wpdb->get_var(
+            $wpdb->prepare('SELECT COUNT(*) FROM %i WHERE file_path = %s', $this->get_table_name(), $file_path)
+        ) > 0;
+    }
+
     public function add( DetectedFile $detected_file ): void {
         global $wpdb;
+        assert($wpdb instanceof wpdb);
 
         if (! $this->table_exists()) {
             return;
         }
 
         try {
+            if ($this->exits($detected_file->path)) {
+                $wpdb->update(
+                    $this->get_table_name(),
+                    array( 
+                        'detection' => $detected_file->detection,
+                        'sha256' => $detected_file->sha256,
+                        'request_id' => $detected_file->request_id
+                    ),
+                    array( 'file_path' => $detected_file->path )
+                );
+                return; 
+            }
             $wpdb->insert(
                 $this->get_table_name(),
                 array( 
@@ -121,7 +147,7 @@ public function get_all(): array {
             return array();
         }
         return $wpdb->get_results(
-            $wpdb->prepare('SELECT file_path, detection, sha256, request_id FROM %i', $this->get_table_name()),
+            $wpdb->prepare('SELECT file_path, detection, sha256, request_id, updated_at FROM %i', $this->get_table_name()),
             ARRAY_A
         );
     }

PluginPage/Findings/FindingsMenuPage.php

@@ -135,6 +135,9 @@ public function findings_list(): void {
 							<th scope="col" id="title_file" class="manage-column column-title column-primary">
 								File
 							</th>
+							<th scope="col" id="title_file" class="manage-column column-title column-primary">
+								Last seen
+							</th>
 							<th scope="col" id="title_detection" class="manage-column column-title column-primary">
 								Detection
 							</th>
@@ -168,6 +171,11 @@ public function findings_list(): void {
 										echo esc_html($finding['file_path']);
 										?>
 									</td>
+									<td>
+										<?php
+										echo esc_html($finding['updated_at']);
+										?>
+									</td>
 									<td>
 										<?php
 										echo esc_html($finding['detection']);

PluginPage/FullScan/FullScanMenuPage.php

@@ -246,7 +246,7 @@ public function full_scan(): void {
 				if ($file_path->isDir()) {
 					continue;
 				}
-				// For testing purposes, we only scan files with eicar in the name
+				// // For testing purposes, we only scan files with eicar in the name
 				// if (str_contains($file_path->getPathname(), "eicar") === false) {
 				// 	continue;
 				// }

Readme.txt

@@ -58,6 +58,7 @@ While the released code is hosted on the WordPress svn, we develop the plugin on
 
 = 2.1.0 =
 * bugfix: [full scan runs in loop](https://github.com/GDATASoftwareAG/wordpress-gdata-antivirus/issues/37)
+* bugifx: fails on duplicate key when detecting the same file twice
 
 = 2.0.9 =
 * bugfix: reconnect on long running scans