Use cancellable crypto RNG from u-root

Fixes #246 Signed-off-by: Andrea Barberio <[email protected]>
insomniacslk · Apr 3, 2019 · 625d653 · 625d653
1 parent 68ad195
commit 625d653
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/dhcpv4/dhcpv4.go b/dhcpv4/dhcpv4.go
@@ -17,7 +17,6 @@ package dhcpv4
 
 import (
 	"bytes"
-	"crypto/rand"
 	"errors"
 	"fmt"
 	"net"
@@ -26,6 +25,7 @@ import (
 
 	"github.com/insomniacslk/dhcp/iana"
 	"github.com/insomniacslk/dhcp/rfc1035label"
+	"github.com/u-root/u-root/pkg/rand"
 	"github.com/u-root/u-root/pkg/uio"
 )
 

diff --git a/dhcpv6/dhcpv6_test.go b/dhcpv6/dhcpv6_test.go
@@ -1,16 +1,15 @@
 package dhcpv6
 
 import (
-	"crypto/rand"
 	"encoding/binary"
 	"errors"
 	"net"
 	"testing"
 
+	"github.com/insomniacslk/dhcp/iana"
 	"github.com/stretchr/testify/require"
 	"github.com/stretchr/testify/suite"
-
-	"github.com/insomniacslk/dhcp/iana"
+	"github.com/u-root/u-root/pkg/rand"
 )
 
 func randomReadMock(value []byte, n int, err error) func([]byte) (int, error) {

diff --git a/dhcpv6/dhcpv6message.go b/dhcpv6/dhcpv6message.go
@@ -1,13 +1,13 @@
 package dhcpv6
 
 import (
-	"crypto/rand"
 	"errors"
 	"fmt"
 	"net"
 	"time"
 
 	"github.com/insomniacslk/dhcp/iana"
+	"github.com/u-root/u-root/pkg/rand"
 	"github.com/u-root/u-root/pkg/uio"
 )