use internal check function for error checking

cmd/wireformat/wireformat.go

@@ -9,6 +9,8 @@ import (
 	"os"
 	"strings"
 
+	check "antoine-roux.ml/projects/go/dns-tools/internal"
+
 	"github.com/miekg/dns"
 )
 
@@ -24,26 +26,20 @@ func main() {
 	var err error
 	if *file != "" {
 		if *file == "-" {
-			*file = "/dev/stdin"
+			*file = os.Stdin.Name()
 		}
 		body, err = ioutil.ReadFile(*file)
-		if err != nil {
+		check.Check(err)
-			log.Fatalln(err)
-		}
 
 		msg := &dns.Msg{}
 		err = msg.Unpack(body)
-		if err != nil {
+		check.Check(err)
-			log.Fatalln(err)
-		}
 
 		log.Println(msg)
 	} else if *domainToResolve != "" {
 		if *domainToResolve == "-" {
-			readDomainToResolve, err := ioutil.ReadFile("/dev/stdin")
+			readDomainToResolve, err := ioutil.ReadFile(os.Stdin.Name())
-			if err != nil {
+			check.Check(err)
-				log.Fatalln(err)
-			}
 			*domainToResolve = strings.TrimSuffix(string(readDomainToResolve), "\n")
 		}
 		msg := &dns.Msg{
@@ -56,9 +52,7 @@ func main() {
 			},
 		}
 		b, err := msg.Pack()
-		if err != nil {
+		check.Check(err)
-			log.Fatalln(err)
-		}
 		binary.Write(os.Stdout, binary.LittleEndian, b)
 	} else {
 		log.Println(errors.New("Too few argument pass"))

go.mod

@@ -3,3 +3,5 @@ module antoine-roux.ml/projects/go/dns-tools
 go 1.15
 
 require github.com/miekg/dns v1.1.35
+
+replace antoine-roux.ml/projects/go/dns-tools/internal => ./internal

internal/check.go

@@ -0,0 +1,9 @@
+package check
+
+import "log"
+
+func Check(err error) {
+	if err != nil {
+		log.Fatalln(err)
+	}
+}