thunderbrewhq/squall
1
0
Fork 0
mirror of https://github.com/thunderbrewhq/squall.git synced 2025-12-12 02:22:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(big): add SBigShr

Browse source
This commit is contained in:
fallenoak 2023-02-03 14:50:34 -06:00
parent 4a1b6a0f19
commit af0d9ec60e
No known key found for this signature in database
GPG key ID: 7628F8E61AEA070D
5 changed files with 202 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
storm/Big.cpp
View file

@ -49,6 +49,10 @@ void SBigMul(BigData* a, BigData* b, BigData* c) {
Mul(a->Primary(), b->Primary(), c->Primary(), a->Stack());
}
void SBigShr(BigData* a, BigData* b, uint32_t shift) {
Shr(a->Primary(), b->Primary(), shift);
}
void SBigSquare(BigData* a, BigData* b) {
Square(a->Primary(), b->Primary(), a->Stack());
}

2
storm/Big.hpp
View file

@ -20,6 +20,8 @@ void SBigMul(BigData* a, BigData* b, BigData* c);
void SBigNew(BigData** num);
void SBigShr(BigData* a, BigData* b, uint32_t shift);
void SBigSquare(BigData* a, BigData* b);
void SBigSub(BigData* a, BigData* b, BigData* c);

18
storm/big/Ops.cpp
View file

@ -170,6 +170,24 @@ void SetZero(BigBuffer& buffer) {
buffer.Clear();
}
void Shr(BigBuffer& a, const BigBuffer& b, uint32_t shift) {
auto v4 = shift >> 5;
auto v9 = shift & 0x1F;
uint32_t i = 0;
for (i = 0; b.IsUsed(i + v4); i++) {
auto v6 = b[i + v4] >> v9;
if (v9) {
v6 += b[i + v4 + 1] << (32 - v9);
}
a[i] = v6;
}
a.SetCount(i);
}
void Square(BigBuffer& a, const BigBuffer& b, BigStack& stack) {
auto& aa = stack.MakeDistinct(a, &a == &b);
aa.Clear();

2
storm/big/Ops.hpp
View file

@ -37,6 +37,8 @@ void SetOne(BigBuffer& buffer);
void SetZero(BigBuffer& buffer);
void Shr(BigBuffer& a, const BigBuffer& b, uint32_t shift);
void Square(BigBuffer& a, const BigBuffer& b, BigStack& stack);
void Sub(BigBuffer& a, const BigBuffer& b, const BigBuffer& c);

176
test/Big.cpp
View file

@ -855,6 +855,27 @@ TEST_CASE("SBigMul", "[big]") {
}
}
TEST_CASE("SBigShr", "[big]") {
SECTION("shifts 256 right 7 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 256);
SBigShr(a, b, 7);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 2);
SBigDel(a);
SBigDel(b);
}
}
TEST_CASE("SBigSquare", "[square]") {
SECTION("squares 0xFFFFFFFF") {
BigData* a;
@ -983,6 +1004,161 @@ TEST_CASE("SetZero", "[big]") {
}
}
TEST_CASE("Shr", "[big]") {
SECTION("shifts 0 right 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 0);
Shr(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 0);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 1 right 0 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 1);
Shr(a->Primary(), b->Primary(), 0);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 1);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 1 right 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 1);
Shr(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 0);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 2 right 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 2);
Shr(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 1);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 2 right 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 2);
Shr(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 1);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 256 right 7 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 256);
Shr(a->Primary(), b->Primary(), 7);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 2);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 0x1000000000000000 right 2 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
uint64_t b_ = 0x1000000000000000;
SBigFromBinary(b, reinterpret_cast<uint8_t*>(&b_), sizeof(b_));
Shr(a->Primary(), b->Primary(), 2);
a->Primary().Trim();
CHECK(a->Primary().Count() == 2);
CHECK(a->Primary()[0] == 0);
CHECK(a->Primary()[1] == 0x4000000);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 0x1000000000000000 right 32 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
uint64_t b_ = 0x1000000000000000;
SBigFromBinary(b, reinterpret_cast<uint8_t*>(&b_), sizeof(b_));
Shr(a->Primary(), b->Primary(), 32);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 0x10000000);
SBigDel(a);
SBigDel(b);
}
}
TEST_CASE("Square", "[big]") {
SECTION("squares 0") {
BigData* a;