git.oblomov.eu Git - wine/blob - dlls/wbemprox/builtin.c

   1 /*
   2  * Copyright 2012 Hans Leidekker for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #define COBJMACROS
  20 #define NONAMELESSUNION
  21 #define NONAMELESSSTRUCT
  22
  23 #include "config.h"
  24 #include <stdarg.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "wbemcli.h"
  29 #include "tlhelp32.h"
  30
  31 #include "wine/debug.h"
  32 #include "wbemprox_private.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(wbemprox);
  35
  36 static const WCHAR class_biosW[] =
  37     {'W','i','n','3','2','_','B','I','O','S',0};
  38 static const WCHAR class_compsysW[] =
  39     {'W','i','n','3','2','_','C','o','m','p','u','t','e','r','S','y','s','t','e','m',0};
  40 static const WCHAR class_osW[] =
  41     {'W','i','n','3','2','_','O','p','e','r','a','t','i','n','g','S','y','s','t','e','m',0};
  42 static const WCHAR class_processW[] =
  43     {'W','i','n','3','2','_','P','r','o','c','e','s','s',0};
  44 static const WCHAR class_processorW[] =
  45     {'W','i','n','3','2','_','P','r','o','c','e','s','s','o','r',0};
  46
  47 static const WCHAR prop_captionW[] =
  48     {'C','a','p','t','i','o','n',0};
  49 static const WCHAR prop_descriptionW[] =
  50     {'D','e','s','c','r','i','p','t','i','o','n',0};
  51 static const WCHAR prop_manufacturerW[] =
  52     {'M','a','n','u','f','a','c','t','u','r','e','r',0};
  53 static const WCHAR prop_modelW[] =
  54     {'M','o','d','e','l',0};
  55 static const WCHAR prop_osarchitectureW[] =
  56     {'O','S','A','r','c','h','i','t','e','c','t','u','r','e',0};
  57 static const WCHAR prop_pprocessidW[] =
  58     {'P','a','r','e','n','t','P','r','o','c','e','s','s','I','D',0};
  59 static const WCHAR prop_processidW[] =
  60     {'P','r','o','c','e','s','s','I','D',0};
  61 static const WCHAR prop_releasedateW[] =
  62     {'R','e','l','e','a','s','e','D','a','t','e',0};
  63 static const WCHAR prop_serialnumberW[] =
  64     {'S','e','r','i','a','l','N','u','m','b','e','r',0};
  65 static const WCHAR prop_threadcountW[] =
  66     {'T','h','r','e','a','d','C','o','u','n','t',0};
  67
  68 static const struct column col_bios[] =
  69 {
  70     { prop_descriptionW,  CIM_STRING },
  71     { prop_manufacturerW, CIM_STRING },
  72     { prop_releasedateW,  CIM_DATETIME },
  73     { prop_serialnumberW, CIM_STRING }
  74 };
  75 static const struct column col_compsys[] =
  76 {
  77     { prop_descriptionW,  CIM_STRING },
  78     { prop_manufacturerW, CIM_STRING },
  79     { prop_modelW,        CIM_STRING }
  80 };
  81 static const struct column col_os[] =
  82 {
  83     { prop_captionW,        CIM_STRING },
  84     { prop_osarchitectureW, CIM_STRING }
  85 };
  86 static const struct column col_process[] =
  87 {
  88     { prop_captionW,     CIM_STRING|COL_FLAG_DYNAMIC },
  89     { prop_descriptionW, CIM_STRING|COL_FLAG_DYNAMIC },
  90     { prop_pprocessidW,  CIM_UINT32 },
  91     { prop_processidW,   CIM_UINT32 },
  92     { prop_threadcountW, CIM_UINT32 }
  93 };
  94 static const struct column col_processor[] =
  95 {
  96     { prop_manufacturerW, CIM_STRING }
  97 };
  98
  99 static const WCHAR bios_descriptionW[] =
 100     {'D','e','f','a','u','l','t',' ','S','y','s','t','e','m',' ','B','I','O','S',0};
 101 static const WCHAR bios_manufacturerW[] =
 102     {'T','h','e',' ','W','i','n','e',' ','P','r','o','j','e','c','t',0};
 103 static const WCHAR bios_releasedateW[] =
 104     {'2','0','1','2','0','6','0','8','0','0','0','0','0','0','.','0','0','0','0','0','0','+','0','0','0',0};
 105 static const WCHAR bios_serialnumberW[] =
 106     {'0',0};
 107 static const WCHAR compsys_descriptionW[] =
 108     {'A','T','/','A','T',' ','C','O','M','P','A','T','I','B','L','E',0};
 109 static const WCHAR compsys_manufacturerW[] =
 110     {'T','h','e',' ','W','i','n','e',' ','P','r','o','j','e','c','t',0};
 111 static const WCHAR compsys_modelW[] =
 112     {'W','i','n','e',0};
 113 static const WCHAR os_captionW[] =
 114     {'W','i','n','e',0};
 115 static const WCHAR os_32bitW[] =
 116     {'3','2','-','b','i','t',0};
 117 static const WCHAR os_64bitW[] =
 118     {'6','4','-','b','i','t',0};
 119 static const WCHAR processor_manufacturerW[] =
 120     {'G','e','n','u','i','n','e','I','n','t','e','l',0};
 121
 122 #include "pshpack1.h"
 123 struct record_bios
 124 {
 125     const WCHAR *description;
 126     const WCHAR *manufacturer;
 127     const WCHAR *releasedate;
 128     const WCHAR *serialnumber;
 129 };
 130 struct record_computersystem
 131 {
 132     const WCHAR *description;
 133     const WCHAR *manufacturer;
 134     const WCHAR *model;
 135 };
 136 struct record_operatingsystem
 137 {
 138     const WCHAR *caption;
 139     const WCHAR *osarchitecture;
 140 };
 141 struct record_process
 142 {
 143     const WCHAR *caption;
 144     const WCHAR *description;
 145     UINT32       pprocess_id;
 146     UINT32       process_id;
 147     UINT32       thread_count;
 148 };
 149 struct record_processor
 150 {
 151     const WCHAR *manufacturer;
 152 };
 153 #include "poppack.h"
 154
 155 static const struct record_bios data_bios[] =
 156 {
 157     { bios_descriptionW, bios_manufacturerW, bios_releasedateW, bios_serialnumberW }
 158 };
 159 static const struct record_computersystem data_compsys[] =
 160 {
 161     { compsys_descriptionW, compsys_manufacturerW, compsys_modelW }
 162 };
 163 static const struct record_processor data_processor[] =
 164 {
 165     { processor_manufacturerW }
 166 };
 167
 168 static void fill_process( struct table *table )
 169 {
 170     struct record_process *rec;
 171     PROCESSENTRY32W entry;
 172     HANDLE snap;
 173     UINT num_rows = 0, offset = 0, count = 8;
 174
 175     snap = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
 176     if (snap == INVALID_HANDLE_VALUE) return;
 177
 178     entry.dwSize = sizeof(entry);
 179     if (!Process32FirstW( snap, &entry )) goto done;
 180     if (!(table->data = heap_alloc( count * sizeof(*rec) ))) goto done;
 181
 182     do
 183     {
 184         if (num_rows > count)
 185         {
 186             BYTE *data;
 187             count *= 2;
 188             if (!(data = heap_realloc( table->data, count * sizeof(*rec) ))) goto done;
 189             table->data = data;
 190         }
 191         rec = (struct record_process *)(table->data + offset);
 192         rec->caption      = heap_strdupW( entry.szExeFile );
 193         rec->description  = heap_strdupW( entry.szExeFile );
 194         rec->process_id   = entry.th32ProcessID;
 195         rec->pprocess_id  = entry.th32ParentProcessID;
 196         rec->thread_count = entry.cntThreads;
 197         offset += sizeof(*rec);
 198         num_rows++;
 199     } while (Process32NextW( snap, &entry ));
 200
 201     TRACE("created %u rows\n", num_rows);
 202     table->num_rows = num_rows;
 203
 204 done:
 205     CloseHandle( snap );
 206 }
 207
 208 static void fill_os( struct table *table )
 209 {
 210     struct record_operatingsystem *rec;
 211     SYSTEM_INFO info;
 212
 213     if (!(table->data = heap_alloc( sizeof(*rec) ))) return;
 214
 215     rec = (struct record_operatingsystem *)table->data;
 216     rec->caption = os_captionW;
 217
 218     GetNativeSystemInfo( &info );
 219     if (info.u.s.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64)
 220         rec->osarchitecture = os_64bitW;
 221     else
 222         rec->osarchitecture = os_32bitW;
 223
 224     TRACE("created 1 row\n");
 225     table->num_rows = 1;
 226 }
 227
 228 static struct table classtable[] =
 229 {
 230     { class_biosW, SIZEOF(col_bios), col_bios, SIZEOF(data_bios), (BYTE *)data_bios, NULL },
 231     { class_compsysW, SIZEOF(col_compsys), col_compsys, SIZEOF(data_compsys), (BYTE *)data_compsys, NULL },
 232     { class_osW, SIZEOF(col_os), col_os, 0, NULL, fill_os },
 233     { class_processW, SIZEOF(col_process), col_process, 0, NULL, fill_process },
 234     { class_processorW, SIZEOF(col_processor), col_processor, SIZEOF(data_processor), (BYTE *)data_processor, NULL }
 235 };
 236
 237 struct table *get_table( const WCHAR *name )
 238 {
 239     UINT i;
 240     struct table *table = NULL;
 241
 242     for (i = 0; i < SIZEOF(classtable); i++)
 243     {
 244         if (!strcmpiW( classtable[i].name, name ))
 245         {
 246             table = &classtable[i];
 247             if (table->fill && !table->data) table->fill( table );
 248             break;
 249         }
 250     }
 251     return table;
 252 }