/** Module Name: NetUtils.cpp Abstract: This is the implementation file for the utility functions for Network APIs. */ #include "NetUtils.h" #include #include #include #include #include #include #include //DNS_MAX_NAME_BUFFER_LENGTH //---------------------------------------------------------------------------------- HRESULT FlatAdd ( DOMAIN_DESC* i_pDomDesc, // Pointer to the Domain Description Structure returned by IBrowserDomainTree::GetDomains() OUT NETNAMELIST* o_pDomainList // Pointer to the list of NETNAME is returned here. ) /*++ Routine Description: This function flattens the domain tree returned by GetDomains() method of IBrowserDomainTree into a NETNAME list. This code produces a preorder traversal method. Arguments: i_pDomDesc - Pointer to the Domain Description Structure returned by IBrowserDomainTree::GetDomains() o_pDomainList - Pointer to the list of NETNAME is returned here. --*/ { RETURN_INVALIDARG_IF_NULL(i_pDomDesc); RETURN_INVALIDARG_IF_NULL(o_pDomainList); HRESULT hr = S_OK; stack Stack; Stack.push(i_pDomDesc); while (!Stack.empty()) { DOMAIN_DESC* pDomDesc = Stack.top(); Stack.pop(); NETNAME* pCurrent = new NETNAME; BREAK_OUTOFMEMORY_IF_NULL(pCurrent, &hr); pCurrent->bstrNetName = pDomDesc->pszName; if (!(pCurrent->bstrNetName)) { delete pCurrent; hr = E_OUTOFMEMORY; break; } // Add to the output list. o_pDomainList->push_back(pCurrent); if (pDomDesc->pdNextSibling) Stack.push(pDomDesc->pdNextSibling); if (pDomDesc->pdChildList) Stack.push(pDomDesc->pdChildList); } if (FAILED(hr)) FreeNetNameList(o_pDomainList); return hr; } HRESULT Get50Domains ( OUT NETNAMELIST* o_pDomains // List of NETNAME structures. ) /*++ Routine Description: Returns a list of all NT 5.0 domains in a list of NETNAME structures Arguments: o_pDomains - Pointer to list of NETNAME structures is returned. --*/ // This method uses the DsDomainTreeBrowser COM Object to get the list // of domain names from the DS. The Domain tree returned is then flatened out // by using preorder traversal algorithm. { RETURN_INVALIDARG_IF_NULL(o_pDomains); FreeNetNameList(o_pDomains); CComPtr pDsDomTree; HRESULT hr = CoCreateInstance( CLSID_DsDomainTreeBrowser, NULL, CLSCTX_INPROC_SERVER, IID_IDsBrowseDomainTree, (void **)&pDsDomTree ); RETURN_IF_FAILED(hr); PDOMAIN_TREE pDomTree = NULL; hr = pDsDomTree->GetDomains(&pDomTree, DBDTF_RETURNEXTERNAL | DBDTF_RETURNINBOUND); RETURN_IF_FAILED(hr); // Flaten the tree in to a list. hr = FlatAdd(&(pDomTree->aDomains[0]), o_pDomains); pDsDomTree->FreeDomains(&pDomTree); return hr; } //---------------------------------------------------------------------------------- HRESULT Is50Domain ( IN BSTR i_bstrDomain, OUT BSTR* o_pbstrDnsDomainName ) { return GetDomainInfo(i_bstrDomain, NULL, o_pbstrDnsDomainName); } //---------------------------------------------------------------------------------- HRESULT GetServerInfo ( IN BSTR i_bstrServer, OUT BSTR *o_pbstrDomain, // = NULL OUT BSTR *o_pbstrNetbiosName, // = NULL OUT BOOL *o_pbValidDSObject, // = NULL OUT BSTR *o_pbstrDnsName, // = NULL OUT BSTR *o_pbstrGuid, // = NULL OUT BSTR *o_pbstrFQDN, // = NULL OUT SUBSCRIBERLIST *o_pFRSRootList, // NULL OUT long *o_lMajorNo, // = NULL OUT long *o_lMinorNo // = NULL ) { // This function uses NetWkstaGetInfo to get the server informaiton. if (!o_pbstrDomain && !o_pbstrNetbiosName && !o_pbValidDSObject && !o_pbstrDnsName && !o_pbstrGuid && !o_pbstrFQDN && !o_pFRSRootList && !o_lMajorNo && !o_lMinorNo ) return(E_INVALIDARG); if (o_pbstrDomain) *o_pbstrDomain = NULL; if (o_pbstrNetbiosName) *o_pbstrNetbiosName = NULL; if (o_pbValidDSObject) *o_pbValidDSObject = FALSE; if (o_pbstrDnsName) *o_pbstrDnsName = NULL; if (o_pbstrGuid) *o_pbstrGuid = NULL; if (o_pbstrFQDN) *o_pbstrFQDN = NULL; if (o_pFRSRootList) FreeSubscriberList(o_pFRSRootList); if (o_lMajorNo) *o_lMajorNo = 0; if (o_lMinorNo) *o_lMinorNo = 0; HRESULT hr = S_OK; PWKSTA_INFO_100 wki100 = NULL; NET_API_STATUS NetStatus = NetWkstaGetInfo(i_bstrServer, 100, (LPBYTE *) &wki100 ); if (ERROR_SUCCESS != NetStatus) hr = HRESULT_FROM_WIN32(NetStatus); else { if (o_lMajorNo) *o_lMajorNo = wki100->wki100_ver_major; if (o_lMinorNo) *o_lMinorNo = wki100->wki100_ver_minor; if (SUCCEEDED(hr) && o_pbstrNetbiosName && wki100->wki100_computername) { *o_pbstrNetbiosName = SysAllocString(wki100->wki100_computername); if (!*o_pbstrNetbiosName) hr = E_OUTOFMEMORY; } if (SUCCEEDED(hr) && (o_pbstrDomain || o_pbValidDSObject || o_pbstrDnsName || o_pbstrGuid || o_pbstrFQDN || o_pFRSRootList) && wki100->wki100_langroup && wki100->wki100_computername) { PDSROLE_PRIMARY_DOMAIN_INFO_BASIC pBuffer = NULL; DWORD dwErr = DsRoleGetPrimaryDomainInformation( i_bstrServer, DsRolePrimaryDomainInfoBasic, (PBYTE *)&pBuffer); if (ERROR_SUCCESS != dwErr) hr = HRESULT_FROM_WIN32(dwErr); else { CComBSTR bstrDomain; // // Verify if this is really a server name. // NetWkstaGetInfo and DsRoleGetPrimaryDomainInformation work with domain name, // they return info of a DC. // if (i_bstrServer && *i_bstrServer && (pBuffer->DomainNameFlat && !lstrcmpi(i_bstrServer, pBuffer->DomainNameFlat) || pBuffer->DomainNameDns && !lstrcmpi(i_bstrServer, pBuffer->DomainNameDns))) { // we're seeing a domain name, not what we expect, return S_FALSE. hr = S_FALSE; // server not in a domain or cannot find an appropriate computer obj. } else { bstrDomain = (pBuffer->DomainNameDns ? pBuffer->DomainNameDns : pBuffer->DomainNameFlat); if (!bstrDomain) { hr = E_OUTOFMEMORY; } else if (!*bstrDomain) { hr = S_FALSE; // server not in a domain or cannot find an appropriate computer obj. } else { // // In case the DNS name is in absolute form, remove the ending dot // int nlen = _tcslen(bstrDomain); if ( *(bstrDomain + nlen - 1) == _T('.') ) *(bstrDomain + nlen - 1) = _T('\0'); } } DsRoleFreeMemory(pBuffer); if (S_OK == hr && o_pbstrDomain) { *o_pbstrDomain = SysAllocString(bstrDomain); if (!*o_pbstrDomain) hr = E_OUTOFMEMORY; } if (S_OK == hr && (o_pbValidDSObject || o_pbstrDnsName || o_pbstrGuid || o_pbstrFQDN || o_pFRSRootList) ) { CComBSTR bstrDC; HANDLE hDS = NULL; hr = DsBindToDS(bstrDomain, &bstrDC, &hDS); if (SUCCEEDED(hr)) { CComBSTR bstrOldName = wki100->wki100_langroup; bstrOldName += _T("\\"); bstrOldName += wki100->wki100_computername; bstrOldName += _T("$"); if (o_pbstrGuid) hr = CrackName(hDS, bstrOldName, DS_NT4_ACCOUNT_NAME, DS_UNIQUE_ID_NAME, o_pbstrGuid); if (SUCCEEDED(hr) && (o_pbValidDSObject || o_pbstrDnsName || o_pbstrFQDN || o_pFRSRootList)) { CComBSTR bstrComputerDN; hr = CrackName(hDS, bstrOldName, DS_NT4_ACCOUNT_NAME, DS_FQDN_1779_NAME, &bstrComputerDN); if (SUCCEEDED(hr) && o_pbstrFQDN) { *o_pbstrFQDN = bstrComputerDN.Copy(); if (!*o_pbstrFQDN) hr = E_OUTOFMEMORY; } if (SUCCEEDED(hr) && (o_pbValidDSObject || o_pbstrDnsName || o_pFRSRootList)) { PLDAP pldap = NULL; hr = LdapConnectToDC(bstrDC, &pldap); if (SUCCEEDED(hr)) { dwErr = ldap_bind_s(pldap, NULL, NULL, LDAP_AUTH_NEGOTIATE); if (LDAP_SUCCESS != dwErr) { DebugOutLDAPError(pldap, dwErr, _T("ldap_bind_s")); hr = HRESULT_FROM_WIN32(dwErr); } else { if (o_pbValidDSObject) { hr = IsValidObject(pldap, bstrComputerDN); *o_pbValidDSObject = (S_OK == hr); } if (SUCCEEDED(hr) && o_pbstrDnsName) { PLDAP_ATTR_VALUE pValues[2] = {0,0}; LDAP_ATTR_VALUE pAttributes[1]; pAttributes[0].bstrAttribute = _T("dNSHostName"); hr = GetValues( pldap, bstrComputerDN, OBJCLASS_SF_COMPUTER, LDAP_SCOPE_BASE, 1, pAttributes, pValues ); if (SUCCEEDED(hr)) { *o_pbstrDnsName = SysAllocString((LPTSTR)(pValues[0]->vpValue)); if (!*o_pbstrDnsName) hr = E_OUTOFMEMORY; FreeAttrValList(pValues[0]); } else { hr = S_OK; // ignore failure, since dNSHostName might not be set } } if (SUCCEEDED(hr) && o_pFRSRootList) { PCTSTR ppszAttributes[] = {ATTR_FRS_SUBSCRIBER_MEMBERREF, ATTR_FRS_SUBSCRIBER_ROOTPATH, 0}; LListElem* pElem = NULL; hr = GetValuesEx(pldap, bstrComputerDN, LDAP_SCOPE_SUBTREE, OBJCLASS_SF_NTFRSSUBSCRIBER, ppszAttributes, &pElem); if (SUCCEEDED(hr) && pElem) { LListElem* pCurElem = pElem; while (pCurElem) { PTSTR** pppszValues = pCurElem->pppszAttrValues; if (!pppszValues || !pppszValues[0] || !*(pppszValues[0]) || !pppszValues[1] || !*(pppszValues[1])) { pCurElem = pCurElem->Next; continue; // corrupted subscriber object, ignore it } SUBSCRIBER* pCurrent = new SUBSCRIBER; BREAK_OUTOFMEMORY_IF_NULL(pCurrent, &hr); pCurrent->bstrMemberDN = *(pppszValues[0]); // frsMemberReference pCurrent->bstrRootPath = *(pppszValues[1]); // frsRootPath if (!(pCurrent->bstrMemberDN) || !(pCurrent->bstrRootPath)) { delete pCurrent; hr = E_OUTOFMEMORY; break; } o_pFRSRootList->push_back(pCurrent); pCurElem = pCurElem->Next; } FreeLListElem(pElem); if (FAILED(hr)) FreeSubscriberList(o_pFRSRootList); } } } ldap_unbind(pldap); } } } DsUnBind(&hDS); } } } // DsRoleGetPrimaryDomainInformation } NetApiBufferFree((LPBYTE)wki100); } //NetWkstaGetInfo return hr; } //---------------------------------------------------------------------------------- HRESULT IsServerRunningDfs ( IN BSTR i_bstrServer ) /*++ Routine Description: Contacts the machine and determines if service Dfs is running. Arguments: i_bstrServer - The server name. --*/ { SC_HANDLE SCMHandle = NULL; SC_HANDLE DFSHandle = NULL; SERVICE_STATUS SStatus; HRESULT hr = S_FALSE; if ((SCMHandle = OpenSCManager(i_bstrServer, NULL, GENERIC_READ)) && (DFSHandle = OpenService(SCMHandle, _T("Dfs"), SERVICE_QUERY_STATUS)) && QueryServiceStatus(DFSHandle, &SStatus) && (SERVICE_RUNNING == SStatus.dwCurrentState) ) { hr = S_OK; } if (DFSHandle) CloseServiceHandle(DFSHandle); if (SCMHandle) CloseServiceHandle(SCMHandle); return hr; } // // TRUE: support NTFS 5 reparse point // FALSE: doesn't support // BOOL CheckReparsePoint(IN BSTR i_bstrServer, IN BSTR i_bstrShare) { if (!i_bstrServer || !*i_bstrServer || !i_bstrShare || !*i_bstrShare) return FALSE; TCHAR szFileSystemName[MAX_PATH + 1] = {0}; DWORD dwMaxCompLength = 0, dwFileSystemFlags = 0; CComBSTR bstrRootPath = _T("\\\\"); bstrRootPath += i_bstrServer; bstrRootPath += _T("\\"); bstrRootPath += i_bstrShare; bstrRootPath += _T("\\"); // a trailing backslash is required BOOL bRet = GetVolumeInformation(bstrRootPath, NULL, 0, NULL, &dwMaxCompLength, &dwFileSystemFlags, szFileSystemName, MAX_PATH); return (bRet && !lstrcmpi(_T("NTFS"), szFileSystemName) && (dwFileSystemFlags & FILE_SUPPORTS_REPARSE_POINTS)); } //---------------------------------------------------------------------------------- // // S_OK: o_pbFound is valid // S_FALSE: share is not eligible to host dfs root // hr: other errors // HRESULT CheckShare ( IN BSTR i_bstrServer, IN BSTR i_bstrShare, OUT BOOL* o_pbFound ) { if (!i_bstrServer || !*i_bstrServer || !i_bstrShare || !*i_bstrShare || !o_pbFound) return E_INVALIDARG; *o_pbFound = FALSE; if (!lstrcmpi(i_bstrShare, _T("SYSVOL")) || !lstrcmpi(i_bstrShare, _T("NETLOGON")) || _istspace(i_bstrShare[0]) || // exclude share with leading space _istspace(i_bstrShare[lstrlen(i_bstrShare) - 1]) // exclude share with trailing space ) return S_FALSE; LPSHARE_INFO_1 lpBuffer = NULL; DWORD dwNumEntries = 0; DWORD dwTotalEntries = 0; DWORD dwResumehandle = NULL; NET_API_STATUS nstatRetVal = NetShareEnum( i_bstrServer, 1L, (LPBYTE*) &lpBuffer, 0xFFFFFFFF, &dwNumEntries, &dwTotalEntries, &dwResumehandle ); if (NERR_Success != nstatRetVal) { if (ERROR_NO_MORE_ITEMS == nstatRetVal) return S_OK; else return HRESULT_FROM_WIN32(nstatRetVal); } HRESULT hr = S_OK; for (DWORD i = 0; i < dwNumEntries; i++) { if (!lstrcmpi(lpBuffer[i].shi1_netname, i_bstrShare)) { if (lpBuffer[i].shi1_type != STYPE_DISKTREE) hr = S_FALSE; else *o_pbFound = TRUE; break; } } NetApiBufferFree ((LPVOID) lpBuffer); return hr; } HRESULT CreateShare ( IN BSTR i_bstrServerName, IN BSTR i_bstrShareName, IN BSTR i_bstrComment, IN BSTR i_bstrPath ) /*++ Routine Description: This function creates an share of the specified name on the specified computer. Arguments: i_bstrServerName - The machine on which the new share is to be created. i_bstrShareName - The name of the new share to be created. i_bstrComment - Comment to be given for the share. i_bstrPath - The physical path of the share. --*/ { RETURN_INVALIDARG_IF_NULL(i_bstrServerName); RETURN_INVALIDARG_IF_NULL(i_bstrShareName); RETURN_INVALIDARG_IF_NULL(i_bstrComment); RETURN_INVALIDARG_IF_NULL(i_bstrPath); SHARE_INFO_2 NewShareInfo; NewShareInfo.shi2_netname = i_bstrShareName; NewShareInfo.shi2_type = STYPE_DISKTREE; NewShareInfo.shi2_remark = i_bstrComment; NewShareInfo.shi2_permissions = ACCESS_ALL; NewShareInfo.shi2_max_uses = (DWORD)-1; NewShareInfo.shi2_current_uses = 0; NewShareInfo.shi2_path = i_bstrPath; NewShareInfo.shi2_passwd = NULL; DWORD dwError = 0; NET_API_STATUS nstatRetVal = NetShareAdd(i_bstrServerName, 2, (LPBYTE) &NewShareInfo, &dwError); return HRESULT_FROM_WIN32(nstatRetVal); } //---------------------------------------------------------------------------------- // Checks if \\\\x\y\z is on a valid disktree share // and return the path local to the server HRESULT GetFolderInfo ( IN BSTR i_bstrServerName, // IN BSTR i_bstrFolder, // \x\y\z OUT BSTR *o_bstrPath // return \x\y\z ) { if (!i_bstrServerName || !*i_bstrServerName || !i_bstrFolder || !*i_bstrFolder) { return(E_INVALIDARG); } // // first, test if folder \\\\x\y\z can be reached // CComBSTR bstrUNC; if (0 != mylstrncmp(i_bstrServerName, _T("\\\\"), 2)) { bstrUNC = _T("\\\\"); RETURN_OUTOFMEMORY_IF_NULL((BSTR)bstrUNC); } bstrUNC += i_bstrServerName; RETURN_OUTOFMEMORY_IF_NULL((BSTR)bstrUNC); bstrUNC += _T("\\"); RETURN_OUTOFMEMORY_IF_NULL((BSTR)bstrUNC); bstrUNC += i_bstrFolder; RETURN_OUTOFMEMORY_IF_NULL((BSTR)bstrUNC); if (-1 == GetFileAttributes(bstrUNC)) return HRESULT_FROM_WIN32(GetLastError()); CComBSTR bstrShareName; CComBSTR bstrRestOfPath; TCHAR *p = _tcschr(i_bstrFolder, _T('\\')); if (p && *(p+1)) { bstrRestOfPath = p+1; RETURN_OUTOFMEMORY_IF_NULL((BSTR)bstrRestOfPath); bstrShareName = CComBSTR(p - i_bstrFolder, i_bstrFolder); } else { bstrShareName = i_bstrFolder; } RETURN_OUTOFMEMORY_IF_NULL((BSTR)bstrShareName); HRESULT hr = S_OK; SHARE_INFO_2 *pShareInfo = NULL; NET_API_STATUS nstatRetVal = NetShareGetInfo (i_bstrServerName, bstrShareName, 2, (LPBYTE *)&pShareInfo); if (nstatRetVal != NERR_Success) { hr = HRESULT_FROM_WIN32(nstatRetVal); } else { if (STYPE_DISKTREE != pShareInfo->shi2_type && STYPE_SPECIAL != pShareInfo->shi2_type) // we allow ADMIN$, C$ to be configured in Dfs/Frs hr = STG_E_NOTFILEBASEDSTORAGE; while (NULL != o_bstrPath) { CComBSTR bstrPath = pShareInfo->shi2_path; BREAK_OUTOFMEMORY_IF_NULL((BSTR)bstrPath, &hr); if (!!bstrRestOfPath) { if ( _T('\\') != *(bstrPath + _tcslen(bstrPath) - 1) ) { bstrPath += _T("\\"); BREAK_OUTOFMEMORY_IF_NULL((BSTR)bstrPath, &hr); } bstrPath += bstrRestOfPath; BREAK_OUTOFMEMORY_IF_NULL((BSTR)bstrPath, &hr); } *o_bstrPath = bstrPath.Detach(); break; } NetApiBufferFree(pShareInfo); } return(hr); } void FreeNetNameList(NETNAMELIST *pList) { if (pList && !pList->empty()) { for (NETNAMELIST::iterator i = pList->begin(); i != pList->end(); i++) delete (*i); pList->clear(); } } void FreeRootInfoList(ROOTINFOLIST *pList) { if (pList && !pList->empty()) { for (ROOTINFOLIST::iterator i = pList->begin(); i != pList->end(); i++) delete (*i); pList->clear(); } } void FreeSubscriberList(SUBSCRIBERLIST *pList) { if (pList && !pList->empty()) { for (SUBSCRIBERLIST::iterator i = pList->begin(); i != pList->end(); i++) delete (*i); pList->clear(); } } HRESULT GetLocalComputerName(OUT BSTR* o_pbstrComputerName) { _ASSERT(o_pbstrComputerName); DWORD dwErr = 0; TCHAR szBuffer[DNS_MAX_NAME_BUFFER_LENGTH]; DWORD dwSize = DNS_MAX_NAME_BUFFER_LENGTH; if ( !GetComputerNameEx(ComputerNameDnsFullyQualified, szBuffer, &dwSize) ) { dwSize = DNS_MAX_NAME_BUFFER_LENGTH; if ( !GetComputerNameEx(ComputerNameNetBIOS, szBuffer, &dwSize) ) dwErr = GetLastError(); } if (0 == dwErr) { *o_pbstrComputerName = SysAllocString(szBuffer); if (!*o_pbstrComputerName) dwErr = ERROR_OUTOFMEMORY; } return HRESULT_FROM_WIN32(dwErr); } HRESULT GetDomainDfsRoots( OUT NETNAMELIST* o_pDfsRootList, IN LPCTSTR i_lpszDomainName ) { HRESULT hr = S_OK; FreeNetNameList(o_pDfsRootList); do { DFS_INFO_200 *pBuffer = NULL; DWORD dwEntriesRead = 0; DWORD dwResumeHandle = 0; DWORD dwErr = NetDfsEnum( const_cast(i_lpszDomainName), 200, // Level, (DWORD)-1, // PrefMaxLen, (LPBYTE *)&pBuffer, &dwEntriesRead, &dwResumeHandle ); if (NERR_Success != dwErr) { hr = (ERROR_NO_MORE_ITEMS == dwErr || ERROR_NOT_FOUND == dwErr) ? S_FALSE : HRESULT_FROM_WIN32(dwErr); break; } NETNAME *pCurrent = NULL; for (DWORD i = 0; i < dwEntriesRead; i++) { pCurrent = new NETNAME; if (!pCurrent) { hr = E_OUTOFMEMORY; break; } pCurrent->bstrNetName = pBuffer[i].FtDfsName; if (!pCurrent->bstrNetName) { delete pCurrent; hr = E_OUTOFMEMORY; break; } o_pDfsRootList->push_back(pCurrent); } NetApiBufferFree(pBuffer); } while (0); if (FAILED(hr)) FreeNetNameList(o_pDfsRootList); return hr; } HRESULT GetMultiDfsRoots( OUT ROOTINFOLIST* o_pDfsRootList, IN LPCTSTR i_lpszScope ) { if (!i_lpszScope || !*i_lpszScope) return E_INVALIDARG; HRESULT hr = S_OK; FreeRootInfoList(o_pDfsRootList); do { DFS_INFO_200 *pBuffer = NULL; DWORD dwEntriesRead = 0; DWORD dwResumeHandle = 0; DWORD dwErr = NetDfsEnum( const_cast(i_lpszScope), 200, // Level, (DWORD)-1, // PrefMaxLen, (LPBYTE *)&pBuffer, &dwEntriesRead, &dwResumeHandle ); if (NERR_Success != dwErr) { hr = (ERROR_NO_MORE_ITEMS == dwErr || ERROR_NOT_FOUND == dwErr) ? S_FALSE : HRESULT_FROM_WIN32(dwErr); break; } ROOTINFO *pCurrent = NULL; for (DWORD i = 0; i < dwEntriesRead; i++) { CComBSTR bstrEntryPath = _T("\\\\"); bstrEntryPath += i_lpszScope; bstrEntryPath += _T("\\"); bstrEntryPath += pBuffer[i].FtDfsName; PDFS_INFO_2 pDfsInfo = NULL; NET_API_STATUS nRet = NetDfsGetInfo( bstrEntryPath, NULL, NULL, 2, (LPBYTE*) &pDfsInfo ); if (NERR_Success != nRet) continue; pCurrent = new ROOTINFO; if (pCurrent) { if ((pDfsInfo->State & DFS_VOLUME_FLAVORS) == DFS_VOLUME_FLAVOR_AD_BLOB) pCurrent->enumRootType = DFS_TYPE_FTDFS; else pCurrent->enumRootType = DFS_TYPE_STANDALONE; pCurrent->bstrRootName = pBuffer[i].FtDfsName; } NetApiBufferFree(pDfsInfo); if (!pCurrent) { hr = E_OUTOFMEMORY; break; } if (!pCurrent->bstrRootName) { delete pCurrent; hr = E_OUTOFMEMORY; break; } o_pDfsRootList->push_back(pCurrent); } NetApiBufferFree(pBuffer); } while (0); if (FAILED(hr)) FreeRootInfoList(o_pDfsRootList); return hr; } /* bool IsDfsPath ( LPTSTR i_lpszNetPath ) { if ( S_OK != CheckUNCPath(i_lpszNetPath) ) return false; CComBSTR bstrPath = i_lpszNetPath; if (!bstrPath) return false; // out of memory TCHAR *s = _tcschr(bstrPath + 2, _T('\\')); // points to the 3rd backslash TCHAR *p = bstrPath + bstrPath.Length(); // points to the ending char '\0' bool bIsDfsPath = false; do { *p = _T('\0'); PDFS_INFO_1 pDfsInfo1 = NULL; DWORD dwStatus = NetDfsGetClientInfo ( bstrPath, // dir path as entrypath NULL, // No server name required NULL, // No share required 1, // level 1 (LPBYTE *)&pDfsInfo1 // out buffer. ); if (NERR_Success == dwStatus) { bIsDfsPath = true; NetApiBufferFree(pDfsInfo1); break; } p--; while (_T('\\') != *p && p > s) { p--; } } while (p > s); return bIsDfsPath; } */ HRESULT CheckUNCPath( IN LPCTSTR i_lpszPath ) /*++ Routine Description: Checks if path is of UNC format. --*/ { //"someserver\someshare\somedir\.." - Invalid //"\\Server - Invalid //"\\someserver\someshare\.. - Valid //"\\someserver\someshare\Somedir\.. - Valid RETURN_INVALIDARG_IF_NULL(i_lpszPath); DWORD dwRetFlags = 0; NET_API_STATUS nStatus = I_NetPathType(NULL, (LPTSTR)i_lpszPath, &dwRetFlags, 0); if (NERR_Success != nStatus) return HRESULT_FROM_WIN32(nStatus); return (ITYPE_UNC == dwRetFlags) ? S_OK : S_FALSE; } HRESULT GetUNCPathComponent( IN LPCTSTR i_lpszPath, OUT BSTR* o_pbstrComponent, IN UINT i_nStartBackslashes, // index starting from 1 IN UINT i_nEndBackslashes // index starting from 1 ) { RETURN_INVALIDARG_IF_NULL(o_pbstrComponent); RETURN_INVALIDARG_IF_TRUE(i_nStartBackslashes < 2); RETURN_INVALIDARG_IF_TRUE(0 != i_nEndBackslashes && i_nEndBackslashes <= i_nStartBackslashes); *o_pbstrComponent = NULL; HRESULT hr = CheckUNCPath(i_lpszPath); RETURN_INVALIDARG_IF_TRUE(S_OK != hr); UINT nDeltaBackslashes = i_nEndBackslashes ? (i_nEndBackslashes - i_nStartBackslashes) : 0; TCHAR *p = (LPTSTR)i_lpszPath + 2; // skip the first 2 backslashes i_nStartBackslashes -= 2; // locate the starting backslash while (*p && i_nStartBackslashes) { if (_T('\\') == *p) i_nStartBackslashes--; p++; } if (!*p) return hr; // locate the ending backslash TCHAR *s = p; if (!nDeltaBackslashes) { s += _tcslen(p); } else { while (*s && nDeltaBackslashes) { if (_T('\\') == *s) nDeltaBackslashes--; s++; } if (*s) s--; } // p points at the first char after the starting backslash, and // s points at the ending backslash or the ending char '\0' *o_pbstrComponent = SysAllocStringLen(p, s - p); RETURN_OUTOFMEMORY_IF_NULL(*o_pbstrComponent); return S_OK; } // return TRUE if string matches the filter string, case-insensitive match BOOL FilterMatch(LPCTSTR lpszString, FILTERDFSLINKS_TYPE lLinkFilterType, LPCTSTR lpszFilter) { BOOL bResult = TRUE; switch (lLinkFilterType) { case FILTERDFSLINKS_TYPE_BEGINWITH: if (!lpszString || !lpszFilter) return FALSE; bResult = !mylstrncmpi(lpszString, lpszFilter, lstrlen(lpszFilter)); break; case FILTERDFSLINKS_TYPE_CONTAIN: { if (!lpszString || !lpszFilter) return FALSE; TCHAR* pszStringLower = _tcsdup(lpszString); TCHAR* pszFilterLower = _tcsdup(lpszFilter); if (!pszStringLower || !pszFilterLower) bResult = FALSE; else bResult = (NULL != _tcsstr(_tcslwr(pszStringLower), _tcslwr(pszFilterLower))); if (pszStringLower) free(pszStringLower); if (pszFilterLower) free(pszFilterLower); } break; default: break; } return bResult; } HRESULT IsHostingDfsRoot(IN BSTR i_bstrServer, OUT BSTR* o_pbstrRootEntryPath) { if (o_pbstrRootEntryPath) *o_pbstrRootEntryPath = NULL; DWORD dwEntriesRead = 0; DWORD dwResumeHandle = 0; DFS_INFO_1* pDfsInfoLevel1 = NULL; NET_API_STATUS dwRet = NetDfsEnum( i_bstrServer, 1, sizeof(DFS_INFO_1), // get 1 entry (LPBYTE *)&pDfsInfoLevel1, &dwEntriesRead, &dwResumeHandle ); if (ERROR_NO_MORE_ITEMS == dwRet || ERROR_NOT_FOUND == dwRet) return S_FALSE; HRESULT hr = S_OK; if (NERR_Success == dwRet) { if (o_pbstrRootEntryPath) { *o_pbstrRootEntryPath = SysAllocString(pDfsInfoLevel1->EntryPath); if (!*o_pbstrRootEntryPath) hr = E_OUTOFMEMORY; } NetApiBufferFree(pDfsInfoLevel1); } else { hr = HRESULT_FROM_WIN32(dwRet); } return hr; } /* void GetTimeDelta(LPCTSTR str, SYSTEMTIME* ptime0, SYSTEMTIME* ptime1) { dfsDebugOut((_T("%s, delta = %d millisec \n"), str, ((ptime1->wMinute - ptime0->wMinute) * 60 + (ptime1->wSecond - ptime0->wSecond)) * 1000 + (ptime1->wMilliseconds - ptime0->wMilliseconds) )); } */