#!/usr/bin/python -tt

# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program 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 Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
# Copyright 2009 Red Hat Inc
# written by Seth Vidal

# look through list of running apps
# report any app which was updated after it was started
# (and therefore needs to be restarted)


# for each /proc/number-dir 
# get stat of create time on that dir
# open up smaps and search for all lines with 'fd:' in them
# take filename
# search for the package owning that file
# make a list of installtimes of all pkgs of the files the program has open
# sort the list
# if the dir create time is < the largest time on the installtimes list
# then output the pid and process cmdline as needing to be restarted


import sys
import os
import yum
import glob
import stat

def return_running_pids():
    mypid = os.getpid()
    pids = []
    for fn in glob.glob('/proc/[0123456789]*'):
        if mypid == os.path.basename(fn):
            continue
        else:
            pids.append(os.path.basename(fn))
    return pids

def get_open_files(pid):
    files = []
    smaps = '/proc/%s/smaps' % pid
    try:
        maps = open(smaps, 'r')
    except (IOError, OSError), e:
        print "Could not open %s" % smaps
        return files

    for line in maps.readlines():
        if line.find('fd:') == -1:
            continue
        line = line.replace('\n', '')
        slash = line.find('/')
        filename = line[slash:]
        filename = filename.replace('(deleted)', '') #only mildly retarded
        filename = filename.strip()
        if filename not in files:
            files.append(filename)
    return files

def main(args):
    
    my = yum.YumBase()
    if hasattr(my, 'setCacheDir'):
        my.setCacheDir()
    my.conf.cache = True

    needing_restart = []

    for pid in return_running_pids():
        try:
            pid_start = os.stat('/proc/' + pid)[stat.ST_CTIME]
        except OSError, e:
            continue
        found_match = False
        for fn in get_open_files(pid):
            if found_match:
                break
            
            for pkg in my.rpmdb.searchFiles(fn):
                if float(pkg.installtime) > float(pid_start):
                    needing_restart.append(pid)
                    found_match = True

    for pid in needing_restart:
        try:
            cmdline = open('/proc/' +pid+ '/cmdline', 'r').read()
        except OSError, e:
            print "Couldn't access process information for %s: %s" % (pid, str(e))
            continue
        print '%s : %s' % (pid, cmdline)
        
    return 0
    
if __name__ == "__main__":
    sys.exit(main(sys.argv))