Source code of Windows XP (NT5)
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.
 
 
 
 
 
 

360 lines
11 KiB

//
// MODULE: APGTSCLS.H
//
// PURPOSE: Class header file
//
// PROJECT: Generic Troubleshooter DLL for Microsoft AnswerPoint
//
// COMPANY: Saltmine Creative, Inc. (206)-284-7511 [email protected]
//
// AUTHOR: Roman Mach, Joe Mabel
//
// ORIGINAL DATE: 8-2-96
//
// NOTES:
// 1. Based on Print Troubleshooter DLL
//
// Version Date By Comments
//--------------------------------------------------------------------
// V0.1 - RM Original
// V3.0 7-22-98 JM Major revision, deprecate IDH.
// V3.1 1-06-99 JM Extract APGTSEXT.H
//
#if !defined(APGTSCLS_H_INCLUDED)
#define APGTSCLS_H_INCLUDED
#if _MSC_VER >= 1000
#pragma once
#endif // _MSC_VER >= 1000
#include"apgtsinf.h"
#include "apgtslog.h"
#include "LogString.h"
#include "apgtspl.h"
#include "maxbuf.h"
#include "apgts.h"
#include <map>
using namespace std;
// string constants involved in commands from sysop to take various actions.
#define SZ_OP_ACTION "TSHOOOT" // Preface to all operator actions. Note the extra "O".
#define SZ_EMERGENCY_DEF SZ_OP_ACTION
#define SZ_RELOAD_TOPIC "E1" // Reload one topic
#define SZ_KILL_THREAD "E2" // Kill (and restart) one pool thread
#define SZ_RELOAD_ALL_TOPICS "E3" // Reload all monitored files.
#define SZ_SET_REG "E4" // Set a registry value.
#define SZ_KILL_STUCK_THREADS "E8" // Kill (and restart) all stuck pool threads
#define SZ_EMERGENCY_REBOOT "E9" // want to reboot this DLL.
// The product version is loaded from the resource file upon DLL startup.
// Used for APGTS logging and status page reporting.
extern CString gstrProductVersion;
// HTTP spec for document type. For validation of incoming HTTP POST request.
#define CONT_TYPE_STR "application/x-www-form-urlencoded"
class CHttpQuery {
public:
CHttpQuery();
~CHttpQuery();
BOOL GetFirst(LPCTSTR szInput, TCHAR *pchName, TCHAR *pchValue);
BOOL GetNext(TCHAR *pchName, TCHAR *pchValue);
void Push(LPCTSTR szPushed);
protected:
BOOL LoopFind(TCHAR *pchName, TCHAR *pchValue);
void AddBuffer( TCHAR ch, TCHAR *tostr);
void PutStr(LPCTSTR instr, TCHAR *addtostr);
static void CleanStr(TCHAR *str);
protected:
enum decstates {
ST_GETNEXT,
ST_FINDNAME,
ST_GETDATA,
ST_DECODEHEX1,
ST_DECODEHEX2,
ST_GETFIRST,
};
decstates m_state; // used to track where we are in putting together
// characters while deciphering HTTP encoding.
CString m_strInput; // The original input buffer, containing name/value pairs.
// It is also possible to "push" a pair onto the front of
// this buffer
int m_nIndex; // index into the string of m_strInput. Keeps track of
// where we are in the parse.
};
// forward declaration
class CDBLoadConfiguration;
class CTopic;
class CSniffConnector;
//
//
class APGTSContext
{
private:
//
// this nested class is an internal manager if nid-value pairs container
//
class CCommandsAddManager;
class CCommands
{
friend class CCommandsAddManager;
private:
//
// this nested class represent name/value pairs we get from an HTML form
//
class NID_VALUE_PAIR
{
friend class CCommands;
private:
NID nid; // Note two special values:
// nidProblem: value is a node
// nidNil: ignore value
int value; // typically a node state, but for nidProblem, it's
// problem node NID
public:
bool operator<(const NID_VALUE_PAIR & pair)const
{return nid<pair.nid || value<pair.value;};
bool operator==(const NID_VALUE_PAIR & pair)const
{return nid==pair.nid || value==pair.value;};
};
private:
vector<NID_VALUE_PAIR>m_arrPair;
private: // CAddManager is managing addition to object of this class
int Add( NID nid, int value );
public:
CCommands() {}
~CCommands() {}
int GetSize() const;
void RemoveAll();
bool GetAt( int nIndex, NID &nid, int &value ) const;
void RotateProblemPageToFront();
};
//
// this nested class is an internal manager of additions to
// "Commands: and "Sniffed" objects of CCommands class
//
class CCommandsAddManager
{
CCommands& m_Commands;
CCommands& m_Sniffed;
public:
CCommandsAddManager(CCommands& commands, CCommands& sniffed) : m_Commands(commands), m_Sniffed(sniffed) {}
~CCommandsAddManager() {}
public:
void Add(NID nid, int value, bool sniffed);
};
//
// this nested class is an internal manager if name-value pairs container
// carrying additional imformation from HTMP form
//
class CAdditionalInfo
{
private:
//
// this nested class represent name/value pairs we get from an HTML form
//
class NAME_VALUE_PAIR
{
friend class CAdditionalInfo;
private:
CString name;
CString value;
public:
bool operator<(const NAME_VALUE_PAIR & pair)const
{return name<pair.name;};
bool operator==(const NAME_VALUE_PAIR & pair)const
{return name==pair.name;};
};
private:
vector<NAME_VALUE_PAIR>m_arrPair;
public:
CAdditionalInfo() {}
~CAdditionalInfo() {}
int GetSize() const;
void RemoveAll();
bool GetAt( int nIndex, CString& name, CString& value ) const;
int Add( const CString& name, const CString& value );
};
protected:
enum eOpAction {eNoOpAction, eReloadTopic, eKillThread, eReloadAllTopics, eSetReg};
public:
APGTSContext( CAbstractECB *pECB,
CDBLoadConfiguration *pConf,
CHTMLLog *pLog,
GTS_STATISTIC *pStat,
CSniffConnector* pSniffConnector);
~APGTSContext();
void ProcessQuery();
static BOOL StrIsDigit(LPCTSTR pSz);
CString RetCurrentTopic() const;
protected:
void CheckAndLogCookie();
void DoContent();
DWORD ProcessCommands(LPTSTR pszCmd, LPTSTR pszValue);
VOID ClearCommandList();
VOID ClearSniffedList();
VOID ClearAdditionalInfoList();
//bool PlaceNodeInCommandList(NID nid, IST ist);
//bool PlaceNodeInSniffedList(NID nid, IST ist);
//bool PlaceInAdditionalInfoList(const CString& name, const CString& value);
VOID SetNodesPerCommandList();
VOID SetNodesPerSniffedList();
VOID ProcessAdditionalInfoList();
VOID ReadPolicyInfo();
VOID LogNodesPerCommandList();
CString GetStartOverLink();
bool StripSniffedNodePrefix(LPTSTR szName);
DWORD DoInference(
LPTSTR pszCmd,
LPTSTR pszValue,
CTopic * pTopic,
bool bUsesIDH);
DWORD NextCommand(LPTSTR pszCmd, LPTSTR pszValue, bool bUsesIDH);
DWORD NextAdditionalInfo(LPTSTR pszCmd, LPTSTR pszValue);
DWORD NextIgnore(LPTSTR pszCmd, LPTSTR pszValue);
NID NIDFromSymbolicName(LPCTSTR szNodeName);
char *GetCookieValue(char *pszName, char *pszNameValue);
void asctimeCookie(const struct tm &gmt, char * szOut);
void SetError(LPCTSTR szMessage);
// Operator actions
eOpAction IdentifyOperatorAction(CAbstractECB *pECB);
eOpAction ParseOperatorAction(CAbstractECB *pECB, CString & strArg);
void ExecuteOperatorAction(
CAbstractECB *pECB,
eOpAction action,
const CString & strArg);
// Status pages: code is in separate StatusPage.cpp
void DisplayFirstPage(bool bHasPwd);
void DisplayFurtherGlobalStatusPage();
void DisplayThreadStatusOverviewPage();
void DisplayTopicStatusPage(LPCTSTR topic_name);
bool ShowFullFirstPage(bool bHasPwd);
void InsertPasswordInForm();
void BeginSelfAddressingForm();
protected:
CAbstractECB *m_pECB; // effectively, everything that came in from
// the user in a submitted HTML form
DWORD m_dwErr;
// The next 2 are arrays of TCHAR rather than being CString, because it's easier
// for when they need to be passed to methods of EXTENSION_CONTROL_BLOCK
TCHAR m_ipstr[MAXBUF]; // Remote IP address (who submitted the form)
TCHAR m_resptype[MAXBUF]; // HTTP response type e.g. "200 OK",
// "302 Object Moved"
CString m_strHeader; // header for response file (indicates whether
// we're sending HTML, setting a cookie, etc.)
// >>> $UNICODE Is it OK that this is CString (based
// on TCHAR) or should it always be char? JM 10/27/98
CString m_strText; // this is where we build the string to pass
// back over the net.
// >>> $UNICODE Is it OK that this is CString (based
// on TCHAR) or should it always be char? JM 10/27/98
CString m_strLocalIPAddress; // IP address (in the dotted form) for the local machine
// If not defined: GetLength() == 0
CLogString m_logstr; // We log to this object & when we're all done
// destructor writes it to the log.
CHttpQuery m_Qry; // takes in raw URL-encoded string, gives us
// functions to get back scanned pairs.
CDBLoadConfiguration *m_pConf; // contains support-file data structures
CString m_strVRoot; // Local URL to this DLL
TCHAR *m_pszQuery; // a copy of what came in via GET or POST
CInfer m_infer; // belief-network handler, unique to this request
// This works out what node to show and builds HTML
// fragments for the HTI template to render.
CHTMLLog *m_pLog; // access to writing to the log.
bool m_bPostType; // TRUE = post, FALSE = get
DWORD m_dwBytes; // length of query string in chars, excluding
// terminating null.
GTS_STATISTIC *m_pStat;
CCommandsAddManager m_CommandsAddManager; // manages adding data to m_Commands and m_Sniffed
CCommands m_Commands; // name/value pairs we get from an HTML form
CCommands m_Sniffed; // name/value pairs (for sniffed nodes) we get
// from an HTML form; "SNIFFED_" already stripped out.
CAdditionalInfo m_AdditionalInfo; // name/value pairs we get from HTML. They represent
// additional info. Additional info is name/value
// pair other then command pair (C_TYPE, C_TOPIC or C_PRELOAD),
// though first in name/value pair's sequence
// will be a command.
bool m_bPreload; // TRUE = name/value pairs aren't really from a
// user, they're from a sniffer.
bool m_bNewCookie; // true = we had to create a new cookie for this
// query: they didn't already have one.
CHourlyDailyCounter * const m_pcountUnknownTopics; // count requests where the topic is not known in
// the LST file.
CHourlyDailyCounter * const m_pcountAllAccessesFinish; // Each time we finish with any sort of request,
// successful or not, this gets incremented.
CHourlyDailyCounter * const m_pcountStatusAccesses; // Each time we finish a request for system status
CHourlyDailyCounter * const m_pcountOperatorActions; // Count operator action requests.
CString m_TopicName;
// You can compile with the NOPWD option to suppress all password checking.
// This is intended mainly for creating test versions with this feature suppressed.
#ifndef NOPWD
CString m_strTempPwd; // temporary password (if this is a status request)
#endif // ifndef NOPWD
// You can compile with the SHOWPROGRESS option to get a report on the progress of this page.
#ifdef SHOWPROGRESS
time_t timeCreateContext;
time_t timeStartInfer;
time_t timeEndInfer;
time_t timeEndRender;
#endif // SHOWPROGRESS
private:
// Functions to set and retrieve an alternate HTI file name.
void SetAltHTIname( const CString& strHTIname );
CString GetAltHTIname() const;
CString m_strAltHTIname; // name of the alternate HTI template file if specified.
typedef map<CString,CString> CCookiePairs;
CCookiePairs m_mapCookiesPairs; // Map of command line cookie name-value pairs.
};
// global prototypes
//UINT PoolTask(LPVOID);
UINT WINAPI PoolTask( LPVOID lpParams );
bool ProcessRequest(CPoolQueue & PoolQueue);
DWORD WINAPI DirNotifyTask( LPDWORD lpParams );
/////////////////////////////////////////////////////////////////////////////
#endif // APGTSCLS_H_INCLUDED