11 #define LISTEN_PORT 2156
12 #define BACKLOG 10 //max pending connections
13 #define RECEIVE_BUFFER_SIZE 2048
15 extern int classsize[];
17 objstr_t *mainobjstore;
21 //todo:initialize main object store
22 //do we want this to be a global variable, or provide
23 //separate access funtions and hide the structure?
24 mainobjstore = objstrCreate(DEFAULT_OBJ_STORE_SIZE);
25 if (mhashCreate(HASH_SIZE, LOADFACTOR))
28 if (lhashCreate(HASH_SIZE, LOADFACTOR))
31 //pthread_t threadListen;
32 //pthread_create(&threadListen, NULL, dstmListen, NULL);
39 int listenfd, acceptfd;
40 struct sockaddr_in my_addr;
41 struct sockaddr_in client_addr;
42 socklen_t addrlength = sizeof(struct sockaddr);
43 pthread_t thread_dstm_accept;
46 listenfd = socket(AF_INET, SOCK_STREAM, 0);
53 my_addr.sin_family = AF_INET;
54 my_addr.sin_port = htons(LISTEN_PORT);
55 my_addr.sin_addr.s_addr = INADDR_ANY;
56 memset(&(my_addr.sin_zero), '\0', 8);
58 if (bind(listenfd, (struct sockaddr *)&my_addr, addrlength) == -1)
64 if (listen(listenfd, BACKLOG) == -1)
70 printf("Listening on port %d, fd = %d\n", LISTEN_PORT, listenfd);
73 acceptfd = accept(listenfd, (struct sockaddr *)&client_addr, &addrlength);
74 pthread_create(&thread_dstm_accept, NULL, dstmAccept, (void *)acceptfd);
79 void *dstmAccept(void *acceptfd)
81 int numbytes,i,choice, oid;
82 char buffer[RECEIVE_BUFFER_SIZE], control;
85 int fd_flags = fcntl((int)acceptfd, F_GETFD), size;
87 printf("Recieved connection: fd = %d\n", (int)acceptfd);
88 numbytes = recv((int)acceptfd, (void *)buffer, sizeof(buffer), 0);
99 oid = *((int *)(buffer+1));
101 printf("DEBUG -> Received oid is %d\n", oid);
103 srcObj = mhashSearch(oid);
104 h = (objheader_t *) srcObj;
106 buffer[0] = OBJECT_NOT_FOUND;
108 buffer[0] = OBJECT_FOUND;
109 size = sizeof(objheader_t) + sizeof(classsize[h->type]);
110 memcpy(buffer+1, srcObj, size);
113 printf("DEBUG -> Sending oid = %d, type %d\n", h->oid, h->type);
116 if(send((int)acceptfd, (void *)buffer, sizeof(buffer), 0) < 0) {
120 case READ_MULT_REQUEST:
124 case MOVE_MULT_REQUEST:
127 printf("Client sent %d\n",buffer[0]);
129 printf("Num Read %d\n",*((short*)(buffer+offset)));
130 offset += sizeof(short);
131 printf("Num modified %d\n",*((short*)(buffer+offset)));
132 handleTransReq(acceptfd, buffer);
139 printf("Error receiving");
141 //printf("Read %d bytes from %d\n", numbytes, (int)acceptfd);
142 //printf("%s", buffer);
144 if (close((int)acceptfd) == -1)
149 printf("Closed connection: fd = %d\n", (int)acceptfd);
153 //TOOD put __FILE__ __LINE__ for all error conditions
154 int handleTransReq(int acceptfd, char *buf) {
155 short numread = 0, nummod = 0;
157 int offset = 0, size,i;
158 int objnotfound = 0, transdis = 0, transabort = 0, transagree = 0;
159 objheader_t *headptr = NULL;
162 char sendbuf[RECEIVE_BUFFER_SIZE];
166 numread = *((short *)(buf+offset));
167 offset += sizeof(short);
168 nummod = *((short *)(buf+offset));
169 offset += sizeof(short);
171 //Make an array to store the object headers for all objects that are only read
172 if ((headptr = calloc(numread, sizeof(objheader_t))) == NULL) {
173 perror("handleTransReq: Calloc error");
176 //Process each object id that is only read
177 for (i = 0; i < numread; i++) {
179 tmp = (objheader_t *) (buf + offset);
180 //find if object is still present in the same machine since TRANS_REQUEST
181 if ((mobj = mhashSearch(tmp->oid)) == NULL) {
184 sendbuf[0] = OBJECT_NOT_FOUND;
185 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
189 } else { // If obj found in machine (i.e. has not moved)
190 //Check if obj is locked
191 if ((((objheader_t *)mobj)->status >> 3) == 1) {
192 //Check version of the object
193 if (tmp->version == ((objheader_t *)mobj)->version) {//If version match
196 sendbuf[0] = TRANS_DISAGREE;
197 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
201 } else {//If versions don't match ..HARD ABORT
204 sendbuf[0] = TRANS_DISAGREE_ABORT;
205 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
210 } else {// If object not locked then lock it
211 ((objheader_t *)mobj)->status |= LOCK;
212 if (tmp->version == ((objheader_t *)mobj)->version) {//If versions match
215 sendbuf[0] = TRANS_AGREE;
216 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
220 } else {//If versions don't match
223 sendbuf[0] = TRANS_DISAGREE_ABORT;
224 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
231 memcpy(headptr, buf+offset, sizeof(objheader_t));
232 offset += sizeof(objheader_t);
236 if((tmpholder = objstrCreate(RECEIVE_BUFFER_SIZE)) == NULL) {
237 perror("handleTransReq: Calloc error");
241 //Process each object id that is only modified
242 for(i = 0; i < nummod; i++) {
244 tmp = (objheader_t *)(buf + offset);
245 //find if object is still present in the same machine since TRANS_REQUEST
246 if ((mobj = mhashSearch(tmp->oid)) == NULL) {
249 sendbuf[0] = OBJECT_NOT_FOUND;
250 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
254 } else { // If obj found in machine (i.e. has not moved)
255 //Check if obj is locked
256 if ((((objheader_t *)mobj)->status >> 3) == 1) {
257 //Check version of the object
258 if (tmp->version == ((objheader_t *)mobj)->version) {//If version match
261 sendbuf[0] = TRANS_DISAGREE;
262 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
266 } else {//If versions don't match ..HARD ABORT
269 sendbuf[0] = TRANS_DISAGREE_ABORT;
270 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
275 } else {// If object not locked then lock it
276 ((objheader_t *)mobj)->status |= LOCK;
277 if (tmp->version == ((objheader_t *)mobj)->version) {//If versions match
280 sendbuf[0] = TRANS_AGREE;
281 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
285 } else {//If versions don't match
288 sendbuf[0] = TRANS_DISAGREE_ABORT;
289 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
297 size = sizeof(objheader_t) + classsize[tmp->type];
298 if ((top = objstrAlloc(tmpholder, size)) == NULL) {
299 perror("handleTransReq: Calloc error");
302 memcpy(top, buf+offset, size);
307 sendbuf[0] = TRANS_DISAGREE_ABORT;
308 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {
313 sendbuf[0] = TRANS_AGREE;
314 if(send((int)acceptfd, (void *)sendbuf, sizeof(sendbuf), 0) < 0) {