2012-04-05 33 views
1

是否有任何簡單的方法來判斷UNC路徑是否指向本地機器。 我發現以下問題SO找出UNC路徑是否指向本地機器

是否有任何WIN32 API會做同樣的事情?

+0

的[PathIsSameRoot](http://msdn.microsoft.com/en-us/library/windows/desktop/bb773687%28v=vs.85%29.aspx)函數聽起來有前途的。不過,我不知道它是否接受UNC路徑。 – 2012-04-05 18:30:53

回答

0
#include <windows.h> 
#include <WinSock.h> 
#include <string> 
#include <algorithm> 

#pragma comment(lib, "wsock32.lib") 

using namespace std; 

std::wstring ExtractHostName(const std::wstring &share) 
{ 
if (share.size() < 3) 
    return L""; 
size_t pos = share.find( L"\\", 2); 
wstring server = (pos != string::npos) ? share.substr(2, pos - 2) : share.substr(2); 
transform(server.begin(),server.end(), server.begin(), tolower); 
return server; 
} 

bool IsIP(const std::wstring &server) 
{ 
size_t invalid = server.find_first_not_of( L"."); 
bool fIsIP = (invalid == string::npos); 
return fIsIP; 
} 

bool IsLocalIP(const std::wstring &ipToCheck) 
{ 

if (ipToCheck == L"127.0.0.1") 
    return true;  

WORD wVersionRequested; 
WSADATA wsaData; 
wVersionRequested = MAKEWORD(1, 1); 
if (WSAStartup(wVersionRequested, &wsaData) != 0) 
    return false; 

bool fIsLocal = false; 

char hostName[255]; 
if(gethostname (hostName, sizeof(hostName)) == 0) 
{ 
    PHOSTENT hostinfo;  
    if((hostinfo = gethostbyname(hostName)) != NULL) 
    { 
     for (int i = 0; hostinfo->h_addr_list[i]; i++) 
     { 
      char *ip = inet_ntoa(*(struct in_addr *)hostinfo->h_addr_list[i]); 
      wchar_t wcIP[100]={0}; 
      ::MultiByteToWideChar(CP_ACP, 0, ip, -1, wcIP, _countof(wcIP)); 
      if (ipToCheck == wcIP) 
      { 
       fIsLocal = true; 
       break; 
      } 
     } 
    } 
} 
WSACleanup(); 
return fIsLocal; 
} 

bool IsLocalHost(const std::wstring &server) 
{ 
if (server == L"localhost") 
    return true;  

bool fIsLocalHost = false; 

wchar_t buffer[MAX_PATH]={0}; 
DWORD dwSize = _countof(buffer); 
BOOL fRet = GetComputerName(buffer, &dwSize); 
transform(buffer, buffer + dwSize, buffer, tolower); 
fIsLocalHost = (server == buffer); 
return fIsLocalHost; 
} 

bool ShareIsLocal(const std::wstring &share) 
{ 
wstring server = ExtractHostName(share); 
bool fIsIp = IsIP(server); 
if (fIsIp) 
    return IsLocalIP(server); 
else 
    return IsLocalHost(server); 
}