57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
|
package packer
|
||
|
|
|
||
|
|
import "net"
|
||
|
|
import "fmt"
|
||
|
|
|
||
|
|
// packerHeader structure to hold our header information.
|
||
|
|
type packerHeader struct {
|
||
|
|
family addressFamily
|
||
|
|
address net.IP
|
||
|
|
Port int
|
||
|
|
Service string
|
||
|
|
}
|
||
|
|
|
||
|
|
type addressFamily int
|
||
|
|
type addressFamilyString string
|
||
|
|
|
||
|
|
//Family -- ENUM for Address Family
|
||
|
|
const (
|
||
|
|
FamilyIPv4 addressFamily = iota
|
||
|
|
FamilyIPv6
|
||
|
|
)
|
||
|
|
|
||
|
|
func newPackerHeader() (p *packerHeader) {
|
||
|
|
p = new(packerHeader)
|
||
|
|
p.SetAddress("127.0.0.1")
|
||
|
|
p.Port = 65535
|
||
|
|
p.Service = "na"
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
//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 {
|
||
|
|
p.family = FamilyIPv4
|
||
|
|
} else {
|
||
|
|
err := p.address.To16()
|
||
|
|
if err != nil {
|
||
|
|
p.family = FamilyIPv6
|
||
|
|
} else {
|
||
|
|
panic(fmt.Sprintf("setAddress does not support %s", addr))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *packerHeader) Address() (address net.IP) {
|
||
|
|
address = p.address
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *packerHeader) Family() (family addressFamily) {
|
||
|
|
family = p.family
|
||
|
|
return
|
||
|
|
}
|