int main(int argc, const char** argv) { int i; int ret; dtn_handle_t handle; dtn_endpoint_id_t source_eid; dtn_reg_info_t reginfo; dtn_reg_id_t regid; dtn_bundle_spec_t ping_spec; dtn_bundle_spec_t reply_spec; dtn_bundle_payload_t ping_payload; ping_payload_t payload_contents; ping_payload_t recv_contents; dtn_bundle_payload_t reply_payload; dtn_bundle_status_report_t* sr_data; dtn_bundle_id_t bundle_id; int debug = 1; char demux[64]; int dest_len = 0; struct timeval send_times[MAX_PINGS_IN_FLIGHT]; dtn_timestamp_t creation_times[MAX_PINGS_IN_FLIGHT]; struct timeval now, recv_start, recv_end; u_int32_t nonce; u_int32_t seqno = 0; int timeout; // force stdout to always be line buffered, even if output is // redirected to a pipe or file setvbuf(stdout, (char *)NULL, _IOLBF, 0); doOptions(argc, argv); memset(&ping_spec, 0, sizeof(ping_spec)); gettimeofday(&now, 0); srand(now.tv_sec); nonce = rand(); // open the ipc handle int err = dtn_open(&handle); if (err != DTN_SUCCESS) { fprintf(stderr, "fatal error opening dtn handle: %s\n", dtn_strerror(err)); exit(1); } // make sure they supplied a valid destination eid or // "localhost", in which case we just use the local daemon if (strcmp(dest_eid_str, "localhost") == 0) { dtn_build_local_eid(handle, &ping_spec.dest, "ping"); } else { if (dtn_parse_eid_string(&ping_spec.dest, dest_eid_str)) { fprintf(stderr, "invalid destination eid string '%s'\n", dest_eid_str); exit(1); } } dest_len = strlen(ping_spec.dest.uri); if ((dest_len < 4) || (strcmp(ping_spec.dest.uri + dest_len - 4, "ping") != 0)) { fprintf(stderr, "\nWARNING: ping destination does not end in \"ping\"\n\n"); } // if the user specified a source eid, register on it. // otherwise, build a local eid based on the configuration of // our dtn router plus the demux string snprintf(demux, sizeof(demux), "/ping.%d", getpid()); if (source_eid_str[0] != '\0') { if (dtn_parse_eid_string(&source_eid, source_eid_str)) { fprintf(stderr, "invalid source eid string '%s'\n", source_eid_str); exit(1); } } else { dtn_build_local_eid(handle, &source_eid, demux); } // set the source and replyto eids in the bundle spec if (debug) printf("source_eid [%s]\n", source_eid.uri); dtn_copy_eid(&ping_spec.source, &source_eid); dtn_copy_eid(&ping_spec.replyto, &source_eid); // now create a new registration based on the source memset(®info, 0, sizeof(reginfo)); dtn_copy_eid(®info.endpoint, &source_eid); reginfo.flags = DTN_REG_DEFER; reginfo.regid = DTN_REGID_NONE; reginfo.expiration = 0; if ((ret = dtn_register(handle, ®info, ®id)) != 0) { fprintf(stderr, "error creating registration: %d (%s)\n", ret, dtn_strerror(dtn_errno(handle))); exit(1); } if (debug) printf("dtn_register succeeded, regid %d\n", regid); // set the expiration time and request deletion status reports ping_spec.expiration = expiration; ping_spec.dopts = DOPTS_DELETE_RCPT; printf("PING [%s] (expiration %u)...\n", ping_spec.dest.uri, expiration); if (interval == 0) { printf("WARNING: zero second interval will result in flooding pings!!\n"); } // loop, sending pings and polling for activity for (i = 0; count == 0 || i < count; ++i) { gettimeofday(&send_times[seqno], NULL); // fill in a short payload string, a nonce, and a sequence number // to verify the echo feature and make sure we're not getting // duplicate responses or ping responses from another app memcpy(&payload_contents.ping, PING_STR, 8); payload_contents.seqno = seqno; payload_contents.nonce = nonce; payload_contents.time = send_times[seqno].tv_sec; memset(&ping_payload, 0, sizeof(ping_payload)); dtn_set_payload(&ping_payload, DTN_PAYLOAD_MEM, (char*)&payload_contents, sizeof(payload_contents)); memset(&bundle_id, 0, sizeof(bundle_id)); if ((ret = dtn_send(handle, regid, &ping_spec, &ping_payload, &bundle_id)) != 0) { fprintf(stderr, "error sending bundle: %d (%s)\n", ret, dtn_strerror(dtn_errno(handle))); exit(1); } creation_times[seqno] = bundle_id.creation_ts; memset(&reply_spec, 0, sizeof(reply_spec)); memset(&reply_payload, 0, sizeof(reply_payload)); // now loop waiting for replies / status reports until it's // time to send again, adding twice the expiration time if we // just sent the last ping timeout = interval * 1000; if (i == count - 1) timeout += expiration * 2000; do { gettimeofday(&recv_start, 0); if ((ret = dtn_recv(handle, &reply_spec, DTN_PAYLOAD_MEM, &reply_payload, timeout)) < 0) { if (dtn_errno(handle) == DTN_ETIMEOUT) { break; // time to send again } fprintf(stderr, "error getting ping reply: %d (%s)\n", ret, dtn_strerror(dtn_errno(handle))); exit(1); } gettimeofday(&recv_end, 0); if (reply_payload.status_report != NULL) { sr_data = reply_payload.status_report; if (sr_data->flags != STATUS_DELETED) { fprintf(stderr, "(bad status report from %s: flags 0x%x)\n", reply_spec.source.uri, sr_data->flags); goto next; } // find the seqno corresponding to the original // transmission time in the status report int j = 0; for (j = 0; j < MAX_PINGS_IN_FLIGHT; ++j) { if (creation_times[j].secs == sr_data->bundle_id.creation_ts.secs && creation_times[j].seqno == sr_data->bundle_id.creation_ts.seqno) { printf("bundle deleted at [%s] (%s): seqno=%d, time=%ld ms\n", reply_spec.source.uri, dtn_status_report_reason_to_str(sr_data->reason), j, TIMEVAL_DIFF_MSEC(recv_end, send_times[j])); goto next; } } printf("bundle deleted at [%s] (%s): ERROR: can't find seqno\n", reply_spec.source.uri, dtn_status_report_reason_to_str(sr_data->reason)); } else { if (reply_payload.buf.buf_len != sizeof(ping_payload_t)) { printf("%d bytes from [%s]: ERROR: length != %zu\n", reply_payload.buf.buf_len, reply_spec.source.uri, sizeof(ping_payload_t)); goto next; } memcpy(&recv_contents, reply_payload.buf.buf_val, sizeof(recv_contents)); if (recv_contents.seqno > MAX_PINGS_IN_FLIGHT) { printf("%d bytes from [%s]: ERROR: invalid seqno %d\n", reply_payload.buf.buf_len, reply_spec.source.uri, recv_contents.seqno); goto next; } if (recv_contents.nonce != nonce) { printf("%d bytes from [%s]: ERROR: invalid nonce %u != %u\n", reply_payload.buf.buf_len, reply_spec.source.uri, recv_contents.nonce, nonce); goto next; } if (recv_contents.time != (u_int32_t)send_times[recv_contents.seqno].tv_sec) { printf("%d bytes from [%s]: ERROR: time mismatch -- " "seqno %u reply time %u != send time %lu\n", reply_payload.buf.buf_len, reply_spec.source.uri, recv_contents.seqno, recv_contents.time, (long unsigned int)send_times[recv_contents.seqno].tv_sec); goto next; } printf("%d bytes from [%s]: '%.*s' seqno=%d, time=%ld ms\n", reply_payload.buf.buf_len, reply_spec.source.uri, (u_int)strlen(PING_STR), reply_payload.buf.buf_val, recv_contents.seqno, TIMEVAL_DIFF_MSEC(recv_end, send_times[recv_contents.seqno])); fflush(stdout); } next: dtn_free_payload(&reply_payload); timeout -= TIMEVAL_DIFF_MSEC(recv_end, recv_start); // once we get status from all the pings we're supposed to // send, we're done reply_count++; if (count != 0 && reply_count == count) { break; } } while (timeout > 0); seqno++; seqno %= MAX_PINGS_IN_FLIGHT; } dtn_close(handle); return 0; }
int main(int argc, const char** argv) { int ret; dtn_handle_t handle; dtn_endpoint_id_t source_eid; dtn_endpoint_id_t replyto_eid; dtn_reg_info_t reginfo; dtn_reg_id_t regid; dtn_bundle_spec_t ping_spec; dtn_bundle_spec_t reply_spec; dtn_bundle_payload_t ping_payload; ping_payload_t payload_contents; ping_payload_t recv_contents; dtn_bundle_payload_t reply_payload; dtn_bundle_status_report_t* sr_data; dtn_bundle_id_t bundle_id; int debug = 1; char demux[64]; int dest_len = 0; struct timeval send_time, now; u_int32_t nonce; int done; time_t clock; struct tm* tm_buf; // force stdout to always be line buffered, even if output is // redirected to a pipe or file setvbuf(stdout, (char *)NULL, _IOLBF, 0); doOptions(argc, argv); memset(&ping_spec, 0, sizeof(ping_spec)); gettimeofday(&now, 0); srand(now.tv_sec); nonce = rand(); // open the ipc handle int err = 0; if (api_IP_set) err = dtn_open_with_IP(api_IP,api_port,&handle); else err = dtn_open(&handle); if (err != DTN_SUCCESS) { fprintf(stderr, "fatal error opening dtn handle: %s\n", dtn_strerror(err)); exit(1); } // make sure they supplied a valid destination eid or // "localhost", in which case we just use the local daemon if (strcmp(dest_eid_str, "localhost") == 0) { dtn_build_local_eid(handle, &ping_spec.dest, "ping"); } else { if (dtn_parse_eid_string(&ping_spec.dest, dest_eid_str)) { fprintf(stderr, "invalid destination eid string '%s'\n", dest_eid_str); exit(1); } } dest_len = strlen(ping_spec.dest.uri); if ((dest_len < 4) || (strcmp(ping_spec.dest.uri + dest_len - 4, "ping") != 0)) { fprintf(stderr, "\nWARNING: ping destination does not end in \"ping\"\n\n"); } // if the user specified a source eid, register on it. // otherwise, build a local eid based on the configuration of // our dtn router plus the demux string snprintf(demux, sizeof(demux), "/traceroute.%d", getpid()); if (source_eid_str[0] != '\0') { if (dtn_parse_eid_string(&source_eid, source_eid_str)) { fprintf(stderr, "invalid source eid string '%s'\n", source_eid_str); exit(1); } } else { dtn_build_local_eid(handle, &source_eid, demux); } if (debug) printf("source_eid [%s]\n", source_eid.uri); dtn_copy_eid(&ping_spec.source, &source_eid); // Make the replyto EID if ( replyto_eid_str[0]!=0 ) { if (dtn_parse_eid_string(&replyto_eid, replyto_eid_str)) { fprintf(stderr, "invalid replyto eid string '%s'\n", replyto_eid_str); exit(1); } dtn_copy_eid(&ping_spec.replyto, &replyto_eid); printf("using user-supplied replyto\n"); } else { dtn_copy_eid(&ping_spec.replyto, &source_eid); printf("using default replyto\n"); } // now create a new registration based on the source // if the replyto EID is unspecified or the same as the // source EID then we'll get status reports, otherwise // they'll go somewhere else. memset(®info, 0, sizeof(reginfo)); dtn_copy_eid(®info.endpoint, &source_eid); reginfo.flags = DTN_REG_DROP; reginfo.regid = DTN_REGID_NONE; reginfo.expiration = 0; if ((ret = dtn_register(handle, ®info, ®id)) != 0) { fprintf(stderr, "error creating registration: %d (%s)\n", ret, dtn_strerror(dtn_errno(handle))); exit(1); } if (debug) printf("dtn_register succeeded, regid %d\n", regid); // set the expiration time and request a bunch of status reports ping_spec.expiration = expiration; ping_spec.dopts = DOPTS_DELETE_RCPT | DOPTS_RECEIVE_RCPT | DOPTS_FORWARD_RCPT | DOPTS_DELIVERY_RCPT; // loop, sending pings and polling for activity gettimeofday(&send_time, NULL); // fill in a short payload string, a nonce, and a sequence number // to verify the echo feature and make sure we're not getting // duplicate responses or ping responses from another app memcpy(&payload_contents.ping, PING_STR, 8); payload_contents.seqno = 0; payload_contents.nonce = nonce; payload_contents.time = send_time.tv_sec; memset(&ping_payload, 0, sizeof(ping_payload)); dtn_set_payload(&ping_payload, DTN_PAYLOAD_MEM, (char*)&payload_contents, sizeof(payload_contents)); memset(&bundle_id, 0, sizeof(bundle_id)); if ((ret = dtn_send(handle, regid, &ping_spec, &ping_payload, &bundle_id)) != 0) { fprintf(stderr, "error sending bundle: %d (%s)\n", ret, dtn_strerror(dtn_errno(handle))); exit(1); } memset(&reply_spec, 0, sizeof(reply_spec)); memset(&reply_payload, 0, sizeof(reply_payload)); clock = time(&clock); tm_buf = gmtime(&clock); printf("%s: sent at %.*s UTC\n", ping_spec.source.uri, 24, asctime(tm_buf)); // now loop waiting for replies / status reports until we're done done = 0; while (1) { int timeout = done ? wait_after_done * 1000 : -1; if ((ret = dtn_recv(handle, &reply_spec, DTN_PAYLOAD_MEM, &reply_payload, timeout)) < 0) { if (done && dtn_errno(handle) == DTN_ETIMEOUT) { break; } fprintf(stderr, "error getting ping reply: %d (%s)\n", ret, dtn_strerror(dtn_errno(handle))); exit(1); } gettimeofday(&now, 0); if (reply_payload.status_report != NULL) { sr_data = reply_payload.status_report; if (sr_data->flags & STATUS_RECEIVED) { clock = sr_data->receipt_ts.secs + DTNTIME_OFFSET; tm_buf = gmtime(&clock); printf("%s: received at %.*s UTC (%ld ms rtt)\n", reply_spec.source.uri, 24, asctime(tm_buf), TIMEVAL_DIFF_MSEC(now, send_time)); } if (sr_data->flags & STATUS_FORWARDED) { clock = sr_data->forwarding_ts.secs + DTNTIME_OFFSET; tm_buf = gmtime(&clock); printf("%s: forwarded at %.*s UTC (%ld ms rtt)\n", reply_spec.source.uri, 24, asctime(tm_buf), TIMEVAL_DIFF_MSEC(now, send_time)); } if (sr_data->flags & STATUS_DELIVERED) { clock = sr_data->delivery_ts.secs + DTNTIME_OFFSET; tm_buf = gmtime(&clock); printf("%s: delivered at %.*s UTC (%ld ms rtt)\n", reply_spec.source.uri, 24, asctime(tm_buf), TIMEVAL_DIFF_MSEC(now, send_time)); } if (sr_data->flags & STATUS_DELETED) { clock = sr_data->deletion_ts.secs + DTNTIME_OFFSET; tm_buf = gmtime(&clock); printf("%s: deleted at %.*s UTC (%s) (%ld ms rtt)\n", reply_spec.source.uri, 24, asctime(tm_buf), dtn_status_report_reason_to_str(sr_data->reason), TIMEVAL_DIFF_MSEC(now, send_time)); break; } } else { if (reply_payload.buf.buf_len != sizeof(ping_payload_t)) { printf("%d bytes from [%s]: ERROR: length != %zu\n", reply_payload.buf.buf_len, reply_spec.source.uri, sizeof(ping_payload_t)); break; } memcpy(&recv_contents, reply_payload.buf.buf_val, sizeof(recv_contents)); if (recv_contents.seqno != 0) { printf("%d bytes from [%s]: ERROR: invalid seqno %d\n", reply_payload.buf.buf_len, reply_spec.source.uri, recv_contents.seqno); break; } if (recv_contents.nonce != nonce) { printf("%d bytes from [%s]: ERROR: invalid nonce %u != %u\n", reply_payload.buf.buf_len, reply_spec.source.uri, recv_contents.nonce, nonce); break; } if ((int)recv_contents.time != (int)send_time.tv_sec) { printf("%d bytes from [%s]: ERROR: time mismatch %u != %lu\n", reply_payload.buf.buf_len, reply_spec.source.uri, recv_contents.time, (long unsigned int)send_time.tv_sec); } clock = reply_spec.creation_ts.secs + DTNTIME_OFFSET; tm_buf = gmtime(&clock); printf("%s: echo reply at %.*s UTC (%ld ms rtt)\n", reply_spec.source.uri, 24, asctime(tm_buf), TIMEVAL_DIFF_MSEC(now, send_time)); done = 1; } dtn_free_payload(&reply_payload); } dtn_close(handle); return 0; }