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
 693 } salt_t;
 694
 695 typedef struct
 696 {
 697   int V;
 698   int R;
 699   int P;
 700
 701   int enc_md;
 702
 703   u32 id_buf[8];
 704   u32 u_buf[32];
 705   u32 o_buf[32];
 706
 707   int id_len;
 708   int o_len;
 709   int u_len;
 710
 711   u32 rc4key[2];
 712   u32 rc4data[2];
 713
 714 } pdf_t;
 715
 716 typedef struct
 717 {
 718   u32 pke[25];
 719   u32 eapol[64];
 720   int eapol_size;
 721   int keyver;
 722   u8  orig_mac1[6];
 723   u8  orig_mac2[6];
 724   u8  orig_nonce1[32];
 725   u8  orig_nonce2[32];
 726
 727 } wpa_t;
 728
 729 typedef struct
 730 {
 731   u32 cry_master_buf[64];
 732   u32 ckey_buf[64];
 733   u32 public_key_buf[64];
 734
 735   u32 cry_master_len;
 736   u32 ckey_len;
 737   u32 public_key_len;
 738
 739 } bitcoin_wallet_t;
 740
 741 typedef struct
 742 {
 743   u32 salt_buf[30];
 744   u32 salt_len;
 745
 746   u32 esalt_buf[38];
 747   u32 esalt_len;
 748
 749 } sip_t;
 750
 751 typedef struct
 752 {
 753   u32 data[384];
 754
 755 } androidfde_t;
 756
 757 typedef struct
 758 {
 759   u32 nr_buf[16];
 760   u32 nr_len;
 761
 762   u32 msg_buf[128];
 763   u32 msg_len;
 764
 765 } ikepsk_t;
 766
 767 typedef struct
 768 {
 769   u32 user_len;
 770   u32 domain_len;
 771   u32 srvchall_len;
 772   u32 clichall_len;
 773
 774   u32 userdomain_buf[64];
 775   u32 chall_buf[256];
 776
 777 } netntlm_t;
 778
 779 typedef struct
 780 {
 781   u32 user[16];
 782   u32 realm[16];
 783   u32 salt[32];
 784   u32 timestamp[16];
 785   u32 checksum[4];
 786
 787 } krb5pa_t;
 788
 789 typedef struct
 790 {
 791   u32 account_info[512];
 792   u32 checksum[4];
 793   u32 edata2[2560];
 794   u32 edata2_len;
 795
 796 } krb5tgs_t;
 797
 798 typedef struct
 799 {
 800   u32 salt_buf[16];
 801   u32 data_buf[112];
 802   u32 keyfile_buf[16];
 803   u32 signature;
 804
 805 } tc_t;
 806
 807 typedef struct
 808 {
 809   u32 salt_buf[16];
 810
 811 } pbkdf2_md5_t;
 812
 813 typedef struct
 814 {
 815   u32 salt_buf[16];
 816
 817 } pbkdf2_sha1_t;
 818
 819 typedef struct
 820 {
 821   u32 salt_buf[16];
 822
 823 } pbkdf2_sha256_t;
 824
 825 typedef struct
 826 {
 827   u32 salt_buf[32];
 828
 829 } pbkdf2_sha512_t;
 830
 831 typedef struct
 832 {
 833   u32 salt_buf[128];
 834   u32 salt_len;
 835
 836 } rakp_t;
 837
 838 typedef struct
 839 {
 840   u32 data_len;
 841   u32 data_buf[512];
 842
 843 } cloudkey_t;
 844
 845 typedef struct
 846 {
 847   u32 encryptedVerifier[4];
 848   u32 encryptedVerifierHash[5];
 849
 850   u32 keySize;
 851
 852 } office2007_t;
 853
 854 typedef struct
 855 {
 856   u32 encryptedVerifier[4];
 857   u32 encryptedVerifierHash[8];
 858
 859 } office2010_t;
 860
 861 typedef struct
 862 {
 863   u32 encryptedVerifier[4];
 864   u32 encryptedVerifierHash[8];
 865
 866 } office2013_t;
 867
 868 typedef struct
 869 {
 870   u32 version;
 871   u32 encryptedVerifier[4];
 872   u32 encryptedVerifierHash[4];
 873   u32 rc4key[2];
 874
 875 } oldoffice01_t;
 876
 877 typedef struct
 878 {
 879   u32 version;
 880   u32 encryptedVerifier[4];
 881   u32 encryptedVerifierHash[5];
 882   u32 rc4key[2];
 883
 884 } oldoffice34_t;
 885
 886 typedef struct
 887 {
 888   u32 salt_buf[128];
 889   u32 salt_len;
 890
 891   u32 pc_digest[5];
 892   u32 pc_offset;
 893
 894 } pstoken_t;
 895
 896 typedef struct
 897 {
 898   u32 type;
 899   u32 mode;
 900   u32 magic;
 901   u32 salt_len;
 902   u32 salt_buf[4];
 903   u32 verify_bytes;
 904   u32 compress_length;
 905   u32 data_len;
 906   u32 data_buf[2048];
 907   u32 auth_len;
 908   u32 auth_buf[4];
 909
 910 } zip2_t;
 911
 912 typedef struct
 913 {
 914   u32 salt_buf[32];
 915
 916 } win8phone_t;
 917
 918 typedef struct
 919 {
 920   u32 version;
 921   u32 algorithm;
 922
 923   /* key-file handling */
 924   u32 keyfile_len;
 925   u32 keyfile[8];
 926
 927   u32 final_random_seed[8];
 928   u32 transf_random_seed[8];
 929   u32 enc_iv[4];
 930   u32 contents_hash[8];
 931
 932   /* specific to version 1 */
 933   u32 contents_len;
 934   u32 contents[75000];
 935
 936   /* specific to version 2 */
 937   u32 expected_bytes[8];
 938
 939 } keepass_t;
 940
 941 typedef struct
 942 {
 943   u32 digest[4];
 944   u32 out[4];
 945
 946 } pdf14_tmp_t;
 947
 948 typedef struct
 949 {
 950   union
 951   {
 952     u32 dgst32[16];
 953     u64 dgst64[8];
 954   };
 955
 956   u32 dgst_len;
 957   u32 W_len;
 958
 959 } pdf17l8_tmp_t;
 960
 961 typedef struct
 962 {
 963   u32 digest_buf[4];
 964
 965 } phpass_tmp_t;
 966
 967 typedef struct
 968 {
 969   u32 digest_buf[4];
 970
 971 } md5crypt_tmp_t;
 972
 973 typedef struct
 974 {
 975   u32 alt_result[8];
 976
 977   u32 p_bytes[4];
 978   u32 s_bytes[4];
 979
 980 } sha256crypt_tmp_t;
 981
 982 typedef struct
 983 {
 984   u64 l_alt_result[8];
 985
 986   u64 l_p_bytes[2];
 987   u64 l_s_bytes[2];
 988
 989 } sha512crypt_tmp_t;
 990
 991 typedef struct
 992 {
 993   u32 ipad[5];
 994   u32 opad[5];
 995
 996   u32 dgst[10];
 997   u32 out[10];
 998
 999 } wpa_tmp_t;
1000
1001 typedef struct
1002 {
1003   u64 dgst[8];
1004
1005 } bitcoin_wallet_tmp_t;
1006
1007 typedef struct
1008 {
1009   u32 ipad[5];
1010   u32 opad[5];
1011
1012   u32 dgst[5];
1013   u32 out[4];
1014
1015 } dcc2_tmp_t;
1016
1017 typedef struct
1018 {
1019   u32 E[18];
1020
1021   u32 P[18];
1022
1023   u32 S0[256];
1024   u32 S1[256];
1025   u32 S2[256];
1026   u32 S3[256];
1027
1028 } bcrypt_tmp_t;
1029
1030 typedef struct
1031 {
1032   u32 digest[2];
1033
1034   u32 P[18];
1035
1036   u32 S0[256];
1037   u32 S1[256];
1038   u32 S2[256];
1039   u32 S3[256];
1040
1041 } pwsafe2_tmp_t;
1042
1043 typedef struct
1044 {
1045   u32 digest_buf[8];
1046
1047 } pwsafe3_tmp_t;
1048
1049 typedef struct
1050 {
1051   u32 digest_buf[5];
1052
1053 } androidpin_tmp_t;
1054
1055 typedef struct
1056 {
1057   u32 ipad[5];
1058   u32 opad[5];
1059
1060   u32 dgst[10];
1061   u32 out[10];
1062
1063 } androidfde_tmp_t;
1064
1065 typedef struct
1066 {
1067   u32 ipad[16];
1068   u32 opad[16];
1069
1070   u32 dgst[64];
1071   u32 out[64];
1072
1073 } tc_tmp_t;
1074
1075 typedef struct
1076 {
1077   u64 ipad[8];
1078   u64 opad[8];
1079
1080   u64 dgst[32];
1081   u64 out[32];
1082
1083 } tc64_tmp_t;
1084
1085 typedef struct
1086 {
1087   u32 ipad[4];
1088   u32 opad[4];
1089
1090   u32 dgst[32];
1091   u32 out[32];
1092
1093 } pbkdf2_md5_tmp_t;
1094
1095 typedef struct
1096 {
1097   u32 ipad[5];
1098   u32 opad[5];
1099
1100   u32 dgst[32];
1101   u32 out[32];
1102
1103 } pbkdf2_sha1_tmp_t;
1104
1105 typedef struct
1106 {
1107   u32 ipad[8];
1108   u32 opad[8];
1109
1110   u32 dgst[32];
1111   u32 out[32];
1112
1113 } pbkdf2_sha256_tmp_t;
1114
1115 typedef struct
1116 {
1117   u64 ipad[8];
1118   u64 opad[8];
1119
1120   u64 dgst[16];
1121   u64 out[16];
1122
1123 } pbkdf2_sha512_tmp_t;
1124
1125 typedef struct
1126 {
1127   u64 out[8];
1128
1129 } ecryptfs_tmp_t;
1130
1131 typedef struct
1132 {
1133   u64 ipad[8];
1134   u64 opad[8];
1135
1136   u64 dgst[16];
1137   u64 out[16];
1138
1139 } oraclet_tmp_t;
1140
1141 typedef struct
1142 {
1143   u32 ipad[5];
1144   u32 opad[5];
1145
1146   u32 dgst[5];
1147   u32 out[5];
1148
1149 } agilekey_tmp_t;
1150
1151 typedef struct
1152 {
1153   u32 ipad[5];
1154   u32 opad[5];
1155
1156   u32 dgst1[5];
1157   u32 out1[5];
1158
1159   u32 dgst2[5];
1160   u32 out2[5];
1161
1162 } mywallet_tmp_t;
1163
1164 typedef struct
1165 {
1166   u32 ipad[5];
1167   u32 opad[5];
1168
1169   u32 dgst[5];
1170   u32 out[5];
1171
1172 } sha1aix_tmp_t;
1173
1174 typedef struct
1175 {
1176   u32 ipad[8];
1177   u32 opad[8];
1178
1179   u32 dgst[8];
1180   u32 out[8];
1181
1182 } sha256aix_tmp_t;
1183
1184 typedef struct
1185 {
1186   u64 ipad[8];
1187   u64 opad[8];
1188
1189   u64 dgst[8];
1190   u64 out[8];
1191
1192 } sha512aix_tmp_t;
1193
1194 typedef struct
1195 {
1196   u32 ipad[8];
1197   u32 opad[8];
1198
1199   u32 dgst[8];
1200   u32 out[8];
1201
1202 } lastpass_tmp_t;
1203
1204 typedef struct
1205 {
1206   u64 digest_buf[8];
1207
1208 } drupal7_tmp_t;
1209
1210 typedef struct
1211 {
1212   u32 ipad[5];
1213   u32 opad[5];
1214
1215   u32 dgst[5];
1216   u32 out[5];
1217
1218 } lotus8_tmp_t;
1219
1220 typedef struct
1221 {
1222   u32 out[5];
1223
1224 } office2007_tmp_t;
1225
1226 typedef struct
1227 {
1228   u32 out[5];
1229
1230 } office2010_tmp_t;
1231
1232 typedef struct
1233 {
1234   u64 out[8];
1235
1236 } office2013_tmp_t;
1237
1238 typedef struct
1239 {
1240   u32 digest_buf[5];
1241
1242 } saph_sha1_tmp_t;
1243
1244 typedef struct
1245 {
1246   u32 block[16];
1247
1248   u32 dgst[8];
1249
1250   u32 block_len;
1251   u32 final_len;
1252
1253 } seven_zip_tmp_t;
1254
1255 typedef struct
1256 {
1257   u32 KEK[4];
1258   u32 lsb[4];
1259   u32 cipher[4];
1260
1261 } axcrypt_tmp_t;
1262
1263 typedef struct
1264 {
1265   u32 tmp_digest[8];
1266
1267 } keepass_tmp_t;
1268
1269 typedef struct
1270 {
1271   u32 Kc[16];
1272   u32 Kd[16];
1273
1274   u32 iv[2];
1275
1276 } bsdicrypt_tmp_t;
1277
1278 typedef struct
1279 {
1280   u32 dgst[17][5];
1281
1282 } rar3_tmp_t;
1283
1284 typedef struct
1285 {
1286   u32 user[16];
1287
1288 } cram_md5_t;
1289
1290 typedef struct
1291 {
1292   u32 iv_buf[4];
1293   u32 iv_len;
1294
1295   u32 salt_buf[4];
1296   u32 salt_len;
1297
1298   u32 crc;
1299
1300   u32 data_buf[96];
1301   u32 data_len;
1302
1303   u32 unpack_size;
1304
1305 } seven_zip_t;
1306
1307 typedef struct
1308 {
1309   u32 key;
1310   u64 val;
1311
1312 } hcstat_table_t;
1313
1314 typedef struct
1315 {
1316   u32 cs_buf[0x100];
1317   u32 cs_len;
1318
1319 } cs_t;
1320
1321 typedef struct
1322 {
1323   u32 cmds[0x100];
1324
1325 } kernel_rule_t;
1326
1327 typedef struct
1328 {
1329   u32 salt_pos;
1330   u32 digest_pos;
1331   u32 hash_pos;
1332   u32 gidvid;
1333   u32 il_pos;
1334
1335 } plain_t;
1336
1337 typedef struct
1338 {
1339   u32 i[16];
1340
1341   u32 pw_len;
1342
1343   u32 alignment_placeholder_1;
1344   u32 alignment_placeholder_2;
1345   u32 alignment_placeholder_3;
1346
1347 } pw_t;
1348
1349 typedef struct
1350 {
1351   u32 i;
1352
1353 } bf_t;
1354
1355 typedef struct
1356 {
1357   u32 i[8];
1358
1359   u32 pw_len;
1360
1361 } comb_t;
1362
1363 typedef struct
1364 {
1365   u32 b[32];
1366
1367 } bs_word_t;
1368
1369 typedef struct
1370 {
1371   #ifndef SCRYPT_TMP_ELEM
1372   #define SCRYPT_TMP_ELEM 1
1373   #endif
1374
1375   uint4 P[SCRYPT_TMP_ELEM];
1376
1377 } scrypt_tmp_t;