1 files changed, 113 insertions, 0 deletions
diff --git a/rcon.c b/rcon.c
new file mode 100644
index 0000000..75e850d
--- /dev/null
+++ b/rcon.c
@@ -0,0 +1,113 @@
+/*
+ * Adopted from mcrcon, Copyright (c) 2012-2020, Tiiffi <tiiffi at gmail>.
+ * https://github.com/Tiiffi/mcrcon/tree/b02201d689b3032bc681b28f175fd3d83d167293
+ */
+
+#include "rcon.h"
+#include "common.h"
+
+#include <sys/socket.h>
+#include <sysexits.h>
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+
+int rcon_send_packet(int sd, struct rc_packet *packet)
+{
+        int len;
+        int total = 0;  // bytes we've sent
+        int bytesleft;  // bytes left to send
+        int ret = -1;
+
+        bytesleft = len = packet->size + sizeof(int);
+
+        while (total < len)
+	{
+                ret = send(sd, (char *) packet + total, bytesleft, 0);
+                if(ret == -1)
+		{
+			fprintf(stderr, _("send(): %s.\n"), strerror(errno));
+			return EX_IOERR;
+		}
+                total += ret;
+                bytesleft -= ret;
+        }
+
+	return EX_OK;
+}
+
+int rcon_build_packet(struct rc_packet *out, int id, int cmd, char *s1)
+{
+        // size + id + cmd + s1 + s2 NULL terminator
+        int s1_len = strlen(s1);
+        if (s1_len > RCON_DATA_BUFFSIZE)
+	{
+                fprintf(stderr, _("Warning: Command string too long (%d). Maximum allowed: %d.\n"), s1_len, RCON_DATA_BUFFSIZE);
+                return EX_DATAERR;
+        }
+
+        out->size = sizeof(int) * 2 + s1_len + 2;
+        out->id = id;
+        out->cmd = cmd;
+        strncpy(out->data, s1, RCON_DATA_BUFFSIZE);
+
+	return EX_OK;
+}
+
+int rcon_recv_packet(struct rc_packet *out, int sd)
+{
+        int psize;
+
+        int ret = recv(sd, (char *) &psize, sizeof(int), 0);
+
+        if (ret == 0)
+	{
+                fprintf(stderr, _("Connection lost.\n"));
+                return EX_IOERR;
+        }
+	if(ret == -1)
+	{
+		fprintf(stderr, _("recv(): %d\n"), errno);
+		return EX_IOERR;
+	}
+
+        if (ret != sizeof(int))
+	{
+                fprintf(stderr, _("Error: recv() failed. Invalid packet size (%d).\n"), ret);
+                return EX_IOERR;
+        }
+
+        if (psize < 10 || psize > RCON_DATA_BUFFSIZE)
+	{
+                fprintf(stderr, _("Warning: invalid packet size (%d). Must over 10 and less than %d.\n"), psize, RCON_DATA_BUFFSIZE);
+
+                if(psize > RCON_DATA_BUFFSIZE  || psize < 0) psize = RCON_DATA_BUFFSIZE;
+		// Former net_clean_incoming.
+		char tmp[psize];
+		ret = recv(sd, tmp, psize, 0);
+
+	        if(ret == 0)
+		{
+			fprintf(stderr, _("Connection lost.\n"));
+		}
+
+                return EX_DATAERR;
+        }
+
+        out->size = psize;
+
+        int received = 0;
+        while (received < psize)
+	{
+                ret = recv(sd, (char *) out + sizeof(int) + received, psize - received, 0);
+                if (ret == 0) // connection closed before completing receving
+		{
+                        fprintf(stderr, _("Connection lost.\n"));
+                        return EX_IOERR;
+                }
+
+                received += ret;
+        }
+
+        return EX_OK;
+}