disk.h

Go to the documentation of this file.

 
 
#ifndef PSTSDK_DISK_DISK_H
#define PSTSDK_DISK_DISK_H
 
#include <cstddef>
 
#include "pstsdk/util/primitives.h"
 
namespace pstsdk
{
namespace disk
{
 
template<typename T>
struct block_reference
{
    typedef T block_id_disk;
    typedef T location;
 
    block_id_disk bid; 
    location ib;       
} PSTSDK_MS_STRUCT;
static_assert(sizeof(block_reference<ulong>) == 8, "block_reference<ulong> incorrect size");
static_assert(sizeof(block_reference<ulonglong>) == 16, "block_reference<ulonglong> incorrect size");
 
//
// header
//
 
 
const size_t header_fmap_entries = 128;
 
const size_t header_fpmap_size = 128;
 
const size_t header_lock_entries = 32;
 
enum database_format
{
    database_format_ansi_min = 14,      
    database_format_ansi = 15,          
    database_format_unicode_min = 20,   
    database_format_unicode = 23,       
};
 
enum database_type
{
    database_ost = 12, 
    database_pst = 19, 
};
 
const ushort pst_magic = 0x4D53;
const ushort ost_magic = 0x4F53;
 
enum crypt_method
{
    crypt_method_none = 0,    
    crypt_method_permute = 1, 
    crypt_method_cyclic = 2,  
};
 
template<typename T>
struct root
{
    typedef T location;
    typedef T count;
 
    ulong cOrphans;             
    location ibFileEof;         
    location ibAMapLast;        
    count cbAMapFree;           
    count cbPMapFree;           
    block_reference<T> brefNBT; 
    block_reference<T> brefBBT; 
    byte fAMapValid;            
    byte bARVec;                
    ushort cARVec;              
} PSTSDK_MS_STRUCT;
static_assert(sizeof(root<ulong>) == 40, "root<ulong> incorrect size");
static_assert(sizeof(root<ulonglong>) == 80, "root<ulonglong> incorrect size");
static_assert(offsetof(root<ulonglong>, cOrphans) == 0, "root<ulonglong> cOrphans at incorrect offset");
static_assert(offsetof(root<ulonglong>, ibFileEof) == 8, "root<ulonglong> ibFileEof at incorrect offset");
static_assert(offsetof(root<ulonglong>, ibAMapLast) == 16, "root<ulonglong> ibAMapLast at incorrect offset");
static_assert(offsetof(root<ulonglong>, cbAMapFree) == 24, "root<ulonglong> cbAMapFree at incorrect offset");
static_assert(offsetof(root<ulonglong>, cbPMapFree) == 32, "root<ulonglong> cbPMapFree at incorrect offset");
static_assert(offsetof(root<ulonglong>, brefNBT) == 40, "root<ulonglong> brefNBT at incorrect offset");
static_assert(offsetof(root<ulonglong>, brefBBT) == 56, "root<ulonglong> brefBBT at incorrect offset");
static_assert(offsetof(root<ulonglong>, fAMapValid) == 72, "root<ulonglong> fAMapValid at incorrect offset");
static_assert(offsetof(root<ulonglong>, bARVec) == 73, "root<ulonglong> bARVec at incorrect offset");
static_assert(offsetof(root<ulonglong>, cARVec) == 74, "root<ulonglong> cARVec at incorrect offset");
 
const uint hlmagic = 0x4e444221;
 
template<typename T>
struct header
{
} PSTSDK_MS_STRUCT;
 
template<>
struct header<ulonglong>
{
    typedef ulonglong block_id_disk;   
    typedef ulonglong location;        
    typedef ulonglong count;           
 
    ulong dwMagic;                     
    ulong dwCRCPartial;
    ushort wMagicClient;               
    ushort wVer;                       
    ushort wVerClient;                 
    byte bPlatformCreate;              
    byte bPlatformAccess;              
    ulong dwOpenDBID;                  
    ulong dwOpenClaimID;               
    block_id_disk bidUnused;           
    block_id_disk bidNextP;            
    ulong dwUnique;
    node_id rgnid[nid_type_max];       
    root<ulonglong> root_info;         
    byte rgbFM[header_fmap_entries];   
    byte rgbFP[header_fpmap_size];     
    byte bSentinel;                    
    byte bCryptMethod;                 
    byte rgbReserved[2];               
#ifdef __GNUC__
    // GCC refuses to pack this next to rgbReserved
    byte bidNextB[8];
#else
#pragma pack(4)
    block_id_disk bidNextB;            
#pragma pack()
#endif
    ulong dwCRCFull;
    byte rgbVersionEncoded[3];
    byte bLockSemaphore;               
    byte rgbLock[header_lock_entries]; 
} PSTSDK_MS_STRUCT;
static_assert(sizeof(header<ulonglong>) == 568, "header<ulonglong> incorrect size");
static_assert(offsetof(header<ulonglong>, rgnid) == 44, "header<ulonglong> rgnid at incorrect offset");
static_assert(offsetof(header<ulonglong>, root_info) == 176, "header<ulonglong> root_info at incorrect offset");
static_assert(offsetof(header<ulonglong>, rgbFM) == 256, "header<ulonglong> rgbFM at incorrect offset");
static_assert(offsetof(header<ulonglong>, dwCRCFull) == 524, "header<ulonglong> dwCRCFull at incorrect offset");
 
template<>
struct header<ulong>
{
    typedef ulong block_id_disk;
    typedef ulong location;
    typedef ulong count;
 
    ulong dwMagic;
    ulong dwCRCPartial;
    ushort wMagicClient;
    ushort wVer;
    ushort wVerClient;
    byte bPlatformCreate;
    byte bPlatformAccess;
    ulong dwOpenDBID;
    ulong dwOpenClaimID;
#ifdef __GNUC__
    byte bidNextB[4];
#else
    block_id_disk bidNextB;
#endif
    block_id_disk bidNextP;
    ulong dwUnique;
    node_id rgnid[nid_type_max];
    root<ulong> root_info;
    byte rgbFM[header_fmap_entries];
    byte rgbFP[header_fpmap_size];
    byte bSentinel;
    byte bCryptMethod;
    byte rgbReserved[2];
    ulonglong ullReserved;
    ulong dwReserved;
    byte rgbVersionEncoded[3];
    byte bLockSemaphore;
    byte rgbLock[header_lock_entries];
} PSTSDK_MS_STRUCT;
static_assert(sizeof(header<ulong>) == 512, "header<ulong> incorrect size");
static_assert(offsetof(header<ulong>, rgnid) == 36, "header<ulong> rgnid at incorrect offset");
static_assert(offsetof(header<ulong>, rgbFM) == 204, "header<ulong> rgbFM at incorrect offset");
 
template<typename T>
struct header_crc_locations
{
} PSTSDK_MS_STRUCT;
 
template<>
struct header_crc_locations<ulong>
{
    static const size_t start = offsetof(header<ulong>, wMagicClient);
    static const size_t end = offsetof(header<ulong>, bLockSemaphore);
    static const size_t length = end - start;
} PSTSDK_MS_STRUCT;
 
template<>
struct header_crc_locations<ulonglong>
{
    static const size_t partial_start = header_crc_locations<ulong>::start;
    static const size_t partial_end = header_crc_locations<ulong>::end;
    static const size_t partial_length = header_crc_locations<ulong>::length;
    static const size_t full_start = offsetof(header<ulonglong>, wMagicClient);
    static const size_t full_end = offsetof(header<ulonglong>, dwCRCFull);
    static const size_t full_length = full_end - full_start;
} PSTSDK_MS_STRUCT;
 
//
// utility functions
//
 
 
const ulong crc_table[] = {
    0x00000000, 0x77073096, 0xEE0E612C, 0x990951BA, 0x076DC419, 0x706AF48F,
    0xE963A535, 0x9E6495A3, 0x0EDB8832, 0x79DCB8A4, 0xE0D5E91E, 0x97D2D988,
    0x09B64C2B, 0x7EB17CBD, 0xE7B82D07, 0x90BF1D91, 0x1DB71064, 0x6AB020F2,
    0xF3B97148, 0x84BE41DE, 0x1ADAD47D, 0x6DDDE4EB, 0xF4D4B551, 0x83D385C7,
    0x136C9856, 0x646BA8C0, 0xFD62F97A, 0x8A65C9EC, 0x14015C4F, 0x63066CD9,
    0xFA0F3D63, 0x8D080DF5, 0x3B6E20C8, 0x4C69105E, 0xD56041E4, 0xA2677172,
    0x3C03E4D1, 0x4B04D447, 0xD20D85FD, 0xA50AB56B, 0x35B5A8FA, 0x42B2986C,
    0xDBBBC9D6, 0xACBCF940, 0x32D86CE3, 0x45DF5C75, 0xDCD60DCF, 0xABD13D59,
    0x26D930AC, 0x51DE003A, 0xC8D75180, 0xBFD06116, 0x21B4F4B5, 0x56B3C423,
    0xCFBA9599, 0xB8BDA50F, 0x2802B89E, 0x5F058808, 0xC60CD9B2, 0xB10BE924,
    0x2F6F7C87, 0x58684C11, 0xC1611DAB, 0xB6662D3D, 0x76DC4190, 0x01DB7106,
    0x98D220BC, 0xEFD5102A, 0x71B18589, 0x06B6B51F, 0x9FBFE4A5, 0xE8B8D433,
    0x7807C9A2, 0x0F00F934, 0x9609A88E, 0xE10E9818, 0x7F6A0DBB, 0x086D3D2D,
    0x91646C97, 0xE6635C01, 0x6B6B51F4, 0x1C6C6162, 0x856530D8, 0xF262004E,
    0x6C0695ED, 0x1B01A57B, 0x8208F4C1, 0xF50FC457, 0x65B0D9C6, 0x12B7E950,
    0x8BBEB8EA, 0xFCB9887C, 0x62DD1DDF, 0x15DA2D49, 0x8CD37CF3, 0xFBD44C65,
    0x4DB26158, 0x3AB551CE, 0xA3BC0074, 0xD4BB30E2, 0x4ADFA541, 0x3DD895D7,
    0xA4D1C46D, 0xD3D6F4FB, 0x4369E96A, 0x346ED9FC, 0xAD678846, 0xDA60B8D0,
    0x44042D73, 0x33031DE5, 0xAA0A4C5F, 0xDD0D7CC9, 0x5005713C, 0x270241AA,
    0xBE0B1010, 0xC90C2086, 0x5768B525, 0x206F85B3, 0xB966D409, 0xCE61E49F,
    0x5EDEF90E, 0x29D9C998, 0xB0D09822, 0xC7D7A8B4, 0x59B33D17, 0x2EB40D81,
    0xB7BD5C3B, 0xC0BA6CAD, 0xEDB88320, 0x9ABFB3B6, 0x03B6E20C, 0x74B1D29A,
    0xEAD54739, 0x9DD277AF, 0x04DB2615, 0x73DC1683, 0xE3630B12, 0x94643B84,
    0x0D6D6A3E, 0x7A6A5AA8, 0xE40ECF0B, 0x9309FF9D, 0x0A00AE27, 0x7D079EB1,
    0xF00F9344, 0x8708A3D2, 0x1E01F268, 0x6906C2FE, 0xF762575D, 0x806567CB,
    0x196C3671, 0x6E6B06E7, 0xFED41B76, 0x89D32BE0, 0x10DA7A5A, 0x67DD4ACC,
    0xF9B9DF6F, 0x8EBEEFF9, 0x17B7BE43, 0x60B08ED5, 0xD6D6A3E8, 0xA1D1937E,
    0x38D8C2C4, 0x4FDFF252, 0xD1BB67F1, 0xA6BC5767, 0x3FB506DD, 0x48B2364B,
    0xD80D2BDA, 0xAF0A1B4C, 0x36034AF6, 0x41047A60, 0xDF60EFC3, 0xA867DF55,
    0x316E8EEF, 0x4669BE79, 0xCB61B38C, 0xBC66831A, 0x256FD2A0, 0x5268E236,
    0xCC0C7795, 0xBB0B4703, 0x220216B9, 0x5505262F, 0xC5BA3BBE, 0xB2BD0B28,
    0x2BB45A92, 0x5CB36A04, 0xC2D7FFA7, 0xB5D0CF31, 0x2CD99E8B, 0x5BDEAE1D,
    0x9B64C2B0, 0xEC63F226, 0x756AA39C, 0x026D930A, 0x9C0906A9, 0xEB0E363F,
    0x72076785, 0x05005713, 0x95BF4A82, 0xE2B87A14, 0x7BB12BAE, 0x0CB61B38,
    0x92D28E9B, 0xE5D5BE0D, 0x7CDCEFB7, 0x0BDBDF21, 0x86D3D2D4, 0xF1D4E242,
    0x68DDB3F8, 0x1FDA836E, 0x81BE16CD, 0xF6B9265B, 0x6FB077E1, 0x18B74777,
    0x88085AE6, 0xFF0F6A70, 0x66063BCA, 0x11010B5C, 0x8F659EFF, 0xF862AE69,
    0x616BFFD3, 0x166CCF45, 0xA00AE278, 0xD70DD2EE, 0x4E048354, 0x3903B3C2,
    0xA7672661, 0xD06016F7, 0x4969474D, 0x3E6E77DB, 0xAED16A4A, 0xD9D65ADC,
    0x40DF0B66, 0x37D83BF0, 0xA9BCAE53, 0xDEBB9EC5, 0x47B2CF7F, 0x30B5FFE9,
    0xBDBDF21C, 0xCABAC28A, 0x53B39330, 0x24B4A3A6, 0xBAD03605, 0xCDD70693,
    0x54DE5729, 0x23D967BF, 0xB3667A2E, 0xC4614AB8, 0x5D681B02, 0x2A6F2B94,
    0xB40BBE37, 0xC30C8EA1, 0x5A05DF1B, 0x2D02EF8D
};
 
const byte table1[] =
{
      65,  54,  19,  98, 168,  33, 110, 187,
     244,  22, 204,   4, 127, 100, 232,  93,
      30, 242, 203,  42, 116, 197,  94,  53,
     210, 149,  71, 158, 150,  45, 154, 136,
      76, 125, 132,  63, 219, 172,  49, 182,
      72,  95, 246, 196, 216,  57, 139, 231,
      35,  59,  56, 142, 200, 193, 223,  37,
     177,  32, 165,  70,  96,  78, 156, 251,
     170, 211,  86,  81,  69, 124,  85,   0,
       7, 201,  43, 157, 133, 155,   9, 160,
     143, 173, 179,  15,  99, 171, 137,  75,
     215, 167,  21,  90, 113, 102,  66, 191,
      38,  74, 107, 152, 250, 234, 119,  83,
     178, 112,   5,  44, 253,  89,  58, 134,
     126, 206,   6, 235, 130, 120,  87, 199,
     141,  67, 175, 180,  28, 212,  91, 205,
     226, 233,  39,  79, 195,   8, 114, 128,
     207, 176, 239, 245,  40, 109, 190,  48,
      77,  52, 146, 213,  14,  60,  34,  50,
     229, 228, 249, 159, 194, 209,  10, 129,
      18, 225, 238, 145, 131, 118, 227, 151,
     230,  97, 138,  23, 121, 164, 183, 220,
     144, 122,  92, 140,   2, 166, 202, 105,
     222,  80,  26,  17, 147, 185,  82, 135,
      88, 252, 237,  29,  55,  73,  27, 106,
     224,  41,  51, 153, 189, 108, 217, 148,
     243,  64,  84, 111, 240, 198, 115, 184,
     214,  62, 101,  24,  68,  31, 221, 103,
      16, 241,  12,  25, 236, 174,   3, 161,
      20, 123, 169,  11, 255, 248, 163, 192,
     162,   1, 247,  46, 188,  36, 104, 117,
      13, 254, 186,  47, 181, 208, 218,  61,
};
 
const byte table2[] =
{
      20,  83,  15,  86, 179, 200, 122, 156,
     235, 101,  72,  23,  22,  21, 159,   2,
     204,  84, 124, 131,   0,  13,  12,  11,
     162,  98, 168, 118, 219, 217, 237, 199,
     197, 164, 220, 172, 133, 116, 214, 208,
     167, 155, 174, 154, 150, 113, 102, 195,
      99, 153, 184, 221, 115, 146, 142, 132,
     125, 165,  94, 209,  93, 147, 177,  87,
      81,  80, 128, 137,  82, 148,  79,  78,
      10, 107, 188, 141, 127, 110,  71,  70,
      65,  64,  68,   1,  17, 203,   3,  63,
     247, 244, 225, 169, 143,  60,  58, 249,
     251, 240,  25,  48, 130,   9,  46, 201,
     157, 160, 134,  73, 238, 111,  77, 109,
     196,  45, 129,  52,  37, 135,  27, 136,
     170, 252,   6, 161,  18,  56, 253,  76,
      66, 114, 100,  19,  55,  36, 106, 117,
     119,  67, 255, 230, 180,  75,  54,  92,
     228, 216,  53,  61,  69, 185,  44, 236,
     183,  49,  43,  41,   7, 104, 163,  14,
     105, 123,  24, 158,  33,  57, 190,  40,
      26,  91, 120, 245,  35, 202,  42, 176,
     175,  62, 254,   4, 140, 231, 229, 152,
      50, 149, 211, 246,  74, 232, 166, 234,
     233, 243, 213,  47, 112,  32, 242,  31,
       5, 103, 173,  85,  16, 206, 205, 227,
      39,  59, 218, 186, 215, 194,  38, 212,
     145,  29, 210,  28,  34,  51, 248, 250,
     241,  90, 239, 207, 144, 182, 139, 181,
     189, 192, 191,   8, 151,  30, 108, 226,
      97, 224, 198, 193,  89, 171, 187,  88,
     222,  95, 223,  96, 121, 126, 178, 138,
};
 
const byte table3[] =
{
      71, 241, 180, 230,  11, 106, 114,  72,
     133,  78, 158, 235, 226, 248, 148,  83,
     224, 187, 160,   2, 232,  90,   9, 171,
     219, 227, 186, 198, 124, 195,  16, 221,
      57,   5, 150,  48, 245,  55,  96, 130,
     140, 201,  19,  74, 107,  29, 243, 251,
     143,  38, 151, 202, 145,  23,   1, 196,
      50,  45, 110,  49, 149, 255, 217,  35,
     209,   0,  94, 121, 220,  68,  59,  26,
      40, 197,  97,  87,  32, 144,  61, 131,
     185,  67, 190, 103, 210,  70,  66, 118,
     192, 109,  91, 126, 178,  15,  22,  41,
      60, 169,   3,  84,  13, 218,  93, 223,
     246, 183, 199,  98, 205, 141,   6, 211,
     105,  92, 134, 214,  20, 247, 165, 102,
     117, 172, 177, 233,  69,  33, 112,  12,
     135, 159, 116, 164,  34,  76, 111, 191,
      31,  86, 170,  46, 179, 120,  51,  80,
     176, 163, 146, 188, 207,  25,  28, 167,
      99, 203,  30,  77,  62,  75,  27, 155,
      79, 231, 240, 238, 173,  58, 181,  89,
       4, 234,  64,  85,  37,  81, 229, 122,
     137,  56, 104,  82, 123, 252,  39, 174,
     215, 189, 250,   7, 244, 204, 142,  95,
     239,  53, 156, 132,  43,  21, 213, 119,
      52,  73, 182,  18,  10, 127, 113, 136,
     253, 157,  24,  65, 125, 147, 216,  88,
      44, 206, 254,  36, 175, 222, 184,  54,
     200, 161, 128, 166, 153, 152, 168,  47,
      14, 129, 101, 115, 228, 194, 162, 138,
     212, 225,  17, 208,   8, 139,  42, 242,
     237, 154, 100,  63, 193, 108, 249, 236
};
 
template<typename T>
ushort compute_signature(T id, T address);
 
template<typename T>
ushort compute_signature(const block_reference<T>& reference) { return compute_signature(reference.bid, reference.ib); }
 
ulong compute_crc(const void * pdata, ulong cb);
 
void permute(void * pdata, ulong cb, bool encrypt);
 
void cyclic(void * pdata, ulong cb, ulong key);
 
 
//
// page structures
//
 
 
 
const size_t page_size = 512;
 
enum page_type
{
    page_type_bbt = 0x80,   
    page_type_nbt = 0x81,   
    page_type_fmap = 0x82,  
    page_type_pmap = 0x83,  
    page_type_amap = 0x84,  
    page_type_fpmap = 0x85, 
    page_type_dlist = 0x86  
};
 
template<typename T>
struct page_trailer
{
} PSTSDK_MS_STRUCT;
 
template<>
struct page_trailer<ulonglong>
{
    typedef ulonglong block_id_disk;
 
    byte page_type;        
    byte page_type_repeat; 
    ushort signature;      
    ulong crc;             
    block_id_disk bid;     
} PSTSDK_MS_STRUCT;
static_assert(sizeof(page_trailer<ulonglong>) == 16, "page_trailer<ulonglong> incorrect size");
 
template<>
struct page_trailer<ulong>
{
    typedef ulong block_id_disk;
 
    byte page_type;
    byte page_type_repeat;
    ushort signature;
    block_id_disk bid;
    ulong crc;
} PSTSDK_MS_STRUCT;
static_assert(sizeof(page_trailer<ulong>) == 12, "page_trailer<ulong> incorrect size");
 
template<typename T>
struct page
{
    static const size_t page_data_size = page_size - sizeof(page_trailer<T>); 
 
    byte data[page_data_size]; 
    page_trailer<T> trailer;   
} PSTSDK_MS_STRUCT;
static_assert(sizeof(page<ulong>) == page_size, "page<ulong> incorrect size");
static_assert(sizeof(page<ulonglong>) == page_size, "page<ulonglong> incorrect size");
 
const size_t bytes_per_slot = 64;
 
const size_t first_amap_page_location = 0x4400;
 
template<typename T>
struct amap_page : public page<T>
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(amap_page<ulong>) == page_size, "amap_page<ulong> incorrect size");
static_assert(sizeof(amap_page<ulonglong>) == page_size, "amap_page<ulonglong> incorrect size");
 
template<typename T>
struct pmap_page : public page<T>
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(pmap_page<ulong>) == page_size, "pmap_page<ulong> incorrect size");
static_assert(sizeof(pmap_page<ulonglong>) == page_size, "pmap_page<ulonglong> incorrect size");
 
template<typename T>
struct fmap_page : public page<T>
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(fmap_page<ulong>) == page_size, "fmap_page<ulong> incorrect size");
static_assert(sizeof(fmap_page<ulonglong>) == page_size, "fmap_page<ulonglong> incorrect size");
 
template<typename T>
struct fpmap_page : public page<T>
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(fpmap_page<ulong>) == page_size, "fpmap_page<ulong> incorrect size");
static_assert(sizeof(fpmap_page<ulonglong>) == page_size, "fpmap_page<ulonglong> incorrect size");
 
const size_t dlist_page_location = 0x4200;
 
template<typename T>
struct dlist_page
{
    static const size_t extra_space = page<T>::page_data_size - 8;
    static const size_t max_entries = extra_space / sizeof(ulong); 
 
    byte flags;                       
    byte num_entries;                 
    union
    {
        ulong current_page;           
        ulong backfill_location;      
    };
    union
    {
        ulong entries[max_entries];   
        byte _ignore[extra_space];
    };
    page_trailer<T> trailer;          
} PSTSDK_MS_STRUCT;
static_assert(sizeof(dlist_page<ulong>) == page_size, "dlist_page<ulong> incorrect size");
static_assert(sizeof(dlist_page<ulonglong>) == page_size, "dlist_page<ulonglong> incorrect size");
 
const ulong dlist_page_num_mask = 0x0000FFFF;
 
const ulong dlist_slots_shift = 20;
 
inline ulong dlist_get_page_num(ulong entry) { return entry & dlist_page_num_mask; } 
 
inline ulong dlist_get_slots(ulong entry) { return entry >> dlist_slots_shift; }
 
template<typename T>
struct bt_entry
{
    typedef T bt_key;
 
    bt_key key;               
    block_reference<T> ref;   
} PSTSDK_MS_STRUCT;
static_assert(sizeof(bt_entry<ulong>) == 12, "bt_entry<ulong> incorrect size");
static_assert(sizeof(bt_entry<ulonglong>) == 24, "bt_entry<ulonglong> incorrect size");
 
template<typename T>
struct nbt_leaf_entry
{
    typedef T nid_index;
    typedef T block_id_disk;
 
    nid_index nid;       
    block_id_disk data;  
    block_id_disk sub;   
    node_id parent_nid;  
} PSTSDK_MS_STRUCT;
static_assert(sizeof(nbt_leaf_entry<ulong>) == 16, "nbt_leaf_entry<ulong> incorrect size");
static_assert(sizeof(nbt_leaf_entry<ulonglong>) == 32, "nbt_leaf_entry<ulonglong> incorrect size");
 
template<typename T>
struct bbt_leaf_entry
{
    block_reference<T> ref; 
    ushort size;            
    ushort ref_count;       
} PSTSDK_MS_STRUCT;
static_assert(sizeof(bbt_leaf_entry<ulong>) == 12, "bbt_leaf_entry<ulong> incorrect size");
static_assert(sizeof(bbt_leaf_entry<ulonglong>) == 24, "bbt_leaf_entry<ulonglong> incorrect size");
 
template<typename T, typename EntryType>
struct bt_page
{
    static const size_t extra_space = page<T>::page_data_size - (sizeof(T) * sizeof(byte));
    static const size_t max_entries = extra_space / sizeof(EntryType); 
    union
    {
        EntryType entries[max_entries];
        byte _ignore[extra_space];
    };
 
    byte num_entries;          
    byte num_entries_max;      
    byte entry_size;           
    byte level;                
 
    page_trailer<T> trailer;   
} PSTSDK_MS_STRUCT;
namespace detail {
    // We can't pass types containing "," to offsetof, because it's a
    // preprocessor macro, and it doesn't really understand how to parse an
    // argument list.  So we declare some aliases in a "detail" namespace
    // where they won't pollute the header.
    typedef bt_page<ulong, bt_entry<ulong> > bt_page_ulong;
    typedef bt_page<ulonglong, bt_entry<ulonglong> > bt_page_ulonglong;
}
static_assert(offsetof(detail::bt_page_ulong, num_entries) == 496, "bt_page<ulong, bt_entry<ulong> > num_entries at incorrect offset");
static_assert(offsetof(detail::bt_page_ulonglong, num_entries) == 488, "bt_page<ulonglong, bt_entry<ulonglong> > num_entries at incorrect offset");
static_assert(offsetof(detail::bt_page_ulong, trailer) == 500, "bt_page<ulong, bt_entry<ulong> > trailer at incorrect offset");
static_assert(offsetof(detail::bt_page_ulonglong, trailer) == 496, "bt_page<ulonglong, bt_entry<ulonglong> > trailer at incorrect offset");
 
template<typename T>
struct nbt_nonleaf_page : public bt_page<T, bt_entry<T> >
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(nbt_nonleaf_page<ulong>) == page_size, "nbt_nonleaf_page<ulong> incorrect size");
static_assert(sizeof(nbt_nonleaf_page<ulonglong>) == page_size, "nbt_nonleaf_page<ulonglong> incorrect size");
 
template<typename T>
struct bbt_nonleaf_page : public bt_page<T, bt_entry<T> >
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(bbt_nonleaf_page<ulong>) == page_size, "bbt_nonleaf_page<ulong> incorrect size");
static_assert(sizeof(bbt_nonleaf_page<ulonglong>) == page_size, "bbt_nonleaf_page<ulonglong> incorrect size");
 
template<typename T>
struct nbt_leaf_page : public bt_page<T, nbt_leaf_entry<T> >
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(nbt_leaf_page<ulong>) == page_size, "nbt_leaf_page<ulong> incorrect size");
static_assert(sizeof(nbt_leaf_page<ulonglong>) == page_size, "nbt_leaf_page<ulonglong> incorrect size");
 
template<typename T>
struct bbt_leaf_page : public bt_page<T, bbt_leaf_entry<T> >
{
} PSTSDK_MS_STRUCT;
static_assert(sizeof(bbt_leaf_page<ulong>) == page_size, "bbt_leaf_page<ulong> incorrect size");
static_assert(sizeof(bbt_leaf_page<ulonglong>) == page_size, "bbt_leaf_page<ulonglong> incorrect size");
 
//
// block structures
//
 
 
const size_t max_block_disk_size = 8 * 1024;
 
enum block_types
{
    block_type_external = 0x00, 
    block_type_extended = 0x01, 
    block_type_sub = 0x02       
};
 
template<typename T>
size_t align_disk(size_t size);
 
size_t align_slot(size_t size);
 
const uint block_id_attached_bit = 0x1;
 
const uint block_id_internal_bit = 0x2;
 
const uint block_id_increment = 0x4;
 
template<typename T>
bool bid_is_external(T bid) { return ((bid & block_id_internal_bit) == 0); }
 
template<typename T>
bool bid_is_internal(T bid) { return !bid_is_external(bid); }
 
template<typename T>
struct block_trailer
{
} PSTSDK_MS_STRUCT;
 
template<>
struct block_trailer<ulonglong>
{
    typedef ulonglong block_id_disk;
 
    ushort cb;           
    ushort signature;    
    ulong crc;           
    block_id_disk bid;   
} PSTSDK_MS_STRUCT;
static_assert(sizeof(block_trailer<ulonglong>) == 16, "block_trailer<ulonglong> incorrect size");
 
template<>
struct block_trailer<ulong>
{
    typedef ulong block_id_disk;
 
    ushort cb;
    ushort signature;
    block_id_disk bid;
    ulong crc;
} PSTSDK_MS_STRUCT;
static_assert(sizeof(block_trailer<ulong>) == 12, "block_trailer<ulong> incorrect size");
 
template<typename T>
struct external_block
{
    static const size_t max_size = max_block_disk_size - sizeof(block_trailer<T>);
    byte data[1];   
} PSTSDK_MS_STRUCT;
 
template<typename T>
struct extended_block
{
} PSTSDK_MS_STRUCT;
 
template<>
struct extended_block<ulonglong>
{
    typedef ulonglong block_id_disk;
 
    static const size_t max_count = (external_block<ulonglong>::max_size - 8) / sizeof(extended_block<ulonglong>::block_id_disk);
    static const size_t max_size = external_block<ulonglong>::max_size * extended_block<ulonglong>::max_count;
 
    byte block_type;       
    byte level;            
    ushort count;          
    ulong total_size;      
    block_id_disk bid[1];  
} PSTSDK_MS_STRUCT;
 
template<>
struct extended_block<ulong>
{
    typedef ulong block_id_disk;
 
    static const size_t max_count = ((4096L) - sizeof(block_trailer<ulong>) - 8) / sizeof(extended_block<ulong>::block_id_disk);
    static const size_t max_size = external_block<ulong>::max_size * extended_block<ulong>::max_count;
 
    byte block_type;
    byte level;
    ushort count;
    ulong total_size;
    block_id_disk bid[1];
} PSTSDK_MS_STRUCT;
 
template<typename T>
struct sub_leaf_entry
{
    typedef T block_id_disk;
 
    node_id nid;        
    block_id_disk data; 
    block_id_disk sub;  
} PSTSDK_MS_STRUCT;
static_assert(sizeof(sub_leaf_entry<ulong>) == 12, "sub_leaf_entry<ulong> incorrect size");
static_assert(sizeof(sub_leaf_entry<ulonglong>) == 24, "sub_leaf_entry<ulonglong> incorrect size");
 
template<typename T>
struct sub_nonleaf_entry
{
    typedef T block_id_disk;
 
    node_id nid_key;             
    block_id_disk sub_block_bid; 
} PSTSDK_MS_STRUCT;
static_assert(sizeof(sub_nonleaf_entry<ulong>) == 8, "sub_nonleaf_entry<ulong> incorrect size");
static_assert(sizeof(sub_nonleaf_entry<ulonglong>) == 16, "sub_nonleaf_entry<ulonglong> incorrect size");
 
template<typename T, typename EntryType>
struct sub_block
{
    byte block_type;    
    byte level;         
    ushort count;       
    EntryType entry[1]; 
} PSTSDK_MS_STRUCT;
 
template<typename T>
struct sub_nonleaf_block : public sub_block<T, sub_nonleaf_entry<T> >
{
} PSTSDK_MS_STRUCT;
 
template<typename T>
struct sub_leaf_block : public sub_block<T, sub_leaf_entry<T> >
{
} PSTSDK_MS_STRUCT;
 
//
// heap structures
//
 
 
const byte heap_signature = 0xEC;
 
const uint heap_max_alloc_size = 3580;
 
const uint heap_max_alloc_size_wver_14 = 3068;
 
enum heap_client_signature
{
   heap_sig_gmp = 0x6C,  //< Internal
   heap_sig_tc = 0x7C,   //< Table context
   heap_sig_smp = 0x8C,  //< Internal
   heap_sig_hmp = 0x9C,  //< Internal
   heap_sig_ch = 0xA5,   //< \deprecated Internal
   heap_sig_chtc = 0xAC, //< \deprecated Internal
   heap_sig_bth = 0xB5,  //< BTree on Heap
   heap_sig_pc = 0xBC,   //< Property Context
};
 
enum heap_fill_level
{
    heap_fill_empty,    //< >= 3584 bytes free
    heap_fill_1,        //< 2560 - 3583 bytes free
    heap_fill_2,        //< 2048 - 2559 bytes free
    heap_fill_3,        //< 1792 - 2047 bytes free
    heap_fill_4,        //< 1536 - 1791 bytes free
    heap_fill_5,        //< 1280 - 1535 bytes free
    heap_fill_6,        //< 1024 - 1279 bytes free
    heap_fill_7,        //< 768 - 1023 bytes free
    heap_fill_8,        //< 512 - 767 bytes free
    heap_fill_9,        //< 256 - 511 bytes free
    heap_fill_10,       //< 128 - 255 bytes free
    heap_fill_11,       //< 64 - 127 bytes free
    heap_fill_12,       //< 32 - 63 bytes free
    heap_fill_13,       //< 16 - 31 bytes free
    heap_fill_14,       //< 8 - 15 bytes free
    heap_fill_full      //< < 8 bytes free
};
 
struct heap_first_header
{
    static const uint fill_level_size = 4; 
 
    ushort page_map_offset;  
    byte signature;          
    byte client_signature;   
    heap_id root_id;         
    byte page_fill_levels[fill_level_size]; 
} PSTSDK_MS_STRUCT;
 
struct heap_page_header
{
    ushort page_map_offset; 
} PSTSDK_MS_STRUCT;
 
struct heap_page_fill_header
{
    static const uint fill_level_size = 64; 
 
    ushort page_map_offset; 
    byte page_fill_levels[fill_level_size]; 
} PSTSDK_MS_STRUCT;
 
struct heap_page_map
{
    ushort num_allocs; 
    ushort num_frees;  
    ushort allocs[1];  
} PSTSDK_MS_STRUCT;
 
//
// bth structures
//
 
 
struct bth_header
{
    byte bth_signature; 
    byte key_size;      
    byte entry_size;    
    byte num_levels;    
    heap_id root;       
} PSTSDK_MS_STRUCT;
 
#pragma pack(2)
template<typename K>
struct bth_nonleaf_entry
{
    K key;        
    heap_id page; 
} PSTSDK_MS_STRUCT;
#pragma pack()
static_assert(sizeof(bth_nonleaf_entry<ushort>) == 6, "bth_nonleaf_entry<ushort> incorrect size");
 
template<typename K, typename V>
struct bth_leaf_entry
{
    K key;   
    V value; 
} PSTSDK_MS_STRUCT;
 
template<typename EntryType>
struct bth_node
{
    EntryType entries[1]; 
} PSTSDK_MS_STRUCT;
 
template<typename K, typename V>
struct bth_leaf_node : bth_node<bth_leaf_entry<K,V> >
{
} PSTSDK_MS_STRUCT;
 
template<typename K>
struct bth_nonleaf_node : bth_node<bth_nonleaf_entry<K> >
{
} PSTSDK_MS_STRUCT;
 
//
// pc structures
//
 
 
#pragma pack(2)
struct prop_entry
{
    ushort type;     //$< Property type
    heapnode_id id;  //$< Heapnode id for variable length properties, or the value directly for fixed size property types
};
#pragma pack()
static_assert(sizeof(bth_leaf_entry<ushort, prop_entry>) == 8, "bth_leaf_entry<ushort, prop_entry> incorrect size");
 
struct sub_object
{
    node_id nid; //$< The subnode id containing the data for the object
    ulong size;  //$< The size of the object
} PSTSDK_MS_STRUCT;
 
struct mv_toc
{
    ulong count;      //$< Number of entries in the TOC
    ulong offsets[1]; //$< Array of offsets for the start of each entry
} PSTSDK_MS_STRUCT;
 
//
// tc structures
//
 
 
enum tc_offsets
{
    tc_offsets_four,    //$< Offset of the end of the four and eight byte columns
    tc_offsets_two,     //$< Offset of the end of the two byte columns
    tc_offsets_one,     //$< Offset of the end of the one byte columns
    tc_offsets_bitmap,  //$< Offset of the end of the existance bitmap
    tc_offsets_max      //$< Number of entries in the offset array
};
 
#pragma pack(2)
struct column_description
{
    ushort type;        //$< Column property type
    prop_id id;         //$< Column property id
    ushort offset;      //$< Offset into the row
    byte size;          //$< Width of the column
    byte bit_offset;    //$< Bit offset into the existance bitmap
};
 
struct gust_column_description
{
    ushort type;
    prop_id id;
    ushort offset;
    byte size;
    byte unused1;
    ushort bit_offset;
    ushort unused2;
    node_id data_subnode;
};
 
struct tc_header
{
    byte signature;             //$< TC signature, \ref heap_sig_tc
    byte num_columns;           //$< Number of columns in this table
    ushort size_offsets[tc_offsets_max]; // Row offset array, see \ref tc_offsets
    heap_id row_btree_id;       //$< The bth_header allocation for the row mapping btree
    heapnode_id row_matrix_id;  //$< The heapnode_id allocation for the row matrix
    byte unused[4];
    column_description columns[1]; //$< Column description array, of length num_columns
};
 
struct gust_header
{
    byte signature;
    byte unused1;
    ushort size_offsets[tc_offsets_max];
    heap_id row_btree_id;
    heapnode_id row_matrix_id;
    byte unused2[4];
    ushort num_columns;
    node_id column_subnode;
    ulong unused3;
    ulong unused4;
};
#pragma pack()
 
//
// nameid structures
//
 
 
struct nameid
{
    union
    {
        ulong id;               
        ulong string_offset;    
    };
    ulong index;                
} PSTSDK_MS_STRUCT;
 
struct nameid_hash_entry
{
    ulong hash_base;            
    ulong index;                
} PSTSDK_MS_STRUCT;
static_assert(sizeof(nameid) == 8, "nameid incorrect size");
static_assert(sizeof(nameid_hash_entry) == 8, "nameid incorrect size");
 
inline ushort nameid_get_prop_index(const nameid& n) { return (ushort)(n.index >> 16); }
 
inline ushort nameid_get_guid_index(const nameid& n) { return (ushort)((ushort)n.index >> 1); }
 
inline bool nameid_is_string(const nameid& n) { return n.index & 0x1; }
 
inline ushort nameid_get_prop_index(const nameid_hash_entry& n) { return (ushort)(n.index >> 16); }
 
inline ushort nameid_get_guid_index(const nameid_hash_entry& n) { return (ushort)((ushort)n.index >> 1); }
 
inline bool nameid_is_string(const nameid_hash_entry& n) { return n.index & 0x1; }
 
} // end disk namespace
} // end pstsdk namespace
 
 
template<typename T>
inline pstsdk::ushort pstsdk::disk::compute_signature(T id, T address)
{
    T value = address ^ id;
 
    return (ushort(ushort(value >> 16) ^ ushort(value)));
}
 
inline pstsdk::ulong pstsdk::disk::compute_crc(const void * pdata, ulong cb)
{
    ulong crc = 0;
    const byte * pb = reinterpret_cast<const byte*>(pdata);
 
    while(cb-- > 0)
        crc = crc_table[(int)(byte)crc ^ *pb++] ^ (crc >> 8);
 
    return crc;
}
 
inline void pstsdk::disk::permute(void * pdata, ulong cb, bool encrypt)
{
    byte * pb = reinterpret_cast<byte*>(pdata);
    const byte * ptable = encrypt ? table1 : table3;
    byte b;
 
    while(cb-- > 0)
    {
        b = *pb;
        *pb++ = ptable[b];
    }
}
 
inline void pstsdk::disk::cyclic(void * pdata, ulong cb, ulong key)
{
    byte * pb = reinterpret_cast<byte*>(pdata);
    byte b;
    ushort w;
 
    w = (ushort)(key ^ (key >> 16));
 
    while (cb-- > 0)
    {
        b = *pb;
        b = (byte)(b + (byte)w);
        b = table1[b];
        b = (byte)(b + (byte)(w >> 8));
        b = table2[b];
        b = (byte)(b - (byte)(w >> 8));
        b = table3[b];
        b = (byte)(b - (byte)w);
        *pb++ = b;
 
        w = (ushort)(w + 1);
    }
}
 
template<typename T>
inline size_t pstsdk::disk::align_disk(size_t size)
{
    return align_slot(size + sizeof(block_trailer<T>));
}
 
inline size_t pstsdk::disk::align_slot(size_t size)
{
    return ((size + bytes_per_slot - 1) & ~(bytes_per_slot - 1));
}
 
#endif