2024-05-24 22:03:55 +02:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2024-05-24 23:55:58 +02:00
|
|
|
"log"
|
2024-05-25 00:22:42 +02:00
|
|
|
"regexp"
|
|
|
|
"strconv"
|
2024-05-24 23:55:58 +02:00
|
|
|
"strings"
|
|
|
|
|
2024-05-24 22:03:55 +02:00
|
|
|
// Uncomment this block to pass the first stage
|
2024-05-24 22:46:34 +02:00
|
|
|
"net"
|
|
|
|
"os"
|
2024-05-24 22:03:55 +02:00
|
|
|
)
|
|
|
|
|
2024-05-24 23:55:58 +02:00
|
|
|
const (
|
|
|
|
HTTPDelimiter = "\r\n"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Header struct {
|
|
|
|
Name string
|
|
|
|
Value string
|
|
|
|
}
|
|
|
|
type Request struct {
|
|
|
|
Method string
|
|
|
|
Path string
|
|
|
|
Version string
|
|
|
|
Body string
|
|
|
|
Headers []Header
|
|
|
|
}
|
|
|
|
|
2024-05-25 00:22:42 +02:00
|
|
|
type HttpCode struct {
|
2024-05-24 23:55:58 +02:00
|
|
|
Code uint
|
|
|
|
Message string
|
|
|
|
}
|
|
|
|
|
|
|
|
var (
|
2024-05-25 00:22:42 +02:00
|
|
|
BadRequest = HttpCode{Code: 400, Message: "Bad Response"}
|
|
|
|
NotFound = HttpCode{Code: 404, Message: "Not Found"}
|
|
|
|
OK = HttpCode{Code: 200, Message: "OK"}
|
2024-05-24 23:55:58 +02:00
|
|
|
)
|
|
|
|
|
2024-05-25 00:22:42 +02:00
|
|
|
type Response struct {
|
|
|
|
Code HttpCode
|
|
|
|
Version string
|
|
|
|
Body string
|
|
|
|
Headers []Header
|
|
|
|
}
|
|
|
|
|
|
|
|
func Respond(conn net.Conn, response Response) {
|
|
|
|
fmt.Fprintf(conn, "%s %d %s%s", response.Version, response.Code.Code, response.Code.Message, HTTPDelimiter)
|
|
|
|
bodySize := len(response.Body)
|
|
|
|
if bodySize > 0 {
|
|
|
|
response.Headers = append(response.Headers, Header{Name: "Content-Length", Value: strconv.Itoa(bodySize)})
|
|
|
|
}
|
|
|
|
for _, header := range response.Headers {
|
|
|
|
fmt.Fprintf(conn, "%s: %s%s", header.Name, header.Value, HTTPDelimiter)
|
|
|
|
}
|
2024-05-24 23:55:58 +02:00
|
|
|
|
2024-05-25 00:22:42 +02:00
|
|
|
fmt.Fprint(conn, HTTPDelimiter)
|
|
|
|
if bodySize > 0 {
|
|
|
|
fmt.Fprint(conn, response.Body)
|
|
|
|
}
|
|
|
|
|
|
|
|
// fmt.Fprintf(conn, "HTTP/1.1 %d %s%s%s", response.Code, response.Message, HTTPDelimiter, HTTPDelimiter)
|
2024-05-24 23:55:58 +02:00
|
|
|
}
|
|
|
|
|
2024-05-24 22:56:28 +02:00
|
|
|
func handleConnection(conn net.Conn) {
|
2024-05-24 23:55:58 +02:00
|
|
|
defer conn.Close()
|
|
|
|
buffer := make([]byte, 1024)
|
|
|
|
n, err := conn.Read(buffer)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
contents := string(buffer[:n])
|
|
|
|
parts := strings.Split(contents, HTTPDelimiter)
|
|
|
|
request := Request{}
|
|
|
|
isBody := false
|
|
|
|
for i, part := range parts {
|
|
|
|
if i == 0 {
|
|
|
|
head := strings.Split(part, " ")
|
|
|
|
if len(head) != 3 {
|
2024-05-25 00:22:42 +02:00
|
|
|
Respond(conn, Response{Version: "HTTP/1.1", Code: BadRequest})
|
2024-05-24 23:55:58 +02:00
|
|
|
return
|
|
|
|
}
|
|
|
|
request.Method = head[0]
|
|
|
|
request.Path = head[1]
|
|
|
|
request.Version = head[2]
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
if isBody {
|
|
|
|
request.Body = part
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
// Headers
|
|
|
|
if part == "" {
|
|
|
|
isBody = true
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
h := strings.SplitN(part, ": ", 2)
|
|
|
|
header := Header{Name: h[0], Value: h[1]}
|
|
|
|
request.Headers = append(request.Headers, header)
|
|
|
|
}
|
|
|
|
fmt.Println(request)
|
|
|
|
|
|
|
|
if request.Path == "/" {
|
2024-05-25 00:22:42 +02:00
|
|
|
Respond(conn, Response{Version: request.Version, Code: OK})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
re := regexp.MustCompile(`^/echo/([A-Za-z]+)$`)
|
|
|
|
matches := re.FindStringSubmatch(request.Path)
|
|
|
|
if len(matches) > 0 {
|
|
|
|
Respond(conn, Response{Version: request.Version, Code: OK, Body: matches[1]})
|
2024-05-24 23:55:58 +02:00
|
|
|
return
|
|
|
|
}
|
2024-05-25 00:22:42 +02:00
|
|
|
|
|
|
|
Respond(conn, Response{Version: request.Version, Code: NotFound})
|
2024-05-24 22:56:28 +02:00
|
|
|
}
|
|
|
|
|
2024-05-24 22:03:55 +02:00
|
|
|
func main() {
|
|
|
|
// You can use print statements as follows for debugging, they'll be visible when running tests.
|
|
|
|
fmt.Println("Logs from your program will appear here!")
|
|
|
|
|
|
|
|
// Uncomment this block to pass the first stage
|
|
|
|
//
|
2024-05-24 22:46:34 +02:00
|
|
|
l, err := net.Listen("tcp", "0.0.0.0:4221")
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println("Failed to bind to port 4221")
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
|
2024-05-24 22:56:28 +02:00
|
|
|
for {
|
|
|
|
conn, err := l.Accept()
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println("Error accepting connection: ", err.Error())
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
go handleConnection(conn)
|
2024-05-24 22:46:34 +02:00
|
|
|
}
|
2024-05-24 22:56:28 +02:00
|
|
|
|
2024-05-24 22:03:55 +02:00
|
|
|
}
|