etw-provider.c

Go to the documentation of this file.

#ifdef _WIN32
 
#include <windows.h>
#include <TraceLoggingProvider.h>
#pragma comment(lib, "Advapi32.lib")
#include <trace-etw-util.h>
#include <stdlib.h>

struct etw_provider
{
    etw_provider_ref_t provider_ref;
};
 
etw_provider_t *TRACE_ETW_UTIL_API
    etw_provider_init(etw_provider_ref_t provider_ref)
{
    etw_provider_t *handle;
    TLG_STATUS status;
 
    if (provider_ref == NULL)
    {
        return NULL;
    }
 
    handle = (etw_provider_t *)malloc(sizeof(etw_provider_t));
    if (handle == NULL)
    {
        return NULL;
    }
 
    handle->provider_ref = provider_ref;
 
    status = TraceLoggingRegister(provider_ref);
    if (status != S_OK)
    {
        free(handle);
        return NULL;
    }
 
    return handle;
}

static void write_trace_event(etw_provider_ref_t ref, int level,
                               const char *service, const char *message)
{
    if (service != NULL)
    {
        switch (level)
        {
        case 1:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(1),
                TraceLoggingString(service, "Service"),
                TraceLoggingString(message, "Message"));
            break;
        case 2:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(2),
                TraceLoggingString(service, "Service"),
                TraceLoggingString(message, "Message"));
            break;
        case 3:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(3),
                TraceLoggingString(service, "Service"),
                TraceLoggingString(message, "Message"));
            break;
        case 4:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(4),
                TraceLoggingString(service, "Service"),
                TraceLoggingString(message, "Message"));
            break;
        default:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(5),
                TraceLoggingString(service, "Service"),
                TraceLoggingString(message, "Message"));
            break;
        }
    }
    else
    {
        switch (level)
        {
        case 1:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(1),
                TraceLoggingString(message, "Message"));
            break;
        case 2:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(2),
                TraceLoggingString(message, "Message"));
            break;
        case 3:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(3),
                TraceLoggingString(message, "Message"));
            break;
        case 4:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(4),
                TraceLoggingString(message, "Message"));
            break;
        default:
            TraceLoggingWrite(ref, "Trace",
                TraceLoggingLevel(5),
                TraceLoggingString(message, "Message"));
            break;
        }
    }
}
 
int TRACE_ETW_UTIL_API
    etw_provider_write(etw_provider_t *handle, int level,
                       const char *service, const char *message)
{
    if (handle == NULL || message == NULL)
    {
        return 0;
    }
 
    write_trace_event(handle->provider_ref, level, service, message);
 
    return 0;
}
 
void TRACE_ETW_UTIL_API
    etw_provider_dispose(etw_provider_t *handle)
{
    if (handle == NULL)
    {
        return;
    }
 
    TraceLoggingUnregister(handle->provider_ref);
    free(handle);
}
 
#endif /* _WIN32 */