1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
|
#include "randommod.h"
#include "byte.h"
#include "dns.h"
static void swap(unsigned char *x, long long len, unsigned char *y) {
unsigned char t[33];
byte_copy(t, len, x);
byte_copy(x, len, y);
byte_copy(y, len, t);
}
void dns_sortip4(unsigned char *s, long long n) {
long long i;
if (n < 0) return;
n >>= 2;
while (n > 1) {
i = randommod(n);
--n;
swap(s + (i << 2), 4, s + (n << 2));
}
}
void dns_sortip(unsigned char *s, long long nn) {
long long i;
long long n = nn;
if (nn < 0) return;
n >>= 4;
while (n > 1) {
i = randommod(n);
--n;
swap(s + (i << 4), 16, s + (n << 4));
}
for (i = 0; i + 16 <= nn; i += 16) {
if (!byte_isequal(s + i, 12, "\0\0\0\0\0\0\0\0\0\0\377\377")) {
swap(s + i, 16, s);
break;
}
}
}
void dns_sortipkey(unsigned char *s, unsigned char *t, long long nn) {
long long i, j, k;
long long n;
unsigned char *key;
nn >>=4;
n = nn;
while (n > 1) {
i = randommod(n);
--n;
swap(s + 16 * i, 16, s + 16 * n);
swap(t + 33 * i, 33, t + 33 * n);
}
n = nn;
j = 0;
k = 0;
for (i = k; i < n; ++i) {
key = t + 33 * i;
if (key[0] == 1) {
swap(s + 16 * i, 16, s + 16 * j);
swap(t + 33 * i, 33, t + 33 * j);
++j;
}
}
for (i = k; i < j; ++i) {
key = t + 33 * i;
if (key[0] != 1) continue;
if (!byte_isequal(s + 16 * i, 12, "\0\0\0\0\0\0\0\0\0\0\377\377")) {
swap(s + 16 * i, 16, s + 16 * k);
swap(t + 33 * i, 33, t + 33 * k);
break;
}
}
k = j;
for (i = k; i < n; ++i) {
key = t + 33 * i;
if (key[0] == 2) {
swap(s + 16 * i, 16, s + 16 * j);
swap(t + 33 * i, 33, t + 33 * j);
++j;
}
}
for (i = k; i < j; ++i) {
key = t + 33 * i;
if (key[0] != 2) continue;
if (!byte_isequal(s + 16 * i, 12, "\0\0\0\0\0\0\0\0\0\0\377\377")) {
swap(s + 16 * i, 16, s + 16 * k);
swap(t + 33 * i, 33, t + 33 * k);
break;
}
}
k = j;
for (i = k; i < n; ++i) {
key = t + 33 * i;
if (!byte_isequal(s + 16 * i, 16, "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0")) {
swap(s + 16 * i, 16, s + 16 * j);
swap(t + 33 * i, 33, t + 33 * j);
++j;
}
}
for (i = k; i < j; ++i) {
key = t + 33 * i;
if (key[0] != 0) continue;
if (!byte_isequal(s + 16 * i, 12, "\0\0\0\0\0\0\0\0\0\0\377\377")) {
swap(s + 16 * i, 16, s + 16 * k);
swap(t + 33 * i, 33, t + 33 * k);
break;
}
}
}
|