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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 其他 > 查看源码
uniqueiv.c
资源名称：1121aircrack-2.41.zip　[点击查看]
上传用户：fubang
上传日期：2009-06-18
资源大小：2071k
文件大小：4k
源码类别：

其他

开发平台：

Unix_Linux

uniqueiv.c：源码内容
  1. /*
  2.  *  IV uniqueness detection method, as designed by Stanislaw Pusep:
  3.  *
  4.  *  Each IV byte is stored in corresponding "level". We have 3 levels with
  5.  *  IV[2] as root index (level 0), IV[1] and IV[2] as level 2 and level 1
  6.  *  indices respectively. Space required to allocate all data is at maximum
  7.  *  2^24/8 (2 MB) and space required by filled index structures is 257 KB.
  8.  */
  10. #include <stdlib.h>
  12. #define IV_NOTHERE  0
  13. #define IV_PRESENT  1
  15. /* select byte within which desired bit is located */
  17. #define BITWISE_OFFT(x)         (x >> 3)
  19. /* mask to extract desired bit */
  21. #define BITWISE_MASK(x)         (1 << (x & 7))
  23. /* allocate root structure */
  25. unsigned char **uniqueiv_init( void )
  26. {
  27.     int i;
  29.     /* allocate root bucket (level 0) as vector of pointers */
  30.     
  31.     unsigned char **uiv_root = (unsigned char **)
  32.         malloc( 256 * sizeof( unsigned char * ) );
  33.     
  34.     if( uiv_root == NULL )
  35.         return( NULL );
  37.     /* setup initial state as empty */
  39.     for( i = 0; i < 256; i++ )
  40.         uiv_root[i] = NULL;
  42.     return( uiv_root );
  43. }
  45. /* update records with new IV */
  47. int uniqueiv_mark( unsigned char **uiv_root, unsigned char IV[3] )
  48. {
  49.     unsigned char **uiv_lvl1;
  50.     unsigned char  *uiv_lvl2;
  51.     short i;
  53.     if( uiv_root == NULL )
  54.         return( 0 );
  56.     /* select bucket from level 1 */
  58.     uiv_lvl1 = (unsigned char **) uiv_root[IV[2]];
  60.     /* create if it doesn't exists */
  62.     if( uiv_lvl1 == NULL )
  63.     {
  64.         /* allocate level 2 bucket being a vector of bits */
  66.         uiv_lvl1 = (unsigned char **) malloc( 256 * sizeof( unsigned char * ) );
  68.         if( uiv_lvl1 == NULL )
  69.             return( 1 );
  71.         /* setup initial state as empty */
  73.         for( i = 0; i < 256; i++ )
  74.             uiv_lvl1[i] = NULL;
  76.         /* link to parent bucket */
  78.         uiv_root[IV[2]] = (unsigned char *) uiv_lvl1;
  79.     }
  81.     /* select bucket from level 2 */
  83.     uiv_lvl2 = (unsigned char *) uiv_lvl1[IV[1]];
  85.     /* create if it doesn't exists */
  87.     if( uiv_lvl2 == NULL )
  88.     {
  89.         /* allocate level 2 bucket as a vector of pointers */
  91.         uiv_lvl2 = (unsigned char *) malloc( 32 * sizeof( unsigned char ) );
  93.         if( uiv_lvl1 == NULL )
  94.             return( 1 );
  96.         /* setup initial state as empty */
  98.         for( i = 0; i < 32; i++ )
  99.             uiv_lvl2[i] = 0;
  101.         /* link to parent bucket */
  103.         uiv_lvl1[IV[1]] = uiv_lvl2;
  104.     }
  106.     /* place single bit into level 2 bucket */
  108.     uiv_lvl2[BITWISE_OFFT( IV[0] )] |= BITWISE_MASK( IV[0] );
  109.         
  110.     return( 0 );
  111. }
  113. /* check if already seen IV */
  115. int uniqueiv_check( unsigned char **uiv_root, unsigned char IV[3] )
  116. {
  117.     unsigned char **uiv_lvl1;
  118.     unsigned char  *uiv_lvl2;
  119.         
  120.     if( uiv_root == NULL )
  121.         return( IV_NOTHERE );
  123.     /* select bucket from level 1 */
  125.     uiv_lvl1 = (unsigned char **) uiv_root[IV[2]];
  127.     /* stop here if not even allocated */
  129.     if( uiv_lvl1 == NULL )
  130.         return( IV_NOTHERE );
  132.     /* select bucket from level 2 */
  134.     uiv_lvl2 = (unsigned char *) uiv_lvl1[IV[1]];
  136.     /* stop here if not even allocated */
  138.     if( uiv_lvl2 == NULL )
  139.         return( IV_NOTHERE );
  141.     /* check single bit from level 2 bucket */
  143.     if( ( uiv_lvl2[ BITWISE_OFFT( IV[0] ) ]
  144.                   & BITWISE_MASK( IV[0] ) ) == 0 )
  145.         return( IV_NOTHERE );
  146.     else
  147.         return( IV_PRESENT );
  148. }
  150. /* unallocate everything */
  152. void uniqueiv_wipe( unsigned char **uiv_root )
  153. {
  154.     int i, j;
  155.     unsigned char **uiv_lvl1;
  156.     unsigned char  *uiv_lvl2;
  157.         
  158.     if( uiv_root == NULL )
  159.         return;
  161.     /* recursively wipe out allocated buckets */
  163.     for( i = 0; i < 256; i++ )
  164.     {
  165.         uiv_lvl1 = (unsigned char **) uiv_root[i];
  167.         if( uiv_lvl1 != NULL )
  168.         {
  169.             for( j = 0; j < 256; j++ )
  170.             {
  171.                 uiv_lvl2 = (unsigned char *) uiv_lvl1[j];
  173.                 if( uiv_lvl2 != NULL )
  174.                     free( uiv_lvl2 );
  175.             }
  177.             free( uiv_lvl1 );
  178.         }
  179.     }
  181.     free( uiv_root );
  183.     return;
  184. }