###########################################################################
#
# This program is part of Zenoss Core, an open source monitoring platform.
# Copyright (C) 2008-2010, Zenoss Inc.
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 2, or (at your
# option) any later version, as published by the Free Software Foundation.
#
# For complete information please visit: http://www.zenoss.com/oss/
#
###########################################################################

from pysamba.library import *
from pysamba.composite_context import composite_context

class com_context(Structure): pass

CIMSTRING = c_char_p
CIM_EMPTY=0
CIM_SINT8=16
CIM_UINT8=17
CIM_SINT16=2
CIM_UINT16=18
CIM_SINT32=3
CIM_UINT32=19
CIM_SINT64=20
CIM_UINT64=21
CIM_REAL32=4
CIM_REAL64=5
CIM_BOOLEAN=11
CIM_STRING=8
CIM_DATETIME=101
CIM_REFERENCE=102
CIM_CHAR16=103
CIM_OBJECT=13
CIM_FLAG_ARRAY=0x2000
CIM_ARR_SINT8=CIM_FLAG_ARRAY|CIM_SINT8
CIM_ARR_UINT8=CIM_FLAG_ARRAY|CIM_UINT8
CIM_ARR_SINT16=CIM_FLAG_ARRAY|CIM_SINT16
CIM_ARR_UINT16=CIM_FLAG_ARRAY|CIM_UINT16
CIM_ARR_SINT32=CIM_FLAG_ARRAY|CIM_SINT32
CIM_ARR_UINT32=CIM_FLAG_ARRAY|CIM_UINT32
CIM_ARR_SINT64=CIM_FLAG_ARRAY|CIM_SINT64
CIM_ARR_UINT64=CIM_FLAG_ARRAY|CIM_UINT64
CIM_ARR_REAL32=CIM_FLAG_ARRAY|CIM_REAL32
CIM_ARR_REAL64=CIM_FLAG_ARRAY|CIM_REAL64
CIM_ARR_BOOLEAN=CIM_FLAG_ARRAY|CIM_BOOLEAN
CIM_ARR_STRING=CIM_FLAG_ARRAY|CIM_STRING
CIM_ARR_DATETIME=CIM_FLAG_ARRAY|CIM_DATETIME
CIM_ARR_REFERENCE=CIM_FLAG_ARRAY|CIM_REFERENCE
CIM_ARR_CHAR16=CIM_FLAG_ARRAY|CIM_CHAR16
CIM_ARR_OBJECT=CIM_FLAG_ARRAY|CIM_OBJECT
CIM_ILLEGAL=0xfff
CIM_TYPEMASK=0x2FFF

class CIMSTRINGS(Structure):
    _fields_ = [
        ('count', uint32_t),
        ('item', POINTER(CIMSTRING))
        ]

class WbemClassObject(Structure): pass

class arr_int8(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(int8_t))
        ]
class arr_uint8(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(uint8_t))
        ]
class arr_int16(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(int16_t))
        ]
class uarr_uint16(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(uint16_t))
        ]
class arr_int32(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(int32_t))
        ]
class arr_uint32(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(uint32_t))
        ]
dlong_t = int64_t
class arr_dlong(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(dlong_t))
        ]
class arr_uint32(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(uint32_t))
        ]
udlong_t = uint64_t
class arr_udlong(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(udlong_t))
        ]
class arr_uint16(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(uint16_t))
        ]
class arr_CIMSTRING(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(CIMSTRING))
        ]
class arr_WbemClassObject(Structure):
    _fields_= [
        ('count', uint32_t),
        ('item', POINTER(WbemClassObject))
        ]

class CIMVAR(Union):
    _fields_ = [
        ('v_sint8', int8_t),            # case (CIM_SINT8)
        ('v_uint8', uint8_t),           # case (CIM_UINT8)
        ('v_sint16', int16_t),          # case (CIM_SINT16)
        ('v_uint16', uint16_t),         # case (CIM_UINT16)
        ('v_sint32', int32_t),          # case (CIM_SINT32)
        ('v_uint32', uint32_t),         # case (CIM_UINT32)
        ('v_sint64', int64_t),          # case (CIM_SINT64)
        ('v_uint64', uint64_t),         # case (CIM_UINT64)
        ('v_real32', uint32_t),         # case (CIM_REAL32)
        ('v_real64', uint64_t),         # case (CIM_REAL64)
        ('v_boolean', uint16_t),        # case (CIM_BOOLEAN)
        ('v_string', CIMSTRING), # relative,string,charset(UTF16),case(CIM_STRING)]
        ('v_datetime', CIMSTRING), # relative,string,charset(UTF16),case(CIM_DATETIME)]
        ('v_reference', CIMSTRING), # relative,string,charset(UTF16),case(CIM_REFERENCE)]
        ('v_object', POINTER(WbemClassObject)), # [relative,subcontext(4),case(CIM_OBJECT)]
        ('a_sint8', POINTER(arr_int8)), # [relative,case(CIM_ARR_SINT8)]
        ('a_uint8', POINTER(arr_uint8)), # [relative,case(CIM_ARR_UINT8)]
        ('a_sint16', POINTER(arr_int16)), # [relative,case(CIM_ARR_SINT16)]
        ('a_uint16', POINTER(arr_uint16)), # [relative,case(CIM_ARR_UINT16)]
        ('a_sint32', POINTER(arr_int32)), # [relative,case(CIM_ARR_SINT32)]
        ('a_uint32', POINTER(arr_uint32)), # [relative,case(CIM_ARR_UINT32)]
        ('a_sint64', POINTER(arr_dlong)), # [relative,case(CIM_ARR_SINT64)]
        ('a_uint64', POINTER(arr_udlong)), # [relative,case(CIM_ARR_UINT64)]
        ('a_real32', POINTER(arr_uint32)), # [relative,case(CIM_ARR_REAL32)]
        ('a_real64', POINTER(arr_udlong)), # [relative,case(CIM_ARR_REAL64)]
        ('a_boolean', POINTER(arr_uint16)), # [relative,case(CIM_ARR_BOOLEAN)]
        ('a_string', POINTER(arr_CIMSTRING)), # [relative,case(CIM_ARR_STRING)]
        ('a_datetime', POINTER(arr_CIMSTRING)), # [relative,case(CIM_ARR_DATETIME)]
        ('a_reference', POINTER(arr_CIMSTRING)), # [relative,case(CIM_ARR_REFERENCE)]
        ('a_object', POINTER(arr_WbemClassObject)), # [relative,case(CIM_ARR_OBJECT)]
        ]


class WbemQualifier(Structure): pass

class WbemQualifiers(Structure):
    _fields_ = [
        ('count', uint32_t),
        ('item',  POINTER(POINTER(WbemQualifier))),
        ]

class WbemPropertyDesc(Structure):
    _fields_ = [
        ('cimtype', uint32_t),
        ('nr', uint16_t),
        ('offset', uint32_t),
        ('depth', uint32_t),
        ('qualifiers', WbemQualifiers),
        ]

class WbemProperty(Structure):
    _fields_ = [
        ('name', CIMSTRING), # [relative,string,charset(UTF16)]
        ('desc', POINTER(WbemPropertyDesc)), # [relative]
        ]

class WbemMethods(Structure): pass

class WbemClass(Structure):
    _fields_ = [
        ('u_0', uint8_t),
        ('__CLASS', CIMSTRING),
        ('data_size', uint32_t),
        ('__DERIVATION', CIMSTRINGS),
        ('qualifiers', WbemQualifiers),
        ('__PROPERTY_COUNT', uint32_t),
        ('properties', POINTER(WbemProperty)),
        ('default_flags', POINTER(uint8_t)),
        ('default_values', POINTER(CIMVAR)),
        ]

class WbemInstance(Structure):
    _fields_ = [
        ('u1_0', uint8_t),
        ('__CLASS', CIMSTRING),
        ('default_flags', POINTER(uint8_t)),
        ('data', POINTER(CIMVAR)),
        ('u2_4', uint32_t),
        ('u3_1', uint8_t),
        ]

WbemClassObject._fields_ = [
        ('flags', uint8_t),
        ('__SERVER', CIMSTRING),
        ('__NAMESPACE', CIMSTRING),
        ('sup_class', POINTER(WbemClass)),
        ('sup_methods', POINTER(WbemMethods)),
        ('obj_class', POINTER(WbemClass)),
        ('obj_methods', POINTER(WbemMethods)),
        ('instance', POINTER(WbemInstance)),
        ]

class IWbemServices(Structure): pass
class IEnumWbemClassObject(Structure): pass
class IWbemClassObject(Structure): pass
class IWbemContext(Structure): pass
class IUnknown(Structure): pass

# hack definition of BSTR, needs more complete structure definition
BSTR = c_char_p

WBEM_FLAG_RETURN_IMMEDIATELY = 0x10
WBEM_FLAG_ENSURE_LOCATABLE = 0x100
WBEM_FLAG_FORWARD_ONLY = 0x20

WBEM_S_TIMEDOUT=0x40004

#WERROR IWbemServices_ExecQuery_recv(struct composite_context *c, struct IEnumWbemClassObject **ppEnum);
library.IWbemServices_ExecQuery_recv.restype = WERROR
library.IWbemServices_ExecQuery_recv.argtypes = [POINTER(composite_context), POINTER(POINTER(IEnumWbemClassObject))]
library.IWbemServices_ExecQuery_recv = logFuncCall(library.IWbemServices_ExecQuery_recv)

#WERROR IEnumWbemClassObject_Reset_recv(struct composite_context *c);
library.IEnumWbemClassObject_Reset_recv.restype = WERROR
library.IEnumWbemClassObject_Reset_recv.argtypes = [POINTER(composite_context)]
library.IEnumWbemClassObject_Reset_recv = logFuncCall(library.IEnumWbemClassObject_Reset_recv)

#uint32_t IUnknown_Release_recv(struct composite_context *c);
library.IUnknown_Release_recv.restype = WERROR
library.IUnknown_Release_recv.argtypes = [POINTER(composite_context)]
library.IUnknown_Release_recv = logFuncCall(library.IUnknown_Release_recv)

library.WBEM_ConnectServer.restype = WERROR
library.WBEM_ConnectServer.argtypes = [POINTER(com_context), c_char_p, c_char_p, c_char_p, c_char_p, c_char_p, c_uint32, c_char_p, POINTER(IWbemContext), c_void_p]
library.WBEM_ConnectServer = logFuncCall(library.WBEM_ConnectServer)
library.WBEM_ConnectServer_recv.restype = WERROR
library.WBEM_ConnectServer_recv.argtypes = [POINTER(composite_context), c_void_p, c_void_p]
library.WBEM_ConnectServer_recv = logFuncCall(library.WBEM_ConnectServer_recv)

#extern struct composite_context *WBEM_ConnectServer_send(struct com_context *ctx,
#        TALLOC_CTX *parent_ctx, const char *server, const char *nspace,
#        const char *user, const char *password, const char *locale,
#        uint32_t flags, const char *authority, struct IWbemContext* wbem_ctx);
library.WBEM_ConnectServer_send.restype = POINTER(composite_context)
library.WBEM_ConnectServer_send.argtypes = [POINTER(com_context), c_void_p, c_char_p, c_char_p, c_char_p, c_char_p, c_char_p, c_uint32, c_char_p, POINTER(IWbemContext)]
library.WBEM_ConnectServer_send = logFuncCall(library.WBEM_ConnectServer_send)

library.IEnumWbemClassObject_SmartNext_send.restype = POINTER(composite_context)
library.IEnumWbemClassObject_SmartNext_send.argtypes = [POINTER(IEnumWbemClassObject), c_void_p, c_int32, c_uint32]
library.IEnumWbemClassObject_SmartNext_send = logFuncCall(library.IEnumWbemClassObject_SmartNext_send)
library.IEnumWbemClassObject_SmartNext_recv.restype = WERROR
library.IEnumWbemClassObject_SmartNext_recv.argtypes = [POINTER(composite_context), c_void_p, c_void_p, c_void_p]
library.IEnumWbemClassObject_SmartNext_recv = logFuncCall(library.IEnumWbemClassObject_SmartNext_recv)
library.IEnumWbemClassObject_SmartNext.restype = WERROR
library.IEnumWbemClassObject_SmartNext.argtypes = [POINTER(IEnumWbemClassObject), c_void_p, c_int32, c_uint32, c_void_p, c_void_p]
library.IEnumWbemClassObject_SmartNext = logFuncCall(library.IEnumWbemClassObject_SmartNext)
library.wmi_errstr.restype = c_char_p
library.wmi_errstr.argtypes = [WERROR]
library.wmi_errstr = logFuncCall(library.wmi_errstr)
library.IWbemClassObject_GetMethod.restype = WERROR
library.IWbemClassObject_GetMethod.argtypes = [POINTER(IWbemClassObject), c_void_p, c_char_p, c_uint32, c_void_p, c_void_p]
library.IWbemClassObject_GetMethod = logFuncCall(library.IWbemClassObject_GetMethod)
library.IWbemClassObject_SpawnInstance.restype = WERROR
library.IWbemClassObject_SpawnInstance.argtypes = [POINTER(IWbemClassObject), c_void_p, c_uint32, c_void_p]
library.IWbemClassObject_SpawnInstance = logFuncCall(library.IWbemClassObject_SpawnInstance)
library.IWbemClassObject_Put.restype = WERROR
library.IWbemClassObject_Put.argtypes = [POINTER(IWbemClassObject), c_void_p, c_char_p, c_uint32, POINTER(CIMVAR), enum]
library.IWbemClassObject_Put = logFuncCall(library.IWbemClassObject_Put)
library.WbemClassObject_Get.restype = WERROR
library.WbemClassObject_Get.argtypes = [POINTER(WbemClassObject), c_void_p, c_char_p, c_uint32, POINTER(CIMVAR), enum, c_void_p]
library.WbemClassObject_Get = logFuncCall(library.WbemClassObject_Get)

library.ConnectAndQuery.restype = WERROR
library.ConnectAndQuery.argtypes = [POINTER(com_context), c_char_p, c_char_p, c_void_p]
library.ConnectAndQuery = logFuncCall(library.ConnectAndQuery)
library.IWbemServices_ExecQuery_send_f.restype = POINTER(composite_context)
library.IWbemServices_ExecQuery_send_f.argtypes = [POINTER(IWbemServices), c_void_p, BSTR, BSTR, c_int32, POINTER(IWbemContext)]
library.IWbemServices_ExecQuery_send_f = logFuncCall(library.IWbemServices_ExecQuery_send_f)
library.IWbemServices_ExecNotificationQuery_send_f.restype = POINTER(composite_context)
library.IWbemServices_ExecNotificationQuery_send_f.argtypes = [POINTER(IWbemServices), c_void_p, BSTR, BSTR, c_int32, POINTER(IWbemContext)]
library.IWbemServices_ExecNotificationQuery_send_f = logFuncCall(library.IWbemServices_ExecNotificationQuery_send_f)
library.IWbemServices_ExecNotificationQuery_recv.restype = WERROR
library.IWbemServices_ExecNotificationQuery_recv.argtypes = [POINTER(composite_context), POINTER(POINTER(IEnumWbemClassObject))]
library.IWbemServices_ExecNotificationQuery_recv = logFuncCall(library.IWbemServices_ExecNotificationQuery_recv)
library.IEnumWbemClassObject_Reset_send_f.restype = POINTER(composite_context)
library.IEnumWbemClassObject_Reset_send_f.argtypes = [POINTER(IEnumWbemClassObject), c_void_p]
library.IEnumWbemClassObject_Reset_send_f = logFuncCall(library.IEnumWbemClassObject_Reset_send_f)
library.IUnknown_Release_send_f.restype = POINTER(composite_context)
library.IUnknown_Release_send_f.argtypes = [POINTER(IUnknown), c_void_p]
library.IUnknown_Release_send_f = logFuncCall(library.IUnknown_Release_send_f)