PMG/RuleDB/MatchFilename.pm

   1 package PMG::RuleDB::MatchFilename;
   2
   3 use strict;
   4 use warnings;
   5 use DBI;
   6 use Digest::SHA;
   7 use MIME::Words;
   8
   9 use PMG::Utils;
  10 use PMG::RuleDB::Object;
  11
  12 use base qw(PMG::RuleDB::Object);
  13
  14 sub otype {
  15     return 3004;
  16 }
  17
  18 sub oclass {
  19     return 'what';
  20 }
  21
  22 sub otype_text {
  23     return 'Match Filename';
  24 }
  25
  26 sub new {
  27     my ($type, $fname, $ogroup) = @_;
  28
  29     my $class = ref($type) || $type;
  30
  31     my $self = $class->SUPER::new($class->otype(), $ogroup);
  32
  33     $self->{fname} = $fname;
  34
  35     return $self;
  36 }
  37
  38 sub load_attr {
  39     my ($type, $ruledb, $id, $ogroup, $value) = @_;
  40
  41     my $class = ref($type) || $type;
  42
  43     defined($value) || die "undefined value: ERROR";;
  44
  45     my $obj = $class->new($value, $ogroup);
  46     $obj->{id} = $id;
  47
  48     $obj->{digest} = Digest::SHA::sha1_hex($id, $value, $ogroup);
  49
  50     return $obj;
  51 }
  52
  53 sub save {
  54     my ($self, $ruledb) = @_;
  55
  56     defined($self->{ogroup}) || die "undefined ogroup: ERROR";
  57
  58     my $new_value = $self->{fname};
  59     $new_value =~ s/\\/\\\\/g;
  60
  61     if (defined($self->{id})) {
  62         # update
  63
  64         $ruledb->{dbh}->do("UPDATE Object SET Value = ? WHERE ID = ?",
  65                            undef, $new_value, $self->{id});
  66
  67     } else {
  68         # insert
  69
  70         my $sth = $ruledb->{dbh}->prepare(
  71             "INSERT INTO Object (Objectgroup_ID, ObjectType, Value) " .
  72             "VALUES (?, ?, ?);");
  73
  74         $sth->execute($self->ogroup, $self->otype, $new_value);
  75
  76         $self->{id} = PMG::Utils::lastid($ruledb->{dbh}, 'object_id_seq');
  77     }
  78
  79     return $self->{id};
  80 }
  81
  82 sub parse_entity {
  83     my ($self, $entity) = @_;
  84
  85     my $res;
  86
  87     if (my $id = $entity->head->mime_attr('x-proxmox-tmp-aid')) {
  88         chomp $id;
  89
  90         if (my $value = PMG::Utils::extract_filename($entity->head)) {
  91             if ($value =~ m|^$self->{fname}$|i) {
  92                 push @$res, $id;
  93             }
  94         }
  95     }
  96
  97     foreach my $part ($entity->parts)  {
  98         if (my $match = $self->parse_entity ($part)) {
  99             push @$res, @$match;
 100         }
 101     }
 102
 103     return $res;
 104 }
 105
 106 sub what_match {
 107     my ($self, $queue, $entity, $msginfo) = @_;
 108
 109     return $self->parse_entity($entity);
 110 }
 111
 112 sub short_desc {
 113     my $self = shift;
 114
 115     return "filename=$self->{fname}";
 116 }
 117
 118 sub properties {
 119     my ($class) = @_;
 120
 121     return {
 122         filename => {
 123             description => "Filename filter",
 124             type => 'string',
 125             maxLength => 1024,
 126         },
 127     };
 128 }
 129
 130 sub get {
 131     my ($self) = @_;
 132
 133     return { filename => $self->{fname} };
 134 }
 135
 136 sub update {
 137     my ($self, $param) = @_;
 138
 139     $self->{fname} = $param->{filename};
 140 }
 141
 142 1;
 143 __END__
 144
 145 =head1 PMG::RuleDB::MatchFilename
 146
 147 Match Header Filename