Hochwasser/main.go

package main

import (
	"flag"
	"net"
	"runtime/pprof"
	"time"
)

import "fmt"
import "image/png"
import "log"
import "os"
import "strconv"
import _ "net/http/pprof"

var err error
var cpuprofile = flag.String("cpuprofile", "", "Destination file for CPU Profile")
var image = flag.String("image", "", "Absolute Path to image")
var canvas_xsize = flag.Int("xsize", 800, "Width of the canvas in px")
var canvas_ysize = flag.Int("ysize", 600, "Height of the canvas in px")
var image_offsetx = flag.Int("xoffset", 0, "Offset of posted image from left border")
var image_offsety = flag.Int("yoffset", 0, "Offset of posted image from top border")
var connections = flag.Int("connections", 10, "Number of simultaneous connections/threads. Each Thread posts a subimage")
var address = flag.String("host", "127.0.0.1:1337", "Server address")
var runtime = flag.String("runtime", "1", "Runtime in Minutes")

func main() {
	flag.Parse()
	if *image == "" || *address == "" {
		log.Fatal("No image or no server address provided")
	}

	// Start cpu profiling if wanted
	if *cpuprofile != "" {
		f, err := os.Create(*cpuprofile)
		if err != nil {
			log.Fatal(err)
		}
		pprof.StartCPUProfile(f)
		defer pprof.StopCPUProfile()
	}
	// Generate and split messages into equal chunks
	msg := splitmessages(genMessages())
	for _, message := range msg {
		go bomb(message)
	}

	// Terminate after 1 Minute to save resources
	timer, err := time.ParseDuration(*runtime + "m")
	if err != nil {
		log.Fatal("Invalid runtime specified: " + err.Error())
	}
	time.Sleep(time.Minute * timer)
}

func bomb(messages []byte) {
	conn, err := net.Dial("tcp", *address)

	if err != nil {
		log.Print("error establishing tcp connection: " + err.Error())
	}
	//TODO: Actually close the connection and not just terminate main thread
	defer conn.Close()

	// Start bombardement
	for {
		_, err := conn.Write(messages)
		if err != nil {
			log.Println(err.Error())
		}
	}
}

// Creates message based on given image
func genMessages() (output []byte) {
	reader, err := os.Open(*image)
	if err != nil {
		log.Fatal(err)
	}

	img, err2 := png.Decode(reader)
	if err2 != nil {
		log.Fatal(err2)
	}

	for x := img.Bounds().Max.X; x != 0; x-- {
		for y := img.Bounds().Max.Y; y != 0; y-- {
			col := img.At(x, y)
			r, g, b, _ := col.RGBA()

			rStr := strconv.FormatInt(int64(r), 16)
			if len(rStr) == 1 {
				rStr = "0" + rStr
			}

			gStr := strconv.FormatInt(int64(g), 16)
			if len(gStr) == 1 {
				gStr = "0" + gStr
			}

			bStr := strconv.FormatInt(int64(b), 16)
			if len(bStr) == 1 {
				bStr = "0" + bStr
			}

			colStr := rStr[0:2]
			colStr += gStr[0:2]
			colStr += bStr[0:2]

			//Do not draw transparent pixels
			if colStr == "000000" {
				continue
			}
			pxStr := fmt.Sprintf("PX %d %d %s\n", x+*image_offsetx, y+*image_offsety, colStr)
			output = append(output, []byte(pxStr)...)
		}
	}
	return output
}

// Splits messages into chunks, splitting on complete commands only
func splitmessages(in []byte) [][]byte {
	index := 0
	equalsplit := len(in) / *connections
	output := make([][]byte, *connections)
	for i := 0; i < *connections; i++ {
		if index+equalsplit > len(in) {
			output[i] = in[index:]
			break
		}

		tmp := index
		for in[index+equalsplit] != 80 {
			index++
		}
		output[i] = in[tmp : index+equalsplit]
		index += equalsplit
	}
	return output
}