Changeset 386

Timestamp:

05/29/07 13:04:51 (1 year ago)

Author:

kindlund

Message:

Updated registry code to include intermediate cleanup of checkpoint operations (when temporary files are created).

Files:

• honeyclient/branches/exp/kindlund-filesystem/lib/HoneyClient/Agent/Integrity/Registry.pm (modified) (2 diffs)

honeyclient/branches/exp/kindlund-filesystem/lib/HoneyClient/Agent/Integrity/Registry.pm

@@ -442 +442 @@
-            }
-            delete($self->{_checkpoint_parsers}->{$hive});
+        }
+    }
+}
+
+# Helper function, designed to cleanup temporary files created
+# by specified parsers.
+#
+# Inputs: HoneyClient::Agent::Integrity::Registry object,
+#         the hashtable collection of specified parsers
+#
+# Outputs: None.
+sub _cleanupParsers {
+    # Extract arguments.
+    my ($self, $parser_collection) = @_;
+
+    # Delete any temporary files created.
+    my $parser = undef;
+    my $fname = undef;
+    foreach my $hive (@{$self->{hives_to_check}}) {
+        $parser = $parser_collection->{$hive};
+        if (defined($parser)) {
+            $fname = $parser->getFilename();
+            if (defined($fname) && (-f $fname)) {
+                # XXX: Change this to debug, eventually.
+                $LOG->info("Deleting temporary '" . $hive . "' hive data in '" .
+                            $fname . "'.");
+                if (!unlink($fname)) {
+                    $LOG->fatal("Error: Unable to unlink '" . $hive . "' hive data in '" . $fname ."'.");
+                    Carp::croak("Error: Unable to unlink '" . $hive . "' hive data in '" . $fname ."'.");
+                }
+            }
+            delete($parser_collection->{$hive});
-        }
-    }
@@ -1464 +1496 @@
-            $LOG->debug("Finished checking '" . $hive . "' hive.");
-        }
+
+        # Cleanup temporary files created by checkpointing.
+        $self->_cleanupParsers($self->{_checkpoint_parsers});
-    }
-