Fix import reader

cybercongress · Jan 31, 2019 · 3b57173 · 3b57173
1 parent 64e5516
commit 3b57173
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/util/file.go b/util/file.go
@@ -18,12 +18,17 @@ func RootifyPath(path string) string {
 
 func ReadExactlyNBytes(reader io.Reader, n uint64) ([]byte, error) {
 	data := make([]byte, n)
-	bytesReaded, err := reader.Read(data)
-	if err != nil {
-		return nil, err
-	}
-	if uint64(bytesReaded) != n {
-		return nil, errors.New("not enough bytes tor read")
+	currentlyReadedBytes := uint64(0)
+	for currentlyReadedBytes < n {
+		readedBytes, err := reader.Read(data[currentlyReadedBytes:n])
+		if err != nil {
+			return nil, err
+		}
+		if readedBytes == 0 {
+			return nil, errors.New("not enough bytes tor read")
+		}
+		currentlyReadedBytes += uint64(readedBytes)
 	}
+
 	return data, nil
 }