|
0
|
1 #ifdef WIN32
|
|
|
2 #include <winsock.h>
|
|
|
3 #else
|
|
|
4 #include <sys/types.h>
|
|
|
5 #include <sys/socket.h>
|
|
|
6 #include <netinet/in.h>
|
|
|
7 #include <netdb.h>
|
|
|
8 #include <arpa/inet.h>
|
|
|
9 #endif
|
|
|
10 #include <string.h>
|
|
|
11 #include <stdlib.h>
|
|
|
12 #include "datum.h"
|
|
|
13 #include "structs.h"
|
|
|
14 #include "vis_threading.h"
|
|
|
15 #include "interp.h"
|
|
|
16
|
|
|
17 #ifdef WIN32
|
|
|
18 #define VIS_CloseSocket(sock) closesocket(sock)
|
|
|
19 #else
|
|
|
20 #define VIS_CloseSocket(sock) close(sock)
|
|
|
21 #endif
|
|
|
22
|
|
|
23 #define NET_SEARCH_BUFFER_SIZE 512
|
|
|
24
|
|
|
25 typedef struct netbuffer {
|
|
|
26 char buffer[NET_SEARCH_BUFFER_SIZE];
|
|
|
27 int index;
|
|
|
28 struct netbuffer * next;
|
|
|
29 } netbuffer;
|
|
|
30 /*
|
|
|
31 typedef struct
|
|
|
32 {
|
|
|
33 int sockfd;
|
|
|
34 netbuffer * buffers;
|
|
|
35 VIS_CRITICAL_SECTION(lock)
|
|
|
36 } net_client_data;*/
|
|
|
37
|
|
|
38 int send_all(int sockfd, char * data, int len)
|
|
|
39 {
|
|
|
40 int sent = 0;
|
|
|
41 int temp;
|
|
|
42 while(sent < len)
|
|
|
43 {
|
|
|
44 temp = send(sockfd, data+sent, len-sent, 0);
|
|
|
45 if(temp < 0)
|
|
|
46 {
|
|
|
47 DEBUGPUTS("send_all() failed\n");
|
|
|
48 return 0-sent;
|
|
|
49 }
|
|
|
50 sent += temp;
|
|
|
51 }
|
|
|
52 return sent;
|
|
|
53 }
|
|
|
54
|
|
|
55 int recv_all(int sockfd, char * data, int len)
|
|
|
56 {
|
|
|
57 int temp;
|
|
|
58 int received = 0;
|
|
|
59 while(received < len)
|
|
|
60 {
|
|
|
61 temp = recv(sockfd, data, len-received, 0);
|
|
|
62 if(temp < 0)
|
|
|
63 {
|
|
|
64 return 0 - received;
|
|
|
65 }
|
|
|
66 received += temp;
|
|
|
67 }
|
|
|
68 return received;
|
|
|
69 }
|
|
|
70
|
|
|
71 int net_client_new(datum ** params, queue_entry * worker_entry)
|
|
|
72 {
|
|
|
73 int sockfd, i;
|
|
|
74 //net_client_data * net;
|
|
|
75 struct sockaddr_in dest;
|
|
|
76 struct hostent * h;
|
|
|
77 char * host = params[0]->c.generic.data;
|
|
|
78 int hostlen = params[0]->c.generic.len;
|
|
|
79 for(i = 0; i < hostlen; ++i)
|
|
|
80 if((host[i] < '0' || host[i] > '0') && host[i] != '.')
|
|
|
81 {
|
|
|
82 h = gethostbyname(host);
|
|
|
83 if(!h)
|
|
|
84 {
|
|
|
85 release_ref(params[0]);
|
|
|
86 release_ref(params[1]);
|
|
|
87 params[0] = NULL;
|
|
|
88 params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, worker_entry->instance->def->program);
|
|
|
89 params[1]->c.integers.num_a = 0;
|
|
|
90 return 0;
|
|
|
91 }
|
|
|
92 host = inet_ntoa(*((struct in_addr *)h->h_addr));
|
|
|
93 break;
|
|
|
94 }
|
|
|
95 sockfd = socket(PF_INET, SOCK_STREAM, 0);
|
|
|
96
|
|
|
97 dest.sin_family = AF_INET;
|
|
|
98 dest.sin_port = htons(params[1]->c.integers.num_a);
|
|
|
99 dest.sin_addr.s_addr = inet_addr(host);
|
|
|
100 memset(&(dest.sin_zero), '\0', 8);
|
|
|
101
|
|
|
102 release_ref(params[0]);
|
|
|
103 release_ref(params[1]);
|
|
|
104 params[0] = params[1] = NULL;
|
|
|
105
|
|
|
106 if(connect(sockfd, (struct sockaddr *)&dest, sizeof(struct sockaddr)) != 1)
|
|
|
107 {
|
|
|
108 params[0] = new_datum(BUILTIN_TYPE_NETCLIENT, 2, 0, worker_entry->instance->def->program);//sizeof(net_client_data));
|
|
|
109 params[0]->c.integers.num_a = sockfd;
|
|
|
110 /*net = params[0]->c.generic.data;
|
|
|
111 net->sockfd = sockfd;
|
|
|
112 net->buffers = NULL;*/
|
|
|
113
|
|
|
114 // VIS_InitializeCriticalSection(net->lock);
|
|
|
115 DEBUGPUTS("Connection established\n");
|
|
|
116 }
|
|
|
117 else
|
|
|
118 {
|
|
|
119 params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, worker_entry->instance->def->program);
|
|
|
120 params[1]->c.integers.num_a = 0;
|
|
|
121 DEBUGPUTS("connect() failed\n");
|
|
|
122 }
|
|
|
123 return 0;
|
|
|
124 }
|
|
|
125
|
|
|
126 int net_client_put_string(datum ** params, queue_entry * worker_entry)
|
|
|
127 {
|
|
|
128 //net_client_data * net = params[0]->c.generic.data;
|
|
|
129 //int sockfd = net->sockfd;
|
|
|
130 int sockfd = params[0]->c.integers.num_a;
|
|
|
131 int to_send = params[1]->c.generic.len-1;
|
|
|
132 char * data = params[1]->c.generic.data;
|
|
|
133 int sent;
|
|
|
134 if(to_send)
|
|
|
135 {
|
|
|
136 sent = send_all(sockfd, data, to_send);
|
|
|
137
|
|
|
138 if(sent != to_send)
|
|
|
139 {
|
|
|
140 release_ref(params[0]);
|
|
|
141 release_ref(params[1]);
|
|
|
142 params[0] = NULL;
|
|
|
143 params[1] = new_datum(BUILTIN_TYPE_WHOLE, 2, 0, worker_entry->instance->def->program);
|
|
|
144 params[1]->c.integers.num_a = 0-sent;
|
|
|
145 DEBUGPUTS("send() failed\n");
|
|
|
146 return 0;
|
|
|
147 }
|
|
|
148 }
|
|
|
149 DEBUGPRINTF("Sent %s\n", data);
|
|
|
150 release_ref(params[1]);
|
|
|
151 params[1] = NULL;
|
|
|
152 return 0;
|
|
|
153 }
|
|
|
154
|
|
|
155 int net_client_get_fstring(datum ** params, queue_entry * worker_entry)
|
|
|
156 {
|
|
|
157 char * buf;
|
|
|
158 //net_client_data * net = params[0]->c.generic.data;
|
|
|
159 //int sockfd = net->sockfd;
|
|
|
160 int sockfd = params[0]->c.integers.num_a;
|
|
|
161 int to_receive = params[1]->c.integers.num_a;
|
|
|
162 int temp, received = 0;
|
|
|
163 datum * output = new_datum(BUILTIN_TYPE_STRING, 1, params[1]->c.integers.num_a+1, worker_entry->instance->def->program);
|
|
|
164 release_ref(params[1]);
|
|
|
165 buf = output->c.generic.data;
|
|
|
166 while(received < to_receive)
|
|
|
167 {
|
|
|
168 temp = recv(sockfd, buf+received, to_receive-received, 0);
|
|
|
169 if(temp < 0)
|
|
|
170 {
|
|
|
171 release_ref(params[0]);
|
|
|
172 params[0] = params[1] = NULL;
|
|
|
173 params[2] = output;
|
|
|
174 return 0;
|
|
|
175 }
|
|
|
176 received += temp;
|
|
|
177 }
|
|
|
178 params[1] = output;
|
|
|
179 params[2] = NULL;
|
|
|
180 return 0;
|
|
|
181 }
|
|
|
182
|
|
|
183 int net_client_get_dstring(datum ** inputlist, queue_entry * worker_entry)
|
|
|
184 {
|
|
|
185 BOOL found = FALSE;
|
|
|
186 netbuffer buf;
|
|
|
187 netbuffer * current, *first, *temp,*temp2;
|
|
|
188 int i,j,k,startk;
|
|
|
189 int found_entry;
|
|
|
190 int string_offset;
|
|
|
191 int search_offset;
|
|
|
192 netbuffer * search_start;
|
|
|
193 int search_start_offset;
|
|
|
194 int *search_offsets;
|
|
|
195 netbuffer ** search_starts;
|
|
|
196 int *search_start_offsets;
|
|
|
197 int read_bytes;
|
|
|
198 int buf_pos;
|
|
|
199 int sockfd = inputlist[0]->c.integers.num_a;
|
|
|
200 //net_client_data * net = inputlist[0]->c.generic.data;
|
|
|
201
|
|
|
202 list_data * list;
|
|
|
203 if(inputlist[1]->company->type_id == BUILTIN_TYPE_LIST)
|
|
|
204 {
|
|
|
205
|
|
|
206 list = ((list_data *)inputlist[1]->c.generic.data);
|
|
|
207 DEBUGPRINTF("Delimeter input is a list with %d entries.\n", list->num_entries);
|
|
|
208 search_offsets = malloc(sizeof(int) * (list->num_entries));
|
|
|
209 DEBUGPRINTF("Allocated %d bytes.\n", sizeof(int) * (list->num_entries));
|
|
|
210 search_starts = malloc(sizeof(netbuffer *) * (list->num_entries));
|
|
|
211 DEBUGPRINTF("Allocated %d bytes.\n", sizeof(netbuffer *) * (list->num_entries));
|
|
|
212 search_start_offsets = malloc(sizeof(int) * (list->num_entries));
|
|
|
213 DEBUGPRINTF("Allocated %d bytes.\n", sizeof(int) * (list->num_entries));
|
|
|
214 for(i = 0; i < list->num_entries; ++i)
|
|
|
215 {
|
|
|
216 DEBUGPRINTF("Setting search_offsets[%d] = 0.\n", i);
|
|
|
217 search_offsets[i] = 0;
|
|
|
218 }
|
|
|
219 }
|
|
|
220 search_offset = 0;
|
|
|
221 first = &buf;
|
|
|
222 first->next = NULL;
|
|
|
223 current = first;
|
|
|
224 current->index = 0;
|
|
|
225 first = current;
|
|
|
226 read_bytes = 1;
|
|
|
227 while(!found && read_bytes == 1)
|
|
|
228 {
|
|
|
229 buf_pos = 0;
|
|
|
230 for(i = 0; i < NET_SEARCH_BUFFER_SIZE && !found; ++i)
|
|
|
231 {
|
|
|
232 if(i >= buf_pos)
|
|
|
233 {
|
|
|
234 read_bytes = recv(sockfd, current->buffer+i, 1, 0);
|
|
|
235 if(read_bytes != 1)
|
|
|
236 break;
|
|
|
237 ++buf_pos;
|
|
|
238 }
|
|
|
239 DEBUGPRINTF("Checking character #%d (%c)\n", i, current->buffer[i]);
|
|
|
240 switch(inputlist[1]->company->type_id)
|
|
|
241 {
|
|
|
242 case BUILTIN_TYPE_WHOLE:
|
|
|
243 if((int)current->buffer[i] == inputlist[1]->c.integers.num_a)
|
|
|
244 {
|
|
|
245 found = TRUE;
|
|
|
246 search_offset = 1;
|
|
|
247 search_start = current;
|
|
|
248 search_start_offset = i;
|
|
|
249 }
|
|
|
250 break;
|
|
|
251 case BUILTIN_TYPE_STRING:
|
|
|
252 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]);
|
|
|
253 if(current->buffer[i] == ((char *)inputlist[1]->c.generic.data)[search_offset])
|
|
|
254 {
|
|
|
255 if(search_offset == 0)
|
|
|
256 {
|
|
|
257 search_start = current;
|
|
|
258 search_start_offset = i;
|
|
|
259 }
|
|
|
260 ++search_offset;
|
|
|
261 DEBUGPRINTF("Search offset is: %d, delim len is: %d\n", search_offset, (inputlist[1]->c.generic.len-1));
|
|
|
262 if(search_offset == (inputlist[1]->c.generic.len-1))
|
|
|
263 {
|
|
|
264 found = TRUE;
|
|
|
265 DEBUGPUTS("Matched delim\n");
|
|
|
266 }
|
|
|
267 }
|
|
|
268 else
|
|
|
269 {
|
|
|
270 if(search_offset > 0)
|
|
|
271 {
|
|
|
272 current = search_start;
|
|
|
273 i = search_start_offset;
|
|
|
274 }
|
|
|
275 search_offset = 0;
|
|
|
276 }
|
|
|
277 break;
|
|
|
278 case BUILTIN_TYPE_LIST:
|
|
|
279 for(j = 0; j < list->num_entries; ++j)
|
|
|
280 {
|
|
|
281 DEBUGPRINTF("Testing list entry %d against character %d in buffer %d\n", j, i, current->index);
|
|
|
282 if(list->entries[j]->company->type_id == BUILTIN_TYPE_WHOLE && (int)current->buffer[i] == list->entries[j]->c.integers.num_a)
|
|
|
283 {
|
|
|
284 DEBUGPUTS("Matched whole number entry.\n");
|
|
|
285 found = TRUE;
|
|
|
286 found_entry = j;
|
|
|
287 search_offset = 1;
|
|
|
288 search_start = current;
|
|
|
289 search_start_offset = i;
|
|
|
290 break;
|
|
|
291 }
|
|
|
292 else if(list->entries[j]->company->type_id == BUILTIN_TYPE_STRING)
|
|
|
293 {
|
|
|
294 DEBUGPUTS("String entry.\n");
|
|
|
295 if(current->buffer[i] == ((char *)list->entries[j]->c.generic.data)[search_offsets[j]])
|
|
|
296 {
|
|
|
297 DEBUGPRINTF("%c in buffer matches character #%d in entry.\n", current->buffer[i], search_offsets[j]);
|
|
|
298 if(search_offsets[j] == 0)
|
|
|
299 {
|
|
|
300 search_starts[j] = current;
|
|
|
301 search_start_offsets[j] = i;
|
|
|
302 }
|
|
|
303 ++search_offsets[j];
|
|
|
304 if(search_offsets[j] == (list->entries[j]->c.generic.len-1))
|
|
|
305 {
|
|
|
306 DEBUGPUTS("Entire string matched.\n");
|
|
|
307 found = TRUE;
|
|
|
308 found_entry = j;
|
|
|
309 search_offset = search_offsets[j];
|
|
|
310 search_start = search_starts[j];
|
|
|
311 search_start_offset = search_start_offsets[j];
|
|
|
312 break;
|
|
|
313 }
|
|
|
314 }
|
|
|
315 else if(search_offsets[j] > 0)
|
|
|
316 {
|
|
|
317 DEBUGPRINTF("%c in bufer does not match character #%d in entry.\n", current->buffer[i], search_offsets[j]);
|
|
|
318 temp = search_starts[j];
|
|
|
319 search_offsets[j] = 0;
|
|
|
320 startk = search_start_offsets[j];
|
|
|
321 while(temp && !found)
|
|
|
322 {
|
|
|
323 DEBUGPRINTF("Scanning block %d for possible missed match from %d to %d.\n", temp->index, startk, (temp == current ? i : NET_SEARCH_BUFFER_SIZE)-1);
|
|
|
324 for(k = startk; k < (temp == current ? i : NET_SEARCH_BUFFER_SIZE); ++k)
|
|
|
325 {
|
|
|
326 if(temp->buffer[k] == ((char *)list->entries[j]->c.generic.data)[search_offsets[j]])
|
|
|
327 {
|
|
|
328 if(!search_offsets[j])
|
|
|
329 {
|
|
|
330 search_starts[j] = temp;
|
|
|
331 search_start_offsets[j] = k;
|
|
|
332 }
|
|
|
333 ++search_offsets[j];
|
|
|
334 if(search_offset == (list->entries[j]->c.generic.len-1))
|
|
|
335 {
|
|
|
336 found = TRUE;
|
|
|
337 found_entry = j;
|
|
|
338 search_start = search_starts[j];
|
|
|
339 search_start_offset = search_start_offsets[j];
|
|
|
340 }
|
|
|
341 }
|
|
|
342 else
|
|
|
343 {
|
|
|
344 if(search_offsets[j] > 0)
|
|
|
345 {
|
|
|
346 temp = search_starts[j];
|
|
|
347 k = search_start_offsets[j];
|
|
|
348 }
|
|
|
349 search_offsets[j] = 0;
|
|
|
350 }
|
|
|
351 }
|
|
|
352 startk = 0;
|
|
|
353 temp = temp->next;
|
|
|
354 }
|
|
|
355
|
|
|
356 }
|
|
|
357 else
|
|
|
358 search_offsets[j] = 0;
|
|
|
359
|
|
|
360 }
|
|
|
361 }
|
|
|
362 break;
|
|
|
363 }
|
|
|
364 }
|
|
|
365 if(!found && read_bytes == 1)
|
|
|
366 {
|
|
|
367 current->next = malloc(sizeof(netbuffer));
|
|
|
368 current->next->index = current->index+1;
|
|
|
369 //current->next->offset = current->next->filled = 0;
|
|
|
370 current->next->next = NULL;
|
|
|
371 current = current->next;
|
|
|
372 }
|
|
|
373 }
|
|
|
374 if(inputlist[1]->company->type_id == BUILTIN_TYPE_LIST)
|
|
|
375 {
|
|
|
376 VIS_FREE(search_offsets, "Get DString@Net Clinet, search offsets");
|
|
|
377 VIS_FREE(search_starts, "Get DString@Net Clinet, search starts");
|
|
|
378 VIS_FREE(search_start_offsets, "Get DString@Net Clinet, search start offsets");
|
|
|
379 }
|
|
|
380 if(found)
|
|
|
381 {
|
|
|
382 if(inputlist[1]->company->type_id == BUILTIN_TYPE_LIST)
|
|
|
383 {
|
|
|
384 inputlist[2] = add_ref(list->entries[found_entry]);
|
|
|
385 release_ref(inputlist[1]);
|
|
|
386 }
|
|
|
387 else
|
|
|
388 inputlist[2] = inputlist[1];
|
|
|
389 inputlist[3] = NULL;
|
|
|
390 }
|
|
|
391 else
|
|
|
392 {
|
|
|
393 release_ref(inputlist[1]);
|
|
|
394 inputlist[3] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, worker_entry->instance->def->program);
|
|
|
395 datum_set_yesno(inputlist[3], 0);
|
|
|
396 inputlist[2] = NULL;
|
|
|
397 release_ref(inputlist[0]);
|
|
|
398 inputlist[0] = NULL;
|
|
|
399 }
|
|
|
400 //Does this need to be here still or was it just working around another bug?
|
|
|
401 if(search_start_offset < 0)
|
|
|
402 search_start_offset = 0;
|
|
|
403 if(!found) {
|
|
|
404 search_start = current;
|
|
|
405 search_start_offset = i;
|
|
|
406 }
|
|
|
407 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);
|
|
|
408 inputlist[1] = new_datum(BUILTIN_TYPE_STRING, 1, NET_SEARCH_BUFFER_SIZE * search_start->index + search_start_offset+1, worker_entry->instance->def->program);
|
|
|
409 temp = first;
|
|
|
410 string_offset = 0;
|
|
|
411 while(temp)
|
|
|
412 {
|
|
|
413 DEBUGPRINTF("Copying from index %d to offset %X\n", temp->index, string_offset);
|
|
|
414 if(temp == search_start)
|
|
|
415 {
|
|
|
416 //if(found)
|
|
|
417 //{
|
|
|
418 temp->buffer[search_start_offset] = '\0';
|
|
|
419 memcpy(((char *)inputlist[1]->c.generic.data)+string_offset, temp->buffer, search_start_offset);
|
|
|
420 string_offset += search_start_offset;
|
|
|
421 /*}
|
|
|
422 else
|
|
|
423 {
|
|
|
424 memcpy(((char *)inputlist[1]->c.generic.data)+string_offset, temp->buffer, i);
|
|
|
425 string_offset += i;
|
|
|
426 }*/
|
|
|
427 break;
|
|
|
428 }
|
|
|
429 else
|
|
|
430 {
|
|
|
431 memcpy(((char *)inputlist[1]->c.generic.data)+string_offset, temp->buffer, NET_SEARCH_BUFFER_SIZE);
|
|
|
432 string_offset += NET_SEARCH_BUFFER_SIZE;
|
|
|
433 }
|
|
|
434 if(temp != first)
|
|
|
435 {
|
|
|
436 temp2 = temp->next;
|
|
|
437 VIS_FREE(temp, "Get DString@Net Client, buffer node");
|
|
|
438 temp = temp2;
|
|
|
439 }
|
|
|
440 else
|
|
|
441 temp = temp->next;
|
|
|
442 }
|
|
|
443 while(temp)
|
|
|
444 {
|
|
|
445 if(temp != first)
|
|
|
446 {
|
|
|
447 temp2 = temp->next;
|
|
|
448 VIS_FREE(temp, "Get DString@Net Client, buffer node");
|
|
|
449 temp = temp2;
|
|
|
450 }
|
|
|
451 else
|
|
|
452 temp = temp->next;
|
|
|
453 }
|
|
|
454 ((char *)inputlist[1]->c.generic.data)[string_offset] = '\0';
|
|
|
455 DEBUGPRINTF("Retrieved: %s\ninputlist[0] = %X\n", inputlist[1]->c.generic.data, inputlist[0]);
|
|
|
456 return 0;
|
|
|
457 }
|
|
|
458
|
|
|
459 int net_client_put_raw(datum ** params, queue_entry * entry)
|
|
|
460 {
|
|
|
461 int sockfd = params[0]->c.integers.num_a;
|
|
|
462 int sent;
|
|
|
463 if(params[1]->union_type == 1 && params[1]->c.generic.len)
|
|
|
464 {
|
|
|
465 sent = send_all(sockfd, params[1]->c.generic.data, params[1]->c.generic.len);
|
|
|
466 if(sent <= 0)
|
|
|
467 {
|
|
|
468 release_ref(params[0]);
|
|
|
469 release_ref(params[1]);
|
|
|
470 params[0] = NULL;
|
|
|
471 params[1] = new_datum(BUILTIN_TYPE_WHOLE, 2, 0, entry->instance->def->program);
|
|
|
472 params[1]->c.integers.num_a = 0-sent;
|
|
|
473 return 0;
|
|
|
474 }
|
|
|
475 }
|
|
|
476 release_ref(params[1]);
|
|
|
477 params[1] = NULL;
|
|
|
478 return 0;
|
|
|
479 }
|
|
|
480
|
|
|
481 int net_client_get_raw(datum ** params, queue_entry * entry)
|
|
|
482 {
|
|
|
483 int sockfd = params[0]->c.integers.num_a;
|
|
|
484 int got;
|
|
|
485 params[1] = copy_datum(params[1], 0);
|
|
|
486 if(params[1]->union_type == 1 && params[1]->c.generic.len)
|
|
|
487 {
|
|
|
488 got = recv_all(sockfd, params[1]->c.generic.data, params[1]->c.generic.len);
|
|
|
489 if(got <= 0)
|
|
|
490 {
|
|
|
491 release_ref(params[0]);
|
|
|
492 release_ref(params[1]);
|
|
|
493 params[0] = params[1] = NULL;
|
|
|
494 params[2] = new_datum(BUILTIN_TYPE_WHOLE, 2, 0, entry->instance->def->program);
|
|
|
495 params[2]->c.integers.num_a = 0-got;
|
|
|
496 return 0;
|
|
|
497 }
|
|
|
498 }
|
|
|
499 params[2] = NULL;
|
|
|
500 return 0;
|
|
|
501 }
|
|
|
502
|
|
|
503 typedef struct
|
|
|
504 {
|
|
|
505 int sockfd;
|
|
|
506 datum * callback;
|
|
|
507 program * program;
|
|
|
508 } net_listen_data;
|
|
|
509
|
|
|
510 DWORD WINAPI listen_thread(net_listen_data * listen)
|
|
|
511 {
|
|
|
512 worker_datum * work = listen->callback->c.generic.data;
|
|
|
513 struct sockaddr_in client_addy;
|
|
|
514 int addy_len;
|
|
|
515 int newsock;
|
|
|
516 int i;
|
|
|
517 queue_entry entry;
|
|
|
518 datum * params[32];
|
|
|
519 BOOL netparam;
|
|
|
520 worker_instance inst;
|
|
|
521 entry.worker_num = 0;
|
|
|
522 entry.instance = &inst;
|
|
|
523 inst.def = listen->program->defs->deflist;//Use Main
|
|
|
524 inst.caller_instance = NULL;
|
|
|
525 inst.trans = NULL;
|
|
|
526 inst.num_workers = inst.num_wires = 0;
|
|
|
527 VIS_InitializeCriticalSection(inst.counter_lock);
|
|
|
528
|
|
|
529 while(execute_active)
|
|
|
530 {
|
|
|
531 addy_len = sizeof(client_addy);
|
|
|
532 newsock = accept(listen->sockfd, (struct sockaddr *)&client_addy, &addy_len);
|
|
|
533 if(newsock != -1)
|
|
|
534 {
|
|
|
535 netparam = FALSE;
|
|
|
536 for(i = 0; i < work->def->num_inputs; ++i)
|
|
|
537 if(work->params[i])
|
|
|
538 params[i] = add_ref(work->params[i]);
|
|
|
539 else if(!netparam)
|
|
|
540 {
|
|
|
541 params[i] = new_datum(BUILTIN_TYPE_NETCLIENT, 2, 0, listen->program);
|
|
|
542 params[i]->c.integers.num_a = newsock;
|
|
|
543 netparam = TRUE;
|
|
|
544 }
|
|
|
545 else
|
|
|
546 params[i] = NULL;
|
|
|
547 //Make sure that our fake instance is never cleaned up
|
|
|
548 inst.in_progress_count = inst.in_queue_count = 1000;
|
|
|
549 execute_def(work->def, entry, params, NULL);
|
|
|
550 }
|
|
|
551 }
|
|
|
552 release_ref(listen->callback);
|
|
|
553 VIS_CloseSocket(listen->sockfd);
|
|
|
554 VIS_FREE(listen, "Net listener object");
|
|
|
555 return 0;
|
|
|
556 }
|
|
|
557
|
|
|
558 int vis_net_listenport(datum ** params, queue_entry * entry)
|
|
|
559 {
|
|
|
560 int junk;
|
|
|
561 struct sockaddr_in my_address;
|
|
|
562 int port = params[0]->c.integers.num_a;
|
|
|
563 net_listen_data * listener = malloc(sizeof(net_listen_data));
|
|
|
564 release_ref(params[0]);
|
|
|
565
|
|
|
566 DEBUGPUTS("calling socket\n");
|
|
|
567 listener->callback = params[1];
|
|
|
568 listener->sockfd = socket(PF_INET, SOCK_STREAM, 0);
|
|
|
569 listener->program = entry->instance->def->program;
|
|
|
570 if(listener->sockfd == -1)
|
|
|
571 {
|
|
|
572 release_ref(params[1]);
|
|
|
573 VIS_FREE(listener, "net listener object");
|
|
|
574 params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
|
|
|
575 params[1]->c.integers.num_a = 0;
|
|
|
576 params[0] = NULL;
|
|
|
577 return 0;
|
|
|
578 }
|
|
|
579 DEBUGPRINTF("Socket: %X\n", listener->sockfd);
|
|
|
580
|
|
|
581 my_address.sin_family = AF_INET;
|
|
|
582 my_address.sin_port = htons(port);
|
|
|
583 my_address.sin_addr.s_addr = INADDR_ANY;
|
|
|
584 memset(&(my_address.sin_zero), '\0', 8);
|
|
|
585
|
|
|
586 DEBUGPRINTF("Calling bind on port: %d\n", port);
|
|
|
587 junk = bind(listener->sockfd, (struct sockaddr *)&my_address, sizeof(struct sockaddr));
|
|
|
588 DEBUGPRINTF("Bind returned: %d\n", junk);
|
|
|
589 if(junk == -1)
|
|
|
590 {
|
|
|
591 perror("bind");
|
|
|
592 DEBUGPUTS("bind failed, releasing ref to callback worker\n");
|
|
|
593 release_ref(params[1]);
|
|
|
594 DEBUGPUTS("Closing socket\n");
|
|
|
595 DEBUGPRINTF("Socket: %X\n", listener->sockfd);
|
|
|
596 VIS_CloseSocket(listener->sockfd);
|
|
|
597 DEBUGPUTS("Freeing listener data\n");
|
|
|
598 VIS_FREE(listener, "net listener object");
|
|
|
599 DEBUGPUTS("Allocating error output\n");
|
|
|
600 params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
|
|
|
601 params[1]->c.integers.num_a = 0;
|
|
|
602 params[0] = NULL;
|
|
|
603 return 0;
|
|
|
604 }
|
|
|
605
|
|
|
606 DEBUGPUTS("Calling listen\n");
|
|
|
607 if(listen(listener->sockfd, 10) == -1)
|
|
|
608 {
|
|
|
609 release_ref(params[1]);
|
|
|
610 VIS_CloseSocket(listener->sockfd);
|
|
|
611 VIS_FREE(listener, "net listener object");
|
|
|
612 params[1] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
|
|
|
613 params[1]->c.integers.num_a = 0;
|
|
|
614 params[0] = NULL;
|
|
|
615 return 0;
|
|
|
616 }
|
|
|
617 DEBUGPUTS("Creating new thread\n");
|
|
|
618 VIS_NewThread(listen_thread, listener);
|
|
|
619 params[0] = new_datum(BUILTIN_TYPE_YESNO, 2, 0, entry->instance->def->program);
|
|
|
620 params[0]->c.integers.num_a = 0;
|
|
|
621 params[1] = NULL;
|
|
|
622 return 0;
|
|
|
623 }
|
|
|
624
|
|
|
625
|
