Files
wmic/pysamba/wbem/wbem.py
T
2019-02-16 00:16:52 +01:00

310 lines
13 KiB
Python

###########################################################################
#
# 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)