update command

Author: Ro0tk1t

cmd/command.go

@@ -1,8 +1,11 @@
 package cmd
 
 import (
+    "os"
     "fmt"
     "github.com/spf13/cobra"
+
+    log "github.com/sirupsen/logrus"
 )
 
 
@@ -14,7 +17,8 @@ var (
 )
 
 var rootCmd = &cobra.Command{
-    Use: "",
+    Use: "bomb",
+    Args: cobra.NoArgs,
 }
 
 var versionCmd = &cobra.Command{
@@ -25,7 +29,14 @@ var versionCmd = &cobra.Command{
 
 
 func Execute(){
-    rootCmd.Execute()
+    if err := rootCmd.Execute(); err != nil {
+        log.Fatal(err)
+    }
+    // the function MarkFlagRequired looks like dosen't work,
+    // so judg it by hand
+    if Phone == "" {
+        log.Fatal("phone is empty")
+    }
 }
 
 func init(){
@@ -34,9 +45,12 @@ func init(){
     rootCmd.Flags().StringVarP(&Phone, "phone", "p", "", "phone number")
     rootCmd.Flags().StringVarP(&AreaCode, "areacode", "a", "", "the phone number area code, example: +86")
     rootCmd.AddCommand(versionCmd)
+
+    rootCmd.MarkFlagRequired("phone")
 }
 
 
 func showVersion(cmd *cobra.Command, args []string) {
     fmt.Println("version 1.0.0")
+    os.Exit(0)
 }

core/core.go

@@ -29,7 +29,7 @@ func bomb(api *models.API) {
     for {
         time.Sleep(time.Duration(gap)* time.Second)
         fmt.Printf("bomb %s with %s\n", cmd.Phone, api.Name)
-        //sender.Run(api)
+        sender.Run(*api)
     }
 }
 

core/sender.go

@@ -11,54 +11,17 @@ import (
     "bomb/models"
 )
 
-type Send struct {}
 
 type HttpSender struct {
-    Send
 }
 
 type HttpsSender struct {
-    Send
 }
 
 type TcpSender struct {
-    Send
-}
-
-func (s Send) Init () {
-    // TODO
-}
-
-//func (s *Send) SendData (api models.API) {
-//    fmt.Println(api.Url)
-//    // TODO
-//}
-
-func (s Send) Recv () {
-    //TODO
-}
-
-func (s *Send) Run (api models.API) {
-//    s.SendData(api)
-}
-
-
-func NewSender(way string) models.Sender {
-    switch way {
-    case "http":
-        return &HttpSender{}
-    case "https": {
-        return &HttpsSender{}
-    };
-    case "tcp": {
-        return &TcpSender{}
-    };
-    default: {
-        return &HttpSender{}
-    }
-    }
 }
 
+func (h HttpSender) Init() {}
 
 func (h *HttpSender) SendData (api models.API) {
     switch api.Method {
@@ -124,10 +87,18 @@ func (h *HttpSender) SendData (api models.API) {
     }
 }
 
+func (h *HttpSender) Run (api models.API) {
+    h.SendData(api)
+}
+
+func (h *HttpsSender) Init () {}
+
 func (h *HttpsSender) SendData (api models.API) {
     switch api.Method {
         case 0: {
-            tr := &http.Transport{}
+            tr := &http.Transport{
+                TLSClientConfig:&tls.Config{InsecureSkipVerify: true},
+            }
             client := &http.Client{Transport: tr}
             req, err := http.NewRequest("GET", api.Url, nil)
             if err != nil {
@@ -175,17 +146,34 @@ func (h *HttpsSender) SendData (api models.API) {
     }
 }
 
-func (h *HttpsSender) Init () {}
-func (h *HttpsSender) Recv () {}
 func (h *HttpsSender) Run (api models.API) {
     h.SendData(api)
 }
-func (h *HttpSender) Run (api models.API) {
-    h.SendData(api)
-}
+
+func (t *TcpSender) Init () {}
 
 func (t *TcpSender) SendData (api models.API) {
     fmt.Println(api.Url)
     // TODO
 }
 
+func (t *TcpSender) Run (api models.API) {
+    t.SendData(api)
+}
+
+func NewSender(way string) models.Sender {
+    switch way {
+    case "http":
+        return &HttpSender{}
+    case "https": {
+        return &HttpsSender{}
+    };
+    case "tcp": {
+        return &TcpSender{}
+    };
+    default: {
+        return &HttpSender{}
+    }
+    }
+}
+

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/coreos/go-etcd v2.0.0+incompatible // indirect
 	github.com/cpuguy83/go-md2man v1.0.10 // indirect
 	github.com/mitchellh/go-homedir v1.1.0 // indirect
+	github.com/sirupsen/logrus v1.8.1 // indirect
 	github.com/spf13/cobra v1.2.1 // indirect
 	github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8 // indirect
 	github.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77 // indirect

go.sum

@@ -209,6 +209,8 @@ github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQD
 github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts=
 github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=
 github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
+github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=
+github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
 github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
 github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
 github.com/spf13/afero v1.6.0 h1:xoax2sJ2DT8S8xA2paPFjDCScCNeWsg75VG0DLRreiY=
@@ -366,6 +368,7 @@ golang.org/x/sys v0.0.0-20190624142023-c5567b49c5d0/go.mod h1:h1NjWce9XRLGQEsW7w
 golang.org/x/sys v0.0.0-20190726091711-fc99dfbffb4e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191001151750-bb3f8db39f24/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191204072324-ce4227a45e2e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20191228213918-04cbcbbfeed8/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20200113162924-86b910548bc1/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=

models/models.go

@@ -34,7 +34,7 @@ type API struct {
 type Sender interface {
     Init()
     SendData(API)
-    Recv()
+    //Recv()
     Run(API)
 }
 

utils/utils.go

@@ -31,10 +31,8 @@ func LoadDatas(){
             newDatas = append(newDatas, api)
         }
     }
-    conf.Datas = newDatas
 
-    fmt.Println(conf.Datas[0].GetMethod())
-    fmt.Println(conf.Datas[0].GetDataType())
+    conf.Datas = newDatas
 }