#!/usr/bin/python
# -*- encoding: utf-8; py-indent-offset: 4 -*-
# +------------------------------------------------------------------+
# |             ____ _               _        __  __ _  __           |
# |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
# |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
# |           | |___| | | |  __/ (__|   <    | |  | | . \            |
# |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
# |                                                                  |
# | Copyright Mathias Kettner 2014             mk@mathias-kettner.de |
# +------------------------------------------------------------------+
#
# This file is part of Check_MK.
# The official homepage is at http://mathias-kettner.de/check_mk.
#
# check_mk 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 in version 2.  check_mk is  distributed
# in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
# out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
# PARTICULAR PURPOSE. See the  GNU General Public License for more de-
# tails. You should have  received  a copy of the  GNU  General Public
# License along with GNU Make; see the file  COPYING.  If  not,  write
# to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
# Boston, MA 02110-1301 USA.


def check_wmic_process(item, params, info):
    name, memwarn, memcrit, pagewarn, pagecrit, cpuwarn, cpucrit = params
    count, mem, page, userc, kernelc = 0, 0, 0, 0, 0
    cpucores = 1
    if len(info) == 0:
        return (3, "No output from agent in section wmic_process")
    legend = info[0]
    for line in info[1:]:
        psinfo = dict(zip(legend, line))
        if psinfo.get("Name") is None:
            continue
        elif "ThreadCount" in legend and psinfo["Name"].lower() == "system idle process":
            cpucores = int(psinfo["ThreadCount"])
        elif psinfo["Name"].lower() == name.lower():
            count   += 1
            mem     += int(psinfo["WorkingSetSize"])
            page    += int(psinfo["PageFileUsage"])
            userc   += int(psinfo["UserModeTime"])
            kernelc += int(psinfo["KernelModeTime"])

    mem_mb = mem / 1048576.0
    page_mb = page / 1048576.0
    user_per_sec = get_rate("wmic_process.user.%s.%d" % (name, count), time.time(), userc)
    kernel_per_sec = get_rate("wmic_process.kernel.%s.%d" % (name, count), time.time(), kernelc)
    user_perc = user_per_sec / 100000.0 / cpucores
    kernel_perc = kernel_per_sec / 100000.0 / cpucores
    cpu_perc = user_perc + kernel_perc
    perfdata = [
            ("mem",    mem_mb,      memwarn,  memcrit),
            ("page",   page_mb,     pagewarn, pagecrit),
            ("user",   user_perc,   cpuwarn,  cpucrit,  0, 100),
            ("kernel", kernel_perc, cpuwarn,  cpucrit,  0, 100) ]

    messages = []
    messages.append("%d processes" % count)
    state = 0

    msg =  "%.0f%%/%.0f%% User/Kernel" % (user_perc, kernel_perc)
    if cpu_perc >= cpucrit:
        state = 2
        msg += "(!!) (critical at %d%%)" % cpucrit
    elif cpu_perc >= cpuwarn:
        state = 1
        msg += "(!) (warning at %d%%)" % cpuwarn
    messages.append(msg)

    msg = "%.1fMB RAM" % mem_mb
    if mem_mb >= memcrit and memcrit > 0:
        state = 2
        msg += "(!!) (critical at %d MB)" % memcrit
    elif mem_mb >= memwarn and memwarn > 0:
        state = max(1, state)
        msg += "(!) (warning at %d MB)" % memwarn
    messages.append(msg)

    msg = "%1.fMB Page" % page_mb
    if page_mb >= pagecrit:
        state = 2
        msg += "(!!) (critical at %d MB)" % pagecrit
    elif page_mb >= pagewarn:
        state = max(state, 1)
        msg += "(!) (warning at %d MB)" % pagewarn
    messages.append(msg)

    return (state, ", ".join(messages), perfdata)

check_info["wmic_process"] = {
    'check_function':          check_wmic_process,
    'service_description':     'Process %s',
    'has_perfdata':            True,
    'group':                   'wmic_process'
}
