split types into extypes
This commit is contained in:
parent
06d1642b16
commit
8f2ac50590
69
exttypes.go
Normal file
69
exttypes.go
Normal file
@ -0,0 +1,69 @@
|
||||
package dsnet
|
||||
|
||||
import (
|
||||
"net"
|
||||
"strings"
|
||||
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
||||
type JSONIPNet struct {
|
||||
IPNet net.IPNet
|
||||
}
|
||||
|
||||
func (n JSONIPNet) MarshalJSON() ([]byte, error) {
|
||||
return []byte("\"" + n.IPNet.String() + "\""), nil
|
||||
}
|
||||
|
||||
func (n *JSONIPNet) UnmarshalJSON(b []byte) error {
|
||||
cidr := strings.Trim(string(b), "\"")
|
||||
IP, IPNet, err := net.ParseCIDR(cidr)
|
||||
IPNet.IP = IP
|
||||
n.IPNet = *IPNet
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *JSONIPNet) String() string {
|
||||
return n.IPNet.String()
|
||||
}
|
||||
|
||||
type JSONKey struct {
|
||||
Key wgtypes.Key
|
||||
}
|
||||
|
||||
func (k JSONKey) MarshalJSON() ([]byte, error) {
|
||||
return []byte("\"" + k.Key.String() + "\""), nil
|
||||
}
|
||||
|
||||
func (k JSONKey) PublicKey() JSONKey {
|
||||
return JSONKey{
|
||||
Key: k.Key.PublicKey(),
|
||||
}
|
||||
}
|
||||
|
||||
func (k *JSONKey) UnmarshalJSON(b []byte) error {
|
||||
b64Key := strings.Trim(string(b), "\"")
|
||||
key, err := wgtypes.ParseKey(b64Key)
|
||||
k.Key = key
|
||||
return err
|
||||
}
|
||||
|
||||
func GenerateJSONPrivateKey() JSONKey {
|
||||
privateKey, err := wgtypes.GeneratePrivateKey()
|
||||
|
||||
check(err)
|
||||
|
||||
return JSONKey{
|
||||
Key: privateKey,
|
||||
}
|
||||
}
|
||||
|
||||
func GenerateJSONKey() JSONKey {
|
||||
privateKey, err := wgtypes.GenerateKey()
|
||||
|
||||
check(err)
|
||||
|
||||
return JSONKey{
|
||||
Key: privateKey,
|
||||
}
|
||||
}
|
62
types.go
62
types.go
@ -4,7 +4,6 @@ import (
|
||||
"encoding/json"
|
||||
"io/ioutil"
|
||||
"net"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
@ -151,64 +150,3 @@ type Dsnet struct {
|
||||
ListenPort int
|
||||
Peers []Peer
|
||||
}
|
||||
|
||||
type JSONIPNet struct {
|
||||
IPNet net.IPNet
|
||||
}
|
||||
|
||||
func (n JSONIPNet) MarshalJSON() ([]byte, error) {
|
||||
return []byte("\"" + n.IPNet.String() + "\""), nil
|
||||
}
|
||||
|
||||
func (n *JSONIPNet) UnmarshalJSON(b []byte) error {
|
||||
cidr := strings.Trim(string(b), "\"")
|
||||
IP, IPNet, err := net.ParseCIDR(cidr)
|
||||
IPNet.IP = IP
|
||||
n.IPNet = *IPNet
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *JSONIPNet) String() string {
|
||||
return n.IPNet.String()
|
||||
}
|
||||
|
||||
type JSONKey struct {
|
||||
Key wgtypes.Key
|
||||
}
|
||||
|
||||
func (k JSONKey) MarshalJSON() ([]byte, error) {
|
||||
return []byte("\"" + k.Key.String() + "\""), nil
|
||||
}
|
||||
|
||||
func (k JSONKey) PublicKey() JSONKey {
|
||||
return JSONKey{
|
||||
Key: k.Key.PublicKey(),
|
||||
}
|
||||
}
|
||||
|
||||
func (k *JSONKey) UnmarshalJSON(b []byte) error {
|
||||
b64Key := strings.Trim(string(b), "\"")
|
||||
key, err := wgtypes.ParseKey(b64Key)
|
||||
k.Key = key
|
||||
return err
|
||||
}
|
||||
|
||||
func GenerateJSONPrivateKey() JSONKey {
|
||||
privateKey, err := wgtypes.GeneratePrivateKey()
|
||||
|
||||
check(err)
|
||||
|
||||
return JSONKey{
|
||||
Key: privateKey,
|
||||
}
|
||||
}
|
||||
|
||||
func GenerateJSONKey() JSONKey {
|
||||
privateKey, err := wgtypes.GenerateKey()
|
||||
|
||||
check(err)
|
||||
|
||||
return JSONKey{
|
||||
Key: privateKey,
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user