mirrors/netboot
1
0
Fork 0
mirror of https://github.com/danderson/netboot.git synced 2025-10-17 02:21:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

Made dhcpv6 packet easier to test

Browse Source
This commit is contained in:
Dmitri Dolguikh 2017-09-28 16:36:54 -07:00 committed by Dave Anderson
parent b6741d78b8
commit bf8701a8c5
2 changed files with 16 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
dhcp6/packet.go
View File

@ -5,6 +5,7 @@ import (
"net"
"encoding/binary"
"bytes"
"golang.org/x/tools/go/gcimporter15/testdata"
)
type MessageType uint8
@ -28,21 +29,16 @@ const (
type Packet struct {
Type MessageType
TransactionID [3]byte
Options []byte
Options Options
}
func MakePacket(bs []byte, len int) *Packet {
ret := &Packet{Type: MessageType(bs[0]), Options: make([]byte, len - 4)}
copy(ret.TransactionID[:], bs[1:4])
copy(ret.Options[:], bs[4:len])
return ret
}
func (p *Packet) UnmarshalOptions() (Options, error) {
ret, err := MakeOptions(p.Options)
func MakePacket(bs []byte, len int) (*Packet, error) {
options, err := MakeOptions(bs[4:]) // 4:len?
if err != nil {
return nil, fmt.Errorf("packet has malformed options section: %s", err)
}
ret := &Packet{Type: MessageType(bs[0]), Options: options}
copy(ret.TransactionID[:], bs[1:4])
return ret, nil
}
@ -62,7 +58,7 @@ func (p *Packet) BuildResponse(serverDuid []byte) ([]byte, error) {
}
func (p *Packet) BuildMsgAdvertise(serverDuid []byte) ([]byte, error) {
in_options, _ := p.UnmarshalOptions()
in_options := p.Options
ret_options := make(Options)
ret_options.AddOption(&Option{Id: OptClientId, Length: uint16(len(in_options[OptClientId].Value)), Value: in_options[OptClientId].Value})
@ -92,7 +88,7 @@ func (p *Packet) BuildMsgAdvertise(serverDuid []byte) ([]byte, error) {
// TODO: OptClientArchType may not be present
func (p *Packet) BuildMsgReply(serverDuid []byte) ([]byte, error) {
in_options, _ := p.UnmarshalOptions()
in_options := p.Options
ret_options := make(Options)
ret_options.AddOption(&Option{Id: OptClientId, Length: uint16(len(in_options[OptClientId].Value)), Value: in_options[OptClientId].Value})
@ -116,7 +112,7 @@ func (p *Packet) BuildMsgReply(serverDuid []byte) ([]byte, error) {
}
func (p *Packet) BuildMsgInformationRequestReply(serverDuid []byte) ([]byte, error) {
in_options, _ := p.UnmarshalOptions()
in_options := p.Options
ret_options := make(Options)
ret_options.AddOption(&Option{Id: OptClientId, Length: uint16(len(in_options[OptClientId].Value)), Value: in_options[OptClientId].Value})
@ -138,7 +134,7 @@ func (p *Packet) BuildMsgInformationRequestReply(serverDuid []byte) ([]byte, err
}
func (p *Packet) BuildMsgReleaseReply(serverDuid []byte) ([]byte, error){
in_options, _ := p.UnmarshalOptions()
in_options := p.Options
ret_options := make(Options)
ret_options.AddOption(&Option{Id: OptClientId, Length: uint16(len(in_options[OptClientId].Value)), Value: in_options[OptClientId].Value})
@ -172,7 +168,7 @@ func (p *Packet) ShouldDiscard(serverDuid []byte) error {
}
func ShouldDiscardSolicit(p *Packet) error {
options, _ := MakeOptions(p.Options)
options := p.Options
if !options.RequestedBootFileUrlOption() {
return fmt.Errorf("'Solicit' packet doesn't have file url option")
}
@ -186,7 +182,7 @@ func ShouldDiscardSolicit(p *Packet) error {
}
func ShouldDiscardRequest(p *Packet, serverDuid []byte) error {
options, _ := MakeOptions(p.Options)
options := p.Options
if !options.RequestedBootFileUrlOption() {
return fmt.Errorf("'Request' packet doesn't have file url option")
}
@ -203,7 +199,7 @@ func ShouldDiscardRequest(p *Packet, serverDuid []byte) error {
}
func ShouldDiscardInformationRequest(p *Packet, serverDuid []byte) error {
options, _ := MakeOptions(p.Options)
options := p.Options
if !options.RequestedBootFileUrlOption() {
return fmt.Errorf("'Information-request' packet doesn't have boot file url option")
}

7
pixiecore/dhcpv6.go
View File

@ -17,8 +17,7 @@ func (s *ServerV6) serveDHCP(conn *dhcp6.Conn) error {
continue
}
opts, _ := pkt.UnmarshalOptions()
s.log("dhcpv6", fmt.Sprintf("Received (%d) packet (%d): %s\n", pkt.Type, pkt.TransactionID, opts.HumanReadable()))
s.log("dhcpv6", fmt.Sprintf("Received (%d) packet (%d): %s\n", pkt.Type, pkt.TransactionID, pkt.Options.HumanReadable()))
response, _ := pkt.BuildResponse(s.Duid)
if err := conn.SendDHCP(src, response); err != nil {
@ -26,8 +25,8 @@ func (s *ServerV6) serveDHCP(conn *dhcp6.Conn) error {
continue
}
reply_packet := dhcp6.MakePacket(response, len(response))
reply_opts,_ := reply_packet.UnmarshalOptions()
reply_packet, _ := dhcp6.MakePacket(response, len(response))
reply_opts := reply_packet.Options
s.log("dhcpv6", fmt.Sprintf("Sent (%d) packet (%d): %s\n", reply_packet.Type, reply_packet.TransactionID, reply_opts.HumanReadable()))
}