new: exposing session environment to proxy scripts

caplets/beef-inject.js

@@ -1,5 +1,6 @@
 function onLoad() {
     console.log( "BeefInject loaded." );
+    console.log("targets: " + env['arp.spoof.targets']);
 }
 
 function onResponse(req, res) {

session/environment.go

@@ -23,6 +23,10 @@ func NewEnvironment() *Environment {
 	return env
 }
 
+func (env *Environment) Storage() *map[string]string {
+	return &env.storage
+}
+
 func (env *Environment) Has(name string) bool {
 	env.lock.Lock()
 	defer env.lock.Unlock()

session/modules/http_proxy.go

@@ -166,7 +166,7 @@ func (p *HttpProxy) Start() error {
 	} else {
 		scriptPath := v.(string)
 		if scriptPath != "" {
-			if err, p.script = LoadProxyScript(scriptPath); err != nil {
+			if err, p.script = LoadProxyScript(scriptPath, p.Session); err != nil {
 				return err
 			} else {
 				log.Debugf("Proxy script %s loaded.", scriptPath)

session/modules/http_proxy_script.go

@@ -5,6 +5,8 @@ import (
 	"net/http"
 	"sync"
 
+	"github.com/evilsocket/bettercap-ng/session"
+
 	"github.com/robertkrimen/otto"
 )
 
@@ -20,7 +22,7 @@ type ProxyScript struct {
 	cbCache          map[string]bool
 }
 
-func LoadProxyScript(path string) (err error, s *ProxyScript) {
+func LoadProxyScript(path string, sess *session.Session) (err error, s *ProxyScript) {
 	log.Infof("Loading proxy script %s ...", path)
 
 	raw, err := ioutil.ReadFile(path)
@@ -32,6 +34,7 @@ func LoadProxyScript(path string) (err error, s *ProxyScript) {
 		Path:   path,
 		Source: string(raw),
 		VM:     otto.New(),
+
 		gil:              &sync.Mutex{},
 		onRequestScript:  nil,
 		onResponseScript: nil,
@@ -45,6 +48,13 @@ func LoadProxyScript(path string) (err error, s *ProxyScript) {
 		return
 	}
 
+	// define session pointer
+	err = s.VM.Set("env", *sess.Env.Storage())
+	if err != nil {
+		log.Errorf("Error while defining environment: %s", err)
+		return
+	}
+
 	// run onLoad if defined
 	if s.hasCallback("onLoad") {
 		_, err = s.VM.Run("onLoad()")