summaryrefslogtreecommitdiff
path: root/vendor/golang.org/x/net/ipv4/packet_go1_8.go
blob: b47d186834d8dd86d251cbfec2e319a02654fcec (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright 2012 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// +build !go1.9

package ipv4

import "net"

func (c *packetHandler) readFrom(b []byte) (h *Header, p []byte, cm *ControlMessage, err error) {
	c.rawOpt.RLock()
	oob := NewControlMessage(c.rawOpt.cflags)
	c.rawOpt.RUnlock()
	n, nn, _, src, err := c.ReadMsgIP(b, oob)
	if err != nil {
		return nil, nil, nil, err
	}
	var hs []byte
	if hs, p, err = slicePacket(b[:n]); err != nil {
		return nil, nil, nil, err
	}
	if h, err = ParseHeader(hs); err != nil {
		return nil, nil, nil, err
	}
	if nn > 0 {
		cm = new(ControlMessage)
		if err := cm.Parse(oob[:nn]); err != nil {
			return nil, nil, nil, err
		}
	}
	if src != nil && cm != nil {
		cm.Src = src.IP
	}
	return
}

func (c *packetHandler) writeTo(h *Header, p []byte, cm *ControlMessage) error {
	oob := cm.Marshal()
	wh, err := h.Marshal()
	if err != nil {
		return err
	}
	dst := new(net.IPAddr)
	if cm != nil {
		if ip := cm.Dst.To4(); ip != nil {
			dst.IP = ip
		}
	}
	if dst.IP == nil {
		dst.IP = h.Dst
	}
	wh = append(wh, p...)
	_, _, err = c.WriteMsgIP(wh, oob, dst)
	return err
}