bocomreg.cpp
资源名称:bo2ksrc.zip [点击查看]
上传用户:jinandeyu
上传日期:2007-01-05
资源大小:620k
文件大小:4k
源码类别:
远程控制编程
开发平台:
WINDOWS
- /* Back Orifice 2000 - Remote Administration Suite
- Copyright (C) 1999, Cult Of The Dead Cow
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 2 of the License, or
- (at your option) any later version.
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
- You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- The author of this program may be contacted at dildog@l0pht.com. */
- // BO2K Command Dispatcher
- #include<windows.h>
- #include<bocomreg.h>
- BO_CMD_HANDLER *command_handler_table=NULL;
- BO_CMD_DESC *command_description_table=NULL;
- HANDLE g_hDispatchMutex=NULL;
- int InitializeCommandDispatcher(void)
- {
- int i;
- command_handler_table=(BO_CMD_HANDLER *)malloc(sizeof(BO_CMD_HANDLER)*MAX_BO_COMMANDS);
- if(command_handler_table==NULL) return -1;
- command_description_table=(BO_CMD_DESC *)malloc(sizeof(BO_CMD_DESC)*MAX_BO_COMMANDS);
- if(command_description_table==NULL) return -1;
- for(i=0;i<MAX_BO_COMMANDS;i++) {
- command_handler_table[i]=NULL;
- }
- g_hDispatchMutex=CreateMutex(NULL,FALSE,NULL);
- if(g_hDispatchMutex==NULL) return -1;
- return 0;
- }
- int KillCommandDispatcher(void)
- {
- CloseHandle(g_hDispatchMutex);
- if(command_handler_table) free(command_handler_table);
- if(command_description_table) free(command_description_table);
- return 0;
- }
- int RegisterNativeCommand(int command, BO_CMD_HANDLER handler)
- {
- if(command<0 || command>MAX_BO_COMMANDS)
- return -1;
- command_handler_table[command]=handler;
- command_description_table[command].svCommName="";
- command_description_table[command].svFolderName="";
- command_description_table[command].svArgDesc1="";
- command_description_table[command].svArgDesc2="";
- command_description_table[command].svArgDesc3="";
- command_description_table[command].bNativeComm=TRUE;
- return 0;
- }
- int RegisterCommand(BO_CMD_HANDLER handler, char *svFolderName, char *svCommName, char *svArgDesc1, char *svArgDesc2, char *svArgDesc3)
- {
- int i;
- for(i=200;i<MAX_BO_COMMANDS;i++) {
- if(command_handler_table[i]==INVALID_CMD_HANDLER) break;
- }
- if(i==MAX_BO_COMMANDS) return -1;
- command_handler_table[i]=handler;
- command_description_table[i].svCommName=svCommName;
- command_description_table[i].svFolderName=svFolderName;
- command_description_table[i].svArgDesc1=svArgDesc1;
- command_description_table[i].svArgDesc2=svArgDesc2;
- command_description_table[i].svArgDesc3=svArgDesc3;
- command_description_table[i].bNativeComm=FALSE;
- return i;
- }
- int UnregisterCommand(int command)
- {
- if(command<0 || command>=MAX_BO_COMMANDS) return -1;
- if(command_handler_table[command]==INVALID_CMD_HANDLER) return -1;
- command_handler_table[command]=INVALID_CMD_HANDLER;
- command_description_table[command].bNativeComm=FALSE;
- return 0;
- }
- int DispatchCommand(int command, CAuthSocket *cas_from, int comid, int nArg1, char *svArg2, char *svArg3)
- {
- BO_CMD_HANDLER handler;
- int ret;
- if(WaitForSingleObject(g_hDispatchMutex,INFINITE)!=WAIT_OBJECT_0) {
- return -1;
- }
- if(cas_from->GetAuthHandler()->pValidateCommand(cas_from->GetUserID(), command)==FALSE) {
- ReleaseMutex(g_hDispatchMutex);
- return -1;
- }
- handler=command_handler_table[command];
- if(handler==INVALID_CMD_HANDLER) {
- ReleaseMutex(g_hDispatchMutex);
- return -1;
- }
- ret=handler(cas_from, comid, nArg1, svArg2, svArg3);
- ReleaseMutex(g_hDispatchMutex);
- return ret;
- }