tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Optimize GCD in pq_factorize.

Browse Source
This commit is contained in:
levlam 2022-03-21 19:38:40 +03:00
parent cb155ebf1a
commit 93a2b9895d
1 changed files with 17 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
tdutils/td/utils/crypto.cpp
View File

@ -55,42 +55,39 @@
namespace td {
static uint64 gcd(uint64 a, uint64 b) {
static uint64 pq_gcd(uint64 a, uint64 b) {
if (a == 0) {
return b;
}
if (b == 0) {
return a;
}
int shift = 0;
while ((a & 1) == 0 && (b & 1) == 0) {
a >>= 1;
b >>= 1;
shift++;
}
while (true) {
while ((a & 1) == 0) {
a >>= 1;
}
DCHECK((b & 1) != 0);
while (true) {
if (a > b) {
a = (a - b) >> 1;
while ((a & 1) == 0) {
a >>= 1;
}
} else if (b > a) {
b = (b - a) >> 1;
while ((b & 1) == 0) {
b >>= 1;
}
if (a > b) {
a -= b;
} else if (b > a) {
b -= a;
} else {
return a << shift;
return a;
}
}
}
uint64 pq_factorize(uint64 pq) {
if (pq < 2 || pq > (static_cast<uint64>(1) << 63)) {
if (pq <= 2 || pq > (static_cast<uint64>(1) << 63)) {
return 1;
}
if ((pq & 1) == 0) {
return 2;
}
uint64 g = 0;
for (int i = 0, iter = 0; i < 3 || iter < 1000; i++) {
uint64 q = Random::fast(17, 32) % (pq - 1);
@ -120,7 +117,7 @@ uint64 pq_factorize(uint64 pq) {
x = c;
uint64 z = x < y ? pq + x - y : x - y;
g = gcd(z, pq);
g = pq_gcd(z, pq);
if (g != 1) {
break;
}