implementing more core functions

Author: Alex Levinson

FUNCTIONS.md

@@ -133,6 +133,52 @@ Basic string deobfuscator function.
 
 
 
+## `MD5(data)`
+
+Perform an MD5() hash on data.
+
+### Argument List
+
+ * **data** *[]byte*
+
+### Returned Object Fields
+
+ * **value** *string*
+
+---
+
+
+
+## `Timestamp()`
+
+Get the system's current timestamp in epoch format.
+
+### Argument List
+
+
+### Returned Object Fields
+
+ * **value** *int64*
+
+---
+
+
+
+## `Halt()`
+
+Stop the current VM from continuing execution.
+
+### Argument List
+
+
+### Returned Object Fields
+
+ * **value** *bool*
+
+---
+
+
+
 ## `ExecuteCommand(baseCmd, cmdArgs)`
 
 Executes system commands.

engine/lib_core.go

@@ -1,10 +1,13 @@
 package engine
 
 import (
+	"crypto/md5"
 	"crypto/rand"
+	"encoding/hex"
 	"errors"
 	"math/big"
 	"strings"
+	"time"
 	"unicode"
 )
 
@@ -86,3 +89,21 @@ func (e *Engine) Asset(filename string) ([]byte, error) {
 	err := errors.New("Asset not found: " + filename)
 	return []byte{}, err
 }
+
+func (e *Engine) Timestamp() int64 {
+	return time.Now().Unix()
+}
+
+func (e *Engine) Halt() bool {
+	e.Halted = true
+	e.VM.Interrupt <- func() {
+		panic(errTimeout)
+	}
+	return true
+}
+
+func (e *Engine) MD5(data []byte) string {
+	hasher := md5.New()
+	hasher.Write(data)
+	return hex.EncodeToString(hasher.Sum(nil))
+}

engine/util.go

@@ -6,7 +6,6 @@ import (
 	"fmt"
 	"net"
 
-	"github.com/davecgh/go-spew/spew"
 	"github.com/robertkrimen/otto"
 )
 
@@ -97,7 +96,6 @@ func (e *Engine) ValueToByteSlice(v otto.Value) []byte {
 		}
 	} else {
 		e.Logger.WithField("trace", "true").Errorf("Unknown class to cast to byte slice")
-		spew.Dump(v)
 	}
 
 	return valueBytes

engine/vm_functions.go

@@ -98,6 +98,35 @@
     - label: value
       gotype: string
       return: true
+- name: MD5
+  description: Perform an MD5() hash on data.
+  author: Alex
+  package: core
+  args:
+    - label: data
+      gotype: '[]byte'
+  returns:
+    - label: value
+      gotype: string
+      return: true
+- name: Timestamp
+  description: Get the system's current timestamp in epoch format.
+  author: Alex
+  package: core
+  args: []
+  returns:
+    - label: value
+      gotype: int64
+      return: true
+- name: Halt
+  description: Stop the current VM from continuing execution.
+  author: Alex
+  package: core
+  args: []
+  returns:
+    - label: value
+      gotype: bool
+      return: true
 #####################################################################
 #
 # exec package

functions.yml

@@ -5,10 +5,12 @@ function BeforeDeploy() {
 }
 
 function Deploy() {
+  LogInfo("Deploy()");
   return true;
 }
 
 function AfterDeploy() {
+  LogInfo("AfterDeploy()");
   return true;
 }
 

spec/halt.gs

@@ -0,0 +1,18 @@
+//import:/Users/flint/Downloads/tater.jpg
+
+function BeforeDeploy() {
+  LogInfo("BeforeDeploy()");
+  return true;
+}
+
+function Deploy() {
+  var firstHash = MD5("helloworld");
+  var secondHash = MD5(Asset("tater.jpg").fileData);
+  DebugConsole();
+  return true;
+}
+
+function AfterDeploy() {
+  LogInfo("AfterDeploy()");
+  return true;
+}

spec/md5.gs

@@ -0,0 +1,16 @@
+function BeforeDeploy() {
+  LogInfo("BeforeDeploy()");
+  return true;
+}
+
+function Deploy() {
+  ts = Timestamp();
+  LogInfo("Deploy()");
+  LogDebug("Timestamp: " + ts.value);
+  return true;
+}
+
+function AfterDeploy() {
+  LogInfo("AfterDeploy()");
+  return true;
+}