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 SBigShl

Browse source
This commit is contained in:
fallenoak 2023-02-03 16:20:07 -06:00
parent 60edadfe06
commit 25c2ad06f1
No known key found for this signature in database
GPG key ID: 7628F8E61AEA070D
5 changed files with 212 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 SBigShl(BigData* a, BigData* b, uint32_t shift) {
Shl(a->Primary(), b->Primary(), shift);
}
void SBigShr(BigData* a, BigData* b, uint32_t shift) {
Shr(a->Primary(), b->Primary(), shift);
}

2
storm/Big.hpp
View file

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

23
storm/big/Ops.cpp
View file

@ -170,6 +170,29 @@ void SetZero(BigBuffer& buffer) {
buffer.Clear();
}
void Shl(BigBuffer& a, const BigBuffer& b, uint32_t shift) {
auto v4 = shift >> 5;
auto v14 = shift & 0x1F;
auto v5 = b.Count() + v4 + 1;
for (uint32_t i = v5 - 1, j = v5 - v4 - 1; i + 1 > 0; i--, j--) {
uint32_t v9 = 0;
if (i >= v4) {
v9 = b[j] << v14;
if (i > v4 && v14) {
v9 += b[j - 1] >> (32 - v14);
}
}
a[i] = v9;
}
a.SetCount(v5);
a.Trim();
}
void Shr(BigBuffer& a, const BigBuffer& b, uint32_t shift) {
auto v4 = shift >> 5;
auto v9 = shift & 0x1F;

2
storm/big/Ops.hpp
View file

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

181
test/Big.cpp
View file

@ -855,6 +855,27 @@ TEST_CASE("SBigMul", "[big]") {
}
}
TEST_CASE("SBigShl", "[big]") {
SECTION("shifts 256 left 7 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 256);
SBigShl(a, b, 7);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 32768);
SBigDel(a);
SBigDel(b);
}
}
TEST_CASE("SBigShr", "[big]") {
SECTION("shifts 256 right 7 bits") {
BigData* a;
@ -1004,6 +1025,166 @@ TEST_CASE("SetZero", "[big]") {
}
}
TEST_CASE("Shl", "[big]") {
SECTION("shifts 0 left 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 0);
Shl(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 0);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 1 left 0 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 1);
Shl(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 left 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 1);
Shl(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 2);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 2 left 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 2);
Shl(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 4);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 256 left 7 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
SBigFromUnsigned(b, 256);
Shl(a->Primary(), b->Primary(), 7);
a->Primary().Trim();
CHECK(a->Primary().Count() == 1);
CHECK(a->Primary()[0] == 32768);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 0xFFFFFFFF left 10 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
uint64_t b_ = 0xFFFFFFFF;
SBigFromBinary(b, reinterpret_cast<uint8_t*>(&b_), sizeof(b_));
Shl(a->Primary(), b->Primary(), 10);
a->Primary().Trim();
CHECK(a->Primary().Count() == 2);
CHECK(a->Primary()[0] == 0xFFFFFC00);
CHECK(a->Primary()[1] == 0x3FF);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 0x1000000000000000 left 1 bit") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
uint64_t b_ = 0x1000000000000000;
SBigFromBinary(b, reinterpret_cast<uint8_t*>(&b_), sizeof(b_));
Shl(a->Primary(), b->Primary(), 1);
a->Primary().Trim();
CHECK(a->Primary().Count() == 2);
CHECK(a->Primary()[0] == 0);
CHECK(a->Primary()[1] == 0x20000000);
SBigDel(a);
SBigDel(b);
}
SECTION("shifts 0x1000000000000000 left 16 bits") {
BigData* a;
SBigNew(&a);
BigData* b;
SBigNew(&b);
uint64_t b_ = 0x1000000000000000;
SBigFromBinary(b, reinterpret_cast<uint8_t*>(&b_), sizeof(b_));
Shl(a->Primary(), b->Primary(), 16);
a->Primary().Trim();
CHECK(a->Primary().Count() == 3);
CHECK(a->Primary()[0] == 0);
CHECK(a->Primary()[1] == 0);
CHECK(a->Primary()[2] == 0x1000);
SBigDel(a);
SBigDel(b);
}
}
TEST_CASE("Shr", "[big]") {
SECTION("shifts 0 right 1 bit") {
BigData* a;