func TestHttpServer(t *testing.T) {
http_req_worker := func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, WsWorker) {
if req != nil {
resp := HttpUtils.MakeResponse(200, "OK", req.HeaderParams, req.Body)
return &resp, nil
}
resp := HttpUtils.MakeResponse(400, "Bad request", nil, nil)
return &resp, nil
}
srv, err := RunNewServer("127.0.0.1", 45000, http_req_worker)
if srv != nil {
defer srv.Close()
if err != nil {
t.Fatalf("Ошибка запуска сервера %s, но возвращён ненулевой интерфейс\n", err.Error())
}
} else {
if err != nil {
t.Fatalf("Ошибка запуска сервера %s\n", err.Error())
} else {
t.Fatal("Ошибка запуска сервера: результат nil, nil")
}
}
// for n := uint64(0); n < 1000; n++ {
// OneConnectionCheck("127.0.0.1:45000", n, t)
// }
for num := 0; num < 10; num++ {
var wg sync.WaitGroup
for step := 0; step < 1000; step++ {
wg.Add(1)
N := uint64(step)
go func() {
defer wg.Done()
OneConnectionCheck("127.0.0.1:45000", N, t)
}()
}
wg.Wait()
}
} // func TestHttpServer(t *testing.T)
func TestHttpWebsockServer(t *testing.T) {
http_req_worker := func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, WsWorker) {
if req != nil {
var ws_worker WsWorker
var resp HttpUtils.HttpResponse
hparams_map := make(map[string]string)
for _, p := range req.HeaderParams {
hparams_map[p.Name] = p.Value
}
check := false
for _, p := range strings.Split(hparams_map["Upgrade"], ", ") {
if p == "websocket" {
check = true
break
}
}
if check {
check = false
for _, p := range strings.Split(hparams_map["Connection"], ", ") {
if p == "Upgrade" {
check = true
break
}
}
if check {
key, check := hparams_map["Sec-WebSocket-Key"]
if check {
// Переходим на протокол веб-сокетов
confirm_key := WebsocketUtils.GetAcceptKey(key)
resp_hparams := make([]HttpUtils.HeaderParam, 3)
resp_hparams[0] = HttpUtils.HeaderParam{Name: "Upgrade", Value: "websocket"}
resp_hparams[1] = HttpUtils.HeaderParam{Name: "Connection", Value: "Upgrade"}
resp_hparams[2] = HttpUtils.HeaderParam{Name: "Sec-WebSocket-Accept", Value: confirm_key}
resp = HttpUtils.MakeResponse(101, "Switching protocols", resp_hparams, nil)
}
}
ws_worker = &echo_ws_worker{}
} else {
t.Error("Некорректный запрос (ожидаем только запрос переключения на вебсокет)")
resp = HttpUtils.MakeResponse(200, "OK", req.HeaderParams, req.Body)
}
return &resp, ws_worker
} // if req != nil
resp := HttpUtils.MakeResponse(400, "Bad request", nil, nil)
return &resp, nil
} // http_req_worker := func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, WsWorker)
srv, err := RunNewServer("127.0.0.1", 45000, http_req_worker)
if srv != nil {
defer srv.Close()
if err != nil {
t.Fatalf("Ошибка запуска сервера %s, но возвращён ненулевой интерфейс\n", err.Error())
}
} else {
if err != nil {
t.Fatalf("Ошибка запуска сервера %s\n", err.Error())
} else {
t.Fatal("Ошибка запуска сервера: результат nil, nil")
}
}
for n := uint64(0); n < 100; n++ {
OneWebsockConnectionCheck("127.0.0.1:45000", n, t)
}
for num := 0; num < 10; num++ {
var wg sync.WaitGroup
for step := 0; step < 1000; step++ {
wg.Add(1)
N := uint64(step)
go func() {
defer wg.Done()
OneWebsockConnectionCheck("127.0.0.1:45000", N, t)
}()
}
wg.Wait()
}
} // func TestHttpWebsockServer(t *testing.T)
// type HttpWorker func(*HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, WsWorker)
func make_handler(root_dir string) func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, CommonWebServer.WsWorker) {
resp_bad_req := HttpUtils.MakeResponse(400, "Bad request", nil, nil)
return func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, CommonWebServer.WsWorker) {
if req == nil {
return &resp_bad_req, nil
}
hparams_map := make(map[string]string)
for _, p := range req.HeaderParams {
hparams_map[p.Name] = p.Value
}
check := false
for _, p := range strings.Split(hparams_map["Upgrade"], ", ") {
if p == "websocket" {
check = true
break
}
}
if check {
check = false
for _, p := range strings.Split(hparams_map["Connection"], ", ") {
if p == "Upgrade" {
check = true
break
}
}
if check {
key, check := hparams_map["Sec-WebSocket-Key"]
if check {
// Переходим на протокол веб-сокетов
confirm_key := WebsocketUtils.GetAcceptKey(key)
resp_hparams := make([]HttpUtils.HeaderParam, 3)
resp_hparams[0] = HttpUtils.HeaderParam{Name: "Upgrade", Value: "websocket"}
resp_hparams[1] = HttpUtils.HeaderParam{Name: "Connection", Value: "Upgrade"}
resp_hparams[2] = HttpUtils.HeaderParam{Name: "Sec-WebSocket-Accept", Value: confirm_key}
resp := HttpUtils.MakeResponse(101, "Switching protocols", resp_hparams, nil)
return &resp, &EchoWsWorker{}
}
}
}
if req.Host == "/" {
h_params := []HttpUtils.HeaderParam{HttpUtils.HeaderParam{Name: "Location", Value: "/index.html"}}
resp := HttpUtils.MakeResponse(301, "Moved permanently", h_params, nil)
return &resp, nil
}
var ftype string = ""
switch path.Ext(req.Host) {
case "html":
ftype = "text/html"
case "txt":
ftype = "text/plain"
case "png":
ftype = "image/png"
case "ico":
ftype = "image/x-icon"
}
f, e := os.Open(root_dir + req.Host)
if e != nil {
resp := HttpUtils.MakeResponse(404, "Нету!", nil, nil)
return &resp, nil
} else {
defer f.Close()
}
var resp HttpUtils.HttpResponse
hexademical := "0123456789ABCDEF"
switch req.Type {
case "POST":
s_body := string(req.Body)
var req_body string
for p := strings.Index(s_body, "%"); p >= 0; p = strings.Index(s_body, "%") {
v := make([]byte, 1)
v[0] = byte(strings.Index(hexademical, string(s_body[p+1]))) << 4
v[0] |= byte(strings.Index(hexademical, string(s_body[p+2])))
req_body = strings.Join([]string{req_body, s_body[:p], string(v)}, "")
s_body = s_body[p+3:]
}
req_body = strings.Join([]string{req_body, s_body}, "")
params := strings.Split(string(req_body), "&")
resp_body := `<!DOCTYPE html>
<html lang="ru">
<head>
<meta charset="UTF-8">
<title>Введённые данные</title>
<link rel="icon" href="favicon.ico" type="image/x-icon">
</head>
<body>
<p><h1>Вы ввели:</h1></p>`
for _, p := range params {
resp_body += strings.Join([]string{"<p>", p, "</p>"}, "")
}
resp_body += `<p><a href="index.html">Перейти на главную страницу</a></p>
<p><a href="form.html">Перейти на страницу с формой</a></p>
</body>
</html>`
body := []byte(resp_body)
header_params := make([]HttpUtils.HeaderParam, 2)
header_params[0] = HttpUtils.HeaderParam{Name: HttpUtils.BodySizeParamName, Value: strconv.Itoa(len(body))}
header_params[1] = HttpUtils.HeaderParam{Name: "Content-Type", Value: "text/html"}
resp = HttpUtils.MakeResponse(200, "OK", header_params, body)
case "GET":
body := make([]byte, 0, 100)
buf := make([]byte, 100)
for e == nil {
sz, e := f.Read(buf)
if sz == 0 {
break
} else if (e == nil) || (e == io.EOF) {
body = append(body, buf[:sz]...)
} else {
resp = HttpUtils.MakeResponse(500, e.Error(), nil, nil)
}
}
if (e != nil) && (e != io.EOF) {
panic("Ошибка, которой быть не должно: " + e.Error())
}
header_params := make([]HttpUtils.HeaderParam, 1, 2)
header_params[0] = HttpUtils.HeaderParam{Name: HttpUtils.BodySizeParamName, Value: strconv.Itoa(len(body))}
if ftype != "" {
header_params = append(header_params, HttpUtils.HeaderParam{Name: "Content-Type", Value: ftype})
}
resp = HttpUtils.MakeResponse(200, "OK", header_params, body)
default:
resp = HttpUtils.MakeResponse(501, "Not supported", nil, nil)
}
return &resp, nil
} // return func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, WsWorker)
} // func make_handler(root_dir string) func(req *HttpUtils.HttpRequest) (*HttpUtils.HttpResponse, WsWorker)
