80 lines
1.4 KiB
Go
80 lines
1.4 KiB
Go
package packer
|
|
|
|
import "net"
|
|
import "fmt"
|
|
|
|
type addressFamily int
|
|
|
|
// packerHeader structure to hold our header information.
|
|
type packerHeader struct {
|
|
family addressFamily
|
|
address net.IP
|
|
Port int
|
|
Service string
|
|
}
|
|
|
|
//Family -- ENUM for Address Family
|
|
const (
|
|
FamilyIPv4 addressFamily = iota
|
|
FamilyIPv6
|
|
)
|
|
|
|
var addressFamilyText = [...]string{
|
|
"ipv4",
|
|
"ipv6",
|
|
}
|
|
|
|
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) AddressBytes() (b []byte) {
|
|
b = make([]byte, 16)
|
|
|
|
switch {
|
|
case p.address.To4() != nil:
|
|
b = make([]byte, 4)
|
|
for pos := range b {
|
|
b[pos] = p.address[pos+12]
|
|
}
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func (p *packerHeader) Address() (address net.IP) {
|
|
address = p.address
|
|
return
|
|
}
|
|
|
|
func (p *packerHeader) Family() (family addressFamily) {
|
|
family = p.family
|
|
return
|
|
}
|
|
|
|
func (p *packerHeader) FamilyText() (familyText string) {
|
|
familyText = addressFamilyText[p.family]
|
|
return
|
|
}
|