Correction réception fichier, ajout méthode util Network.go

pkg/utils/FileReceiver.go

@@ -8,24 +8,31 @@ import (
 
 // ReceiveFile Permet de récupérer un fichier sur un reader
 func ReceiveFile(fileName string, fileSize int, reader *bufio.Reader) bool {
-	file, fileErr := os.Create(fmt.Sprintf("/home/benjamin/sbe/%s", fileName))
+	file, fileErr := os.Create(fmt.Sprintf("D:\\sbe\\%s", fileName))
 	if fileErr != nil {
 		return false
 	}
-
 	defer file.Close()
-	currentSize := 0
-	buffer := make([]byte, 1024)
 
+	// Params
+	buffer := make([]byte, 1024)
+	currentSize := 0
+
+	// Retrieving file
 	for currentSize < fileSize {
 		length, err := reader.Read(buffer)
 		if err != nil {
 			return false
 		}
-
+		_, err = file.WriteAt(buffer, int64(currentSize))
+		if err != nil {
+			return false
+		}
 		currentSize += length
-		file.WriteAt(buffer, int64(currentSize))
+		check := fileSize - currentSize
+		if 1024 > check && check > 0 {
+			buffer = make([]byte, check)
+		}
 	}
-
 	return true
 }

pkg/utils/Network.go

@@ -3,8 +3,10 @@ package utils
 import (
 	"fmt"
 	"net"
+	"strings"
 )
 
+// NetworkLister List all network interface available
 func NetworkLister() {
 	// Retrieve Interfaces
 	inter, err := net.Interfaces()
@@ -22,3 +24,8 @@ func NetworkLister() {
 	}
 	print("\n")
 }
+
+// IsIPv4 Check if the ip is a v4 or v6 ip
+func IsIPv4(addr string) bool {
+	return strings.Count(addr, ":") < 2
+}