Add precompile impl for constraints control

Author: MishkaRogachev

arbos/constraints/storage.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"io"
 
+	"github.com/ethereum/go-ethereum/arbitrum/multigas"
 	"github.com/ethereum/go-ethereum/rlp"
 )
 
@@ -129,3 +130,40 @@ func (src *StorageResourceConstraints) Write(rc *ResourceConstraints) error {
 	}
 	return src.storage.Set(data)
 }
+
+// SetConstraint adds or updates a resource constraint in storage.
+func (src *StorageResourceConstraints) SetConstraint(resourceKinds []uint8, resourceWeights []uint64, periodSecs uint32, targetPerSec uint64) error {
+	if len(resourceKinds) != len(resourceWeights) {
+		return fmt.Errorf("resourceKinds and resourceWeights must have the same length")
+	}
+	if len(resourceKinds) == 0 {
+		return fmt.Errorf("at least one resourceKind is required")
+	}
+	resSet := EmptyResourceSet()
+	for i, r := range resourceKinds {
+		resSet = resSet.WithResource(multigas.ResourceKind(r), ResourceWeight(resourceWeights[i]))
+	}
+	rc, err := src.Load()
+	if err != nil {
+		return err
+	}
+	rc.Set(resSet, PeriodSecs(periodSecs), targetPerSec)
+	return src.Write(rc)
+}
+
+func (src *StorageResourceConstraints) ClearConstraint(resourceKinds []uint8, periodSecs uint32) error {
+	if len(resourceKinds) == 0 {
+		return fmt.Errorf("at least one resourceKind is required")
+	}
+
+	resSet := EmptyResourceSet()
+	for _, r := range resourceKinds {
+		resSet = resSet.WithResource(multigas.ResourceKind(r), 1)
+	}
+	rc, err := src.Load()
+	if err != nil {
+		return err
+	}
+	rc.Clear(resSet, PeriodSecs(periodSecs))
+	return src.Write(rc)
+}

go-ethereum

@@ -453,3 +453,13 @@ func (con ArbOwner) SetChainConfig(c ctx, evm mech, serializedChainConfig []byte
 func (con ArbOwner) SetCalldataPriceIncrease(c ctx, _ mech, enable bool) error {
 	return c.State.Features().SetCalldataPriceIncrease(enable)
 }
+
+// SetResourceConstraint adds or updates a resource constraint with specified resource kinds and weights
+func (con ArbOwner) SetResourceConstraint(c ctx, evm mech, resourceKinds []uint8, resourceWeights []uint64, periodSecs uint32, targetPerSec uint64) error {
+	return c.State.ResourceConstraints().SetConstraint(resourceKinds, resourceWeights, periodSecs, targetPerSec)
+}
+
+// ClearConstraint removes a resource constraint
+func (con ArbOwner) ClearConstraint(c ctx, evm mech, resourceKinds []uint8, periodSecs uint32) error {
+	return c.State.ResourceConstraints().ClearConstraint(resourceKinds, periodSecs)
+}