pasteholejones/termbin.go
2022-05-25 17:46:18 +09:30

55 lines
919 B
Go

package main
import (
"log"
"net"
"time"
)
func handleTermbinConn(conn net.Conn) {
defer conn.Close()
tmp := make([]byte, 1024)
data := make([]byte, 0)
for {
conn.SetReadDeadline(time.Now().Add(time.Second * 1))
if len(data) > 1024*1024*5 {
conn.Write([]byte("5MiB size limit exceeded. Please try again.\n"))
return
}
n, err := conn.Read(tmp)
if err != nil {
break
}
data = append(data, tmp[:n]...)
}
p, err := NewPaste(data)
if err != nil {
conn.Write([]byte("Error occurred writing your paste\n"))
return
}
conn.Write([]byte("https://p.fuckhole.org/" + p.Uid + "\nAll Pastehole Jones expire in 24 hours.\n"))
return
}
func listenTermbin() {
sock, err := net.Listen("tcp", "0.0.0.0:9999")
if err != nil {
log.Fatal(err)
}
defer sock.Close()
for {
conn, err := sock.Accept()
if err != nil {
log.Println(err)
return
}
go handleTermbinConn(conn)
}
}