File Coverage

blib/lib/Devel/Leak/Object.pm

Criterion	Covered	Total	%
statement	154	227	67.8
branch	64	114	56.1
condition	13	23	56.5
subroutine	17	17	100.0
pod	0	5	0.0
total	248	386	64.2

package Devel::Leak::Object;

use 5.005;
# We abuse refs a LOT
use strict qw{ vars subs };
use Carp         ();
use Scalar::Util ();

use vars qw{ $VERSION @ISA @EXPORT_OK };
use vars qw{ %OBJECT_COUNT %TRACKED %DESTROY_ORIGINAL %DESTROY_STUBBED %DESTROY_NEXT %IGNORE_CLASS %OBJECT_COUNT_CHECKPOINT };
BEGIN {
	$VERSION     = '1.01';

	# Set up exports
	require Exporter;
	@ISA         = qw(Exporter);
	@EXPORT_OK   = qw(track bless status checkpoint);

	# Set up state storage (primary for clarity)
	%OBJECT_COUNT     = ();
	%OBJECT_COUNT_CHECKPOINT     = ();
	%TRACKED          = ();
	%DESTROY_ORIGINAL = ();
	%DESTROY_STUBBED  = ();
	%DESTROY_NEXT     = ();
	%IGNORE_CLASS     = ();
}

sub import {
	my $class  = shift;
	my @import = ();
	while ( @_ ) {
		my $function = shift;
		unless ( $function =~ /^GLOBAL_(.*)$/ ) {
			push @import, $function;
			next;
		}
		my $global = $1;
		*{'CORE::GLOBAL::' . $global} = \&{$global};
	}
	return $class->SUPER::import(@import);
}

sub bless {
	my $reference = shift;
	my $class     = @_ ? shift : scalar caller;
	my $object    = CORE::bless($reference, $class);
	Devel::Leak::Object::track($object);
	return $object;
};

sub track {
	my $object = shift;
	my $class  = Scalar::Util::blessed($object);
	unless ( defined $class ) {
		Carp::carp("Devel::Leak::Object::track was passed a non-object");
	}
	return if (defined($IGNORE_CLASS{$class}));
	my $address = Scalar::Util::refaddr($object);
	if ( $TRACKED{$address} ) {
		$TRACKED{$address}->{class} ||= ''; # avoid warnings about uninitialised strings
		if ( $class eq $TRACKED{$address}->{class} ) {
			# Reblessing into the same class, ignore
			return $OBJECT_COUNT{$class};
		} else {
			# Reblessing into a different class
			$OBJECT_COUNT{$TRACKED{$address}->{class}}--;
		}
	}

	# Set or over-write the class name for the tracked object
	my ($package, $srcfile, $srcline, $subroutine) = caller(1);
	$package ||= '';
	$subroutine ||= '';
	#don't just tell us that we called it from our own new..
	if ($package eq $class) {
		my ($next_package, $next_srcfile, $next_srcline, $next_subroutine) = caller(2);
		if ($next_subroutine eq $class.'::new') {
			($package, $srcfile, $srcline, $subroutine) = ($next_package, $next_srcfile, $next_srcline, $next_subroutine);
		}
	}
	$TRACKED{$address} = { class => $class, file => $srcfile, line => $srcline, package=>$package, subroutine=>$subroutine };

	# If needed, initialise the new class
	unless ( $DESTROY_STUBBED{$class} ) {
		if ( exists ${$class.'::'}{DESTROY} and *{$class.'::DESTROY'}{CODE} ) {
			# Stash the pre-existing DESTROY function
			$DESTROY_ORIGINAL{$class} = \&{$class . '::DESTROY'};
		}
		$DESTROY_STUBBED{$class} = 1;
		eval <<"END_DESTROY";
package $class;\
no warnings;
sub DESTROY {
	my \$class   = Scalar::Util::blessed(\$_[0]);
	my \$address = Scalar::Util::refaddr(\$_[0]);
	unless ( defined \$class ) {
		Carp::carp("Unexpected error: First param to DESTROY is no an object");
		return;
	}
	unless ( defined \$class ) {
		die "Unexpected error: First param to DESTROY is no an object";
	}

	# Don't do anything unless tracking for the specific object is set
	my \$original = \$Devel::Leak::Object::TRACKED{\$address}->{class};
	if ( \$original ) {
		### TODO - We COULD add a check that $class eq
		#          \$Devel::Leak::Object::TRACKED{\$address}->{class}
		#          and then not decrement unless it is the same.
		#          However, in practice it should ALWAYS be the same if
		#          we already have \$Devel::Leak::Object::TRACKED{\$address}
		#          true still, and if for some reason this is wrong, we get
		#          a false positive in the leak counting.
		#          This additional check may be able to be added at a later
		#          date if it turns out to be needed.
		#          if ( \$class eq \$Devel::Leak::Object::TRACKED{\$address} ) { ... }
		if ( \$class ne \$original ) {
			warn "Object class '\$class' does not match original ".\$Devel::Leak::Object::TRACKED{\$address}->{class};
		}
		\$Devel::Leak::Object::OBJECT_COUNT{\$original}--;
		if ( \$Devel::Leak::Object::OBJECT_COUNT{\$original} < 0 ) {
			warn "Object count for ".\$Devel::Leak::Object::TRACKED{\$address}->{class}." negative (\$Devel::Leak::Object::OBJECT_COUNT{\$original})";
		}
		delete \$Devel::Leak::Object::TRACKED{\$address};

		# Hand of to the regular DESTROY method, or pass up to the SUPERclass if there isn't one
		if ( \$Devel::Leak::Object::DESTROY_ORIGINAL{\$original} ) {
			goto \&{\$Devel::Leak::Object::DESTROY_ORIGINAL{\$original}};
		}
	} else {
		\$original = \$class;
	}

	# If we don't have the DESTROY_NEXT for this class, populate it
	unless ( \$Devel::Leak::Object::DESTROY_NEXT{\$original} ) {
		Devel::Leak::Object::make_next(\$original);
	}
	my \$super = \$Devel::Leak::Object::DESTROY_NEXT{\$original}->{'$class'};
	unless (( defined \$super ) or (defined(\$Devel::Leak::Object::IGNORE_CLASS{\$class}))) {
		warn "Failed to find super-method for class \$class in package $class";
		\$Devel::Leak::Object::IGNORE_CLASS{\$class} = 1;
	}
	if ( \$super ) {
		goto \&{\$super.'::DESTROY'};
	}
	return;
}
END_DESTROY
		if ( $@ ) {
			die "Failed to generate DESTROY method for $class: $@";
		}

		# Pre-emptively populate the DESTROY_NEXT map
		unless ( $DESTROY_NEXT{$class} ) {
			make_next($class);
		}
	}

	$OBJECT_COUNT{$TRACKED{$address}->{class}}++;
}

sub make_next {
		my $class = shift;

		# Build the %DESTROY_NEXT entries to support DESTROY_stub
		$DESTROY_NEXT{$class} = {};
		my @stack = ( $class );
		my %seen  = ( UNIVERSAL => 1 );
		my @queue = ();
		while ( my $c = shift @stack ) {
			next if $seen{$c}++;

			# Does the class have it's own DESTROY method
			my $has_destroy = $DESTROY_STUBBED{$c}
				? !! exists $DESTROY_ORIGINAL{$c}
				: !! (exists ${"${c}::"}{DESTROY} and *{"${c}::DESTROY"}{CODE});
			if ( $has_destroy ) {
				# Everything in the queue has this class as it's next call
				while ( @queue ) {
					$DESTROY_NEXT{$class}->{shift(@queue)} = $c;
				}
			} else {
				# This class goes onto the queue
				push @queue, $c;
			}

			# Add the @ISA to the search stack.
			unshift @stack, @{"${c}::ISA"};
		}

		# Any else has no target to go to
		while ( @queue ) {
			$DESTROY_NEXT{$class}->{shift @queue} = '';
		}

		return 1;
}

sub checkpoint {
	my $first;
	for (sort keys %OBJECT_COUNT) {
		next unless $OBJECT_COUNT{$_}; # Don't list class with count zero
		$OBJECT_COUNT_CHECKPOINT{$_} ||= 0;
		next unless ($OBJECT_COUNT{$_} > $OBJECT_COUNT_CHECKPOINT{$_});

		print STDERR "checkpoint:\n" unless ($first++);;
		printf STDERR "\t%-40s %d\n", $_, $OBJECT_COUNT{$_}-$OBJECT_COUNT_CHECKPOINT{$_};

		$OBJECT_COUNT_CHECKPOINT{$_} = $OBJECT_COUNT{$_};
	}
}

sub status {
	print STDERR "Tracked objects by class:\n";
	for (sort keys %OBJECT_COUNT) {
		next if ($_ eq '');	#Don't know what these are..
		next unless $OBJECT_COUNT{$_}; # Don't list class with count zero
		printf STDERR "\t%-40s %d\n", $_, $OBJECT_COUNT{$_};
	}
	if($Devel::Leak::Object::TRACKSOURCELINES) {
		print STDERR "\nSources of leaks:\n";
		my %classes = ();
		foreach my $obj (values(%TRACKED)) {
			#TODO: no, I don't know why there are some undefined
			next unless defined($obj->{class});
			$classes{$obj->{class}} ||= {};
			my $line = $obj->{file}.' line: '.$obj->{line}; #.' ('.$obj->{package}.' -> '.$obj->{subroutine}.')';
			$classes{$obj->{class}}->{$line}++;
		}
		foreach my $class (sort keys(%classes)) {
			printf STDERR "%s\n", $class;
			my %lines = %{$classes{$class}};
			foreach my $line (sort keys(%lines)) {
	   			printf STDERR "%6d from %s\n", $lines{$line}, $line;
			}
		}
	}
}

END {
	status();
}

1;

__END__