home *** CD-ROM | disk | FTP | other *** search
/ io Programmo 68 / IOPROG_68.ISO / soft / Codice / SPS.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2003-03-04  |  11.7 KB  |  394 lines
  1. //SPS.cpp
  2. //Simple Packet Sniffer (SPS)
  3. //(c) Elia Florio (eflorio@edmaster.it)
  4. #include <stdio.h>
  5.  
  6. //WINSOCK 2 - RAW SOCKET SUPPORT
  7. #include <winsock2.h>
  8. #pragma comment (lib, "ws2_32.lib")
  9. #define SIO_RCVALL _WSAIOW(IOC_VENDOR,1)    //required by winsock2.h
  10. #define MAX_PACKET_SIZE (65536-20)            //2^16 - ip_header_length
  11.  
  12. //log file C:\SPSLOG.TXT
  13. FILE *file=NULL;
  14. char* filename;
  15.  
  16. //menu' variables
  17. bool filelog;
  18. bool hexlog;
  19. char ipaddr[255];
  20. int  filter_type;
  21.  
  22. //-----------------------------------------------------------------
  23. // INTERNET HEADERS
  24. //-----------------------------------------------------------------
  25.  
  26. //IPv4 HEADER (20 BYTES)
  27. typedef struct iphdr
  28. {
  29.     unsigned char  VerIHL;            // 8-bit - Version (4-bit) + IP Header Length (4-bit)
  30.     unsigned char  Tos;                // 8-bit - Type of Service
  31.     unsigned short Total_len;        //16-bit - Total Length
  32.     unsigned short ID;                //16-bit - Identification
  33.     unsigned short Flags_and_Frags; //16-bit - Flags (3-bit) + Fragment offset (13-bit)
  34.     unsigned char  TTL;                // 8-bit - Time To Live 
  35.     unsigned char  Protocol;        // 8-bit - Protocol
  36.     unsigned short Checksum;        //16-bit - Header Checksum
  37.     unsigned long  SrcIP;            //32-bit - Source IP Address
  38.     unsigned long  DstIP;            //32-bit - Destination IP Address
  39.     //unsigned long Opt_and_Padd;
  40. } IpHeader;
  41.  
  42.  
  43. //TCP HEADER (20 BYTES + DATA)
  44. typedef struct tcphd
  45. {
  46.     unsigned short SrcPort;            //16-bit - Source Port
  47.     unsigned short DstPort;            //16-bit - Destination Port
  48.     unsigned long  SeqNumber;        //32-bit - Seq Number
  49.     unsigned long  AckNumber;        //32-bit - Ack Number
  50.     unsigned short Off_Res_Ctrl;    //16-bit - D-Offset (4-bit) + Reserved (6-bit) + Ctrl (6-bit)
  51.     unsigned short Window;            //16-bit - Window
  52.     unsigned short Checksum;        //16-bit - Checksum
  53.     unsigned short Urgent;            //16-bit - Urgent Pointer
  54.     //unsigned long Opt_and_Padd;
  55.     unsigned char dati[MAX_PACKET_SIZE];
  56. } TcpHeader;
  57.  
  58.  
  59. //UDP HEADER (8 BYTES + DATA)
  60. typedef struct udphd
  61. {
  62.     unsigned short SrcPort;            //16-bit - Source Port
  63.     unsigned short DstPort;            //16-bit - Destination Port
  64.     unsigned short Length;            //16-bit - Datagram Length
  65.     unsigned short Checksum;        //16-bit - Checksum
  66.     unsigned char dati[MAX_PACKET_SIZE];
  67. } UdpHeader;
  68.  
  69.  
  70. // ICMP Header (8 BYTES)
  71. typedef struct icmphd
  72. { 
  73.     unsigned char  icmp_type;         //  8-bit - Type of message
  74.     unsigned char  icmp_code;        //  8-bit - Code
  75.     unsigned short icmp_cksum;       // 16-bit - Checksum
  76.     unsigned short icmp_id;          // 16-bit - Identifer
  77.     unsigned short icmp_seq;         // 16-bit - Sequence number
  78. } IcmpHeader;
  79. //-----------------------------------------------------------------
  80.  
  81.  
  82. //PACKET FILTERING/ANALYSING FUNCTION
  83. void PacketFilter(char* Buffer, int Size)
  84. {
  85.     IpHeader *iphdr;
  86.     TcpHeader *tcphd;
  87.     UdpHeader *udphd;
  88.     IcmpHeader *icmphd;
  89.     struct sockaddr_in SockAddr;
  90.     unsigned short iphdrlen;
  91.     char C;
  92.  
  93.     //Analyze packet
  94.     iphdr = (IpHeader *)Buffer;
  95.     iphdrlen = (iphdr->VerIHL << 4);
  96.     memcpy(&C, &iphdrlen, 1);
  97.     iphdrlen = (C >> 4) * 4; //20 bytes
  98.  
  99.     //Analyze IP Header
  100.     printf("\n=====CAPTURED PACKET======\n");
  101.     if (filelog) fprintf(file,"\n=====CAPTURED PACKET======\n");
  102.     
  103.     printf("* [Header IP] \n");
  104.     if (filelog) fprintf(file,"* [Header IP] \n");
  105.     
  106.     memset(&SockAddr, 0, sizeof(SockAddr));
  107.     SockAddr.sin_addr.s_addr = iphdr->SrcIP;
  108.     printf("  From: %s ", inet_ntoa(SockAddr.sin_addr));
  109.     if (filelog) fprintf(file,"  From: %s ", inet_ntoa(SockAddr.sin_addr));
  110.  
  111.     memset(&SockAddr, 0, sizeof(SockAddr));
  112.     SockAddr.sin_addr.s_addr = iphdr->DstIP;
  113.     printf("  To: %s \n", inet_ntoa(SockAddr.sin_addr));
  114.     if (filelog) fprintf(file,"  To: %s \n", inet_ntoa(SockAddr.sin_addr));
  115.  
  116.     printf("  ID: %i \n", ntohs(iphdr->ID));
  117.     if (filelog) fprintf(file,"  ID: %i \n", ntohs(iphdr->ID));
  118.  
  119.     printf("  TTL: %i \n", ntohs(iphdr->TTL));
  120.     if (filelog) fprintf(file,"  TTL: %i \n", ntohs(iphdr->TTL));
  121.  
  122.     printf("  Total Length: %i \n\n", ntohs(iphdr->Total_len));
  123.     if (filelog) fprintf(file,"  Total Length: %i \n\n", ntohs(iphdr->Total_len));
  124.  
  125.  
  126.     //Analyze IP Protocol
  127.     switch (iphdr->Protocol)
  128.     {
  129.     case 1:
  130.         printf("* [Protocol: ICMP] \n");
  131.         if (filelog) fprintf(file,"* [Protocol: ICMP] \n");
  132.  
  133.         if(filter_type==4 || filter_type==3) {
  134.         if (Size > iphdrlen)
  135.         {
  136.             //Analyze ICMP
  137.             icmphd = (IcmpHeader *)(Buffer + iphdrlen);
  138.  
  139.             printf("  Type: %i  ", icmphd->icmp_type);
  140.             if (filelog) fprintf(file,"  Type: %i  ", icmphd->icmp_type);
  141.  
  142.             printf("  Code: %i  \n", icmphd->icmp_code);
  143.             if (filelog) fprintf(file,"  Code: %i  \n", icmphd->icmp_code);
  144.  
  145.             printf("  IcmpSeq: %i \n", icmphd->icmp_seq);
  146.             if (filelog) fprintf(file,"  IcmpSeq: %i \n", icmphd->icmp_seq);
  147.  
  148.             if(icmphd->icmp_type==0)  {printf("  (EchoReply)"); if(filelog) fprintf(file,"  (EchoReply)");}
  149.             if(icmphd->icmp_type==3)  {printf("  (DestUnreach)"); if(filelog) fprintf(file,"  (DestUnreach)");}
  150.             if(icmphd->icmp_type==8)  {printf("  (EchoRequest)"); if(filelog) fprintf(file,"  (EchoRequest)");}
  151.             if(icmphd->icmp_type==11) {printf("  (TimeExceed)"); if(filelog) fprintf(file,"  (TimeExceed)");}
  152.             if(icmphd->icmp_type==30) {printf("  (TraceRoute)"); if(filelog) fprintf(file,"  (TraceRoute)");}
  153.             }
  154.         }
  155.  
  156.         break;
  157.  
  158.  
  159.     case 6:
  160.         printf("* [Protocol: TCP] \n");
  161.         if (filelog) fprintf(file,"* [Protocol: TCP] \n");
  162.  
  163.         if(filter_type==4 || filter_type==1) {
  164.         if (Size > iphdrlen)
  165.         {
  166.             //Analyze TCP
  167.             tcphd = (TcpHeader *)(Buffer + iphdrlen);
  168.             
  169.             printf("  Data Offset = %i\n",ntohs(tcphd->Off_Res_Ctrl)>>12);
  170.             if(filelog) fprintf(file,"  Data Offset = %i\n",ntohs(tcphd->Off_Res_Ctrl)>>12);            
  171.  
  172.             printf("  SrcPort: %i  =>  ", ntohs(tcphd->SrcPort));
  173.             if (filelog) fprintf(file,"  SrcPort: %i  =>  ", ntohs(tcphd->SrcPort));
  174.             
  175.             //try to guess upper-layer protocol looking src port number
  176.             if(ntohs(tcphd->SrcPort)==25)   {printf("(SMTP)"); if(filelog) fprintf(file,"(SMTP)");}
  177.             if(ntohs(tcphd->SrcPort)==80)   {printf("(HTTP)"); if(filelog) fprintf(file,"(HTTP)");}
  178.             if(ntohs(tcphd->SrcPort)==110)  {printf("(POP3)"); if(filelog) fprintf(file,"(POP3)");}
  179.             if(ntohs(tcphd->SrcPort)==6667) {printf("(IRC)");  if(filelog) fprintf(file,"(IRC)");}
  180.  
  181.             printf("\n  DstPort: %i \n", ntohs(tcphd->DstPort));
  182.             if(filelog) fprintf(file,"\n  DstPort: %i \n", ntohs(tcphd->DstPort));
  183.  
  184.             printf("  SeqNumber: %i \n  AckNumber: %i \n", ntohl(tcphd->SeqNumber), ntohl(tcphd->AckNumber));
  185.             if(filelog) fprintf(file,"  SeqNumber: %i \n  AckNumber: %i \n", ntohl(tcphd->SeqNumber), ntohl(tcphd->AckNumber));
  186.             
  187.             //calculate data offset and length
  188.             int data_start = ((ntohs(tcphd->Off_Res_Ctrl)>>12)-5)*4;
  189.             int data_end = ntohs(iphdr->Total_len) - iphdrlen - 20 - data_start;
  190.             printf("  Data Dump (%i): \n",data_end);
  191.             if(filelog) fprintf(file,"  Data Dump (%i): \n",data_end);
  192.  
  193.             //get data bytes
  194.             for(int j=data_start;j<data_end;j++) {
  195.                 if(!hexlog) printf("%c",tcphd->dati[j]);
  196.                 else {printf("%X ",tcphd->dati[j]);
  197.                       if(j%16==0 && j!=0) printf("\n");}
  198.                 
  199.                 if(filelog) {
  200.                     if(!hexlog) fprintf(file,"%c",tcphd->dati[j]);
  201.                     else {fprintf(file, "%X ",tcphd->dati[j]);
  202.                           if(j%16==0 && j!=0) fprintf(file,"\n");}
  203.                 }
  204.                 }
  205.             printf("\n");
  206.             }
  207.         }
  208.         break;
  209.  
  210.     
  211.     case 17:
  212.         printf("* [Protocol: UDP] \n");
  213.         if (filelog) fprintf(file,"* [Protocol: UDP] \n");
  214.  
  215.         if(filter_type==4 || filter_type==2) {
  216.         if (Size > iphdrlen)
  217.         {
  218.             //Analyze UDP
  219.             udphd = (UdpHeader *)(Buffer + iphdrlen);
  220.             
  221.             printf("  SrcPort: %i  =>  ", ntohs(udphd->SrcPort));
  222.             if (filelog) fprintf(file,"  SrcPort: %i  =>  ", ntohs(udphd->SrcPort));
  223.  
  224.             //try to guess upper-layer protocol looking src port number            
  225.             if(ntohs(udphd->SrcPort)==53)  {printf("(DNS)"); if(filelog) fprintf(file,"(DNS)");}
  226.             if(ntohs(udphd->SrcPort)==137)  {printf("(NetBIOS)"); if(filelog) fprintf(file,"(NetBIOS)");}
  227.  
  228.             printf("\n  DstPort: %i \n", ntohs(udphd->DstPort));
  229.             if(filelog) fprintf(file,"\n  DstPort: %i \n", ntohs(udphd->DstPort));
  230.  
  231.             printf("  Length: %i \n", ntohs(udphd->Length));
  232.             if(filelog) fprintf(file,"  Length: %i \n", ntohs(udphd->Length));
  233.  
  234.             int data_start = 0;
  235.             int data_end = ntohs(udphd->Length) - 8;
  236.             printf("  Data Dump (%i): \n",data_end);
  237.             if(filelog) fprintf(file,"  Data Dump (%i): \n",data_end);
  238.  
  239.             for(int j=data_start;j<data_end;j++) {
  240.                 if(!hexlog) printf("%c",udphd->dati[j]);
  241.                 else {printf("%X ",udphd->dati[j]);
  242.                       if(j%16==0 && j!=0) printf("\n");}
  243.  
  244.                 if(filelog) {
  245.                     if(!hexlog) fprintf(file,"%c",udphd->dati[j]);
  246.                     else {fprintf(file, "%X ",udphd->dati[j]);
  247.                           if(j%16==0 && j!=0) fprintf(file,"\n");}
  248.                     }
  249.                 }
  250.             printf("\n");
  251.             }
  252.         }
  253.         break;
  254.  
  255.  
  256.     default:
  257.         printf("* [Protocol: %i ] \n", iphdr->Protocol); 
  258.         if (filelog) fprintf(file,"* [Protocol: %i ] \n", iphdr->Protocol); 
  259.     }
  260.  
  261.     printf("\n");
  262.     if(filelog) fprintf(file,"\n");
  263. }
  264.  
  265.  
  266. // SNIFFER LOOP
  267. void Sniffing(SOCKET Sock)
  268. {
  269.     char *RecvBuffer = (char *)malloc(MAX_PACKET_SIZE + 1);
  270.     int BytesRecv, FromLen;
  271.     struct sockaddr_in From;
  272.  
  273.     if (RecvBuffer == NULL)
  274.     {    printf("malloc() failed.\n");
  275.         exit(-1);
  276.     }
  277.  
  278.     FromLen = sizeof(From);
  279.  
  280.     
  281.     //packet-capture loop
  282.     do
  283.     {
  284.         memset(RecvBuffer, 0, MAX_PACKET_SIZE + 1);
  285.         memset(&From, 0, sizeof(From));
  286.  
  287.         BytesRecv = recvfrom(Sock, RecvBuffer, MAX_PACKET_SIZE, 0, (sockaddr *)&From, &FromLen);
  288.         
  289.         //capture a packet and pass it to filtering function
  290.         if (BytesRecv > 0)
  291.         {    PacketFilter(RecvBuffer, BytesRecv);}
  292.         else
  293.         {    printf( "recvfrom() failed.\n");}
  294.  
  295.     } while (BytesRecv > 0);
  296.  
  297.  
  298.     free(RecvBuffer);
  299. }
  300.  
  301.  
  302. // MAIN PROGRAM
  303. // - NETWORK CARD SETUP
  304. // - RAW SOCKET BINDING
  305. int main()
  306. {
  307.     WSAData wsaData;
  308.     SOCKET Sock;
  309.     struct sockaddr_in SockAddr;
  310.     DWORD BytesReturned;
  311.     int I = 1;
  312.     int sfres,sfsel;
  313.  
  314.     try
  315.     {
  316.         if (WSAStartup(MAKEWORD(2, 1), &wsaData) != 0)
  317.         {    printf("Error: WSAStartup() failed\n");
  318.             exit(-1);
  319.         }
  320.  
  321.         Sock = socket(AF_INET, SOCK_RAW, IPPROTO_IP);
  322.  
  323.         if (Sock == INVALID_SOCKET)
  324.         {    printf("Error: socket() failed\n");
  325.             exit(-1);
  326.         }
  327.  
  328.         memset(&SockAddr, 0, sizeof(SockAddr));
  329.  
  330.         printf("\nSIMPLE PACKET SNIFFER\n");
  331.         printf("---------------------  \n");
  332.         printf("Written by Elia Florio \n");
  333.         printf("(eflorio@edmaster.it)  \n\n");
  334.         printf("Local IP Adrress  ? ");
  335.         sfres = scanf("%s",&ipaddr);
  336.  
  337.         SockAddr.sin_addr.s_addr = inet_addr(ipaddr);
  338.         SockAddr.sin_family = AF_INET;
  339.         SockAddr.sin_port = htons(0); //anything...
  340.  
  341.         //binding socket in PROMISCUOUS MODE
  342.         if (bind(Sock, (SOCKADDR *)&SockAddr, sizeof(SockAddr)) == SOCKET_ERROR)
  343.             {    printf("Error: bind(%s) failed\n", inet_ntoa(SockAddr.sin_addr));
  344.                 exit(-1);
  345.             }
  346.         
  347.         if (WSAIoctl(Sock, SIO_RCVALL, &I, sizeof(I), NULL, NULL, &BytesReturned, NULL, NULL) == SOCKET_ERROR)
  348.         {    printf("Error: WSAIoctl() failed\n");
  349.             exit(-1);
  350.         }
  351.  
  352.         printf("\n1) Log immediato solo su console\n");
  353.         printf("2) Log su console e su file (C:\\SPSLOG.TXT)\n");
  354.         printf("Modalita' di logging ? ");
  355.         sfres = scanf("%d",&sfsel);
  356.         if (sfsel==1) filelog=false;
  357.         else filelog=true;
  358.  
  359.         printf("\n1) Dump in formato ASCII leggibile\n");
  360.         printf("2) Dump in formato HEX\n");
  361.         printf("Modalita' di dump ? ");
  362.         sfres = scanf("%d",&sfsel);
  363.         if (sfsel==1) hexlog=false;
  364.         else hexlog=true;
  365.  
  366.         printf("\n1) TCP\n");
  367.         printf("2) UDP\n");
  368.         printf("3) ICMP\n");
  369.         printf("4) TCP / UDP / ICMP\n");
  370.         printf("Pacchetti da analizzare ? ");
  371.         sfres = scanf("%d",&filter_type);
  372.         
  373.         printf("\n\n\n\n\n-------------------------------------------BEGIN SESSION\n");
  374.         
  375.         if (filelog) {
  376.             filename="C:\\SPSLOG.TXT";
  377.             file=fopen(filename,"a+");
  378.             fprintf(file,"\n-------------------------------------------BEGIN SESSION\n");
  379.         }
  380.         
  381.         Sniffing(Sock);
  382.     }
  383.     
  384.     catch (...)
  385.     {
  386.         printf("Error: Unknown exception\n");
  387.     }
  388.  
  389.     closesocket(Sock);
  390.     WSACleanup();
  391.  
  392.     return 0;
  393. }
  394.