void tw_net_cancel(tw_event *e) { tw_pe *src_pe = e->src_lp->pe; switch (e->state.owner) { case TW_net_outq: /* Cancelled before we could transmit it. Do not * transmit the event and instead just release the * buffer back into our own free list. */ tw_eventq_delete_any(&outq, e); tw_event_free(src_pe, e); return; break; case TW_net_asend: /* Too late. We've already let MPI start to send * this event over the network. We can't pull it * back now without sending another message to do * the cancel. * * Setting the cancel_q flag will signal us to do * another message send once the current send of * this message is completed. */ e->state.cancel_asend = 1; break; case TW_pe_sevent_q: /* Way late; the event was already sent and is in * our sent event queue. Mark it as a cancel and * place it at the front of the outq. */ e->state.cancel_q = 1; tw_eventq_unshift(&outq, e); break; default: /* Huh? Where did you come from? Why are we being * told about you? We did not send you so we cannot * cancel you! */ tw_error( TW_LOC, "Don't know how to cancel event owned by %u", e->state.owner); } service_queues(src_pe); }
void tw_net_send(tw_event *e) { tw_pe * me = e->src_lp->pe; int changed = 0; e->state.remote = 0; e->state.owner = TW_net_outq; tw_eventq_unshift(&outq, e); do { changed = test_q(&posted_sends, me, send_finish); changed |= send_begin(me); } while (changed); }
tw_event * tw_socket_read_event(tw_pe * me) { tw_net_node *node = g_tw_net_node[me->id]; tw_event *recv_event; tw_event *cancel_event; #ifdef ROSS_MEMORY_LIB tw_memory *last; tw_memory *memory; #endif //tw_message *temp_message; void *temp_data; //tw_pe *send_pe; tw_peid send_peid; tw_pe *dest_pe; int rv; unsigned int i; #ifdef ROSS_MEMORY_LIB void *temp_mem_data; size_t mem_size; tw_fd mem_fd; #endif rv = 0; /* * Get a free event from our freeq and save the pointers * to the message and the data for later use. */ if(me->abort_event == (recv_event = tw_event_grab(me))) return NULL; //temp_message = recv_event->message; //temp_data = recv_event->message->data; temp_data = recv_event + 1; /* * Attempt to read an event, and return NULL if no more events to recv. */ for (i = 0; i < nnet_nodes - g_tw_npe; i++) { rv = tw_socket_read(node->clients[i], (char *) recv_event, sizeof(tw_event) + g_tw_msg_sz, 100); if (rv > 0) break; } /* * Check to see if we actually read an event */ if (1 > rv) { if(recv_event != me->abort_event) { recv_event->event_id = 0; tw_eventq_unshift(&me->free_q, recv_event); } return NULL; } if (recv_event == me->abort_event) tw_error(TW_LOC, "Out of memory! Allocate more events!"); if(recv_event->recv_ts < me->GVT) tw_error(TW_LOC, "Received straggler event!"); /* * Restore recv'ed event's pointers * * on recv'rs side: have dest_lp ptr, not src_lp ptr */ //recv_event->dest_lp = tw_getlp((tw_lpid)recv_event->dest_lp); //recv_event->src_lp = tw_getlp((tw_lpid)recv_event->src_lp); //recv_event->message = temp_message; //recv_event->message->data = temp_data; recv_event->dest_lp = tw_getlocal_lp((tw_lpid) recv_event->dest_lp); //send_pe = recv_event->src_lp->pe; send_peid = (recv_event->dest_lp->type.map) ((tw_lpid) recv_event->src_lp); if(send_peid == me->id) tw_error(TW_LOC, "Sent event over network to self?"); if (recv_event->recv_ts > g_tw_ts_end) tw_error(TW_LOC, "%d: Received remote event at %d, end=%d!", recv_event->dest_lp->id, recv_event->recv_ts, g_tw_ts_end); if(recv_event->dest_lp->pe != me) tw_error(TW_LOC, "Not destination PE!"); /* * If a CANCEL message, just get the event out of hash table * and call * tw_event_cancel() on it, which rolls it back if nec */ if(recv_event->state.owner == TW_net_acancel) { #if VERIFY_SOCKET_TCP printf ("\t\t\t\t\t\t\t\tREAD CANCEL: dest p%d l%d: ts=%f sn=%d\n", recv_event->dest_lp->pe->id, recv_event->dest_lp->id, recv_event->recv_ts, recv_event->event_id); #endif cancel_event = NULL; cancel_event = tw_hash_remove(me->hash_t, recv_event, send_peid); dest_pe = cancel_event->dest_lp->pe; cancel_event->state.cancel_q = 1; cancel_event->state.remote = 0; if(cancel_event == recv_event) tw_error(TW_LOC, "cancel_event == recv_event!"); if(cancel_event->state.owner == 0 || cancel_event->state.owner == TW_pe_free_q) tw_error(TW_LOC, "cancel_event no owner!"); tw_mutex_lock(&dest_pe->cancel_q_lck); cancel_event->cancel_next = dest_pe->cancel_q; dest_pe->cancel_q = cancel_event; tw_mutex_unlock(&dest_pe->cancel_q_lck); recv_event->event_id = recv_event->state.cancel_q = 0; recv_event->state.remote = 0; tw_event_free(me, recv_event); return cancel_event; } recv_event->next = NULL; //recv_event->lp_state = NULL; recv_event->cancel_next = NULL; recv_event->caused_by_me = NULL; recv_event->cause_next = NULL; // signals for on-the-fly fossil collection recv_event->state.remote = 1; tw_hash_insert(me->hash_t, recv_event, send_peid); #if VERIFY_SOCKET_TCP printf ("\t\t\t\t\t\t\t\tREAD NORMAL: dest p%d l%d: ts=%f sn=%d src p%d l%d \n", recv_event->dest_lp->pe->id, recv_event->dest_lp->id, recv_event->recv_ts, recv_event->seq_num, recv_event->src_lp->pe->id, recv_event->src_lp->id); #endif #ifdef ROSS_MEMORY_LIB mem_size = (size_t) recv_event->memory; mem_fd = (tw_fd) recv_event->prev; last = NULL; while(mem_size) { memory = tw_memory_alloc(recv_event->src_lp, mem_fd); temp_mem_data = memory->data; if(last) last->next = memory; else recv_event->memory = memory; rv = 0; while(rv != mem_size) { rv = tw_socket_read(node->clients[i], (char *) memory, mem_size, 100); } memory->data = temp_mem_data; memory->prev = (tw_memory *) mem_fd; #if VERIFY_SOCKET_TCP printf("recv\'d mem buf of size %d on event %f\n", rv, recv_event->recv_ts); #endif mem_size = (size_t) memory->next; mem_fd = (tw_fd) memory->prev; last = memory; } #endif recv_event->prev = NULL; return recv_event; }