]>
Commit | Line | Data |
---|---|---|
f4fe6fc4 DM |
1 | package PMG::RuleDB::MatchFilename; |
2 | ||
3 | use strict; | |
4 | use warnings; | |
f4fe6fc4 DM |
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 | ||
f4fe6fc4 DM |
26 | sub new { |
27 | my ($type, $fname, $ogroup) = @_; | |
28 | ||
29 | my $class = ref($type) || $type; | |
30 | ||
7a2cf7e6 | 31 | my $self = $class->SUPER::new($class->otype(), $ogroup); |
f4fe6fc4 DM |
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 | ||
9ef3f143 | 43 | defined($value) || die "undefined value: ERROR";; |
f4fe6fc4 DM |
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 | ||
9ef3f143 | 56 | defined($self->{ogroup}) || die "undefined ogroup: ERROR"; |
f4fe6fc4 DM |
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 | ||
682a1397 DC |
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 | ||
f4fe6fc4 DM |
142 | 1; |
143 | __END__ | |
144 | ||
145 | =head1 PMG::RuleDB::MatchFilename | |
146 | ||
147 | Match Header Filename |