X-Git-Url: https://git.ucc.asn.au/?p=progcomp2012.git;a=blobdiff_plain;f=judge%2Fmanager%2Fnetwork.cpp;fp=judge%2Fmanager%2Fnetwork.cpp;h=e349fea4cabfb85749929a7918cd0951189105a3;hp=0000000000000000000000000000000000000000;hb=341297b4dce9528d54fe9dceeff0d6dc33f70abe;hpb=73d5a5dab8881b9ed5f93a675d64f86a9b45e6bc

diff --git a/judge/manager/network.cpp b/judge/manager/network.cpp
new file mode 100644
index 0000000..e349fea
--- /dev/null
+++ b/judge/manager/network.cpp
@@ -0,0 +1,186 @@
+#include "network.h"
+#include <stdarg.h>
+
+using namespace std;
+
+Network::Network(int newPort) : sfd(-1), port(newPort), file(NULL)
+{
+    	sfd = socket(PF_INET, SOCK_STREAM, 0);
+	if (sfd < 0)
+	{
+		perror("Network::Network - Error creating TCP socket");
+		exit(EXIT_FAILURE);
+	}
+}
+
+Network::~Network()
+{
+	if (Valid())
+	{
+		if (shutdown(sfd, SHUT_RDWR) == -1)
+		{
+			perror("Network::~Network - shutting down socket... ");
+			close(sfd);
+			sfd = -1;
+		}
+	}
+	close(sfd);
+}
+
+Server::Server(int newPort) : Network(newPort)
+{
+	struct   sockaddr_in name;
+//	char   buf[1024];
+//	int    cc;
+
+	
+	name.sin_family = AF_INET;
+	name.sin_addr.s_addr = htonl(INADDR_ANY);
+	name.sin_port = htons(port);
+
+	if (bind( sfd, (struct sockaddr *) &name, sizeof(name) ) < 0)
+	{
+		perror("Server::Server - Error binding socket");
+		close(sfd); sfd = -1;
+		exit(EXIT_FAILURE);
+	}
+
+	if (listen(sfd,1) < 0)
+	{
+		perror("Server::Server - Error listening on socket");
+		close(sfd); sfd = -1;
+		exit(EXIT_FAILURE);
+	}
+	int psd = accept(sfd, 0, 0);
+	close(sfd);
+	sfd = psd;
+	if (sfd < 0)
+	{
+		perror("Server::Server - Error accepting connection");
+		close(sfd); sfd = -1;
+		exit(EXIT_FAILURE);
+	}
+
+
+
+	/*
+	for(;;) 
+	{
+		cc=recv(sfd,buf,sizeof(buf), 0) ;
+		if (cc == 0) exit (0);
+		buf[cc] = '\0';
+		printf("message received: %s\n", buf);
+	}
+	*/
+}
+
+Client::Client(const char * serverAddress, int newPort) : Network(newPort)
+{
+	struct	sockaddr_in server;
+	struct  hostent *hp;
+
+
+	server.sin_family = AF_INET;
+	hp = gethostbyname(serverAddress);
+	bcopy ( hp->h_addr, &(server.sin_addr.s_addr), hp->h_length);
+	server.sin_port = htons(port);
+
+	if (connect(sfd, (struct sockaddr *) &server, sizeof(server)) < 0)
+	{
+		fprintf(stderr, "Client::Client - Error connecting to server at address %s: ", serverAddress);
+		perror("");
+		close(sfd); sfd = -1;
+		exit(EXIT_FAILURE);
+	}
+
+
+
+
+	/*
+        for (;;) {
+	   send(sfd, "HI", 2,0 );
+           sleep(2);
+        }
+	*/
+}
+
+/**
+ * Sends a message accross the network
+ * WARNING: Always terminates the message with a newline '\n'
+ * @returns true if the message was successfully sent; false if it was not (ie: the network was not connected!)
+ */
+bool Network::SendMessage(const char * print, ...)
+{
+	if (!Valid()) //Is the process running...
+		return false; 
+
+	if (file == NULL)
+	{
+		file = fdopen(sfd, "r+");
+		setbuf(file, NULL);
+	}
+
+	va_list ap;
+	va_start(ap, print);
+
+	if (vfprintf(file, print, ap) < 0 || fprintf(file, "\n") < 0)
+	{
+		va_end(ap);
+		return false;
+	}
+	va_end(ap);
+
+	return true;
+}
+
+/**
+ * Retrieves a message from the network, waiting a maximum amount of time
+ * @param buffer - C++ string to store the resultant message in
+ * @param timeout - Maximum amount of time to wait before failure. If timeout <= 0, then GetMessage will wait indefinately.
+ * @returns true if the response was recieved within the specified time, false if it was not, or an EOF was recieved, or the process was not running.
+ */
+bool Network::GetMessage(string & buffer, double timeout)
+{
+	if (!Valid() || timeout == 0)
+		return false;
+
+	if (file == NULL)
+	{
+		file = fdopen(sfd, "r+");
+		setbuf(file, NULL);
+	}
+
+	assert(&buffer != NULL);
+	GetterThread getterThread(file, buffer);
+	assert(&(getterThread.buffer) != NULL);
+
+	TimerThread timerThread(timeout*1000000);
+
+	getterThread.Start();
+	if (timeout > 0)
+		timerThread.Start();
+
+	
+	while (!getterThread.Finished())
+	{
+		if (timeout > 0 && timerThread.Finished())
+		{
+			getterThread.Stop();
+			timerThread.Stop();
+			return false;
+		}
+	}
+
+	getterThread.Stop();
+	if (timeout > 0)
+		timerThread.Stop();
+
+	
+
+	if (buffer.size() == 1 && buffer[0] == EOF)
+		return false;
+	return true;
+
+
+}
+