M: Added sanity check for $action argument of setHandler method.

2012-09-02 19:23:32 +02:00
parent 6b24a7950b
commit 1c890bec84
1 changed files with 4 additions and 0 deletions
--- a/lib/File/DirWalk.pm
+++ b/lib/File/DirWalk.pm
@@ -49,6 +49,10 @@ sub new {
 sub setHandler {
 	my ($self,$action,$func) = @_;
    
+	if ($action !~ /onBeginWalk|onLink|onFile|onDirEnter|onDirLeave/) {
+    	croak("Invalid action argument: $action");
+	}
+	
 	if (ref($func) ne 'CODE') {
    	croak("Second argument must be CODE reference.");
 	}