feat: throw error if trying to write to kv storage in a view function

warp-contracts · Nov 9, 2023 · 0b13ca4 · 0b13ca4
1 parent 8c2d62e
commit 0b13ca4
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/src/core/modules/impl/handler/JsHandlerApi.ts b/src/core/modules/impl/handler/JsHandlerApi.ts
@@ -238,6 +238,7 @@ export class JsHandlerApi<State> extends AbstractContractHandler<State> {
 
   private setupSwGlobal<Input>({ interaction, interactionTx }: InteractionData<Input>) {
     this.swGlobal._activeTx = interactionTx;
+    this.swGlobal.interactionType = interaction.interactionType;
     this.swGlobal.caller = interaction.caller; // either contract tx id (for internal writes) or transaction.owner
   }
 

diff --git a/src/core/modules/impl/handler/WasmHandlerApi.ts b/src/core/modules/impl/handler/WasmHandlerApi.ts
@@ -33,6 +33,7 @@ export class WasmHandlerApi<State> extends AbstractContractHandler<State> {
 
       this.swGlobal._activeTx = interactionTx;
       this.swGlobal.caller = interaction.caller; // either contract tx id (for internal writes) or transaction.owner
+      this.swGlobal.interactionType = interaction.interactionType;
       this.swGlobal.gasLimit = executionContext.evaluationOptions.gasLimit;
       this.swGlobal.gasUsed = 0;
 

diff --git a/src/legacy/smartweave-global.ts b/src/legacy/smartweave-global.ts
@@ -6,6 +6,7 @@ import { CacheKey, SortKeyCache } from '../cache/SortKeyCache';
 import { SortKeyCacheRangeOptions } from '../cache/SortKeyCacheRangeOptions';
 import { InteractionState } from '../contract/states/InteractionState';
 import { safeGet } from '../utils/utils';
+import { ContractError, InteractionType } from "../core/modules/impl/HandlerExecutorFactory";
 
 /**
  *
@@ -66,6 +67,8 @@ export class SmartWeaveGlobal {
 
   kv: KV;
 
+  interactionType: InteractionType;
+
   constructor(
     arweave: Arweave,
     contract: { id: string; owner: string },
@@ -218,6 +221,13 @@ export class SWTransaction {
     }
     return this.smartWeaveGlobal._activeTx.source === 'redstone-sequencer' ? 'L2' : 'L1';
   }
+
+  get interactionType(): InteractionType {
+    if (!this.smartWeaveGlobal._activeTx) {
+      throw new Error('No current Tx');
+    }
+    return this.smartWeaveGlobal.interactionType;
+  }
 }
 
 // tslint:disable-next-line: max-classes-per-file
@@ -282,7 +292,13 @@ export class KV {
   ) {}
 
   async put(key: string, value: any): Promise<void> {
+    if (this._transaction.interactionType === 'view') {
+      throw new ContractError('Forbidden write operation for "view" function: "KV.put"');
+    }
+
     this.checkStorageAvailable();
+
+
     await this._storage.put(new CacheKey(key, this._transaction.sortKey), value);
   }
 
@@ -304,6 +320,10 @@ export class KV {
   }
 
   async del(key: string): Promise<void> {
+    if (this._transaction.interactionType === 'view') {
+      throw new ContractError('Forbidden write operation for "view" function: "KV.del"');
+    }
+
     this.checkStorageAvailable();
     const sortKey = this._transaction.sortKey;