home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 1996 December / PCWKCD1296.iso / vjplusb / activex / inetsdk / samples / wininet / headdump / headdump.cpp next >
Encoding:
C/C++ Source or Header  |  1996-07-11  |  7.2 KB  |  257 lines
  1. /****************************************************************************
  2. *
  3. * HeadDump: requests the default HTML document from the server and prints 
  4. *            along with HTTP headers.
  5. *            Microsoft Corporation (C). Leon Braginski.
  6. *    
  7. * PURPOSE: 
  8. * This sample demonstrates how to create and submit HTTP 
  9. * request. Sample requests the default HTML document from the
  10. * server and then display it along with the HTTP transaction headers.
  11. * This sample also allows to access password protected pages. It 
  12. * checks for HTTP server response code and it is "401 Access Denied"
  13. * it asks password and user name and then resubmit request.
  14. * Note that is only works with the Basic authentication scheme, since 
  15. * this scheme is built into WinInet APIs. Please see documentation for
  16. * WinInet APIs for more details.
  17. *
  18. * COMMENTS: 
  19. * This sample can either work with the direct access to the Internet or
  20. * via proxy server.
  21. * 
  22. * USAGE:
  23. * With direct connection to the Internet:
  24. *    C:> headdump.exe www.server.name
  25. * Via proxy server:
  26. *    C:> headdump.exe www.server.name http://MyProxyServer:ProxyPort
  27. *
  28. * Note that sample continuously output information on the screen, so use like
  29. * this:
  30. *     C:> headdump.exe www.server.name | more
  31. *
  32. * This sample has been tested under Windows NT 4.0 Beta and Windows 95.
  33. * 
  34. ****************************************************************************/
  35.  
  36.  
  37.  
  38. #include <windows.h>
  39. #include <wininet.h>
  40. #include <iostream.h>
  41.  
  42. BOOL ErrorOut  ( DWORD dError, TCHAR * szCallingFunc);
  43.  
  44. int main (int argc, char *argv[])
  45. {
  46.  HINTERNET hOpen, hConnect, hReq;
  47.  DWORD  dwSize, dwCode;
  48.  CHAR *lpBuffer, szData[51];
  49.  
  50.  
  51. switch (argc)
  52. {
  53.     case 1:
  54.     {
  55.         cerr << "Usage: " << argv[0] <<" host [proxy]" << endl;
  56.         cerr << "\twhere host is a HTTP server such as www.server.com" << endl;
  57.         cerr << "\tand proxy is optional proxy in form: http://proxy:80" << endl;
  58.         return 0;
  59.     }
  60.     case 2:
  61.     {
  62.         if ( !(hOpen = InternetOpen ( "HeadDump",  LOCAL_INTERNET_ACCESS , NULL, 0, 0) ) )
  63.         {
  64.             ErrorOut ( GetLastError(), "InternetOpen");
  65.             return 0;
  66.         }
  67.         break;
  68.     }
  69.     case 3:
  70.     {
  71.        if ( !(hOpen = InternetOpen ( "HeadDump",  CERN_PROXY_INTERNET_ACCESS, argv[2], NULL, 0) ) )
  72.        {
  73.             ErrorOut ( GetLastError(), "InternetOpen");
  74.            return 0;
  75.        }
  76.        break;
  77.     }
  78.  }   
  79.         
  80.  
  81. if ( !(hConnect = InternetConnect ( hOpen, argv[1] , INTERNET_INVALID_PORT_NUMBER, "",  "", INTERNET_SERVICE_HTTP, 0  , 0) ) )
  82. {
  83.     ErrorOut (GetLastError(), "InternetConnect");
  84.     return 0;
  85. }
  86.  
  87. if ( !(hReq = HttpOpenRequest (hConnect, "GET", "", HTTP_VERSION, "", NULL, INTERNET_FLAG_RELOAD | INTERNET_FLAG_NO_CACHE_WRITE ,0 )))
  88. {
  89.     ErrorOut (GetLastError(), "HttpOpenRequest");
  90.     return 0;
  91. }
  92.  
  93.  
  94. again:
  95. if (!HttpSendRequest (hReq, NULL, 0, NULL, 0) )
  96. {
  97.     ErrorOut (GetLastError(), "HttpSend");
  98.     return 0;
  99. }
  100.  
  101. dwSize = sizeof (DWORD) ;  
  102. if ( !HttpQueryInfo (hReq, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &dwCode, &dwSize, NULL))
  103. {
  104.     ErrorOut (GetLastError(), "HttpQueryInfo");
  105.     return FALSE;
  106. }
  107.  
  108. if ( dwCode == HTTP_STATUS_DENIED || dwCode == HTTP_STATUS_PROXY_AUTH_REQ)
  109. {  
  110.     // This is a secure page.
  111.     cerr << "This page is password protected. " << endl;
  112.     CHAR szUser[50]="";
  113.     CHAR szPass[50]="";
  114.  
  115.     cerr << "User: ";
  116.     cin  >> szUser;
  117.     cerr << "Pass: ";
  118.     cin  >> szPass;
  119.  
  120.  
  121.     // We have to read all outstanding data on the Internet handle
  122.     // before we can resubmit request. Just discard the data.
  123.     do
  124.     {
  125.         InternetReadFile (hReq, (LPVOID)szData, 50, &dwSize);
  126.     }
  127.     while (dwSize != 0);
  128.  
  129.  
  130.     if ( !InternetSetOption (hConnect, INTERNET_OPTION_USERNAME, (LPVOID) szUser, lstrlen (szUser) ))
  131.     {
  132.         cerr << "InternetSetOptionFailed: " << GetLastError() << endl;
  133.         return FALSE;
  134.     }
  135.     
  136.     if ( !InternetSetOption (hConnect, INTERNET_OPTION_PASSWORD, (LPVOID) szPass, lstrlen (szPass) ))
  137.     {
  138.         cerr << "InternetSetOptionFailed: " << GetLastError() << endl;
  139.         return FALSE;
  140.     }
  141.     goto again;
  142. }
  143.  
  144.  
  145.  
  146.  
  147. // First time we will find out the size of the headers.
  148. HttpQueryInfo (hReq,HTTP_QUERY_RAW_HEADERS_CRLF,  NULL, &dwSize, NULL);
  149. lpBuffer =  new char [dwSize + 1 ];
  150.  
  151. // Now we call HttpQueryInfo again to get the headers.
  152. if (!HttpQueryInfo (hReq,HTTP_QUERY_RAW_HEADERS_CRLF, (LPVOID) lpBuffer,
  153.     &dwSize, NULL))
  154. {
  155.     ErrorOut (GetLastError(), "HttpQueryInfo");
  156.     return FALSE;
  157. }
  158. *(lpBuffer + dwSize) = '\0';
  159. cout << lpBuffer << endl;
  160.  
  161. // We can relay on the CONTENTS_LENGTH header to find the size of the 
  162. // HTML file, since this header may not exist (it is optional for HTTP/1.0
  163. do
  164. {
  165.     if (!InternetReadFile (hReq, (LPVOID)szData, 50, &dwSize) )
  166.     {
  167.          ErrorOut (GetLastError (), "InternetReadFile");
  168.         return FALSE;
  169.     }
  170.     if ( !dwSize)
  171.         break;
  172.     else
  173.     {
  174.         szData [dwSize] = '\0'; 
  175.         cout << szData;
  176.     }
  177. }
  178. while (TRUE);
  179. cout << endl;
  180.  
  181. if (!InternetCloseHandle (hReq) )
  182. {
  183.     ErrorOut (GetLastError (), "CloseHandle on hReq");
  184.     return FALSE;
  185. }
  186. if (!InternetCloseHandle (hConnect) )
  187. {
  188.     ErrorOut (GetLastError (), "CloseHandle on hConnect");
  189.     return FALSE;
  190. }
  191. if (!InternetCloseHandle (hOpen) )
  192. {
  193.     ErrorOut (GetLastError (), "CloseHandle on hOpen");
  194.     return FALSE;
  195. }
  196.  
  197.  
  198. free (lpBuffer);
  199. return TRUE;
  200. }
  201.  
  202. /****************************************************************************
  203. *
  204. *    FUNCTION: ErrorOut
  205. *
  206. *    PURPOSE: This function is used to get extended Internet error.
  207. *
  208. *    COMMENTS:  Function returns TRUE on success and FALSE on failure.
  209. *
  210. ****************************************************************************/
  211.  
  212. BOOL ErrorOut ( DWORD dError, TCHAR * szCallFunc)
  213. {
  214.     TCHAR szTemp[100] = "", *szBuffer=NULL, *szBufferFinal = NULL;
  215.     DWORD  dwIntError , dwLength = 0; 
  216.     wsprintf (szTemp,  "%s error %d\n ", szCallFunc, dError );
  217.     if (dError == ERROR_INTERNET_EXTENDED_ERROR)
  218.     {
  219.         InternetGetLastResponseInfo (&dwIntError, NULL, &dwLength);
  220.         if (dwLength)
  221.         {
  222.             if ( !(szBuffer = (TCHAR *) LocalAlloc ( LPTR,  dwLength) ) )
  223.             {
  224.                 lstrcat (szTemp, TEXT ( "Unable to allocate memory to display Internet error code. Error code: ") );
  225.                 lstrcat (szTemp, TEXT (_itoa (GetLastError(), szBuffer, 10)  ) );
  226.                 lstrcat (szTemp, TEXT ("\n") );
  227.                 cerr << szTemp << endl;
  228.                 return FALSE;
  229.             }
  230.             if (!InternetGetLastResponseInfo (&dwIntError, (LPTSTR) szBuffer, &dwLength))
  231.             {
  232.                 lstrcat (szTemp, TEXT ( "Unable to get Intrnet error. Error code: ") );
  233.                 lstrcat (szTemp, TEXT (_itoa (GetLastError(), szBuffer, 10)  ) );
  234.                 lstrcat (szTemp, TEXT ("\n") );
  235.                 cerr << szTemp << endl;
  236.                 return FALSE;
  237.             }
  238.             if ( !(szBufferFinal = (TCHAR *) LocalAlloc ( LPTR,  (strlen (szBuffer) +strlen (szTemp) + 1)  ) )  )
  239.             {
  240.                 lstrcat (szTemp, TEXT ( "Unable to allocate memory. Error code: ") );
  241.                 lstrcat (szTemp, TEXT (_itoa (GetLastError(), szBuffer, 10)  ) );
  242.                 lstrcat (szTemp, TEXT ("\n") );
  243.                 cerr << szTemp << endl;
  244.                 return FALSE;
  245.             }
  246.             lstrcpy (szBufferFinal, szTemp);
  247.             lstrcat (szBufferFinal, szBuffer);
  248.             LocalFree (szBuffer);
  249.             cerr <<  szBufferFinal  << endl;
  250.             LocalFree (szBufferFinal);
  251.         }
  252.     }
  253.     else
  254.         cerr << szTemp << endl;
  255.     return TRUE;
  256. }
  257.