shell.h

来自「费尔个人防火墙 2.1 源程序 版权所有 (C) 费尔安全实验室 」· C头文件 代码 · 共 64 行

// Shell.h
//
// copyright (c) 2002 Passeck Technology
// 
// all right reserved.
//
// author: tony
// 
// create date: 2002/11/05
//
// email: [e-mail]
//
//
//

#ifndef _SHELL_H
#define _SHELL_H

#define SHELL_INSTALL_KEY		"SOFTWARE\\FilSecLab"
#define SHELL_REG_PROGRAM		"RegProgram"
#define SHELL_UPDATE_PROGRAM	"UpdateProgram"
#define SHELL_USER_REG_PROGRAM  "UserRegProgram"

typedef enum{
	SHELL_ERROR_SUCCESS = 0,			// 成功
	SHELL_ERROR_CANT_GET_PATH,			// 不能从注册表得到程序路径
	SHELL_ERROR_CANT_RUN,				// 不能运行程序，程序不存在或者被损坏
	SHELL_ERROR_INVALID_PROGRAM_TYPE,	// 无效的类型
} SHELL_ERROR;

typedef enum{
	PROGRAM_REGISTER,
	PROGRAM_UPDATE,
	PROGRAM_USERREG,
} PROGRAM_TYPE;

class CShell
{
public:
	CShell();
	virtual ~CShell();
public:
	int RunProgram(BYTE bProgramType, BOOL bIsWait = FALSE);
	LPCTSTR GetCmdLine(){return m_CmdLine;}
	LPCTSTR GetProgram(){return m_Program;}
private:
	char m_Program[MAX_PATH];
	char m_CmdLine[MAX_PATH];
public:
	static BOOL GetUpdateProgram(char* szProgram, int nMaxLen);
	static BOOL GetRegProgram(char* szProgram, int nMaxLen);
	static BOOL GetUserReg(char* szProgram, int nMaxLen);
	static HANDLE Pg_CreateProcess(LPCTSTR szCommandLine, BOOL bIsWait = FALSE);
	static BOOL ReadReg(
		TCHAR	*sKey,
		void	*pBuffer,
		DWORD	dwBufSize, 
		HKEY	hkey	 = HKEY_LOCAL_MACHINE, 
		TCHAR	*sSubKey = SHELL_INSTALL_KEY,
		DWORD	ulType	 = REG_SZ
		);
};

#endif //_SHELL_H