/*++

Copyright (c) Microsoft Corporation.  All rights reserved.

Module Name:

    Driver.cpp

Abstract:

    Sample Soundwire Extension Unit Driver.

Environment:

    Kernel mode only

--*/

#include "private.h"
#include "trace.h"
#include "driver.h"

#ifndef __INTELLISENSE__
#include "driver.tmh"
#endif

RECORDER_LOG g_SDCAVXuLog;

// When creating private drops, use set C_DEFINES=-DDROP_STAMP=nnnn to add the drop timestamp to the initial trace
#ifndef DROP_STAMP
#define DROP_STAMP 0
#endif

PAGED_CODE_SEG
void SdcaXu_DriverUnload (_In_ WDFDRIVER Driver)
{
    PAGED_CODE(); 

    if (!Driver)
    {
        ASSERT(FALSE);
        return;
    }

    if (g_RegistryPath.Buffer != NULL)
    {
        ExFreePool(g_RegistryPath.Buffer);
        RtlZeroMemory(&g_RegistryPath, sizeof(g_RegistryPath));
    }

    WPP_CLEANUP(WdfDriverWdmGetDriverObject(Driver));

    return;
}

INIT_CODE_SEG
NTSTATUS
DriverEntry(
    _In_ PDRIVER_OBJECT  DriverObject,
    _In_ PUNICODE_STRING RegistryPath
    )
/*++

Routine Description:
    DriverEntry initializes the driver and is the first routine called by the
    system after the driver is loaded.

Parameters Description:

    DriverObject - represents the instance of the function driver that is loaded
    into memory. DriverEntry must initialize members of DriverObject before it
    returns to the caller. DriverObject is allocated by the system before the
    driver is loaded, and it is released by the system after the system unloads
    the function driver from memory.

    RegistryPath - represents the driver specific path in the Registry.
    The function driver can use the path to store driver related data between
    reboots. The path does not store hardware instance specific data.

Return Value:

    STATUS_SUCCESS if successful,
    STATUS_UNSUCCESSFUL otherwise.

--*/
{
    PAGED_CODE(); 
    
    NTSTATUS status = STATUS_SUCCESS;

    WPP_INIT_TRACING(DriverObject, RegistryPath);

    auto exit = scope_exit([&status, &DriverObject]() {
        if (!NT_SUCCESS(status))
        {
            if (g_RegistryPath.Buffer != NULL)
            {
                ExFreePool(g_RegistryPath.Buffer);
                RtlZeroMemory(&g_RegistryPath, sizeof(g_RegistryPath));
            }
        
            WPP_CLEANUP(DriverObject);
        }
    });

    RETURN_NTSTATUS_IF_FAILED(CopyRegistrySettingsPath(RegistryPath));

    //
    // Initiialize driver config to control the attributes that
    // are global to the driver. Note that framework by default
    // provides a driver unload routine. If you create any resources
    // in the DriverEntry and want to be cleaned in driver unload,
    // you can override that by manually setting the EvtDriverUnload in the
    // config structure. In general xxx_CONFIG_INIT macros are provided to
    // initialize most commonly used members.
    //

    WDF_DRIVER_CONFIG wdfCfg;
    WDF_DRIVER_CONFIG_INIT(&wdfCfg, SdcaXu_EvtBusDeviceAdd);
    wdfCfg.EvtDriverUnload = SdcaXu_DriverUnload;
    
    //
    // Add a driver context. (for illustration purposes only).
    //
    WDF_OBJECT_ATTRIBUTES attributes;
    WDF_OBJECT_ATTRIBUTES_INIT_CONTEXT_TYPE(&attributes, SDCAXU_DRIVER_CONTEXT);

    //
    // Create a framework driver object to represent our driver.
    //
    WDFDRIVER driver;
    RETURN_NTSTATUS_IF_FAILED(WdfDriverCreate(
        DriverObject,
        RegistryPath,
        &attributes,            // Driver Attributes
        &wdfCfg,                // Driver Config Info
        &driver                 // hDriver
        ));

    RECORDER_CONFIGURE_PARAMS recorderConfig;
    RECORDER_CONFIGURE_PARAMS_INIT(&recorderConfig);
    recorderConfig.CreateDefaultLog = FALSE;
    WppRecorderConfigure(&recorderConfig); 

    RECORDER_LOG_CREATE_PARAMS recorderLogCreateParams;
    RECORDER_LOG_CREATE_PARAMS_INIT(&recorderLogCreateParams, NULL);
    recorderLogCreateParams.TotalBufferSize = WPP_TOTAL_BUFFER_SIZE;
    recorderLogCreateParams.ErrorPartitionSize = WPP_ERROR_PARTITION_SIZE;

    RtlStringCbPrintfA(recorderLogCreateParams.LogIdentifier,
        RECORDER_LOG_IDENTIFIER_MAX_CHARS,
        "SDCAVXu");

    RECORDER_LOG logHandle = NULL;
    status = WppRecorderLogCreate(&recorderLogCreateParams, &logHandle);
    if (!NT_SUCCESS(status))
    {
        logHandle = NULL;

        // Non fatal failure
        status = STATUS_SUCCESS;
    }

    g_SDCAVXuLog = logHandle;

    DrvLogInfo(g_SDCAVXuLog, FLAG_INIT, "SdcaVXu Driver loaded, %lld", DROP_STAMP);

    //
    // Post init.
    //
    ACX_DRIVER_CONFIG acxCfg;
    ACX_DRIVER_CONFIG_INIT(&acxCfg);
    
    RETURN_NTSTATUS_IF_FAILED(AcxDriverInitialize(driver, &acxCfg));
    
    return status;
}