From c3434f38325856dcdc19adc29b5e31944614221b Mon Sep 17 00:00:00 2001 From: Pablo Mazzini Date: Mon, 30 Jul 2018 19:05:13 +0200 Subject: fix OptIPAddressLeaseTime --- dhcpv4/option_ip_address_lease_time.go | 2 +- dhcpv4/option_ip_address_lease_time_test.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'dhcpv4') diff --git a/dhcpv4/option_ip_address_lease_time.go b/dhcpv4/option_ip_address_lease_time.go index a171b04..7562c58 100644 --- a/dhcpv4/option_ip_address_lease_time.go +++ b/dhcpv4/option_ip_address_lease_time.go @@ -17,7 +17,7 @@ type OptIPAddressLeaseTime struct { // sequence of bytes and returns it, or an error. func ParseOptIPAddressLeaseTime(data []byte) (*OptIPAddressLeaseTime, error) { // Should at least have code, length, and lease time. - if len(data) < 4 { + if len(data) < 6 { return nil, ErrShortByteStream } code := OptionCode(data[0]) diff --git a/dhcpv4/option_ip_address_lease_time_test.go b/dhcpv4/option_ip_address_lease_time_test.go index 66715bd..7d507bf 100644 --- a/dhcpv4/option_ip_address_lease_time_test.go +++ b/dhcpv4/option_ip_address_lease_time_test.go @@ -20,7 +20,7 @@ func TestParseOptIPAddressLeaseTime(t *testing.T) { require.Equal(t, &OptIPAddressLeaseTime{LeaseTime: 43200}, o) // Short byte stream - data = []byte{51, 4} + data = []byte{51, 4, 168, 192} _, err = ParseOptIPAddressLeaseTime(data) require.Error(t, err, "should get error from short byte stream") -- cgit v1.2.3