2423 lines
62 KiB
C++
2423 lines
62 KiB
C++
//
|
|
// Copyright (C) 1997, Microsoft Corporation
|
|
//
|
|
// File: ftsup.cxx
|
|
//
|
|
// Contents: This module contains the functions which deal with ftdfs removal
|
|
//
|
|
// History: 07-Dec-1998 JHarper Created.
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
#include "headers.hxx"
|
|
#pragma hdrstop
|
|
|
|
#include "marshal.hxx"
|
|
#include "winldap.h"
|
|
#include "dsgetdc.h"
|
|
#include "setup.hxx"
|
|
#include "ftsup.hxx"
|
|
#include "dfsmwml.h"
|
|
|
|
#if DBG
|
|
|
|
void
|
|
DumpBuf(
|
|
PCHAR cp,
|
|
ULONG len);
|
|
|
|
#endif // DBG
|
|
|
|
extern PLDAP pLdapConnection;
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsGetDsBlob
|
|
//
|
|
// Synopsis: Reads a Dfs BLOB from the DS
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsGetDsBlob(
|
|
LPWSTR wszFtDfsName,
|
|
LPWSTR wszDcName,
|
|
ULONG *pcbBlob,
|
|
BYTE **ppBlob)
|
|
{
|
|
DWORD dwErr;
|
|
LDAP *pldap = NULL;
|
|
PLDAPMessage pMsg = NULL;
|
|
PLDAP_BERVAL *rgldapPktBlob = NULL;
|
|
PLDAP_BERVAL pldapPktBlob;
|
|
LDAPMessage *pmsgServers;
|
|
DWORD i;
|
|
WCHAR *wszConfigurationDN = NULL;
|
|
WCHAR wszDfsConfigDN[MAX_PATH+1];
|
|
LPWSTR rgAttrs[5];
|
|
BYTE *pBlob = NULL;
|
|
ULONG cLen = 0;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetDsBlob(%ws,%ws)\n", wszFtDfsName, wszDcName);
|
|
#endif
|
|
|
|
dwErr = DfspLdapOpen(wszDcName, &pldap, &wszConfigurationDN);
|
|
|
|
if (dwErr != ERROR_SUCCESS)
|
|
goto Cleanup;
|
|
|
|
//
|
|
// Build the entry name we want to search in
|
|
//
|
|
|
|
cLen = wcslen(L"CN=") +
|
|
wcslen(wszFtDfsName) +
|
|
wcslen(L",") +
|
|
wcslen(wszConfigurationDN) +
|
|
1;
|
|
|
|
if (cLen > MAX_PATH) {
|
|
dwErr = ERROR_DS_NAME_TOO_LONG;
|
|
goto Cleanup;
|
|
}
|
|
|
|
wcscpy(wszDfsConfigDN,L"CN=");
|
|
wcscat(wszDfsConfigDN,wszFtDfsName);
|
|
wcscat(wszDfsConfigDN,L",");
|
|
wcscat(wszDfsConfigDN,wszConfigurationDN);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("wszDfsConfigDN=[%ws]\n", wszDfsConfigDN);
|
|
#endif
|
|
|
|
rgAttrs[0] = L"pKT";
|
|
rgAttrs[1] = NULL;
|
|
|
|
if (DfsSvcLdap)
|
|
DbgPrint("DfsGetDsBlob:ldap_search_s(%ws)\n", rgAttrs[0]);
|
|
|
|
dwErr = ldap_search_sW(
|
|
pldap,
|
|
wszDfsConfigDN,
|
|
LDAP_SCOPE_BASE,
|
|
L"(objectClass=*)",
|
|
rgAttrs,
|
|
0,
|
|
&pMsg);
|
|
|
|
DFSM_TRACE_ERROR_HIGH(dwErr, ALL_ERROR,
|
|
DfsGetDsBlob_Error_ldap_search_sW,
|
|
LOGULONG(dwErr));
|
|
|
|
if (dwErr != LDAP_SUCCESS) {
|
|
dwErr = LdapMapErrorToWin32(dwErr);
|
|
goto Cleanup;
|
|
}
|
|
|
|
pmsgServers = ldap_first_entry(pldap, pMsg);
|
|
|
|
if (pmsgServers == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
rgldapPktBlob = ldap_get_values_lenW(
|
|
pldap,
|
|
pMsg,
|
|
L"pKT");
|
|
|
|
if (rgldapPktBlob == NULL || *rgldapPktBlob == NULL) {
|
|
dwErr = ERROR_INTERNAL_DB_CORRUPTION;
|
|
goto Cleanup;
|
|
}
|
|
|
|
pldapPktBlob = rgldapPktBlob[0];
|
|
|
|
pBlob = (BYTE *)malloc(pldapPktBlob->bv_len);
|
|
|
|
if (pBlob == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
RtlCopyMemory(pBlob, pldapPktBlob->bv_val, pldapPktBlob->bv_len);
|
|
|
|
*ppBlob = pBlob;
|
|
*pcbBlob = pldapPktBlob->bv_len;
|
|
|
|
Cleanup:
|
|
|
|
if (rgldapPktBlob != NULL)
|
|
ldap_value_free_len(rgldapPktBlob);
|
|
|
|
if (pMsg != NULL)
|
|
ldap_msgfree( pMsg );
|
|
|
|
if (pldap != NULL && pldap != pLdapConnection) {
|
|
if (DfsSvcLdap)
|
|
DbgPrint("DfsGetDsBlob:ldap_unbind()\n");
|
|
ldap_unbind( pldap );
|
|
}
|
|
|
|
if (wszConfigurationDN != NULL)
|
|
free(wszConfigurationDN);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetDsBlob returning %d\n", dwErr);
|
|
#endif
|
|
|
|
return (dwErr);
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsPutDsBlob
|
|
//
|
|
// Synopsis: Updates a Dfs BLOB in the DS
|
|
//
|
|
// History: 12/8/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsPutDsBlob(
|
|
LPWSTR wszFtDfsName,
|
|
LPWSTR wszDcName,
|
|
ULONG cbBlob,
|
|
BYTE *pBlob)
|
|
{
|
|
|
|
DWORD dwErr;
|
|
LDAP *pldap = NULL;
|
|
GUID idNewPkt;
|
|
LDAP_BERVAL ldapVal;
|
|
PLDAP_BERVAL rgldapVals[2];
|
|
LDAPModW ldapMod;
|
|
LDAP_BERVAL ldapIdVal;
|
|
PLDAP_BERVAL rgldapIdVals[2];
|
|
LDAPModW ldapIdMod;
|
|
PLDAPModW rgldapMods[3];
|
|
WCHAR *wszConfigurationDN = NULL;
|
|
WCHAR wszDfsConfigDN[MAX_PATH+1];
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsPutDsBlob(%ws,%ws)\n", wszFtDfsName, wszDcName);
|
|
#endif
|
|
|
|
dwErr = DfspLdapOpen(wszDcName, &pldap, &wszConfigurationDN);
|
|
|
|
if (dwErr != ERROR_SUCCESS)
|
|
goto Cleanup;
|
|
|
|
//
|
|
// Build the entry name
|
|
//
|
|
|
|
wcscpy(wszDfsConfigDN,L"CN=");
|
|
wcscat(wszDfsConfigDN,wszFtDfsName);
|
|
wcscat(wszDfsConfigDN,L",");
|
|
wcscat(wszDfsConfigDN,wszConfigurationDN);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("wszDfsConfigDN=[%ws]\n", wszDfsConfigDN);
|
|
#endif
|
|
|
|
UuidCreate( &idNewPkt );
|
|
|
|
ldapIdVal.bv_len = sizeof(GUID);
|
|
ldapIdVal.bv_val = (PCHAR) &idNewPkt;
|
|
rgldapIdVals[0] = &ldapIdVal;
|
|
rgldapIdVals[1] = NULL;
|
|
|
|
ldapIdMod.mod_op = LDAP_MOD_REPLACE | LDAP_MOD_BVALUES;
|
|
ldapIdMod.mod_type = L"pKTGuid";
|
|
ldapIdMod.mod_vals.modv_bvals = rgldapIdVals;
|
|
|
|
ldapVal.bv_len = cbBlob;
|
|
ldapVal.bv_val = (PCHAR) pBlob;
|
|
rgldapVals[0] = &ldapVal;
|
|
rgldapVals[1] = NULL;
|
|
|
|
ldapMod.mod_op = LDAP_MOD_REPLACE | LDAP_MOD_BVALUES;
|
|
ldapMod.mod_type = L"pKT";
|
|
ldapMod.mod_vals.modv_bvals = rgldapVals;
|
|
|
|
rgldapMods[0] = &ldapMod;
|
|
rgldapMods[1] = &ldapIdMod;
|
|
rgldapMods[2] = NULL;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("Writing BLOB of %d bytes\n", cbBlob);
|
|
#endif
|
|
|
|
if (DfsSvcLdap)
|
|
DbgPrint("DfsPutDsBlob:ldap_modify(%ws)\n", L"pKTGuid and pKT");
|
|
|
|
dwErr = ldap_modify_sW(
|
|
pldap,
|
|
wszDfsConfigDN,
|
|
rgldapMods);
|
|
|
|
DFSM_TRACE_ERROR_HIGH(dwErr, ALL_ERROR,
|
|
DfsPutDsBlob_Error_ldap_modify_sW,
|
|
LOGULONG(dwErr));
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose) {
|
|
DbgPrint("ldap_modify_sW returned %d\n", dwErr);
|
|
}
|
|
#endif
|
|
|
|
if (pldap != pLdapConnection) {
|
|
if (DfsSvcLdap)
|
|
DbgPrint("DfsPutDsBlob(1):ldap_unbind()\n");
|
|
ldap_unbind(pldap);
|
|
pldap = NULL;
|
|
}
|
|
|
|
if (dwErr != LDAP_SUCCESS) {
|
|
dwErr = LdapMapErrorToWin32(dwErr);
|
|
} else {
|
|
dwErr = ERROR_SUCCESS;
|
|
}
|
|
|
|
Cleanup:
|
|
|
|
if (pldap != NULL && pldap != pLdapConnection) {
|
|
if (DfsSvcLdap)
|
|
DbgPrint("DfsPutDsBlob(2):ldap_unbind()\n");
|
|
ldap_unbind( pldap );
|
|
}
|
|
|
|
if (wszConfigurationDN != NULL)
|
|
free(wszConfigurationDN);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsPutDsBlob returning %d\n", dwErr);
|
|
#endif
|
|
|
|
return (dwErr);
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsGetVolList
|
|
//
|
|
// Synopsis: Unserializes an FtDfs BLOB and creates
|
|
// a volume list representing an FtDfs.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
DWORD
|
|
DfsGetVolList(
|
|
ULONG cbBlob,
|
|
BYTE *pBlob,
|
|
PDFS_VOLUME_LIST pDfsVolList)
|
|
{
|
|
DWORD dwErr;
|
|
DWORD cObj;
|
|
DWORD cVol;
|
|
LDAP_PKT LdapPkt;
|
|
MARSHAL_BUFFER marshalBuffer;
|
|
NTSTATUS NtStatus;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetVolList()\n");
|
|
#endif
|
|
|
|
RtlZeroMemory(&LdapPkt, sizeof(LDAP_PKT));
|
|
|
|
if (cbBlob > sizeof(DWORD)) {
|
|
|
|
_GetULong(pBlob, pDfsVolList->Version);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose) {
|
|
DbgPrint("Blob Version = %d\n", pDfsVolList->Version);
|
|
DbgPrint("BLOB is %d bytes:\n", cbBlob);
|
|
}
|
|
#endif
|
|
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
cbBlob - sizeof(DWORD),
|
|
pBlob + sizeof(DWORD));
|
|
|
|
NtStatus = DfsRtlGet(&marshalBuffer, &MiLdapPkt, &LdapPkt);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose) {
|
|
|
|
DbgPrint(" %d objects found\n", LdapPkt.cLdapObjects);
|
|
|
|
for (cObj = 0; cObj < LdapPkt.cLdapObjects; cObj++) {
|
|
|
|
DbgPrint("%d:name=%ws size=%d p=0x%x\n",
|
|
cObj,
|
|
LdapPkt.rgldapObjects[cObj].wszObjectName,
|
|
LdapPkt.rgldapObjects[cObj].cbObjectData,
|
|
LdapPkt.rgldapObjects[cObj].pObjectData);
|
|
|
|
// DumpBuf(
|
|
// LdapPkt.rgldapObjects[cObj].pObjectData,
|
|
// LdapPkt.rgldapObjects[cObj].cbObjectData);
|
|
}
|
|
|
|
}
|
|
#endif
|
|
|
|
for (cObj = 0; cObj < LdapPkt.cLdapObjects; cObj++) {
|
|
|
|
if (wcscmp(LdapPkt.rgldapObjects[cObj].wszObjectName, L"\\siteroot") != 0) {
|
|
pDfsVolList->VolCount++;
|
|
}
|
|
|
|
}
|
|
|
|
pDfsVolList->Volumes = (PDFS_VOLUME) malloc(pDfsVolList->VolCount * sizeof(DFS_VOLUME));
|
|
|
|
if (pDfsVolList->Volumes == NULL) {
|
|
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
|
|
}
|
|
|
|
RtlZeroMemory(pDfsVolList->Volumes, pDfsVolList->VolCount * sizeof(DFS_VOLUME));
|
|
|
|
//
|
|
// Save the true/allocated size so to optimize deletions/additions
|
|
//
|
|
pDfsVolList->AllocatedVolCount = pDfsVolList->VolCount;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("===============================\n");
|
|
#endif
|
|
|
|
for (cVol = cObj = 0; cObj < LdapPkt.cLdapObjects; cObj++) {
|
|
|
|
if (wcscmp(LdapPkt.rgldapObjects[cObj].wszObjectName, L"\\siteroot") == 0) {
|
|
|
|
dwErr = DfsGetSiteTable(
|
|
pDfsVolList,
|
|
&LdapPkt.rgldapObjects[cObj]);
|
|
|
|
} else {
|
|
|
|
dwErr = DfsGetVolume(
|
|
&pDfsVolList->Volumes[cVol],
|
|
&LdapPkt.rgldapObjects[cObj]);
|
|
|
|
}
|
|
|
|
if (dwErr != ERROR_SUCCESS)
|
|
goto Cleanup;
|
|
|
|
cVol++;
|
|
|
|
}
|
|
|
|
} else if (cbBlob == sizeof(DWORD)) {
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("pKT BLOB is simply one DWORD\n");
|
|
#endif
|
|
dwErr = ERROR_INTERNAL_DB_CORRUPTION;
|
|
|
|
} else {
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("pKT BLOB is corrupt!\n");
|
|
#endif
|
|
dwErr = ERROR_INTERNAL_DB_CORRUPTION;
|
|
|
|
}
|
|
|
|
Cleanup:
|
|
|
|
FreeLdapPkt(&LdapPkt);
|
|
|
|
//
|
|
// Do any recovery needed
|
|
//
|
|
|
|
if(pDfsVolList && pDfsVolList->Volumes) {
|
|
DWORD dwErr2 = DfsRecoverVolList(pDfsVolList);
|
|
if(dwErr == ERROR_SUCCESS) {
|
|
// this is to prevent masking the previous error.
|
|
dwErr = dwErr2;
|
|
}
|
|
}
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetVolList returning %d\n", dwErr);
|
|
#endif
|
|
|
|
return( dwErr );
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsGetVolume
|
|
//
|
|
// Synopsis: Unserializes the data in a buffer/blob to a volume
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsGetVolume(
|
|
PDFS_VOLUME pVolume,
|
|
PLDAP_OBJECT pLdapObject)
|
|
{
|
|
DFS_VOLUME_PROPERTIES VolProps;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
MARSHAL_BUFFER marshalBuffer;
|
|
NTSTATUS NtStatus;
|
|
PBYTE pBuffer = NULL;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetVolume(%ws,%d)\n",
|
|
pLdapObject->wszObjectName,
|
|
pLdapObject->cbObjectData);
|
|
#endif
|
|
|
|
RtlZeroMemory(&VolProps, sizeof(DFS_VOLUME_PROPERTIES));
|
|
|
|
pVolume->wszObjectName = (WCHAR *) malloc(
|
|
(wcslen(pLdapObject->wszObjectName)+1) * sizeof(WCHAR));
|
|
|
|
if (pVolume->wszObjectName == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
wcscpy(
|
|
pVolume->wszObjectName,
|
|
pLdapObject->wszObjectName);
|
|
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
pLdapObject->cbObjectData,
|
|
pLdapObject->pObjectData);
|
|
|
|
NtStatus = DfsRtlGet(
|
|
&marshalBuffer,
|
|
&MiVolumeProperties,
|
|
&VolProps);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
VolProps.dwTimeout = 300;
|
|
|
|
if (
|
|
(marshalBuffer.Current < marshalBuffer.Last)
|
|
&&
|
|
(marshalBuffer.Last - marshalBuffer.Current) == sizeof(ULONG)
|
|
) {
|
|
|
|
DfsRtlGetUlong(&marshalBuffer, &VolProps.dwTimeout);
|
|
|
|
}
|
|
|
|
pVolume->idVolume = VolProps.idVolume;
|
|
pVolume->wszPrefix = VolProps.wszPrefix;
|
|
pVolume->wszShortPrefix = VolProps.wszShortPrefix;
|
|
pVolume->dwType = VolProps.dwType;
|
|
pVolume->dwState = VolProps.dwState;
|
|
pVolume->wszComment = VolProps.wszComment;
|
|
pVolume->dwTimeout = VolProps.dwTimeout;
|
|
pVolume->ftPrefix = VolProps.ftPrefix;
|
|
pVolume->ftState = VolProps.ftState;
|
|
pVolume->ftComment = VolProps.ftComment;
|
|
pVolume->dwVersion = VolProps.dwVersion;
|
|
pVolume->cbRecovery = VolProps.cbRecovery;
|
|
pVolume->pRecovery = VolProps.pRecovery;
|
|
|
|
pBuffer = VolProps.pSvc;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("VolProps.cbSvc = %d\n", VolProps.cbSvc);
|
|
#endif
|
|
|
|
dwErr = UnSerializeReplicaList(
|
|
&pVolume->ReplCount,
|
|
&pVolume->AllocatedReplCount,
|
|
&pVolume->ReplicaInfo,
|
|
&pVolume->FtModification,
|
|
&pBuffer);
|
|
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
goto Cleanup;
|
|
}
|
|
|
|
//
|
|
// Get deleted replicas
|
|
//
|
|
|
|
if (pBuffer < (pBuffer + VolProps.cbSvc)) {
|
|
|
|
dwErr = UnSerializeReplicaList(
|
|
&pVolume->DelReplCount,
|
|
&pVolume->AllocatedDelReplCount,
|
|
&pVolume->DelReplicaInfo,
|
|
&pVolume->DelFtModification,
|
|
&pBuffer);
|
|
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
goto Cleanup;
|
|
}
|
|
|
|
}
|
|
|
|
Cleanup:
|
|
|
|
if (VolProps.pSvc != NULL)
|
|
MarshalBufferFree(VolProps.pSvc);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetVolume returning %d\n", dwErr);
|
|
#endif
|
|
|
|
return dwErr;
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsFreeVolList
|
|
//
|
|
// Synopsis: Frees the volume list and associated substructures representing
|
|
// an FtDfs.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
VOID
|
|
DfsFreeVolList(
|
|
PDFS_VOLUME_LIST pDfsVolList)
|
|
{
|
|
ULONG cVol;
|
|
ULONG cSite;
|
|
ULONG i;
|
|
PLIST_ENTRY pListHead;
|
|
PDFSM_SITE_ENTRY pSiteEntry;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsFreeVolList()\n");
|
|
#endif
|
|
|
|
if (pDfsVolList->VolCount > 0 && pDfsVolList->Volumes) {
|
|
for (cVol = 0; cVol < pDfsVolList->VolCount; cVol++) {
|
|
DfsFreeVol(&pDfsVolList->Volumes[cVol]);
|
|
}
|
|
}
|
|
|
|
if (pDfsVolList->Volumes != NULL)
|
|
free(pDfsVolList->Volumes);
|
|
|
|
pListHead = &pDfsVolList->SiteList;
|
|
|
|
if (pListHead->Flink != NULL) {
|
|
|
|
while (pListHead->Flink != pListHead) {
|
|
pSiteEntry = CONTAINING_RECORD(pListHead->Flink, DFSM_SITE_ENTRY, Link);
|
|
RemoveEntryList(pListHead->Flink);
|
|
for (i = 0; i < pSiteEntry->Info.cSites; i++) {
|
|
if (pSiteEntry->Info.Site[i].SiteName != NULL)
|
|
MarshalBufferFree(pSiteEntry->Info.Site[i].SiteName);
|
|
}
|
|
if (pSiteEntry->ServerName != NULL)
|
|
MarshalBufferFree(pSiteEntry->ServerName);
|
|
free(pSiteEntry);
|
|
}
|
|
|
|
}
|
|
|
|
RtlZeroMemory(pDfsVolList, sizeof(DFS_VOLUME_LIST));
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsFreeVolList exit\n");
|
|
#endif
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsFreeVol
|
|
//
|
|
// Synopsis: Frees the volume and associated substructures representing
|
|
// an FtDfs volume
|
|
//
|
|
// History: 12/16/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
VOID
|
|
DfsFreeVol(
|
|
PDFS_VOLUME pVol)
|
|
{
|
|
ULONG cRepl;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose & 0x80000000)
|
|
DbgPrint("DfsFreeVol()\n");
|
|
#endif
|
|
|
|
if (pVol->ReplCount > 0 && pVol->ReplicaInfo != NULL) {
|
|
for (cRepl = 0; cRepl < pVol->ReplCount; cRepl++) {
|
|
DfsFreeRepl(&pVol->ReplicaInfo[cRepl]);
|
|
}
|
|
}
|
|
|
|
if (pVol->DelReplCount > 0 && pVol->DelReplicaInfo != NULL) {
|
|
for (cRepl = 0; cRepl < pVol->DelReplCount; cRepl++) {
|
|
DfsFreeRepl(&pVol->DelReplicaInfo[cRepl]);
|
|
}
|
|
}
|
|
|
|
if (pVol->wszPrefix != NULL)
|
|
MarshalBufferFree(pVol->wszPrefix);
|
|
if (pVol->wszShortPrefix != NULL)
|
|
MarshalBufferFree(pVol->wszShortPrefix);
|
|
if (pVol->wszComment != NULL)
|
|
MarshalBufferFree(pVol->wszComment);
|
|
if (pVol->pRecovery != NULL)
|
|
MarshalBufferFree(pVol->pRecovery);
|
|
|
|
if (pVol->ReplicaInfo != NULL)
|
|
free(pVol->ReplicaInfo);
|
|
if (pVol->DelReplicaInfo != NULL)
|
|
free(pVol->DelReplicaInfo);
|
|
if (pVol->FtModification != NULL)
|
|
free(pVol->FtModification);
|
|
if (pVol->DelFtModification != NULL)
|
|
free(pVol->DelFtModification);
|
|
if (pVol->wszObjectName != NULL)
|
|
free(pVol->wszObjectName);
|
|
|
|
RtlZeroMemory(pVol, sizeof(DFS_VOLUME));
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose & 0x80000000)
|
|
DbgPrint("DfsFreeVol exit\n");
|
|
#endif
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsFreeRepl
|
|
//
|
|
// Synopsis: Frees the Replica and associated substructures representing
|
|
// an FtDfs replica
|
|
//
|
|
// History: 12/16/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
VOID
|
|
DfsFreeRepl(
|
|
PDFS_REPLICA_INFO pRepl)
|
|
{
|
|
#if DBG
|
|
if (DfsSvcVerbose & 0x80000000)
|
|
DbgPrint("DfsFreeRepl()\n");
|
|
#endif
|
|
|
|
if (pRepl->pwszServerName != NULL)
|
|
MarshalBufferFree(pRepl->pwszServerName);
|
|
|
|
if (pRepl->pwszShareName != NULL)
|
|
MarshalBufferFree(pRepl->pwszShareName);
|
|
|
|
RtlZeroMemory(pRepl, sizeof(DFS_REPLICA_INFO));
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose & 0x80000000)
|
|
DbgPrint("DfsFreeRepl exit\n");
|
|
#endif
|
|
}
|
|
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsPutVolList
|
|
//
|
|
// Synopsis: Serializes the structs representing a Dfs volume and
|
|
// creates a BLOB.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsPutVolList(
|
|
ULONG *pcbBlob,
|
|
BYTE **ppBlob,
|
|
PDFS_VOLUME_LIST pDfsVolList)
|
|
{
|
|
ULONG cVol;
|
|
ULONG cRepl;
|
|
ULONG cObj;
|
|
ULONG cBuffer;
|
|
ULONG cSite;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
PBYTE Buffer = NULL;
|
|
MARSHAL_BUFFER marshalBuffer;
|
|
NTSTATUS NtStatus;
|
|
DFS_VOLUME_PROPERTIES VolProps;
|
|
LDAP_PKT LdapPkt;
|
|
PLIST_ENTRY pListHead;
|
|
PLIST_ENTRY pLink;
|
|
PDFSM_SITE_ENTRY pSiteEntry;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsPutVolList()\n");
|
|
#endif
|
|
|
|
LdapPkt.cLdapObjects = pDfsVolList->VolCount + 1;
|
|
LdapPkt.rgldapObjects = (PLDAP_OBJECT) malloc(LdapPkt.cLdapObjects * sizeof(LDAP_OBJECT));
|
|
|
|
if (LdapPkt.rgldapObjects == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
RtlZeroMemory(LdapPkt.rgldapObjects, LdapPkt.cLdapObjects * sizeof(LDAP_OBJECT));
|
|
|
|
//
|
|
// For each volume, serialize the replicas, then the volume
|
|
//
|
|
|
|
for (cVol = 0; cVol < pDfsVolList->VolCount; cVol++) {
|
|
|
|
RtlZeroMemory(&VolProps, sizeof(DFS_VOLUME_PROPERTIES));
|
|
|
|
LdapPkt.rgldapObjects[cVol].wszObjectName = (WCHAR *) MarshalBufferAllocate(
|
|
|
|
(wcslen(pDfsVolList->Volumes[cVol].wszObjectName)+1) * sizeof(WCHAR));
|
|
|
|
if (LdapPkt.rgldapObjects[cVol].wszObjectName == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
wcscpy(LdapPkt.rgldapObjects[cVol].wszObjectName, pDfsVolList->Volumes[cVol].wszObjectName);
|
|
|
|
//
|
|
// Serialize the replicas
|
|
//
|
|
|
|
dwErr = SerializeReplicaList(
|
|
pDfsVolList->Volumes[cVol].ReplCount,
|
|
pDfsVolList->Volumes[cVol].ReplicaInfo,
|
|
pDfsVolList->Volumes[cVol].FtModification,
|
|
pDfsVolList->Volumes[cVol].DelReplCount,
|
|
pDfsVolList->Volumes[cVol].DelReplicaInfo,
|
|
pDfsVolList->Volumes[cVol].DelFtModification,
|
|
&VolProps.cbSvc,
|
|
&VolProps.pSvc);
|
|
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint(" cbSvc = %d\n", VolProps.cbSvc);
|
|
#endif
|
|
|
|
VolProps.idVolume = pDfsVolList->Volumes[cVol].idVolume;
|
|
VolProps.wszPrefix = pDfsVolList->Volumes[cVol].wszPrefix;
|
|
VolProps.wszShortPrefix = pDfsVolList->Volumes[cVol].wszShortPrefix;
|
|
VolProps.dwType = pDfsVolList->Volumes[cVol].dwType;
|
|
VolProps.dwState = pDfsVolList->Volumes[cVol].dwState;
|
|
VolProps.wszComment = pDfsVolList->Volumes[cVol].wszComment;
|
|
VolProps.dwTimeout = pDfsVolList->Volumes[cVol].dwTimeout;
|
|
VolProps.ftPrefix = pDfsVolList->Volumes[cVol].ftPrefix;
|
|
VolProps.ftState = pDfsVolList->Volumes[cVol].ftState;
|
|
VolProps.ftComment = pDfsVolList->Volumes[cVol].ftComment;
|
|
VolProps.dwVersion = pDfsVolList->Volumes[cVol].dwVersion;
|
|
VolProps.cbRecovery = pDfsVolList->Volumes[cVol].cbRecovery;
|
|
VolProps.pRecovery = pDfsVolList->Volumes[cVol].pRecovery;
|
|
|
|
//
|
|
// Now serialize the volume
|
|
|
|
cBuffer = 0;
|
|
NtStatus = DfsRtlSize(&MiVolumeProperties, &VolProps, &cBuffer);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
MarshalBufferFree(VolProps.pSvc);
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
cBuffer += sizeof(ULONG);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("VolProps marshaled size = %d\n", cBuffer);
|
|
#endif
|
|
|
|
Buffer = (PBYTE) MarshalBufferAllocate(cBuffer);
|
|
|
|
if (Buffer == NULL) {
|
|
MarshalBufferFree(VolProps.pSvc);
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
LdapPkt.rgldapObjects[cVol].pObjectData = (PCHAR) Buffer;
|
|
LdapPkt.rgldapObjects[cVol].cbObjectData = cBuffer;
|
|
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
cBuffer,
|
|
Buffer);
|
|
|
|
NtStatus = DfsRtlPut(&marshalBuffer, &MiVolumeProperties, &VolProps);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
MarshalBufferFree(VolProps.pSvc);
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
NtStatus = DfsRtlPutUlong(&marshalBuffer, &VolProps.dwTimeout);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
MarshalBufferFree(VolProps.pSvc);
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
MarshalBufferFree(VolProps.pSvc);
|
|
|
|
}
|
|
|
|
//
|
|
// Serialize the site table
|
|
//
|
|
|
|
cBuffer = sizeof(ULONG) + sizeof(GUID);
|
|
|
|
//
|
|
// Add the size of all the entries
|
|
//
|
|
|
|
pListHead = &pDfsVolList->SiteList;
|
|
|
|
for (pLink = pListHead->Flink; pLink != pListHead; pLink = pLink->Flink) {
|
|
pSiteEntry = CONTAINING_RECORD(pLink, DFSM_SITE_ENTRY, Link);
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("pSiteEntry for %ws\n", pSiteEntry->ServerName);
|
|
#endif
|
|
NtStatus = DfsRtlSize(&MiDfsmSiteEntry, pSiteEntry, &cBuffer);
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Get a buffer big enough
|
|
//
|
|
|
|
Buffer = (BYTE *) MarshalBufferAllocate(cBuffer);
|
|
|
|
if (Buffer == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
LdapPkt.rgldapObjects[pDfsVolList->VolCount].wszObjectName =
|
|
(WCHAR *)MarshalBufferAllocate(sizeof(L"\\siteroot"));
|
|
if (LdapPkt.rgldapObjects[pDfsVolList->VolCount].wszObjectName == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
wcscpy(LdapPkt.rgldapObjects[pDfsVolList->VolCount].wszObjectName, L"\\siteroot");
|
|
LdapPkt.rgldapObjects[pDfsVolList->VolCount].pObjectData = (PCHAR) Buffer;
|
|
LdapPkt.rgldapObjects[pDfsVolList->VolCount].cbObjectData = cBuffer;
|
|
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
cBuffer,
|
|
Buffer);
|
|
|
|
//
|
|
// Put the guid, then the object count in the beginning of the buffer
|
|
//
|
|
|
|
DfsRtlPutGuid(&marshalBuffer, &pDfsVolList->SiteGuid);
|
|
DfsRtlPutUlong(&marshalBuffer, &pDfsVolList->SiteCount);
|
|
|
|
for (pLink = pListHead->Flink; pLink != pListHead; pLink = pLink->Flink) {
|
|
pSiteEntry = CONTAINING_RECORD(pLink, DFSM_SITE_ENTRY, Link);
|
|
DfsRtlPut(&marshalBuffer,&MiDfsmSiteEntry, pSiteEntry);
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
}
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose) {
|
|
|
|
DbgPrint("After reserialization, %d objects found\n", LdapPkt.cLdapObjects);
|
|
|
|
for (cObj = 0; cObj < LdapPkt.cLdapObjects; cObj++) {
|
|
|
|
DbgPrint("%d:name=%ws size=%d p=0x%x\n",
|
|
cObj,
|
|
LdapPkt.rgldapObjects[cObj].wszObjectName,
|
|
LdapPkt.rgldapObjects[cObj].cbObjectData,
|
|
LdapPkt.rgldapObjects[cObj].pObjectData);
|
|
|
|
// DumpBuf(
|
|
// LdapPkt.rgldapObjects[cObj].pObjectData,
|
|
// LdapPkt.rgldapObjects[cObj].cbObjectData);
|
|
}
|
|
|
|
}
|
|
#endif
|
|
|
|
//
|
|
// Finally, serialize all the volumes and the site table into a blob
|
|
//
|
|
|
|
cBuffer = 0;
|
|
NtStatus = DfsRtlSize(&MiLdapPkt, &LdapPkt, &cBuffer);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(dwErr);
|
|
goto Cleanup;
|
|
}
|
|
|
|
cBuffer += sizeof(DWORD);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("New ldap size = %d\n", cBuffer);
|
|
#endif
|
|
|
|
Buffer = (PBYTE) malloc(cBuffer);
|
|
|
|
if (Buffer == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
// 447486. Fix prefix bug. use memory after checking for null
|
|
|
|
*((PDWORD) Buffer) = 2; // Version #
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
cBuffer,
|
|
Buffer + sizeof(DWORD));
|
|
|
|
NtStatus = DfsRtlPut(&marshalBuffer, &MiLdapPkt, &LdapPkt);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(dwErr);
|
|
goto Cleanup;
|
|
}
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("Remarshal succeeded cBuffer = %d\n", cBuffer);
|
|
#endif
|
|
|
|
*pcbBlob = cBuffer;
|
|
*ppBlob = Buffer;
|
|
Buffer = NULL;
|
|
|
|
Cleanup:
|
|
|
|
FreeLdapPkt(&LdapPkt);
|
|
|
|
if (dwErr != ERROR_SUCCESS && Buffer != NULL)
|
|
free(Buffer);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsPutVolList exit %d\n", dwErr);
|
|
#endif
|
|
|
|
return dwErr;
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: FreeLdapPkt
|
|
//
|
|
// Synopsis: Frees an LDAP_PKT structure and all substructures.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
VOID
|
|
FreeLdapPkt(
|
|
LDAP_PKT *pLdapPkt)
|
|
{
|
|
ULONG cObj;
|
|
|
|
if (pLdapPkt->rgldapObjects != NULL) {
|
|
|
|
for (cObj = 0; cObj < pLdapPkt->cLdapObjects; cObj++) {
|
|
if (pLdapPkt->rgldapObjects[cObj].wszObjectName != NULL)
|
|
MarshalBufferFree(pLdapPkt->rgldapObjects[cObj].wszObjectName);
|
|
if (pLdapPkt->rgldapObjects[cObj].pObjectData != NULL)
|
|
MarshalBufferFree(pLdapPkt->rgldapObjects[cObj].pObjectData);
|
|
}
|
|
MarshalBufferFree(pLdapPkt->rgldapObjects);
|
|
}
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: SerializeReplicaList
|
|
//
|
|
// Synopsis: This method serializes the replica info list and the
|
|
// deleted replica info list into a buffer.
|
|
//
|
|
// Returns: [ERROR_SUCCESS] -- If all went well.
|
|
//
|
|
// [ERROR_OUTOFMEMORY] - If unable to allocate the target buffer.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
DWORD
|
|
SerializeReplicaList(
|
|
ULONG ReplCount,
|
|
DFS_REPLICA_INFO *pReplicaInfo,
|
|
FILETIME *pFtModification,
|
|
ULONG DelReplCount,
|
|
DFS_REPLICA_INFO *pDelReplicaInfo,
|
|
FILETIME *pDelFtModification,
|
|
ULONG *cBuffer,
|
|
PBYTE *ppBuffer)
|
|
{
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
ULONG TotalSize;
|
|
ULONG i;
|
|
ULONG *pSize;
|
|
BYTE *Buffer = NULL;
|
|
ULONG *SizeBuffer = NULL;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("SerializeReplicaList(%d,%d)\n", ReplCount, DelReplCount);
|
|
#endif
|
|
|
|
pSize = SizeBuffer = (PULONG) malloc(sizeof(ULONG) * (ReplCount + DelReplCount));
|
|
|
|
if (SizeBuffer == NULL) {
|
|
|
|
return ERROR_OUTOFMEMORY;
|
|
|
|
}
|
|
|
|
//
|
|
// Need all the size values now and later for marshalling stuff.
|
|
// So we collect them here into an array.
|
|
//
|
|
|
|
TotalSize = 0;
|
|
pSize = SizeBuffer;
|
|
|
|
for (i = 0; i < ReplCount; i++) {
|
|
|
|
*pSize = GetReplicaMarshalSize(&pReplicaInfo[i], &pFtModification[i]);
|
|
TotalSize += *pSize;
|
|
pSize++;
|
|
|
|
}
|
|
|
|
for (i = 0; i < DelReplCount; i++) {
|
|
|
|
*pSize = GetReplicaMarshalSize(&pDelReplicaInfo[i], &pDelFtModification[i]);
|
|
TotalSize += *pSize;
|
|
pSize++;
|
|
|
|
}
|
|
|
|
//
|
|
// Allocate the byte Buffer we need to pass back
|
|
//
|
|
// TotalSize is the size required just to marshal all the replicas and
|
|
// their last-modification-timestamps.
|
|
//
|
|
// In addition, we need:
|
|
//
|
|
// 1 ULONG for storing the count of replicas
|
|
// ReplCount ULONGs for storing the marshal size of each replica.
|
|
// 1 ULONG for count of deleted replicas
|
|
// DelReplCount ULONGS for storing the marshal size of each deleted Repl
|
|
//
|
|
|
|
//
|
|
// First calc the size of the Buffer.
|
|
//
|
|
|
|
TotalSize += sizeof(ULONG) * (1 + ReplCount + 1 + DelReplCount);
|
|
|
|
*ppBuffer = Buffer = (PBYTE) malloc(TotalSize);
|
|
|
|
if (Buffer == NULL) {
|
|
free(SizeBuffer);
|
|
return ERROR_OUTOFMEMORY;
|
|
}
|
|
|
|
//
|
|
// Set the number of entries to follow in the Buffer at the start.
|
|
//
|
|
|
|
_PutULong(Buffer, ReplCount);
|
|
Buffer += sizeof(ULONG);
|
|
|
|
pSize = SizeBuffer;
|
|
for (i = 0; i < ReplCount; i++) {
|
|
|
|
//
|
|
// Marshall each replica Entry into the Buffer.
|
|
// Remember we first need to put the size of the marshalled
|
|
// replica entry to follow, then the FILETIME for the replica,
|
|
// and finally, the marshalled replica entry structure.
|
|
//
|
|
|
|
_PutULong(Buffer, *pSize);
|
|
Buffer += sizeof(ULONG);
|
|
dwErr = SerializeReplica(&pReplicaInfo[i], &pFtModification[i], Buffer, *pSize);
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
free(*ppBuffer);
|
|
return dwErr;
|
|
}
|
|
Buffer += *pSize;
|
|
pSize++;
|
|
|
|
}
|
|
|
|
//
|
|
// Now marshal the deleted Repl list.
|
|
//
|
|
|
|
_PutULong(Buffer, DelReplCount);
|
|
Buffer += sizeof(ULONG);
|
|
|
|
for (i = 0; i < DelReplCount; i++) {
|
|
|
|
_PutULong(Buffer, *pSize);
|
|
Buffer += sizeof(ULONG);
|
|
dwErr = SerializeReplica(&pDelReplicaInfo[i], &pDelFtModification[i], Buffer, *pSize);
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
free(*ppBuffer);
|
|
return dwErr;
|
|
}
|
|
Buffer += *pSize;
|
|
pSize++;
|
|
}
|
|
|
|
*cBuffer = TotalSize;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("SerializeReplicaList exit %d\n", dwErr);
|
|
#endif
|
|
|
|
return( dwErr );
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Member: SerializeReplica
|
|
//
|
|
// Synopsis: Serializes a replica info structure
|
|
//
|
|
// Notes: The size of the buffer should have been calculated using
|
|
// the function GetReplicaMarshalSize()
|
|
//
|
|
// History: 13-May-93 SudK Created.
|
|
// 19-Nov-98 Jharper Modified.
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
DWORD
|
|
SerializeReplica(
|
|
DFS_REPLICA_INFO *pDfsReplicaInfo,
|
|
FILETIME *pFtModification,
|
|
PBYTE Buffer,
|
|
ULONG Size)
|
|
{
|
|
DWORD dwErr;
|
|
MARSHAL_BUFFER MarshalBuffer;
|
|
NTSTATUS NtStatus;
|
|
|
|
MarshalBufferInitialize(&MarshalBuffer, Size, Buffer);
|
|
NtStatus = DfsRtlPut(&MarshalBuffer, &MiFileTime, pFtModification);
|
|
|
|
if (NT_SUCCESS(NtStatus))
|
|
NtStatus = DfsRtlPut(&MarshalBuffer, &MiDfsReplicaInfo, pDfsReplicaInfo);
|
|
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
|
|
return dwErr;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------
|
|
//
|
|
// Member: GetReplicaMarshalSize, public
|
|
//
|
|
// Synopsis: Returns the size of a buffer required to marshal this
|
|
// replica.
|
|
//
|
|
// History: 12-May-93 SudK Created.
|
|
// 19-Nov-98 Jharper Modified.
|
|
//
|
|
//-----------------------------------------------------------------------
|
|
|
|
ULONG
|
|
GetReplicaMarshalSize(
|
|
DFS_REPLICA_INFO *pDfsReplicaInfo,
|
|
FILETIME *pFtModification)
|
|
{
|
|
ULONG Size = 0;
|
|
NTSTATUS NtStatus;
|
|
|
|
NtStatus = DfsRtlSize(&MiFileTime, pFtModification, &Size);
|
|
|
|
if (NT_SUCCESS(NtStatus))
|
|
NtStatus = DfsRtlSize(&MiDfsReplicaInfo, pDfsReplicaInfo, &Size);
|
|
|
|
return(Size);
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Member: UnSerializeReplicaList
|
|
//
|
|
// Synopsis: Unserializes a buffer into a relica list.
|
|
//
|
|
// History: 20-Nov-98 Jharper created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
UnSerializeReplicaList(
|
|
ULONG *pReplCount,
|
|
ULONG *pAllocatedReplCount,
|
|
DFS_REPLICA_INFO **ppReplicaInfo,
|
|
FILETIME **ppFtModification,
|
|
BYTE **ppBuffer)
|
|
{
|
|
|
|
DFS_REPLICA_INFO *pReplicaInfo = NULL;
|
|
FILETIME *pFtModification = NULL;
|
|
ULONG ReplCount;
|
|
BYTE *pBuffer = *ppBuffer;
|
|
ULONG cRepl;
|
|
ULONG Size;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
MARSHAL_BUFFER marshalBuffer;
|
|
NTSTATUS NtStatus;
|
|
|
|
//
|
|
// Get # replicas
|
|
//
|
|
|
|
_GetULong(pBuffer, ReplCount);
|
|
pBuffer += sizeof(ULONG);
|
|
|
|
pReplicaInfo = (DFS_REPLICA_INFO *) malloc(sizeof(DFS_REPLICA_INFO) * ReplCount);
|
|
|
|
pFtModification = (FILETIME *) malloc(sizeof(FILETIME) * ReplCount);
|
|
|
|
if(pReplicaInfo == NULL || pFtModification == NULL) {
|
|
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
|
|
}
|
|
|
|
//
|
|
// Now get each replica
|
|
//
|
|
|
|
for (cRepl = 0; cRepl < ReplCount; cRepl++) {
|
|
|
|
_GetULong(pBuffer, Size);
|
|
pBuffer += sizeof(ULONG);
|
|
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
Size,
|
|
pBuffer);
|
|
|
|
NtStatus = DfsRtlGet(
|
|
&marshalBuffer,
|
|
&MiFileTime,
|
|
&pFtModification[cRepl]);
|
|
|
|
if (NT_SUCCESS(NtStatus)) {
|
|
|
|
NtStatus = DfsRtlGet(
|
|
&marshalBuffer,
|
|
&MiDfsReplicaInfo,
|
|
&pReplicaInfo[cRepl]);
|
|
|
|
}
|
|
|
|
if (NT_SUCCESS(NtStatus)) {
|
|
|
|
pBuffer += Size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*ppReplicaInfo = pReplicaInfo;
|
|
*ppFtModification = pFtModification;
|
|
*ppBuffer = pBuffer;
|
|
*pAllocatedReplCount = *pReplCount = ReplCount;
|
|
|
|
Cleanup:
|
|
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
if (pReplicaInfo != NULL)
|
|
free(pReplicaInfo);
|
|
if (pFtModification != NULL)
|
|
free(pFtModification);
|
|
}
|
|
|
|
return dwErr;
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsGetSiteTable
|
|
//
|
|
// Synopsis: Unserializes the buffer passed in into a dfs site table.
|
|
//
|
|
// Returns: [ERROR_SUCCESS] -- If all went well.
|
|
//
|
|
// [ERROR_OUTOFMEMORY] - If unable to allocate the target buffer.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsGetSiteTable(
|
|
PDFS_VOLUME_LIST pDfsVolList,
|
|
PLDAP_OBJECT pLdapObject)
|
|
{
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
NTSTATUS NtStatus;
|
|
ULONG cSite;
|
|
PDFSM_SITE_ENTRY pSiteEntry;
|
|
PDFSM_SITE_ENTRY pTmpSiteEntry;
|
|
MARSHAL_BUFFER marshalBuffer;
|
|
BYTE *pBuffer = NULL;
|
|
ULONG Size;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetSiteTable(%d)\n", pLdapObject->cbObjectData);
|
|
#endif
|
|
|
|
InitializeListHead(&pDfsVolList->SiteList);
|
|
|
|
MarshalBufferInitialize(
|
|
&marshalBuffer,
|
|
pLdapObject->cbObjectData,
|
|
pLdapObject->pObjectData);
|
|
|
|
NtStatus = DfsRtlGetGuid(&marshalBuffer, &pDfsVolList->SiteGuid);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
NtStatus = DfsRtlGetUlong(&marshalBuffer, &pDfsVolList->SiteCount);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
pBuffer = (BYTE *)malloc(pLdapObject->cbObjectData);
|
|
|
|
if (pBuffer == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
pTmpSiteEntry = (PDFSM_SITE_ENTRY)pBuffer;
|
|
|
|
for (cSite = 0; cSite < pDfsVolList->SiteCount; cSite++) {
|
|
|
|
RtlZeroMemory(pBuffer, pLdapObject->cbObjectData);
|
|
|
|
NtStatus = DfsRtlGet(
|
|
&marshalBuffer,
|
|
&MiDfsmSiteEntry,
|
|
pBuffer);
|
|
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
dwErr = RtlNtStatusToDosError(NtStatus);
|
|
goto Cleanup;
|
|
}
|
|
|
|
Size = sizeof(DFSM_SITE_ENTRY) + (pTmpSiteEntry->Info.cSites * sizeof(DFS_SITENAME_INFO));
|
|
|
|
pSiteEntry = (PDFSM_SITE_ENTRY) malloc(Size);
|
|
|
|
if (pSiteEntry == NULL) {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto Cleanup;
|
|
}
|
|
|
|
RtlCopyMemory(pSiteEntry, pBuffer, Size);
|
|
InsertHeadList(&pDfsVolList->SiteList, &pSiteEntry->Link);
|
|
|
|
}
|
|
|
|
Cleanup:
|
|
|
|
if (pBuffer != NULL)
|
|
free(pBuffer);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsGetSiteTable exit dwErr=%d\n", dwErr);
|
|
#endif
|
|
|
|
return dwErr;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: GetNetInfoEx
|
|
//
|
|
// Synopsis: Gets information about the volume.
|
|
//
|
|
// Arguments: [Level] -- Level of Information desired.
|
|
//
|
|
// [pDfsVol] - Pointer to DFS_VOLUME to use to fill the info
|
|
//
|
|
// [pInfo] -- Pointer to info struct to be filled. Pointer
|
|
// members will be allocated using MIDL_user_allocate.
|
|
// The type of this variable is LPDFS_INFO_3, but one
|
|
// can pass in pointers to lower levels, and only the
|
|
// fields appropriate for the level will be touched.
|
|
//
|
|
// [pcbInfo] -- On successful return, contains the size in
|
|
// bytes of the returned info. The returned size does
|
|
// not include the size of the DFS_INFO_3 struct itself.
|
|
//
|
|
// Returns: ERROR_SUCCESS -- Successfully returning info
|
|
//
|
|
// ERROR_OUTOFMEMORY -- Out of memory
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
GetNetInfoEx(
|
|
PDFS_VOLUME pDfsVol,
|
|
DWORD Level,
|
|
LPDFS_INFO_3 pInfo,
|
|
LPDWORD pcbInfo)
|
|
{
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
DWORD cbInfo = 0;
|
|
DWORD cbItem;
|
|
ULONG i;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("GetNetInfoEx(%ws,%d)\n", pDfsVol->wszPrefix, Level);
|
|
#endif
|
|
|
|
//
|
|
// See if this is a Level 100 or 101. If so, we handle them right away
|
|
// and return
|
|
|
|
if (Level == 100) {
|
|
|
|
LPDFS_INFO_100 pInfo100 = (LPDFS_INFO_100) pInfo;
|
|
|
|
if (pDfsVol->wszComment != NULL) {
|
|
cbItem = (wcslen(pDfsVol->wszComment) + 1) * sizeof(WCHAR);
|
|
pInfo100->Comment = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo100->Comment != NULL) {
|
|
wcscpy(pInfo100->Comment, pDfsVol->wszComment);
|
|
cbInfo += cbItem;
|
|
goto AllDone;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto AllDone;
|
|
}
|
|
} else {
|
|
pInfo100->Comment = (LPWSTR) MIDL_user_allocate(sizeof(WCHAR));
|
|
if (pInfo100->Comment != NULL) {
|
|
pInfo100->Comment[0] = UNICODE_NULL;
|
|
cbInfo += sizeof(WCHAR);
|
|
goto AllDone;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto AllDone;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if (Level == 101) {
|
|
|
|
LPDFS_INFO_101 pInfo101 = (LPDFS_INFO_101) pInfo;
|
|
|
|
pInfo->State = pDfsVol->dwState;
|
|
goto AllDone;
|
|
|
|
}
|
|
|
|
//
|
|
// level 4 isn't just an extension of 3, so handle it seperately
|
|
//
|
|
|
|
if (Level == 4) {
|
|
|
|
LPDFS_INFO_4 pInfo4 = (LPDFS_INFO_4) pInfo;
|
|
|
|
cbItem = sizeof(UNICODE_PATH_SEP) + (wcslen(pDfsVol->wszPrefix) + 1) * sizeof(WCHAR);
|
|
pInfo4->EntryPath = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo4->EntryPath != NULL) {
|
|
pInfo4->EntryPath[0] = UNICODE_PATH_SEP;
|
|
wcscpy(&pInfo4->EntryPath[1], pDfsVol->wszPrefix);
|
|
cbInfo += cbItem;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto AllDone;
|
|
}
|
|
|
|
if (pDfsVol->wszComment != NULL) {
|
|
cbItem = (wcslen(pDfsVol->wszComment)+1) * sizeof(WCHAR);
|
|
pInfo4->Comment = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo4->Comment != NULL) {
|
|
wcscpy( pInfo4->Comment, pDfsVol->wszComment );
|
|
cbInfo += cbItem;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto AllDone;
|
|
}
|
|
} else {
|
|
pInfo4->Comment = (LPWSTR) MIDL_user_allocate(sizeof(WCHAR));
|
|
if (pInfo4->Comment != NULL) {
|
|
pInfo4->Comment[0] = UNICODE_NULL;
|
|
cbInfo += sizeof(WCHAR);
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto AllDone;
|
|
}
|
|
}
|
|
|
|
pInfo4->State = pDfsVol->dwState;
|
|
pInfo4->Timeout = pDfsVol->dwTimeout;
|
|
pInfo4->Guid = pDfsVol->idVolume;
|
|
pInfo4->NumberOfStorages = pDfsVol->ReplCount;
|
|
cbItem = pInfo4->NumberOfStorages * sizeof(DFS_STORAGE_INFO);
|
|
pInfo4->Storage = (LPDFS_STORAGE_INFO) MIDL_user_allocate(cbItem);
|
|
if (pInfo4->Storage != NULL) {
|
|
RtlZeroMemory(pInfo4->Storage, cbItem);
|
|
cbInfo += cbItem;
|
|
for (i = 0; i < pDfsVol->ReplCount; i++) {
|
|
dwErr = GetNetStorageInfo(&pDfsVol->ReplicaInfo[i],&pInfo4->Storage[i],&cbItem);
|
|
cbInfo += cbItem;
|
|
}
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
for (; i > 0; i--) {
|
|
MIDL_user_free(pInfo4->Storage[i-1].ServerName);
|
|
MIDL_user_free(pInfo4->Storage[i-1].ShareName);
|
|
}
|
|
}
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
}
|
|
|
|
//
|
|
// See if we need to clean up...
|
|
//
|
|
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
if (pInfo4->EntryPath != NULL) {
|
|
MIDL_user_free(pInfo4->EntryPath);
|
|
}
|
|
if (pInfo4->Storage != NULL) {
|
|
MIDL_user_free(pInfo4->Storage);
|
|
}
|
|
goto AllDone;
|
|
} else {
|
|
*pcbInfo = cbInfo;
|
|
}
|
|
goto AllDone;
|
|
|
|
}
|
|
|
|
//
|
|
// Level is 1,2 or 3
|
|
//
|
|
|
|
//
|
|
// Fill in the Level 1 stuff
|
|
//
|
|
|
|
cbItem = sizeof(UNICODE_PATH_SEP) + (wcslen(pDfsVol->wszPrefix)+1) * sizeof(WCHAR);
|
|
pInfo->EntryPath = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo->EntryPath != NULL) {
|
|
pInfo->EntryPath[0] = UNICODE_PATH_SEP;
|
|
wcscpy(&pInfo->EntryPath[1], pDfsVol->wszPrefix);
|
|
cbInfo += cbItem;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
goto AllDone;
|
|
}
|
|
|
|
//
|
|
// Fill in the Level 2 stuff if needed
|
|
//
|
|
|
|
if (Level > 1) {
|
|
pInfo->State = pDfsVol->dwState;
|
|
pInfo->NumberOfStorages = pDfsVol->ReplCount;
|
|
if (pDfsVol->wszComment != NULL) {
|
|
cbItem = (wcslen(pDfsVol->wszComment)+1) * sizeof(WCHAR);
|
|
pInfo->Comment = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo->Comment != NULL) {
|
|
wcscpy( pInfo->Comment, pDfsVol->wszComment );
|
|
cbInfo += cbItem;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
}
|
|
} else {
|
|
pInfo->Comment = (LPWSTR) MIDL_user_allocate(sizeof(WCHAR));
|
|
if (pInfo->Comment != NULL) {
|
|
pInfo->Comment[0] = UNICODE_NULL;
|
|
cbInfo += sizeof(WCHAR);
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// Fill in the Level 3 stuff if needed
|
|
//
|
|
|
|
if (dwErr == ERROR_SUCCESS && Level > 2) {
|
|
cbItem = pInfo->NumberOfStorages * sizeof(DFS_STORAGE_INFO);
|
|
pInfo->Storage = (LPDFS_STORAGE_INFO) MIDL_user_allocate(cbItem);
|
|
if (pInfo->Storage != NULL) {
|
|
RtlZeroMemory(pInfo->Storage, cbItem);
|
|
cbInfo += cbItem;
|
|
for (i = 0; i < pDfsVol->ReplCount; i++) {
|
|
dwErr = GetNetStorageInfo(&pDfsVol->ReplicaInfo[i], &pInfo->Storage[i], &cbItem);
|
|
cbInfo += cbItem;
|
|
}
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
for (; i > 0; i--) {
|
|
MIDL_user_free(pInfo->Storage[i-1].ServerName);
|
|
MIDL_user_free(pInfo->Storage[i-1].ShareName);
|
|
}
|
|
}
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
}
|
|
}
|
|
|
|
//
|
|
// See if we need to clean up...
|
|
//
|
|
|
|
if (dwErr != ERROR_SUCCESS) {
|
|
if (Level > 1) {
|
|
if (pInfo->EntryPath != NULL) {
|
|
MIDL_user_free(pInfo->EntryPath);
|
|
}
|
|
}
|
|
if (Level > 2) {
|
|
if (pInfo->Storage != NULL) {
|
|
MIDL_user_free(pInfo->Storage);
|
|
}
|
|
}
|
|
}
|
|
|
|
AllDone:
|
|
|
|
//
|
|
// Finally, we are done
|
|
//
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
*pcbInfo = cbInfo;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("GetNetInfoEx returning %d\n", dwErr);
|
|
#endif
|
|
|
|
return(dwErr);
|
|
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: GetNetStorageInfo, public
|
|
//
|
|
// Synopsis: Returns the replica info in a DFS_STORAGE_INFO struct.
|
|
// Useful for NetDfsXXX APIs.
|
|
//
|
|
// Arguments: [pInfo] -- Pointer to DFS_STORAGE_INFO to fill. Pointer
|
|
// members will be allocated using MIDL_user_allocate.
|
|
//
|
|
// [pcbInfo] -- On successful return, set to size in bytes of
|
|
// returned info. The size does not include the size
|
|
// of the DFS_STORAGE_INFO struct itself.
|
|
//
|
|
// Returns:
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
GetNetStorageInfo(
|
|
PDFS_REPLICA_INFO pRepl,
|
|
LPDFS_STORAGE_INFO pInfo,
|
|
LPDWORD pcbInfo)
|
|
{
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
LPWSTR wszShare;
|
|
DWORD cbInfo = 0, cbItem;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("GetNetStorageInfo(\\\\%ws\\%ws)\n",
|
|
pRepl->pwszServerName,
|
|
pRepl->pwszShareName);
|
|
#endif
|
|
|
|
pInfo->State = pRepl->ulReplicaState;
|
|
|
|
cbItem = (wcslen(pRepl->pwszServerName) + 1) * sizeof(WCHAR);
|
|
pInfo->ServerName = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo->ServerName != NULL) {
|
|
wcscpy(pInfo->ServerName, pRepl->pwszServerName);
|
|
cbInfo += cbItem;
|
|
} else {
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
}
|
|
|
|
if (dwErr == ERROR_SUCCESS) {
|
|
cbItem = (wcslen(pRepl->pwszShareName) + 1) * sizeof(WCHAR);
|
|
pInfo->ShareName = (LPWSTR) MIDL_user_allocate(cbItem);
|
|
if (pInfo->ShareName != NULL) {
|
|
wcscpy( pInfo->ShareName, pRepl->pwszShareName );
|
|
cbInfo += cbItem;
|
|
} else {
|
|
MIDL_user_free( pInfo->ServerName );
|
|
pInfo->ServerName = NULL;
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
}
|
|
}
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
*pcbInfo = cbInfo;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("GetStorageInfo returning %d\n", dwErr);
|
|
#endif
|
|
|
|
return( dwErr );
|
|
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsRemoveRootReplica
|
|
//
|
|
// Synopsis: Removes a Dfs root replica from the dfs root replica list
|
|
//
|
|
// History: 12/16/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsRemoveRootReplica(
|
|
PDFS_VOLUME_LIST pDfsVolList,
|
|
LPWSTR RootName)
|
|
{
|
|
ULONG cVol;
|
|
ULONG cRepl;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsRemoveRootReplica(%ws)\n", RootName);
|
|
#endif
|
|
|
|
|
|
for (cVol = 0; cVol < pDfsVolList->VolCount; cVol++) {
|
|
|
|
if (wcscmp(pDfsVolList->Volumes[cVol].wszObjectName,L"\\domainroot") != 0)
|
|
continue;
|
|
|
|
ScanReplicas:
|
|
|
|
for (cRepl = 0; cRepl < pDfsVolList->Volumes[cVol].ReplCount; cRepl++) {
|
|
|
|
if (
|
|
_wcsicmp(
|
|
pDfsVolList->Volumes[cVol].ReplicaInfo[cRepl].pwszServerName,
|
|
RootName) == 0
|
|
) {
|
|
DfsReplDeleteByIndex(&pDfsVolList->Volumes[cVol], cRepl);
|
|
goto ScanReplicas;
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsRemoveRootReplica exit %d\n", dwErr);
|
|
#endif
|
|
|
|
return dwErr;
|
|
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsRecoverVolList, public
|
|
//
|
|
// Synopsis: Walks the dfs_volume list, checking the recovery params, and
|
|
// applying any recovery needed. Also applies the deleted replica
|
|
// list to the volume's replica list.
|
|
//
|
|
// Arguments: [pDfsVolList] -- Pointer to DFS_VOLUME_LIST work on.
|
|
//
|
|
// Returns:
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsRecoverVolList(
|
|
PDFS_VOLUME_LIST pDfsVolList)
|
|
{
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
ULONG iVol;
|
|
ULONG iRepl;
|
|
ULONG iDelRepl;
|
|
PDFS_VOLUME pVol;
|
|
ULONG RecoveryState;
|
|
ULONG Operation;
|
|
ULONG OperState;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsRecoverVolList(%d volumes)\n", pDfsVolList->VolCount);
|
|
#endif
|
|
|
|
ReStart:
|
|
for (iVol = 0; dwErr == ERROR_SUCCESS && iVol < pDfsVolList->VolCount; iVol++) {
|
|
pVol = &pDfsVolList->Volumes[iVol];
|
|
if (pVol->cbRecovery >= sizeof(ULONG)) {
|
|
_GetULong(pVol->pRecovery, RecoveryState);
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("RecoveryState[%d]=0x%x\n", iVol, RecoveryState);
|
|
#endif
|
|
if (DFS_GET_RECOVERY_STATE(RecoveryState) != DFS_RECOVERY_STATE_NONE) {
|
|
Operation = DFS_GET_RECOVERY_STATE(RecoveryState);
|
|
OperState = DFS_GET_OPER_STAGE(RecoveryState);
|
|
switch (Operation) {
|
|
case DFS_RECOVERY_STATE_CREATING:
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DFS_RECOVERY_STATE_CREATING\n");
|
|
#endif
|
|
dwErr = DfsVolDelete(pDfsVolList, iVol);
|
|
goto ReStart;
|
|
case DFS_RECOVERY_STATE_ADD_SERVICE:
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DFS_RECOVERY_STATE_ADD_SERVICE\n");
|
|
#endif
|
|
// dwErr = RecoverFromAddService(OperState);
|
|
ASSERT(L"DFS_RECOVERY_STATE_ADD_SERVICE - WHY?\n");
|
|
break;
|
|
case DFS_RECOVERY_STATE_REMOVE_SERVICE:
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DFS_RECOVERY_STATE_REMOVE_SERVICE\n");
|
|
#endif
|
|
// dwErr = RecoverFromRemoveService(OperState);
|
|
ASSERT(L"DFS_RECOVERY_STATE_REMOVE_SERVICE - WHY?\n");
|
|
break;
|
|
case DFS_RECOVERY_STATE_DELETE:
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DFS_RECOVERY_STATE_DELETE\n");
|
|
#endif
|
|
dwErr = DfsVolDelete(pDfsVolList, iVol);
|
|
goto ReStart;
|
|
default:
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("default\n");
|
|
#endif
|
|
dwErr = ERROR_INTERNAL_DB_CORRUPTION;
|
|
}
|
|
}
|
|
}
|
|
if (pVol->pRecovery != NULL)
|
|
MarshalBufferFree(pVol->pRecovery);
|
|
pVol->pRecovery = NULL;
|
|
pVol->cbRecovery = 0;
|
|
}
|
|
|
|
if (dwErr != ERROR_SUCCESS)
|
|
goto AllDone;
|
|
|
|
//
|
|
// Now apply deleted replica list to each volume
|
|
//
|
|
|
|
for (iVol = 0; dwErr == ERROR_SUCCESS && iVol < pDfsVolList->VolCount; iVol++) {
|
|
pVol = &pDfsVolList->Volumes[iVol];
|
|
for (iDelRepl = 0; dwErr == ERROR_SUCCESS && iDelRepl < pVol->DelReplCount; iDelRepl++) {
|
|
#if 0 // XXX This appears to be wrong - don't do it.
|
|
dwErr = DfsReplDeleteByName(
|
|
pVol,
|
|
pVol->DelReplicaInfo[iDelRepl].pwszServerName,
|
|
pVol->DelReplicaInfo[iDelRepl].pwszShareName);
|
|
#endif
|
|
DfsFreeRepl(&pVol->DelReplicaInfo[iDelRepl]);
|
|
}
|
|
if (pVol->DelReplicaInfo != NULL) {
|
|
free(pVol->DelReplicaInfo);
|
|
pVol->DelReplicaInfo = NULL;
|
|
}
|
|
if (pVol->DelFtModification != NULL) {
|
|
free(pVol->DelFtModification);
|
|
pVol->DelFtModification = NULL;
|
|
}
|
|
pVol->DelReplCount = 0;
|
|
}
|
|
|
|
AllDone:
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsRecoverVolList returning %d\n", dwErr);
|
|
#endif
|
|
return dwErr;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsVolDelete, public
|
|
//
|
|
// Synopsis: Removes a DFS_VOLUME from DFS_VOLUME_LIST
|
|
//
|
|
// Arguments: [pDfsVolList] -- Pointer to DFS_VOLUME_LIST work on.
|
|
// [iVol] -- Index of volume to delete
|
|
//
|
|
// Returns:
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsVolDelete(
|
|
PDFS_VOLUME_LIST pDfsVolList,
|
|
ULONG iVol)
|
|
{
|
|
ULONG i;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
ASSERT(iVol < pDfsVolList->VolCount);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsVolDelete(%d)\n", iVol);
|
|
#endif
|
|
|
|
//
|
|
// Free any memory this Volume is using
|
|
//
|
|
|
|
DfsFreeVol(&pDfsVolList->Volumes[iVol]);
|
|
|
|
//
|
|
// Since we're shrinking the list, we simply clip out the entry we don't want
|
|
// and adjust the count.
|
|
//
|
|
for (i = iVol+1; i < pDfsVolList->VolCount; i++)
|
|
pDfsVolList->Volumes[i-1] = pDfsVolList->Volumes[i];
|
|
|
|
pDfsVolList->VolCount--;
|
|
RtlZeroMemory(&pDfsVolList->Volumes[pDfsVolList->VolCount], sizeof(DFS_VOLUME));
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsVolDelete returning %d\n", dwErr);
|
|
#endif
|
|
return dwErr;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsReplDeleteByIndex, public
|
|
//
|
|
// Synopsis: Removes a PDFS_REPLICA_INFO from a DFS_VOLUME's replica list
|
|
//
|
|
// Arguments: [pDfsVol] -- Pointer to DFS_VOLUME work on.
|
|
// [iRepl] -- Index of replica to delete
|
|
//
|
|
// Returns:
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsReplDeleteByIndex(
|
|
PDFS_VOLUME pVol,
|
|
ULONG iRepl)
|
|
{
|
|
ULONG i;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
ASSERT(iRepl < pVol->ReplCount);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsReplDeleteByIndex(%d)\n", iRepl);
|
|
#endif
|
|
|
|
//
|
|
// Free any memory this replica is using
|
|
//
|
|
|
|
DfsFreeRepl(&pVol->ReplicaInfo[iRepl]);
|
|
|
|
//
|
|
// Since we're shrinking the list(s), we simply clip out the entry we don't want
|
|
// and adjust the count.
|
|
//
|
|
for (i = iRepl+1; i < pVol->ReplCount; i++) {
|
|
pVol->ReplicaInfo[i-1] = pVol->ReplicaInfo[i];
|
|
pVol->FtModification[i-1] = pVol->FtModification[i];
|
|
}
|
|
|
|
pVol->ReplCount--;
|
|
RtlZeroMemory(&pVol->ReplicaInfo[pVol->ReplCount], sizeof(DFS_REPLICA_INFO));
|
|
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsReplDeleteByIndex returning %d\n", dwErr);
|
|
#endif
|
|
return dwErr;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsReplDeleteByName, public
|
|
//
|
|
// Synopsis: Removes a PDFS_REPLICA_INFO from a DFS_VOLUME's replica list. Simply
|
|
// looks up the replica by name and then calls DfsReplDeleteByIndex()
|
|
//
|
|
// Arguments: [pDfsVol] -- Pointer to DFS_VOLUME work on.
|
|
// [pwszServername] - Server Name
|
|
// [pwszShareName] - Share Name
|
|
//
|
|
// Returns:
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsReplDeleteByName(
|
|
PDFS_VOLUME pVol,
|
|
LPWSTR pwszServerName,
|
|
LPWSTR pwszShareName)
|
|
{
|
|
ULONG iRep;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsReplDeleteByName(%ws,%ws)\n", pwszServerName, pwszShareName);
|
|
#endif
|
|
|
|
if (pwszServerName == NULL || pwszShareName == NULL) {
|
|
dwErr = ERROR_INVALID_PARAMETER;
|
|
goto AllDone;
|
|
}
|
|
|
|
ReStart:
|
|
|
|
//
|
|
// Scan the volume's replica list, and if there is a match on the
|
|
// passed-in servername and sharename, remove the replica.
|
|
//
|
|
for (iRep = 0; iRep < pVol->ReplCount; iRep++) {
|
|
if (_wcsicmp(pVol->ReplicaInfo[iRep].pwszServerName, pwszServerName) == 0
|
|
&&
|
|
_wcsicmp(pVol->ReplicaInfo[iRep].pwszShareName, pwszShareName) == 0
|
|
) {
|
|
dwErr = DfsReplDeleteByIndex(pVol, iRep);
|
|
goto ReStart;
|
|
}
|
|
}
|
|
|
|
AllDone:
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsReplDeleteByName returning %d\n", dwErr);
|
|
#endif
|
|
return dwErr;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsDelReplDelete, public
|
|
//
|
|
// Synopsis: Removes a PDFS_REPLICA_INFO from a DFS_VOLUME's deleted replica list
|
|
//
|
|
// Arguments: [pDfsVol] -- Pointer to DFS_VOLUME work on.
|
|
// [iDelRepl] -- Index of Deleted replica to delete
|
|
//
|
|
// Returns:
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
DWORD
|
|
DfsDelReplDelete(
|
|
PDFS_VOLUME pVol,
|
|
ULONG iDelRepl)
|
|
{
|
|
ULONG i;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
ASSERT(iDelRepl < pVol->DelReplCount);
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsDelReplDelete(%d)\n", iDelRepl);
|
|
#endif
|
|
|
|
//
|
|
// Free any memory this replica is using
|
|
//
|
|
|
|
DfsFreeRepl(&pVol->DelReplicaInfo[iDelRepl]);
|
|
|
|
//
|
|
// Since we're shrinking the list(s), we simply clip out the entry we don't want
|
|
// and adjust the count.
|
|
//
|
|
for (i = iDelRepl+1; i < pVol->DelReplCount; i++) {
|
|
pVol->DelReplicaInfo[i-1] = pVol->DelReplicaInfo[i];
|
|
pVol->DelFtModification[i-1] = pVol->DelFtModification[i];
|
|
}
|
|
|
|
pVol->DelReplCount--;
|
|
RtlZeroMemory(&pVol->DelReplicaInfo[pVol->DelReplCount], sizeof(DFS_REPLICA_INFO));
|
|
|
|
#if DBG
|
|
if (DfsSvcVerbose)
|
|
DbgPrint("DfsDelReplDelete returning %d\n", dwErr);
|
|
#endif
|
|
return dwErr;
|
|
}
|
|
|
|
#if DBG
|
|
|
|
void
|
|
DumpBuf(PCHAR cp, ULONG len)
|
|
{
|
|
ULONG i, j, c;
|
|
|
|
for (i = 0; i < len; i += 16) {
|
|
DbgPrint("%08x ", i /* +(ULONG)cp */);
|
|
for (j = 0; j < 16; j++) {
|
|
if (j == 8)
|
|
DbgPrint(" ");
|
|
if (i+j < len) {
|
|
c = cp[i+j] & 0xff;
|
|
DbgPrint("%02x ", c);
|
|
} else {
|
|
DbgPrint(" ");
|
|
}
|
|
}
|
|
DbgPrint(" ");
|
|
for (j = 0; j < 16; j++) {
|
|
if (j == 8)
|
|
DbgPrint("|");
|
|
if (i+j < len) {
|
|
c = cp[i+j] & 0xff;
|
|
if (c < ' ' || c > '~')
|
|
c = '.';
|
|
DbgPrint("%c", c);
|
|
} else {
|
|
DbgPrint(" ");
|
|
}
|
|
}
|
|
DbgPrint("\n");
|
|
}
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// Function: DfsDumpVolList
|
|
//
|
|
// Synopsis: Prints the volume information represented by the volume
|
|
// list passed in.
|
|
//
|
|
// Returns: [ERROR_SUCCESS] -- If all went well.
|
|
//
|
|
// History: 11/19/98 JHarper Created
|
|
//
|
|
//-------------------------------------------------------------------------
|
|
|
|
VOID
|
|
DfsDumpVolList(
|
|
PDFS_VOLUME_LIST pDfsVolList)
|
|
{
|
|
ULONG cVol;
|
|
ULONG cRepl;
|
|
ULONG cSite;
|
|
ULONG i;
|
|
GUID guid;
|
|
PLIST_ENTRY pListHead;
|
|
PLIST_ENTRY pLink;
|
|
PDFSM_SITE_ENTRY pSiteEntry;
|
|
|
|
DbgPrint("****************************************\n");
|
|
|
|
for (cVol = 0; cVol < pDfsVolList->VolCount; cVol++) {
|
|
|
|
DbgPrint("Name=%ws\n", pDfsVolList->Volumes[cVol].wszObjectName);
|
|
DbgPrint("Prefix: %ws\n", pDfsVolList->Volumes[cVol].wszPrefix);
|
|
DbgPrint("Comment: %ws\n", pDfsVolList->Volumes[cVol].wszComment);
|
|
DbgPrint("cbRecovery: %d\n", pDfsVolList->Volumes[cVol].cbRecovery);
|
|
|
|
DbgPrint("ReplCount=%d\n", pDfsVolList->Volumes[cVol].ReplCount);
|
|
DbgPrint("AllocatedReplCount=%d\n", pDfsVolList->Volumes[cVol].AllocatedReplCount);
|
|
|
|
for (cRepl = 0; cRepl < pDfsVolList->Volumes[cVol].ReplCount; cRepl++) {
|
|
|
|
DbgPrint("[%d]ServerName=%ws\n",
|
|
cRepl, pDfsVolList->Volumes[cVol].ReplicaInfo[cRepl].pwszServerName);
|
|
DbgPrint("[%d]ShareName=%ws\n",
|
|
cRepl, pDfsVolList->Volumes[cVol].ReplicaInfo[cRepl].pwszShareName);
|
|
|
|
}
|
|
|
|
DbgPrint("DelReplCount=%d\n", pDfsVolList->Volumes[cVol].DelReplCount);
|
|
DbgPrint("AllocatedDelReplCount=%d\n", pDfsVolList->Volumes[cVol].AllocatedDelReplCount);
|
|
|
|
for (cRepl = 0; cRepl < pDfsVolList->Volumes[cVol].DelReplCount; cRepl++) {
|
|
|
|
DbgPrint("[%d]ServerName=%ws\n",
|
|
cRepl, pDfsVolList->Volumes[cVol].DelReplicaInfo[cRepl].pwszServerName);
|
|
DbgPrint("[%d]ShareName=%ws\n",
|
|
cRepl, pDfsVolList->Volumes[cVol].DelReplicaInfo[cRepl].pwszShareName);
|
|
|
|
}
|
|
|
|
DbgPrint("\n");
|
|
|
|
}
|
|
|
|
DbgPrint("---------SiteTable-------\n");
|
|
|
|
guid = pDfsVolList->SiteGuid;
|
|
DbgPrint("SiteTableGuid:"
|
|
"%08lX-%04X-%04X-%02X%02X%02X%02X%02X%02X%02X%02X\n",
|
|
guid.Data1, guid.Data2, guid.Data3, (int) guid.Data4[0],
|
|
(int) guid.Data4[1], (int) guid.Data4[2], (int) guid.Data4[3],
|
|
(int) guid.Data4[4], (int) guid.Data4[5],
|
|
(int) guid.Data4[6], (int) guid.Data4[7]);
|
|
|
|
|
|
pListHead = &pDfsVolList->SiteList;
|
|
|
|
if (pListHead->Flink == pListHead)
|
|
return;
|
|
|
|
for (pLink = pListHead->Flink; pLink != pListHead; pLink = pLink->Flink) {
|
|
|
|
pSiteEntry = CONTAINING_RECORD(pLink, DFSM_SITE_ENTRY, Link);
|
|
|
|
DbgPrint("[%ws-%d sites]\n",
|
|
pSiteEntry->ServerName,
|
|
pSiteEntry->Info.cSites);
|
|
|
|
for (i = 0; i < pSiteEntry->Info.cSites; i++) {
|
|
DbgPrint("\t%ws\n", pSiteEntry->Info.Site[i].SiteName);
|
|
}
|
|
|
|
}
|
|
|
|
DbgPrint("****************************************\n");
|
|
|
|
}
|
|
|
|
#endif // DBG
|