diff options
Diffstat (limited to 'device/send.go')
| -rw-r--r-- | device/send.go | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/device/send.go b/device/send.go index 72633be..9e29d77 100644 --- a/device/send.go +++ b/device/send.go @@ -515,10 +515,18 @@ func (device *Device) RoutineEncryption() { // pad content to multiple of 16 mtu := int(atomic.LoadInt32(&device.tun.mtu)) - lastUnit := len(elem.packet) % mtu - paddedSize := (lastUnit + PaddingMultiple - 1) & ^(PaddingMultiple - 1) - if paddedSize > mtu { - paddedSize = mtu + var paddedSize int + if mtu == 0 { + paddedSize = (len(elem.packet) + PaddingMultiple - 1) & ^(PaddingMultiple - 1) + } else { + lastUnit := len(elem.packet) + if lastUnit > mtu { + lastUnit %= mtu + } + paddedSize := (lastUnit + PaddingMultiple - 1) & ^(PaddingMultiple - 1) + if paddedSize > mtu { + paddedSize = mtu + } } for i := len(elem.packet); i < paddedSize; i++ { elem.packet = append(elem.packet, 0) |