File: dns_sortip.c

package info (click to toggle)
dq 20230101-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,020 kB
  • sloc: ansic: 8,269; makefile: 363; sh: 176; python: 82
file content (125 lines) | stat: -rw-r--r-- 2,817 bytes parent folder | download | duplicates (4)
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;
        }
    }
}