OpenCL/inc_types.cl

   1 /**
   2  * Authors.....: Jens Steube <jens.steube@gmail.com>
   3  *               magnum <john.magnum@hushmail.com>
   4  *
   5  * License.....: MIT
   6  */
   7
   8 typedef uchar  u8;
   9 typedef ushort u16;
  10 typedef uint   u32;
  11 typedef ulong  u64;
  12
  13 #ifndef NEW_SIMD_CODE
  14 #undef  VECT_SIZE
  15 #define VECT_SIZE 1
  16 #endif
  17
  18 #define CONCAT(a, b)       a##b
  19 #define VTYPE(type, width) CONCAT(type, width)
  20
  21 #if VECT_SIZE == 1
  22 typedef uchar   u8x;
  23 typedef ushort  u16x;
  24 typedef uint    u32x;
  25 typedef ulong   u64x;
  26 #else
  27 typedef VTYPE(uchar,  VECT_SIZE)  u8x;
  28 typedef VTYPE(ushort, VECT_SIZE) u16x;
  29 typedef VTYPE(uint,   VECT_SIZE) u32x;
  30 typedef VTYPE(ulong,  VECT_SIZE) u64x;
  31 #endif
  32
  33 inline u32 l32_from_64_S (u64 a)
  34 {
  35   const u32 r = (u32) (a);
  36
  37   return r;
  38 }
  39
  40 inline u32 h32_from_64_S (u64 a)
  41 {
  42   a >>= 32;
  43
  44   const u32 r = (u32) (a);
  45
  46   return r;
  47 }
  48
  49 inline u64 hl32_to_64_S (const u32 a, const u32 b)
  50 {
  51   return as_ulong ((uint2) (b, a));
  52 }
  53
  54 inline u32x l32_from_64 (u64x a)
  55 {
  56   u32x r;
  57
  58   #if VECT_SIZE == 1
  59   r    = (u32) a;
  60   #endif
  61
  62   #if VECT_SIZE >= 2
  63   r.s0 = (u32) a.s0;
  64   r.s1 = (u32) a.s1;
  65   #endif
  66
  67   #if VECT_SIZE >= 4
  68   r.s2 = (u32) a.s2;
  69   r.s3 = (u32) a.s3;
  70   #endif
  71
  72   #if VECT_SIZE >= 8
  73   r.s4 = (u32) a.s4;
  74   r.s5 = (u32) a.s5;
  75   r.s6 = (u32) a.s6;
  76   r.s7 = (u32) a.s7;
  77   #endif
  78
  79   #if VECT_SIZE >= 16
  80   r.s8 = (u32) a.s8;
  81   r.s9 = (u32) a.s9;
  82   r.sa = (u32) a.sa;
  83   r.sb = (u32) a.sb;
  84   r.sc = (u32) a.sc;
  85   r.sd = (u32) a.sd;
  86   r.se = (u32) a.se;
  87   r.sf = (u32) a.sf;
  88   #endif
  89
  90   return r;
  91 }
  92
  93 inline u32x h32_from_64 (u64x a)
  94 {
  95   a >>= 32;
  96
  97   u32x r;
  98
  99   #if VECT_SIZE == 1
 100   r    = (u32) a;
 101   #endif
 102
 103   #if VECT_SIZE >= 2
 104   r.s0 = (u32) a.s0;
 105   r.s1 = (u32) a.s1;
 106   #endif
 107
 108   #if VECT_SIZE >= 4
 109   r.s2 = (u32) a.s2;
 110   r.s3 = (u32) a.s3;
 111   #endif
 112
 113   #if VECT_SIZE >= 8
 114   r.s4 = (u32) a.s4;
 115   r.s5 = (u32) a.s5;
 116   r.s6 = (u32) a.s6;
 117   r.s7 = (u32) a.s7;
 118   #endif
 119
 120   #if VECT_SIZE >= 16
 121   r.s8 = (u32) a.s8;
 122   r.s9 = (u32) a.s9;
 123   r.sa = (u32) a.sa;
 124   r.sb = (u32) a.sb;
 125   r.sc = (u32) a.sc;
 126   r.sd = (u32) a.sd;
 127   r.se = (u32) a.se;
 128   r.sf = (u32) a.sf;
 129   #endif
 130
 131   return r;
 132 }
 133
 134 inline u64x hl32_to_64 (const u32x a, const u32x b)
 135 {
 136   u64x r;
 137
 138   #if VECT_SIZE == 1
 139   r    = as_ulong ((uint2) (b,    a));
 140   #endif
 141
 142   #if VECT_SIZE >= 2
 143   r.s0 = as_ulong ((uint2) (b.s0, a.s0));
 144   r.s1 = as_ulong ((uint2) (b.s1, a.s1));
 145   #endif
 146
 147   #if VECT_SIZE >= 4
 148   r.s2 = as_ulong ((uint2) (b.s2, a.s2));
 149   r.s3 = as_ulong ((uint2) (b.s3, a.s3));
 150   #endif
 151
 152   #if VECT_SIZE >= 8
 153   r.s4 = as_ulong ((uint2) (b.s4, a.s4));
 154   r.s5 = as_ulong ((uint2) (b.s5, a.s5));
 155   r.s6 = as_ulong ((uint2) (b.s6, a.s6));
 156   r.s7 = as_ulong ((uint2) (b.s7, a.s7));
 157   #endif
 158
 159   #if VECT_SIZE >= 16
 160   r.s8 = as_ulong ((uint2) (b.s8, a.s8));
 161   r.s9 = as_ulong ((uint2) (b.s9, a.s9));
 162   r.sa = as_ulong ((uint2) (b.sa, a.sa));
 163   r.sb = as_ulong ((uint2) (b.sb, a.sb));
 164   r.sc = as_ulong ((uint2) (b.sc, a.sc));
 165   r.sd = as_ulong ((uint2) (b.sd, a.sd));
 166   r.se = as_ulong ((uint2) (b.se, a.se));
 167   r.sf = as_ulong ((uint2) (b.sf, a.sf));
 168   #endif
 169
 170   return r;
 171 }
 172
 173 #ifdef IS_AMD
 174 inline u32 swap32_S (const u32 v)
 175 {
 176   return (as_uint (as_uchar4 (v).s3210));
 177 }
 178
 179 inline u64 swap64_S (const u64 v)
 180 {
 181   return (as_ulong (as_uchar8 (v).s76543210));
 182 }
 183
 184 inline u32 rotr32_S (const u32 a, const u32 n)
 185 {
 186   return rotate (a, 32 - n);
 187 }
 188
 189 inline u32 rotl32_S (const u32 a, const u32 n)
 190 {
 191   return rotate (a, n);
 192 }
 193
 194 inline u64 rotr64_S (const u64 a, const u32 n)
 195 {
 196   const u32 a0 = h32_from_64_S (a);
 197   const u32 a1 = l32_from_64_S (a);
 198
 199   const u32 t0 = (n >= 32) ? amd_bitalign (a0, a1, n - 32) : amd_bitalign (a1, a0, n);
 200   const u32 t1 = (n >= 32) ? amd_bitalign (a1, a0, n - 32) : amd_bitalign (a0, a1, n);
 201
 202   const u64 r = hl32_to_64_S (t0, t1);
 203
 204   return r;
 205 }
 206
 207 inline u64 rotl64_S (const u64 a, const u32 n)
 208 {
 209   return rotr64_S (a, 64 - n);
 210 }
 211
 212 inline u32x swap32 (const u32x v)
 213 {
 214   return ((v >> 24) & 0x000000ff)
 215        | ((v >>  8) & 0x0000ff00)
 216        | ((v <<  8) & 0x00ff0000)
 217        | ((v << 24) & 0xff000000);
 218 }
 219
 220 inline u64x swap64 (const u64x v)
 221 {
 222   return ((v >> 56) & 0x00000000000000ff)
 223        | ((v >> 40) & 0x000000000000ff00)
 224        | ((v >> 24) & 0x0000000000ff0000)
 225        | ((v >>  8) & 0x00000000ff000000)
 226        | ((v <<  8) & 0x000000ff00000000)
 227        | ((v << 24) & 0x0000ff0000000000)
 228        | ((v << 40) & 0x00ff000000000000)
 229        | ((v << 56) & 0xff00000000000000);
 230 }
 231
 232 inline u32x rotr32 (const u32x a, const u32 n)
 233 {
 234   return rotate (a, 32 - n);
 235 }
 236
 237 inline u32x rotl32 (const u32x a, const u32 n)
 238 {
 239   return rotate (a, n);
 240 }
 241
 242 inline u64x rotr64 (const u64x a, const u32 n)
 243 {
 244   const u32x a0 = h32_from_64 (a);
 245   const u32x a1 = l32_from_64 (a);
 246
 247   const u32x t0 = (n >= 32) ? amd_bitalign (a0, a1, n - 32) : amd_bitalign (a1, a0, n);
 248   const u32x t1 = (n >= 32) ? amd_bitalign (a1, a0, n - 32) : amd_bitalign (a0, a1, n);
 249
 250   const u64x r = hl32_to_64 (t0, t1);
 251
 252   return r;
 253 }
 254
 255 inline u64x rotl64 (const u64x a, const u32 n)
 256 {
 257   return rotr64 (a, 64 - n);
 258 }
 259
 260 inline u32x __bfe (const u32x a, const u32x b, const u32x c)
 261 {
 262   return amd_bfe (a, b, c);
 263 }
 264
 265 inline u32 __bfe_S (const u32 a, const u32 b, const u32 c)
 266 {
 267   return amd_bfe (a, b, c);
 268 }
 269
 270 inline u32 amd_bytealign_S (const u32 a, const u32 b, const u32 c)
 271 {
 272   return amd_bytealign (a, b, c);
 273 }
 274 #endif
 275
 276 #ifdef IS_NV
 277 inline u32 swap32_S (const u32 v)
 278 {
 279   u32 r;
 280
 281   asm ("prmt.b32 %0, %1, 0, 0x0123;" : "=r"(r) : "r"(v));
 282
 283   return r;
 284 }
 285
 286 inline u64 swap64_S (const u64 v)
 287 {
 288   u32 il;
 289   u32 ir;
 290
 291   asm ("mov.b64 {%0, %1}, %2;" : "=r"(il), "=r"(ir) : "l"(v));
 292
 293   u32 tl;
 294   u32 tr;
 295
 296   asm ("prmt.b32 %0, %1, 0, 0x0123;" : "=r"(tl) : "r"(il));
 297   asm ("prmt.b32 %0, %1, 0, 0x0123;" : "=r"(tr) : "r"(ir));
 298
 299   u64 r;
 300
 301   asm ("mov.b64 %0, {%1, %2};" : "=l"(r) : "r"(tr), "r"(tl));
 302
 303   return r;
 304 }
 305
 306 inline u32 rotr32_S (const u32 a, const u32 n)
 307 {
 308   return rotate (a, 32 - n);
 309 }
 310
 311 inline u32 rotl32_S (const u32 a, const u32 n)
 312 {
 313   return rotate (a, n);
 314 }
 315
 316 inline u64 rotr64_S (const u64 a, const u32 n)
 317 {
 318   return rotate (a, (u64) 64 - n);
 319 }
 320
 321 inline u64 rotl64_S (const u64 a, const u32 n)
 322 {
 323   return rotr64_S (a, 64 - n);
 324 }
 325
 326 inline u32x swap32 (const u32x v)
 327 {
 328   return ((v >> 24) & 0x000000ff)
 329        | ((v >>  8) & 0x0000ff00)
 330        | ((v <<  8) & 0x00ff0000)
 331        | ((v << 24) & 0xff000000);
 332 }
 333
 334 inline u64x swap64 (const u64x v)
 335 {
 336   return ((v >> 56) & 0x00000000000000ff)
 337        | ((v >> 40) & 0x000000000000ff00)
 338        | ((v >> 24) & 0x0000000000ff0000)
 339        | ((v >>  8) & 0x00000000ff000000)
 340        | ((v <<  8) & 0x000000ff00000000)
 341        | ((v << 24) & 0x0000ff0000000000)
 342        | ((v << 40) & 0x00ff000000000000)
 343        | ((v << 56) & 0xff00000000000000);
 344 }
 345
 346 inline u32x rotr32 (const u32x a, const u32 n)
 347 {
 348   return rotate (a, 32 - n);
 349 }
 350
 351 inline u32x rotl32 (const u32x a, const u32 n)
 352 {
 353   return rotate (a, n);
 354 }
 355
 356 inline u64x rotr64 (const u64x a, const u32 n)
 357 {
 358   return rotate (a, (u64) 64 - n);
 359 }
 360
 361 inline u64x rotl64 (const u64x a, const u32 n)
 362 {
 363   return rotate (a, (u64) n);
 364 }
 365
 366 inline u32x __byte_perm (const u32x a, const u32x b, const u32x c)
 367 {
 368   u32x r;
 369
 370   #if VECT_SIZE == 1
 371   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r)    : "r"(a),    "r"(b),    "r"(c)   );
 372   #endif
 373
 374   #if VECT_SIZE >= 2
 375   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s0) : "r"(a.s0), "r"(b.s0), "r"(c.s0));
 376   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s1) : "r"(a.s1), "r"(b.s1), "r"(c.s1));
 377   #endif
 378
 379   #if VECT_SIZE >= 4
 380   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s2) : "r"(a.s2), "r"(b.s2), "r"(c.s2));
 381   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s3) : "r"(a.s3), "r"(b.s3), "r"(c.s3));
 382   #endif
 383
 384   #if VECT_SIZE >= 8
 385   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s4) : "r"(a.s4), "r"(b.s4), "r"(c.s4));
 386   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s5) : "r"(a.s5), "r"(b.s5), "r"(c.s5));
 387   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s6) : "r"(a.s6), "r"(b.s6), "r"(c.s6));
 388   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s7) : "r"(a.s7), "r"(b.s7), "r"(c.s7));
 389   #endif
 390
 391   #if VECT_SIZE >= 16
 392   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s8) : "r"(a.s8), "r"(b.s8), "r"(c.s8));
 393   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.s9) : "r"(a.s9), "r"(b.s9), "r"(c.s9));
 394   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.sa) : "r"(a.sa), "r"(b.sa), "r"(c.sa));
 395   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.sb) : "r"(a.sb), "r"(b.sb), "r"(c.sb));
 396   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.sc) : "r"(a.sc), "r"(b.sc), "r"(c.sc));
 397   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.sd) : "r"(a.sd), "r"(b.sd), "r"(c.sd));
 398   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.se) : "r"(a.se), "r"(b.se), "r"(c.se));
 399   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r.sf) : "r"(a.sf), "r"(b.sf), "r"(c.sf));
 400   #endif
 401
 402   return r;
 403 }
 404
 405 inline u32 __byte_perm_S (const u32 a, const u32 b, const u32 c)
 406 {
 407   u32 r;
 408
 409   asm ("prmt.b32 %0, %1, %2, %3;" : "=r"(r) : "r"(a), "r"(b), "r"(c));
 410
 411   return r;
 412 }
 413
 414 inline u32x __bfe (const u32x a, const u32x b, const u32x c)
 415 {
 416   u32x r;
 417
 418   #if VECT_SIZE == 1
 419   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r)    : "r"(a),    "r"(b),    "r"(c));
 420   #endif
 421
 422   #if VECT_SIZE >= 2
 423   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s0) : "r"(a.s0), "r"(b.s0), "r"(c.s0));
 424   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s1) : "r"(a.s1), "r"(b.s1), "r"(c.s1));
 425   #endif
 426
 427   #if VECT_SIZE >= 4
 428   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s2) : "r"(a.s2), "r"(b.s2), "r"(c.s2));
 429   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s3) : "r"(a.s3), "r"(b.s3), "r"(c.s3));
 430   #endif
 431
 432   #if VECT_SIZE >= 8
 433   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s4) : "r"(a.s4), "r"(b.s4), "r"(c.s4));
 434   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s5) : "r"(a.s5), "r"(b.s5), "r"(c.s5));
 435   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s6) : "r"(a.s6), "r"(b.s6), "r"(c.s6));
 436   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s7) : "r"(a.s7), "r"(b.s7), "r"(c.s7));
 437   #endif
 438
 439   #if VECT_SIZE >= 16
 440   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s8) : "r"(a.s8), "r"(b.s8), "r"(c.s8));
 441   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.s9) : "r"(a.s9), "r"(b.s9), "r"(c.s9));
 442   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.sa) : "r"(a.sa), "r"(b.sa), "r"(c.sa));
 443   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.sb) : "r"(a.sb), "r"(b.sb), "r"(c.sb));
 444   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.sc) : "r"(a.sc), "r"(b.sc), "r"(c.sc));
 445   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.sd) : "r"(a.sd), "r"(b.sd), "r"(c.sd));
 446   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.se) : "r"(a.se), "r"(b.se), "r"(c.se));
 447   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r.sf) : "r"(a.sf), "r"(b.sf), "r"(c.sf));
 448   #endif
 449
 450   return r;
 451 }
 452
 453 inline u32 __bfe_S (const u32 a, const u32 b, const u32 c)
 454 {
 455   u32 r;
 456
 457   asm ("bfe.u32 %0, %1, %2, %3;" : "=r"(r) : "r"(a), "r"(b), "r"(c));
 458
 459   return r;
 460 }
 461
 462 inline u32x amd_bytealign (const u32x a, const u32x b, const u32x c)
 463 {
 464   u32x r;
 465
 466   #if CUDA_ARCH >= 350
 467
 468   #if VECT_SIZE == 1
 469   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r)    : "r"(b),    "r"(a),    "r"((c & 3) * 8));
 470   #endif
 471
 472   #if VECT_SIZE >= 2
 473   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s0) : "r"(b.s0), "r"(a.s0), "r"((c.s0 & 3) * 8));
 474   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s1) : "r"(b.s1), "r"(a.s1), "r"((c.s1 & 3) * 8));
 475   #endif
 476
 477   #if VECT_SIZE >= 4
 478   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s2) : "r"(b.s2), "r"(a.s2), "r"((c.s2 & 3) * 8));
 479   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s3) : "r"(b.s3), "r"(a.s3), "r"((c.s3 & 3) * 8));
 480   #endif
 481
 482   #if VECT_SIZE >= 8
 483   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s4) : "r"(b.s4), "r"(a.s4), "r"((c.s4 & 3) * 8));
 484   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s5) : "r"(b.s5), "r"(a.s5), "r"((c.s5 & 3) * 8));
 485   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s6) : "r"(b.s6), "r"(a.s6), "r"((c.s6 & 3) * 8));
 486   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s7) : "r"(b.s7), "r"(a.s7), "r"((c.s7 & 3) * 8));
 487   #endif
 488
 489   #if VECT_SIZE >= 16
 490   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s8) : "r"(b.s8), "r"(a.s8), "r"((c.s8 & 3) * 8));
 491   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.s9) : "r"(b.s9), "r"(a.s9), "r"((c.s9 & 3) * 8));
 492   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.sa) : "r"(b.sa), "r"(a.sa), "r"((c.sa & 3) * 8));
 493   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.sb) : "r"(b.sb), "r"(a.sb), "r"((c.sb & 3) * 8));
 494   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.sc) : "r"(b.sc), "r"(a.sc), "r"((c.sc & 3) * 8));
 495   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.sd) : "r"(b.sd), "r"(a.sd), "r"((c.sd & 3) * 8));
 496   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.se) : "r"(b.se), "r"(a.se), "r"((c.se & 3) * 8));
 497   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r.sf) : "r"(b.sf), "r"(a.sf), "r"((c.sf & 3) * 8));
 498   #endif
 499
 500   #else
 501
 502   r = __byte_perm (b, a, ((u32x) (0x76543210) >> ((c & 3) * 4)) & 0xffff);
 503
 504   #endif
 505
 506   return r;
 507 }
 508
 509 inline u32 amd_bytealign_S (const u32 a, const u32 b, const u32 c)
 510 {
 511   u32 r;
 512
 513   #if CUDA_ARCH >= 350
 514
 515   asm ("shf.r.wrap.b32 %0, %1, %2, %3;" : "=r"(r) : "r"(b), "r"(a), "r"((c & 3) * 8));
 516
 517   #else
 518
 519   r = __byte_perm_S (b, a, (0x76543210 >> ((c & 3) * 4)) & 0xffff);
 520
 521   #endif
 522
 523   return r;
 524 }
 525 #endif
 526
 527 #ifdef IS_GENERIC
 528 inline u32 swap32_S (const u32 v)
 529 {
 530   return (as_uint (as_uchar4 (v).s3210));
 531 }
 532
 533 inline u64 swap64_S (const u64 v)
 534 {
 535   return (as_ulong (as_uchar8 (v).s76543210));
 536 }
 537
 538 inline u32 rotr32_S (const u32 a, const u32 n)
 539 {
 540   return rotate (a, 32 - n);
 541 }
 542
 543 inline u32 rotl32_S (const u32 a, const u32 n)
 544 {
 545   return rotate (a, n);
 546 }
 547
 548 inline u64 rotr64_S (const u64 a, const u32 n)
 549 {
 550   return rotate (a, (u64) 64 - n);
 551 }
 552
 553 inline u64 rotl64_S (const u64 a, const u32 n)
 554 {
 555   return rotate (a, (u64) n);
 556 }
 557
 558 inline u32x swap32 (const u32x v)
 559 {
 560   return ((v >> 24) & 0x000000ff)
 561        | ((v >>  8) & 0x0000ff00)
 562        | ((v <<  8) & 0x00ff0000)
 563        | ((v << 24) & 0xff000000);
 564 }
 565
 566 inline u64x swap64 (const u64x v)
 567 {
 568   return ((v >> 56) & 0x00000000000000ff)
 569        | ((v >> 40) & 0x000000000000ff00)
 570        | ((v >> 24) & 0x0000000000ff0000)
 571        | ((v >>  8) & 0x00000000ff000000)
 572        | ((v <<  8) & 0x000000ff00000000)
 573        | ((v << 24) & 0x0000ff0000000000)
 574        | ((v << 40) & 0x00ff000000000000)
 575        | ((v << 56) & 0xff00000000000000);
 576 }
 577
 578 inline u32x rotr32 (const u32x a, const u32 n)
 579 {
 580   return rotate (a, 32 - n);
 581 }
 582
 583 inline u32x rotl32 (const u32x a, const u32 n)
 584 {
 585   return rotate (a, n);
 586 }
 587
 588 inline u64x rotr64 (const u64x a, const u32 n)
 589 {
 590   return rotate (a, (u64) 64 - n);
 591 }
 592
 593 inline u64x rotl64 (const u64x a, const u32 n)
 594 {
 595   return rotate (a, (u64) n);
 596 }
 597
 598 inline u32x __bfe (const u32x a, const u32x b, const u32x c)
 599 {
 600   #define BIT(x)      ((u32x) (1u) << (x))
 601   #define BIT_MASK(x) (BIT (x) - 1)
 602   #define BFE(x,y,z)  (((x) >> (y)) & BIT_MASK (z))
 603
 604   return BFE (a, b, c);
 605
 606   #undef BIT
 607   #undef BIT_MASK
 608   #undef BFE
 609 }
 610
 611 inline u32 __bfe_S (const u32 a, const u32 b, const u32 c)
 612 {
 613   #define BIT(x)      (1u << (x))
 614   #define BIT_MASK(x) (BIT (x) - 1)
 615   #define BFE(x,y,z)  (((x) >> (y)) & BIT_MASK (z))
 616
 617   return BFE (a, b, c);
 618
 619   #undef BIT
 620   #undef BIT_MASK
 621   #undef BFE
 622 }
 623
 624 inline u32x amd_bytealign (const u32x a, const u32x b, const u32 c)
 625 {
 626   #if VECT_SIZE == 1
 627   const u64x tmp = ((((u64x) (a)) << 32) | ((u64x) (b))) >> ((c & 3) * 8);
 628
 629   return (u32x) (tmp);
 630   #endif
 631
 632   #if VECT_SIZE == 2
 633   const u64x tmp = ((((u64x) (a.s0, a.s1)) << 32) | ((u64x) (b.s0, b.s1))) >> ((c & 3) * 8);
 634
 635   return (u32x) (tmp.s0, tmp.s1);
 636   #endif
 637
 638   #if VECT_SIZE == 4
 639   const u64x tmp = ((((u64x) (a.s0, a.s1, a.s2, a.s3)) << 32) | ((u64x) (b.s0, b.s1, b.s2, b.s3))) >> ((c & 3) * 8);
 640
 641   return (u32x) (tmp.s0, tmp.s1, tmp.s2, tmp.s3);
 642   #endif
 643
 644   #if VECT_SIZE == 8
 645   const u64x tmp = ((((u64x) (a.s0, a.s1, a.s2, a.s3, a.s4, a.s5, a.s6, a.s7)) << 32) | ((u64x) (b.s0, b.s1, b.s2, b.s3, b.s4, b.s5, b.s6, b.s7))) >> ((c & 3) * 8);
 646
 647   return (u32x) (tmp.s0, tmp.s1, tmp.s2, tmp.s3, tmp.s4, tmp.s5, tmp.s6, tmp.s7);
 648   #endif
 649
 650   #if VECT_SIZE == 16
 651   const u64x tmp = ((((u64x) (a.s0, a.s1, a.s2, a.s3, a.s4, a.s5, a.s6, a.s7, a.s8, a.s9, a.sa, a.sb, a.sc, a.sd, a.se, a.sf)) << 32) | ((u64x) (b.s0, b.s1, b.s2, b.s3, b.s4, b.s5, b.s6, b.s7, b.s8, b.s9, b.sa, b.sb, b.sc, b.sd, b.se, b.sf))) >> ((c & 3) * 8);
 652
 653   return (u32x) (tmp.s0, tmp.s1, tmp.s2, tmp.s3, tmp.s4, tmp.s5, tmp.s6, tmp.s7, tmp.s8, tmp.s9, tmp.sa, tmp.sb, tmp.sc, tmp.sd, tmp.se, tmp.sf);
 654   #endif
 655 }
 656
 657 inline u32 amd_bytealign_S (const u32 a, const u32 b, const u32 c)
 658 {
 659   const u64 tmp = ((((u64) a) << 32) | ((u64) b)) >> ((c & 3) * 8);
 660
 661   return (u32) (tmp);
 662 }
 663
 664 #endif
 665
 666 typedef struct
 667 {
 668   u32 digest_buf[DGST_ELEM];
 669
 670 } digest_t;
 671
 672 typedef struct
 673 {
 674   u32 salt_buf[16];
 675   u32 salt_buf_pc[8];
 676
 677   u32 salt_len;
 678   u32 salt_iter;
 679   u32 salt_sign[2];
 680
 681   u32 keccak_mdlen;
 682   u32 truecrypt_mdlen;
 683
 684   u32 digests_cnt;
 685   u32 digests_done;
 686
 687   u32 digests_offset;
 688
 689   u32 scrypt_N;
 690   u32 scrypt_r;
 691   u32 scrypt_p;
 692   u32 scrypt_tmto;
 693   u32 scrypt_phy;
 694
 695 } salt_t;
 696
 697 typedef struct
 698 {
 699   int V;
 700   int R;
 701   int P;
 702
 703   int enc_md;
 704
 705   u32 id_buf[8];
 706   u32 u_buf[32];
 707   u32 o_buf[32];
 708
 709   int id_len;
 710   int o_len;
 711   int u_len;
 712
 713   u32 rc4key[2];
 714   u32 rc4data[2];
 715
 716 } pdf_t;
 717
 718 typedef struct
 719 {
 720   u32 pke[25];
 721   u32 eapol[64];
 722   int eapol_size;
 723   int keyver;
 724   u8  orig_mac1[6];
 725   u8  orig_mac2[6];
 726   u8  orig_nonce1[32];
 727   u8  orig_nonce2[32];
 728
 729 } wpa_t;
 730
 731 typedef struct
 732 {
 733   u32 cry_master_buf[64];
 734   u32 ckey_buf[64];
 735   u32 public_key_buf[64];
 736
 737   u32 cry_master_len;
 738   u32 ckey_len;
 739   u32 public_key_len;
 740
 741 } bitcoin_wallet_t;
 742
 743 typedef struct
 744 {
 745   u32 salt_buf[30];
 746   u32 salt_len;
 747
 748   u32 esalt_buf[38];
 749   u32 esalt_len;
 750
 751 } sip_t;
 752
 753 typedef struct
 754 {
 755   u32 data[384];
 756
 757 } androidfde_t;
 758
 759 typedef struct
 760 {
 761   u32 nr_buf[16];
 762   u32 nr_len;
 763
 764   u32 msg_buf[128];
 765   u32 msg_len;
 766
 767 } ikepsk_t;
 768
 769 typedef struct
 770 {
 771   u32 user_len;
 772   u32 domain_len;
 773   u32 srvchall_len;
 774   u32 clichall_len;
 775
 776   u32 userdomain_buf[64];
 777   u32 chall_buf[256];
 778
 779 } netntlm_t;
 780
 781 typedef struct
 782 {
 783   u32 user[16];
 784   u32 realm[16];
 785   u32 salt[32];
 786   u32 timestamp[16];
 787   u32 checksum[4];
 788
 789 } krb5pa_t;
 790
 791 typedef struct
 792 {
 793   u32 account_info[512];
 794   u32 checksum[4];
 795   u32 edata2[2560];
 796   u32 edata2_len;
 797
 798 } krb5tgs_t;
 799
 800 typedef struct
 801 {
 802   u32 salt_buf[16];
 803   u32 data_buf[112];
 804   u32 keyfile_buf[16];
 805   u32 signature;
 806
 807 } tc_t;
 808
 809 typedef struct
 810 {
 811   u32 salt_buf[16];
 812
 813 } pbkdf2_md5_t;
 814
 815 typedef struct
 816 {
 817   u32 salt_buf[16];
 818
 819 } pbkdf2_sha1_t;
 820
 821 typedef struct
 822 {
 823   u32 salt_buf[16];
 824
 825 } pbkdf2_sha256_t;
 826
 827 typedef struct
 828 {
 829   u32 salt_buf[32];
 830
 831 } pbkdf2_sha512_t;
 832
 833 typedef struct
 834 {
 835   u32 salt_buf[128];
 836   u32 salt_len;
 837
 838 } rakp_t;
 839
 840 typedef struct
 841 {
 842   u32 data_len;
 843   u32 data_buf[512];
 844
 845 } cloudkey_t;
 846
 847 typedef struct
 848 {
 849   u32 encryptedVerifier[4];
 850   u32 encryptedVerifierHash[5];
 851
 852   u32 keySize;
 853
 854 } office2007_t;
 855
 856 typedef struct
 857 {
 858   u32 encryptedVerifier[4];
 859   u32 encryptedVerifierHash[8];
 860
 861 } office2010_t;
 862
 863 typedef struct
 864 {
 865   u32 encryptedVerifier[4];
 866   u32 encryptedVerifierHash[8];
 867
 868 } office2013_t;
 869
 870 typedef struct
 871 {
 872   u32 version;
 873   u32 encryptedVerifier[4];
 874   u32 encryptedVerifierHash[4];
 875   u32 rc4key[2];
 876
 877 } oldoffice01_t;
 878
 879 typedef struct
 880 {
 881   u32 version;
 882   u32 encryptedVerifier[4];
 883   u32 encryptedVerifierHash[5];
 884   u32 rc4key[2];
 885
 886 } oldoffice34_t;
 887
 888 typedef struct
 889 {
 890   u32 salt_buf[128];
 891   u32 salt_len;
 892
 893   u32 pc_digest[5];
 894   u32 pc_offset;
 895
 896 } pstoken_t;
 897
 898 typedef struct
 899 {
 900   u32 type;
 901   u32 mode;
 902   u32 magic;
 903   u32 salt_len;
 904   u32 salt_buf[4];
 905   u32 verify_bytes;
 906   u32 compress_length;
 907   u32 data_len;
 908   u32 data_buf[2048];
 909   u32 auth_len;
 910   u32 auth_buf[4];
 911
 912 } zip2_t;
 913
 914 typedef struct
 915 {
 916   u32 salt_buf[32];
 917
 918 } win8phone_t;
 919
 920 typedef struct
 921 {
 922   u32 version;
 923   u32 algorithm;
 924
 925   /* key-file handling */
 926   u32 keyfile_len;
 927   u32 keyfile[8];
 928
 929   u32 final_random_seed[8];
 930   u32 transf_random_seed[8];
 931   u32 enc_iv[4];
 932   u32 contents_hash[8];
 933
 934   /* specific to version 1 */
 935   u32 contents_len;
 936   u32 contents[75000];
 937
 938   /* specific to version 2 */
 939   u32 expected_bytes[8];
 940
 941 } keepass_t;
 942
 943 typedef struct
 944 {
 945   u32 digest[4];
 946   u32 out[4];
 947
 948 } pdf14_tmp_t;
 949
 950 typedef struct
 951 {
 952   union
 953   {
 954     u32 dgst32[16];
 955     u64 dgst64[8];
 956   };
 957
 958   u32 dgst_len;
 959   u32 W_len;
 960
 961 } pdf17l8_tmp_t;
 962
 963 typedef struct
 964 {
 965   u32 digest_buf[4];
 966
 967 } phpass_tmp_t;
 968
 969 typedef struct
 970 {
 971   u32 digest_buf[4];
 972
 973 } md5crypt_tmp_t;
 974
 975 typedef struct
 976 {
 977   u32 alt_result[8];
 978
 979   u32 p_bytes[4];
 980   u32 s_bytes[4];
 981
 982 } sha256crypt_tmp_t;
 983
 984 typedef struct
 985 {
 986   u64 l_alt_result[8];
 987
 988   u64 l_p_bytes[2];
 989   u64 l_s_bytes[2];
 990
 991 } sha512crypt_tmp_t;
 992
 993 typedef struct
 994 {
 995   u32 ipad[5];
 996   u32 opad[5];
 997
 998   u32 dgst[10];
 999   u32 out[10];
1000
1001 } wpa_tmp_t;
1002
1003 typedef struct
1004 {
1005   u64 dgst[8];
1006
1007 } bitcoin_wallet_tmp_t;
1008
1009 typedef struct
1010 {
1011   u32 ipad[5];
1012   u32 opad[5];
1013
1014   u32 dgst[5];
1015   u32 out[4];
1016
1017 } dcc2_tmp_t;
1018
1019 typedef struct
1020 {
1021   u32 E[18];
1022
1023   u32 P[18];
1024
1025   u32 S0[256];
1026   u32 S1[256];
1027   u32 S2[256];
1028   u32 S3[256];
1029
1030 } bcrypt_tmp_t;
1031
1032 typedef struct
1033 {
1034   u32 digest[2];
1035
1036   u32 P[18];
1037
1038   u32 S0[256];
1039   u32 S1[256];
1040   u32 S2[256];
1041   u32 S3[256];
1042
1043 } pwsafe2_tmp_t;
1044
1045 typedef struct
1046 {
1047   u32 digest_buf[8];
1048
1049 } pwsafe3_tmp_t;
1050
1051 typedef struct
1052 {
1053   u32 digest_buf[5];
1054
1055 } androidpin_tmp_t;
1056
1057 typedef struct
1058 {
1059   u32 ipad[5];
1060   u32 opad[5];
1061
1062   u32 dgst[10];
1063   u32 out[10];
1064
1065 } androidfde_tmp_t;
1066
1067 typedef struct
1068 {
1069   u32 ipad[16];
1070   u32 opad[16];
1071
1072   u32 dgst[64];
1073   u32 out[64];
1074
1075 } tc_tmp_t;
1076
1077 typedef struct
1078 {
1079   u64 ipad[8];
1080   u64 opad[8];
1081
1082   u64 dgst[32];
1083   u64 out[32];
1084
1085 } tc64_tmp_t;
1086
1087 typedef struct
1088 {
1089   u32 ipad[4];
1090   u32 opad[4];
1091
1092   u32 dgst[32];
1093   u32 out[32];
1094
1095 } pbkdf2_md5_tmp_t;
1096
1097 typedef struct
1098 {
1099   u32 ipad[5];
1100   u32 opad[5];
1101
1102   u32 dgst[32];
1103   u32 out[32];
1104
1105 } pbkdf2_sha1_tmp_t;
1106
1107 typedef struct
1108 {
1109   u32 ipad[8];
1110   u32 opad[8];
1111
1112   u32 dgst[32];
1113   u32 out[32];
1114
1115 } pbkdf2_sha256_tmp_t;
1116
1117 typedef struct
1118 {
1119   u64 ipad[8];
1120   u64 opad[8];
1121
1122   u64 dgst[16];
1123   u64 out[16];
1124
1125 } pbkdf2_sha512_tmp_t;
1126
1127 typedef struct
1128 {
1129   u64 out[8];
1130
1131 } ecryptfs_tmp_t;
1132
1133 typedef struct
1134 {
1135   u64 ipad[8];
1136   u64 opad[8];
1137
1138   u64 dgst[16];
1139   u64 out[16];
1140
1141 } oraclet_tmp_t;
1142
1143 typedef struct
1144 {
1145   u32 ipad[5];
1146   u32 opad[5];
1147
1148   u32 dgst[5];
1149   u32 out[5];
1150
1151 } agilekey_tmp_t;
1152
1153 typedef struct
1154 {
1155   u32 ipad[5];
1156   u32 opad[5];
1157
1158   u32 dgst1[5];
1159   u32 out1[5];
1160
1161   u32 dgst2[5];
1162   u32 out2[5];
1163
1164 } mywallet_tmp_t;
1165
1166 typedef struct
1167 {
1168   u32 ipad[5];
1169   u32 opad[5];
1170
1171   u32 dgst[5];
1172   u32 out[5];
1173
1174 } sha1aix_tmp_t;
1175
1176 typedef struct
1177 {
1178   u32 ipad[8];
1179   u32 opad[8];
1180
1181   u32 dgst[8];
1182   u32 out[8];
1183
1184 } sha256aix_tmp_t;
1185
1186 typedef struct
1187 {
1188   u64 ipad[8];
1189   u64 opad[8];
1190
1191   u64 dgst[8];
1192   u64 out[8];
1193
1194 } sha512aix_tmp_t;
1195
1196 typedef struct
1197 {
1198   u32 ipad[8];
1199   u32 opad[8];
1200
1201   u32 dgst[8];
1202   u32 out[8];
1203
1204 } lastpass_tmp_t;
1205
1206 typedef struct
1207 {
1208   u64 digest_buf[8];
1209
1210 } drupal7_tmp_t;
1211
1212 typedef struct
1213 {
1214   u32 ipad[5];
1215   u32 opad[5];
1216
1217   u32 dgst[5];
1218   u32 out[5];
1219
1220 } lotus8_tmp_t;
1221
1222 typedef struct
1223 {
1224   u32 out[5];
1225
1226 } office2007_tmp_t;
1227
1228 typedef struct
1229 {
1230   u32 out[5];
1231
1232 } office2010_tmp_t;
1233
1234 typedef struct
1235 {
1236   u64 out[8];
1237
1238 } office2013_tmp_t;
1239
1240 typedef struct
1241 {
1242   u32 digest_buf[5];
1243
1244 } saph_sha1_tmp_t;
1245
1246 typedef struct
1247 {
1248   u32 block[16];
1249
1250   u32 dgst[8];
1251
1252   u32 block_len;
1253   u32 final_len;
1254
1255 } seven_zip_tmp_t;
1256
1257 typedef struct
1258 {
1259   u32 KEK[4];
1260   u32 lsb[4];
1261   u32 cipher[4];
1262
1263 } axcrypt_tmp_t;
1264
1265 typedef struct
1266 {
1267   u32 tmp_digest[8];
1268
1269 } keepass_tmp_t;
1270
1271 typedef struct
1272 {
1273   u32 Kc[16];
1274   u32 Kd[16];
1275
1276   u32 iv[2];
1277
1278 } bsdicrypt_tmp_t;
1279
1280 typedef struct
1281 {
1282   u32 dgst[17][5];
1283
1284 } rar3_tmp_t;
1285
1286 typedef struct
1287 {
1288   u32 user[16];
1289
1290 } cram_md5_t;
1291
1292 typedef struct
1293 {
1294   u32 iv_buf[4];
1295   u32 iv_len;
1296
1297   u32 salt_buf[4];
1298   u32 salt_len;
1299
1300   u32 crc;
1301
1302   u32 data_buf[96];
1303   u32 data_len;
1304
1305   u32 unpack_size;
1306
1307 } seven_zip_t;
1308
1309 typedef struct
1310 {
1311   u32 key;
1312   u64 val;
1313
1314 } hcstat_table_t;
1315
1316 typedef struct
1317 {
1318   u32 cs_buf[0x100];
1319   u32 cs_len;
1320
1321 } cs_t;
1322
1323 typedef struct
1324 {
1325   u32 cmds[0x100];
1326
1327 } kernel_rule_t;
1328
1329 typedef struct
1330 {
1331   u32 salt_pos;
1332   u32 digest_pos;
1333   u32 hash_pos;
1334   u32 gidvid;
1335   u32 il_pos;
1336
1337 } plain_t;
1338
1339 typedef struct
1340 {
1341   u32 i[16];
1342
1343   u32 pw_len;
1344
1345   u32 alignment_placeholder_1;
1346   u32 alignment_placeholder_2;
1347   u32 alignment_placeholder_3;
1348
1349 } pw_t;
1350
1351 typedef struct
1352 {
1353   u32 i;
1354
1355 } bf_t;
1356
1357 typedef struct
1358 {
1359   u32 i[8];
1360
1361   u32 pw_len;
1362
1363 } comb_t;
1364
1365 typedef struct
1366 {
1367   u32 b[32];
1368
1369 } bs_word_t;
1370
1371 typedef struct
1372 {
1373   uint4 P[64];
1374
1375 } scrypt_tmp_t;