main.c File Reference

#include <stdio.h>
#include "windows.h"
#include "ocidl.h"
#include "initguid.h"
#include "objidl.h"
#include "wbemcli.h"
#include "wmic.h"
#include "wine/debug.h"
#include "wine/unicode.h"

Include dependency graph for main.c:

Go to the source code of this file.

Macros
#define	COBJMACROS

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (wmic)
static const WCHAR *	find_class (const WCHAR *alias)
static WCHAR *	strdupW (const WCHAR *src)
static WCHAR *	find_prop (IWbemClassObject *class, const WCHAR *prop)
static int	output_string (HANDLE handle, const WCHAR *msg,...)
static int	output_error (int msg)
static int	output_header (const WCHAR *prop, ULONG column_width)
static int	output_line (const WCHAR *str, ULONG column_width)
static int	query_prop (const WCHAR *class, const WCHAR *propname)
int	wmain (int argc, WCHAR *argv[])

Variables
static const WCHAR	biosW []
static const WCHAR	computersystemW []
static const WCHAR	cpuW []
static const WCHAR	logicaldiskW []
static const WCHAR	nicW []
static const WCHAR	osW []
static const WCHAR	processW []
static const WCHAR	win32_biosW []
static const WCHAR	win32_computersystemW []
static const WCHAR	win32_logicaldiskW []
static const WCHAR	win32_networkadapterW []
static const WCHAR	win32_operatingsystemW []
static const WCHAR	win32_processW []
static const WCHAR	win32_processorW []
struct {
const WCHAR *   alias
const WCHAR *   class
}	alias_map []

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 20 of file main.c.

Function Documentation

find_class()

static const WCHAR * find_class ( const WCHAR *  alias )

static

Definition at line 81 of file main.c.

{
    unsigned int i;
 
    for (i = 0; i < ARRAY_SIZE(alias_map); i++)
    {
        if (!strcmpiW( alias, alias_map[i].alias )) return alias_map[i].class;
    }
    return NULL;
}

Referenced by wmain().

find_prop()

static WCHAR * find_prop ( IWbemClassObject *  class, const WCHAR *  prop  )

static

Definition at line 101 of file main.c.

{
    SAFEARRAY *sa;
    WCHAR *ret = NULL;
    LONG i, last_index = 0;
    BSTR str;
 
    if (IWbemClassObject_GetNames( class, NULL, WBEM_FLAG_ALWAYS, NULL, &sa ) != S_OK) return NULL;
 
    SafeArrayGetUBound( sa, 1, &last_index );
    for (i = 0; i <= last_index; i++)
    {
        SafeArrayGetElement( sa, &i, &str );
        if (!strcmpiW( str, prop ))
        {
            ret = strdupW( str );
            break;
        }
    }
    SafeArrayDestroy( sa );
    return ret;
}

Referenced by query_prop().

output_error()

static int output_error ( int  msg )

static

Definition at line 141 of file main.c.

{
    static const WCHAR fmtW[] = {'%','s',0};
    WCHAR buffer[8192];
 
    LoadStringW( GetModuleHandleW(NULL), msg, buffer, ARRAY_SIZE(buffer));
    return output_string( GetStdHandle(STD_ERROR_HANDLE), fmtW, buffer );
}

Referenced by query_prop(), and wmain().

output_header()

static int output_header ( const WCHAR *  prop, ULONG  column_width  )

static

Definition at line 150 of file main.c.

{
    static const WCHAR bomW[] = {0xfeff}, fmtW[] = {'%','-','*','s','\r','\n',0};
    int len;
    DWORD count;
    WCHAR buffer[8192];
 
    len = snprintfW( buffer, ARRAY_SIZE(buffer), fmtW, column_width, prop );
 
    if (!WriteConsoleW( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len, &count, NULL )) /* redirected */
    {
        WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), bomW, sizeof(bomW), &count, FALSE );
        WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len * sizeof(WCHAR), &count, FALSE );
        count += sizeof(bomW);
    }
 
    return count;
}

Referenced by query_prop(), and test_encode().

output_line()

static int output_line ( const WCHAR *  str, ULONG  column_width  )

static

Definition at line 169 of file main.c.

{
    static const WCHAR fmtW[] = {'%','-','*','s','\r','\n',0};
    return output_string( GetStdHandle(STD_OUTPUT_HANDLE), fmtW, column_width, str );
}

Referenced by query_prop().

output_string()

static int output_string ( HANDLE  handle, const WCHAR *  msg,   ...  )

static

Definition at line 124 of file main.c.

{
    va_list va_args;
    int len;
    DWORD count;
    WCHAR buffer[8192];
 
    va_start( va_args, msg );
    len = vsnprintfW( buffer, ARRAY_SIZE(buffer), msg, va_args );
    va_end( va_args );
 
    if (!WriteConsoleW( handle, buffer, len, &count, NULL ))
        WriteFile( handle, buffer, len * sizeof(WCHAR), &count, FALSE );
 
    return count;
}

Referenced by output_error(), and output_line().

query_prop()

static int query_prop ( const WCHAR *  class, const WCHAR *  propname  )

static

Definition at line 175 of file main.c.

{
    static const WCHAR select_allW[] = {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',0};
    static const WCHAR cimv2W[] = {'R','O','O','T','\\','C','I','M','V','2',0};
    static const WCHAR wqlW[] = {'W','Q','L',0};
    HRESULT hr;
    IWbemLocator *locator = NULL;
    IWbemServices *services = NULL;
    IEnumWbemClassObject *result = NULL;
    LONG flags = WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY;
    BSTR path = NULL, wql = NULL, query = NULL;
    WCHAR *prop = NULL;
    BOOL first = TRUE;
    int len, ret = -1;
    IWbemClassObject *obj;
    ULONG count, width = 0;
    VARIANT v;
 
    WINE_TRACE("%s, %s\n", debugstr_w(class), debugstr_w(propname));
 
    CoInitialize( NULL );
    CoInitializeSecurity( NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT,
                          RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL );
 
    hr = CoCreateInstance( &CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemLocator,
                           (void **)&locator );
    if (hr != S_OK) goto done;
 
    if (!(path = SysAllocString( cimv2W ))) goto done;
    hr = IWbemLocator_ConnectServer( locator, path, NULL, NULL, NULL, 0, NULL, NULL, &services );
    if (hr != S_OK) goto done;
 
    len = strlenW( class ) + ARRAY_SIZE(select_allW);
    if (!(query = SysAllocStringLen( NULL, len ))) goto done;
    strcpyW( query, select_allW );
    strcatW( query, class );
 
    if (!(wql = SysAllocString( wqlW ))) goto done;
    hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, &result );
    if (hr != S_OK) goto done;
 
    for (;;) /* get column width */
    {
        IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
        if (!count) break;
 
        if (!prop && !(prop = find_prop( obj, propname )))
        {
            output_error( STRING_INVALID_QUERY );
            goto done;
        }
        if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
        {
            VariantChangeType( &v, &v, 0, VT_BSTR );
            width = max( strlenW( V_BSTR( &v ) ), width );
            VariantClear( &v );
        }
        IWbemClassObject_Release( obj );
    }
    width += 2;
 
    IEnumWbemClassObject_Reset( result );
    for (;;)
    {
        IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
        if (!count) break;
 
        if (first)
        {
            output_header( prop, width );
            first = FALSE;
        }
        if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
        {
            VariantChangeType( &v, &v, 0, VT_BSTR );
            output_line( V_BSTR( &v ), width );
            VariantClear( &v );
        }
        IWbemClassObject_Release( obj );
    }
    ret = 0;
 
done:
    if (result) IEnumWbemClassObject_Release( result );
    if (services) IWbemServices_Release( services );
    if (locator) IWbemLocator_Release( locator );
    SysFreeString( path );
    SysFreeString( query );
    SysFreeString( wql );
    HeapFree( GetProcessHeap(), 0, prop );
    CoUninitialize();
    return ret;
}

Referenced by wmain().

strdupW()

static WCHAR * strdupW ( const WCHAR *  src )

inlinestatic

Definition at line 92 of file main.c.

{
    WCHAR *dst;
    if (!src) return NULL;
    if (!(dst = HeapAlloc( GetProcessHeap(), 0, (strlenW( src ) + 1) * sizeof(WCHAR) ))) return NULL;
    strcpyW( dst, src );
    return dst;
}

Referenced by add_cookie(), add_domain(), assembly_create(), BackgroundCopyFileConstructor(), BackgroundCopyJob_SetCredentials(), BackgroundCopyJobConstructor(), build_path(), build_source_filename(), downloadcb_OnProgress(), find_prop(), CShellLink::GetIconLocation(), handle_redirect(), HtmlHelpW(), http_options_SetCustomHeaders(), IAssemblyName_SetPath(), insert_header(), ldap_encode_sort_controlW(), CShellLink::Load(), LoadWinTypeFromCHM(), MergeChmString(), name_GetVersion(), CShellLink::OnNotify(), open_connection(), OpenCHM(), parse_culture(), parse_display_name(), path_SetClassName(), path_SetNamespaceAt(), path_SetServer(), CShellLink::Resolve(), resolve_filename(), CShellLink::Save(), set_cookies(), set_credentials(), set_server_for_hostname(), CShellLink::SetArguments(), SetChmPath(), CShellLink::SetDescription(), CShellLink::SetIconLocation(), CShellLink::SetPath(), CShellLink::SetRelativePath(), CShellLink::SetTargetFromPIDLOrPath(), CShellLink::SetWorkingDirectory(), winhttp_request_Open(), winhttp_request_SetProxy(), WinHttpConnect(), WinHttpOpen(), WinHttpOpenRequest(), and WinHttpSendRequest().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

wmic

)

wmain()

int wmain	(	int	argc,
		WCHAR *	argv[]
	)

Definition at line 269 of file main.c.

{
    static const WCHAR getW[] = {'g','e','t',0};
    static const WCHAR quitW[] = {'q','u','i','t',0};
    static const WCHAR exitW[] = {'e','x','i','t',0};
    static const WCHAR pathW[] = {'p','a','t','h',0};
    static const WCHAR classW[] = {'c','l','a','s','s',0};
    static const WCHAR contextW[] = {'c','o','n','t','e','x','t',0};
    const WCHAR *class, *value;
    int i;
 
    for (i = 1; i < argc && argv[i][0] == '/'; i++)
        WINE_FIXME( "command line switch %s not supported\n", debugstr_w(argv[i]) );
 
    if (i >= argc)
        goto not_supported;
 
    if (!strcmpiW( argv[i], quitW ) ||
        !strcmpiW( argv[i], exitW ))
    {
        return 0;
    }
 
    if (!strcmpiW( argv[i], classW) ||
        !strcmpiW( argv[i], contextW ))
    {
        WINE_FIXME( "command %s not supported\n", debugstr_w(argv[i]) );
        goto not_supported;
    }
 
    if (!strcmpiW( argv[i], pathW ))
    {
        if (++i >= argc)
        {
            output_error( STRING_INVALID_PATH );
            return 1;
        }
        class = argv[i];
    }
    else
    {
        class = find_class( argv[i] );
        if (!class)
        {
            output_error( STRING_ALIAS_NOT_FOUND );
            return 1;
        }
    }
 
    if (++i >= argc)
        goto not_supported;
 
    if (!strcmpiW( argv[i], getW ))
    {
        if (++i >= argc)
            goto not_supported;
        value = argv[i];
        return query_prop( class, value );
    }
 
not_supported:
    output_error( STRING_CMDLINE_NOT_SUPPORTED );
    return 1;
}

Variable Documentation

alias

const WCHAR* alias

Definition at line 67 of file main.c.

Referenced by CLSIDFromProgID(), DoGetAliasPIDLs(), DoInitAliasPIDLs(), find_class(), fnIMultiLanguage3_GetCharsetInfo(), MCI_FinishOpen(), write_progid_record(), xmlAddEncodingAlias(), xmlCleanupEncodingAliases(), xmlDelEncodingAlias(), xmlFindCharEncodingHandler(), xmlGetEncodingAlias(), xmlNewCharEncodingHandler(), xmlParseCharEncoding(), and xsltNamespaceAlias().

const struct { ... } alias_map[]

Initial value:

=
{
    { biosW, win32_biosW },
    { computersystemW, win32_computersystemW },
    { cpuW, win32_processorW },
    { logicaldiskW, win32_logicaldiskW },
    { nicW, win32_networkadapterW },
    { osW, win32_operatingsystemW },
    { processW, win32_processW }
}

Referenced by find_class().

biosW

const WCHAR biosW[]

static

Initial value:

=
    {'b','i','o','s',0}

Definition at line 35 of file main.c.

Referenced by test_ParseDisplayName().

class

static JOBOBJECTINFOCLASS class

Definition at line 68 of file main.c.

Referenced by check_script_collisions(), compile_script(), get_coclass_threading(), has_class_installed(), has_class_removed(), IsInterfaceExpected(), lookup_script_identifier(), test_wm_notify(), TestManualInstantiation(), TestModuleInterfaces(), TestModuleRegistry(), write_coclass(), and write_progid().

computersystemW

const WCHAR computersystemW[]

static

Initial value:

=
    {'c','o','m','p','u','t','e','r','s','y','s','t','e','m',0}

Definition at line 37 of file main.c.

cpuW

const WCHAR cpuW[]

static

Initial value:

=
    {'c','p','u',0}

Definition at line 39 of file main.c.

logicaldiskW

const WCHAR logicaldiskW[]

static

Initial value:

=
    {'L','o','g','i','c','a','l','D','i','s','k',0}

Definition at line 41 of file main.c.

nicW

const WCHAR nicW[]

static

Initial value:

=
    {'n','i','c',0}

Definition at line 43 of file main.c.

osW

const WCHAR osW[]

static

Initial value:

=
    {'o','s',0}

Definition at line 45 of file main.c.

processW

const WCHAR processW[]

static

Initial value:

=
    {'p','r','o','c','e','s','s',0}

Definition at line 47 of file main.c.

Referenced by test_Win32_Process().

win32_biosW

const WCHAR win32_biosW[]

static

Initial value:

=
    {'W','i','n','3','2','_','B','I','O','S',0}

Definition at line 50 of file main.c.

win32_computersystemW

const WCHAR win32_computersystemW[]

static

Initial value:

=
    {'W','i','n','3','2','_','C','o','m','p','u','t','e','r','S','y','s','t','e','m',0}

Definition at line 52 of file main.c.

win32_logicaldiskW

const WCHAR win32_logicaldiskW[]

static

Initial value:

=
    {'W','i','n','3','2','_','L','o','g','i','c','a','l','D','i','s','k',0}

Definition at line 54 of file main.c.

win32_networkadapterW

const WCHAR win32_networkadapterW[]

static

Initial value:

=
    {'W','i','n','3','2','_','N','e','t','w','o','r','k','A','d','a','p','t','e','r',0}

Definition at line 56 of file main.c.

win32_operatingsystemW

const WCHAR win32_operatingsystemW[]

static

Initial value:

=
    {'W','i','n','3','2','_','O','p','e','r','a','t','i','n','g','S','y','s','t','e','m',0}

Definition at line 58 of file main.c.

win32_processorW

const WCHAR win32_processorW[]

static

Initial value:

=
    {'W','i','n','3','2','_','P','r','o','c','e','s','s','o','r',0}

Definition at line 62 of file main.c.

win32_processW

const WCHAR win32_processW[]

static

Initial value:

=
    {'W','i','n','3','2','_','P','r','o','c','e','s','s',0}

Definition at line 60 of file main.c.