133 lines
3 KiB
C++
133 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2016-2020 Marco Hladik <marco@icculus.org>
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
|
|
* IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
|
|
* OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
#define TCP_BUFFER_LENGTH 32
|
|
|
|
typedef struct
|
|
{
|
|
filestream m_fSocket;
|
|
string m_strBuffer[TCP_BUFFER_LENGTH];
|
|
int m_iBufferLines;
|
|
int m_iState;
|
|
} tcpinfo_t;
|
|
|
|
enum
|
|
{
|
|
STATE_DISCONNECTED,
|
|
STATE_CONNECTING,
|
|
STATE_CONNECTED
|
|
};
|
|
|
|
int
|
|
TCP_Connect(tcpinfo_t *in, string path)
|
|
{
|
|
in.m_fSocket = fopen(path, -1);
|
|
|
|
if (in.m_fSocket < 0) {
|
|
dprint(sprintf("^1TCP_Connect^7: Unable to access %s\n", path));
|
|
} else {
|
|
/* we got at least this far */
|
|
in.m_iState = STATE_CONNECTING;
|
|
}
|
|
|
|
return (int)in.m_fSocket;
|
|
}
|
|
|
|
void
|
|
TCP_Disconnect(tcpinfo_t *in)
|
|
{
|
|
if (in.m_iState == STATE_DISCONNECTED)
|
|
return;
|
|
|
|
in.m_iState = STATE_DISCONNECTED;
|
|
fclose(in.m_fSocket);
|
|
in.m_fSocket = -1;
|
|
}
|
|
|
|
int
|
|
TCP_Send(tcpinfo_t *in, string msg)
|
|
{
|
|
if (in.m_iState == STATE_DISCONNECTED) {
|
|
dprint("^1TCP_Send^7: Can't send, we're disconnected\n");
|
|
return 0;
|
|
}
|
|
|
|
/* the buffer is full. we've been unable to send messages for a while */
|
|
if (in.m_iBufferLines >= (TCP_BUFFER_LENGTH - 1)) {
|
|
dprint("^1TCP_Send^7: Buffer full.\n");
|
|
return 0;
|
|
}
|
|
|
|
in.m_strBuffer[in.m_iBufferLines] = msg;
|
|
in.m_iBufferLines++;
|
|
return 1;
|
|
}
|
|
|
|
string
|
|
TCP_Receive(tcpinfo_t *in)
|
|
{
|
|
/* not even open */
|
|
if (in.m_fSocket <= 0)
|
|
return "";
|
|
|
|
return fgets(in.m_fSocket);
|
|
}
|
|
|
|
/* you want to run this every frame */
|
|
void
|
|
TCP_Frame(tcpinfo_t *in)
|
|
{
|
|
string out;
|
|
|
|
/* not even open */
|
|
if (in.m_iState == STATE_DISCONNECTED)
|
|
return;
|
|
|
|
/* got nothing worth sending */
|
|
if (in.m_strBuffer[0] == "" || in.m_iBufferLines <= 0)
|
|
return;
|
|
|
|
/* if the send was unsuccessful, try next frame */
|
|
out = (string)memalloc(strlen(in.m_strBuffer[0]));
|
|
|
|
for (float i = 0; i < strlen(in.m_strBuffer[0]); i++)
|
|
out[i] = str2chr(in.m_strBuffer[0], i);
|
|
|
|
if (fwrite(in.m_fSocket, (void *)out, strlen(in.m_strBuffer[0])) <= 0) {
|
|
dprint("^1TCP_Frame^7: Unsuccessful frame\n");
|
|
memfree((__variant *)out);
|
|
return;
|
|
}
|
|
|
|
print(sprintf("^1TCP_Frame^7: Sent %s", in.m_strBuffer[0]));
|
|
|
|
/* at least one successful write went through */
|
|
in.m_iState = STATE_CONNECTED;
|
|
|
|
/* a send was successful after here */
|
|
for (int i = 0; i < (TCP_BUFFER_LENGTH - 1); i++) {
|
|
in.m_strBuffer[i] = in.m_strBuffer[i+1];
|
|
}
|
|
|
|
in.m_iBufferLines--;
|
|
memfree((__variant *)out);
|
|
}
|
|
|
|
int
|
|
TCP_GetState(tcpinfo_t *in)
|
|
{
|
|
return in.m_iState;
|
|
}
|