AfxParseURL的源码

/////////////////////////////////////////////////////////////////////////////

// Global Functions

AFX_STATIC BOOL AFXAPI _AfxParseURLWorker(LPCTSTR pstrURL,

LPURL_COMPONENTS lpComponents, DWORD& dwServiceType,

INTERNET_PORT& nPort, DWORD dwFlags)

{

// this function will return bogus stuff if lpComponents

// isn't set up to copy the components

ASSERT(lpComponents != NULL && pstrURL != NULL);

if (lpComponents == NULL || pstrURL == NULL)

return FALSE;

ASSERT(lpComponents->dwHostNameLength == 0 ||

lpComponents->lpszHostName != NULL);

ASSERT(lpComponents->dwUrlPathLength == 0 ||

lpComponents->lpszUrlPath != NULL);

ASSERT(lpComponents->dwUserNameLength == 0 ||

lpComponents->lpszUserName != NULL);

ASSERT(lpComponents->dwPasswordLength == 0 ||

lpComponents->lpszPassword != NULL);

ASSERT(AfxIsValidAddress(lpComponents, sizeof(URL_COMPONENTS), TRUE));

LPTSTR pstrCanonicalizedURL;

TCHAR szCanonicalizedURL[INTERNET_MAX_URL_LENGTH];

DWORD dwNeededLength = INTERNET_MAX_URL_LENGTH;

BOOL bRetVal;

BOOL bMustFree = FALSE;

DWORD dwCanonicalizeFlags = dwFlags &

(ICU_NO_ENCODE | ICU_DECODE | ICU_NO_META |

ICU_ENCODE_SPACES_ONLY | ICU_BROWSER_MODE);

DWORD dwCrackFlags = dwFlags & (ICU_ESCAPE | ICU_USERNAME);

bRetVal = InternetCanonicalizeUrl(pstrURL, szCanonicalizedURL,

&dwNeededLength, dwCanonicalizeFlags);

if (!bRetVal)

{

if (::GetLastError() != ERROR_INSUFFICIENT_BUFFER)

return FALSE;

pstrCanonicalizedURL = new TCHAR[dwNeededLength];

bMustFree = TRUE;

bRetVal = InternetCanonicalizeUrl(pstrURL, pstrCanonicalizedURL,

&dwNeededLength, dwCanonicalizeFlags);

if (!bRetVal)

{

delete [] pstrCanonicalizedURL;

return FALSE;

}

}

else

pstrCanonicalizedURL = szCanonicalizedURL;

// now that it's safely canonicalized, crack it

bRetVal = InternetCrackUrl(pstrCanonicalizedURL, 0,

dwCrackFlags, lpComponents);

if (bMustFree)

delete [] pstrCanonicalizedURL;

// convert to MFC-style service ID

if (!bRetVal)

dwServiceType = AFX_INET_SERVICE_UNK;

else

{

nPort = lpComponents->nPort;

switch (lpComponents->nScheme)

{

case INTERNET_SCHEME_FTP:

dwServiceType = AFX_INET_SERVICE_FTP;

break;

case INTERNET_SCHEME_GOPHER:

dwServiceType = AFX_INET_SERVICE_GOPHER;

break;

case INTERNET_SCHEME_HTTP:

dwServiceType = AFX_INET_SERVICE_HTTP;

break;

case INTERNET_SCHEME_HTTPS:

dwServiceType = AFX_INET_SERVICE_HTTPS;

break;

case INTERNET_SCHEME_FILE:

dwServiceType = AFX_INET_SERVICE_FILE;

break;

case INTERNET_SCHEME_NEWS:

dwServiceType = AFX_INET_SERVICE_NNTP;

break;

case INTERNET_SCHEME_MAILTO:

dwServiceType = AFX_INET_SERVICE_MAILTO;

break;

default:

dwServiceType = AFX_INET_SERVICE_UNK;

}

}

return bRetVal;

}

BOOL AFXAPI AfxParseURLEx(LPCTSTR pstrURL, DWORD& dwServiceType,

CString& strServer, CString& strObject, INTERNET_PORT& nPort,

CString& strUsername, CString& strPassword, DWORD dwFlags/* = 0*/)

{

dwServiceType = AFX_INET_SERVICE_UNK;

ASSERT(pstrURL != NULL);

if (pstrURL == NULL)

return FALSE;

URL_COMPONENTS urlComponents;

memset(&urlComponents, 0, sizeof(URL_COMPONENTS));

urlComponents.dwStructSize = sizeof(URL_COMPONENTS);

urlComponents.dwHostNameLength = INTERNET_MAX_HOST_NAME_LENGTH;

urlComponents.lpszHostName = strServer.GetBuffer(INTERNET_MAX_HOST_NAME_LENGTH+1);

urlComponents.dwUrlPathLength = INTERNET_MAX_PATH_LENGTH;

urlComponents.lpszUrlPath = strObject.GetBuffer(INTERNET_MAX_PATH_LENGTH+1);

urlComponents.dwUserNameLength = INTERNET_MAX_USER_NAME_LENGTH;

urlComponents.lpszUserName = strUsername.GetBuffer(INTERNET_MAX_USER_NAME_LENGTH+1);

urlComponents.dwPasswordLength = INTERNET_MAX_PASSWORD_LENGTH;

urlComponents.lpszPassword = strPassword.GetBuffer(INTERNET_MAX_PASSWORD_LENGTH+1);

BOOL bRetVal = _AfxParseURLWorker(pstrURL, &urlComponents,

dwServiceType, nPort, dwFlags);

strServer.ReleaseBuffer();

strObject.ReleaseBuffer();

strUsername.ReleaseBuffer();

strPassword.ReleaseBuffer();

return bRetVal;

}

BOOL AFXAPI AfxParseURL(LPCTSTR pstrURL, DWORD& dwServiceType,

CString& strServer, CString& strObject, INTERNET_PORT& nPort)

{

dwServiceType = AFX_INET_SERVICE_UNK;

ASSERT(pstrURL != NULL);

if (pstrURL == NULL)

return FALSE;

URL_COMPONENTS urlComponents;

memset(&urlComponents, 0, sizeof(URL_COMPONENTS));

urlComponents.dwStructSize = sizeof(URL_COMPONENTS);

urlComponents.dwHostNameLength = INTERNET_MAX_URL_LENGTH;

urlComponents.lpszHostName = strServer.GetBuffer(INTERNET_MAX_URL_LENGTH+1);

urlComponents.dwUrlPathLength = INTERNET_MAX_URL_LENGTH;

urlComponents.lpszUrlPath = strObject.GetBuffer(INTERNET_MAX_URL_LENGTH+1);

BOOL bRetVal = _AfxParseURLWorker(pstrURL, &urlComponents,

dwServiceType, nPort, ICU_BROWSER_MODE);

strServer.ReleaseBuffer();

strObject.ReleaseBuffer();

return bRetVal;

}