summaryrefslogtreecommitdiff
path: root/os-win.c
diff options
context:
space:
mode:
authorpommicket <pommicket@gmail.com>2023-09-10 13:40:05 -0400
committerpommicket <pommicket@gmail.com>2023-09-10 13:40:05 -0400
commit6586f9e66d87017984f57b3b4579c78897ba66f2 (patch)
treef84ab9a9b4a6449e123cfed0a3e9245750a19fd6 /os-win.c
parent5e22cc67e92a87a0448dea70c89a4472b961fa14 (diff)
LSP over TCP on windows
Diffstat (limited to 'os-win.c')
-rw-r--r--os-win.c104
1 files changed, 104 insertions, 0 deletions
diff --git a/os-win.c b/os-win.c
index 3f93c53..fe08be3 100644
--- a/os-win.c
+++ b/os-win.c
@@ -431,3 +431,107 @@ bool open_with_default_application(const char *path) {
bool change_directory(const char *path) {
return _chdir(path) == 0;
}
+
+struct Socket {
+ SOCKET sock;
+ char error[256];
+};
+
+Socket *socket_connect_tcp(const char *address, u16 port) {
+ Socket *s = calloc(1, sizeof *s);
+ if (!s) return NULL;
+ s->sock = INVALID_SOCKET;
+ if (!address) address = "127.0.0.1";
+
+
+ struct sockaddr_in addr = {
+ .sin_family = AF_INET,
+ .sin_port = htons(port),
+ };
+ if (inet_pton(AF_INET, address, &addr.sin_addr) != 1) {
+ strbuf_printf(s->error, "Invalid address");
+ return s;
+ }
+
+ SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);
+ if (sock == INVALID_SOCKET) {
+ strbuf_printf(s->error, "Couldn't create socket (error code %d)",
+ WSAGetLastError());
+ return s;
+ }
+ if (connect(sock, (SOCKADDR *)&addr, sizeof addr) != 0) {
+ strbuf_printf(s->error, "Couldn't connect to %s:%u (error code %d)",
+ address, port, WSAGetLastError());
+ closesocket(sock);
+ return s;
+ }
+
+ u_long mode = 1; // non-blocking
+ ioctlsocket(sock, FIONBIO, &mode);
+ s->sock = sock;
+ return s;
+}
+const char *socket_get_error(Socket *s) {
+ return s->error;
+}
+
+long long socket_read(Socket *s, char *data, size_t size) {
+ if (size > INT_MAX) {
+ strbuf_printf(s->error, "Too much data to read.");
+ return -2;
+ }
+ size_t so_far = 0;
+ while (so_far < size) {
+ int bytes_read = recv(s->sock, data + so_far, (int)(size - so_far), 0);
+ if (bytes_read > 0) {
+ so_far += (size_t)bytes_read;
+ } else if (bytes_read == 0) {
+ return (long long)so_far;
+ } else {
+ int err = WSAGetLastError();
+ if (err == WSAEWOULDBLOCK) {
+ return so_far == 0 ? -1 : (long long)so_far;
+ } else if (err == WSAECONNRESET) {
+ return 0;
+ } else {
+ strbuf_printf(s->error, "recv failed (error code %d)", err);
+ return -2;
+ }
+ }
+ }
+ return (long long)size;
+}
+
+long long socket_write(Socket *s, const char *data, size_t size) {
+ if (size > INT_MAX) {
+ strbuf_printf(s->error, "too much data to write");
+ return -2;
+ }
+ size_t so_far = 0;
+ while (so_far < size) {
+ int bytes_written = send(s->sock, data + so_far, (int)(size - so_far), 0);
+ if (bytes_written > 0) {
+ so_far += (size_t)bytes_written;
+ } else {
+ int err = WSAGetLastError();
+ if (err == WSAEWOULDBLOCK) {
+ return (long long)so_far;
+ } else if (err == WSAEHOSTUNREACH || err == WSAECONNRESET) {
+ return -1;
+ } else {
+ strbuf_printf(s->error, "write failed (error code %d)", err);
+ return -2;
+ }
+ }
+ }
+ return (long long)size;
+}
+
+void socket_close(Socket **psocket) {
+ Socket *s = *psocket;
+ if (!s) return;
+ if (s->sock != INVALID_SOCKET)
+ closesocket(s->sock);
+ free(s);
+ *psocket = NULL;
+}