From bb7b4c59ad8be634c15e83d949525bdad632fca5 Mon Sep 17 00:00:00 2001
From: Stefano <stefano.scafiti96@gmail.com>
Date: Thu, 3 Feb 2022 13:35:27 +0100
Subject: [PATCH] Add flag allow-unknown-authority to get command. Print usage
 when no file url has been specified

---
 main.go | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index d5c2090..c0c325f 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"bufio"
+	"crypto/x509"
 	"flag"
 	"fmt"
 	"os"
@@ -94,11 +95,18 @@ func transferLoop(transfer *XdccTransfer) {
 	// TODO: do clean-up operations here
 }
 
+func suggestUnknownAuthoritySwitch(err error) {
+	if err.Error() == (x509.UnknownAuthorityError{}.Error()) {
+		fmt.Println("use the --allow-unknown-authority flag to skip certificate verification")
+	}
+}
+
 func doTransfer(transfer *XdccTransfer) {
 	err := transfer.Start()
 
 	if err != nil {
 		fmt.Println(err)
+		suggestUnknownAuthoritySwitch(err)
 		return
 	}
 
@@ -145,11 +153,19 @@ func loadUrlListFile(filePath string) []string {
 	return urlList
 }
 
+func printGetUsageAndExit(flagSet *flag.FlagSet) {
+	fmt.Printf("usage: get url1 url2 ... [-o path] [-i file] [--allow-unknown-authority]\n\nFlag set:\n")
+	flagSet.PrintDefaults()
+	os.Exit(0)
+}
+
 func getCommand(args []string) {
 	getCmd := flag.NewFlagSet("get", flag.ExitOnError)
 	path := getCmd.String("o", ".", "output folder of dowloaded file")
 	inputFile := getCmd.String("i", "", "input file containing a list of urls")
 
+	skipCertificateCheck := getCmd.Bool("allow-unknown-authority", false, "skip x509 certificate check during tls connection")
+
 	urlList := parseFlags(getCmd, args)
 
 	if *inputFile != "" {
@@ -157,8 +173,7 @@ func getCommand(args []string) {
 	}
 
 	if len(urlList) == 0 {
-		fmt.Println("no file url provided")
-		os.Exit(1)
+		printGetUsageAndExit(getCmd)
 	}
 
 	wg := sync.WaitGroup{}
@@ -172,7 +187,7 @@ func getCommand(args []string) {
 			}
 
 			wg.Add(1)
-			transfer := NewXdccTransfer(*url, *path)
+			transfer := NewXdccTransfer(*url, *path, *skipCertificateCheck)
 			go func(transfer *XdccTransfer) {
 				doTransfer(transfer)
 				wg.Done()