sam-forwarder/config/tunconf.go

package i2ptunconf

import (
	"github.com/eyedeekay/sam-forwarder"
	"github.com/zieckey/goini"
	"strings"
)

type Conf struct {
	config             *goini.INI
	saveFile           bool
	TargetHost         string
	TargetPort         string
	SamHost            string
	SamPort            string
	TunName            string
	encryptLeaseSet    bool
	inAllowZeroHop     bool
	outAllowZeroHop    bool
	inLength           int
	outLength          int
	inQuantity         int
	outQuantity        int
	inVariance         int
	outVariance        int
	inBackupQuantity   int
	outBackupQuantity  int
	useCompression     bool
	reduceIdle         bool
	reduceIdleTime     int
	reduceIdleQuantity int
	accessListType     string
	accessList         []string
}

func (c *Conf) configParse(path string) (*goini.INI, error) {
	ini := goini.New()
	if err := ini.ParseFile(path); err != nil {
		return nil, err
	}
	return ini, nil
}

// Get passes directly through to goini.Get
func (c *Conf) Get(key string) (string, bool) {
	return c.config.Get(key)
}

// GetBool passes directly through to goini.GetBool
func (c *Conf) GetBool(key string) (bool, bool) {
	return c.config.GetBool(key)
}

// GetInt passes directly through to goini.GetInt
func (c *Conf) GetInt(key string) (int, bool) {
	return c.config.GetInt(key)
}

func NewI2PTunConf(iniFile string) (*Conf, error) {
	var config *goini.INI
	var err error
	var c Conf

	if iniFile != "none" {
		config, err = c.configParse(iniFile)
		if err != nil {
			return nil, err
		}
		if v, ok := config.GetBool("keys"); ok {
			c.saveFile = v
		} else {
			c.saveFile = false
		}

		if v, ok := config.Get("host"); ok {
			c.TargetHost = strings.Replace(v, ":", "", -1)
		} else {
			c.TargetHost = "127.0.0.1"
		}
		if v, ok := config.Get("port"); !ok {
			c.TargetPort = strings.Replace(v, ":", "", -1)
		} else {
			c.TargetPort = "8081"
		}

		if v, ok := config.Get("samhost"); ok {
			c.SamHost = strings.Replace(v, ":", "", -1)
		} else {
			c.SamHost = "127.0.0.1"
		}
		if v, ok := config.Get("samport"); ok {
			c.SamPort = strings.Replace(v, ":", "", -1)
		} else {
			c.SamPort = "7656"
		}

		if v, ok := config.Get("keys"); ok {
			c.TunName = v
		} else {
			c.TunName = "fowarder"
		}
		if v, ok := config.GetBool("i2cp.encryptLeaseSet"); ok {
			c.encryptLeaseSet = v
		} else {
			c.encryptLeaseSet = false
		}

		if v, ok := config.GetBool("inbound.allowZeroHop"); ok {
			c.inAllowZeroHop = v
		} else {
			c.inAllowZeroHop = false
		}
		if v, ok := config.GetBool("outbound.allowZeroHop"); ok {
			c.outAllowZeroHop = v
		} else {
			c.outAllowZeroHop = false
		}

		if v, ok := config.GetInt("inbound.length"); ok {
			c.inLength = v
		} else {
			c.inLength = 3
		}
		if v, ok := config.GetInt("outbound.length"); ok {
			c.outLength = v
		} else {
			c.outLength = 3
		}

		if v, ok := config.GetInt("inbound.quantity"); ok {
			c.inQuantity = v
		} else {
			c.inQuantity = 5
		}
		if v, ok := config.GetInt("outbound.quantity"); ok {
			c.outQuantity = v
		} else {
			c.outQuantity = 5
		}

		if v, ok := config.GetInt("inbound.variance"); ok {
			c.inVariance = v
		} else {
			c.inVariance = 0
		}
		if v, ok := config.GetInt("outbound.variance"); ok {
			c.outVariance = v
		} else {
			c.outVariance = 0
		}

		if v, ok := config.GetInt("inbound.backupQuantity"); ok {
			c.inBackupQuantity = v
		} else {
			c.inBackupQuantity = 2
		}
		if v, ok := config.GetInt("outbound.backupQuantity"); ok {
			c.outBackupQuantity = v
		} else {
			c.outBackupQuantity = 2
		}

		if v, ok := config.GetBool("gzip"); ok {
			c.useCompression = v
		} else {
			c.useCompression = true
		}

		if v, ok := config.GetBool("i2cp.reduceOnIdle"); ok {
			c.reduceIdle = v
		} else {
			c.reduceIdle = false
		}
		if v, ok := config.GetInt("i2cp.reduceIdleTime"); ok {
			c.reduceIdleTime = (v / 1000) / 60
		} else {
			c.reduceIdleTime = (6 * 60) * 1000
		}
		if v, ok := config.GetInt("i2cp.reduceQuantity"); ok {
			c.reduceIdleQuantity = v
		} else {
			c.reduceIdleQuantity = 3
		}

		if v, ok := config.GetBool("i2cp.enableBlackList"); ok {
			if v {
				c.accessListType = "blacklist"
			}
		}
		if v, ok := config.GetBool("i2cp.enableAccessList"); ok {
			if v {
				c.accessListType = "whitelist"
			}
		}
		if c.accessListType != "whitelist" && c.accessListType != "blacklist" {
			c.accessListType = "none"
		}
		if v, ok := config.Get("i2cp.accessList"); ok {
			csv := strings.Split(v, ",")
			for _, z := range csv {
				c.accessList = append(c.accessList, z)
			}
		}

		return &c, nil
	}
	return nil, nil
}

func NewSAMForwarderFromConfig(iniFile, SamHost, SamPort string) (*samforwarder.SAMForwarder, error) {
	if iniFile != "none" {
		config, err := NewI2PTunConf(iniFile)
		if err != nil {
			return nil, err
		}
		return samforwarder.NewSAMForwarderFromOptions(
			samforwarder.SetSaveFile(config.saveFile),
			samforwarder.SetHost(config.TargetHost),
			samforwarder.SetPort(config.TargetPort),
			samforwarder.SetSAMHost(config.SamHost),
			samforwarder.SetSAMPort(config.SamPort),
			samforwarder.SetName(config.TunName),
			samforwarder.SetInLength(config.inLength),
			samforwarder.SetOutLength(config.outLength),
			samforwarder.SetInVariance(config.inVariance),
			samforwarder.SetOutVariance(config.outVariance),
			samforwarder.SetInQuantity(config.inQuantity),
			samforwarder.SetOutQuantity(config.outQuantity),
			samforwarder.SetInBackups(config.inBackupQuantity),
			samforwarder.SetOutBackups(config.outBackupQuantity),
			samforwarder.SetEncrypt(config.encryptLeaseSet),
			samforwarder.SetAllowZeroIn(config.inAllowZeroHop),
			samforwarder.SetAllowZeroOut(config.outAllowZeroHop),
			samforwarder.SetCompress(config.useCompression),
			samforwarder.SetReduceIdle(config.reduceIdle),
			samforwarder.SetReduceIdleTime(config.reduceIdleTime),
			samforwarder.SetReduceIdleQuantity(config.reduceIdleQuantity),
			samforwarder.SetAccessListType(config.accessListType),
			samforwarder.SetAccessList(config.accessList),
		)
	}
	return nil, nil
}