From 9eeca0ebc8b48677cbde1f621457b499ef7d850d Mon Sep 17 00:00:00 2001
From: "daniel.vladco" <vladcodaniel@gmail.com>
Date: Wed, 22 Jan 2025 19:16:51 +0200
Subject: [PATCH] feat: implement missing instr with 2 regs v0.5.4

---
 internal/polkavm/interpreter/mutator.go | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/internal/polkavm/interpreter/mutator.go b/internal/polkavm/interpreter/mutator.go
index 2b795e4..9b97912 100644
--- a/internal/polkavm/interpreter/mutator.go
+++ b/internal/polkavm/interpreter/mutator.go
@@ -3,6 +3,7 @@ package interpreter
 import (
 	"log"
 	"math"
+	"math/bits"
 
 	"github.com/eigerco/strawberry/pkg/serialization/codec/jam"
 
@@ -150,34 +151,34 @@ func (m *Mutator) Sbrk(dst polkavm.Reg, sizeReg polkavm.Reg) error {
 	return nil
 }
 func (m *Mutator) CountSetBits64(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, uint64(bits.OnesCount64(m.get64(s))))
 }
 func (m *Mutator) CountSetBits32(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set32(d, uint32(bits.OnesCount32(m.get32(s))))
 }
 func (m *Mutator) LeadingZeroBits64(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, uint64(bits.LeadingZeros64(m.get64(s))))
 }
 func (m *Mutator) LeadingZeroBits32(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set32(d, uint32(bits.LeadingZeros32(m.get32(s))))
 }
 func (m *Mutator) TrailingZeroBits64(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, uint64(bits.TrailingZeros64(m.get64(s))))
 }
 func (m *Mutator) TrailingZeroBits32(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set32(d, uint32(bits.TrailingZeros32(m.get32(s))))
 }
 func (m *Mutator) SignExtend8(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, uint64(int64(int8(uint8(m.get64(s))))))
 }
 func (m *Mutator) SignExtend16(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, uint64(int64(int16(uint16(m.get64(s))))))
 }
 func (m *Mutator) ZeroExtend16(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, uint64(int16(m.get64(s))))
 }
 func (m *Mutator) ReverseBytes(d polkavm.Reg, s polkavm.Reg) {
-	panic("todo: implement")
+	m.set64(d, bits.ReverseBytes64(m.get64(s)))
 }
 func (m *Mutator) MoveReg(d polkavm.Reg, s polkavm.Reg) {
 	m.setNext32(d, m.get32(s))