ObjectCustomFieldValue_Overlay.pm 13.6 KB
Newer Older
1
# BEGIN BPS TAGGED BLOCK {{{
2
# 
Jesse Vincent's avatar
Jesse Vincent committed
3
# COPYRIGHT:
Jesse Vincent's avatar
Jesse Vincent committed
4
# 
Thomas Sibley's avatar
Thomas Sibley committed
5
# This software is Copyright (c) 1996-2010 Best Practical Solutions, LLC
6
#                                          <jesse@bestpractical.com>
7
# 
Jesse Vincent's avatar
Jesse Vincent committed
8
9
10
11
# (Except where explicitly superseded by other copyright notices)
# 
# 
# LICENSE:
12
13
14
15
16
17
18
19
20
21
22
# 
# This work is made available to you under the terms of Version 2 of
# the GNU General Public License. A copy of that license should have
# been provided with this software, but in any event can be snarfed
# from www.gnu.org.
# 
# This work is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
# 
Jesse Vincent's avatar
Jesse Vincent committed
23
24
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
Jesse Vincent's avatar
Jesse Vincent committed
25
26
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301 or visit their web page on the internet at
Ruslan Zakirov's avatar
Ruslan Zakirov committed
27
# http://www.gnu.org/licenses/old-licenses/gpl-2.0.html.
Jesse Vincent's avatar
Jesse Vincent committed
28
29
30
31
32
33
34
35
36
# 
# 
# CONTRIBUTION SUBMISSION POLICY:
# 
# (The following paragraph is not intended to limit the rights granted
# to you to modify and distribute this software under the terms of
# the GNU General Public License and is only of importance to you if
# you choose to contribute your changes and enhancements to the
# community by submitting them to Best Practical Solutions, LLC.)
37
# 
Jesse Vincent's avatar
Jesse Vincent committed
38
39
40
41
42
43
44
45
# By intentionally submitting any modifications, corrections or
# derivatives to this work, or any other work intended for use with
# Request Tracker, to Best Practical Solutions, LLC, you confirm that
# you are the copyright holder for those contributions and you grant
# Best Practical Solutions,  LLC a nonexclusive, worldwide, irrevocable,
# royalty-free, perpetual, license to use, copy, create derivative
# works based on those contributions, and sublicense and distribute
# those contributions and any derivatives thereof.
46
# 
47
# END BPS TAGGED BLOCK }}}
48

49
50
package RT::ObjectCustomFieldValue;

51
use strict;
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
52
use warnings;
53
use RT::Interface::Web;
54
use Regexp::Common qw(RE_net_IPv4);
sunnavy's avatar
sunnavy committed
55
use Regexp::IPv6 qw($IPv6_re);
56
57
58
use Regexp::Common::net::CIDR;
require Net::CIDR;

59

60
61
no warnings qw(redefine);

62
63
64
sub Create {
    my $self = shift;
    my %args = (
65
66
67
68
69
70
71
72
73
74
75
        CustomField     => 0,
        ObjectType      => '',
        ObjectId        => 0,
        Disabled        => 0,
        Content         => '',
        LargeContent    => undef,
        ContentType     => '',
        ContentEncoding => '',
        @_,
    );

76
77
78

    my $cf_as_sys = RT::CustomField->new(RT->SystemUser);
    $cf_as_sys->Load($args{'CustomField'});
sunnavy's avatar
sunnavy committed
79
80

    if($cf_as_sys->Type eq 'IPAddress') {
sunnavy's avatar
sunnavy committed
81
82
83
84
        if ( $args{'Content'} ) {
            $args{'Content'} = $self->ParseIP( $args{'Content'} );
        }

sunnavy's avatar
sunnavy committed
85
86
87
88
89
90
91
92
93
        unless ( defined $args{'Content'} ) {
            return
              wantarray
              ? ( 0, $self->loc("Content can't be empty for IPAddress") )
              : 0;
        }
    }

    if($cf_as_sys->Type eq 'IPAddressRange') {
94
95
        if ($args{'Content'}) {
            ($args{'Content'}, $args{'LargeContent'}) = $self->ParseIPRange( $args{'Content'} );
sunnavy's avatar
sunnavy committed
96
        }
97

sunnavy's avatar
sunnavy committed
98
99
100
        unless ( defined $args{'Content'} ) {
            return
              wantarray
sunnavy's avatar
sunnavy committed
101
              ? ( 0, $self->loc("Content can't be empty for IPAddressRange") )
sunnavy's avatar
sunnavy committed
102
              : 0;
103
104
105
        }
    }

106
    if ( defined $args{'Content'} && length( Encode::encode_utf8($args{'Content'}) ) > 255 ) {
107
        if ( defined $args{'LargeContent'} && length $args{'LargeContent'} ) {
108
            $RT::Logger->error("Content is longer than 255 bytes and LargeContent specified");
109
110
111
112
113
114
        }
        else {
            $args{'LargeContent'} = $args{'Content'};
            $args{'Content'} = '';
            $args{'ContentType'} ||= 'text/plain';
        }
115
116
117
    }

    ( $args{'ContentEncoding'}, $args{'LargeContent'} ) =
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
118
119
120
121
122
123
124
125
126
127
128
129
130
        $self->_EncodeLOB( $args{'LargeContent'}, $args{'ContentType'} )
            if defined $args{'LargeContent'};

    return $self->SUPER::Create(
        CustomField     => $args{'CustomField'},
        ObjectType      => $args{'ObjectType'},
        ObjectId        => $args{'ObjectId'},
        Disabled        => $args{'Disabled'},
        Content         => $args{'Content'},
        LargeContent    => $args{'LargeContent'},
        ContentType     => $args{'ContentType'},
        ContentEncoding => $args{'ContentEncoding'},
    );
131
132
133
134
135
}


sub LargeContent {
    my $self = shift;
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
136
137
138
139
140
    return $self->_DecodeLOB(
        $self->ContentType,
        $self->ContentEncoding,
        $self->_Value( 'LargeContent', decode_utf8 => 0 )
    );
141
142
}

143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168

=head2 LoadByCols

=cut

sub LoadByCols {
    my $self = shift;
    my %args = (@_);
    my $cf;
    if ( $args{CustomField} ) {
        $cf = RT::CustomField->new( $self->CurrentUser );
        $cf->Load( $args{CustomField} );
        if ( $cf->Type eq 'IPAddressRange' ) {

            my ( $sIP, $eIP ) = $cf->ParseIPRange( $args{'Content'} );
            if ( $sIP && $eIP ) {
                $self->SUPER::LoadByCols( %args,
                                          Content      => $sIP,
                                          LargeContent => $eIP
                                        );
            }
        }
    }
    return $self->SUPER::LoadByCols(%args);
}

169
170
171
172
173
174
175
176
177
=head2 LoadByTicketContentAndCustomField { Ticket => TICKET, CustomField => CUSTOMFIELD, Content => CONTENT }

Loads a custom field value by Ticket, Content and which CustomField it's tied to

=cut


sub LoadByTicketContentAndCustomField {
    my $self = shift;
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
178
179
180
181
182
183
    my %args = (
        Ticket => undef,
        CustomField => undef,
        Content => undef,
        @_
    );
184

Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
185
186
187
188
189
190
191
    return $self->LoadByCols(
        Content => $args{'Content'},
        CustomField => $args{'CustomField'},
        ObjectType => 'RT::Ticket',
        ObjectId => $args{'Ticket'},
        Disabled => 0
    );
192
193
194
195
}

sub LoadByObjectContentAndCustomField {
    my $self = shift;
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
196
197
198
199
200
201
    my %args = (
        Object => undef,
        CustomField => undef,
        Content => undef,
        @_
    );
202
203
204

    my $obj = $args{'Object'} or return;

Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
205
206
207
208
209
210
211
    return $self->LoadByCols(
        Content => $args{'Content'},
        CustomField => $args{'CustomField'},
        ObjectType => ref($obj),
        ObjectId => $obj->Id,
        Disabled => 0
    );
212
213
}

214
215
216
217
218
219
220
221
222
223
224
225
226
227
=head2 CustomFieldObj

Returns the CustomField Object which has the id returned by CustomField

=cut

sub CustomFieldObj {
    my $self = shift;
    my $CustomField = RT::CustomField->new( $self->CurrentUser );
    $CustomField->SetContextObject( $self->Object );
    $CustomField->Load( $self->__Value('CustomField') );
    return $CustomField;
}

228
229
230
231
232
233
234
235

=head2 Content

Return this custom field's content. If there's no "regular"
content, try "LargeContent"

=cut

236
237
238
my $re_ip_sunit = qr/[0-1][0-9][0-9]|2[0-4][0-9]|25[0-5]/;
my $re_ip_serialized = qr/$re_ip_sunit(?:\.$re_ip_sunit){3}/;

239
240
sub Content {
    my $self = shift;
241

242
    my $content = $self->_Value('Content');
sunnavy's avatar
sunnavy committed
243
244
245
246
247
248
249
    if (   $self->CustomFieldObj->Type eq 'IPAddress'
        || $self->CustomFieldObj->Type eq 'IPAddressRange' )
    {

        if ( $content =~ /^\s*($re_ip_serialized)\s*$/o ) {
            $content = sprintf "%d.%d.%d.%d", split /\./, $1;
        }
250

sunnavy's avatar
sunnavy committed
251
        return $content if $self->CustomFieldObj->Type eq 'IPAddress';
252

sunnavy's avatar
sunnavy committed
253
        my $large_content = $self->LargeContent;
sunnavy's avatar
sunnavy committed
254
255
256
        if ( $large_content =~ /^\s*($re_ip_serialized)\s*$/o ) {
            my $eIP = sprintf "%d.%d.%d.%d", split /\./, $1;
            if ( $content eq $eIP ) {
257
258
                return $content;
            }
sunnavy's avatar
sunnavy committed
259
260
261
262
            else {
                return $content . "-" . $eIP;
            }
        }
sunnavy's avatar
sunnavy committed
263
264
265
266
267
268
269
270
271
        elsif ( $large_content =~ /^\s*($IPv6_re)\s*$/o ) {
            my $eIP = $1;
            if ( $content eq $eIP ) {
                return $content;
            }
            else {
                return $content . "-" . $eIP;
            }
        }
sunnavy's avatar
sunnavy committed
272
273
274
        else {
            return $content;
        }
275
    }
sunnavy's avatar
sunnavy committed
276

277
    if ( !(defined $content && length $content) && $self->ContentType && $self->ContentType eq 'text/plain' ) {
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
278
        return $self->LargeContent;
279
280
    } else {
        return $content;
281
282
283
    }
}

Jesse Vincent's avatar
Jesse Vincent committed
284
285
286
287
288
289
=head2 Object

Returns the object this value applies to

=cut

290
291
sub Object {
    my $self  = shift;
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
292
293
294
    my $Object = $self->__Value('ObjectType')->new( $self->CurrentUser );
    $Object->LoadById( $self->__Value('ObjectId') );
    return $Object;
295
296
}

Jesse Vincent's avatar
Jesse Vincent committed
297
298
299
300
301
302
303
304

=head2 Delete

Disable this value. Used to remove "current" values from records while leaving them in the history.

=cut


Audrey Tang's avatar
Audrey Tang committed
305
306
sub Delete {
    my $self = shift;
Ruslan Zakirov's avatar
minor    
Ruslan Zakirov committed
307
    return $self->SetDisabled(1);
Audrey Tang's avatar
Audrey Tang committed
308
309
}

Jesse Vincent's avatar
Jesse Vincent committed
310
311
312
=head2 _FillInTemplateURL URL

Takes a URL containing placeholders and returns the URL as filled in for this 
313
ObjectCustomFieldValue. The values for the placeholders will be URI-escaped.
Jesse Vincent's avatar
Jesse Vincent committed
314
315
316
317
318
319
320
321
322
323
324
325
326

Available placeholders:

=over

=item __id__

The id of the object in question.

=item __CustomField__

The value of this custom field for the object in question.

327
328
329
330
=item __WebDomain__, __WebPort__, __WebPath__, __WebBaseURL__ and __WebURL__

The value of the config option.

Jesse Vincent's avatar
Jesse Vincent committed
331
332
333
334
=back

=cut

335
336
337
338
339
340
341
342
343
344
345
{
my %placeholders = (
    id          => { value => sub { $_[0]->ObjectId }, escape => 1 },
    CustomField => { value => sub { $_[0]->Content }, escape => 1 },
    WebDomain   => { value => sub { RT->Config->Get('WebDomain') } },
    WebPort     => { value => sub { RT->Config->Get('WebPort') } },
    WebPath     => { value => sub { RT->Config->Get('WebPath') } },
    WebBaseURL  => { value => sub { RT->Config->Get('WebBaseURL') } },
    WebURL      => { value => sub { RT->Config->Get('WebURL') } },
);

Jesse Vincent's avatar
Jesse Vincent committed
346
347
348
349
sub _FillInTemplateURL {
    my $self = shift;
    my $url = shift;

350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
    return undef unless defined $url && length $url;

    # special case, whole value should be an URL
    if ( $url =~ /^__CustomField__/ ) {
        my $value = $self->Content;
        # protect from javascript: URLs
        if ( $value =~ /^\s*javascript:/i ) {
            my $object = $self->Object;
            $RT::Logger->error(
                "Dangerouse value with JavaScript in custom field '". $self->CustomFieldObj->Name ."'"
                ." on ". ref($object) ." #". $object->id
            );
            return undef;
        }
        $url =~ s/^__CustomField__/$value/;
    }

367
    # default value, uri-escape
368
    for my $key (keys %placeholders) {
369
370
371
372
373
374
        $url =~ s{__${key}__}{
            my $value = $placeholders{$key}{'value'}->( $self );
            $value = '' if !defined($value);
            RT::Interface::Web::EscapeURI(\$value) if $placeholders{$key}{'escape'};
            $value
        }gxe;
375
    }
Jesse Vincent's avatar
Jesse Vincent committed
376
377

    return $url;
378
} }
Jesse Vincent's avatar
Jesse Vincent committed
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408


=head2 ValueLinkURL

Returns a filled in URL template for this ObjectCustomFieldValue, suitable for 
constructing a hyperlink in RT's webui. Returns undef if this custom field doesn't have
a LinkValueTo

=cut

sub LinkValueTo {
    my $self = shift;
    return $self->_FillInTemplateURL($self->CustomFieldObj->LinkValueTo);
}



=head2 ValueIncludeURL

Returns a filled in URL template for this ObjectCustomFieldValue, suitable for 
constructing a hyperlink in RT's webui. Returns undef if this custom field doesn't have
a IncludeContentForValue

=cut

sub IncludeContentForValue {
    my $self = shift;
    return $self->_FillInTemplateURL($self->CustomFieldObj->IncludeContentForValue);
}

409
410
411

sub ParseIPRange {
    my $self = shift;
sunnavy's avatar
sunnavy committed
412
413
414
415
    my $value = shift or return;
    $value = lc $value;
    $value =~ s!^\s+!!;
    $value =~ s!\s+$!!;
416
    
sunnavy's avatar
sunnavy committed
417
    if ( $value =~ /^$RE{net}{CIDR}{IPv4}{-keep}$/go ) {
418
        my $cidr = join( '.', map $_||0, (split /\./, $1)[0..3] ) ."/$2";
sunnavy's avatar
sunnavy committed
419
420
421
422
        $value = (Net::CIDR::cidr2range( $cidr ))[0] || $value;
    }
    elsif ( $value =~ /^$IPv6_re(?:\/\d+)?$/o ) {
        $value = (Net::CIDR::cidr2range( $value ))[0] || $value;
423
424
425
    }
    
    my ($sIP, $eIP);
sunnavy's avatar
sunnavy committed
426
    if ( $value =~ /^($RE{net}{IPv4})$/o ) {
427
428
        $sIP = $eIP = sprintf "%03d.%03d.%03d.%03d", split /\./, $1;
    }
sunnavy's avatar
sunnavy committed
429
    elsif ( $value =~ /^($RE{net}{IPv4})-($RE{net}{IPv4})$/o ) {
430
431
432
        $sIP = sprintf "%03d.%03d.%03d.%03d", split /\./, $1;
        $eIP = sprintf "%03d.%03d.%03d.%03d", split /\./, $2;
    }
sunnavy's avatar
sunnavy committed
433
434
435
436
437
438
439
440
441
    elsif ( $value =~ /^($IPv6_re)$/o ) {
        $sIP = $self->ParseIP( $1 );
        $eIP = $sIP;
    }
    elsif ( $value =~ /^($IPv6_re)-($IPv6_re)$/o ) {
        ($sIP, $eIP) = ( $1, $2 );
        $sIP = $self->ParseIP( $sIP );
        $eIP = $self->ParseIP( $eIP );
    }
442
    else {
sunnavy's avatar
sunnavy committed
443
        return;
444
    }
sunnavy's avatar
sunnavy committed
445

446
447
448
449
450
    ($sIP, $eIP) = ($eIP, $sIP) if $sIP gt $eIP;
    
    return $sIP, $eIP;
}

sunnavy's avatar
sunnavy committed
451
452
453
454
455
456
457
458
459
460
sub ParseIP {
    my $self = shift;
    my $value = shift or return;
    $value = lc $value;
    $value =~ s!^\s+!!;
    $value =~ s!\s+$!!;

    if ( $value =~ /^($RE{net}{IPv4})$/o ) {
        return sprintf "%03d.%03d.%03d.%03d", split /\./, $1;
    }
sunnavy's avatar
sunnavy committed
461
    elsif ( $value =~ /^$IPv6_re$/o ) {
sunnavy's avatar
sunnavy committed
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498

        # up_fields are before '::'
        # low_fields are after '::' but without v4
        # v4_fields are the v4
        my ( @up_fields, @low_fields, @v4_fields );
        my $v6;
        if ( $value =~ /(.*:)(\d+\..*)/ ) {
            ( $v6, my $v4 ) = ( $1, $2 );
            chop $v6 unless $v6 =~ /::$/;
            while ( $v4 =~ /(\d+)\.(\d+)/g ) {
                push @v4_fields, sprintf '%.2x%.2x', $1, $2;
            }
        }
        else {
            $v6 = $value;
        }

        my ( $up, $low );
        if ( $v6 =~ /::/ ) {
            ( $up, $low ) = split /::/, $v6;
        }
        else {
            $up = $v6;
        }

        @up_fields = split /:/, $up;
        @low_fields = split /:/, $low if $low;

        my @zero_fields =
          ('0000') x ( 8 - @v4_fields - @up_fields - @low_fields );
        my @fields = ( @up_fields, @zero_fields, @low_fields, @v4_fields );

        return join ':', map { sprintf "%.4x", hex "0x$_" } @fields;
    }
    return;
}

499
1;