207 lines
5.9 KiB
Go
207 lines
5.9 KiB
Go
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
|
|
// Use of this source code is governed by an MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
module http
|
|
|
|
import time
|
|
|
|
#flag -lwininet -lShlwapi
|
|
#flag -lurlmon
|
|
#include <WinInet.h>
|
|
#include "urlmon.h"
|
|
#include <shlwapi.h>
|
|
// #LPWSTR winstring(string s);
|
|
// # bool ok = InternetReadFile(request, buf, BUF_MAX, &nr_read);
|
|
import const (
|
|
INTERNET_OPEN_TYPE_PRECONFIG
|
|
INTERNET_DEFAULT_HTTP_PORT
|
|
INTERNET_DEFAULT_HTTPS_PORT
|
|
INTERNET_SERVICE_HTTP
|
|
)
|
|
|
|
fn (req &Request) do() Response {
|
|
emptyresp := Response{}
|
|
mut url := req.url
|
|
//println('\n\nhttp.do() WIN URL="$url" TYP=$req.typ data="$req.data" headers.len=req.headers.len"')
|
|
//println(req.headers)
|
|
is_ssl := req.url.starts_with('https://')
|
|
//println('is ssl=$is_ssl')
|
|
mut pos := url.index('://')
|
|
if pos == -1 {return emptyresp}
|
|
url = url.right(pos + 3)
|
|
mut host := url
|
|
mut path := '/'
|
|
pos = url.index('/')
|
|
if pos > -1 {
|
|
host = url.left(pos)
|
|
host = host.clone()
|
|
path = url.right(pos)
|
|
}
|
|
// println('HOST="$host"')
|
|
// println('PATH="$path"')
|
|
mut headers := ''
|
|
mut resp_headers := ''
|
|
// for header in req.headers {
|
|
for key, val in req.headers {
|
|
headers += '$key: $val\r\n'
|
|
}
|
|
if req.typ == 'POST' {
|
|
headers += 'Content-Type: application/x-www-form-urlencoded'
|
|
}
|
|
// headers = headers.trim_space()
|
|
// println('!!! OLO REQ HEADERS WIN="$headers"')
|
|
data := req.data
|
|
// Retrieve default http user agent
|
|
// char httpUseragent[512];
|
|
// # char httpUseragent []= "";
|
|
user_agent := ''
|
|
// DWORD szhttpUserAgent = sizeof(httpUseragent);
|
|
// ObtainUserAgentString(0, httpUseragent, &szhttpUserAgent);
|
|
// # HINTERNET internet = InternetOpenA(httpUseragent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
|
|
internet := C.InternetOpenA(user_agent.str, INTERNET_OPEN_TYPE_PRECONFIG, 0, 0, 0)
|
|
// # if (!internet)
|
|
if isnil(internet) {
|
|
println('InternetOpen() failed')
|
|
return emptyresp
|
|
}
|
|
// # INTERNET_PORT port = INTERNET_DEFAULT_HTTP_PORT;
|
|
port := int(if is_ssl{INTERNET_DEFAULT_HTTPS_PORT} else { INTERNET_DEFAULT_HTTP_PORT})
|
|
// if is_ssl {
|
|
// # port = INTERNET_DEFAULT_HTTPS_PORT;
|
|
// }
|
|
connect := C.InternetConnectA(internet, host.str, port, 0, 0, INTERNET_SERVICE_HTTP, 0, 0)
|
|
// # HINTERNET connect = InternetConnectA(internet, host.str, port, NULL, NULL,
|
|
// # INTERNET_SERVICE_HTTP, 0, 0);
|
|
# if (!connect)
|
|
if isnil(connect) {
|
|
e := C.GetLastError()
|
|
println('[windows] InternetConnect() failed')
|
|
C.printf('err=%d\n', e)
|
|
return emptyresp
|
|
}
|
|
flags := 0
|
|
// # DWORD flags =
|
|
#flags =
|
|
# INTERNET_FLAG_HYPERLINK | INTERNET_FLAG_IGNORE_CERT_CN_INVALID |
|
|
# INTERNET_FLAG_IGNORE_CERT_DATE_INVALID |
|
|
# INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTP |
|
|
# INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTPS | INTERNET_FLAG_NO_AUTH |
|
|
# INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_NO_UI |
|
|
# INTERNET_FLAG_NO_COOKIES | // FUCK YOU MICROSOFT
|
|
# INTERNET_FLAG_KEEP_CONNECTION |
|
|
# INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_RELOAD ;
|
|
if is_ssl {
|
|
#flags = flags | INTERNET_FLAG_SECURE;
|
|
}
|
|
request := C.HttpOpenRequest(connect, req.typ.str, path.str, 'HTTP/1.1', 0, 0, flags, 0)
|
|
// request := C.InternetOpenUrl(connect, req.typ.str, path.str, 'HTTP/1.1', 0, 0, flags, 0)
|
|
// # HINTERNET request = HttpOpenRequest(connect, req->typ.str, path.str, "HTTP/1.1",
|
|
// # NULL, NULL, flags, NULL);
|
|
// # if (!request)
|
|
if isnil(request) {
|
|
println('HttpOpenRequest() failed')
|
|
return emptyresp
|
|
}
|
|
// println('LEN BEFORE SEND=$headers.len ; $headers')
|
|
# bool ret =HttpSendRequest(request, headers.str, -1, data.str, data.len);
|
|
// # printf("RET=%d\n", ret);
|
|
// # int e = GetLastError();
|
|
// # printf("e=%d\n", e);
|
|
// Get response headers
|
|
// Todo call twice to get len
|
|
# LPSTR h_buf = malloc(1024);
|
|
# DWORD dwSize = 1024;
|
|
// LPVOID lpOutBuffer=malloc(dwSize);
|
|
# HttpQueryInfo(request, HTTP_QUERY_RAW_HEADERS_CRLF,
|
|
# h_buf,&dwSize,NULL);
|
|
//# printf(" resp HEADERS %s\n", h_buf);
|
|
// Get response body
|
|
// # const int BUF_MAX = 1024;
|
|
// # TCHAR buf[BUF_MAX + 1];
|
|
mut buf := [1025]byte
|
|
mut nr_read := 0
|
|
BUF_MAX := 1024
|
|
// ok := C.InternetReadFile(request, buf, BUF_MAX, &nr_read)
|
|
// # DWORD dwRead = 0;
|
|
// /println('calling InternetReadFile()')
|
|
// # bool ok = InternetReadFile(request, buf, BUF_MAX, &nr_read);
|
|
// # if (!ok)
|
|
// {
|
|
// println('read not ok')
|
|
// # int e = GetLastError();
|
|
// # printf("%d\n", e);
|
|
// }
|
|
// # printf("dwread=%d\n", dwRead);
|
|
// # while ((InternetReadFile(request, buf, BUF_MAX, &nr_read)) && nr_read > 0)
|
|
mut s := ''
|
|
for
|
|
{
|
|
ok := C.InternetReadFile(request, buf, BUF_MAX, &nr_read)
|
|
if !ok {
|
|
println('InternetReadFile() not ok ')
|
|
}
|
|
if ok && nr_read == 0 {
|
|
//println('ok && nr read == 0, breaking')
|
|
//C.printf('buf broken="%s"\n', buf)
|
|
if req.url.contains('websocket') {
|
|
println('win sleeping 2')
|
|
time.sleep(2)
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
//println('ireadfile()')
|
|
buf[nr_read] = 0
|
|
//C.printf('buf="%s"\n', buf)
|
|
|
|
s += tos(buf, nr_read) // TODO perf
|
|
nr_read = 0
|
|
}
|
|
C.InternetCloseHandle(request)
|
|
C.InternetCloseHandle(connect)
|
|
C.InternetCloseHandle(internet)
|
|
# resp_headers = tos2(h_buf);
|
|
hh := resp_headers.split('\n')
|
|
mut resp := Response {
|
|
body: s
|
|
headers: map[string]string{}
|
|
// headers: resp_headers
|
|
}
|
|
// println('gen hh')
|
|
for h in hh {
|
|
// println('\n!')
|
|
// println(h)
|
|
vals := h.split(':')
|
|
hpos := h.index(':')
|
|
if hpos == -1 {
|
|
continue
|
|
}
|
|
key := h.left(hpos)
|
|
val := h.right(hpos + 1)
|
|
// println('$key => $val')
|
|
resp.headers[key] = val.trim_space()
|
|
}
|
|
//println('END OF WIN req.do($req.url)')
|
|
return resp
|
|
}
|
|
|
|
fn escape(s string) string {
|
|
# DWORD size=1;
|
|
# char *escaped = NULL;
|
|
# char *empty_string = NULL;
|
|
# HRESULT res = UrlEscapeA(s.str, empty_string, &size, URL_ESCAPE_PERCENT | URL_ESCAPE_SEGMENT_ONLY);
|
|
# if (res == E_POINTER)
|
|
{
|
|
# escaped = HeapAlloc(GetProcessHeap(), 0, size);
|
|
# if (!escaped)
|
|
# return s;
|
|
# UrlEscapeA(s.str, escaped, &size, URL_ESCAPE_PERCENT | URL_ESCAPE_SEGMENT_ONLY);
|
|
# return tos2(escaped);
|
|
}
|
|
return ''
|
|
}
|
|
|
|
fn C.InternetReadFile(voidptr, voidptr, int, intptr) bool
|
|
|