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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
Check.cpp
资源名称：FireWallSys.rar　[点击查看]
上传用户：yitai_qhd
上传日期：2008-04-24
资源大小：31k
文件大小：6k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

Check.cpp：源码内容
  1. #include "check.h"
  2. #include "packet.h"
  3. #include "windef.h"
  4. #include "stdio.h"
  5. #include "string.h"
  7. int CheckTcp(
  8. PIP_HEADER pIpHeader, 
  9. PTCP_HEADER pTcpHeader, 
  10. BOOLEAN IsSend, 
  11. UINT LookaheadBufferSize,
  12. PVOID pVoid
  13. )
  14. {
  15. NTSTATUS Status;
  16. NTSTATUS LogStatus;
  17.     UNICODE_STRING UnicodeFilespec;
  18. UNICODE_STRING UnicodeLogFile;
  19.     OBJECT_ATTRIBUTES ObjectAttributes;
  20. OBJECT_ATTRIBUTES LogObjectAttributes;
  21.     HANDLE FileHandle;
  22. HANDLE LogFileHandle;
  23. DWORD IP;
  25.     IO_STATUS_BLOCK Iosb;
  26. IO_STATUS_BLOCK LogIosb;
  28.     RtlInitUnicodeString(&UnicodeFilespec, L"\DosDevices\C:\ip.dat");
  29. RtlInitUnicodeString(&UnicodeLogFile, L"\DosDevices\C:\iplog.dat");
  32.     InitializeObjectAttributes(&ObjectAttributes,           // ptr to structure
  33.                                &UnicodeFilespec,            // ptr to file spec
  34.                                OBJ_CASE_INSENSITIVE,        // attributes
  35.                                NULL,                        // root directory handle
  36.                                NULL );                      // ptr to security descriptor
  37. InitializeObjectAttributes(&LogObjectAttributes,           // ptr to structure
  38.                                &UnicodeLogFile,            // ptr to file spec
  39.                                OBJ_CASE_INSENSITIVE,        // attributes
  40.                                NULL,                        // root directory handle
  41.                                NULL );                      // ptr to security descriptor
  44.     Status = ZwCreateFile(&FileHandle,                      // returned file handle
  45.                           (GENERIC_READ| SYNCHRONIZE),    // desired access
  46.                           &ObjectAttributes,                // ptr to object attributes
  47.                           &Iosb,                            // ptr to I/O status block
  48.                           0,                                // allocation size
  49.                           FILE_ATTRIBUTE_NORMAL,            // file attributes
  50.                           0,                                // share access
  51.                           FILE_SUPERSEDE,                   // create disposition
  52.                           FILE_SYNCHRONOUS_IO_NONALERT,     // create options
  53.                           NULL,                             // ptr to extended attributes
  54.                           0);                               // length of ea buffer
  56.     LogStatus = ZwCreateFile(&LogFileHandle,                      // returned file handle
  57.                           (FILE_APPEND_DATA| SYNCHRONIZE),    // desired access
  58.                           &LogObjectAttributes,                // ptr to object attributes
  59.                           &LogIosb,                            // ptr to I/O status block
  60.                           0,                                // allocation size
  61.                           FILE_ATTRIBUTE_NORMAL,            // file attributes
  62.                           0,                                // share access
  63.                           FILE_SUPERSEDE,                   // create disposition
  64.                           FILE_SYNCHRONOUS_IO_NONALERT,     // create options
  65.                           NULL,                             // ptr to extended attributes
  66.                           0);      
  67.     //
  68.     // Check the system service status
  69.     //
  70.     if( !NT_SUCCESS(Status) )
  71.         {
  73.       }
  76.     //
  77.     // Check the returned status too...
  78.     //
  79.     if(!NT_SUCCESS(Iosb.Status) )
  80.         {
  82.        }
  85.     Status = ZwReadFile(FileHandle,                   
  86.                          0,                           
  87.                          NULL,                        
  88.                          NULL,                        
  89.                          &Iosb,                       
  90.                          &IP,                      
  91.                          4,                
  92.                          0,                            
  93.                          NULL);                        
  96. if(IsSend)
  97. {
  98. while(!NT_SUCCESS(Status)&&!NT_SUCCESS(Iosb.Status))
  99. {
  100. Status = ZwReadFile(FileHandle,                   
  101.             0,                           
  102. NULL,                        
  103. NULL,                        
  104. &Iosb,                       
  105. &IP,                      
  106. 4,                
  107. 0,                            
  108. NULL);
  109. if(strcmp((char *)pIpHeader->SourceIp,(char *)IPTrans(IP)))
  110. {
  111. LogStatus = ZwWriteFile(LogFileHandle,                   
  112.             0,                           
  113. NULL,                        
  114. NULL,                        
  115. &Iosb,                       
  116. &IP,                      
  117. 4,                
  118. 0,                            
  119. NULL);
  120. return -1;
  121. }
  122. }
  123. }
  124. else
  125. {
  126. while(!NT_SUCCESS(Status)&&!NT_SUCCESS(Iosb.Status))
  127. {
  128. Status = ZwReadFile(FileHandle,                   
  129.             0,                           
  130. NULL,                        
  131. NULL,                        
  132. &Iosb,                       
  133. &IP,                      
  134. 4,                
  135. 0,                            
  136. NULL);
  137. if(strcmp((char *)pIpHeader->DestinationIp,(char *)IPTrans(IP)))
  138. {
  139. LogStatus = ZwWriteFile(LogFileHandle,                   
  140.             0,                           
  141. NULL,                        
  142. NULL,                        
  143. &Iosb,                       
  144. &IP,                      
  145. 4,                
  146. 0,                            
  147. NULL);
  148. return -1;
  149. }
  150. }
  152. }
  153.     //
  154.     // Well, That's all folks!
  155.     //
  156.     Status = ZwClose(FileHandle);
  157. return 1;
  158. }
  160. int CheckUdp(
  161. PIP_HEADER pIpHeader, 
  162. PUDP_HEADER pUdpHeader, 
  163. BOOLEAN IsSend,
  164. UINT LookaheadBufferSize,
  165. void *pVoid
  166. )
  167. {
  168. if(pUdpHeader->Length!=LookaheadBufferSize)
  169. return 1;
  171. return 0;
  172. }
  174. int CheckIcmp(
  175. PIP_HEADER pIpHeader, 
  176. PICMP_HEADER pIcmpHeader, 
  177. BOOLEAN IsSend,
  178. UINT LookaheadBufferSize 
  179. )
  180. {
  181. if(!IsSend) return -1;
  182. return 1;
  183. }
  184. unsigned char* IPTrans(DWORD IP)
  185. {
  186. int ip1=(IP&0x000000ff);
  187. int ip2=(IP&0x0000ff00)>>8;
  188. int ip3=(IP&0x00ff0000)>>16;
  189. int ip4=(IP&0xff000000)>>24;
  190. unsigned char Tmp[4];
  191. sprintf((char*)Tmp,"%c",(char)ip1);
  192. sprintf((char *)&Tmp[1],"%c",(char)ip2);
  193. sprintf((char *)&Tmp[2],"%c",(char)ip3);
  194. sprintf((char *)&Tmp[3],"%c",(char)ip4);
  196. return Tmp;
  197. }