mirror of
https://github.com/bettercap/bettercap
synced 2025-07-08 05:51:37 -07:00
use ip6tables for any_proxy to ipv6 addresses
This commit is contained in:
parent
ed4239fad5
commit
2bc3d871ef
2 changed files with 23 additions and 7 deletions
|
@ -4,6 +4,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/bettercap/bettercap/core"
|
"github.com/bettercap/bettercap/core"
|
||||||
"github.com/bettercap/bettercap/network"
|
"github.com/bettercap/bettercap/network"
|
||||||
|
@ -73,10 +74,18 @@ func (f LinuxFirewall) EnableForwarding(enabled bool) error {
|
||||||
|
|
||||||
func (f *LinuxFirewall) getCommandLine(r *Redirection, enabled bool) (cmdLine []string) {
|
func (f *LinuxFirewall) getCommandLine(r *Redirection, enabled bool) (cmdLine []string) {
|
||||||
action := "-A"
|
action := "-A"
|
||||||
|
destination := ""
|
||||||
|
|
||||||
if !enabled {
|
if !enabled {
|
||||||
action = "-D"
|
action = "-D"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if strings.Count(r.DstAddress, ":") < 2 {
|
||||||
|
destination = r.DstAddress
|
||||||
|
} else {
|
||||||
|
destination = fmt.Sprintf("[%s]", r.DstAddress)
|
||||||
|
}
|
||||||
|
|
||||||
if r.SrcAddress == "" {
|
if r.SrcAddress == "" {
|
||||||
cmdLine = []string{
|
cmdLine = []string{
|
||||||
"-t", "nat",
|
"-t", "nat",
|
||||||
|
@ -85,7 +94,7 @@ func (f *LinuxFirewall) getCommandLine(r *Redirection, enabled bool) (cmdLine []
|
||||||
"-p", r.Protocol,
|
"-p", r.Protocol,
|
||||||
"--dport", fmt.Sprintf("%d", r.SrcPort),
|
"--dport", fmt.Sprintf("%d", r.SrcPort),
|
||||||
"-j", "DNAT",
|
"-j", "DNAT",
|
||||||
"--to", fmt.Sprintf("%s:%d", r.DstAddress, r.DstPort),
|
"--to", fmt.Sprintf("%s:%d", destination, r.DstPort),
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
cmdLine = []string{
|
cmdLine = []string{
|
||||||
|
@ -96,7 +105,7 @@ func (f *LinuxFirewall) getCommandLine(r *Redirection, enabled bool) (cmdLine []
|
||||||
"-d", r.SrcAddress,
|
"-d", r.SrcAddress,
|
||||||
"--dport", fmt.Sprintf("%d", r.SrcPort),
|
"--dport", fmt.Sprintf("%d", r.SrcPort),
|
||||||
"-j", "DNAT",
|
"-j", "DNAT",
|
||||||
"--to", fmt.Sprintf("%s:%d", r.DstAddress, r.DstPort),
|
"--to", fmt.Sprintf("%s:%d", destination, r.DstPort),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -107,6 +116,13 @@ func (f *LinuxFirewall) EnableRedirection(r *Redirection, enabled bool) error {
|
||||||
cmdLine := f.getCommandLine(r, enabled)
|
cmdLine := f.getCommandLine(r, enabled)
|
||||||
rkey := r.String()
|
rkey := r.String()
|
||||||
_, found := f.redirections[rkey]
|
_, found := f.redirections[rkey]
|
||||||
|
cmd := ""
|
||||||
|
|
||||||
|
if strings.Count(r.DstAddress, ":") < 2 {
|
||||||
|
cmd = "iptables"
|
||||||
|
} else {
|
||||||
|
cmd = "ip6tables"
|
||||||
|
}
|
||||||
|
|
||||||
if enabled {
|
if enabled {
|
||||||
if found {
|
if found {
|
||||||
|
@ -116,9 +132,9 @@ func (f *LinuxFirewall) EnableRedirection(r *Redirection, enabled bool) error {
|
||||||
f.redirections[rkey] = r
|
f.redirections[rkey] = r
|
||||||
|
|
||||||
// accept all
|
// accept all
|
||||||
if _, err := core.Exec("iptables", []string{"-P", "FORWARD", "ACCEPT"}); err != nil {
|
if _, err := core.Exec(cmd, []string{"-P", "FORWARD", "ACCEPT"}); err != nil {
|
||||||
return err
|
return err
|
||||||
} else if _, err := core.Exec("iptables", cmdLine); err != nil {
|
} else if _, err := core.Exec(cmd, cmdLine); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
|
@ -128,7 +144,7 @@ func (f *LinuxFirewall) EnableRedirection(r *Redirection, enabled bool) error {
|
||||||
|
|
||||||
delete(f.redirections, r.String())
|
delete(f.redirections, r.String())
|
||||||
|
|
||||||
if _, err := core.Exec("iptables", cmdLine); err != nil {
|
if _, err := core.Exec(cmd, cmdLine); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -43,8 +43,8 @@ func NewAnyProxy(s *session.Session) *AnyProxy {
|
||||||
"Leave empty to intercept any source address."))
|
"Leave empty to intercept any source address."))
|
||||||
|
|
||||||
mod.AddParam(session.NewStringParameter("any.proxy.dst_address",
|
mod.AddParam(session.NewStringParameter("any.proxy.dst_address",
|
||||||
session.ParamIfaceAddress,
|
"",
|
||||||
session.IPv4Validator,
|
"",
|
||||||
"Address where the proxy is listening."))
|
"Address where the proxy is listening."))
|
||||||
|
|
||||||
mod.AddParam(session.NewIntParameter("any.proxy.dst_port",
|
mod.AddParam(session.NewIntParameter("any.proxy.dst_port",
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue