#!/usr/bin/python

import subprocess
import re
from preupg.script_api import *


"""Preupgrade Assistant performs system upgradability assessment
and gathers information required for successful operating system upgrade.
Copyright (C) 2013 Red Hat Inc.
John Dennis <jdennis@redhat.com>

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 3 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>."""
check_applies_to (check_applies="freeradius")
check_rpm_to (check_rpm="python",check_bin="")
#END GENERATED SECTION

RADIUSD = '/usr/sbin/radiusd'

def get_int(string):
    try:
        value = int(string)
        return value
    except ValueError:
        match = re.search(r'\d+', string)
        if match:
            value = int(match.group(0))
            return value
        else:
            raise ValueError("Cannot find an integer in '%s'" % string)
        
def get_radiusd_version():
    args = [RADIUSD, '-v']
    p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = p.communicate()
    returncode = p.returncode
    if returncode != 0:
        raise subprocess.CalledProcessError(returncode=returncode,
                                            cmd=' '.join(args),
                                            output=stderr)

    match = re.search(r'FreeRADIUS\s+Version\s+(\d+)\.(\d+)\.(\d+)(\S*?),', stdout, re.MULTILINE)
    if match:
        major = match.group(1)
        minor = match.group(2)
        micro = match.group(3)
        extra = match.group(4)
    else:
        raise ValueError("Unable to parse FreeRADIUS version")
    
    return major, minor, micro, extra

def main():
    try:
        major, minor, micro, extra = get_radiusd_version()
    except Exception as e:
        log_error("Unable to query FreeRADIUS version: %s" % e)
        exit_error()

    log_info("Found FreeRADIUS version %s.%s.%s%s" % (
        major, minor, micro, extra))

    try:
        major = get_int(major)
        minor = get_int(minor)
        micro = get_int(micro)
    except Exception as e:
        log_error("Unable to determine numeric version values: %s" % e)
        exit_error()


    if major < 3:
        log_high_risk("The configuration of FreeRadius %d is not compatible with"
                      " version 3 in Red Hat Enterprise Linux 7. See the remediation"
                      " description."  % (major))
        exit_fail()

if __name__ == "__main__":
    main()
    exit_pass()
