Go pot

gopot.go

package main

import (
	"bufio"
	"fmt"
	"log"
	"net"
	"time"
)
var lport = ":8883"
var replyDummy = true

func main()  {
	sock, err := net.Listen("tcp", lport)
	if err != nil {
		log.Fatal(err)
	}
	log.Printf("Listening on %s", lport)
	for {
		conn, err := sock.Accept()
		if err != nil {
			continue
		}
		go handleConnection(conn)
	}
}

func handleConnection(conn net.Conn) {
	defer conn.Close()
	err := conn.SetReadDeadline(time.Now().Add(2*time.Second))
	if err != nil {
		log.Fatal(err)
	}
	receiveBuffer := []byte{}
	scanner := bufio.NewScanner(conn)
	for scanner.Scan() {
		if scanner.Text() == "" {
			if replyDummy { writeDummy(conn) }
			conn.Close()
			break
		}
		receiveBuffer = append(receiveBuffer, scanner.Bytes()...)
		receiveBuffer = append(receiveBuffer, '\n')
		if len(receiveBuffer) > 128*1024 {
			conn.Close()
			break
		}
	}
	fmt.Println(string(receiveBuffer))
}

func writeDummy(conn net.Conn) {
	err := conn.SetWriteDeadline(time.Now().Add(2*time.Second))
	if err != nil {
		return
	}
	conn.Write([]byte("HTTP/1.1 200 OK\r\n\r\nCucumber\r\n"))
}