From d0c3ad8f0f946db7b12e1d74a122605e47ab94a5 Mon Sep 17 00:00:00 2001
From: Andrii Ursulenko <a.ursulenko@gmail.com>
Date: Mon, 2 Sep 2019 14:20:19 +0300
Subject: [PATCH] common,contarct: optimize save memory contract snapshot

---
 common.go   |  2 ++
 contract.go | 38 ++++++++++++++------------------------
 2 files changed, 16 insertions(+), 24 deletions(-)

diff --git a/common.go b/common.go
index dc11976b..a87a096a 100644
--- a/common.go
+++ b/common.go
@@ -50,4 +50,6 @@ var (
 	ZeroSignature     Signature
 
 	ZeroRoundPtr = &Round{}
+
+	ZeroPage = make([]byte, PageSize)
 )
diff --git a/contract.go b/contract.go
index 307d2233..9549296e 100644
--- a/contract.go
+++ b/contract.go
@@ -20,6 +20,7 @@
 package wavelet
 
 import (
+	"bytes"
 	"crypto/sha256"
 	"crypto/sha512"
 	"encoding/binary"
@@ -38,7 +39,6 @@ import (
 )
 
 var (
-	ErrNotSmartContract         = errors.New("contract: specified account ID is not a smart contract")
 	ErrContractFunctionNotFound = errors.New("contract: smart contract func not found")
 
 	_ exec.ImportResolver = (*ContractExecutor)(nil)
@@ -445,39 +445,29 @@ func SaveContractMemorySnapshot(snapshot *avl.Tree, id AccountID, mem []byte) {
 
 	WriteAccountContractNumPages(snapshot, id, numPages)
 
+	var (
+		identical, allZero bool
+		pageStart          uint64
+	)
+
 	for pageIdx := uint64(0); pageIdx < numPages; pageIdx++ {
 		old, _ := ReadAccountContractPage(snapshot, id, pageIdx)
+		identical, allZero = true, false
+		pageStart = pageIdx * PageSize
 
-		identical := true
-
-		for idx := uint64(0); idx < PageSize; idx++ {
-			if len(old) == 0 && mem[pageIdx*PageSize+idx] != 0 {
-				identical = false
-				break
-			}
-
-			if len(old) != 0 && mem[pageIdx*PageSize+idx] != old[idx] {
-				identical = false
-				break
-			}
+		if len(old) == 0 {
+			allZero = bytes.Equal(ZeroPage, mem[pageStart:pageStart+PageSize])
+			identical = allZero
+		} else {
+			identical = bytes.Equal(old, mem[pageStart:pageStart+PageSize])
 		}
 
 		if !identical {
-			allZero := true
-
-			for idx := uint64(0); idx < PageSize; idx++ {
-				if mem[pageIdx*PageSize+idx] != 0 {
-					allZero = false
-					break
-				}
-			}
-
 			// If the page is empty, save an empty byte array. Otherwise, save the pages content.
-
 			if allZero {
 				WriteAccountContractPage(snapshot, id, pageIdx, []byte{})
 			} else {
-				WriteAccountContractPage(snapshot, id, pageIdx, mem[pageIdx*PageSize:(pageIdx+1)*PageSize])
+				WriteAccountContractPage(snapshot, id, pageIdx, mem[pageStart:pageStart+PageSize])
 			}
 		}
 	}