source4/lib/wmi/wbemdata.c

   1 /*
   2    WMI Implementation
   3    Copyright (C) 2006 Andrzej Hajda <andrzej.hajda@wp.pl>
   4
   5    This program is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 2 of the License, or
   8    (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18 */
  19
  20 #include "includes.h"
  21 #include "librpc/gen_ndr/dcom.h"
  22 #include "librpc/gen_ndr/com_dcom.h"
  23 #include "librpc/ndr/libndr.h"
  24 #include "librpc/ndr/libndr_proto.h"
  25 #include "lib/com/com.h"
  26 #include "lib/com/dcom/dcom.h"
  27 #include "lib/util/dlinklist.h"
  28 #include "librpc/ndr/libndr.h"
  29 #include "librpc/gen_ndr/ndr_dcom.h"
  30 #include "librpc/rpc/dcerpc.h"
  31 #include "librpc/gen_ndr/ndr_misc.h"
  32 #include "lib/talloc/talloc.h"
  33 #include "libcli/composite/composite.h"
  34 #include "wmi/proto.h"
  35
  36 NTSTATUS ndr_pull_WbemClassObject_Object(struct ndr_pull *ndr, int ndr_flags, struct WbemClassObject *r);
  37 void duplicate_CIMVAR(TALLOC_CTX *mem_ctx, const union CIMVAR *src, union CIMVAR *dst, enum CIMTYPE_ENUMERATION cimtype);
  38 void duplicate_WbemClassObject(TALLOC_CTX *mem_ctx, const struct WbemClassObject *src, struct WbemClassObject *dst);
  39
  40 #define NDR_CHECK_LEN(n) do { if (p + (n) > pend) { \
  41                                 DEBUG(0, ("%s(%d): WBEMDATA_ERR(0x%08X): Buffer too small(0x%04X)\n", __FILE__, __LINE__, ndr->offset, p + (n) - pend)); \
  42                                 status = NT_STATUS_UNSUCCESSFUL; \
  43                                 goto end; \
  44                             } \
  45                         } while(0)
  46
  47 #define NDR_CHECK_EXPR(expr) do { if (!(expr)) {\
  48                                         DEBUG(0, ("%s(%d): WBEMDATA_ERR(0x%08X): Error parsing(%s)\n", __FILE__, __LINE__, ndr->offset, #expr)); \
  49                                         status = NT_STATUS_UNSUCCESSFUL; \
  50                                         goto end; \
  51                                         } \
  52                                     } while(0)
  53
  54 #define NDR_CHECK_CONST(val, exp) NDR_CHECK_EXPR((val) == (exp))
  55 #define NDR_CHECK_RSTRING(rstring) NDR_CHECK_EXPR((rstring) >= 0)
  56
  57 #define NTERR_CHECK(call) status = call; if (!NT_STATUS_IS_OK(status)) goto end;
  58
  59 enum {
  60         DATATYPE_CLASSOBJECT = 2,
  61         DATATYPE_OBJECT = 3,
  62
  63         COFLAG_IS_CLASS = 4,
  64 };
  65
  66 static NTSTATUS marshal(struct IUnknown *pv, struct OBJREF *o)
  67 {
  68         struct ndr_push *ndr;
  69         NTSTATUS status;
  70         struct WbemClassObject *wco;
  71         TALLOC_CTX *mem_ctx;
  72         struct MInterfacePointer *mp;
  73
  74         mp = (struct MInterfacePointer *)((char *)o - offsetof(struct MInterfacePointer, obj)); // FIXME:high remove this Mumbo Jumbo
  75         wco = pv->object_data;
  76         mem_ctx = talloc_new(0);
  77         ndr = talloc_zero(mem_ctx, struct ndr_push);
  78         ndr->flags = 0;
  79         ndr->alloc_size = 1024;
  80         ndr->data = talloc_array(mp, uint8_t, ndr->alloc_size);
  81
  82         if (wco) {
  83                 uint32_t ofs;
  84                 NTERR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, 0x12345678));
  85                 NTERR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, 0));
  86                 NTERR_CHECK(ndr_push_WbemClassObject(ndr, NDR_SCALARS | NDR_BUFFERS, wco));
  87                 ofs = ndr->offset;
  88                 ndr->offset = 4;
  89                 NTERR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, ofs - 8));
  90                 ndr->offset = ofs;
  91         } else {
  92                 NTERR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, 0));
  93         }
  94         o->u_objref.u_custom.pData = talloc_realloc(mp, ndr->data, uint8_t, ndr->offset);
  95         o->u_objref.u_custom.size = ndr->offset;
  96         mp->size = sizeof(struct OBJREF) - sizeof(union OBJREF_Types) + sizeof(struct u_custom) + o->u_objref.u_custom.size - 4;
  97         if (DEBUGLVL(9)) {
  98                 NDR_PRINT_DEBUG(WbemClassObject, wco);
  99         }
 100 end:
 101         talloc_free(mem_ctx);
 102         return status;
 103 }
 104
 105 static NTSTATUS unmarshal(struct OBJREF *o, struct IUnknown **pv)
 106 {
 107         struct ndr_pull *ndr;
 108         TALLOC_CTX *mem_ctx;
 109         struct WbemClassObject *wco;
 110         NTSTATUS status;
 111         uint32_t u;
 112
 113         mem_ctx = talloc_new(0);
 114         ndr = talloc_zero(mem_ctx, struct ndr_pull);
 115         ndr->current_mem_ctx = mem_ctx;
 116         ndr->data = o->u_objref.u_custom.pData;
 117         ndr->data_size = o->u_objref.u_custom.size;
 118
 119         NTERR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 120         if (!u) {
 121                 talloc_free(*pv);
 122                 *pv = NULL;
 123                 status = NT_STATUS_OK;
 124                 goto end;
 125         }
 126         NTERR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 127         if (u + 8 > ndr->data_size) {
 128                 DEBUG(1, ("unmarshall_IWbemClassObject: Incorrect data_size"));
 129                 status = NT_STATUS_BUFFER_TOO_SMALL;
 130                 goto end;
 131         }
 132         wco = talloc_zero(*pv, struct WbemClassObject);
 133         ndr->current_mem_ctx = wco;
 134         status = ndr_pull_WbemClassObject(ndr, NDR_SCALARS | NDR_BUFFERS, wco);
 135
 136         if (NT_STATUS_IS_OK(status) && (DEBUGLVL(9))) {
 137                 NDR_PRINT_DEBUG(WbemClassObject, wco);
 138         }
 139
 140         if (NT_STATUS_IS_OK(status)) {
 141                 (*pv)->object_data = wco;
 142         } else {
 143                 talloc_free(wco);
 144         }
 145 end:
 146         talloc_free(mem_ctx);
 147         return status;
 148 }
 149
 150 WERROR dcom_IWbemClassObject_from_WbemClassObject(struct com_context *ctx, struct IWbemClassObject **_p, struct WbemClassObject *wco)
 151 {
 152         struct IWbemClassObject *p;
 153
 154
 155         p = talloc_zero(ctx, struct IWbemClassObject);
 156         p->ctx = ctx;
 157         p->obj.signature = 0x574f454d;
 158         p->obj.flags = OBJREF_CUSTOM;
 159         GUID_from_string("dc12a681-737f-11cf-884d-00aa004b2e24", &p->obj.iid);
 160         GUID_from_string("4590f812-1d3a-11d0-891f-00aa004b2e24", &p->obj.u_objref.u_custom.clsid);
 161         p->object_data = (void *)wco;
 162         talloc_steal(p, p->object_data);
 163         *_p = p;
 164         return WERR_OK;
 165 }
 166
 167 WERROR IWbemClassObject_GetMethod(struct IWbemClassObject *d, TALLOC_CTX *mem_ctx, const char *name, uint32_t flags, struct IWbemClassObject **in, struct IWbemClassObject **out)
 168 {
 169         uint32_t i;
 170         struct WbemClassObject *wco;
 171
 172         wco = (struct WbemClassObject *)d->object_data;
 173         for (i = 0; i < wco->obj_methods->count; ++i)
 174                 if (!strcmp(wco->obj_methods->method[i].name, name)) {
 175                         if (in) dcom_IWbemClassObject_from_WbemClassObject(d->ctx, in, wco->obj_methods->method[i].in);
 176                         if (out) dcom_IWbemClassObject_from_WbemClassObject(d->ctx, out, wco->obj_methods->method[i].out);
 177                         return WERR_OK;
 178                 }
 179         return WERR_NOT_FOUND;
 180 }
 181
 182 void WbemClassObject_CreateInstance(struct WbemClassObject *wco)
 183 {
 184         uint32_t i;
 185
 186         wco->instance = talloc_zero(wco, struct WbemInstance);
 187         wco->instance->default_flags = talloc_array(wco->instance, uint8_t, wco->obj_class->__PROPERTY_COUNT);
 188         wco->instance->data = talloc_array(wco->instance, union CIMVAR, wco->obj_class->__PROPERTY_COUNT);
 189         memset(wco->instance->data, 0, sizeof(union CIMVAR) * wco->obj_class->__PROPERTY_COUNT);
 190         for (i = 0; i < wco->obj_class->__PROPERTY_COUNT; ++i) {
 191                 wco->instance->default_flags[i] = 1; // FIXME:high resolve this magic
 192         }
 193         wco->instance->__CLASS = wco->obj_class->__CLASS;
 194         wco->instance->u2_4 = 4;
 195         wco->instance->u3_1 = 1;
 196 }
 197
 198 WERROR IWbemClassObject_Clone(struct IWbemClassObject *d, TALLOC_CTX *mem_ctx, struct IWbemClassObject **copy)
 199 {
 200         return WERR_NOT_SUPPORTED;
 201 }
 202
 203 WERROR IWbemClassObject_SpawnInstance(struct IWbemClassObject *d, TALLOC_CTX *mem_ctx, uint32_t flags, struct IWbemClassObject **instance)
 204 {
 205         struct WbemClassObject *wco, *nwco;
 206
 207         wco = (struct WbemClassObject *)d->object_data;
 208         nwco = talloc_zero(mem_ctx, struct WbemClassObject);
 209         nwco->flags = WCF_INSTANCE;
 210         nwco->obj_class = wco->obj_class;
 211         WbemClassObject_CreateInstance(nwco);
 212         dcom_IWbemClassObject_from_WbemClassObject(d->ctx, instance, nwco);
 213         return WERR_OK;
 214 }
 215
 216 void duplicate_WbemQualifier(TALLOC_CTX *mem_ctx, const struct WbemQualifier *src, struct WbemQualifier *dst)
 217 {
 218         dst->name = src->name;
 219         if (src->name) dst->name = talloc_strdup(mem_ctx, src->name);
 220
 221         dst->flavors = src->flavors;
 222
 223         dst->cimtype = src->cimtype;
 224
 225         duplicate_CIMVAR(mem_ctx, &src->value, &dst->value, src->cimtype);
 226 }
 227
 228 void duplicate_CIMSTRINGS(TALLOC_CTX *mem_ctx, const struct CIMSTRINGS *src, struct CIMSTRINGS *dst)
 229 {
 230         uint32_t i;
 231
 232         dst->count = src->count;
 233         for (i = 0; i < src->count; ++i)
 234                 dst->item[i] = talloc_strdup(mem_ctx, src->item[i]);
 235 }
 236
 237 void duplicate_WbemQualifiers(TALLOC_CTX *mem_ctx, const struct WbemQualifiers *src, struct WbemQualifiers *dst)
 238 {
 239         uint32_t i;
 240
 241         dst->count = src->count;
 242         for (i = 0; i < src->count; ++i) {
 243                 dst->item[i] = talloc_zero(mem_ctx, struct WbemQualifier);
 244                 duplicate_WbemQualifier(dst->item[i], src->item[i], dst->item[i]);
 245         }
 246 }
 247
 248 void duplicate_WbemClass(TALLOC_CTX *mem_ctx, const struct WbemClass *src, struct WbemClass *dst)
 249 {
 250         uint32_t i;
 251
 252         dst->u_0 = src->u_0;
 253
 254         dst->__CLASS = src->__CLASS;
 255         if (src->__CLASS) dst->__CLASS = talloc_strdup(mem_ctx, src->__CLASS);
 256
 257         duplicate_CIMSTRINGS(mem_ctx, &src->__DERIVATION, &dst->__DERIVATION);
 258         duplicate_WbemQualifiers(mem_ctx, &src->qualifiers, &dst->qualifiers);
 259
 260         dst->__PROPERTY_COUNT = src->__PROPERTY_COUNT;
 261
 262         dst->properties = talloc_array(mem_ctx, struct WbemProperty, src->__PROPERTY_COUNT);
 263         for (i = 0; i < src->__PROPERTY_COUNT; ++i) {
 264                 dst->properties[i].name = talloc_strdup(dst->properties, src->properties[i].name);
 265                 dst->properties[i].desc = talloc_memdup(dst->properties, src->properties[i].desc, sizeof(*src->properties[i].desc));
 266                 duplicate_WbemQualifiers(dst->properties[i].desc, &src->properties[i].desc->qualifiers, &dst->properties[i].desc->qualifiers);
 267         }
 268
 269         dst->default_flags = talloc_array(mem_ctx, uint8_t, src->__PROPERTY_COUNT);
 270         dst->default_values = talloc_array(mem_ctx, union CIMVAR, src->__PROPERTY_COUNT);
 271         for (i = 0; i < src->__PROPERTY_COUNT; ++i) {
 272                 dst->default_flags[i] = src->default_flags[i];
 273                 duplicate_CIMVAR(dst->default_values, &src->default_values[i], &dst->default_values[i], src->properties[i].desc->cimtype);
 274         }
 275 }
 276
 277 void duplicate_WbemMethod(TALLOC_CTX *mem_ctx, const struct WbemMethod *src, struct WbemMethod *dst)
 278 {
 279         dst->name = src->name;
 280         if (src->name) dst->name = talloc_strdup(mem_ctx, src->name);
 281
 282         dst->u0 = src->u0;
 283         dst->u1 = src->u1;
 284
 285         dst->qualifiers = talloc_zero(mem_ctx, struct WbemQualifiers);
 286         duplicate_WbemQualifiers(dst->qualifiers, src->qualifiers, dst->qualifiers);
 287
 288         dst->in = src->in;
 289         if (src->in) {
 290                 dst->in = talloc_zero(mem_ctx, struct WbemClassObject);
 291                 duplicate_WbemClassObject(dst->in, src->in, dst->in);
 292         }
 293
 294         dst->out = src->out;
 295         if (src->out) {
 296                 dst->out = talloc_zero(mem_ctx, struct WbemClassObject);
 297                 duplicate_WbemClassObject(dst->out, src->out, dst->out);
 298         }
 299 }
 300
 301 void duplicate_WbemMethods(TALLOC_CTX *mem_ctx, const struct WbemMethods *src, struct WbemMethods *dst)
 302 {
 303         uint32_t i;
 304
 305         dst->count = src->count;
 306         dst->u0 = src->u0;
 307         for (i = 0; i < src->count; ++i)
 308                 duplicate_WbemMethod(mem_ctx, &src->method[i], &dst->method[i]);
 309 }
 310
 311 void duplicate_WbemInstance(TALLOC_CTX *mem_ctx, const struct WbemInstance *src, struct WbemInstance *dst, const struct WbemClass *cls)
 312 {
 313         uint32_t i;
 314
 315         dst->u1_0 = src->u1_0;
 316
 317         dst->__CLASS = src->__CLASS;
 318         if (src->__CLASS) dst->__CLASS = talloc_strdup(mem_ctx, src->__CLASS);
 319
 320         dst->default_flags = talloc_array(mem_ctx, uint8_t, cls->__PROPERTY_COUNT);
 321         dst->data = talloc_array(mem_ctx, union CIMVAR, cls->__PROPERTY_COUNT);
 322         for (i = 0; i < cls->__PROPERTY_COUNT; ++i) {
 323                 dst->default_flags[i] = src->default_flags[i];
 324                 duplicate_CIMVAR(dst->data, &src->data[i], &dst->data[i], cls->properties[i].desc->cimtype);
 325         }
 326
 327         dst->u2_4 = src->u2_4;
 328         dst->u3_1 = src->u3_1;
 329 }
 330
 331 void duplicate_WbemClassObject(TALLOC_CTX *mem_ctx, const struct WbemClassObject *src, struct WbemClassObject *dst)
 332 {
 333         dst->flags = src->flags;
 334         if (src->flags & WCF_CLASS) {
 335                 dst->__SERVER = talloc_strdup(mem_ctx, src->__SERVER);
 336                 dst->__NAMESPACE = talloc_strdup(mem_ctx, src->__NAMESPACE);
 337         }
 338         if (src->flags & WCF_DECORATIONS) {
 339                 dst->sup_class = talloc_zero(mem_ctx, struct WbemClass);
 340                 duplicate_WbemClass(dst->sup_class, src->sup_class, dst->sup_class);
 341
 342                 dst->sup_methods = talloc_zero(mem_ctx, struct WbemMethods);
 343                 duplicate_WbemMethods(dst->sup_methods, src->sup_methods, dst->sup_methods);
 344
 345                 dst->obj_methods = talloc_zero(mem_ctx, struct WbemMethods);
 346                 duplicate_WbemMethods(dst->obj_methods, src->obj_methods, dst->obj_methods);
 347         }
 348         if (src->flags & (WCF_CLASS | WCF_INSTANCE)) {
 349                 dst->obj_class = talloc_zero(mem_ctx, struct WbemClass);
 350                 duplicate_WbemClass(dst->obj_class, src->obj_class, dst->obj_class);
 351         }
 352         if (src->flags & WCF_INSTANCE) {
 353                 dst->instance = talloc_zero(mem_ctx, struct WbemInstance);
 354                 duplicate_WbemInstance(dst->instance, src->instance, dst->instance, src->obj_class);
 355         }
 356 }
 357
 358 void duplicate_CIMVAR(TALLOC_CTX *mem_ctx, const union CIMVAR *src, union CIMVAR *dst, enum CIMTYPE_ENUMERATION cimtype)
 359 {
 360         uint32_t i;
 361
 362         switch (cimtype & CIM_TYPEMASK) {
 363         case CIM_SINT8:
 364         case CIM_UINT8:
 365         case CIM_SINT16:
 366         case CIM_UINT16:
 367         case CIM_SINT32:
 368         case CIM_UINT32:
 369         case CIM_SINT64:
 370         case CIM_UINT64:
 371         case CIM_REAL32:
 372         case CIM_REAL64:
 373         case CIM_BOOLEAN:
 374                 *dst = *src;
 375                 break;
 376         case CIM_STRING:
 377         case CIM_DATETIME:
 378         case CIM_REFERENCE:
 379                 dst->v_string = talloc_strdup(mem_ctx, src->v_string);
 380                 break;
 381         case CIM_OBJECT:
 382                 dst->v_object = talloc_zero(mem_ctx, struct WbemClassObject);
 383                 duplicate_WbemClassObject(dst->v_object, src->v_object, dst->v_object);
 384                 break;
 385         case CIM_ARR_SINT8:
 386         case CIM_ARR_UINT8:
 387                 dst->a_uint8 = talloc_memdup(mem_ctx, src->a_uint8, sizeof(struct arr_uint8));
 388                 dst->a_uint8->item = talloc_memdup(dst->a_uint8, src->a_uint8->item, src->a_uint8->count);
 389                 break;
 390         case CIM_ARR_SINT16:
 391         case CIM_ARR_UINT16:
 392         case CIM_ARR_BOOLEAN:
 393                 dst->a_uint8 = talloc_memdup(mem_ctx, src->a_uint8, sizeof(struct arr_uint8));
 394                 dst->a_uint8->item = talloc_memdup(dst->a_uint8, src->a_uint8->item, 2*src->a_uint8->count);
 395                 break;
 396         case CIM_ARR_SINT32:
 397         case CIM_ARR_UINT32:
 398         case CIM_ARR_REAL32:
 399                 dst->a_uint8 = talloc_memdup(mem_ctx, src->a_uint8, sizeof(struct arr_uint8));
 400                 dst->a_uint8->item = talloc_memdup(dst->a_uint8, src->a_uint8->item, 4*src->a_uint8->count);
 401                 break;
 402         case CIM_ARR_SINT64:
 403         case CIM_ARR_UINT64:
 404         case CIM_ARR_REAL64:
 405                 dst->a_uint8 = talloc_memdup(mem_ctx, src->a_uint8, sizeof(struct arr_uint8));
 406                 dst->a_uint8->item = talloc_memdup(dst->a_uint8, src->a_uint8->item, 8*src->a_uint8->count);
 407                 break;
 408         case CIM_ARR_STRING:
 409         case CIM_ARR_DATETIME:
 410         case CIM_ARR_REFERENCE:
 411                 dst->a_uint8 = talloc_memdup(mem_ctx, src->a_uint8, sizeof(struct arr_uint8));
 412                 dst->a_uint8->item = talloc_memdup(dst->a_uint8, src->a_uint8->item, 4*src->a_uint8->count);
 413                 for (i = 0; i < src->a_uint8->count; ++i)
 414                         dst->a_string->item[i] = talloc_strdup(dst->a_uint8->item, src->a_string->item[i]);
 415                 break;
 416         default:
 417                 DEBUG(0, ("duplicate_CIMVAR: cimtype 0x%04X not supported\n", cimtype & CIM_TYPEMASK));
 418                 break;
 419         }
 420 }
 421
 422 WERROR WbemClassObject_Get(struct WbemClassObject *d, TALLOC_CTX *mem_ctx, const char *name, uint32_t flags, union CIMVAR *val, enum CIMTYPE_ENUMERATION *cimtype, uint32_t *flavor)
 423 {
 424         uint32_t i;
 425         for (i = 0; i < d->obj_class->__PROPERTY_COUNT; ++i) {
 426                 if (!strcmp(d->obj_class->properties[i].name, name)) {
 427                         duplicate_CIMVAR(mem_ctx, &d->instance->data[i], val, d->obj_class->properties[i].desc->cimtype);
 428                         if (cimtype) *cimtype = d->obj_class->properties[i].desc->cimtype;
 429                         if (flavor) *flavor = 0; // FIXME:avg implement flavor
 430                         return WERR_OK;
 431                 }
 432         }
 433         return WERR_NOT_FOUND;
 434 }
 435
 436 WERROR IWbemClassObject_Put(struct IWbemClassObject *d, TALLOC_CTX *mem_ctx, const char *name, uint32_t flags, union CIMVAR *val, enum CIMTYPE_ENUMERATION cimtype)
 437 {
 438         struct WbemClassObject *wco;
 439         uint32_t i;
 440
 441         wco = (struct WbemClassObject *)d->object_data;
 442         for (i = 0; i < wco->obj_class->__PROPERTY_COUNT; ++i) {
 443                 if (!strcmp(wco->obj_class->properties[i].name, name)) {
 444                         if (cimtype && cimtype != wco->obj_class->properties[i].desc->cimtype) return WERR_INVALID_PARAM;
 445                         wco->instance->default_flags[i] = 0;
 446                         duplicate_CIMVAR(wco->instance, val, &wco->instance->data[i], wco->obj_class->properties[i].desc->cimtype);
 447                         return WERR_OK;
 448                 }
 449         }
 450         return WERR_NOT_FOUND;
 451 }
 452
 453 #define WERR_CHECK(msg) if (!W_ERROR_IS_OK(result)) { \
 454                             DEBUG(1, ("ERROR: %s - %s\n", msg, wmi_errstr(result))); \
 455                             goto end; \
 456                         } else { \
 457                             DEBUG(1, ("OK   : %s\n", msg)); \
 458                         }
 459
 460 struct pair_guid_ptr {
 461         struct GUID guid;
 462         void *ptr;
 463         struct pair_guid_ptr *next, *prev;
 464 };
 465
 466 static void *get_ptr_by_guid(struct pair_guid_ptr *list, struct GUID *uuid)
 467 {
 468         for (; list; list = list->next) {
 469                 if (GUID_equal(&list->guid, uuid))
 470                         return list->ptr;
 471         }
 472         return NULL;
 473 }
 474
 475 static void add_pair_guid_ptr(TALLOC_CTX *mem_ctx, struct pair_guid_ptr **list, struct GUID *uuid, void *ptr)
 476 {
 477         struct pair_guid_ptr *e;
 478
 479         e = talloc(mem_ctx, struct pair_guid_ptr);
 480         e->guid = *uuid;
 481         e->ptr = ptr;
 482         talloc_steal(e, ptr);
 483         DLIST_ADD(*list, e);
 484 }
 485
 486 struct IEnumWbemClassObject_data {
 487         struct GUID guid;
 488         struct IWbemFetchSmartEnum *pFSE;
 489         struct IWbemWCOSmartEnum *pSE;
 490         struct pair_guid_ptr *cache;
 491 };
 492
 493 static NTSTATUS WBEMDATA_Parse(uint8_t *data, uint32_t size, struct IEnumWbemClassObject *d, uint32_t uCount, struct WbemClassObject **apObjects)
 494 {
 495         struct ndr_pull *ndr;
 496         TALLOC_CTX *mem_ctx;
 497         uint32_t u, i, ofs_next;
 498         uint8_t u8, datatype;
 499         NTSTATUS status;
 500         struct GUID guid;
 501         struct IEnumWbemClassObject_data *ecod;
 502
 503         if (!uCount) return NT_STATUS_NOT_IMPLEMENTED;
 504
 505         ecod = d->object_data;
 506         mem_ctx = talloc_new(0);
 507
 508         ndr = talloc_zero(mem_ctx, struct ndr_pull);
 509         ndr->current_mem_ctx = d->ctx;
 510         ndr->data = data;
 511         ndr->data_size = size;
 512         ndr_set_flags(&ndr->flags, LIBNDR_FLAG_NOALIGN);
 513
 514         NDR_CHECK_set_shift(0x18);
 515         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 516         NDR_CHECK_CONST(u, 0x0);
 517         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 518         NDR_CHECK_CONST(u, *(const uint32_t *)"WBEM");
 519         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 520         NDR_CHECK_CONST(u, *(const uint32_t *)"DATA");
 521         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 522         NDR_CHECK_CONST(u, 0x1A); /* Length of header */
 523         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 524         NDR_PULL_NEED_BYTES(ndr, u + 6);
 525         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 526         NDR_CHECK_CONST(u, 0x0);
 527         NDR_CHECK(ndr_pull_uint8(ndr, NDR_SCALARS, &u8));
 528         NDR_CHECK_CONST(u8, 0x01); /* Major Version */
 529         NDR_CHECK(ndr_pull_uint8(ndr, NDR_SCALARS, &u8));
 530         NDR_CHECK_EXPR(u8 <= 1); /* Minor Version 0 - Win2000, 1 - XP/2003 */
 531         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 532         NDR_CHECK_CONST(u, 0x8); /* Length of header */
 533         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 534         NDR_PULL_NEED_BYTES(ndr, u);
 535         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 536         NDR_CHECK_CONST(u, 0xC); /* Length of header */
 537         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 538         NDR_PULL_NEED_BYTES(ndr, u + 4);
 539         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 540         NDR_CHECK_CONST(u, uCount);
 541         for (i = 0; i < uCount; ++i) {
 542                 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 543                 NDR_CHECK_CONST(u, 0x9); /* Length of header */
 544                 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 545                 NDR_PULL_NEED_BYTES(ndr, u + 1);
 546                 NDR_CHECK(ndr_pull_uint8(ndr, NDR_SCALARS, &datatype));
 547                 ofs_next = ndr->offset + u;
 548                 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 549                 NDR_CHECK_CONST(u, 0x18); /* Length of header */
 550                 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &u));
 551                 NDR_PULL_NEED_BYTES(ndr, u + 16);
 552                 NDR_CHECK(ndr_pull_GUID(ndr, NDR_SCALARS, &guid));
 553                 switch (datatype) {
 554                 case DATATYPE_CLASSOBJECT:
 555                         apObjects[i] = talloc_zero(d->ctx, struct WbemClassObject);
 556                         ndr->current_mem_ctx = apObjects[i];
 557                         NDR_CHECK(ndr_pull_WbemClassObject(ndr, NDR_SCALARS|NDR_BUFFERS, apObjects[i]));
 558                         ndr->current_mem_ctx = d->ctx;
 559                         add_pair_guid_ptr(ecod, &ecod->cache, &guid, apObjects[i]->obj_class);
 560                         break;
 561                 case DATATYPE_OBJECT:
 562                         apObjects[i] = talloc_zero(d->ctx, struct WbemClassObject);
 563                         apObjects[i]->obj_class = get_ptr_by_guid(ecod->cache, &guid);
 564                         (void)talloc_reference(apObjects[i], apObjects[i]->obj_class);
 565                         ndr->current_mem_ctx = apObjects[i];
 566                         NDR_CHECK(ndr_pull_WbemClassObject_Object(ndr, NDR_SCALARS|NDR_BUFFERS, apObjects[i]));
 567                         ndr->current_mem_ctx = d->ctx;
 568                         break;
 569                 default:
 570                         DEBUG(0, ("WBEMDATA_Parse: Data type %d not supported\n", datatype));
 571                         status = NT_STATUS_NOT_SUPPORTED;
 572                         goto end;
 573                 }
 574                 ndr->offset = ofs_next;
 575                 if (DEBUGLVL(9)) {
 576                         NDR_PRINT_DEBUG(WbemClassObject, apObjects[i]);
 577                 }
 578         }
 579         status = NT_STATUS_OK;
 580 end:
 581         talloc_free(mem_ctx);
 582         return status;
 583 }
 584
 585 struct composite_context *dcom_proxy_IEnumWbemClassObject_Release_send(struct IUnknown *d, TALLOC_CTX *mem_ctx);
 586
 587 WERROR IEnumWbemClassObject_SmartNext(struct IEnumWbemClassObject *d, TALLOC_CTX *mem_ctx, int32_t lTimeout, uint32_t uCount, struct WbemClassObject **apObjects, uint32_t *puReturned)
 588 {
 589         WERROR result;
 590         NTSTATUS status;
 591         struct IEnumWbemClassObject_data *ecod;
 592         TALLOC_CTX *loc_ctx;
 593         uint32_t size;
 594         uint8_t *data;
 595
 596         loc_ctx = talloc_new(0);
 597         ecod = d->object_data;
 598         if (!ecod) {
 599                 struct GUID iid;
 600                 WERROR coresult;
 601
 602                 d->object_data = ecod = talloc_zero(d, struct IEnumWbemClassObject_data);
 603                 GUID_from_string(COM_IWBEMFETCHSMARTENUM_UUID, &iid);
 604                 result = dcom_query_interface((struct IUnknown *)d, 5, 1, &iid, (struct IUnknown **)&ecod->pFSE, &coresult);
 605                 WERR_CHECK("dcom_query_interface.");
 606                 result = coresult;
 607                 WERR_CHECK("Retrieve enumerator of result(IWbemFetchSmartEnum).");
 608
 609                 result = IWbemFetchSmartEnum_Fetch(ecod->pFSE, mem_ctx, &ecod->pSE);
 610                 WERR_CHECK("Retrieve enumerator of result(IWbemWCOSmartEnum).");
 611
 612                 ecod->guid = GUID_random();
 613                 d->vtable->Release_send = dcom_proxy_IEnumWbemClassObject_Release_send;
 614         }
 615
 616         result = IWbemWCOSmartEnum_Next(ecod->pSE, loc_ctx, &ecod->guid, lTimeout, uCount, puReturned, &size, &data);
 617         if (!W_ERROR_EQUAL(result, WERR_BADFUNC)) {
 618                 WERR_CHECK("IWbemWCOSmartEnum_Next.");
 619         }
 620
 621         if (data) {
 622                 status = WBEMDATA_Parse(data, size, d, *puReturned, apObjects);
 623                 result = ntstatus_to_werror(status);
 624                 WERR_CHECK("WBEMDATA_Parse.");
 625         }
 626 end:
 627         if (!W_ERROR_IS_OK(result)) {
 628                 status = werror_to_ntstatus(result);
 629                 DEBUG(9, ("dcom_proxy_IEnumWbemClassObject_Next: %s - %s\n", nt_errstr(status), get_friendly_nt_error_msg(status)));
 630         }
 631         talloc_free(loc_ctx);
 632         return result;
 633 }
 634
 635 struct composite_context *dcom_proxy_IEnumWbemClassObject_Release_send(struct IUnknown *d, TALLOC_CTX *mem_ctx)
 636 {
 637         struct composite_context *c, *cr;
 638         struct REMINTERFACEREF iref[3];
 639         struct dcom_object_exporter *ox;
 640         struct IEnumWbemClassObject_data *ecod;
 641         int n;
 642
 643         c = composite_create(d->ctx, d->ctx->event_ctx);
 644         if (c == NULL) return NULL;
 645         c->private_data = d;
 646
 647         ox = object_exporter_by_ip(d->ctx, d);
 648         iref[0].ipid = IUnknown_ipid(d);
 649         iref[0].cPublicRefs = 5;
 650         iref[0].cPrivateRefs = 0;
 651         n = 1;
 652
 653         ecod = d->object_data;
 654         if (ecod) {
 655                 if (ecod->pFSE) {
 656                         talloc_steal(d, ecod->pFSE);
 657                         iref[n].ipid = IUnknown_ipid(ecod->pFSE);
 658                         iref[n].cPublicRefs = 5;
 659                         iref[n].cPrivateRefs = 0;
 660                         ++n;
 661                 }
 662                 if (ecod->pSE) {
 663                         talloc_steal(d, ecod->pSE);
 664                         iref[n].ipid = IUnknown_ipid(ecod->pSE);
 665                         iref[n].cPublicRefs = 5;
 666                         iref[n].cPrivateRefs = 0;
 667                         ++n;
 668                 }
 669         }
 670         cr = IRemUnknown_RemRelease_send(ox->rem_unknown, mem_ctx, n, iref);
 671
 672         composite_continue(c, cr, dcom_release_continue, c);
 673         return c;
 674 }
 675
 676 NTSTATUS dcom_proxy_IWbemClassObject_init()
 677 {
 678         struct GUID clsid;
 679         GUID_from_string("4590f812-1d3a-11d0-891f-00aa004b2e24", &clsid);
 680         dcom_register_marshal(&clsid, marshal, unmarshal);
 681
 682 #if 0
 683         struct IEnumWbemClassObject_vtable *proxy_vtable;
 684         proxy_vtable = (struct IEnumWbemClassObject_vtable *)dcom_proxy_vtable_by_iid((struct GUID *)&dcerpc_table_IEnumWbemClassObject.syntax_id.uuid);
 685         if (proxy_vtable)
 686                 proxy_vtable->Release_send = dcom_proxy_IEnumWbemClassObject_Release_send;
 687         else
 688                 DEBUG(0, ("WARNING: IEnumWbemClassObject should be initialized before IWbemClassObject."));
 689 #endif
 690
 691         return NT_STATUS_OK;
 692 }