#include "pch.h"

#ifdef UNICODE
#error UNICODE not allowed
#endif

#define PNPREPT_HWND    (HWND)1

BOOL
WINAPI
MemDb_Entry (
    IN HINSTANCE hinstDLL,
    IN DWORD dwReason,
    IN LPVOID lpv
    );

BOOL
WINAPI
MigUtil_Entry (
    HINSTANCE hinstDLL,
    DWORD fdwReason,
    LPVOID lpvReserved
    );


VOID
pInitProgBarVars (
    VOID
    );

INT
pCallMains (
    DWORD Reason
    )
{
    if (!MigUtil_Entry (g_hInst, Reason, NULL)) {
        fprintf (stderr, "MigUtil_Entry error!\n");
        return 254;
    }

    if (!MemDb_Entry (g_hInst, Reason, NULL)) {
        fprintf (stderr, "MemDb_Entry error!\n");
        return 254;
    }

    if (!HwComp_Entry (g_hInst, Reason, NULL)) {
        fprintf (stderr, "HwComp_Entry error!\n");
        return 254;
    }

    pInitProgBarVars();

    return 0;
}

BOOL CancelFlag = FALSE;
BOOL *g_CancelFlagPtr = &CancelFlag;

#ifdef PRERELEASE
BOOL g_Stress;
#endif

HANDLE g_hHeap;
HINSTANCE g_hInst;
HINF g_OverrideInf = INVALID_HANDLE_VALUE;
BOOL g_ManualOverrideMode = FALSE;

CHAR   g_TempDirBuf[MAX_MBCHAR_PATH];      // location for hwcomp.dat
CHAR   g_TempDirWackBuf[MAX_MBCHAR_PATH];
CHAR   g_WinDirBuf[MAX_MBCHAR_PATH];
INT    g_TempDirWackChars;
PCSTR  g_SourceDirectories[MAX_SOURCE_COUNT];    // location of INFs
DWORD  g_SourceDirectoryCount;
PSTR g_TempDir;
PSTR g_TempDirWack;
PSTR g_WinDir;

USEROPTIONS g_ConfigOptions; // Needed by migutil. Unused.

extern HWND g_Component, g_SubComponent;

void
HelpAndExit (
    void
    )
{
    fprintf (stderr,
        "Command line syntax:\n\n"
        "pnprept {-a|-s|-u|-c} [-e<n>] <Inf Dir 1> <Inf Dir 2>\n\n"
        "Optional Arguments:\n"
        "  <Inf Dir 1> - Specifies the directory containing the first set of INFs\n"
        "  <Inf Dir 2> - Specifies the directory containing the second set of INFs\n"
        "\nOutput Options (specify at least one):\n"
        "  -a    - Dumps all devices for both Dir 1 and Dir 2.\n"
        "  -1    - Dumps devices supported by Dir 1 only (devices unsupported by Dir 2)\n"
        "  -2    - Dumps devices supported by Dir 2 only (devices unsupported by Dir 1)\n"
        "  -c    - Dumps devices common to Dir 1 and Dir 2\n"
        "\nOther Options:\n"
        "  -i:<path> - Specifies path to win95upg.inf for override list\n"
        "  -m        - Dumps only PNP IDs that are manually overridden (requires -i)\n"
        "  -e<n>     - Puts an equals instead of a tab after column <n>\n"
        "\n"
        );

    exit(255);
}

DWORD g_Mode = 0;
#define MODE_DIR1       0x0001
#define MODE_DIR2       0x0002

#define MAX_SEPARATORS 3
CHAR g_Separators[MAX_SEPARATORS];
LPSTR g_Dir1Path, g_Dir2Path;

VOID
Dump (
    DWORD MustHave,
    DWORD MustNotHave
    )
{
    MEMDB_ENUM e, e2;
    CHAR PnpOutput[MAX_ENCODED_PNPID_LENGTH];
    CHAR DescOutput[MAX_INF_DESCRIPTION*2];
    BOOL IsDir1, IsDir2;
    LPSTR Dir1File, Dir2File;
    GROWBUFFER Dir1List = GROWBUF_INIT;
    GROWBUFFER Dir2List = GROWBUF_INIT;
    BOOL Dir1Flag, Dir2Flag;
    DWORD Flags;
    CHAR Node[MEMDB_MAX];
    INFCONTEXT ic;

    if (!g_ManualOverrideMode) {
        printf ("%s%c%s%cPNP ID\tDevice Description%c\n",
                g_Dir1Path, g_Separators[0], g_Dir2Path, g_Separators[1], g_Separators[2]);
    } else {
        printf ("Overridden PNP IDs\n");
    }

    if (MemDbEnumFirstValue (&e, TEXT("Devices\\*"), MEMDB_THIS_LEVEL_ONLY, MEMDB_ALL_BUT_PROXY)) {

        do {
            //
            // Test this PNP ID to see if it should be displayed
            //

            if (((e.dwValue & MustHave) == MustHave) &&
                ((e.dwValue & MustNotHave) == 0)) {

                //
                // Prepare display string for PNP ID
                //

                StringCopyA (PnpOutput, e.szName);
                DecodePnpId (PnpOutput);

                //
                // Is PNP ID suppressed?  If so, continue memdb enum.
                //

                if (g_OverrideInf != INVALID_HANDLE_VALUE) {
                    if (SetupFindFirstLine (
                            g_OverrideInf,
                            "Standard PNP IDs",
                            PnpOutput,
                            &ic
                            )) {
                        if (g_ManualOverrideMode) {
                            printf ("%s\n", PnpOutput);
                        }
                        continue;
                    }
                }

                //
                // If only dumping manually overridden PNP IDs, continue.
                //

                if (g_ManualOverrideMode) {
                    continue;
                }

                //
                // Enumerate each description for the PNP ID
                //

                IsDir1 = (e.dwValue & MODE_DIR1) != 0;
                IsDir2 = (e.dwValue & MODE_DIR2) != 0;

                wsprintf (Node, TEXT("Devices\\%s\\*"), e.szName);

                if (MemDbEnumFirstValue (
                        &e2,
                        Node,
                        MEMDB_ALL_SUBLEVELS,
                        MEMDB_ENDPOINTS_ONLY
                        )) {

                    //
                    // Prepare display string for description by stripping off
                    // the sequencer and decoding it, then reset INF file name
                    // buffers.
                    //

                    StringCopyA (DescOutput, e2.szName);
                    *_mbschr (DescOutput, '\\') = 0;
                    DecodePnpId (DescOutput);

                    Dir1List.End = 0;
                    Dir2List.End = 0;

                    do {
                        //
                        // For each description, get the value of the string
                        // specified by the description sequencer's offset.
                        //
                        // We store the file names in a table, so we can organize
                        // matches correctly.  After the table is complete, we
                        // then dump it out.
                        //

                        MemDbBuildKeyFromOffset (e2.dwValue, Node, 1, &Flags);

                        if (((Flags & MustHave) || !MustHave) &&
                            ((Flags & MustNotHave) == 0)) {

                            Dir1Flag = (Flags & MODE_DIR1) != 0;
                            Dir2Flag = (Flags & MODE_DIR2) != 0;

                            if (Dir1Flag) {
                                MultiSzAppend (&Dir1List, Node);
                            }
                            if (Dir2Flag) {
                                MultiSzAppend (&Dir2List, Node);
                            }
                        }
                    } while (MemDbEnumNextValue (&e2));

                    MultiSzAppend (&Dir1List, "");
                    MultiSzAppend (&Dir2List, "");

                    //
                    // Dump all matches
                    //

                    Dir1File = (LPSTR) Dir1List.Buf;
                    if (!Dir1File) {
                        Dir1File = "";
                    }
                    Dir2File = (LPSTR) Dir2List.Buf;
                    if (!Dir2File) {
                        Dir2File = "";
                    }

                    while (*Dir1File || *Dir2File) {

                        printf (
                            "%s%c%s%c%s%c%s\n",
                            Dir1File,
                            g_Separators[0],
                            Dir2File,
                            g_Separators[1],
                            PnpOutput,
                            g_Separators[2],
                            DescOutput
                            );

                        if (*Dir1File) {
                            Dir1File = GetEndOfStringA (Dir1File) + 1;
                        }
                        if (*Dir2File) {
                            Dir2File = GetEndOfStringA (Dir2File) + 1;
                        }
                    }
                }
            }
        } while (MemDbEnumNextValue (&e));
    } else {
        printf ("No devices found.\n");
    }

    FreeGrowBuffer (&Dir1List);
    FreeGrowBuffer (&Dir2List);
}

int
__cdecl
main (
    int argc,
    char *argv[]
    )
{
    LPSTR Dir1InputPath;
    LPSTR Dir2InputPath;
    INT i, j;
    LONG rc;
    INT UIMode;
    BOOL AllFlag = FALSE;
    BOOL Dir2Only = FALSE;
    BOOL Dir1Only = FALSE;
    BOOL Common = FALSE;
    PCSTR OverrideList = NULL;
    DWORD Count;

    //
    // Init project globals
    //

    GetTempPathA (MAX_MBCHAR_PATH, g_TempDirBuf);
    g_TempDir = g_TempDirBuf;

    StringCopyA (g_TempDirWackBuf, g_TempDir);
    AppendWack (g_TempDirWackBuf);
    g_TempDirWackChars = CharCountA (g_TempDirWackBuf);
    g_TempDirWack = g_TempDirWackBuf;

    GetWindowsDirectoryA (g_WinDirBuf, MAX_MBCHAR_PATH);
    g_WinDir = g_WinDirBuf;

    ZeroMemory(&g_ConfigOptions,sizeof(USEROPTIONS));

    UIMode = PNPREPT_OUTPUT;
    g_Component = NULL;
    g_SubComponent = PNPREPT_HWND;

    //
    // Parse command line
    //

    Dir1InputPath = NULL;
    Dir2InputPath = NULL;

    for (i = 0 ; i < MAX_SEPARATORS ; i++) {
        g_Separators[i] = '\t';
    }

    for (i = 1 ; i < argc ; i++) {
        if (argv[i][0] == '-' || argv[i][0] == '/') {
            switch (tolower (argv[i][1])) {
            case 'a':
                AllFlag = TRUE;
                break;

            case '2':
                Dir2Only = TRUE;
                break;

            case '1':
                Dir1Only = TRUE;
                break;

            case 'c':
                Common = TRUE;
                break;

            case 'm':
                g_ManualOverrideMode = TRUE;
                break;

            case 'e':
                j = atoi (&argv[i][2]);
                if (j < 1 || j > MAX_SEPARATORS) {
                    HelpAndExit();
                }
                g_Separators[j] = '=';
                break;

            case 'i':
                if (!argv[i][2] && (i + 1) < argc) {
                    i++;
                    OverrideList = argv[i];
                } else if (argv[i][2] == ':') {
                    OverrideList = &argv[i][3];
                } else {
                    HelpAndExit();
                }
                break;

            default:
                HelpAndExit();
            }
        } else {
            if (Dir1InputPath && Dir2InputPath) {
                HelpAndExit();
            } else if (Dir1InputPath) {
                Dir2InputPath = argv[i];
            } else {
                Dir1InputPath = argv[i];
            }
        }
    }

    if (!AllFlag && !Dir2Only && !Dir1Only && !Common) {
        HelpAndExit();
    }

    if (g_ManualOverrideMode && !OverrideList) {
        HelpAndExit();
    }

    if (!Dir2InputPath) {
        HelpAndExit();
    }

    //
    // Init globals and libs
    //

    g_hHeap = GetProcessHeap();
    g_hInst = GetModuleHandle (NULL);

    g_SourceDirectories[0]   = Dir1InputPath;
    Count = 1;
    g_SourceDirectoryCount = Count;

    if (pCallMains (DLL_PROCESS_ATTACH)) {
        fprintf (stderr, "Initialization error!\n");
        return 254;
    }

    g_Dir1Path = _mbsrchr (Dir1InputPath, '\\');
    if (!g_Dir1Path) {
        g_Dir1Path = Dir1InputPath;
    } else {
        g_Dir2Path++;
    }

    g_Dir2Path = _mbsrchr (Dir2InputPath, '\\');
    if (!g_Dir2Path) {
        g_Dir2Path = Dir2InputPath;
    } else {
        g_Dir2Path++;
    }

    if (StringIMatch (g_Dir1Path, g_Dir2Path)) {
        g_Dir1Path = Dir1InputPath;
        g_Dir2Path = Dir2InputPath;
    }

    if (StringIMatch (g_Dir1Path, g_Dir2Path)) {
        fprintf (stderr, "Dir 1 and Dir 2 must be different\n");
        return 247;
    }

    if (OverrideList) {
        CHAR FullPath[MAX_MBCHAR_PATH];
        PSTR DontCare;

        if (!SearchPathA (NULL, OverrideList, NULL, MAX_MBCHAR_PATH, FullPath, &DontCare)) {
            StringCopyA (FullPath, OverrideList);
        }

        g_OverrideInf = SetupOpenInfFile (
                            FullPath,
                            NULL,
                            INF_STYLE_OLDNT|INF_STYLE_WIN4,
                            NULL
                            );
        if (g_OverrideInf == INVALID_HANDLE_VALUE) {
            fprintf (stderr, "Cannot open %s\n", FullPath);
            return 246;
        }
    }

    __try {
        //
        // Generate memdb entries for Dir 1 INFs
        //

        fprintf (stderr, "Processing...\n", Dir1InputPath);

        g_Mode = MODE_DIR1;
        if (!CreateNtHardwareList (&Dir1InputPath, 1, NULL, UIMode)) {
            rc = GetLastError();
            fprintf (stderr, "Could not build complete %s device list.  Win32 Error Code: %xh\n", Dir1InputPath, rc);
            return 1;
        } else {
            fprintf (stderr, "   %s processed\n", Dir1InputPath);
        }

        //
        // Restart hwcomp.lib
        //

        if (!HwComp_Entry (g_hInst, DLL_PROCESS_DETACH, NULL)) {
            fprintf (stderr, "Termination error!\n");
            return 253;
        }

        if (!HwComp_Entry (g_hInst, DLL_PROCESS_ATTACH, NULL)) {
            fprintf (stderr, "Initialization error!\n");
            return 252;
        }

        //
        // Generate memdb entries for Dir 2 INFs
        //

        g_Mode = MODE_DIR2;
        if (!CreateNtHardwareList (&Dir2InputPath, 1, NULL, UIMode)) {
            rc = GetLastError();
            fprintf (stderr, "Could not build complete %s device list.  Win32 Error Code: %xh\n", Dir2InputPath, rc);
            return 2;
        } else {
            fprintf (stderr, "   %s processed\n", Dir2InputPath);
        }

        //
        // Dump output
        //

        if (AllFlag) {
            Dump (0, 0);
        }

        if (Dir2Only) {
            Dump (MODE_DIR2, MODE_DIR1);
        }

        if (Dir1Only) {
            Dump (MODE_DIR1, MODE_DIR2);
        }

        if (Common) {
            Dump (MODE_DIR2|MODE_DIR1, 0);
        }

        //
        // Terminate hwcomp.lib
        //

        if (pCallMains (DLL_PROCESS_DETACH)) {
            fprintf (stderr, "Initialization error!\n");
            return 251;
        }

        fprintf (stderr, "Done\n");
    }

    __finally {
        if (g_OverrideInf != INVALID_HANDLE_VALUE) {
            SetupCloseInfFile (g_OverrideInf);
        }
    }

    return 0;
}

BOOL
ProcessPnpId (
    IN PCSTR  SubComponent
    )
{
    DWORD Value;
    TCHAR Node[MEMDB_MAX];
    LPTSTR PnpId, Desc, File;
    LPTSTR p;
    DWORD FileOffset;
    static DWORD d = 0;

    if (SubComponent) {
        PnpId = (LPTSTR) SubComponent;

        p = _tcschr (PnpId, TEXT('\\'));
        if (!p) {
            return TRUE;
        }
        *p = 0;
        Desc = p+1;

        p = _tcschr (Desc, TEXT('\\'));
        if (!p) {
            return TRUE;
        }
        *p = 0;
        File = p+1;

        //
        // Add the file (it may already exist) and remember the offset.
        // Keep the files in two separate lists.
        //

        MemDbSetValueEx (
            g_Mode == MODE_DIR1 ? TEXT("Dir1") : TEXT("Dir2"),
            File,
            NULL,
            NULL,
            g_Mode,
            &FileOffset
            );

        //
        // Add the PNP ID and OR the mode
        //

        wsprintf (Node, TEXT("Devices\\%s"), PnpId);
        if (!MemDbGetValue (Node, &Value)) {
            Value = 0;
        }

        Value |= g_Mode;

        MemDbSetValue (Node, Value);

        //
        // Add the description, and attach a sequencer to make sure
        // the description is unique.  Make the description point
        // to the file offset.
        //

        d++;
        wsprintf (Node, TEXT("Devices\\%s\\%s\\%u"), PnpId, Desc, d);
        MemDbSetValue (Node, FileOffset);
    }

    return TRUE;
}

//
// Stubs
//

HWND    g_Component;
HWND    g_SubComponent;
HANDLE  g_ComponentCancelEvent;
HANDLE  g_SubComponentCancelEvent;

VOID
pInitProgBarVars (
    VOID
    )
{
    g_Component = NULL;
    g_SubComponent = NULL;
    g_ComponentCancelEvent = CreateEvent (NULL, FALSE, FALSE, NULL);
    g_SubComponentCancelEvent = CreateEvent (NULL, FALSE, FALSE, NULL);
}


BOOL
ProgressBar_SetWindowStringA (
    IN HWND Window,
    IN HANDLE CancelEvent,
    IN PCSTR Message,            OPTIONAL
    IN DWORD MessageId           OPTIONAL
    )
{

    return TRUE;
}


BOOL
TickProgressBar (
    VOID
    )
{
    return TRUE;
}


BOOL
TickProgressBarDelta (
    IN      UINT TickCount
    )
{
    return TRUE;
}

VOID
InitializeProgressBar (
    IN      HWND ProgressBar,
    IN      HWND Component,             OPTIONAL
    IN      HWND SubComponent,          OPTIONAL
    IN      BOOL *CancelFlagPtr         OPTIONAL
    )
{
    return;
}

VOID
TerminateProgressBar (
    VOID
    )
{
    return;
}

VOID
EndSliceProcessing (
    VOID
    )
{
    return;
}

UINT
RegisterProgressBarSlice (
    IN      UINT InitialEstimate
    )
{
    return 0;
}


VOID
ReviseSliceEstimate (
    IN      UINT SliceId,
    IN      UINT RevisedEstimate
    )
{
    return;
}


VOID
BeginSliceProcessing (
    IN      UINT SliceId
    )
{
}