diff net.c @ 0:76568becd6d6

Rhope Alpha 2a source import
author Mike Pavone <pavone@retrodev.com>
date Tue, 28 Apr 2009 23:06:07 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/net.c	Tue Apr 28 23:06:07 2009 +0000
@@ -0,0 +1,625 @@
+#ifdef WIN32
+	#include <winsock.h>
+#else
+	#include <sys/types.h>
+	#include <sys/socket.h>
+	#include <netinet/in.h>
+	#include <netdb.h>
+	#include <arpa/inet.h>
+#endif
+#include <string.h>
+#include <stdlib.h>
+#include "datum.h"
+#include "structs.h"
+#include "vis_threading.h"
+#include "interp.h"
+
+#ifdef WIN32
+	#define VIS_CloseSocket(sock)	closesocket(sock)
+#else
+	#define VIS_CloseSocket(sock)	close(sock)
+#endif
+
+#define NET_SEARCH_BUFFER_SIZE 512
+
+typedef struct netbuffer {
+	char buffer[NET_SEARCH_BUFFER_SIZE];
+	int index;
+	struct netbuffer * next;	
+} netbuffer;
+/*
+typedef struct 
+{
+	int sockfd;
+	netbuffer * buffers;
+	VIS_CRITICAL_SECTION(lock)
+} net_client_data;*/
+
+int send_all(int sockfd, char * data, int len)
+{
+	int sent = 0;
+	int temp;
+	while(sent < len)
+	{
+		temp = send(sockfd, data+sent, len-sent, 0);
+		if(temp < 0)
+		{
+			DEBUGPUTS("send_all() failed\n");
+			return 0-sent;
+		}
+		sent += temp;
+	}
+	return sent;
+}
+
+int recv_all(int sockfd, char * data, int len)
+{
+	int temp;
+	int received = 0;
+	while(received < len)
+	{
+		temp = recv(sockfd, data, len-received, 0);
+		if(temp < 0)
+		{
+			return 0 - received;
+		}
+		received += temp;	
+	}
+	return received;
+}
+
+int net_client_new(datum ** params, queue_entry * worker_entry)
+{
+	int sockfd, i;
+	//net_client_data * net;
+	struct sockaddr_in dest;
+	struct hostent * h;
+	char * host = params[0]->c.generic.data;
+	int hostlen = params[0]->c.generic.len;
+	for(i = 0; i < hostlen; ++i)
+		if((host[i] < '0' || host[i] > '0') && host[i] != '.')
+		{
+			h = gethostbyname(host);
+			if(!h)
+			{
+				release_ref(params[0]);
+				release_ref(params[1]);
+				params[0] = NULL;
+				params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, worker_entry->instance->def->program);
+				params[1]->c.integers.num_a = 0;
+				return 0;
+			}
+			host = inet_ntoa(*((struct in_addr *)h->h_addr));
+			break;
+		}
+	sockfd = socket(PF_INET, SOCK_STREAM, 0);
+	
+	dest.sin_family = AF_INET;
+	dest.sin_port = htons(params[1]->c.integers.num_a);
+	dest.sin_addr.s_addr = inet_addr(host);
+	memset(&(dest.sin_zero), '\0', 8);
+	
+	release_ref(params[0]);
+	release_ref(params[1]);
+	params[0] = params[1] = NULL;
+	
+	if(connect(sockfd, (struct sockaddr *)&dest, sizeof(struct sockaddr)) != 1)
+	{
+		params[0] = new_datum(BUILTIN_TYPE_NETCLIENT, 2, 0, worker_entry->instance->def->program);//sizeof(net_client_data));
+		params[0]->c.integers.num_a = sockfd;
+		/*net = params[0]->c.generic.data;
+		net->sockfd = sockfd;
+		net->buffers = NULL;*/
+		
+//		VIS_InitializeCriticalSection(net->lock);
+		DEBUGPUTS("Connection established\n");
+	}
+	else
+	{
+		params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, worker_entry->instance->def->program);
+		params[1]->c.integers.num_a = 0;
+		DEBUGPUTS("connect() failed\n");
+	}
+	return 0;
+}
+
+int net_client_put_string(datum ** params, queue_entry * worker_entry)
+{
+	//net_client_data * net = params[0]->c.generic.data;
+	//int sockfd = net->sockfd;
+	int sockfd = params[0]->c.integers.num_a;
+	int to_send = params[1]->c.generic.len-1;
+	char * data = params[1]->c.generic.data;
+	int sent;
+	if(to_send)
+	{
+		sent = send_all(sockfd, data, to_send);
+		
+		if(sent != to_send)
+		{
+			release_ref(params[0]);
+			release_ref(params[1]);
+			params[0] = NULL;
+			params[1] = new_datum(BUILTIN_TYPE_WHOLE, 2, 0, worker_entry->instance->def->program);
+			params[1]->c.integers.num_a = 0-sent;
+			DEBUGPUTS("send() failed\n");
+			return 0;
+		}
+	}
+	DEBUGPRINTF("Sent %s\n", data);
+	release_ref(params[1]);
+	params[1] = NULL;
+	return 0;
+}
+
+int net_client_get_fstring(datum ** params, queue_entry * worker_entry)
+{
+	char * buf;
+	//net_client_data * net = params[0]->c.generic.data;
+	//int sockfd = net->sockfd;
+	int sockfd = params[0]->c.integers.num_a;
+	int to_receive = params[1]->c.integers.num_a;
+	int temp, received = 0;
+	datum * output = new_datum(BUILTIN_TYPE_STRING, 1, params[1]->c.integers.num_a+1, worker_entry->instance->def->program);
+	release_ref(params[1]);
+	buf = output->c.generic.data;
+	while(received < to_receive)
+	{
+		temp = recv(sockfd, buf+received, to_receive-received, 0);
+		if(temp < 0)
+		{
+			release_ref(params[0]);
+			params[0] = params[1] = NULL;
+			params[2] = output;
+			return 0;
+		}
+		received += temp;	
+	}
+	params[1] = output;
+	params[2] = NULL;
+	return 0;
+}
+
+int net_client_get_dstring(datum ** inputlist, queue_entry * worker_entry)
+{
+	BOOL found = FALSE;
+	netbuffer buf;
+	netbuffer * current, *first, *temp,*temp2;
+	int i,j,k,startk;
+	int found_entry;
+	int string_offset;
+	int search_offset;
+	netbuffer * search_start;
+	int search_start_offset;
+	int *search_offsets;
+	netbuffer ** search_starts;
+	int *search_start_offsets;
+	int read_bytes;
+	int buf_pos;
+	int sockfd = inputlist[0]->c.integers.num_a;
+	//net_client_data * net = inputlist[0]->c.generic.data;
+	
+	list_data * list;
+	if(inputlist[1]->company->type_id == BUILTIN_TYPE_LIST)
+	{
+		
+		list = ((list_data *)inputlist[1]->c.generic.data);
+		DEBUGPRINTF("Delimeter input is a list with %d entries.\n", list->num_entries);
+		search_offsets = malloc(sizeof(int) * (list->num_entries));
+		DEBUGPRINTF("Allocated %d bytes.\n", sizeof(int) * (list->num_entries));
+		search_starts = malloc(sizeof(netbuffer *) * (list->num_entries));
+		DEBUGPRINTF("Allocated %d bytes.\n", sizeof(netbuffer *) * (list->num_entries));
+		search_start_offsets = malloc(sizeof(int) * (list->num_entries));
+		DEBUGPRINTF("Allocated %d bytes.\n", sizeof(int) * (list->num_entries));
+		for(i = 0; i < list->num_entries; ++i)
+		{
+			DEBUGPRINTF("Setting search_offsets[%d] = 0.\n", i);
+			search_offsets[i] = 0;
+		}
+	}
+	search_offset = 0;
+		first = &buf;
+		first->next = NULL;
+		current = first;
+		current->index = 0;
+		first = current;
+		read_bytes = 1;
+		while(!found && read_bytes == 1)
+		{
+			buf_pos = 0;
+			for(i = 0; i < NET_SEARCH_BUFFER_SIZE && !found; ++i)
+			{
+				if(i >= buf_pos)
+				{
+					read_bytes = recv(sockfd, current->buffer+i, 1, 0);
+					if(read_bytes != 1)
+						break;
+					++buf_pos;
+				}
+				DEBUGPRINTF("Checking character #%d (%c)\n", i, current->buffer[i]);
+				switch(inputlist[1]->company->type_id)
+				{
+				case BUILTIN_TYPE_WHOLE:
+					if((int)current->buffer[i] == inputlist[1]->c.integers.num_a)
+					{
+						found = TRUE;
+						search_offset = 1;
+						search_start = current;
+						search_start_offset = i;
+					}
+					break;
+				case BUILTIN_TYPE_STRING:
+					DEBUGPRINTF("Comparing with character %d of delim: %c (%X)\n", search_offset, ((char *)inputlist[1]->c.generic.data)[search_offset], ((char *)inputlist[1]->c.generic.data)[search_offset]);
+					if(current->buffer[i] == ((char *)inputlist[1]->c.generic.data)[search_offset])
+					{
+						if(search_offset == 0)
+						{
+							search_start = current;
+							search_start_offset = i;
+						}
+						++search_offset;
+						DEBUGPRINTF("Search offset is: %d, delim len is: %d\n", search_offset, (inputlist[1]->c.generic.len-1));
+						if(search_offset == (inputlist[1]->c.generic.len-1))
+						{
+							found = TRUE;
+							DEBUGPUTS("Matched delim\n");
+						}
+					}
+					else
+					{
+						if(search_offset > 0)
+						{
+							current = search_start;
+							i = search_start_offset;
+						}
+						search_offset = 0;
+					}
+					break;
+				case BUILTIN_TYPE_LIST:
+					for(j = 0; j < list->num_entries; ++j)
+					{
+						DEBUGPRINTF("Testing list entry %d against character %d in buffer %d\n", j, i, current->index);
+						if(list->entries[j]->company->type_id == BUILTIN_TYPE_WHOLE && (int)current->buffer[i] == list->entries[j]->c.integers.num_a)
+						{
+							DEBUGPUTS("Matched whole number entry.\n");
+							found = TRUE;
+							found_entry = j;
+							search_offset = 1;
+							search_start = current;
+							search_start_offset = i;
+							break;
+						}
+						else if(list->entries[j]->company->type_id == BUILTIN_TYPE_STRING)
+						{
+							DEBUGPUTS("String entry.\n");
+							if(current->buffer[i] == ((char *)list->entries[j]->c.generic.data)[search_offsets[j]])
+							{
+								DEBUGPRINTF("%c in buffer matches character #%d in entry.\n", current->buffer[i], search_offsets[j]);
+								if(search_offsets[j] == 0)
+								{
+									search_starts[j] = current;
+									search_start_offsets[j] = i;
+								}
+								++search_offsets[j];
+								if(search_offsets[j] == (list->entries[j]->c.generic.len-1))
+								{
+									DEBUGPUTS("Entire string matched.\n");
+									found = TRUE;
+									found_entry = j;
+									search_offset = search_offsets[j];
+									search_start = search_starts[j];
+									search_start_offset = search_start_offsets[j];
+									break;
+								}
+							}
+							else if(search_offsets[j] > 0)
+							{
+								DEBUGPRINTF("%c in bufer does not match character #%d in entry.\n", current->buffer[i], search_offsets[j]);
+								temp = search_starts[j];
+								search_offsets[j] = 0;
+								startk = search_start_offsets[j];
+								while(temp && !found)
+								{
+									DEBUGPRINTF("Scanning block %d for possible missed match from %d to %d.\n", temp->index, startk, (temp == current ? i : NET_SEARCH_BUFFER_SIZE)-1);
+									for(k = startk; k < (temp == current ? i : NET_SEARCH_BUFFER_SIZE); ++k)
+									{
+										if(temp->buffer[k] == ((char *)list->entries[j]->c.generic.data)[search_offsets[j]])
+										{
+											if(!search_offsets[j])
+											{
+												search_starts[j] = temp;
+												search_start_offsets[j] = k;
+											}
+											++search_offsets[j];
+											if(search_offset == (list->entries[j]->c.generic.len-1))
+											{
+												found = TRUE;
+												found_entry = j;
+												search_start = search_starts[j];
+												search_start_offset = search_start_offsets[j];
+											}
+										}
+										else
+										{
+											if(search_offsets[j] > 0)
+											{
+												temp = search_starts[j];
+												k = search_start_offsets[j];
+											}
+											search_offsets[j] = 0;
+										}
+									}
+									startk = 0;
+									temp = temp->next;
+								}
+								
+							}
+							else
+								search_offsets[j] = 0;
+							
+						}
+					}
+					break;
+				}
+			}
+			if(!found && read_bytes == 1)
+			{
+				current->next = malloc(sizeof(netbuffer));
+				current->next->index = current->index+1;
+				//current->next->offset = current->next->filled = 0;
+				current->next->next = NULL;
+				current = current->next;
+			}
+		}
+	if(inputlist[1]->company->type_id == BUILTIN_TYPE_LIST)
+	{
+		VIS_FREE(search_offsets, "Get DString@Net Clinet, search offsets");
+		VIS_FREE(search_starts, "Get DString@Net Clinet, search starts");
+		VIS_FREE(search_start_offsets, "Get DString@Net Clinet, search start offsets");
+	}
+	if(found)
+	{
+		if(inputlist[1]->company->type_id == BUILTIN_TYPE_LIST)
+		{
+			inputlist[2] = add_ref(list->entries[found_entry]);
+			release_ref(inputlist[1]);
+		}
+		else
+			inputlist[2] = inputlist[1];
+		inputlist[3] = NULL;
+	}
+	else
+	{
+		release_ref(inputlist[1]);
+		inputlist[3] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, worker_entry->instance->def->program);
+		datum_set_yesno(inputlist[3], 0);
+		inputlist[2] = NULL;
+		release_ref(inputlist[0]);
+		inputlist[0] = NULL;
+	}
+	//Does this need to be here still or was it just working around another bug?
+	if(search_start_offset < 0)
+		search_start_offset = 0;
+	if(!found) {
+		search_start = current;
+		search_start_offset = i;
+	}
+	DEBUGPRINTF("Allocating string of length: %d = %d * %d + %d + 1\n", NET_SEARCH_BUFFER_SIZE * search_start->index + search_start_offset+1, NET_SEARCH_BUFFER_SIZE , search_start->index, search_start_offset);
+	inputlist[1] = new_datum(BUILTIN_TYPE_STRING, 1, NET_SEARCH_BUFFER_SIZE * search_start->index + search_start_offset+1, worker_entry->instance->def->program);
+	temp = first;
+	string_offset = 0;
+	while(temp)
+	{
+		DEBUGPRINTF("Copying from index %d to offset %X\n", temp->index, string_offset);
+		if(temp == search_start)
+		{
+			//if(found)
+			//{
+				temp->buffer[search_start_offset] = '\0';
+				memcpy(((char *)inputlist[1]->c.generic.data)+string_offset, temp->buffer, search_start_offset);
+				string_offset += search_start_offset;
+			/*}
+			else
+			{
+				memcpy(((char *)inputlist[1]->c.generic.data)+string_offset, temp->buffer, i);
+				string_offset += i;
+			}*/
+			break;
+		}
+		else
+		{
+			memcpy(((char *)inputlist[1]->c.generic.data)+string_offset, temp->buffer, NET_SEARCH_BUFFER_SIZE);
+			string_offset += NET_SEARCH_BUFFER_SIZE;
+		}
+		if(temp != first)
+		{
+			temp2 = temp->next;
+			VIS_FREE(temp, "Get DString@Net Client, buffer node");
+			temp = temp2;
+		}
+		else
+			temp = temp->next;
+	}
+	while(temp)
+	{
+		if(temp != first)
+		{
+			temp2 = temp->next;
+			VIS_FREE(temp, "Get DString@Net Client, buffer node");
+			temp = temp2;
+		}
+		else
+			temp = temp->next;
+	}
+	((char *)inputlist[1]->c.generic.data)[string_offset] = '\0';
+	DEBUGPRINTF("Retrieved: %s\ninputlist[0] = %X\n", inputlist[1]->c.generic.data, inputlist[0]);
+	return 0;
+}
+
+int net_client_put_raw(datum ** params, queue_entry * entry)
+{
+	int sockfd = params[0]->c.integers.num_a;
+	int sent;
+	if(params[1]->union_type == 1 && params[1]->c.generic.len)
+	{
+		sent = send_all(sockfd, params[1]->c.generic.data, params[1]->c.generic.len);
+		if(sent <= 0)
+		{
+			release_ref(params[0]);
+			release_ref(params[1]);
+			params[0] = NULL;
+			params[1] = new_datum(BUILTIN_TYPE_WHOLE, 2, 0, entry->instance->def->program);
+			params[1]->c.integers.num_a = 0-sent;
+			return 0;
+		}
+	}
+	release_ref(params[1]);
+	params[1] = NULL;
+	return 0;
+}
+
+int net_client_get_raw(datum ** params, queue_entry * entry)
+{
+	int sockfd = params[0]->c.integers.num_a;
+	int got;
+	params[1] = copy_datum(params[1], 0);
+	if(params[1]->union_type == 1 && params[1]->c.generic.len)
+	{
+		got = recv_all(sockfd, params[1]->c.generic.data, params[1]->c.generic.len);
+		if(got <= 0)
+		{
+			release_ref(params[0]);
+			release_ref(params[1]);
+			params[0] = params[1] = NULL;
+			params[2] = new_datum(BUILTIN_TYPE_WHOLE, 2, 0, entry->instance->def->program);
+			params[2]->c.integers.num_a = 0-got;
+			return 0;
+		}
+	}
+	params[2] = NULL;
+	return 0;
+}
+
+typedef struct
+{
+	int sockfd;
+	datum * callback;
+	program * program;
+} net_listen_data;
+
+DWORD WINAPI listen_thread(net_listen_data * listen)
+{
+	worker_datum * work = listen->callback->c.generic.data;
+	struct sockaddr_in client_addy;
+	int addy_len;
+	int newsock;
+	int i;
+	queue_entry entry;
+	datum * params[32];
+	BOOL netparam;
+	worker_instance inst;
+	entry.worker_num = 0;
+	entry.instance = &inst;
+	inst.def = listen->program->defs->deflist;//Use Main
+	inst.caller_instance = NULL;
+	inst.trans = NULL;
+	inst.num_workers = inst.num_wires = 0;
+	VIS_InitializeCriticalSection(inst.counter_lock);
+
+	while(execute_active)
+	{
+		addy_len = sizeof(client_addy);
+		newsock = accept(listen->sockfd, (struct sockaddr *)&client_addy, &addy_len);
+		if(newsock != -1)
+		{
+			netparam = FALSE;
+			for(i = 0; i < work->def->num_inputs; ++i)
+				if(work->params[i])
+					params[i] = add_ref(work->params[i]);
+				else if(!netparam)
+				{
+					params[i] = new_datum(BUILTIN_TYPE_NETCLIENT, 2, 0, listen->program);
+					params[i]->c.integers.num_a = newsock;
+					netparam = TRUE;
+				}
+				else
+					params[i] = NULL;
+			//Make sure that our fake instance is never cleaned up
+			inst.in_progress_count = inst.in_queue_count = 1000;
+			execute_def(work->def, entry, params, NULL);
+		}
+	}
+	release_ref(listen->callback);
+	VIS_CloseSocket(listen->sockfd);
+	VIS_FREE(listen, "Net listener object");
+	return 0;
+}
+
+int vis_net_listenport(datum ** params, queue_entry * entry)
+{
+	int junk;
+	struct sockaddr_in my_address;
+	int port = params[0]->c.integers.num_a;
+	net_listen_data * listener = malloc(sizeof(net_listen_data));
+	release_ref(params[0]);
+	
+	DEBUGPUTS("calling socket\n");
+	listener->callback = params[1];
+	listener->sockfd = socket(PF_INET, SOCK_STREAM, 0);
+	listener->program = entry->instance->def->program;
+	if(listener->sockfd == -1)
+	{
+		release_ref(params[1]);
+		VIS_FREE(listener, "net listener object");
+		params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
+		params[1]->c.integers.num_a = 0;
+		params[0] = NULL;
+		return 0;
+	}
+	DEBUGPRINTF("Socket: %X\n", listener->sockfd);
+	
+	my_address.sin_family = AF_INET;
+	my_address.sin_port = htons(port);
+	my_address.sin_addr.s_addr = INADDR_ANY;
+	memset(&(my_address.sin_zero), '\0', 8);
+	
+	DEBUGPRINTF("Calling bind on port: %d\n", port);
+	junk = bind(listener->sockfd, (struct sockaddr *)&my_address, sizeof(struct sockaddr));
+	DEBUGPRINTF("Bind returned: %d\n", junk);
+	if(junk == -1)
+	{
+		perror("bind");
+		DEBUGPUTS("bind failed, releasing ref to callback worker\n");
+		release_ref(params[1]);
+		DEBUGPUTS("Closing socket\n");
+		DEBUGPRINTF("Socket: %X\n", listener->sockfd);
+		VIS_CloseSocket(listener->sockfd);
+		DEBUGPUTS("Freeing listener data\n");
+		VIS_FREE(listener, "net listener object");
+		DEBUGPUTS("Allocating error output\n");
+		params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
+		params[1]->c.integers.num_a = 0;
+		params[0] = NULL;
+		return 0;
+	}
+	
+	DEBUGPUTS("Calling listen\n");
+	if(listen(listener->sockfd, 10) == -1)
+	{
+		release_ref(params[1]);
+		VIS_CloseSocket(listener->sockfd);
+		VIS_FREE(listener, "net listener object");
+		params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
+		params[1]->c.integers.num_a = 0;
+		params[0] = NULL;
+		return 0;
+	}
+	DEBUGPUTS("Creating new thread\n");
+	VIS_NewThread(listen_thread, listener);
+	params[0] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
+	params[0]->c.integers.num_a = 0;
+	params[1] = NULL;
+	return 0;
+}
+
+