Skip to content

Commit

Permalink
add version check to ui
Browse files Browse the repository at this point in the history
  • Loading branch information
@Dregu
Dregu committed Nov 9, 2023
1 parent cec8389 commit b89f0ec
Show file tree
Hide file tree
Showing 5 changed files with 388 additions and 227 deletions.
192 changes: 4 additions & 188 deletions src/game_api/script/usertypes/socket_lua.cpp
View file
Original file line number Diff line number Diff line change
@@ -1,184 +1,13 @@
#include "socket_lua.hpp"
#include "socket.hpp"

#include <Windows.h> // for GetModuleHandleA, GetProcAddress
#include <algorithm> // for max
#include <detours.h> // for DetourAttach, DetourTransactionBegin
#include <exception> // for exception
#include <new> // for operator new
#include <sockpp/inet_address.h> // for inet_address
#include <sockpp/udp_socket.h> // for udp_socket
#include <sol/sol.hpp> // for global_table, proxy_key_t, function
#include <sys/types.h> // for ssize_t
#include <thread> // for thread
#include <tuple> // for get
#include <type_traits> // for move
#include <utility> // for max, min
#include <wininet.h> // for InternetCloseHandle, InternetOpenA, InternetG...
#include <winsock2.h> // for sockaddr_in, SOCKET
#include <ws2tcpip.h> // for inet_ntop
#include <sol/sol.hpp> // for global_table, proxy_key_t, function
#include <sys/types.h> // for ssize_t

#include "logger.h" // for DEBUG, ByteStr
#include "script/lua_backend.hpp" // for LuaBackend
#include "script/safe_cb.hpp" // for make_safe_cb

#pragma comment(lib, "wininet.lib")

void udp_data(sockpp::udp_socket socket, UdpServer* server)
{
ssize_t n;
char buf[500];
sockpp::inet_address src;
while (server->kill_thr.test(std::memory_order_acquire) && (n = socket.recv_from(buf, sizeof(buf), &src)) > 0)
{
std::optional<std::string> ret = server->cb(std::string(buf, n));
if (ret)
{
socket.send_to(ret.value(), src);
}
}
}

UdpServer::UdpServer(std::string host_, in_port_t port_, std::function<SocketCb> cb_)
: host(host_), port(port_), cb(cb_)
{
sock.bind(sockpp::inet_address(host, port));
kill_thr.test_and_set();
thr = std::thread(udp_data, std::move(sock), this);
}
void UdpServer::clear() // TODO: fix and expose: this and the destructor causes deadlock
{
kill_thr.clear(std::memory_order_release);
thr.join();
}
UdpServer::~UdpServer()
{
if (thr.joinable())
{
kill_thr.clear(std::memory_order_release);
thr.join();
}
}

using NetFun = int(SOCKET, char*, int, int, sockaddr_in*, int*);
NetFun* g_sendto_trampoline{nullptr};
NetFun* g_recvfrom_trampoline{nullptr};
int mySendto(SOCKET s, char* buf, int len, int flags, sockaddr_in* addr, int* tolen)
{
auto ret = g_sendto_trampoline(s, buf, len, flags, addr, tolen);
char ip[16] = "";
inet_ntop(addr->sin_family, &addr->sin_addr, ip, sizeof(ip));
DEBUG("SEND: {}:{} | {}", ip, addr->sin_port, ByteStr{buf});
return ret;
}

int myRecvfrom(SOCKET s, char* buf, int len, int flags, sockaddr_in* addr, int* fromlen)
{
auto ret = g_recvfrom_trampoline(s, buf, len, flags, addr, fromlen);
char ip[16] = "";
inet_ntop(addr->sin_family, &addr->sin_addr, ip, sizeof(ip));
DEBUG("RECV: {}:{} | {}", ip, addr->sin_port, ByteStr{buf});
return ret;
}

bool http_get(const char* sURL, std::string& out, std::string& err)
{
const int BUFFER_SIZE = 32768;
DWORD iFlags;
const char* sAgent = "curl";
const char* sHeader = NULL;
HINTERNET hInternet;
HINTERNET hConnect;
char acBuffer[BUFFER_SIZE];
DWORD iReadBytes;
DWORD iBytesToRead = 0;
DWORD iReadBytesOfRq = 4;

// Get connection state
InternetGetConnectedState(&iFlags, 0);
if (iFlags & INTERNET_CONNECTION_OFFLINE)
{
err = "Can't connect to the internet";
return false;
}

// Open internet session
if (!(iFlags & INTERNET_CONNECTION_PROXY))
{
hInternet = InternetOpenA(sAgent, INTERNET_OPEN_TYPE_PRECONFIG_WITH_NO_AUTOPROXY, NULL, NULL, 0);
}
else
{
hInternet = InternetOpenA(sAgent, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
}
if (hInternet)
{
if (sHeader == NULL)
{
sHeader = "Accept: */*\r\n\r\n";
}

hConnect = InternetOpenUrlA(hInternet, sURL, sHeader, lstrlenA(sHeader), INTERNET_FLAG_DONT_CACHE | INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_RELOAD, 0);
if (!hConnect)
{
InternetCloseHandle(hInternet);
err = "Can't connect to the url";
return false;
}

// Get content size
if (!HttpQueryInfo(hConnect, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, (LPVOID)&iBytesToRead, &iReadBytesOfRq, NULL))
{
iBytesToRead = 0;
}

do
{
if (!InternetReadFile(hConnect, acBuffer, BUFFER_SIZE, &iReadBytes))
{
InternetCloseHandle(hInternet);
err = "GET request failed";
return false;
}
if (iReadBytes > 0)
{
out += std::string(acBuffer, iReadBytes);
}
if (iReadBytes <= 0)
{
break;
}
} while (TRUE);
InternetCloseHandle(hInternet);
}
else
{
err = "Can't connect to the internet";
return false;
}

return true;
}

void http_get_async(HttpRequest* req)
{
if (http_get(req->url.c_str(), req->response, req->error))
{
req->cb(req->response, std::nullopt);
}
else
{
req->cb(std::nullopt, req->error);
}
delete req;
}

HttpRequest::HttpRequest(std::string url_, std::function<HttpCb> cb_)
: url(url_), cb(cb_)
{
std::thread thr(http_get_async, this);
thr.detach();
}

namespace NSocket
{
void register_usertypes(sol::state& lua)
Expand All @@ -204,20 +33,7 @@ void register_usertypes(sol::state& lua)
};

/// Hook the sendto and recvfrom functions and start dumping network data to terminal
lua["dump_network"] = []()
{
g_sendto_trampoline = (NetFun*)GetProcAddress(GetModuleHandleA("ws2_32.dll"), "sendto");
g_recvfrom_trampoline = (NetFun*)GetProcAddress(GetModuleHandleA("ws2_32.dll"), "recvfrom");
DetourTransactionBegin();
DetourUpdateThread(GetCurrentThread());
DetourAttach((void**)&g_sendto_trampoline, mySendto);
DetourAttach((void**)&g_recvfrom_trampoline, myRecvfrom);
const LONG error = DetourTransactionCommit();
if (error != NO_ERROR)
{
DEBUG("Failed hooking network: {}\n", error);
}
};
lua["dump_network"] = dump_network;

/// Send a synchronous HTTP GET request and return response as a string or nil on an error
lua["http_get"] = [&lua](std::string url) -> sol::optional<std::string>
Expand Down
38 changes: 1 addition & 37 deletions src/game_api/script/usertypes/socket_lua.hpp
View file
Original file line number Diff line number Diff line change
@@ -1,42 +1,6 @@
#pragma once

#include <atomic> // for atomic_flag
#include <functional> // for function
#include <sockpp/platform.h> // for in_port_t
#include <sol/sol.hpp> // for state, optional
#include <string> // for string
#include <thread> // for thread

#include "sockpp/udp_socket.h" // for udp_socket

class UdpServer
{
public:
using SocketCb = std::optional<std::string>(std::string);

UdpServer(std::string host, in_port_t port, std::function<SocketCb> cb);
~UdpServer();
void clear();

std::string host;
in_port_t port;
std::function<SocketCb> cb;
std::thread thr;
std::atomic_flag kill_thr;
sockpp::udp_socket sock;
};

class HttpRequest
{
public:
using HttpCb = void(std::optional<std::string>, std::optional<std::string>);

HttpRequest(std::string url, std::function<HttpCb> cb);
std::string url;
std::function<HttpCb> cb;
std::string response;
std::string error;
};
#include <sol/sol.hpp> // for state, optional

namespace NSocket
{
Expand Down
Loading

0 comments on commit b89f0ec

Please sign in to comment.