podpora downServera v gameServery ;)

1 files changed, 312 insertions, 0 deletions

diff --git a/src/base/downServer.c b/src/base/downServer.c
new file mode 100644
index 0000000..f0625a9
--- /dev/null
+++ b/src/base/downServer.c
@@ -0,0 +1,312 @@
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <sys/types.h>
+
+#include <unistd.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <sys/time.h>
+#include <sys/stat.h>
+
+#include "main.h"
+#include "list.h"
+#include "server.h"
+#include "myTimer.h"
+#include "arena.h"
+#include "serverSelect.h"
+#include "downServer.h"
+#include "arenaFile.h"
+
+#include "tcp.h"
+#include "buffer.h"
+
+#define DOWN_SERVER_LIMIT_BUFFER	10240
+#define DOWN_SERVER_SEND_FILE_BUFFER	4096
+
+#define NET_STATUS_OK		0
+#define NET_STATUS_ZOMBIE	1
+
+static sock_tcp_t *sock_server_tcp;
+static list_t *listClient;
+
+typedef struct client_down_server_struct
+{
+	int status_connect;
+	sock_tcp_t *sock;
+	buffer_t *recvBuffer;
+	buffer_t *sendBuffer;
+
+	FILE *file;
+	int size;
+	int send;
+} client_ds_t;
+
+static int getFileSize(char *s)
+{
+	struct stat buf;
+
+	if( lstat(s, &buf) < 0 )
+	{
+		return -1;
+	}
+
+	return buf.st_size;
+}
+
+static void proto_getarena(client_ds_t *client, char *msg)
+{
+	arenaFile_t *arenaFile;
+	char msg_out[STR_PROTO_SIZE];
+	char *arenaName;
+
+	arenaName = msg+9;
+
+	//printf("arena = >>%s<<\n", arenaName);
+	arenaFile = getArenaFileFormNetName(arenaName);
+
+	if( arenaFile == NULL )
+	{
+		char *error_msg = _("ERR Arena not found !\n");
+		addBuffer(client->sendBuffer, error_msg, strlen(error_msg));
+		return;
+	}
+
+	client->file = fopen(arenaFile->path, "rb");
+
+	if( client->file == NULL )
+	{
+		char *error_msg = _("ERR Server has a problem !\n");
+		addBuffer(client->sendBuffer, error_msg, strlen(error_msg));
+		return;
+	}
+
+	client->size = getFileSize(arenaFile->path);
+	sprintf(msg_out, "OK %d\n", client->size);
+	addBuffer(client->sendBuffer, msg_out, strlen(msg_out));
+}
+
+static void do_proto(client_ds_t *client, char *msg)
+{
+	if( strncmp(msg, "GETARENA ", 9) == 0 && client->file == NULL )
+	{
+		proto_getarena(client, msg);
+	}
+}
+
+static client_ds_t* newClient(sock_tcp_t *sock)
+{
+	client_ds_t *new;
+
+	new = malloc( sizeof(client_ds_t) );
+	memset(new, 0, sizeof(client_ds_t));
+
+	new->status_connect = NET_STATUS_OK;
+	new->sock = sock;
+	new->recvBuffer = newBuffer(DOWN_SERVER_LIMIT_BUFFER);
+	new->sendBuffer = newBuffer(DOWN_SERVER_LIMIT_BUFFER);
+
+	return new;
+}
+
+static void destroyClient(client_ds_t *client)
+{
+	closeTcpSocket(client->sock);
+	destroyBuffer(client->recvBuffer);
+	destroyBuffer(client->sendBuffer);
+
+	if( client->file != NULL )
+	{
+		fclose(client->file);
+	}
+
+	free(client);
+}
+
+int initDownServer(char *ip4, int port4)
+{
+	sock_server_tcp = bindTcpSocket(ip4, port4, PROTO_TCPv4);
+
+	if( sock_server_tcp == NULL )
+	{
+		fprintf(stderr, _("down server do not inicialized !\n"));
+		return -1;
+	}
+
+	setTcpSockNonBlock(sock_server_tcp);
+
+	listClient = newList();
+
+	return 0;
+}
+
+int setDownServerSelect()
+{
+	int i;
+
+	addSockToSelectRead(sock_server_tcp->sock);
+
+	for( i = 0 ; i < listClient->count ; i++ )
+	{
+		client_ds_t *this;
+
+		this = (client_ds_t *)listClient->list[i];
+
+		addSockToSelectRead(this->sock->sock);
+
+		if( getBufferSize(this->sendBuffer) > 0 )
+		{
+			addSockToSelectWrite(this->sock->sock);
+		}
+	}
+
+	return 0;
+}
+
+static void eventNewClient(sock_tcp_t *sock_server)
+{
+	sock_tcp_t *sock_client;
+	client_ds_t *client;
+
+	sock_client = getTcpNewClient(sock_server);
+	setTcpSockNonBlock(sock_client);
+	client = newClient(sock_client);
+
+	addList(listClient, client);
+}
+
+static void eventReadClient(client_ds_t * client)
+{
+	char buffer[STR_PROTO_SIZE];
+	int ret;
+
+	memset(buffer, 0, STR_PROTO_SIZE);
+
+	ret = readTcpSocket(client->sock, buffer, STR_PROTO_SIZE - 1);
+	//printf("readTcpSocket = %d\n", ret);
+
+	if (ret <= 0)
+	{
+		client->status_connect = NET_STATUS_ZOMBIE;
+		return;
+	}
+
+	if( addBuffer(client->recvBuffer, buffer, ret) < 0 )
+	{
+		client->status_connect = NET_STATUS_ZOMBIE;
+		return;
+	}
+
+	while( getBufferLine(client->recvBuffer, buffer, STR_PROTO_SIZE) >= 0 )
+	{
+		int len;
+
+		len = strlen(buffer);
+
+		if( buffer[len-1] == '\n' )
+		{
+			buffer[len-1] = '\0';
+		}
+
+		do_proto(client, buffer);
+		//addBuffer(client->sendBuffer, buffer, strlen(buffer));
+	}
+}
+
+static void eventWriteClient(client_ds_t *client)
+{
+	void *data;
+	int len;
+	int res;
+
+	len = getBufferSize(client->sendBuffer);
+
+	if( len == 0 )
+	{
+		return;
+	}
+
+	data = getBufferData(client->sendBuffer);
+
+	res = writeTcpSocket(client->sock, data, len);
+	//printf("writeTcpSocket = %d\n", res);
+
+	if( res < 0 )
+	{
+		client->status_connect = NET_STATUS_ZOMBIE;
+		return;
+	}
+
+	cutBuffer(client->sendBuffer, res);
+}
+
+static void sendFile(client_ds_t *client)
+{
+	char buffer[DOWN_SERVER_SEND_FILE_BUFFER];
+	int count;
+
+	count = client->size - client->send;
+
+	if( count > DOWN_SERVER_SEND_FILE_BUFFER )
+	{
+		count = DOWN_SERVER_SEND_FILE_BUFFER;
+	}
+
+	if( fread(buffer, count, 1, client->file) != 1 )
+	{
+		client->status_connect = NET_STATUS_ZOMBIE;
+		return;
+	}
+
+	addBuffer(client->sendBuffer, buffer, count);
+	client->send += count;
+}
+
+int selectDownServerSocket()
+{
+	int i;
+
+	if( isChangeSockInSelectRead(sock_server_tcp->sock) )
+	{
+		eventNewClient(sock_server_tcp);
+	}
+
+	for( i = 0 ; i < listClient->count ; i++ )
+	{
+		client_ds_t *this;
+
+		this = (client_ds_t *)listClient->list[i];
+
+		if( isChangeSockInSelectRead(this->sock->sock) )
+		{
+			eventReadClient(this);
+		}
+
+		if( isChangeSockInSelectWrite(this->sock->sock) )
+		{
+			eventWriteClient(this);
+		}
+
+		if( this->file != NULL && getBufferSize(this->sendBuffer) == 0 )
+		{
+			sendFile(this);
+		}
+
+		if( this->status_connect == NET_STATUS_ZOMBIE )
+		{
+			delListItem(listClient, i, destroyClient);
+		}
+	}
+
+	return 0;
+}
+
+int quitDownServer()
+{
+	closeTcpSocket(sock_server_tcp);
+	destroyListItem(listClient, destroyClient);
+
+	return 0;
+}