diff --git a/lib/File/DirWalk.pm b/lib/File/DirWalk.pm index f872a64..48b39d9 100644 --- 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."); }