'''apport package hook for apparmor

(c) 2009-2013 Canonical Ltd.
Author: Steve Beattie <[email protected]>
Jamie Strandboge <[email protected]>
License: GPLv2
'''

from apport.hookutils import *
import os
import re

def recent_kernlog(pattern):
'''Extract recent messages from kern.log or message which match a regex.
pattern should be a "re" object. '''
lines = ''
if os.path.exists('/var/log/kern.log'):
file = '/var/log/kern.log'
elif os.path.exists('/var/log/messages'):
file = '/var/log/messages'
else:
return lines

for line in open(file):
if pattern.search(line):
lines += line
return lines

def recent_syslog(pattern):
'''Extract recent messages from syslog which match a regex.
pattern should be a "re" object. '''
lines = ''
if os.path.exists('/var/log/syslog'):
file = '/var/log/syslog'
else:
return lines

for line in open(file):
if pattern.search(line):
lines += line
return lines

def add_info(report):
attach_file(report, '/proc/version_signature', 'ProcVersionSignature')
attach_file(report, '/proc/cmdline', 'ProcKernelCmdline')

sec_re = re.compile('audit\(|apparmor|selinux|security', re.IGNORECASE)
report['KernLog'] = recent_kernlog(sec_re)
# DBus messages are reported to syslog
dbus_sec_re = re.compile('dbus.* apparmor', re.IGNORECASE)
report['Syslog'] = recent_syslog(dbus_sec_re)

packages=['apparmor', 'apparmor-utils', 'libapparmor1',
'libapparmor-dev', 'libapparmor-perl', 'apparmor-utils',
'apparmor-docs', 'apparmor-profiles', 'libapache2-mod-apparmor',
'libpam-apparmor', 'auditd', 'libaudit0']

versions = ''
for package in packages:
try:
version = packaging.get_version(package)
except ValueError:
version = 'N/A'
if version is None:
version = 'N/A'
versions += '%s %s\n' % (package, version)
report['ApparmorPackages'] = versions

# These need to be run as root
report['ApparmorStatusOutput'] = root_command_output(['/usr/sbin/apparmor_status'])
report['PstreeP'] = command_output(['/usr/bin/pstree', '-p'])
attach_file_if_exists(report, '/var/log/audit/audit.log', 'audit.log')