end of reformatting
[IRC.git] / Robust / src / Runtime / DSTM / interface / dstmserver.c
1 /* Coordinator => Machine that initiates the transaction request call for commiting a transaction
2  * Participant => Machines that host the objects involved in a transaction commit */
3
4 #include <netinet/tcp.h>
5 #include "dstm.h"
6 #include "mlookup.h"
7 #include "llookup.h"
8 #include "threadnotify.h"
9 #ifdef COMPILER
10 #include "thread.h"
11 #endif
12
13 #define BACKLOG 10 //max pending connections
14 #define RECEIVE_BUFFER_SIZE 2048
15
16 extern int classsize[];
17 extern int numHostsInSystem;
18 extern pthread_mutex_t notifymutex;
19
20 objstr_t *mainobjstore;
21 pthread_mutex_t mainobjstore_mutex;
22 pthread_mutex_t lockObjHeader;
23 pthread_mutexattr_t mainobjstore_mutex_attr; /* Attribute for lock to make it a recursive lock */
24
25 sockPoolHashTable_t *transPResponseSocketPool;
26
27 /* This function initializes the main objects store and creates the
28  * global machine and location lookup table */
29
30 int dstmInit(void) {
31   mainobjstore = objstrCreate(DEFAULT_OBJ_STORE_SIZE);
32   /* Initialize attribute for mutex */
33   pthread_mutexattr_init(&mainobjstore_mutex_attr);
34   pthread_mutexattr_settype(&mainobjstore_mutex_attr, PTHREAD_MUTEX_RECURSIVE_NP);
35   pthread_mutex_init(&mainobjstore_mutex, &mainobjstore_mutex_attr);
36   pthread_mutex_init(&lockObjHeader,NULL);
37   if (mhashCreate(HASH_SIZE, LOADFACTOR))
38     return 1;             //failure
39
40   if (lhashCreate(HASH_SIZE, LOADFACTOR))
41     return 1;             //failure
42
43   if (notifyhashCreate(N_HASH_SIZE, N_LOADFACTOR))
44     return 1;             //failure
45
46   //Initialize socket pool
47   if((transPResponseSocketPool = createSockPool(transPResponseSocketPool, DEFAULTSOCKPOOLSIZE)) == NULL) {
48     printf("Error in creating new socket pool at  %s line %d\n", __FILE__, __LINE__);
49     return 0;
50   }
51
52   return 0;
53 }
54
55
56 int startlistening() {
57   int listenfd;
58   struct sockaddr_in my_addr;
59   socklen_t addrlength = sizeof(struct sockaddr);
60   int setsockflag=1;
61
62   listenfd = socket(AF_INET, SOCK_STREAM, 0);
63   if (listenfd == -1) {
64     perror("socket");
65     exit(1);
66   }
67
68   if (setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &setsockflag, sizeof (setsockflag)) < 0) {
69     perror("socket");
70     exit(1);
71   }
72 #ifdef MAC
73   if (setsockopt(listenfd, SOL_SOCKET, SO_NOSIGPIPE, &setsockflag, sizeof (setsockflag)) < 0) {
74     perror("socket");
75     exit(1);
76   }
77 #endif
78
79   my_addr.sin_family = AF_INET;
80   my_addr.sin_port = htons(LISTEN_PORT);
81   my_addr.sin_addr.s_addr = INADDR_ANY;
82   memset(&(my_addr.sin_zero), '\0', 8);
83
84   if (bind(listenfd, (struct sockaddr *)&my_addr, addrlength) == -1) {
85     perror("bind");
86     exit(1);
87   }
88
89   if (listen(listenfd, BACKLOG) == -1) {
90     perror("listen");
91     exit(1);
92   }
93   return listenfd;
94 }
95
96 /* This function starts the thread to listen on a socket
97  * for tranaction calls */
98 void *dstmListen(void *lfd) {
99   int listenfd=(int)lfd;
100   int acceptfd;
101   struct sockaddr_in client_addr;
102   socklen_t addrlength = sizeof(struct sockaddr);
103   pthread_t thread_dstm_accept;
104
105   printf("Listening on port %d, fd = %d\n", LISTEN_PORT, listenfd);
106   while(1) {
107     int retval;
108     int flag=1;
109     acceptfd = accept(listenfd, (struct sockaddr *)&client_addr, &addrlength);
110     setsockopt(acceptfd, IPPROTO_TCP, TCP_NODELAY, (char *) &flag, sizeof(flag));
111     do {
112       retval=pthread_create(&thread_dstm_accept, NULL, dstmAccept, (void *)acceptfd);
113     } while(retval!=0);
114     pthread_detach(thread_dstm_accept);
115   }
116 }
117 /* This function accepts a new connection request, decodes the control message in the connection
118  * and accordingly calls other functions to process new requests */
119 void *dstmAccept(void *acceptfd) {
120   int val, retval, size, sum, sockid;
121   unsigned int oid;
122   char *buffer;
123   char control,ctrl;
124   char *ptr;
125   void *srcObj;
126   objheader_t *h;
127   trans_commit_data_t transinfo;
128   unsigned short objType, *versionarry, version;
129   unsigned int *oidarry, numoid, mid, threadid;
130
131   /* Receive control messages from other machines */
132   while(1) {
133     int ret=recv_data_errorcode((int)acceptfd, &control, sizeof(char));
134     if (ret==0)
135       break;
136     if (ret==-1) {
137       printf("DEBUG -> RECV Error!.. retrying\n");
138       break;
139     }
140     switch(control) {
141     case READ_REQUEST:
142       /* Read oid requested and search if available */
143       recv_data((int)acceptfd, &oid, sizeof(unsigned int));
144       if((srcObj = mhashSearch(oid)) == NULL) {
145         printf("Error: Object 0x%x is not found in Main Object Store %s, %d\n", oid, __FILE__, __LINE__);
146         break;
147       }
148       h = (objheader_t *) srcObj;
149       GETSIZE(size, h);
150       size += sizeof(objheader_t);
151       sockid = (int) acceptfd;
152       if (h == NULL) {
153         ctrl = OBJECT_NOT_FOUND;
154         send_data(sockid, &ctrl, sizeof(char));
155       } else {
156         // Type
157         char msg[]={OBJECT_FOUND, 0, 0, 0, 0};
158         *((int *)&msg[1])=size;
159         send_data(sockid, &msg, sizeof(msg));
160         send_data(sockid, h, size);
161       }
162       break;
163
164     case READ_MULT_REQUEST:
165       break;
166
167     case MOVE_REQUEST:
168       break;
169
170     case MOVE_MULT_REQUEST:
171       break;
172
173     case TRANS_REQUEST:
174       /* Read transaction request */
175       transinfo.objlocked = NULL;
176       transinfo.objnotfound = NULL;
177       transinfo.modptr = NULL;
178       transinfo.numlocked = 0;
179       transinfo.numnotfound = 0;
180       if((val = readClientReq(&transinfo, (int)acceptfd)) != 0) {
181         printf("Error: In readClientReq() %s, %d\n", __FILE__, __LINE__);
182         pthread_exit(NULL);
183       }
184       break;
185
186     case TRANS_PREFETCH:
187       if((val = prefetchReq((int)acceptfd)) != 0) {
188         printf("Error: In prefetchReq() %s, %d\n", __FILE__, __LINE__);
189         break;
190       }
191       break;
192
193     case TRANS_PREFETCH_RESPONSE:
194       if((val = getPrefetchResponse((int) acceptfd)) != 0) {
195         printf("Error: In getPrefetchResponse() %s, %d\n", __FILE__, __LINE__);
196         break;
197       }
198       break;
199
200     case START_REMOTE_THREAD:
201       recv_data((int)acceptfd, &oid, sizeof(unsigned int));
202       objType = getObjType(oid);
203       startDSMthread(oid, objType);
204       break;
205
206     case THREAD_NOTIFY_REQUEST:
207       recv_data((int)acceptfd, &numoid, sizeof(unsigned int));
208       size = (sizeof(unsigned int) + sizeof(unsigned short)) * numoid + 2 * sizeof(unsigned int);
209       if((buffer = calloc(1,size)) == NULL) {
210         printf("%s() Calloc error at %s, %d\n", __func__, __FILE__, __LINE__);
211         pthread_exit(NULL);
212       }
213
214       recv_data((int)acceptfd, buffer, size);
215
216       oidarry = calloc(numoid, sizeof(unsigned int));
217       memcpy(oidarry, buffer, sizeof(unsigned int) * numoid);
218       size = sizeof(unsigned int) * numoid;
219       versionarry = calloc(numoid, sizeof(unsigned short));
220       memcpy(versionarry, buffer+size, sizeof(unsigned short) * numoid);
221       size += sizeof(unsigned short) * numoid;
222       mid = *((unsigned int *)(buffer+size));
223       size += sizeof(unsigned int);
224       threadid = *((unsigned int *)(buffer+size));
225       processReqNotify(numoid, oidarry, versionarry, mid, threadid);
226       free(buffer);
227
228       break;
229
230     case THREAD_NOTIFY_RESPONSE:
231       size = sizeof(unsigned short) + 2 * sizeof(unsigned int);
232       if((buffer = calloc(1,size)) == NULL) {
233         printf("%s() Calloc error at %s, %d\n", __func__, __FILE__, __LINE__);
234         pthread_exit(NULL);
235       }
236
237       recv_data((int)acceptfd, buffer, size);
238
239       oid = *((unsigned int *)buffer);
240       size = sizeof(unsigned int);
241       version = *((unsigned short *)(buffer+size));
242       size += sizeof(unsigned short);
243       threadid = *((unsigned int *)(buffer+size));
244       threadNotify(oid,version,threadid);
245       free(buffer);
246       break;
247
248     case CLOSE_CONNECTION:
249       goto closeconnection;
250
251     default:
252       printf("Error: dstmAccept() Unknown opcode %d at %s, %d\n", control, __FILE__, __LINE__);
253     }
254   }
255
256 closeconnection:
257   /* Close connection */
258   if (close((int)acceptfd) == -1)
259     perror("close");
260   pthread_exit(NULL);
261 }
262
263 /* This function reads the information available in a transaction request
264  * and makes a function call to process the request */
265 int readClientReq(trans_commit_data_t *transinfo, int acceptfd) {
266   char *ptr;
267   void *modptr;
268   unsigned int *oidmod, oid;
269   fixed_data_t fixed;
270   objheader_t *headaddr;
271   int sum, i, size, n, val;
272
273   oidmod = NULL;
274
275   /* Read fixed_data_t data structure */
276   size = sizeof(fixed) - 1;
277   ptr = (char *)&fixed;;
278   fixed.control = TRANS_REQUEST;
279   recv_data((int)acceptfd, ptr+1, size);
280
281   /* Read list of mids */
282   int mcount = fixed.mcount;
283   size = mcount * sizeof(unsigned int);
284   unsigned int listmid[mcount];
285   ptr = (char *) listmid;
286   recv_data((int)acceptfd, ptr, size);
287
288   /* Read oid and version tuples for those objects that are not modified in the transaction */
289   int numread = fixed.numread;
290   size = numread * (sizeof(unsigned int) + sizeof(unsigned short));
291   char objread[size];
292   if(numread != 0) { //If pile contains more than one object to be read,
293     // keep reading all objects
294     recv_data((int)acceptfd, objread, size);
295   }
296
297   /* Read modified objects */
298   if(fixed.nummod != 0) {
299     if ((modptr = calloc(1, fixed.sum_bytes)) == NULL) {
300       printf("calloc error for modified objects %s, %d\n", __FILE__, __LINE__);
301       return 1;
302     }
303     size = fixed.sum_bytes;
304     recv_data((int)acceptfd, modptr, size);
305   }
306
307   /* Create an array of oids for modified objects */
308   oidmod = (unsigned int *) calloc(fixed.nummod, sizeof(unsigned int));
309   if (oidmod == NULL) {
310     printf("calloc error %s, %d\n", __FILE__, __LINE__);
311     return 1;
312   }
313   ptr = (char *) modptr;
314   for(i = 0 ; i < fixed.nummod; i++) {
315     int tmpsize;
316     headaddr = (objheader_t *) ptr;
317     oid = OID(headaddr);
318     oidmod[i] = oid;
319     GETSIZE(tmpsize, headaddr);
320     ptr += sizeof(objheader_t) + tmpsize;
321   }
322
323   /*Process the information read */
324   if((val = processClientReq(&fixed, transinfo, listmid, objread, modptr, oidmod, acceptfd)) != 0) {
325     printf("Error: In processClientReq() %s, %d\n", __FILE__, __LINE__);
326     /* Free resources */
327     if(oidmod != NULL) {
328       free(oidmod);
329     }
330     return 1;
331   }
332
333   /* Free resources */
334   if(oidmod != NULL) {
335     free(oidmod);
336   }
337
338   return 0;
339 }
340
341 /* This function processes the Coordinator's transaction request using "handleTransReq"
342  * function and sends a reply to the co-ordinator.
343  * Following this it also receives a new control message from the co-ordinator and processes this message*/
344 int processClientReq(fixed_data_t *fixed, trans_commit_data_t *transinfo,
345                      unsigned int *listmid, char *objread, void *modptr, unsigned int *oidmod, int acceptfd) {
346   char control, sendctrl, retval;
347   objheader_t *tmp_header;
348   void *header;
349   int i = 0, val;
350
351   /* Send reply to the Coordinator */
352   if((retval = handleTransReq(fixed, transinfo, listmid, objread, modptr,acceptfd)) == 0 ) {
353     printf("Error: In handleTransReq() %s, %d\n", __FILE__, __LINE__);
354     return 1;
355   }
356
357   recv_data((int)acceptfd, &control, sizeof(char));
358
359   /* Process the new control message */
360   switch(control) {
361   case TRANS_ABORT:
362     if (fixed->nummod > 0)
363       free(modptr);
364     /* Unlock objects that was locked due to this transaction */
365     for(i = 0; i< transinfo->numlocked; i++) {
366       if((header = mhashSearch(transinfo->objlocked[i])) == NULL) {
367         printf("mhashSearch returns NULL at %s, %d\n", __FILE__, __LINE__);                              // find the header address
368         return 1;
369       }
370       UnLock(STATUSPTR(header));
371     }
372
373     /* Send ack to Coordinator */
374     sendctrl = TRANS_UNSUCESSFUL;
375     send_data((int)acceptfd, &sendctrl, sizeof(char));
376     break;
377
378   case TRANS_COMMIT:
379     /* Invoke the transCommit process() */
380     if((val = transCommitProcess(modptr, oidmod, transinfo->objlocked, fixed->nummod, transinfo->numlocked, (int)acceptfd)) != 0) {
381       printf("Error: In transCommitProcess() %s, %d\n", __FILE__, __LINE__);
382       /* Free memory */
383       if (transinfo->objlocked != NULL) {
384         free(transinfo->objlocked);
385       }
386       if (transinfo->objnotfound != NULL) {
387         free(transinfo->objnotfound);
388       }
389       return 1;
390     }
391     break;
392
393   case TRANS_ABORT_BUT_RETRY_COMMIT_WITH_RELOCATING:
394     break;
395
396   default:
397     printf("Error: No response to TRANS_AGREE OR DISAGREE protocol %s, %d\n", __FILE__, __LINE__);
398     //TODO Use fixed.trans_id  TID since Client may have died
399     break;
400   }
401
402   /* Free memory */
403   if (transinfo->objlocked != NULL) {
404     free(transinfo->objlocked);
405   }
406   if (transinfo->objnotfound != NULL) {
407     free(transinfo->objnotfound);
408   }
409   return 0;
410 }
411
412 /* This function increments counters while running a voting decision on all objects involved
413  * in TRANS_REQUEST and If a TRANS_DISAGREE sends the response immediately back to the coordinator */
414 char handleTransReq(fixed_data_t *fixed, trans_commit_data_t *transinfo, unsigned int *listmid, char *objread, void *modptr, int acceptfd) {
415   int val, i = 0, j;
416   unsigned short version;
417   char control = 0, *ptr;
418   unsigned int oid;
419   unsigned int *oidnotfound, *oidlocked, *oidvernotmatch;
420   void *mobj;
421   objheader_t *headptr;
422
423   /* Counters and arrays to formulate decision on control message to be sent */
424   oidnotfound = (unsigned int *) calloc(fixed->numread + fixed->nummod, sizeof(unsigned int));
425   oidlocked = (unsigned int *) calloc(fixed->numread + fixed->nummod, sizeof(unsigned int));
426   oidvernotmatch = (unsigned int *) calloc(fixed->numread + fixed->nummod, sizeof(unsigned int));
427   int objnotfound = 0, objlocked = 0, objvernotmatch = 0;
428   int v_nomatch = 0, v_matchlock = 0, v_matchnolock = 0;
429   int numBytes = 0;
430   /* modptr points to the beginning of the object store
431    * created at the Pariticipant.
432    * Object store holds the modified objects involved in the transaction request */
433   ptr = (char *) modptr;
434
435   /* Process each oid in the machine pile/ group per thread */
436   for (i = 0; i < fixed->numread + fixed->nummod; i++) {
437     if (i < fixed->numread) {            //Objs only read and not modified
438       int incr = sizeof(unsigned int) + sizeof(unsigned short);                  // Offset that points to next position in the objread array
439       incr *= i;
440       oid = *((unsigned int *)(objread + incr));
441       incr += sizeof(unsigned int);
442       version = *((unsigned short *)(objread + incr));
443     } else {            //Objs modified
444       int tmpsize;
445       headptr = (objheader_t *) ptr;
446       oid = OID(headptr);
447       version = headptr->version;
448       GETSIZE(tmpsize, headptr);
449       ptr += sizeof(objheader_t) + tmpsize;
450     }
451
452     /* Check if object is still present in the machine since the beginning of TRANS_REQUEST */
453
454     if ((mobj = mhashSearch(oid)) == NULL) {    /* Obj not found */
455       /* Save the oids not found and number of oids not found for later use */
456       oidnotfound[objnotfound] = oid;
457       objnotfound++;
458     } else {     /* If Obj found in machine (i.e. has not moved) */
459       /* Check if Obj is locked by any previous transaction */
460       if (test_and_set(STATUSPTR(mobj))) {
461         //don't have lock
462         if (version == ((objheader_t *)mobj)->version) {          /* If locked then match versions */
463           v_matchlock++;
464         } else {    /* If versions don't match ...HARD ABORT */
465           v_nomatch++;
466           oidvernotmatch[objvernotmatch] = oid;
467           objvernotmatch++;
468           int size;
469           GETSIZE(size, mobj);
470           size += sizeof(objheader_t);
471           numBytes += size;
472           /* Send TRANS_DISAGREE to Coordinator */
473           control = TRANS_DISAGREE;
474         }
475       } else {    /* If Obj is not locked then lock object */
476         /* Save all object oids that are locked on this machine during this transaction request call */
477         oidlocked[objlocked] = OID(((objheader_t *)mobj));
478         objlocked++;
479         if (version == ((objheader_t *)mobj)->version) {     /* Check if versions match */
480           v_matchnolock++;
481         } else {     /* If versions don't match ...HARD ABORT */
482           v_nomatch++;
483           oidvernotmatch[objvernotmatch] = oid;
484           objvernotmatch++;
485           int size;
486           GETSIZE(size, mobj);
487           size += sizeof(objheader_t);
488           numBytes += size;
489           control = TRANS_DISAGREE;
490         }
491       }
492     }
493   }
494
495   /* send TRANS_DISAGREE and objs*/
496   if(v_nomatch > 0) {
497 #ifdef CACHE
498     char *objs = calloc(1, numBytes);
499     int j, offset = 0;
500     for(j = 0; j<objvernotmatch; j++) {
501       objheader_t *header = mhashSearch(oidvernotmatch[j]);
502       int size = 0;
503       GETSIZE(size, header);
504       size += sizeof(objheader_t);
505       memcpy(objs+offset, header, size);
506       offset += size;
507     }
508 #endif
509     if (objlocked > 0) {
510       for(j = 0; j < objlocked; j++) {
511         if((headptr = mhashSearch(oidlocked[j])) == NULL) {
512           printf("mhashSearch returns NULL at %s, %d\n", __FILE__, __LINE__);
513           return 0;
514         }
515         UnLock(STATUSPTR(headptr));
516       }
517       free(oidlocked);
518     }
519     send_data(acceptfd, &control, sizeof(char));
520 #ifdef CACHE
521     send_data(acceptfd, &numBytes, sizeof(int));
522     send_data(acceptfd, objs, numBytes);
523     transinfo->objvernotmatch = oidvernotmatch;
524     transinfo->numvernotmatch = objvernotmatch;
525     free(objs);
526     free(transinfo->objvernotmatch);
527 #endif
528     return control;
529   }
530
531   /* Decide what control message to send to Coordinator */
532   if ((control = decideCtrlMessage(fixed, transinfo, &v_matchnolock, &v_matchlock, &v_nomatch, &objnotfound, &objlocked,
533                                    modptr, oidnotfound, oidlocked, acceptfd)) == 0) {
534     printf("Error: In decideCtrlMessage() %s, %d\n", __FILE__, __LINE__);
535     return 0;
536   }
537
538   return control;
539 }
540 /* This function decides what control message such as TRANS_AGREE, TRANS_DISAGREE or TRANS_SOFT_ABORT
541  * to send to Coordinator based on the votes of oids involved in the transaction */
542 char decideCtrlMessage(fixed_data_t *fixed, trans_commit_data_t *transinfo, int *v_matchnolock, int *v_matchlock,
543                        int *v_nomatch, int *objnotfound, int *objlocked, void *modptr,
544                        unsigned int *oidnotfound, unsigned int *oidlocked, int acceptfd) {
545   int val;
546   char control = 0;
547
548   /* Condition to send TRANS_AGREE */
549   if(*(v_matchnolock) == fixed->numread + fixed->nummod) {
550     control = TRANS_AGREE;
551     /* Send control message */
552     send_data(acceptfd, &control, sizeof(char));
553   }
554   /* Condition to send TRANS_SOFT_ABORT */
555   if((*(v_matchlock) > 0 && *(v_nomatch) == 0) || (*(objnotfound) > 0 && *(v_nomatch) == 0)) {
556     control = TRANS_SOFT_ABORT;
557
558     /* Send control message */
559     send_data(acceptfd, &control, sizeof(char));
560
561     /*  FIXME how to send objs Send number of oids not found and the missing oids if objects are missing in the machine */
562     if(*(objnotfound) != 0) {
563       int msg[1];
564       msg[0] = *(objnotfound);
565       send_data(acceptfd, &msg, sizeof(int));
566       int size = sizeof(unsigned int)* *(objnotfound);
567       send_data(acceptfd, oidnotfound, size);
568     }
569   }
570
571   /* Fill out the trans_commit_data_t data structure. This is required for a trans commit process
572    * if Participant receives a TRANS_COMMIT */
573   transinfo->objlocked = oidlocked;
574   transinfo->objnotfound = oidnotfound;
575   transinfo->modptr = modptr;
576   transinfo->numlocked = *(objlocked);
577   transinfo->numnotfound = *(objnotfound);
578
579   return control;
580 }
581
582 /* This function processes all modified objects involved in a TRANS_COMMIT and updates pointer
583  * addresses in lookup table and also changes version number
584  * Sends an ACK back to Coordinator */
585 int transCommitProcess(void *modptr, unsigned int *oidmod, unsigned int *oidlocked, int nummod, int numlocked, int acceptfd) {
586   objheader_t *header;
587   objheader_t *newheader;
588   int i = 0, offset = 0;
589   char control;
590   int tmpsize;
591
592   /* Process each modified object saved in the mainobject store */
593   for(i = 0; i < nummod; i++) {
594     if((header = (objheader_t *) mhashSearch(oidmod[i])) == NULL) {
595       printf("Error: mhashsearch returns NULL at %s, %d\n", __FILE__, __LINE__);
596       return 1;
597     }
598     GETSIZE(tmpsize,header);
599     memcpy((char*)header + sizeof(objheader_t), ((char *)modptr + sizeof(objheader_t) + offset), tmpsize);
600     header->version += 1;
601     /* If threads are waiting on this object to be updated, notify them */
602     if(header->notifylist != NULL) {
603       notifyAll(&header->notifylist, OID(header), header->version);
604     }
605     offset += sizeof(objheader_t) + tmpsize;
606   }
607
608   if (nummod > 0)
609     free(modptr);
610
611   /* Unlock locked objects */
612   for(i = 0; i < numlocked; i++) {
613     if((header = (objheader_t *) mhashSearch(oidlocked[i])) == NULL) {
614       printf("Error: mhashsearch returns NULL at %s, %d\n", __FILE__, __LINE__);
615       return 1;
616     }
617     UnLock(STATUSPTR(header));
618   }
619   //TODO Update location lookup table
620
621   /* Send ack to coordinator */
622   control = TRANS_SUCESSFUL;
623   send_data((int)acceptfd, &control, sizeof(char));
624   return 0;
625 }
626
627 /* This function recevies the oid and offset tuples from the Coordinator's prefetch call.
628  * Looks for the objects to be prefetched in the main object store.
629  * If objects are not found then record those and if objects are found
630  * then use offset values to prefetch references to other objects */
631
632 int prefetchReq(int acceptfd) {
633   int i, size, objsize, numoffset = 0;
634   int length;
635   char *recvbuffer, control;
636   unsigned int oid, mid=-1;
637   objheader_t *header;
638   oidmidpair_t oidmid;
639   int sd = -1;
640
641   while(1) {
642     recv_data((int)acceptfd, &numoffset, sizeof(int));
643     if(numoffset == -1)
644       break;
645     recv_data((int)acceptfd, &oidmid, 2*sizeof(unsigned int));
646     oid = oidmid.oid;
647     if (mid != oidmid.mid) {
648       if (mid!=-1) {
649         freeSockWithLock(transPResponseSocketPool, mid, sd);
650       }
651       mid=oidmid.mid;
652       sd = getSockWithLock(transPResponseSocketPool, mid);
653     }
654     short offsetarry[numoffset];
655     recv_data((int) acceptfd, offsetarry, numoffset*sizeof(short));
656
657     /*Process each oid */
658     if ((header = mhashSearch(oid)) == NULL) { /* Obj not found */
659       /* Save the oids not found in buffer for later use */
660       size = sizeof(int) + sizeof(char) + sizeof(unsigned int) ;
661       char sendbuffer[size];
662       *((int *) sendbuffer) = size;
663       *((char *)(sendbuffer + sizeof(int))) = OBJECT_NOT_FOUND;
664       *((unsigned int *)(sendbuffer + sizeof(int) + sizeof(char))) = oid;
665       control = TRANS_PREFETCH_RESPONSE;
666       sendPrefetchResponse(sd, &control, sendbuffer, &size);
667     } else { /* Object Found */
668       int incr = 0;
669       GETSIZE(objsize, header);
670       size = sizeof(int) + sizeof(char) + sizeof(unsigned int) + sizeof(objheader_t) + objsize;
671       char sendbuffer[size];
672       *((int *)(sendbuffer + incr)) = size;
673       incr += sizeof(int);
674       *((char *)(sendbuffer + incr)) = OBJECT_FOUND;
675       incr += sizeof(char);
676       *((unsigned int *)(sendbuffer+incr)) = oid;
677       incr += sizeof(unsigned int);
678       memcpy(sendbuffer + incr, header, objsize + sizeof(objheader_t));
679
680       control = TRANS_PREFETCH_RESPONSE;
681       sendPrefetchResponse(sd, &control, sendbuffer, &size);
682
683       /* Calculate the oid corresponding to the offset value */
684       for(i = 0 ; i< numoffset ; i++) {
685         /* Check for arrays  */
686         if(TYPE(header) > NUMCLASSES) {
687           int elementsize = classsize[TYPE(header)];
688           struct ArrayObject *ao = (struct ArrayObject *) (((char *)header) + sizeof(objheader_t));
689           unsigned short length = ao->___length___;
690           /* Check if array out of bounds */
691           if(offsetarry[i]< 0 || offsetarry[i] >= length) {
692             break;
693           }
694           oid = *((unsigned int *)(((char *)header) + sizeof(objheader_t) + sizeof(struct ArrayObject) + (elementsize*offsetarry[i])));
695         } else {
696           oid = *((unsigned int *)(((char *)header) + sizeof(objheader_t) + offsetarry[i]));
697         }
698
699         /* Don't continue if we hit a NULL pointer */
700         if (oid==0)
701           break;
702
703         if((header = mhashSearch(oid)) == NULL) {
704           size = sizeof(int) + sizeof(char) + sizeof(unsigned int) ;
705           char sendbuffer[size];
706           *((int *) sendbuffer) = size;
707           *((char *)(sendbuffer + sizeof(int))) = OBJECT_NOT_FOUND;
708           *((unsigned int *)(sendbuffer + sizeof(int) + sizeof(char))) = oid;
709
710           control = TRANS_PREFETCH_RESPONSE;
711           sendPrefetchResponse(sd, &control, sendbuffer, &size);
712           break;
713         } else { /* Obj Found */
714           int incr = 0;
715           GETSIZE(objsize, header);
716           size = sizeof(int) + sizeof(char) + sizeof(unsigned int) + sizeof(objheader_t) + objsize;
717           char sendbuffer[size];
718           *((int *)(sendbuffer + incr)) = size;
719           incr += sizeof(int);
720           *((char *)(sendbuffer + incr)) = OBJECT_FOUND;
721           incr += sizeof(char);
722           *((unsigned int *)(sendbuffer+incr)) = oid;
723           incr += sizeof(unsigned int);
724           memcpy(sendbuffer + incr, header, objsize + sizeof(objheader_t));
725
726           control = TRANS_PREFETCH_RESPONSE;
727           sendPrefetchResponse(sd, &control, sendbuffer, &size);
728         }
729       }
730     }
731   }
732   //Release socket
733   if (mid!=-1)
734     freeSockWithLock(transPResponseSocketPool, mid, sd);
735
736   return 0;
737 }
738
739 void sendPrefetchResponse(int sd, char *control, char *sendbuffer, int *size) {
740   send_data(sd, control, sizeof(char));
741   /* Send the buffer with its size */
742   int length = *(size);
743   send_data(sd, sendbuffer, length);
744 }
745
746 void processReqNotify(unsigned int numoid, unsigned int *oidarry, unsigned short *versionarry, unsigned int mid, unsigned int threadid) {
747   objheader_t *header;
748   unsigned int oid;
749   unsigned short newversion;
750   char msg[1+  2 * sizeof(unsigned int) + sizeof(unsigned short)];
751   int sd;
752   struct sockaddr_in remoteAddr;
753   int bytesSent;
754   int size;
755   int i = 0;
756
757   while(i < numoid) {
758     oid = *(oidarry + i);
759     if((header = (objheader_t *) mhashSearch(oid)) == NULL) {
760       printf("Error: mhashsearch returns NULL at %s, %d\n", __FILE__, __LINE__);
761       return;
762     } else {
763       /* Check to see if versions are same */
764 checkversion:
765       if (test_and_set(STATUSPTR(header))==0) {
766         //have lock
767         newversion = header->version;
768         if(newversion == *(versionarry + i)) {
769           //Add to the notify list
770           if((header->notifylist = insNode(header->notifylist, threadid, mid)) == NULL) {
771             printf("Error: Obj notify list points to NULL %s, %d\n", __FILE__, __LINE__);
772             return;
773           }
774           UnLock(STATUSPTR(header));
775         } else {
776           UnLock(STATUSPTR(header));
777           if ((sd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
778             perror("processReqNotify():socket()");
779             return;
780           }
781           bzero(&remoteAddr, sizeof(remoteAddr));
782           remoteAddr.sin_family = AF_INET;
783           remoteAddr.sin_port = htons(LISTEN_PORT);
784           remoteAddr.sin_addr.s_addr = htonl(mid);
785
786           if (connect(sd, (struct sockaddr *)&remoteAddr, sizeof(remoteAddr)) < 0) {
787             printf("Error: processReqNotify():error %d connecting to %s:%d\n", errno,
788                    inet_ntoa(remoteAddr.sin_addr), LISTEN_PORT);
789             close(sd);
790             return;
791           } else {
792             //Send Update notification
793             msg[0] = THREAD_NOTIFY_RESPONSE;
794             *((unsigned int *)&msg[1]) = oid;
795             size = sizeof(unsigned int);
796             *((unsigned short *)(&msg[1]+size)) = newversion;
797             size += sizeof(unsigned short);
798             *((unsigned int *)(&msg[1]+size)) = threadid;
799             size = 1+ 2*sizeof(unsigned int) + sizeof(unsigned short);
800             send_data(sd, msg, size);
801           }
802           close(sd);
803         }
804       } else {
805         randomdelay();
806         goto checkversion;
807       }
808     }
809     i++;
810   }
811   free(oidarry);
812   free(versionarry);
813 }