cleaned up a few of the packer functions

rvpn/packer/packer_data.go

@@ -1,6 +1,8 @@
 package packer
 
-import "bytes"
+import (
+	"bytes"
+)
 
 //packerData -- Contains packer data
 type packerData struct {
@@ -14,18 +16,15 @@ func newPackerData() (p *packerData) {
 	return
 }
 
-func (p *packerData) AppendString(dataString string) (n int, err error) {
+func (p *packerData) AppendString(dataString string) (int, error) {
-	n, err = p.buffer.WriteString(dataString)
+	return p.buffer.WriteString(dataString)
-	return
 }
 
-func (p *packerData) AppendBytes(dataBytes []byte) (n int, err error) {
+func (p *packerData) AppendBytes(dataBytes []byte) (int, error) {
-	n, err = p.buffer.Write(dataBytes)
+	return p.buffer.Write(dataBytes)
-	return
 }
 
 //Data --
-func (p *packerData) Data() (b []byte) {
+func (p *packerData) Data() []byte {
-	b = p.buffer.Bytes()
+	return p.buffer.Bytes()
-	return
 }

rvpn/packer/packer_header.go

@@ -1,7 +1,9 @@
 package packer
 
-import "net"
+import (
-import "fmt"
+	"fmt"
+	"net"
+)
 
 type addressFamily int
 
@@ -37,45 +39,36 @@ func newPackerHeader() (p *packerHeader) {
 //SetAddress -- Set Address. which sets address family automatically
 func (p *packerHeader) SetAddress(addr string) {
 	p.address = net.ParseIP(addr)
-	err := p.address.To4()
 
-	if err != nil {
+	if p.address.To4() != nil {
 		p.family = FamilyIPv4
-	} else {
+	} else if p.address.To16() != nil {
-		err := p.address.To16()
-		if err != nil {
 		p.family = FamilyIPv6
 	} else {
-			panic(fmt.Sprintf("setAddress does not support %s", addr))
+		panic(fmt.Sprintf("setAddress does not support %q", addr))
-		}
 	}
 }
 
-func (p *packerHeader) AddressBytes() (b []byte) {
+func (p *packerHeader) AddressBytes() []byte {
-	b = make([]byte, 16)
+	if ip4 := p.address.To4(); ip4 != nil {
-
+		p.address = ip4
-	switch {
-	case p.address.To4() != nil:
-		b = make([]byte, 4)
-		for pos := range b {
-			b[pos] = p.address[pos+12]
 	}
-		return
+
-	}
+	return []byte(p.address)
-	return
 }
 
-func (p *packerHeader) Address() (address net.IP) {
+func (p *packerHeader) AddressString() string {
-	address = p.address
+	return p.address.String()
-	return
 }
 
-func (p *packerHeader) Family() (family addressFamily) {
+func (p *packerHeader) Address() net.IP {
-	family = p.family
+	return p.address
-	return
 }
 
-func (p *packerHeader) FamilyText() (familyText string) {
+func (p *packerHeader) Family() addressFamily {
-	familyText = addressFamilyText[p.family]
+	return p.family
-	return
+}
+
+func (p *packerHeader) FamilyText() string {
+	return addressFamilyText[p.family]
 }

rvpn/packer/setup.go

@@ -1,7 +1,9 @@
 package packer
 
-import "log"
+import (
-import "os"
+	"log"
+	"os"
+)
 
 var (
 	loginfo  *log.Logger