57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
/* SPDX-License-Identifier: MIT
|
|
*
|
|
* Copyright (C) 2017-2020 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
package device
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"golang.zx2c4.com/wireguard/conn"
|
|
)
|
|
|
|
type DummyDatagram struct {
|
|
msg []byte
|
|
endpoint conn.Endpoint
|
|
}
|
|
|
|
type DummyBind struct {
|
|
in6 chan DummyDatagram
|
|
in4 chan DummyDatagram
|
|
closed bool
|
|
}
|
|
|
|
func (b *DummyBind) SetMark(v uint32) error {
|
|
return nil
|
|
}
|
|
|
|
func (b *DummyBind) ReceiveIPv6(buff []byte) (int, conn.Endpoint, error) {
|
|
datagram, ok := <-b.in6
|
|
if !ok {
|
|
return 0, nil, errors.New("closed")
|
|
}
|
|
copy(buff, datagram.msg)
|
|
return len(datagram.msg), datagram.endpoint, nil
|
|
}
|
|
|
|
func (b *DummyBind) ReceiveIPv4(buff []byte) (int, conn.Endpoint, error) {
|
|
datagram, ok := <-b.in4
|
|
if !ok {
|
|
return 0, nil, errors.New("closed")
|
|
}
|
|
copy(buff, datagram.msg)
|
|
return len(datagram.msg), datagram.endpoint, nil
|
|
}
|
|
|
|
func (b *DummyBind) Close() error {
|
|
close(b.in6)
|
|
close(b.in4)
|
|
b.closed = true
|
|
return nil
|
|
}
|
|
|
|
func (b *DummyBind) Send(buff []byte, end conn.Endpoint) error {
|
|
return nil
|
|
}
|