diff --git a/README.md b/README.md
index 299e1d78..ed1a583b 100644
--- a/README.md
+++ b/README.md
@@ -16,6 +16,8 @@
+ 
+
diff --git a/core/banner.go b/core/banner.go
index 860d1e49..1a63f0c8 100644
--- a/core/banner.go
+++ b/core/banner.go
@@ -2,7 +2,7 @@ package core
const (
Name = "bettercap"
- Version = "2.41.2"
+ Version = "2.41.4"
Author = "Simone 'evilsocket' Margaritelli"
Website = "https://bettercap.org/"
)
diff --git a/modules/tcp_proxy/tcp_proxy_script.go b/modules/tcp_proxy/tcp_proxy_script.go
index 7121a1a4..50956ea0 100644
--- a/modules/tcp_proxy/tcp_proxy_script.go
+++ b/modules/tcp_proxy/tcp_proxy_script.go
@@ -1,6 +1,7 @@
package tcp_proxy
import (
+ "encoding/json"
"net"
"strings"
@@ -55,23 +56,36 @@ func (s *TcpProxyScript) OnData(from, to net.Addr, data []byte, callback func(ca
log.Error("error while executing onData callback: %s", err)
return nil
} else if ret != nil {
- // thanks to @LucasParsy for his code and patience :)
- if array, ok := ret.([]interface{}); ok {
- result := make([]byte, len(array))
- for i, v := range array {
- if num, ok := v.(float64); ok && num >= 0 && num <= 255 {
- result[i] = byte(num)
- } else {
- log.Error("array element at index %d is not a valid byte value %+v", i, v)
- return nil
- }
- }
-
- return result
- } else {
- log.Error("error while casting exported value to array of interface: value = %+v error = %+v", ret, err)
- }
+ return toByteArray(ret)
}
}
return nil
}
+
+func toByteArray(ret interface{}) []byte {
+ // this approach is a bit hacky but it handles all cases
+
+ // serialize ret to JSON
+ if jsonData, err := json.Marshal(ret); err == nil {
+ // attempt to deserialize as []float64
+ var back2Array []float64
+ if err := json.Unmarshal(jsonData, &back2Array); err == nil {
+ result := make([]byte, len(back2Array))
+ for i, num := range back2Array {
+ if num >= 0 && num <= 255 {
+ result[i] = byte(num)
+ } else {
+ log.Error("array element at index %d is not a valid byte value %d", i, num)
+ return nil
+ }
+ }
+ return result
+ } else {
+ log.Error("failed to deserialize %+v to []float64: %v", ret, err)
+ }
+ } else {
+ log.Error("failed to serialize %+v to JSON: %v", ret, err)
+ }
+
+ return nil
+}
diff --git a/modules/tcp_proxy/tcp_proxy_script_test.go b/modules/tcp_proxy/tcp_proxy_script_test.go
new file mode 100644
index 00000000..27bdc099
--- /dev/null
+++ b/modules/tcp_proxy/tcp_proxy_script_test.go
@@ -0,0 +1,169 @@
+package tcp_proxy
+
+import (
+ "net"
+ "testing"
+
+ "github.com/evilsocket/islazy/plugin"
+)
+
+func TestOnData_NoReturn(t *testing.T) {
+ jsCode := `
+ function onData(from, to, data, callback) {
+ // don't return anything
+ }
+ `
+
+ plug, err := plugin.Parse(jsCode)
+ if err != nil {
+ t.Fatalf("Failed to parse plugin: %v", err)
+ }
+
+ script := &TcpProxyScript{
+ Plugin: plug,
+ doOnData: plug.HasFunc("onData"),
+ }
+
+ from := &net.TCPAddr{IP: net.ParseIP("192.168.1.1"), Port: 1234}
+ to := &net.TCPAddr{IP: net.ParseIP("192.168.1.2"), Port: 5678}
+ data := []byte("test data")
+
+ result := script.OnData(from, to, data, nil)
+ if result != nil {
+ t.Errorf("Expected nil result when callback returns nothing, got %v", result)
+ }
+}
+
+func TestOnData_ReturnsArrayOfIntegers(t *testing.T) {
+ jsCode := `
+ function onData(from, to, data, callback) {
+ // Return modified data as array of integers
+ return [72, 101, 108, 108, 111]; // "Hello" in ASCII
+ }
+ `
+
+ plug, err := plugin.Parse(jsCode)
+ if err != nil {
+ t.Fatalf("Failed to parse plugin: %v", err)
+ }
+
+ script := &TcpProxyScript{
+ Plugin: plug,
+ doOnData: plug.HasFunc("onData"),
+ }
+
+ from := &net.TCPAddr{IP: net.ParseIP("192.168.1.1"), Port: 1234}
+ to := &net.TCPAddr{IP: net.ParseIP("192.168.1.2"), Port: 5678}
+ data := []byte("test data")
+
+ result := script.OnData(from, to, data, nil)
+ expected := []byte("Hello")
+
+ if result == nil {
+ t.Fatal("Expected non-nil result when callback returns array of integers")
+ }
+
+ if len(result) != len(expected) {
+ t.Fatalf("Expected result length %d, got %d", len(expected), len(result))
+ }
+
+ for i, b := range result {
+ if b != expected[i] {
+ t.Errorf("Expected byte at index %d to be %d, got %d", i, expected[i], b)
+ }
+ }
+}
+
+func TestOnData_ReturnsDynamicArray(t *testing.T) {
+ jsCode := `
+ function onData(from, to, data, callback) {
+ var result = [];
+ for (var i = 0; i < data.length; i++) {
+ result.push((data[i] + 1) % 256);
+ }
+ return result;
+ }
+ `
+
+ plug, err := plugin.Parse(jsCode)
+ if err != nil {
+ t.Fatalf("Failed to parse plugin: %v", err)
+ }
+
+ script := &TcpProxyScript{
+ Plugin: plug,
+ doOnData: plug.HasFunc("onData"),
+ }
+
+ from := &net.TCPAddr{IP: net.ParseIP("192.168.1.1"), Port: 1234}
+ to := &net.TCPAddr{IP: net.ParseIP("192.168.1.2"), Port: 5678}
+ data := []byte{10, 20, 30, 40, 255}
+
+ result := script.OnData(from, to, data, nil)
+ expected := []byte{11, 21, 31, 41, 0} // 255 + 1 = 256 % 256 = 0
+
+ if result == nil {
+ t.Fatal("Expected non-nil result when callback returns array of integers")
+ }
+
+ if len(result) != len(expected) {
+ t.Fatalf("Expected result length %d, got %d", len(expected), len(result))
+ }
+
+ for i, b := range result {
+ if b != expected[i] {
+ t.Errorf("Expected byte at index %d to be %d, got %d", i, expected[i], b)
+ }
+ }
+}
+
+func TestOnData_ReturnsMixedArray(t *testing.T) {
+ jsCode := `
+ function charToInt(value) {
+ return value.charCodeAt()
+ }
+
+ function onData(from, to, data) {
+ st_data = String.fromCharCode.apply(null, data)
+ if( st_data.indexOf("mysearch") != -1 ) {
+ payload = "mypayload";
+ st_data = st_data.replace("mysearch", payload);
+ res_int_arr = st_data.split("").map(charToInt) // []uint16
+ res_int_arr[0] = payload.length + 1; // first index is float64 and rest []uint16
+ return res_int_arr;
+ }
+ return data;
+ }
+ `
+
+ plug, err := plugin.Parse(jsCode)
+ if err != nil {
+ t.Fatalf("Failed to parse plugin: %v", err)
+ }
+
+ script := &TcpProxyScript{
+ Plugin: plug,
+ doOnData: plug.HasFunc("onData"),
+ }
+
+ from := &net.TCPAddr{IP: net.ParseIP("192.168.1.1"), Port: 1234}
+ to := &net.TCPAddr{IP: net.ParseIP("192.168.1.6"), Port: 5678}
+ data := []byte("Hello mysearch world")
+
+ result := script.OnData(from, to, data, nil)
+ expected := []byte("\x0aello mypayload world")
+
+ if result == nil {
+ t.Fatal("Expected non-nil result when callback returns array of integers")
+ }
+
+ if len(result) != len(expected) {
+ t.Fatalf("Expected result length %d, got %d", len(expected), len(result))
+ }
+
+ for i, b := range result {
+ if b != expected[i] {
+ t.Errorf("Expected byte at index %d to be %d, got %d", i, expected[i], b)
+ }
+ }
+}
diff --git a/modules/wifi/wifi_test.go b/modules/wifi/wifi_test.go
index 2a580f32..afd5322c 100644
--- a/modules/wifi/wifi_test.go
+++ b/modules/wifi/wifi_test.go
@@ -518,37 +518,6 @@ func TestWiFiModuleProbe(t *testing.T) {
}
}
-func TestWiFiModuleChannelSwitchAnnounce(t *testing.T) {
- sess := createMockSession()
- mod := NewWiFiModule(sess)
-
- // Test CSA handler
- handlers := mod.Handlers()
- var csaHandler session.ModuleHandler
- for _, h := range handlers {
- if h.Name == "wifi.channel_switch_announce bssid channel " {
- csaHandler = h
- break
- }
- }
-
- if csaHandler.Name == "" {
- t.Fatal("CSA handler not found")
- }
-
- // Test with valid parameters
- err := csaHandler.Exec([]string{"aa:bb:cc:dd:ee:ff", "11"})
- if err == nil {
- t.Error("Expected error when running CSA without running module")
- }
-
- // Test with invalid channel
- err = csaHandler.Exec([]string{"aa:bb:cc:dd:ee:ff", "999"})
- if err == nil {
- t.Error("Expected error with invalid channel")
- }
-}
-
func TestWiFiModuleFakeAuth(t *testing.T) {
sess := createMockSession()
mod := NewWiFiModule(sess)