httpfileobj.cpp
上传用户:dangjiwu
上传日期:2013-07-19
资源大小:42019k
文件大小:54k
源码类别:

Symbian

开发平台:

Visual C++

  1. /* ***** BEGIN LICENSE BLOCK *****
  2.  * Version: RCSL 1.0/RPSL 1.0
  3.  *
  4.  * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved.
  5.  *
  6.  * The contents of this file, and the files included with this file, are
  7.  * subject to the current version of the RealNetworks Public Source License
  8.  * Version 1.0 (the "RPSL") available at
  9.  * http://www.helixcommunity.org/content/rpsl unless you have licensed
  10.  * the file under the RealNetworks Community Source License Version 1.0
  11.  * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl,
  12.  * in which case the RCSL will apply. You may also obtain the license terms
  13.  * directly from RealNetworks.  You may not use this file except in
  14.  * compliance with the RPSL or, if you have a valid RCSL with RealNetworks
  15.  * applicable to this file, the RCSL.  Please see the applicable RPSL or
  16.  * RCSL for the rights, obligations and limitations governing use of the
  17.  * contents of the file.
  18.  *
  19.  * This file is part of the Helix DNA Technology. RealNetworks is the
  20.  * developer of the Original Code and owns the copyrights in the portions
  21.  * it created.
  22.  *
  23.  * This file, and the files included with this file, is distributed and made
  24.  * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  25.  * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  26.  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS
  27.  * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  28.  *
  29.  * Technology Compatibility Kit Test Suite(s) Location:
  30.  *    http://www.helixcommunity.org/content/tck
  31.  *
  32.  * Contributor(s):
  33.  *
  34.  * ***** END LICENSE BLOCK ***** */
  35. /****************************************************************************
  36.  * Includes
  37.  */
  38. #include "hlxclib/stdio.h"              /* FILE                     */
  39. #include "hlxclib/string.h"             /* strcpy, etc.             */
  40. #include "hlxclib/sys/stat.h"           /* stat, etc.               */
  41. #include "hxtypes.h"                    /* UINT32, BOOL, etc        */
  42. #include "hxcom.h"                      /* IUnknown                 */
  43. #include "hxcomm.h"                     /* IHXCommonClassFactory    */
  44. #include "ihxpckts.h"                   /* IHXValues, IHXBuffers    */
  45. #include "hxbuffer.h"                   /* CHXBuffer                */
  46. #include "hxurl.h"                      /* CHXURL                   */
  47. #include "chxpckts.h"                   /* CHXHeader                */
  48. #include "hxcache2.h"                   /* IHXCache2, IHXCacheObject,
  49.                                            IHXCacheObjectResponse   */
  50. #include "httpfilesys.h"                /* FILE_SYS_PROTOCOL        */
  51. #include "httpfileobj.h"                /* CHXHTTPFileObject        */
  52. #include "debug.h"                      /* DPRINTF                  */
  53. #include "chxcache2.h"                  /* CHXCache2                */
  54. #include "hxprefs.h"                    /* IHXPreferences           */
  55. #include "multilog.h"                   /* INIT_MULTILOG_GROUP_NO_COREDEBUG(...) */
  56. #include "mlog_http.h"                  /* MLOG_HTTP(...)           */
  57. #ifdef _SYMBIAN
  58.     #include "symbhxdir.h"              /*  OS_SEPARATOE_CHAR       */
  59. #else
  60.     #include "hxdir.h"
  61. #endif
  62. #define D_HTTP_FO 0x1000000
  63. // The capacity of the Cache in bytes. IMPORTANT >> See the comment for CHUNK_SIZE.
  64. #define CACHE_CAPACITY (12 * 1024)
  65. // The amount of data requested through the TCP socket read operation.
  66. // IMPORTANT !!! The chunk_size should be <= (1 - THRESHOLD/100)*CACHE_CAPACITY
  67. // Otherwise we may get into a position wherein the cache has less than
  68. // THRESHOLD read-data but it won't ever be able to accept any new data as the
  69. // new data buffer size (CHUNK_SIZE) will overflow the cache. Also, since
  70. // read-data is below threshold, the cache will never discard any more data.
  71. #define CHUNK_SIZE (2 * 1024)
  72. // 70%  -> The percentage of read data we would like the cache
  73. // to hold. It is made high because when a request for old data can't
  74. // be satisfied, then the whole connection has to be teared down
  75. // everything has to be restrated.  So, it's better to have this value high.
  76. // Note that some fileformat objects start seeking from start whatever be
  77. // the direction of the User Interface seek. In such cases, threshold doesn't
  78. // matter.
  79. #define THRESHOLD 70
  80. #define DEFAULT_HTTP_PORT 80
  81. #define DEFAULT_CALLBACK_INTERVAL 10 // ms
  82. // A utility function which does searches for a *string* in a
  83. // *character buffer* of size haystacksize. strnstr() wouldn't work since
  84. // it requires both the parameters to be strings.
  85. char* bufnstr(char *haystackBuffer, char* needleString, int haystackSize);
  86. // CHXHTTPFileObject Class Methods
  87. /****************************************************************************
  88.  *  CHXHTTPFileObject::CHXHTTPFileObject
  89.  *
  90.  *  Constructor
  91.  */
  92. CHXHTTPFileObject::CHXHTTPFileObject(IUnknown* pContext, IHXValues* pOptions)
  93.     : m_RefCount                (0),        // See header file for the
  94.       m_pContext                (NULL),     // purpose of these variables.
  95.       m_pOptions                (NULL),
  96.       m_pClassFactory           (NULL),
  97.       m_pFileResponse           (NULL),
  98.       m_pRequest                (NULL),
  99.       m_pCache                  (NULL),
  100.       m_pSocket                 (NULL),
  101.       m_pCHXURL                 (NULL),
  102.       m_ulCurrentReadOffset     (0),
  103.       m_lNewReadOffset          (-1),
  104.       m_bInSeekDone             (FALSE),
  105.       m_bInReadDone             (FALSE),
  106.       m_bReadPending            (FALSE),
  107.       m_bIncompleteReadPending  (FALSE),
  108.       m_ulFileLength            (0),
  109.       m_pHdrListRoot            (NULL),
  110.       m_bHeaderCompletelyRead   (FALSE),
  111.       m_pHeader                 (NULL),
  112.       m_bStartAllOverAgain      (FALSE),
  113.       m_pScheduler              (NULL),
  114.       m_ulCallbackHandle        (0),
  115.       m_ulCallBackInterval      (DEFAULT_CALLBACK_INTERVAL),  // 10 ms
  116.       m_bFirstChunk             (TRUE),
  117.       m_bInitResponsePending    (FALSE),
  118.       m_bInitialized            (FALSE),
  119.       m_ulFileDataRead          (0),
  120.       m_bAddBlockPending        (FALSE),
  121.       m_pPendingAddBlock        (NULL),
  122.       m_bDisconnected           (FALSE)
  123. {
  124.     MLOG_HTTP("CHXHTTPFileObject::CHXHTTPFileObject()n");
  125.     m_pContext = pContext;
  126.     m_pOptions = pOptions;
  127.     // Signify that we need to keep a reference to this object
  128.     if (m_pOptions != NULL)
  129.     {
  130. m_pOptions->AddRef();
  131.     }
  132.     if (m_pContext != NULL)
  133.     {
  134. m_pContext->AddRef();
  135.         m_pContext->QueryInterface(IID_IHXScheduler,
  136.                                  (void**) &m_pScheduler);
  137.         m_pContext->QueryInterface(IID_IHXCommonClassFactory,
  138.  (void**)&m_pClassFactory);
  139.     }
  140. } // CHXHTTPFileObject()
  141. /****************************************************************************
  142.  *  CHXHTTPFileObject::~CHXHTTPFileObject
  143.  *
  144.  *  Destructor.
  145.  */
  146. CHXHTTPFileObject::~CHXHTTPFileObject(void)
  147. {
  148.     MLOG_HTTP("CHXHTTPFileObject::~CHXHTTPFileObject()n");
  149.     _CleanUp();
  150. } // ~CHXHTTPFileObject()
  151. /****************************************************************************
  152. * IHXFileObject::_CleanUp()
  153. *
  154. * Cleans up the object by relesing all member objects. This is a helper
  155. * function used by the Close() and the destructor.
  156. *
  157. */
  158. STDMETHODIMP
  159. CHXHTTPFileObject::_CleanUp(void)
  160. {
  161.     MLOG_HTTP("CHXHTTPFileObject::_CleanUp()n");
  162.     m_bAddBlockPending = FALSE;
  163.     HX_RELEASE(m_pPendingAddBlock);
  164.     
  165.     if (m_ulCallbackHandle)
  166.     {
  167. m_pScheduler->Remove(m_ulCallbackHandle);
  168. m_ulCallbackHandle = 0;
  169.     }
  170.     m_bInitialized = FALSE;
  171.     
  172.     HX_RELEASE(m_pContext);
  173.     HX_RELEASE(m_pOptions);
  174.     HX_RELEASE(m_pClassFactory);
  175.     HX_RELEASE(m_pFileResponse);
  176.     HX_RELEASE(m_pRequest);
  177.     HX_RELEASE(m_pCache);
  178.     HX_RELEASE(m_pSocket);
  179.     HX_DELETE(m_pCHXURL);
  180.     HX_RELEASE(m_pPendingReadInfo.pPendingReadBuff);
  181.     HX_RELEASE(m_pHeader);
  182.     HX_RELEASE(m_pScheduler);
  183.     HX_RELEASE(m_pPendingAddBlock);
  184.     return HXR_OK;
  185. } // _CleanUp()
  186. // IHXFileObject Interface Methods
  187. /****************************************************************************
  188.  *  IHXFileObject::Init
  189.  *
  190.  *  This routine associates this File Object with a File Response object
  191.  *  which is notified when file operations (read, write, seek, etc.) are
  192.  *  complete. This method also checks the validity of the file by actually
  193.  *  opening it.
  194.  */
  195. STDMETHODIMP
  196. CHXHTTPFileObject::Init(UINT32            fileAccessMode, // We ignore access mode here
  197. IHXFileResponse*  pFileResponse)
  198. {
  199.     MLOG_HTTP("CHXHTTPFileObject::Init()n");
  200.     HX_RESULT res = HXR_OK;
  201.     if (pFileResponse != NULL)
  202.     {
  203. // Release any previous File Response objects
  204. if (m_pFileResponse != NULL)
  205. {
  206.     m_pFileResponse->Release();
  207. }
  208. m_pFileResponse = pFileResponse;
  209. m_pFileResponse->AddRef();
  210.     }
  211.     else
  212.     {
  213. return HXR_INVALID_PARAMETER;
  214.     }
  215.     if(m_bInitialized) // Init() was previously called atleast once
  216.     {
  217.         // There is no point continuing with a pending read as
  218.         // the caller has changed.
  219.         m_bReadPending = FALSE;
  220.         m_bIncompleteReadPending = FALSE;
  221.         HX_RELEASE(m_pPendingReadInfo.pPendingReadBuff);
  222.         /* If we have already opened a file, then seek back
  223.          * to zero during re-initialization
  224.          */
  225.         m_ulCurrentReadOffset = 0;
  226.         m_pFileResponse->InitDone(HXR_OK);
  227.         return HXR_OK;
  228.      }
  229.     m_pPendingReadInfo.pPendingReadBuff = NULL;
  230.     // Else, this is the first time Init() is being called
  231.     if(m_pCHXURL == NULL) // Init() being called before calling SetRequest()
  232.     {
  233.         return HXR_INVALID_OPERATION;
  234.     }
  235.     /*
  236.      * We have to check the validity of the file. What we do is we
  237.      * contact the HTTP server and request for the HTTP file header.
  238.      * This lets us do two things:
  239.      *      (1) It lets us know if the file really exists on the server.
  240.      *      (2) If file exists, the response contains the stats of the file.
  241.      */
  242.     res = _Start(); // Commence the action - connect to server, send GET req, etc
  243.     return res;
  244. } // Init()
  245. // Assumes the state of the network related state variables is
  246. // as it would be if this method is called for the first time
  247. // (for eg., from the constructor). So, m_pSocket, etc should all
  248. // be NULL
  249. STDMETHODIMP
  250. CHXHTTPFileObject::_Start()
  251. {
  252.     MLOG_HTTP("CHXHTTPFileObject::_Start()n");
  253.     IHXNetworkServices* pNetworkServices = NULL;
  254.     if (HXR_OK != m_pContext->QueryInterface( IID_IHXNetworkServices,
  255.                                             (void **)&pNetworkServices))
  256.     {
  257.         return HXR_FAIL;
  258.     }
  259.     HX_RESULT res = HXR_OK;
  260.     res = pNetworkServices->CreateTCPSocket(&m_pSocket);
  261.     pNetworkServices->Release();
  262.     if( (res != HXR_OK) || (m_pSocket == NULL) )
  263.     {
  264.         return HXR_FAIL;
  265.     }
  266.     // Identify yourself as the object to whom the results of all
  267.     // socket related operations should be reported.
  268.     res = m_pSocket->Init( (IHXTCPResponse *)this );
  269.     if( ( res != HXR_OK) || (m_pCHXURL == NULL) )
  270.     {
  271.         HX_RELEASE(m_pSocket);
  272.         return HXR_FAIL;
  273.     }
  274.     // Parse the URL to get server address and port
  275.     IHXValues* pHeader = m_pCHXURL->GetProperties();
  276.     IHXBuffer* pBuffer = NULL;
  277.     char *serverAddress = NULL;
  278.     pHeader->GetPropertyBuffer(PROPERTY_HOST, pBuffer);
  279.     if(pBuffer != NULL)
  280.     {
  281.         serverAddress = (char*)(pBuffer->GetBuffer());
  282.     }
  283.     ULONG32 serverPort = 0;
  284.     pHeader->GetPropertyULONG32(PROPERTY_PORT, serverPort);
  285.     // Connect to the HTTP server
  286.     res = m_pSocket->Connect(serverAddress, (UINT16)serverPort);
  287.     if(res != HXR_OK)
  288.     {
  289.         HX_RELEASE(m_pSocket);
  290.         return HXR_FAIL;
  291.     }
  292.     return HXR_OK;
  293. } // _Start()
  294. /****************************************************************************
  295.  *  IHXFileObject::GetFilename
  296.  *
  297.  *  This routine returns the name of the requested file (without any path
  298.  *  information). This method may be called by the File Format plug-in if the
  299.  *  short name of the file is required.
  300.  */
  301. STDMETHODIMP
  302. CHXHTTPFileObject::GetFilename(REF(const char*) pFileName)
  303. {
  304.     MLOG_HTTP("CHXHTTPFileObject::GetFilename()n");
  305.     pFileName = NULL;
  306.     HX_RESULT res = HXR_OK;
  307.     //  From the File path, extract the file name and return it.
  308.     if(m_pCHXURL != NULL)
  309.     {
  310.         IHXBuffer* pBuffer = NULL;
  311.         IHXValues* pHeader = m_pCHXURL->GetProperties();
  312.         pHeader->GetPropertyBuffer(PROPERTY_RESOURCE, pBuffer);
  313.         if(pBuffer != NULL)
  314.         {
  315.             pFileName = (char*)(pBuffer->GetBuffer());
  316.         }
  317.     }
  318.     else
  319.     {
  320.         res = HXR_FAIL;
  321.     }
  322.     return res;
  323. }
  324. /****************************************************************************
  325.  *  IHXFileObject::Read
  326.  *
  327.  *  This routine reads a block of data of the specified length from the file.
  328.  *  When reading has completed, the caller is asynchronously notified via the
  329.  *  File Response object associated with this File Object. This method is
  330.  *  called by the File Format plug-in when it needs to read from the file.
  331.  */
  332. STDMETHODIMP
  333. CHXHTTPFileObject::Read(UINT32 byteCount)
  334. {
  335.     MLOG_HTTP("CHXHTTPFileObject::Read(%u)n", byteCount);
  336.     // The whole file has already been read
  337.     if(m_ulCurrentReadOffset == m_ulFileLength)
  338.     {
  339.         HX_RESULT res = HXR_FAIL;
  340.         if (m_pFileResponse)
  341.         {
  342.             m_pFileResponse->ReadDone(HXR_FAIL, 0);
  343.             res = HXR_OK;
  344.         }
  345.         return res;
  346.     }
  347.     // Can't have a read when a previous read is outstanding
  348.     if(m_bReadPending)
  349.     {
  350.         return HXR_INVALID_OPERATION;
  351.     }
  352.     m_bIncompleteReadPending = FALSE;
  353.     // Can't read more than the (remaining) file length
  354.     int actualLen = m_ulFileLength - m_ulCurrentReadOffset;
  355.     if(actualLen > byteCount)
  356.         actualLen = byteCount;
  357.     m_bReadPending = TRUE;
  358.     m_pPendingReadInfo.pPendingReadBuff = NULL;
  359.     m_pPendingReadInfo.ulWriteOffset = 0;
  360.     m_pPendingReadInfo.ulReadOffset = m_ulCurrentReadOffset;
  361.     m_pPendingReadInfo.ulSize = (UINT32)actualLen;
  362.     HX_RESULT r = m_pCache->ReadBlock(m_ulCurrentReadOffset, (UINT32)actualLen);
  363.     return r;
  364. } // Read()
  365. /****************************************************************************
  366.  *  IHXFileObject::Write
  367.  *
  368.  *  This routine writes a block of data to the file. When writing has
  369.  *  completed, the caller is asynchronously notified via the File Response
  370.  *  object associated with this File Object. This method called by the File
  371.  *  Format plug-in when it needs to write to the file.
  372.  */
  373. STDMETHODIMP
  374. CHXHTTPFileObject::Write(IHXBuffer* pDataToWrite)
  375. {
  376.     MLOG_HTTP("CHXHTTPFileObject::Write()n");
  377.     return HXR_NOTIMPL;
  378. }
  379. /****************************************************************************
  380.  *  IHXFileObject::Seek
  381.  *
  382.  *  This routine moves to a given position in the file. The move can be
  383.  *  either from the beginning of the file (absolute), or relative to the
  384.  *  current file position. When seeking has completed, the caller is
  385.  *  asynchronously notified via the File Response object associated with this
  386.  *  File Object. This method called by the File Format plug-in when it needs
  387.  *  to seek to a location within the file.
  388.  */
  389. STDMETHODIMP
  390. CHXHTTPFileObject::Seek(UINT32 offset, BOOL   bIsRelative)
  391. {
  392.     const char *seekType = NULL;
  393.     seekType = (bIsRelative == TRUE)? "REL": "ABS";
  394.     MLOG_HTTP("CHXHTTPFileObject::Seek(%u, %s)n", offset, seekType);
  395.     m_lNewReadOffset = offset;
  396.     if(bIsRelative)
  397.     {
  398.         m_lNewReadOffset += m_ulCurrentReadOffset;
  399.     }
  400.     // Can't seek to outside the file
  401.     if( (m_lNewReadOffset < 0) || (m_lNewReadOffset > m_ulFileLength) )
  402.     {
  403.         m_lNewReadOffset = -1;
  404.         return HXR_FAIL;
  405.     }
  406.     if(!m_bInSeekDone)
  407.     {
  408.         while(m_lNewReadOffset >= 0)
  409.         {
  410.             // Cancel any pending reads
  411.             m_bReadPending = FALSE;
  412.             m_bIncompleteReadPending = FALSE;
  413.             HX_RELEASE(m_pPendingReadInfo.pPendingReadBuff);
  414.             m_ulCurrentReadOffset = m_lNewReadOffset;
  415.             m_lNewReadOffset = -1;
  416.             m_bInSeekDone = TRUE;
  417.             if (m_pFileResponse)
  418.             {
  419.                 m_pFileResponse->SeekDone(HXR_OK);
  420.             }
  421.             m_bInSeekDone = FALSE;
  422.         }
  423.     }
  424.     return HXR_OK;
  425. }
  426. /****************************************************************************
  427.  *  IHXFileObject::Advise
  428.  *
  429.  *  This routine is passed information about the intended usage of this
  430.  *  object. The useage will indicate whether sequential or random requests
  431.  *  for information will be made. This may be useful, for example, in
  432.  *  developing a caching scheme.
  433.  */
  434. STDMETHODIMP
  435. CHXHTTPFileObject::Advise(UINT32  usage)
  436. {
  437.     MLOG_HTTP("CHXHTTPFileObject::Advice(%u)n", usage);
  438.     HX_RESULT res = HXR_FAILED;
  439.     // Because of the simple cache, only
  440.     // linear useage is allowed.
  441.     if(HX_FILEADVISE_RANDOMACCESS == usage)
  442.     {
  443. res = HXR_ADVISE_PREFER_LINEAR;
  444.     }
  445.     return res;
  446. } // Advise()
  447. /****************************************************************************
  448.  *  IHXFileObject::Close
  449.  *
  450.  *  This routine closes the file resource and releases all resources
  451.  *  associated with the object. This routine is crucial and must be called
  452.  *  before the File Object is destroyed.
  453.  */
  454. STDMETHODIMP
  455. CHXHTTPFileObject::Close(void)
  456. {
  457.     MLOG_HTTP("CHXHTTPFileObject::Close()n");
  458.     /*
  459.      * Store this in temp since calling _CleanUp()
  460.      * causes m_pFileResponse to get released. We will
  461.      * have pCallCloseDone on the stack to call
  462.      * CloseDone() and then to safely release.
  463.      */
  464.     IHXFileResponse* pCallCloseDone = m_pFileResponse;
  465.     if(pCallCloseDone)
  466.         pCallCloseDone->AddRef();
  467.     HX_RELEASE(m_pFileResponse);
  468.     if(pCallCloseDone)
  469.     {
  470.         pCallCloseDone->CloseDone(HXR_OK);
  471.         pCallCloseDone->Release();
  472.     }
  473.     
  474.     return HXR_OK;
  475. } // Close()
  476. // IHXRequestHandler Interface Methods
  477. /****************************************************************************
  478.  *  IHXRequestHandler::SetRequest
  479.  *
  480.  *  This routine associates this File Object with the file Request object
  481.  *  passed to it from the Helix core. This Request object is primarily used to
  482.  *  obtain the requested URL. This method is called just after the File
  483.  *  Object is created.
  484.  */
  485. STDMETHODIMP
  486. CHXHTTPFileObject::SetRequest(IHXRequest* pRequest)
  487. {
  488.     MLOG_HTTP("CHXHTTPFileObject::SetRequest()n");
  489.     HX_RESULT res = HXR_OK;
  490.     if(pRequest == NULL)
  491.     {
  492.         res = HXR_INVALID_PARAMETER;
  493.     }
  494.     else
  495.     {
  496.         // Release any previous request object, URL
  497.         HX_RELEASE(m_pRequest);
  498.         if(m_pCHXURL != NULL)
  499.         {
  500.             delete m_pCHXURL;
  501.             m_pCHXURL = NULL;
  502.         }
  503.         // Store a reference to the object
  504.         m_pRequest = pRequest;
  505.         m_pRequest->AddRef();
  506.         const char *pURL = NULL;
  507.         if(m_pRequest->GetURL(pURL) != HXR_OK)
  508.         {
  509.             HX_RELEASE(m_pRequest);
  510.             return HXR_FAIL;
  511.         }
  512.         // Store the URL in a CHXURL object for parsing.
  513.         m_pCHXURL = new CHXURL(pURL);
  514.         if(m_pCHXURL == NULL)
  515.         {
  516.             HX_RELEASE(m_pRequest);
  517.             return HXR_FAIL;
  518.             //return HXR_OUTOFMEMORY;
  519.         }
  520.     }
  521.     return res;
  522. } // SetRequest()
  523. /****************************************************************************
  524.  *  IHXRequestHandler::GetRequest
  525.  *
  526.  *  This routine retrieves the Request object associated with this File
  527.  *  Object. It is called just after the SetRequest() method.
  528.  */
  529. STDMETHODIMP
  530. CHXHTTPFileObject::GetRequest(REF(IHXRequest*) pRequest)
  531. {
  532.     MLOG_HTTP("CHXHTTPFileObject::GetRequest()n");
  533.     pRequest = m_pRequest;
  534.     if (pRequest != NULL)
  535.     {
  536. pRequest->AddRef();
  537.     }
  538.     return HXR_OK;
  539. }
  540. // IUnknown COM Interface Methods
  541. /****************************************************************************
  542.  *  IUnknown::AddRef
  543.  *
  544.  *  This routine increases the object reference count in a thread safe
  545.  *  manner. The reference count is used to manage the lifetime of an object.
  546.  *  This method must be explicitly called by the user whenever a new
  547.  *  reference to an object is used.
  548.  */
  549. STDMETHODIMP_(UINT32) CHXHTTPFileObject::AddRef(void)
  550. {
  551.     MLOG_HTTP("CHXHTTPFileObject::AddRef()n");
  552.     return InterlockedIncrement(&m_RefCount);
  553. }
  554. /****************************************************************************
  555.  *  IUnknown::Release
  556.  *
  557.  *  This routine decreases the object reference count in a thread safe
  558.  *  manner, and deletes the object if no more references to it exist. It must
  559.  *  be called explicitly by the user whenever an object is no longer needed.
  560.  */
  561. STDMETHODIMP_(UINT32) CHXHTTPFileObject::Release(void)
  562. {
  563.     MLOG_HTTP("CHXHTTPFileObject::Release()n");
  564.     if (InterlockedDecrement(&m_RefCount) > 0)
  565.     {
  566. return m_RefCount;
  567.     }
  568.     delete this;
  569.     return 0;
  570. }
  571. /****************************************************************************
  572.  *  IUnknown::QueryInterface
  573.  *
  574.  *  This routine indicates which interfaces this object supports. If a given
  575.  *  interface is supported, the object's reference count is incremented, and
  576.  *  a reference to that interface is returned. Otherwise a NULL object and
  577.  *  error code are returned. This method is called by other objects to
  578.  *  discover the functionality of this object.
  579.  */
  580. STDMETHODIMP CHXHTTPFileObject::QueryInterface(REFIID interfaceID,
  581.        void** ppInterfaceObj)
  582. {
  583.     MLOG_HTTP("CHXHTTPFileObject::QueryInterface()n");
  584.     // Initially assume no interfaces are supported
  585.     HX_RESULT res = HXR_NOINTERFACE;
  586.     *ppInterfaceObj = NULL;
  587.     // By definition all COM objects support the IUnknown interface
  588.     if (IsEqualIID(interfaceID, IID_IUnknown))
  589.     {
  590. *ppInterfaceObj = (IUnknown*)(IHXFileObject*)this;
  591.     }
  592.     // IHXFileObject interface is supported
  593.     else if (IsEqualIID(interfaceID, IID_IHXFileObject))
  594.     {
  595. *ppInterfaceObj = (IHXFileObject*)this;
  596.     }
  597.     // IHXRequestHandler interface is supported
  598.     else if (IsEqualIID(interfaceID, IID_IHXRequestHandler))
  599.     {
  600. *ppInterfaceObj = (IHXRequestHandler*)this;
  601.     }
  602.     // IHXFileStat interface is supported
  603.     else if (IsEqualIID(interfaceID, IID_IHXFileStat))
  604.     {
  605. *ppInterfaceObj = (IHXFileStat*)this;
  606.     }
  607.     // IHXCacheObjectResponse interface is supported
  608.     else if(IsEqualIID(interfaceID, IID_IHXCacheObjectResponse))
  609.     {
  610. *ppInterfaceObj = (IHXCacheObjectResponse*)this;
  611.     }
  612.     // IHXTCPResponse interface is supported
  613.     else if (IsEqualIID(interfaceID, IID_IHXTCPResponse))
  614.     {
  615. *ppInterfaceObj = (IHXTCPResponse*)this;
  616.     }
  617.     // IHXCallback interface is supported
  618.     else if (IsEqualIID(interfaceID, IID_IHXCallback))
  619.     {
  620. *ppInterfaceObj = (IHXCallback*)this;
  621.     }
  622.     if (*ppInterfaceObj)
  623.     {
  624. AddRef();
  625. res = HXR_OK;
  626.     }
  627.     return res;
  628. }
  629. /************************************************************************
  630.  * Method:
  631.  * IHXFileStat::Stat
  632.  * Purpose:
  633.  * Collects information about the file that is returned to the
  634.  * caller in an IHXStat object
  635.  */
  636. STDMETHODIMP
  637. CHXHTTPFileObject::Stat(IHXFileStatResponse* pFileStatResponse)
  638. {
  639.     MLOG_HTTP("CHXHTTPFileObject::Stat()n");
  640.     if(pFileStatResponse == NULL)
  641.     {
  642.         return HXR_INVALID_PARAMETER;
  643.     }
  644.     // It shouldn't call this function before it gets a InitDone() (by which time
  645.     // header would have been completely read).
  646.     // >>> Later, give proper values for last four parameters too...
  647.     if(m_ulFileLength > 0)
  648.     {
  649.         pFileStatResponse->StatDone(HXR_OK, m_ulFileLength, 0, 0, 0, HX_S_IFREG);
  650.     }
  651.     else
  652.     {
  653.         return HXR_FAIL;
  654.     }
  655.     return HXR_OK;
  656. }
  657. /*
  658.  * IHXCacheObjectResponse methods
  659.  */
  660. /************************************************************************
  661.  * Method:
  662.  *
  663.  *     IHXCacheObjectResponse::InitDone
  664.  *
  665.  * Purpose:
  666.  *
  667.  *     Notification that IHXCacheObject::Init call has completed.
  668.  */
  669. STDMETHODIMP
  670. CHXHTTPFileObject::InitDone(HX_RESULT   /*IN*/ status)
  671. {
  672.     MLOG_HTTP("CHXHTTPFileObject::InitDone(%s)n", StrRep(status));
  673.     return HXR_OK;
  674. } // InitDone()
  675. /************************************************************************
  676.  * Method:
  677.  *
  678.  *     IHXCacheObjectResponse::ChangeCapacityDone
  679.  *
  680.  * Purpose:
  681.  *
  682.  *     Notification that IHXCacheObject::ChangeCapacity call has completed.
  683.  */
  684. STDMETHODIMP
  685. CHXHTTPFileObject::ChangeCapacityDone(HX_RESULT /*IN*/ status)
  686. {
  687.     MLOG_HTTP("CHXHTTPFileObject::ChangeCapacityDone(%s)n", StrRep(status));
  688.     return HXR_NOTIMPL;
  689. } // ChangeCapacityDone()
  690. /************************************************************************
  691.  * Method:
  692.  *
  693.  *     IHXCacheObjectResponse::AddBlockDone
  694.  *
  695.  * Purpose:
  696.  *
  697.  *     Notification that IHXCacheObject::AddBlock operation has completed.
  698.  */
  699. STDMETHODIMP
  700. CHXHTTPFileObject::AddBlockDone(HX_RESULT /*IN*/ status)
  701. {
  702.     MLOG_HTTP("CHXHTTPFileObject::AddBlockDone(%s)n", StrRep(status));
  703.     // Can the Cache accomodate more data that might arrive
  704.     // should we request the network for some.
  705.     UINT32 uReadSize = m_ulFileLength - m_ulFileDataRead;
  706.     if(uReadSize > CHUNK_SIZE)
  707.         uReadSize = CHUNK_SIZE;
  708.     if(status == HXR_OUTOFMEMORY)
  709.     {
  710.         // Try to add this block again when you read
  711.         // something of the cache (i.e., in ReadBlockDone()
  712.         m_bAddBlockPending = TRUE;
  713.     }
  714.     else
  715.     {
  716.         HX_RELEASE(m_pPendingAddBlock);
  717.         // Ask the network services to get more data.
  718.         m_pSocket->Read((UINT16)uReadSize);
  719.     }
  720.     return HXR_OK;
  721. } // AddBlockDone()
  722. /************************************************************************
  723.  * Method:
  724.  *
  725.  *     IHXCacheObjectResponse::VerifyBlockDone
  726.  *
  727.  * Purpose:
  728.  *
  729.  *     Notification that IHXCacheObject::VerifyBlock operation has
  730.  *     completed.
  731.  */
  732. STDMETHODIMP
  733. CHXHTTPFileObject::VerifyBlockDone(BOOL /*IN*/ bExist)
  734. {
  735.     MLOG_HTTP("CHXHTTPFileObject::VerifyBlockDone(%d)n", bExist);
  736.     return HXR_NOTIMPL;
  737. } // VerifyBlockDone()
  738. /************************************************************************
  739.  * Method:
  740.  *
  741.  *     IHXCacheObjectResponse::ReadBlockDone
  742.  *
  743.  * Purpose:
  744.  *
  745.  *     Notification that IHXCacheObject::ReadBlock operation has
  746.  *     completed.
  747.  */
  748. STDMETHODIMP
  749. CHXHTTPFileObject::ReadBlockDone(HX_RESULT /*IN*/ status,
  750.  IHXBuffer* /*IN*/ pBuffer)
  751. {
  752.     MLOG_HTTP("CHXHTTPFileObject::ReadBlockDone(%s)n", StrRep(status));
  753.     // If the cache has failed to give us anydata and it will not do
  754.     // so in the future and at the same time we notice that the server
  755.     // has disconnected, then raise alarm. Note that we need to check
  756.     // for all these conditions because if the server disconnected after
  757.     // serving all data and if it is stored in the cache, then we shouldn't
  758.     // be worried.
  759.     if(
  760.             (status == HXR_INCOMPLETE) && (pBuffer == NULL) &&
  761.             (m_bDisconnected == TRUE)  && (m_pCache->IsEmpty())
  762.         )
  763.     {
  764.         if (m_pFileResponse)
  765.         {
  766.             m_pFileResponse->ReadDone(HXR_SERVER_DISCONNECTED, NULL);
  767.         }
  768.         return HXR_OK;
  769.     }
  770.     if(pBuffer != NULL)
  771.         pBuffer->AddRef();
  772.     HX_RESULT res = HXR_OK;
  773.     // This is a response to a ReadBlock() called from Func(). Func() does this
  774.     // only when there is an incomplete Read.
  775.     if(m_bIncompleteReadPending)
  776.     {
  777.         if( ((status == HXR_OK) || (status == HXR_INCOMPLETE)) && (pBuffer != NULL) )
  778.         {
  779.             // Copy data into the buffer set aside for the incomplete read.
  780.             memcpy((void*)(m_pPendingReadInfo.pPendingReadBuff->GetBuffer() + m_pPendingReadInfo.ulWriteOffset),
  781.                            (void*)(pBuffer->GetBuffer()), pBuffer->GetSize());
  782.             // Update offsets and sizes for the next read installment
  783.             m_pPendingReadInfo.ulWriteOffset += pBuffer->GetSize(); // Next installment buffer-write offset
  784.             m_pPendingReadInfo.ulReadOffset += pBuffer->GetSize();  // Next installment cache-read offset
  785.             m_pPendingReadInfo.ulSize -= pBuffer->GetSize();        // Next installment max-size
  786.         }
  787.         if(HXR_OK == status)
  788.         {
  789.             // This installment finally fills up the incomplete read buffer.
  790.             m_ulCurrentReadOffset = m_pPendingReadInfo.ulReadOffset;
  791.             m_bIncompleteReadPending = FALSE;
  792.             m_bReadPending = FALSE;
  793.             // Get a handle since there is a possibilily of m_pPendingReadInfo
  794.             // being modified in some method called from ReadDone().
  795.             IHXBuffer* pBuff = m_pPendingReadInfo.pPendingReadBuff;
  796.             m_pPendingReadInfo.pPendingReadBuff = NULL;
  797.             
  798.             if (m_pFileResponse)
  799.             {
  800.                 m_pFileResponse->ReadDone(HXR_OK, pBuff);
  801.             }
  802.             HX_RELEASE(pBuff);
  803.             pBuff = NULL;
  804.         }
  805.         else if(status == HXR_INCOMPLETE) // Need to get more installments to fill up ReadBuffer
  806.         {
  807.             m_ulCallbackHandle = m_pScheduler->RelativeEnter(this, m_ulCallBackInterval);
  808.         }
  809.         else
  810.         {
  811.             // Can't happen
  812.         }
  813.     }
  814.     else // !m_bIncompleteReadPending
  815.     {
  816.         if(status == HXR_OK) // Cool, hand data over to the FF object
  817.         {
  818.             m_ulCurrentReadOffset = m_pPendingReadInfo.ulReadOffset + m_pPendingReadInfo.ulSize;
  819.             m_bReadPending = FALSE;
  820.             if (m_pFileResponse)
  821.             {
  822.                 m_pFileResponse->ReadDone(HXR_OK, pBuffer);
  823.             }
  824.         }
  825.         // This means that the Cache has only partial data. Create a buffer to store this partial data.
  826.         // Mark this read as incomplete and try to read from the cache in installments.
  827.         // Request scheduler to signal you after some time. When the scheduler signals
  828.         // you (via Func()), check to see if Cache has recieved anymore data.
  829.         else if(status == HXR_INCOMPLETE)
  830.         {
  831.             if(m_pClassFactory != NULL)
  832.             {
  833.                 // Create a buffer to fill it in installments.
  834.                 m_pClassFactory->CreateInstance(CLSID_IHXBuffer,
  835.                 (void**)&m_pPendingReadInfo.pPendingReadBuff);
  836.                 if (m_pPendingReadInfo.pPendingReadBuff != NULL)
  837.                 {
  838.                     res =  m_pPendingReadInfo.pPendingReadBuff->SetSize(m_pPendingReadInfo.ulSize);
  839.                     if(HXR_OK == res)
  840.                     {
  841.                         if(pBuffer != NULL)
  842.                         {
  843.                             // Copy the partial data into the pending read buffer.
  844.                             memcpy((void*)(m_pPendingReadInfo.pPendingReadBuff->GetBuffer()),
  845.                                    (void*)(pBuffer->GetBuffer()), pBuffer->GetSize());
  846.                             m_pPendingReadInfo.ulWriteOffset = pBuffer->GetSize();  // Next installment buffer-write offset
  847.                             m_pPendingReadInfo.ulReadOffset += pBuffer->GetSize();  // Next installment cache-read offset
  848.                             m_pPendingReadInfo.ulSize -= pBuffer->GetSize();        // Next installment max-size
  849.                         }
  850.                     }
  851.                     else
  852.                     {
  853.                         HX_RELEASE(m_pPendingReadInfo.pPendingReadBuff);
  854.                         res = HXR_OUTOFMEMORY;
  855.                     }
  856.                 }
  857.                 else
  858.                 {
  859.                     res = HXR_OUTOFMEMORY;
  860.                 }
  861.             }
  862.             else
  863.             {
  864.                 res = HXR_UNEXPECTED;
  865.             }
  866.             // Not enough memory to create buffers.
  867.             // Let's do the best we can do! Return the incomplete data.
  868.             if(HXR_OK != res)
  869.             {
  870.                 m_bReadPending = FALSE;
  871.                 m_ulCurrentReadOffset = m_pPendingReadInfo.ulReadOffset;
  872.                 if (m_pFileResponse)
  873.                 {
  874.                     m_pFileResponse->ReadDone(HXR_INCOMPLETE, pBuffer);
  875.                 }
  876.             }
  877.             else
  878.             {
  879.                 // We will try to fill the remaining data in the buffer after waiting for some
  880.                 // time. We ask the scheduler to call us back after some time and then we check if the
  881.                 // Cache has recieved any new data.
  882.                 m_bIncompleteReadPending = TRUE;
  883.                 m_ulCallbackHandle = m_pScheduler->RelativeEnter(this, m_ulCallBackInterval);
  884.             }
  885.         }
  886.         // status = HXR_FAIL, i.e, the Cache had discarded some part of the data
  887.         // we requeted. The cache has already discarded the data and so there is no hope
  888.         // of getting that data. The only way to satisfy the Read request
  889.         // is to tear down the connection and start-all-over-again.
  890.         else //status = HXR_FAIL and the reason is request for past data
  891.         {
  892.             // Set this flag so that other activities are all paralysed
  893.             m_bStartAllOverAgain = TRUE;
  894.             m_bAddBlockPending = FALSE;
  895.             HX_RELEASE(m_pPendingAddBlock);
  896.             m_pPendingAddBlock = NULL;
  897.             // Close the socket
  898.             HX_RELEASE(m_pSocket);
  899.             m_bIncompleteReadPending = TRUE;
  900.             if(m_pClassFactory != NULL)
  901.             {
  902.                 // Create a buffer and fill it in installments.
  903.                 m_pClassFactory->CreateInstance(CLSID_IHXBuffer,
  904.                 (void**)&m_pPendingReadInfo.pPendingReadBuff);
  905.                 m_pPendingReadInfo.pPendingReadBuff->SetSize(m_pPendingReadInfo.ulSize);
  906.             }
  907.             // Flush the cache so that we can freshly start
  908.             // storing the new info into it.
  909.             m_pCache->Flush();
  910.             // Rest of the the "start-all-over-again" continues in FlushDone()
  911.         }
  912.     } // !m_bIncompleteReadPending
  913.     if(m_bAddBlockPending)
  914.     {
  915.         // Now that we have read some data out of cache, maybe the cache would accept
  916.         // the previously rejected data.
  917.         if( (pBuffer != NULL) && (pBuffer->GetSize() > 0) )
  918.         {
  919.             if(m_pCache->GetUnusedCapacity() >= m_pPendingAddBlock->GetSize())
  920.             {
  921.                 m_ulCallbackHandle = m_pScheduler->RelativeEnter(this, 0);
  922.             }
  923.         }
  924.     }
  925.     HX_RELEASE(pBuffer);
  926.     pBuffer = NULL;
  927.     return HXR_OK;
  928. } // ReadBlockDone()
  929. /************************************************************************
  930.  * Method:
  931.  *
  932.  *     IHXCacheObjectResponse::FlushDone
  933.  *
  934.  * Purpose:
  935.  *
  936.  *     Notification that IHXCacheObject::Flush operation has completed.
  937.  */
  938. STDMETHODIMP
  939. CHXHTTPFileObject::FlushDone(HX_RESULT  /*IN*/ status)
  940. {
  941.     MLOG_HTTP("CHXHTTPFileObject::FlushDone(%s)n", StrRep(status));
  942.     // Discard the old header so that the incoming new
  943.     // hdr is not confused for data.
  944.     HX_RELEASE(m_pHeader);
  945.     m_pHeader = NULL;
  946.     m_bHeaderCompletelyRead = FALSE;
  947.     m_ulFileDataRead = 0;
  948.     m_bFirstChunk = TRUE;   // The TCP chunk that will be recived later
  949.                             // will be the first one from the new connection.
  950.     m_bStartAllOverAgain = FALSE;
  951.     // Start the whole process (viz., connecting, GET request, etc) all over again.
  952.     _Start();
  953.     return HXR_OK;
  954. } // FlushDone()
  955. /*
  956.  * IHXTCPResponse methods
  957.  */
  958. /************************************************************************
  959.  * Method:
  960.  *     IHXTCPResponse::ConnectDone
  961.  * Purpose:
  962.  *     A Connect operation has been completed or an error has occurred.
  963.  */
  964. STDMETHODIMP
  965. CHXHTTPFileObject::ConnectDone(HX_RESULT status)
  966. {
  967.     MLOG_HTTP("CHXHTTPFileObject::ConnectDone(%s)n", StrRep(status));
  968.     if(status != HXR_OK)
  969.     {
  970.         if(!m_bInitialized && m_pFileResponse)
  971.         {
  972.            m_pFileResponse->InitDone(status);
  973.         }
  974.         return HXR_OK;
  975.     }
  976.     // OK, the connection has been established. Now send a GET request to
  977.     // start getting the file.
  978.     char* HttpGetReq = NULL;
  979.     UINT32 ulReqLen = 0;
  980.     HX_RESULT res = HXR_OK;
  981.     res = _PrepareHTTP10GetMessage(HttpGetReq, ulReqLen);
  982.     if(res != HXR_OK)
  983.     {
  984.         if(!m_bInitialized && m_pFileResponse)
  985.         {
  986.             m_pFileResponse->InitDone(HXR_OUTOFMEMORY);
  987.         }
  988.         return HXR_OK;
  989.     }
  990.     CHXBuffer* pSendBuffer = new CHXBuffer((UCHAR*)HttpGetReq, ulReqLen);
  991.     if(pSendBuffer == NULL)
  992.     {
  993.         if(!m_bInitialized && m_pFileResponse)
  994.         {
  995.             m_pFileResponse->InitDone(HXR_OUTOFMEMORY);
  996.         }
  997.         return HXR_OK;
  998.     }
  999.     pSendBuffer->AddRef();
  1000.     // Now write the HTTP/1.0 GET request into the socket
  1001.     res = m_pSocket->Write( (IHXBuffer *)pSendBuffer );
  1002.     // Ask for the response
  1003.     res = m_pSocket->Read((UINT16)CHUNK_SIZE);
  1004.     HX_RELEASE(pSendBuffer);
  1005.     return res;
  1006. } //  ConnectDone()
  1007. /************************************************************************
  1008.  * Method:
  1009.  *     IHXTCPResponse::ReadDone
  1010.  * Purpose:
  1011.  *     A Read operation has been completed or an error has occurred.
  1012.  *     The data is returned in the IHXBuffer.
  1013.  */
  1014. STDMETHODIMP
  1015. CHXHTTPFileObject::ReadDone(HX_RESULT status,
  1016.     IHXBuffer* pBuffer)
  1017. {
  1018.     MLOG_HTTP("CHXHTTPFileObject::ReadDone(%s)n", StrRep(status));
  1019.     if(pBuffer != NULL)
  1020.         pBuffer->AddRef();
  1021.     // A backward seek had failed. Which means the Cache is going
  1022.     // to be flushed, connection tore down and everythings going to
  1023.     // start all over again. So, no point storing this in the cache.
  1024.     if(m_bStartAllOverAgain)
  1025.     {
  1026.         HX_RELEASE(pBuffer);
  1027.         return HXR_OK;
  1028.     }
  1029.     if(!m_bHeaderCompletelyRead) // The TCP data we recvd is a chunk of HTTP header
  1030.     {
  1031.         if( (status != HXR_OK) || (pBuffer == NULL) )
  1032.         {
  1033.             if(!m_bInitialized && m_pFileResponse)
  1034.             {
  1035.                 m_pFileResponse->InitDone(HXR_FAIL);
  1036.             }
  1037.             return HXR_OK;
  1038.         }
  1039.         // Check if response indicates success
  1040.         if(m_bFirstChunk)  // If it is the first chunk from the server
  1041.         {
  1042.             m_bFirstChunk = FALSE;
  1043.             const char pSuccess[] = "200 OKrn"; // HTTP success code
  1044.             // Is the starting line "HTTP/1.x 200 OKrn" ?
  1045.             if( memcmp((void*)(pBuffer->GetBuffer() + 9), (void*)pSuccess, strlen(pSuccess)) != 0 )
  1046.             {
  1047.                 if(!m_bInitialized)
  1048.                 {
  1049.                     m_bFirstChunk = TRUE;
  1050.                     // The requested file is not present on the web server
  1051.                     if (m_pFileResponse)
  1052.                     {
  1053.                         m_pFileResponse->InitDone(HXR_DOC_MISSING);
  1054.                     }
  1055.                 }
  1056.                 return HXR_OK;
  1057.             }
  1058.         }
  1059.         // Append this header chunk to the list of such chunks being maintained.
  1060.         HdrChunk *hi = new HdrChunk;
  1061.         hi->pPartOfHdr = pBuffer;
  1062.         hi->next = NULL;
  1063.         if(m_pHdrListRoot == NULL)
  1064.         {
  1065.             m_pHdrListRoot = hi;
  1066.         }
  1067.         else
  1068.         {
  1069.             HdrChunk *temp = NULL;
  1070.             for(temp = m_pHdrListRoot; temp->next != NULL; temp = temp->next);
  1071.             temp->next = hi;
  1072.         }
  1073.         // Is this the last header chunk? If so, does this chunk
  1074.         // also contain some non-header data? If true, store this
  1075.         // non-hdr data in the Cache and consolidate all the previous
  1076.         // header chunks into a single buffer. Then destroy the
  1077.         // header-chunks list.
  1078.         // Note!!!! GetBuffer() doesn't return a NULL terminated string. So,
  1079.         // shouldn't use strstr(). That's why I used my utility function bufnstr()!
  1080.         char *marker = bufnstr((char*)(hi->pPartOfHdr->GetBuffer()), "rnrn",
  1081.                                 hi->pPartOfHdr->GetSize());
  1082.         if(marker == NULL) // End of hdr not found in this chunk
  1083.         {
  1084.             return m_pSocket->Read((UINT16)CHUNK_SIZE); // Read more data
  1085.         }
  1086.         else
  1087.         {
  1088.             m_bHeaderCompletelyRead = TRUE;
  1089.         }
  1090.         // This chunk has the end-of-header marker.
  1091.         // Consolidate all the header chunks
  1092.         marker += 4; // Move the pointer just after the header end-marker.
  1093.         int hdrLen = 0;
  1094.         int dataLen = 0;
  1095.         char *hdr = NULL;
  1096.         char *data = NULL;
  1097.         CHXBuffer* dataBuff = NULL;
  1098.         int len = 0;
  1099.         for(HdrChunk *temp = m_pHdrListRoot; temp != hi; temp = temp->next)
  1100.             len += temp->pPartOfHdr->GetSize();
  1101.         int lastPktHdrLen = marker - (char *)(hi->pPartOfHdr->GetBuffer());
  1102.         hdrLen = len + lastPktHdrLen;
  1103.         dataLen = hi->pPartOfHdr->GetSize() - lastPktHdrLen;
  1104.         hdr = new char[hdrLen];
  1105.         data = new char[dataLen];
  1106.         if( (hdr == NULL) || (data == NULL) )
  1107.         {
  1108.             if(!m_bInitialized && m_pFileResponse)
  1109.             {
  1110.                 m_pFileResponse->InitDone(HXR_OUTOFMEMORY);
  1111.             }
  1112.             return HXR_OK;
  1113.         }
  1114.         int i = 0;
  1115.         for(HdrChunk *temp2 = m_pHdrListRoot; temp2 != hi; temp2 = temp2->next)
  1116.         {
  1117.             int buffLen = temp2->pPartOfHdr->GetSize();
  1118.             memcpy((void*)(hdr + i), (void*)(temp2->pPartOfHdr->GetBuffer()),buffLen);
  1119.             i += buffLen;
  1120.         }
  1121.         // m_ulFileDataRead is the amount of file data that has been read
  1122.         // so far.
  1123.         m_ulFileDataRead = dataLen;
  1124.         memcpy((void*)hdr, (void*)(hi->pPartOfHdr->GetBuffer()), lastPktHdrLen);
  1125.         memcpy((void*)data, (void*)(hi->pPartOfHdr->GetBuffer() + lastPktHdrLen), dataLen);
  1126.         m_pHeader = new CHXBuffer((UCHAR*)hdr, hdrLen);
  1127.         m_pHeader->AddRef();
  1128.         dataBuff = new CHXBuffer((UCHAR*)data, dataLen);
  1129.         dataBuff->AddRef();
  1130.         if( (!m_pHeader) || (!dataBuff) )
  1131.         {
  1132.             delete hdr;
  1133.             delete data;
  1134.             if(!m_bInitialized && m_pFileResponse)
  1135.             {
  1136.                 m_pFileResponse->InitDone(HXR_OUTOFMEMORY);
  1137.             }
  1138.             return HXR_OK;
  1139.         }
  1140.         // Destroy the header-chunks list
  1141.         HdrChunk *temp3 = m_pHdrListRoot;
  1142.         HdrChunk *temp4 = NULL;
  1143.         while(temp3 != NULL)
  1144.         {
  1145.             temp4 = temp3;
  1146.             temp3 = temp3->next;
  1147.             temp4->pPartOfHdr->Release();
  1148.             delete temp4;
  1149.         }
  1150.         m_pHdrListRoot = NULL;
  1151.         // Parse the header for file statistics
  1152.         marker = bufnstr((char*)(m_pHeader->GetBuffer()), "rnContent-Length: ",
  1153.                             m_pHeader->GetSize());
  1154.         if(marker == NULL)
  1155.         {
  1156.             m_ulFileLength = 0;
  1157.             HX_RELEASE(m_pHeader);
  1158.             m_pHeader = NULL;
  1159.             HX_RELEASE(dataBuff);
  1160.             if(!m_bInitialized && m_pFileResponse)
  1161.             {
  1162.                 m_pFileResponse->InitDone(HXR_INVALID_FILE);
  1163.             }
  1164.             return HXR_OK;
  1165.         }
  1166.         marker += 18;
  1167.         int remLen = m_pHeader->GetSize() - (marker - (char*)(m_pHeader->GetBuffer()));
  1168.         char *marker2 = bufnstr(marker, "rn", remLen);
  1169.         if(marker2 == NULL)
  1170.         {
  1171.             m_ulFileLength = 0;
  1172.             HX_RELEASE(m_pHeader);
  1173.             m_pHeader = NULL;
  1174.             HX_RELEASE(dataBuff);
  1175.             if(!m_bInitialized && m_pFileResponse)
  1176.             {
  1177.                 m_pFileResponse->InitDone(HXR_INVALID_FILE);
  1178.             }
  1179.             return HXR_OK;
  1180.         }
  1181.         int l = marker2 - marker;
  1182.         char *flen = new char[l + 1];
  1183.         memcpy((void*)flen, (void*)marker, l);
  1184.         flen[l] = '';
  1185.         m_ulFileLength = atol(flen);
  1186.         delete flen;
  1187.         if(m_pCache == NULL)
  1188.         {
  1189.             CHXCache2 *pObj = new CHXCache2();
  1190.             pObj->AddRef();
  1191.             #ifdef HELIX_FEATURE_HTTP_FILECACHE
  1192.                 BOOL bCacheEnabled = FALSE;
  1193.                 char *pCacheFile = NULL;
  1194.                 _GetCachePreferences(bCacheEnabled, pCacheFile);
  1195.                 if(bCacheEnabled)
  1196.                 {
  1197.                     pObj->CreateFileCacheObject(&m_pCache, m_pClassFactory,
  1198.                                         m_ulFileLength, pCacheFile);
  1199.                 }
  1200.             #endif
  1201.             if(m_pCache == NULL)
  1202.             {
  1203.                 pObj->CreateMemCacheObject(&m_pCache, m_pClassFactory);
  1204.             }
  1205.             pObj->Release();
  1206.             m_pCache->Init((IHXCacheObjectResponse*)this, CACHE_CAPACITY,
  1207.                                         THRESHOLD);
  1208.         }
  1209.         // You are on the network thread, so you can't call InitDone(). Let the
  1210.         // scheduler do it.
  1211.         if(!m_bInitialized)
  1212.         {
  1213.             m_bInitResponsePending = TRUE;
  1214.             m_ulCallbackHandle = m_pScheduler->RelativeEnter(this, 0);
  1215.         }
  1216.         // Store the non-header portion of the latest chunk in cache.
  1217.         m_pPendingAddBlock = (IHXBuffer*)dataBuff;
  1218.         m_pCache->AddBlock((IHXBuffer*)dataBuff);
  1219.         if(m_bIncompleteReadPending)
  1220.         {
  1221.             // Now that we have got the data for which we had to "startAllOverAgain",
  1222.             // let Func() take over and satisfy the "failed" Read().
  1223.             m_ulCallbackHandle = m_pScheduler->RelativeEnter(this, 0);
  1224.         }
  1225.     }
  1226.     else // The TCP data we recvd is applications data (i.e., not part of HTTP header)
  1227.     {
  1228.         if( (status == HXR_OK) && (pBuffer != NULL) )
  1229.         {
  1230.             m_ulFileDataRead += pBuffer->GetSize();
  1231.             m_pPendingAddBlock = pBuffer;
  1232.             m_pCache->AddBlock(pBuffer);
  1233.         }
  1234.     }
  1235.     return HXR_OK;
  1236. } //  ReadDone()
  1237. /************************************************************************
  1238.  * Method:
  1239.  *     IHXTCPResponse::WriteReady
  1240.  * Purpose:
  1241.  *     This is the response method for WantWrite.
  1242.  *     If HX_RESULT is ok, then the TCP channel is ok to Write to.
  1243.  */
  1244. STDMETHODIMP
  1245. CHXHTTPFileObject::WriteReady(HX_RESULT status)
  1246. {
  1247.     MLOG_HTTP("CHXHTTPFileObject::WriteReady(%s)n", StrRep(status));
  1248.     return HXR_NOTIMPL;
  1249. } //  WriteReady()
  1250. /************************************************************************
  1251.  * Method:
  1252.  *     IHXTCPResponse::Closed
  1253.  * Purpose:
  1254.  *     This method is called to inform you that the TCP channel has
  1255.  *     been closed by the peer or closed due to error.
  1256.  */
  1257. STDMETHODIMP
  1258. CHXHTTPFileObject::Closed(HX_RESULT status)
  1259. {
  1260.     MLOG_HTTP("CHXHTTPFileObject::Closed(%s)n", StrRep(status));
  1261.     m_bDisconnected = TRUE;
  1262.     return HXR_OK;
  1263. } //  Closed()
  1264. /************************************************************************
  1265.  * IHXCallback::Func()
  1266.  */
  1267. STDMETHODIMP
  1268. CHXHTTPFileObject::Func()
  1269. {
  1270.     MLOG_HTTP("CHXHTTPFileObject::Func()n");
  1271.     if(m_bInitResponsePending && m_pFileResponse)
  1272.     {
  1273.         m_bInitResponsePending = FALSE;
  1274.         m_bInitialized = TRUE;
  1275.         m_pFileResponse->InitDone(HXR_OK);
  1276.     }
  1277.     // Is there any incomplete read pending?
  1278.     if(m_bIncompleteReadPending && m_pCache && m_pFileResponse)
  1279.     {
  1280.         // Ask the cache if it can supply the next installment for the
  1281.         // incomplete read.
  1282.         HX_RESULT res = m_pCache->ReadBlock(m_pPendingReadInfo.ulReadOffset,
  1283.                                             m_pPendingReadInfo.ulSize);
  1284.         if(res != HXR_OK)
  1285.         {
  1286.             m_bIncompleteReadPending = FALSE;
  1287.             IHXBuffer* pBuff = m_pPendingReadInfo.pPendingReadBuff;
  1288.             m_pPendingReadInfo.pPendingReadBuff = NULL;
  1289.             if (m_pFileResponse)
  1290.             {
  1291.                 m_pFileResponse->ReadDone(HXR_INCOMPLETE, pBuff);
  1292.             }
  1293.             HX_RELEASE(pBuff);
  1294.             pBuff = NULL;
  1295.         }
  1296.     }
  1297.     // Is there any block which was rejected by the cache earlier because it
  1298.     // was full? If so, let's try inserting into the cache again. Maybe, the
  1299.     // cache has since freed some space.
  1300.     if(m_bAddBlockPending && m_pCache)
  1301.     {
  1302.         m_bAddBlockPending = FALSE;
  1303.         m_pCache->AddBlock(m_pPendingAddBlock);
  1304.     }
  1305.     return HXR_OK;
  1306. } // Func()
  1307. // A utility function which does searches for a *string* in a
  1308. // *character buffer* of size haystacksize. strnstr() wouldn't work since
  1309. // it requires both the parameters to be strings.
  1310. char* bufnstr(char *haystackBuffer, char* needleString, int haystackSize)
  1311. {
  1312.     char lastchar = haystackBuffer[haystackSize];
  1313.     haystackBuffer[haystackSize] = ''; // Now, it's a string
  1314.     char* marker = strstr(haystackBuffer, needleString);
  1315.     if(marker != NULL)
  1316.     {
  1317.         haystackBuffer[haystackSize] = lastchar;
  1318.         return marker;
  1319.     }
  1320.     // Perhaps the pattern occurs at the very end?
  1321.     haystackBuffer[haystackSize] = lastchar;
  1322.     int needleSize = strlen(needleString);
  1323.     marker = haystackBuffer + haystackSize - needleSize;
  1324.     char *tailPiece = new char[needleSize + 1];
  1325.     if(tailPiece == NULL)
  1326.         return NULL;
  1327.     memcpy((void*)tailPiece, (void*)marker, needleSize);
  1328.     tailPiece[needleSize] = '';
  1329.     if( strcmp(tailPiece, needleString) != 0 )
  1330.         marker = NULL;
  1331.     delete tailPiece;
  1332.     return marker;
  1333. } // bufnstr()
  1334. // From the preferences, get the directory and file where the
  1335. // contents should be cached.
  1336. STDMETHODIMP
  1337. CHXHTTPFileObject::_GetCachePreferences(BOOL& bCacheEnabled, char* &pCacheFile )
  1338. {
  1339.     IHXPreferences* pPreferences;
  1340.     m_pContext->QueryInterface(IID_IHXPreferences, (void **)&pPreferences);
  1341.     HX_ASSERT(pPreferences);
  1342.     // Get cache directory, if not set and available
  1343.     IHXBuffer* pBuffer  = NULL;
  1344.     // Check to see if the cache is enabled
  1345.     bCacheEnabled = TRUE;
  1346.     if (pPreferences->ReadPref("CacheEnabled", pBuffer) == HXR_OK)
  1347.     {
  1348.         // Cache enabled entry exists
  1349.         if (atoi((const char*)pBuffer->GetBuffer()))
  1350.         {
  1351.             // Cache enabled
  1352.             bCacheEnabled = TRUE;
  1353.         }
  1354.         else
  1355.         {
  1356.             // Cache disabled
  1357.             bCacheEnabled = FALSE;
  1358.         }
  1359.     }
  1360.     else
  1361.     {
  1362.         bCacheEnabled = FALSE;
  1363.     }
  1364.     if(bCacheEnabled == FALSE)
  1365.     {
  1366.         HX_RELEASE(pBuffer);
  1367.         return HXR_OK;
  1368.     }
  1369.     char separator[2] = {OS_SEPARATOR_CHAR, ''};
  1370.     BOOL bCacheDirSpecified = FALSE;
  1371.     if (pPreferences->ReadPref("CacheFolder", pBuffer) == HXR_OK)
  1372.     {
  1373.         // Make sure that the given filename belongs to a directory
  1374.         struct stat statbuf = { 0 };
  1375.         // If cache database filename does not refer to a file, drop the filename part
  1376.         if (!stat ((const char*)pBuffer->GetBuffer(), &statbuf))
  1377.         {
  1378.           if (statbuf.st_mode & S_IFDIR)
  1379.           {
  1380.               // Valid cache directory specified
  1381.               bCacheDirSpecified = TRUE;
  1382.           }
  1383.         }
  1384.     }
  1385.     if( bCacheDirSpecified == FALSE )
  1386.     {
  1387.         return HXR_FAIL;
  1388.     }
  1389.     char *cacheDir = (char*)(pBuffer->GetBuffer());
  1390.     // Now append the file name
  1391.     HX_ASSERT(m_pCHXURL);
  1392.     IHXValues* pHeader = m_pCHXURL->GetProperties();
  1393.     HX_ASSERT(pHeader);
  1394.     IHXBuffer* pPath = NULL;
  1395.     char *path = NULL;
  1396.     IHXBuffer* pFullPath = NULL;
  1397.     char* fullPath = NULL;
  1398.     pHeader->GetPropertyBuffer(PROPERTY_PATH, pPath);
  1399.     if(pPath != NULL)
  1400.     {
  1401.         path = (char*)(pPath->GetBuffer());
  1402.     }
  1403.     pHeader->GetPropertyBuffer(PROPERTY_FULLPATH, pFullPath);
  1404.     if(pFullPath!= NULL)
  1405.     {
  1406.         fullPath = (char*)(pFullPath->GetBuffer());
  1407.     }
  1408.     char *justFileName = &fullPath[strlen(path)];
  1409.     UINT32 l1 = strlen(cacheDir);
  1410.     UINT32 l2 = strlen(separator);
  1411.     UINT32 l3 = strlen(justFileName);
  1412.     pCacheFile = new char[l1 + l2 + l3 + 1];
  1413.     HX_ASSERT(pCacheFile);
  1414.     memcpy( (void*)pCacheFile, (void*)cacheDir, l1);
  1415.     memcpy( (void*)(pCacheFile + l1), (void*)separator, l2);
  1416.     memcpy( (void*)(pCacheFile + l1 + l2), (void*)justFileName, l3);
  1417.     pCacheFile[l1 + l2 + l3] = '';
  1418.     HX_RELEASE(pBuffer);
  1419.     HX_RELEASE(pPath);
  1420.     HX_RELEASE(pFullPath);
  1421.     return HXR_OK;
  1422. } // _GetCachePreferences()
  1423. // Prepare a HTTP/1.0 GET Request Message with the appropriate headers.
  1424. STDMETHODIMP
  1425. CHXHTTPFileObject::_PrepareHTTP10GetMessage(char* &pGetMsg,
  1426.                                             UINT32 &ulMsgLen)
  1427. {
  1428.     char* SP = " ";     // space
  1429.     char* CRLF = "rn";
  1430.     char* CRLFCRLF = "rnrn";
  1431.     char* resource = NULL;
  1432.     IHXBuffer* pBuffer = NULL;
  1433.     IHXValues* pHeader = m_pCHXURL->GetProperties();
  1434.     pHeader->GetPropertyBuffer(PROPERTY_RESOURCE, pBuffer);
  1435.     if(pBuffer != NULL)
  1436.     {
  1437.         resource = (char*)(pBuffer->GetBuffer());
  1438.     }
  1439.     if(resource == NULL)
  1440.     {
  1441.         return HXR_UNEXPECTED;
  1442.     }
  1443.     CHXString ReqStr;
  1444.     ReqStr += (const char*)"GET /";
  1445.     ReqStr += (const char*)resource;
  1446.     ReqStr += (const char*)SP;
  1447.     ReqStr += (const char*)"HTTP/1.0";
  1448.     IHXValues* pHeaders = NULL;
  1449.     m_pRequest->GetRequestHeaders(pHeaders);
  1450.     const char* pKey = NULL;
  1451.     IHXBuffer* pStr = NULL;
  1452.     IHXKeyValueList* pKeyVals = NULL;
  1453.     m_pClassFactory->CreateInstance(CLSID_IHXKeyValueList,
  1454.     (void**)&pKeyVals);
  1455.     pKeyVals->ImportValues(pHeaders);
  1456.     IHXKeyValueListIter* pIter;
  1457.     pKeyVals->GetIter(pIter);
  1458.     // Attach the appropriate HTTP Request headers
  1459.     HX_RESULT retVal = pIter->GetNextPair(pKey, pStr);
  1460.     while(retVal == HXR_OK)
  1461.     {
  1462.         ReqStr += (const char*)CRLF;
  1463.         ReqStr += (const char*)pKey;
  1464.         ReqStr += (const char*)": ";
  1465.         ReqStr += (const char*)pStr->GetBuffer();
  1466.         retVal = pIter->GetNextPair(pKey, pStr);
  1467.     }
  1468.     ReqStr += (const char*)CRLFCRLF;
  1469.     // Set the function's out parameters
  1470.     ulMsgLen = ReqStr.GetLength();
  1471.     pGetMsg = NULL;
  1472.     pGetMsg = new char[ulMsgLen];
  1473.     if(pGetMsg == NULL)
  1474.     {
  1475.         return HXR_OUTOFMEMORY;
  1476.     }
  1477.     memcpy(pGetMsg, ReqStr.GetBuffer(ulMsgLen), ulMsgLen);
  1478.     return HXR_OK;
  1479. } // _PrepareHTTP10GetMessage()