0
0
mirror of https://github.com/vx3r/wg-gen-web.git synced 2024-12-23 01:03:24 +00:00
wg-gen-web/util/util.go

136 lines
2.8 KiB
Go
Raw Normal View History

2020-01-30 06:45:49 +00:00
package util
import (
"errors"
"io/ioutil"
"net"
"os"
"regexp"
)
var (
RegexpEmail = regexp.MustCompile("^[a-zA-Z0-9.!#$%&'*+/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$")
// pushed build time
Version string
2020-01-30 06:45:49 +00:00
)
2020-01-30 07:17:29 +00:00
// ReadFile file content
2020-01-30 06:45:49 +00:00
func ReadFile(path string) (bytes []byte, err error) {
bytes, err = ioutil.ReadFile(path)
if err != nil {
return nil, err
}
return bytes, nil
}
2020-01-30 07:17:29 +00:00
// WriteFile content to file
2020-01-30 06:45:49 +00:00
func WriteFile(path string, bytes []byte) (err error) {
err = ioutil.WriteFile(path, bytes, 0644)
if err != nil {
return err
}
return nil
}
2020-01-30 07:17:29 +00:00
// FileExists check if file exists
2020-01-30 06:45:49 +00:00
func FileExists(name string) bool {
info, err := os.Stat(name)
if os.IsNotExist(err) {
return false
}
return !info.IsDir()
}
2020-01-30 07:17:29 +00:00
// DirectoryExists check if directory exists
2020-01-30 06:45:49 +00:00
func DirectoryExists(name string) bool {
info, err := os.Stat(name)
if os.IsNotExist(err) {
return false
}
return info.IsDir()
}
2020-01-30 07:17:29 +00:00
// GetAvailableIp search for an available in cidr against a list of reserved ips
2020-01-30 06:45:49 +00:00
func GetAvailableIp(cidr string, reserved []string) (string, error) {
ip, ipnet, err := net.ParseCIDR(cidr)
2020-01-30 06:45:49 +00:00
if err != nil {
return "", err
}
// this two addresses are not usable
broadcastAddr := BroadcastAddr(ipnet).String()
networkAddr := ipnet.IP.String()
for ip := ip.Mask(ipnet.Mask); ipnet.Contains(ip); inc(ip) {
2020-01-30 06:45:49 +00:00
ok := true
address := ip.String()
2020-01-30 06:45:49 +00:00
for _, r := range reserved {
if address == r {
2020-01-30 06:45:49 +00:00
ok = false
break
}
}
if ok && address != networkAddr && address != broadcastAddr {
return address, nil
2020-01-30 06:45:49 +00:00
}
}
return "", errors.New("no more available address from cidr")
}
// IsIPv6 check if given ip is IPv6
func IsIPv6(address string) bool {
ip := net.ParseIP(address)
if ip == nil {
return false
}
return ip.To4() == nil
}
// IsValidIp check if ip is valid
func IsValidIp(ip string) bool {
return net.ParseIP(ip) != nil
}
// IsValidCidr check if CIDR is valid
func IsValidCidr(cidr string) bool {
_, _, err := net.ParseCIDR(cidr)
return err == nil
}
// GetIpFromCidr get ip from cidr
func GetIpFromCidr(cidr string) (string, error) {
ip, _, err := net.ParseCIDR(cidr)
if err != nil {
return "", err
}
return ip.String(), nil
}
2020-01-30 06:45:49 +00:00
// http://play.golang.org/p/m8TNTtygK0
func inc(ip net.IP) {
for j := len(ip) - 1; j >= 0; j-- {
ip[j]++
if ip[j] > 0 {
break
}
}
}
// BroadcastAddr returns the last address in the given network, or the broadcast address.
func BroadcastAddr(n *net.IPNet) net.IP {
// The golang net package doesn't make it easy to calculate the broadcast address. :(
var broadcast net.IP
if len(n.IP) == 4 {
broadcast = net.ParseIP("0.0.0.0").To4()
} else {
broadcast = net.ParseIP("::")
}
for i := 0; i < len(n.IP); i++ {
broadcast[i] = n.IP[i] | ^n.Mask[i]
}
return broadcast
}