gecko-dev/netwerk/cache/src/nsCacheEntry.h

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 *
 * The contents of this file are subject to the Mozilla Public
 * License Version 1.1 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of
 * the License at http://www.mozilla.org/MPL/
 * 
 * Software distributed under the License is distributed on an "AS
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * rights and limitations under the License.
 * 
 * The Original Code is nsCacheEntry.h, released February 22, 2001.
 * 
 * The Initial Developer of the Original Code is Netscape Communications
 * Corporation.  Portions created by Netscape are
 * Copyright (C) 2001 Netscape Communications Corporation.  All
 * Rights Reserved.
 * 
 * Contributor(s): 
 *    Gordon Sheridan, 22-February-2001
 */

#ifndef _nsCacheEntry_h_
#define _nsCacheEntry_h_

#include "nspr.h"
#include "pldhash.h"
#include "nscore.h"
#include "nsCOMPtr.h"
#include "nsString.h"
#include "nsAReadableString.h"

#include "nsICache.h"
#include "nsICacheEntryDescriptor.h"
#include "nsCacheMetaData.h"


class nsCacheDevice;
class nsCacheMetaData;
class nsCacheRequest;
class nsCacheEntryDescriptor;


class nsCacheEntry
{
public:

    nsCacheEntry(nsCString *          key,
                 PRBool               streamBased,
                 nsCacheStoragePolicy storagePolicy);
    ~nsCacheEntry();

    nsCString *  Key(void)  { return mKey; }

    PRInt32  FetchCount(void)                     { return mFetchCount;}
    void     SetFetchCount( PRInt32   count)      { mFetchCount = count;}

    PRTime   LastFetched(void)                    { return mLastFetched;}
    void     SetLastFetched( PRTime lastFetched)  { mLastFetched = lastFetched;}

    PRTime   LastValidated(void)                     { return mLastValidated;}
    void     SetLastValidated( PRTime lastValidated) { mLastValidated = lastValidated;}

    PRTime   ExpirationTime(void)               { return mExpirationTime;}
    void     SetExpirationTime( PRTime expires) { mExpirationTime = expires;}

    PRUint32 DataSize(void)                     { return mDataSize;}
    void     SetDataSize( PRUint32   size)      { mDataSize = size;}

    PRUint32 MetaDataSize(void)                 { return mMetaSize;}
    void     SetMetaDataSize( PRUint32   size)  { mMetaSize = size;}

    nsCacheDevice * CacheDevice(void)                        { return mCacheDevice;}
    void            SetCacheDevice( nsCacheDevice * device)  { mCacheDevice = device;}

    nsresult GetData( nsISupports ** result);
    nsresult SetData( nsISupports *  data);

    nsresult GetMetaDataElement( const nsAReadableCString&  key,
                                 nsAReadableCString **      value);
    nsresult SetMetaDataElement( const nsAReadableCString&  key,
                                 const nsAReadableCString&  value);

    nsresult FlattenMetaData(char ** data, PRUint32 * size);
    nsresult UnflattenMetaData(char * data, PRUint32 size);

    //** enumerate MetaData method


    enum CacheEntryFlags {
        eDoomedMask          = 0x00000001,
        eEntryDirtyMask      = 0x00000002,
        eDataDirtyMask       = 0x00000004,
        eMetaDataDirtyMask   = 0x00000008,
        eStreamDataMask      = 0x00000010,
        eActiveMask          = 0x00000020,
        eInitializedMask     = 0x00000040,
        eValidMask           = 0x00000080,
        //        eStoragePolicyMask = 0x00000300
        eAllowedInMemoryMask = 0x00000100,
        eAllowedOnDiskMask   = 0x00000200
        
    };

    void MarkDoomed()          { mFlags |=  eDoomedMask; }
    void MarkEntryDirty()      { mFlags |=  eEntryDirtyMask; }
    void MarkEntryClean()      { mFlags &= ~eEntryDirtyMask; }
    void MarkDataDirty()       { mFlags |=  eDataDirtyMask; }
    void MarkDataClean()       { mFlags &= ~eDataDirtyMask; }
    void MarkMetaDataDirty()   { mFlags |=  eMetaDataDirtyMask; }
    void MarkMetaDataClean()   { mFlags &= ~eMetaDataDirtyMask; }
    void MarkStreamData()      { mFlags |=  eStreamDataMask; }
    void MarkActive()          { mFlags |=  eActiveMask; }
    void MarkInactive()        { mFlags &= ~eActiveMask; }
    void MarkValid()           { mFlags |=  eValidMask; }
    void MarkInvalid()         { mFlags &= ~eValidMask; }
    void MarkAllowedInMemory() { mFlags |=  eAllowedInMemoryMask; }
    void MarkAllowedOnDisk()   { mFlags |=  eAllowedOnDiskMask; }

    PRBool IsDoomed()          { return (mFlags & eDoomedMask) != 0; }
    PRBool IsEntryDirty()      { return (mFlags & eEntryDirtyMask) != 0; }
    PRBool IsDataDirty()       { return (mFlags & eDataDirtyMask) != 0; }
    PRBool IsMetaDataDirty()   { return (mFlags & eMetaDataDirtyMask) != 0; }
    PRBool IsStreamData()      { return (mFlags & eStreamDataMask) != 0; }
    PRBool IsActive()          { return (mFlags & eActiveMask) != 0; }
    PRBool IsInitialized()     { return (mFlags & eInitializedMask) != 0; }
    PRBool IsValid()           { return (mFlags & eValidMask) != 0; }
    PRBool IsAllowedInMemory() { return (mFlags & eAllowedInMemoryMask) != 0; }
    PRBool IsAllowedOnDisk()   { return (mFlags & eAllowedOnDiskMask) !=0; } 

    // methods for nsCacheService
    nsresult Open(nsCacheRequest *request, nsICacheEntryDescriptor ** result);
    nsresult AsyncOpen(nsCacheRequest *request);
    nsresult Doom(void);
    PRBool   RemoveRequest( nsCacheRequest * request);
    PRBool   RemoveDescriptor( nsCacheEntryDescriptor * descriptor);

    PRCList*              GetListNode(void)        { return &mListLink;   }
    static nsCacheEntry*  GetInstance(PRCList* qp) {
        return (nsCacheEntry*) ((char*)qp - offsetof(nsCacheEntry, mListLink));
    }

    
private:
    friend class nsCacheEntryHashTable;
    friend class nsCacheService;

    // internal methods
    nsresult CommonOpen(nsCacheRequest * request, nsCacheAccessMode *accessGranted);
    void MarkStreamBased() { mFlags |= eStreamDataMask; }
    void MarkInitialized() { mFlags |= eInitializedMask; }

    PRCList                mListLink;       // 8  for holding entry on various lists
    nsCString *            mKey;            // 4  //** ask scc about const'ness
    PRUint32               mFetchCount;     // 4
    PRTime                 mLastFetched;    // 8
    PRTime                 mLastValidated;  // 8
    PRTime                 mExpirationTime; // 8
    PRUint32               mFlags;          // 4
    PRUint32               mDataSize;       // 4
    PRUint32               mMetaSize;       // 4
    nsCacheDevice *        mCacheDevice;    // 4
    nsCOMPtr<nsISupports>  mData;           // 4
    nsCacheMetaData *      mMetaData;       // 4
    PRCList                mRequestQ;       // 8 = 64
    PRCList                mDescriptorQ;    // 8 = 72
};


typedef struct {
    PLDHashNumber  keyHash;
    nsCacheEntry  *cacheEntry;
} nsCacheEntryHashTableEntry;


class nsCacheEntryHashTable
{
public:
    nsCacheEntryHashTable();
    ~nsCacheEntryHashTable();

    nsresult      Init();

    nsCacheEntry *GetEntry( const nsCString * key);
    nsresult      AddEntry( nsCacheEntry *entry);
    nsresult      RemoveEntry( nsCacheEntry *entry);
    //** enumerate entries?

private:

    // PLDHashTable operation callbacks
    static const void *   GetKey( PLDHashTable *table, PLDHashEntryHdr *entry);

    static PLDHashNumber  HashKey( PLDHashTable *table, const void *key);

    static PRBool         MatchEntry( PLDHashTable *           table,
                                      const PLDHashEntryHdr *  entry,
                                      const void *             key);

    static void           MoveEntry( PLDHashTable *table,
                                     const PLDHashEntryHdr *from,
                                     PLDHashEntryHdr       *to);

    static void           ClearEntry( PLDHashTable *table, PLDHashEntryHdr *entry);

    static void           Finalize( PLDHashTable *table);

    static
    PLDHashOperator       FreeCacheEntries(PLDHashTable *    table,
                                           PLDHashEntryHdr * hdr,
                                           PRUint32          number,
                                           void *            arg);

    // member variables
    static PLDHashTableOps ops;
    PLDHashTable           table;
    PRBool                 initialized;
};

#endif // _nsCacheEntry_h_