Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

874 lines
17 KiB

//
// MODULE: CACHEGEN.CPP
//
// PURPOSE: Cache File Generator and Reader for BN Networks
//
// PROJECT: Generic Troubleshooter DLL for Microsoft AnswerPoint
//
// COMPANY: Saltmine Creative, Inc. (206)-284-7511 [email protected]
//
// AUTHOR: Roman Mach
//
// ORIGINAL DATE: 8/7/97
//
// NOTES:
// 1.
//
// Version Date By Comments
//--------------------------------------------------------------------
// V0.2 8/7/97 RM Local Version for Memphis
// V0.3 04/09/98 JM/OK+ Local Version for NT5
//
#include "stdafx.h"
#include <afxwin.h>
#include <afxtempl.h>
#include "apgts.h"
#include "bnts.h"
#include "cachegen.h"
#include "BackupInfo.h"
#include "apgtsinf.h"
#include "ChmRead.h"
int nodecomp(const void *elem1, const void *elem2);
//
//
GTSCacheGenerator::GTSCacheGenerator(BOOL bScanAll, const char *szLogFile, const char *szBNTSLogFile)
{
m_fp = NULL;
m_bScanAll = bScanAll;
m_szBNTSLogFile = szBNTSLogFile;
if (szLogFile != NULL)
m_fp = fopen(szLogFile, "w");
m_nCount = 0;
m_nItemCount = 0;
m_headposition = 0;
m_filedata = NULL;
m_netstartoffset = 0;
m_nodeorder = NULL;
}
//
//
GTSCacheGenerator::~GTSCacheGenerator()
{
if (m_fp)
fclose(m_fp);
if (m_filedata)
free(m_filedata);
if (m_nodeorder)
free(m_nodeorder);
}
bool GTSCacheGenerator::TcharToChar(char szOut[], LPCTSTR szIn, int &OutLen)
{
int x = 0;
while(NULL != szIn[x] && x < OutLen)
{
szOut[x] = (char) szIn[x];
x++;
}
if (x < OutLen)
szOut[x] = NULL;
return x < OutLen;
}
//
//
/*
void GTSCacheGenerator::LogOut(TCHAR *szcFormat, ...)
{
va_list ptr;
if (!m_fp)
return;
if (!szcFormat)
return;
va_start(ptr, szcFormat);
vfprintf(m_fp, szcFormat, ptr);
va_end(ptr);
}
*/
#ifdef _DEBUG
#define LogOut ::AfxTrace
//#define LogOut 1 ? (void)0 : ::AfxTrace
#else
#define LogOut 1 ? (void)0 : ::AfxTrace
#endif
//
//
int nodecomp(const void *elem1, const void *elem2)
{
return (((GTS_NODE_ORDER *)elem1)->depth - ((GTS_NODE_ORDER *)elem2)->depth);
}
//
//
void GTSCacheGenerator::SaveNetItem(CPtrList *nsp, BNTS *bp, FILE *fp, LPCSTR name)
{
GTS_NODE_ITEM *ni;
if (!bp->BNetPropItemStr( name, 0 ))
return;
ni = new GTS_NODE_ITEM(name);
int j = 0;
while (bp->BNetPropItemStr( name, j++ ))
{
CString sTemp = bp->SzcResult();
ni->sStringArr.Add(sTemp);
}
nsp->AddTail(ni);
}
//
//
BOOL GTSCacheGenerator::NodeTraverse( FILE *fp,
BNTS *bp,
int depth,
CArray<int,int> &newnodes,
CArray<int,int> &newstates,
int currnode,
int currstate)
{
BOOL bEnd = FALSE;
int i, j;
if (depth > 30)
{
LogOut(_T("Depth Exceeded\n"));
return FALSE;
}
// uninstantiate
UninstantiateAll(bp);
newnodes.Add(currnode);
newstates.Add(currstate);
depth++;
for (i=0;i<newnodes.GetSize();i++)
{
if (bp->BNodeSetCurrent(newnodes[i]))
{
bp->NodeSymName();
CString sTemp = bp->SzcResult();
LogOut(_T("%s"), sTemp);
ESTDLBL albl = bp->ELblNode();
if (albl == ESTDLBL_problem)
{
LogOut(_T("(prob)"));
}
else if (albl == ESTDLBL_info)
{
LogOut(_T("(info)"));
}
else if (albl == ESTDLBL_fixobs ||
albl == ESTDLBL_fixunobs ||
albl == ESTDLBL_unfix)
{
LogOut(_T("(fix)"));
}
else
{
LogOut(_T("(?)"));
}
}
LogOut(_T("(%d=%d) "), newnodes[i], newstates[i]);
}
SetNodes( bp, newnodes, newstates );
BOOL bRec = FALSE;
if (!bp->BGetRecommendations())
{
bRec = TRUE;
}
else
{
}
const int *rg = bp->RgInt();
i = 0;
if (bRec)
{
if (bp->BImpossible())
{
LogOut(_T("IMPOSSIBLE\n"));
}
else
{
LogOut(_T("RECOMMENDATION ERROR\n"));
}
}
else
{
if (bp->BImpossible())
{
LogOut(_T("IMPOSSIBLE (Have Rec)\n"));
}
else
{
int reccount = bp->CInt();
if (reccount)
{
int nodecount = (int)newstates.GetSize();
if (nodecount)
{
BOOL bFound = FALSE;
for (i=0;i<reccount;i++)
{
bFound = FALSE;
for (j=0;j<nodecount;j++)
{
if (newnodes[j] == rg[i])
{
bFound = TRUE;
break;
}
}
if (!bFound)
break;
}
if (!bFound)
{
LogOut(_T("RECOMMENDATION: "));
if (bp->BNodeSetCurrent(rg[i]))
{
CString sTemp1 = "Error";
CString sTemp2 = "Error";
if (bp->BNodePropItemStr( "HNodeHd", 0 ))
sTemp1 = bp->SzcResult();
bp->NodeSymName();
sTemp2 = bp->SzcResult();
LogOut(_T("%s (%s) (Node: %d)"), sTemp1, sTemp2, rg[i]);
if (m_nodeorder)
{
if (m_nodeorder[rg[i]].depth > depth)
m_nodeorder[rg[i]].depth = depth;
}
UINT realcount = 0;
for (j=0;j<nodecount;j++)
if (newstates[j] != STATE_UNKNOWN)
realcount++;
fwrite(&realcount, sizeof (UINT), 1, fp);
for (j=0;j<nodecount;j++)
if (newstates[j] != STATE_UNKNOWN)
{
GTS_CACHE_NODE cnode;
cnode.node = newnodes[j];
cnode.state = newstates[j];
fwrite(&cnode, sizeof (cnode), 1, fp);
}
// write to cache out file
UINT ucount = reccount;
fwrite(&ucount, sizeof (UINT), 1, fp);
for (j=0;j<reccount;j++)
{
UINT rgval = rg[j];
fwrite(&rgval, sizeof (UINT), 1, fp);
}
m_nCount++;
}
LogOut(_T("\n"));
}
else
{
LogOut(_T("NO RECOMMENDATIONS - ALL AVAILABLE MARKED AS UNKNOWN\n"));
bEnd = TRUE;
}
}
else
LogOut(_T("STATE COUNT 0 - INTERNAL ERROR\n"));
}
else
LogOut(_T("NO RECOMMENDATIONS\n"));
}
}
LogOut(_T("\n"));
// recommendations - let's figure out what to do with them
// the first rec in the returned array will be array[1] in the set array
// so we have to watch for that
if (bp->CInt() && !bEnd)
{
// have rec
CArray<int,int> states;
int node = rg[i];
if (bp->BNodeSetCurrent(node))
{
ESTDLBL albl = bp->ELblNode();
if (albl == ESTDLBL_info)
{
states.Add(0);
states.Add(1);
if (m_bScanAll)
states.Add(STATE_UNKNOWN);
}
else if (albl == ESTDLBL_fixobs ||
albl == ESTDLBL_fixunobs ||
albl == ESTDLBL_unfix)
{
states.Add(0);
if (m_bScanAll)
states.Add(STATE_UNKNOWN);
}
else
{
LogOut(_T("Unexpected Node Type\n"));
}
}
else
LogOut(_T("Can't set node current\n"));
int count = (int)states.GetSize();
for (i=0;i<count;i++)
{
if (!NodeTraverse( fp, bp, depth, newnodes, newstates, node, states[i] ))
return FALSE;
}
}
// done, remove references to our current node
newnodes.RemoveAt(depth - 1);
newstates.RemoveAt(depth - 1);
return TRUE;
}
//
//
void GTSCacheGenerator::UninstantiateAll(BNTS *bp)
{
int count = (int)m_oldnodes.GetSize();
// uninstantiate all nodes
if (!count)
return;
for (int j=0;j<count;j++)
{
if (bp->BNodeSetCurrent(m_oldnodes[j]))
{
if (bp->BNodeSet(-1))
{
}
else
{
LogOut(_T("Can't uninstantiate node\n"));
}
}
else
{
LogOut(_T("Can't set node %d to uninstantiate\n"), m_oldnodes[j]);
}
}
}
//
//
void GTSCacheGenerator::SetNodes(BNTS *bp, CArray<int,int> &nodes, CArray<int,int> &states)
{
m_oldnodes.Copy(nodes);
int count = (int)nodes.GetSize();
if (!count)
return;
LogOut(_T("\nSetNodes:"));
for (int j=0;j<count;j++)
{
if (bp->BNodeSetCurrent(nodes[j]))
{
if (states[j] != STATE_UNKNOWN)
{
LogOut(_T("(%d=%d)"), nodes[j], states[j]);
if (bp->BNodeSet(states[j], false))
{
}
else
{
LogOut(_T("Can't set node\n"));
}
}
else
LogOut(_T("(%d=X)"), nodes[j]);
}
else
LogOut(_T("Can't set node %d\n"), nodes[j]);
}
LogOut(_T("\n"));
}
//
//
BOOL GTSCacheGenerator::ReadCacheFileHeader(CString &sCacheFilename, const CString& strCacheFileWithinCHM)
{
GTS_CACHE_FILE_HEADER header;
bool bUseCHM = strCacheFileWithinCHM.GetLength() != 0;
if (bUseCHM)
{
DWORD size =0;
if (S_OK != ::ReadChmFile(sCacheFilename, strCacheFileWithinCHM, (void**)&m_filedata, &size))
{
return FALSE;
}
}
else
{
UINT size;
// must be binary
FILE *cfp = _tfopen(sCacheFilename, _T("rb"));
if (cfp==NULL)
{
LogOut(_T("Error opening cache file for reading\n"));
return FALSE;
}
// get file size
if (fseek(cfp, 0, SEEK_END))
{
LogOut(_T("Can't set pos to end of file\n"));
fclose(cfp);
return FALSE;
}
fpos_t position;
if (fgetpos(cfp, &position))
{
LogOut(_T("Can't get pos at end of file\n"));
fclose(cfp);
return FALSE;
}
size = (UINT) position;
rewind(cfp);
// allocate space for file
m_filedata = (char *) malloc(size);
if (m_filedata == NULL)
{
LogOut(_T("Error allocating memory\n"));
fclose(cfp);
return FALSE;
}
if (fread(m_filedata, size, 1, cfp) != 1)
{
LogOut(_T("Error reading file into memory\n"));
fclose(cfp);
return FALSE;
}
fclose(cfp);
}
memcpy(&header, m_filedata, sizeof(header));
if (memcmp(header.signature, GTS_CACHE_SIG, sizeof (header.signature)) != 0)
{
LogOut(_T("Bad file signature!\n"));
return FALSE;
}
if (!header.count)
{
LogOut(_T("No items in file!\n"));
return FALSE;
}
m_netstartoffset = header.netoffset;
LogOut(_T("ItemCount: %d\n"), header.count);
m_nItemCount = header.count;
m_cachepos = (GTS_CACHE_FILE_SETDATA *) (m_filedata + sizeof (GTS_CACHE_FILE_HEADER));
return TRUE;
}
//
//
BOOL GTSCacheGenerator::FindNetworkProperty(LPCSTR szName, CString &sResult, int index)
{
if (m_filedata == NULL)
return FALSE;
GTS_CACHE_PROP_NETSTART_BLK *netstart =
(GTS_CACHE_PROP_NETSTART_BLK *)(m_filedata + m_netstartoffset);
GTS_CACHE_PROP_STR_BLK *netstrblk =
(GTS_CACHE_PROP_STR_BLK *) (m_filedata + netstart->netpropoffset);
for (UINT i = 0; i < netstart->netpropcount; i++, netstrblk++)
{
LPCSTR szItem = (LPCSTR) (m_filedata + netstrblk->nameoffset);
if (!strcmp(szName, szItem))
{
if (!index)
{
sResult = (LPCSTR) (m_filedata + netstrblk->stringoffset);
return TRUE;
}
else
{
LPCSTR szStrArrItem = (LPCSTR) (m_filedata + netstrblk->stringoffset);
int len = strlen(szStrArrItem);
for (int j=0;(j < index) && len ;j++)
{
szStrArrItem += len + 1;
len = strlen(szStrArrItem);
}
if (!len)
return FALSE;
sResult = szStrArrItem;
return TRUE;
}
}
}
return FALSE;
}
//
//
BOOL GTSCacheGenerator::FindNodeProperty(UINT nodeid, LPCSTR szName, CString &sResult, int index)
{
if (m_filedata == NULL)
return FALSE;
GTS_CACHE_PROP_NETSTART_BLK *netstart =
(GTS_CACHE_PROP_NETSTART_BLK *)(m_filedata + m_netstartoffset);
GTS_CACHE_PROP_NODEOFF_BLK *nodeblk =
(GTS_CACHE_PROP_NODEOFF_BLK *) (m_filedata + m_netstartoffset + sizeof (GTS_CACHE_PROP_NETSTART_BLK));
for (UINT i = 0; i < netstart->nodecountfile; i++, nodeblk++)
{
if (nodeid == nodeblk->nodeid)
{
GTS_CACHE_PROP_NODESTART_BLK *nodestart =
(GTS_CACHE_PROP_NODESTART_BLK *)(m_filedata + nodeblk->nodeoffset);
GTS_CACHE_PROP_STR_BLK *nodestr =
(GTS_CACHE_PROP_STR_BLK *)(m_filedata + nodeblk->nodeoffset + sizeof (GTS_CACHE_PROP_NODESTART_BLK));
// now try to find string
for (UINT j=0;j<nodestart->nodestringcount;j++, nodestr++)
{
LPCSTR szItem = (LPCSTR) (m_filedata + nodestr->nameoffset);
if (!strcmp(szName, szItem))
{
if (!index)
{
sResult = (LPCSTR) (m_filedata + nodestr->stringoffset);
return TRUE;
}
else
{
LPCSTR szStrArrItem = (LPCSTR) (m_filedata + nodestr->stringoffset);
int len = strlen(szStrArrItem);
for (int j=0;(j < index) && len ;j++)
{
szStrArrItem += len + 1;
len = strlen(szStrArrItem);
}
if (!len)
return FALSE;
sResult = szStrArrItem;
return TRUE;
}
}
}
return FALSE;
}
}
return FALSE;
}
//
//
BOOL GTSCacheGenerator::IsNodePresent(UINT nodeid)
{
if (m_filedata == NULL)
return FALSE;
GTS_CACHE_PROP_NETSTART_BLK *netstart =
(GTS_CACHE_PROP_NETSTART_BLK *)(m_filedata + m_netstartoffset);
GTS_CACHE_PROP_NODEOFF_BLK *nodeblk =
(GTS_CACHE_PROP_NODEOFF_BLK *) (m_filedata + m_netstartoffset + sizeof (GTS_CACHE_PROP_NETSTART_BLK));
for (UINT i = 0; i < netstart->nodecountfile; i++, nodeblk++)
{
if (nodeid == nodeblk->nodeid)
return TRUE;
}
return FALSE;
}
// returns the node count for the network, not what's in the file
//
int GTSCacheGenerator::GetNodeCount()
{
if (m_filedata == NULL)
return 0;
GTS_CACHE_PROP_NETSTART_BLK *netstart =
(GTS_CACHE_PROP_NETSTART_BLK *)(m_filedata + m_netstartoffset);
return netstart->nodecountnetwork;
}
//
//
BOOL GTSCacheGenerator::GetNodeIDFromSymName(LPCTSTR szSymName, UINT &nodeid)
{
char sznSymName[MAX_SYM_NAME_BUF_LEN];
int nSymLen = MAX_SYM_NAME_BUF_LEN;
if (m_filedata == NULL)
return FALSE;
if (!TcharToChar(sznSymName, szSymName, nSymLen))
return FALSE;
GTS_CACHE_PROP_NETSTART_BLK *netstart =
(GTS_CACHE_PROP_NETSTART_BLK *)(m_filedata + m_netstartoffset);
GTS_CACHE_PROP_NODEOFF_BLK *nodeblk =
(GTS_CACHE_PROP_NODEOFF_BLK *) (m_filedata + m_netstartoffset + sizeof (GTS_CACHE_PROP_NETSTART_BLK));
for (UINT i = 0; i < netstart->nodecountfile; i++, nodeblk++)
{
GTS_CACHE_PROP_NODESTART_BLK *nodestart =
(GTS_CACHE_PROP_NODESTART_BLK *)(m_filedata + nodeblk->nodeoffset);
GTS_CACHE_PROP_STR_BLK *nodestr =
(GTS_CACHE_PROP_STR_BLK *)(m_filedata + nodeblk->nodeoffset + sizeof (GTS_CACHE_PROP_NODESTART_BLK));
// now try to find string
for (UINT j=0;j<nodestart->nodestringcount;j++, nodestr++)
{
LPCSTR szItem = (LPCSTR) (m_filedata + nodestr->nameoffset);
if (!strcmp(G_SYMBOLIC_NAME, szItem))
{
if (!strcmp(sznSymName, (LPCSTR) (m_filedata + nodestr->stringoffset)))
{
nodeid = nodeblk->nodeid;
return TRUE;
}
break;
}
}
}
return FALSE;
}
//
//
BOOL GTSCacheGenerator::GetLabelOfNode(UINT nodeid, UINT &lbl)
{
if (m_filedata == NULL)
return FALSE;
GTS_CACHE_PROP_NETSTART_BLK *netstart =
(GTS_CACHE_PROP_NETSTART_BLK *)(m_filedata + m_netstartoffset);
GTS_CACHE_PROP_NODEOFF_BLK *nodeblk =
(GTS_CACHE_PROP_NODEOFF_BLK *) (m_filedata + m_netstartoffset + sizeof (GTS_CACHE_PROP_NETSTART_BLK));
for (UINT i = 0; i < netstart->nodecountfile; i++, nodeblk++)
{
if (nodeid == nodeblk->nodeid)
{
GTS_CACHE_PROP_NODESTART_BLK *nodestart =
(GTS_CACHE_PROP_NODESTART_BLK *)(m_filedata + nodeblk->nodeoffset);
lbl = nodestart->labelnode;
return TRUE;
}
}
return FALSE;
}
//
//
BOOL GTSCacheGenerator::GetNextCacheEntryFromFile(BOOL &bErr, CBNCache *pCache)
{
BOOL bStat = FALSE;
bErr = TRUE;
if (m_filedata == NULL)
return FALSE;
if (!m_nItemCount)
{
bErr = FALSE;
return FALSE;
}
m_nItemCount--;
BN_CACHE_ITEM CacheItem;
// initialize
CacheItem.uNodeCount = 0;
CacheItem.uRecCount = 0;
CacheItem.uName = NULL;
CacheItem.uValue = NULL;
CacheItem.uRec = NULL;
bStat = GetNCEFF(&CacheItem, pCache);
// free allocated space as necessary
if (CacheItem.uName)
free(CacheItem.uName);
if (CacheItem.uValue)
free(CacheItem.uValue);
if (CacheItem.uRec)
free(CacheItem.uRec);
if (!bStat)
{
bErr = TRUE;
return FALSE;
}
bErr = FALSE;
return TRUE;
}
//
//
BOOL GTSCacheGenerator::GetNCEFF(BN_CACHE_ITEM *pCacheItem, CBNCache *pCache)
{
UINT j;
UINT setcount;
UINT reccount;
GTS_CACHE_FILE_SETDATA *setp = m_cachepos;
setcount = setp->count;
if (!setcount || setcount > 1000)
{
LogOut(_T("Set Count out of bounds: %d\n"), setcount);
return FALSE;
}
// initialize
pCacheItem->uNodeCount = setcount;
pCacheItem->uName = (UINT *)malloc(setcount * sizeof (UINT));
pCacheItem->uValue = (UINT *)malloc(setcount * sizeof (UINT));
LogOut(_T("Count: %d\n"), setcount);
GTS_CACHE_NODE *cachenode = &setp->item[0];
// second, read in node = state pairs
for (j=0;j<setcount;j++, cachenode++)
{
pCacheItem->uName[j] = cachenode->node;
pCacheItem->uValue[j] = cachenode->state;
LogOut(_T("(%d,%d)"), cachenode->node, cachenode->state);
}
LogOut(_T("\n"));
GTS_CACHE_FILE_RECDATA *recp = (GTS_CACHE_FILE_RECDATA *) cachenode;
reccount = recp->count;
if (!reccount || reccount > 1000)
{
LogOut(_T("Rec Count out of bounds: %d\n"), reccount);
return FALSE;
}
pCacheItem->uRecCount = reccount;
pCacheItem->uRec = (UINT *)malloc(reccount * sizeof (UINT));
UINT *uitem = &recp->item[0];
for (j=0;j<reccount;j++, uitem++)
{
pCacheItem->uRec[j] = *uitem;
LogOut(_T("(%d)"), *uitem);
}
LogOut(_T("\n"));
if (!pCache->AddCacheItem(pCacheItem))
{
LogOut(_T("Error Adding Item To Cache\n"));
return FALSE;
}
m_cachepos = (GTS_CACHE_FILE_SETDATA *) uitem;
// success!
return TRUE;
}