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