145 lines
4.2 KiB
C++
145 lines
4.2 KiB
C++
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Microsoft Windows
|
||
|
|
// Copyright (C) Microsoft Corporation, 1997.
|
||
|
|
//
|
||
|
|
// File: N C I N F . C P P
|
||
|
|
//
|
||
|
|
// Contents: ???
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
// Author: ???
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#include <pch.h>
|
||
|
|
#pragma hdrstop
|
||
|
|
#include "ncinf.h"
|
||
|
|
#include "ncsetup.h"
|
||
|
|
#include "ncstring.h"
|
||
|
|
|
||
|
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: HrProcessInfExtension
|
||
|
|
//
|
||
|
|
// Purpose: Given the appropriate keywords, returns information about an
|
||
|
|
// INF file that contains extended commands to add and remove
|
||
|
|
// fixed components such as WinSock or SNMP Agent support.
|
||
|
|
//
|
||
|
|
// Arguments:
|
||
|
|
// hinfInstallFile [in] The handle to the inf file to install
|
||
|
|
// from
|
||
|
|
// pszSectionName [in] The Base install section name.
|
||
|
|
// pszSuffix [in] Suffix to append to base. (i.e. "Winsock")
|
||
|
|
// pszAddLabel [in] Label for Add command (i.e. "AddSock")
|
||
|
|
// pszRemoveLabel [in] Label for Remove command (i.e. "DelSock")
|
||
|
|
// pfnHrAdd [in] Callback function to be called when adding.
|
||
|
|
// pfnHrRemove [in] Callback function to be called when removing.
|
||
|
|
//
|
||
|
|
// Returns: HRESULT, S_OK on success
|
||
|
|
//
|
||
|
|
// Author: danielwe 27 Apr 1997
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
HRESULT
|
||
|
|
HrProcessInfExtension (
|
||
|
|
IN HINF hinfInstallFile,
|
||
|
|
IN PCWSTR pszSectionName,
|
||
|
|
IN PCWSTR pszSuffix,
|
||
|
|
IN PCWSTR pszAddLabel,
|
||
|
|
IN PCWSTR pszRemoveLabel,
|
||
|
|
IN PFNADDCALLBACK pfnHrAdd,
|
||
|
|
IN PFNREMOVECALLBACK pfnHrRemove)
|
||
|
|
{
|
||
|
|
Assert(IsValidHandle(hinfInstallFile));
|
||
|
|
|
||
|
|
BOOL fAdd;
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
tstring strSectionName;
|
||
|
|
INFCONTEXT infContext;
|
||
|
|
WCHAR szCmd[LINE_LEN]; // LINE_LEN defined in setupapi.h as 256
|
||
|
|
|
||
|
|
// Construct the section name for which we're looking
|
||
|
|
// (ie. "Inst_Section.Winsock")
|
||
|
|
strSectionName = pszSectionName;
|
||
|
|
strSectionName += L".";
|
||
|
|
strSectionName += pszSuffix;
|
||
|
|
|
||
|
|
// Loop over the elements of the section and process the
|
||
|
|
// appropriate AddSock/DelSock sections found
|
||
|
|
hr = HrSetupFindFirstLine(hinfInstallFile, strSectionName.c_str(),
|
||
|
|
NULL, &infContext);
|
||
|
|
if (S_OK == hr)
|
||
|
|
{
|
||
|
|
tstring strName;
|
||
|
|
|
||
|
|
do
|
||
|
|
{
|
||
|
|
// Retrieve a line from the section, hopefully in the format:
|
||
|
|
// AddSock=section_name or DelSock=section_name
|
||
|
|
hr = HrSetupGetStringField(infContext, 0, szCmd, celems(szCmd),
|
||
|
|
NULL);
|
||
|
|
if (FAILED(hr))
|
||
|
|
{
|
||
|
|
goto Done;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check for the <add> or <remove> command
|
||
|
|
szCmd[celems(szCmd)-1] = L'\0';
|
||
|
|
if (!lstrcmpiW(szCmd, pszAddLabel))
|
||
|
|
{
|
||
|
|
fAdd = TRUE;
|
||
|
|
}
|
||
|
|
else if (!lstrcmpiW(szCmd, pszRemoveLabel))
|
||
|
|
{
|
||
|
|
fAdd = FALSE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
continue; // Other things are in this install section
|
||
|
|
}
|
||
|
|
|
||
|
|
// Query the Add/Remove value from the .inf
|
||
|
|
hr = HrSetupGetStringField(infContext, 1, &strName);
|
||
|
|
if (S_OK == hr)
|
||
|
|
{
|
||
|
|
if (fAdd)
|
||
|
|
{
|
||
|
|
// Call Add callback
|
||
|
|
hr = pfnHrAdd(hinfInstallFile, strName.c_str());
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Call remove callback
|
||
|
|
hr = pfnHrRemove(hinfInstallFile, strName.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
if (FAILED(hr))
|
||
|
|
{
|
||
|
|
goto Done;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
goto Done;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
while (S_OK == (hr = HrSetupFindNextLine(infContext, &infContext)));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (hr == S_FALSE)
|
||
|
|
{
|
||
|
|
// S_FALSE will terminate the loop successfully, so convert it to S_OK
|
||
|
|
// here.
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
Done:
|
||
|
|
TraceHr (ttidError, FAL, hr, (SPAPI_E_LINE_NOT_FOUND == hr),
|
||
|
|
"HrProcessInfExtension");
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|