开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 钩子与API截获 > 查看源码
KeyManager.cpp
资源名称：HookAPI　[点击查看]
上传用户：nbcables
上传日期：2007-01-11
资源大小：1243k
文件大小：2k
源码类别：

钩子与API截获

开发平台：

Visual C++

KeyManager.cpp：源码内容
  1. #include "stdafx.h"
  2. #include <time.h>
  4. #include "keymanager.h"
  7. CKeyManager::CKeyManager(void)
  8. {
  9. m_pkey_info =NULL;
  10. m_key_info_count =0;
  11. InitializeCriticalSection(&m_cs_key);
  12. }
  14. CKeyManager::~CKeyManager(void)
  15. {
  16. RemoveAllKey();
  17. DeleteCriticalSection(&m_cs_key);
  18. }
  20. // 添加一个请求连接数据
  21. int CKeyManager::AddKey(int sd, unsigned char *key, int key_len)
  22. {
  23. KEY_INFO **ppkey =&m_pkey_info;
  24. EnterCriticalSection(&m_cs_key);
  25. while(1)
  26. {
  27. if(*ppkey ==NULL)
  28. {
  29. *ppkey =(KEY_INFO *)malloc(sizeof(KEY_INFO));
  30. memset(*ppkey, 0, sizeof(KEY_INFO));
  31. (*ppkey)->sd =sd;
  32. memcpy((*ppkey)->key, key, key_len);
  33. (*ppkey)->key_len =key_len;
  34. break;
  35. }
  36. ppkey =&((*ppkey)->pnext);
  37. }
  38. LeaveCriticalSection(&m_cs_key);
  40. return 0;
  41. }
  43. // 删除一个请求连接数据
  44. int CKeyManager::RemoveKey(int sd)
  45. {
  46. KEY_INFO *pkey =m_pkey_info, *pprev_info =NULL, **ppkey =&m_pkey_info;
  47. EnterCriticalSection(&m_cs_key);
  48. while(pkey)
  49. {
  50. if(pkey->sd ==sd)
  51. {
  52. if(pprev_info)
  53. pprev_info->pnext =pkey->pnext;
  54. free(pkey);
  55. *ppkey =NULL;
  56. break;
  57. }
  58. pprev_info =pkey;
  59. pkey =pkey->pnext;
  60. ppkey =&pkey->pnext;
  61. }
  63. LeaveCriticalSection(&m_cs_key);
  65. return 0;
  66. }
  68. void CKeyManager::RemoveAllKey()
  69. {
  70. KEY_INFO *pkey =m_pkey_info, *pprev_key =NULL;
  72. EnterCriticalSection(&m_cs_key);
  73. while(pkey)
  74. {
  75. pprev_key =pkey;
  76. pkey =pkey->pnext;
  77. free(pprev_key);
  78. }
  79. m_pkey_info =NULL;
  81. LeaveCriticalSection(&m_cs_key);
  83. }
  86. int CKeyManager::GenKey(unsigned char * key)
  87. {
  88. srand((unsigned int)time(NULL));
  89. for(int i =0; i<16; i++)   // 3des key size =16
  90. key[i] =rand()%255+1;
  92. return i;
  93. }
  95. KEY_INFO *CKeyManager::GetKey(int sd)
  96. {
  97. KEY_INFO *pkey =m_pkey_info;
  99. EnterCriticalSection(&m_cs_key);
  101. while(pkey)
  102. {
  103. if(pkey->sd ==sd)
  104. return pkey;
  105. pkey =pkey->pnext;
  106. }
  108. LeaveCriticalSection(&m_cs_key);
  110. return pkey;
  111. }