struct PeerContext * p1;
struct PeerContext * p2;
GNUNET_SCHEDULER_TaskIdentifier tct;
+ GNUNET_TRANSPORT_TESTING_connect_cb cb;
+ void * cb_cls;
+
+ struct GNUNET_TRANSPORT_Handle *th_p1;
+ struct GNUNET_TRANSPORT_Handle *th_p2;
+ int c;
};
+static void
+exchange_hello_last (void *cb_cls,
+ const struct GNUNET_MessageHeader *message);
+static void
+exchange_hello (void *cb_cls,
+ const struct GNUNET_MessageHeader *message);
static void
-exchange_hello_last (void *cls,
- const struct GNUNET_MessageHeader *message)
+notify_connect_internal (void *cls,
+ const struct GNUNET_PeerIdentity *peer,
+ const struct GNUNET_TRANSPORT_ATS_Information *ats,
+ uint32_t ats_count)
{
struct ConnectingContext * cc = cls;
+// /void * cb_cls = cc->cb_cls;
+
+ GNUNET_assert(cc != NULL);
+ cc->c++;
+
+ if (cc->c == 2)
+ {
+ /* clean up */
+ GNUNET_TRANSPORT_get_hello_cancel (cc->th_p2, &exchange_hello_last, cc);
+ GNUNET_TRANSPORT_get_hello_cancel (cc->th_p1, &exchange_hello, cc);
+
+ if (cc->tct != GNUNET_SCHEDULER_NO_TASK)
+ GNUNET_SCHEDULER_cancel(cc->tct);
+
+ cc->tct = GNUNET_SCHEDULER_NO_TASK;
+
+ GNUNET_TRANSPORT_disconnect( cc->th_p1);
+ GNUNET_TRANSPORT_disconnect( cc->th_p2);
+
+ if (cc->cb != NULL)
+ cc->cb (cc->p1, cc->p2, cc->cb_cls);
+
+ GNUNET_free(cc);
+ }
+
+}
+
+static void
+notify_connect (void *cls,
+ const struct GNUNET_PeerIdentity *peer,
+ const struct GNUNET_TRANSPORT_ATS_Information *ats,
+ uint32_t ats_count)
+{
+ struct PeerContext * p = cls;
+ if (p == NULL)
+ return;
+ if (p->nc != NULL)
+ p->nc (p->cb_cls, peer, ats, ats_count);
+}
+
+static void
+notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
+{
+ struct PeerContext * p = cls;
+ if (p == NULL)
+ return;
+ if (p->nd != NULL)
+ p->nd (p->cb_cls, peer);
+}
+
+static void
+notify_receive (void *cls,
+ const struct GNUNET_PeerIdentity *peer,
+ const struct GNUNET_MessageHeader *message,
+ const struct GNUNET_TRANSPORT_ATS_Information *ats,
+ uint32_t ats_count)
+{
+ struct PeerContext * p = cls;
+ if (p == NULL)
+ return;
+ if (p->rec != NULL)
+ p->rec (p->cb_cls, peer, message, ats, ats_count);
+}
+
+
+static void
+exchange_hello_last (void *cb_cls,
+ const struct GNUNET_MessageHeader *message)
+{
+ struct ConnectingContext * cc = cb_cls;
struct PeerContext *me = cc->p2;
- struct PeerContext *p1 = cc->p1;
+ //struct PeerContext *p1 = cc->p1;
GNUNET_assert (message != NULL);
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
GNUNET_assert (GNUNET_OK ==
GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *)
message, &me->id));
- GNUNET_TRANSPORT_offer_hello (p1->th, message, NULL, NULL);
+ GNUNET_TRANSPORT_offer_hello (cc->th_p1, message, NULL, NULL);
}
static void
-exchange_hello (void *cls,
+exchange_hello (void *cb_cls,
const struct GNUNET_MessageHeader *message)
{
- struct ConnectingContext * cc = cls;
+ struct ConnectingContext * cc = cb_cls;
struct PeerContext *me = cc->p1;
- struct PeerContext *p2 = cc->p2;
+ //struct PeerContext *p2 = cc->p2;
GNUNET_assert (message != NULL);
GNUNET_assert (GNUNET_OK ==
"Exchanging HELLO of size %d from peer %s!\n",
(int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message),
GNUNET_i2s (&me->id));
- GNUNET_TRANSPORT_offer_hello (p2->th, message, NULL, NULL);
+ GNUNET_TRANSPORT_offer_hello (cc->th_p2, message, NULL, NULL);
}
static void
struct PeerContext *p1 = cc->p1;
struct PeerContext *p2 = cc->p2;
+ cc->tct = GNUNET_SCHEDULER_NO_TASK;
+ if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
+ return;
+
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
"Asking peers to connect...\n");
/* FIXME: 'pX.id' may still be all-zeros here... */
- GNUNET_TRANSPORT_try_connect (p2->th,
- &p1->id);
- GNUNET_TRANSPORT_try_connect (p1->th,
+ GNUNET_TRANSPORT_try_connect (cc->th_p1,
&p2->id);
+ GNUNET_TRANSPORT_try_connect (cc->th_p2,
+ &p1->id);
+
cc->tct = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
&try_connect,
cc);
}
-static struct PeerContext *
-GNUNET_TRANSPORT_TESTING_start_peer (const char * cfgname)
+
+/**
+ * Start a peer with the given configuration
+ * @param rec receive callback
+ * @param nc connect callback
+ * @param nd disconnect callback
+ * @param cb_cls closure for callback
+ * @return the peer context
+ */
+struct PeerContext *
+GNUNET_TRANSPORT_TESTING_start_peer (const char * cfgname,
+ GNUNET_TRANSPORT_ReceiveCallback rec,
+ GNUNET_TRANSPORT_NotifyConnect nc,
+ GNUNET_TRANSPORT_NotifyDisconnect nd,
+ void * cb_cls)
{
struct PeerContext * p = GNUNET_malloc (sizeof (struct PeerContext));
GNUNET_DISK_directory_remove (p->servicehome);
p->arm_proc = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-arm",
"gnunet-service-arm",
- "-c", cfgname, NULL);
+ "-c", cfgname,
+#if VERBOSE_PEERS
+ "-L", "DEBUG",
+#else
+ "-L", "ERROR",
+#endif
+ NULL);
+ p->nc = nc;
+ p->nd = nd;
+ p->rec = rec;
+ p->cb_cls = cb_cls;
+
+ p->th = GNUNET_TRANSPORT_connect(p->cfg, NULL,
+ p,
+ ¬ify_receive,
+ ¬ify_connect,
+ ¬ify_disconnect);
+ GNUNET_assert (p->th != NULL);
return p;
}
-static void
+/**
+ * shutdown the given peer
+ * @param p the peer
+ */
+void
GNUNET_TRANSPORT_TESTING_stop_peer (struct PeerContext * p)
{
+ if (p->th != NULL)
+ GNUNET_TRANSPORT_disconnect(p->th);
+
if (NULL != p->arm_proc)
{
if (0 != GNUNET_OS_process_kill (p->arm_proc, SIGTERM))
GNUNET_DISK_directory_remove (p->servicehome);
GNUNET_free(p->servicehome);
}
+ GNUNET_free (p);
}
-static void
+/**
+ * Connect the two given peers and call the callback when both peers report the
+ * inbound connect. Remarks: start_peer's notify_connect callback can be called
+ * before.
+ * @param p1 peer 1
+ * @param p2 peer 2
+ * @param cb the callback to call
+ * @param cb_cls callback cls
+ */
+void
GNUNET_TRANSPORT_TESTING_connect_peers (struct PeerContext * p1,
struct PeerContext * p2,
- GNUNET_TRANSPORT_TESTING_connect_cb * cb,
- void * cls)
+ GNUNET_TRANSPORT_TESTING_connect_cb cb,
+ void * cb_cls)
{
struct ConnectingContext * cc = GNUNET_malloc (sizeof (struct ConnectingContext));
GNUNET_assert (p1 != NULL);
- GNUNET_assert (p1->th != NULL);
-
GNUNET_assert (p2 != NULL);
- GNUNET_assert (p2->th != NULL);
cc->p1 = p1;
cc->p2 = p2;
- GNUNET_TRANSPORT_get_hello (p1->th, &exchange_hello, cc);
- GNUNET_TRANSPORT_get_hello (p2->th, &exchange_hello_last, cc);
+
+ cc->cb = cb;
+ cc->cb_cls = cb_cls;
+
+ cc->th_p1 = GNUNET_TRANSPORT_connect(cc->p1->cfg, NULL,
+ cc,
+ NULL,
+ ¬ify_connect_internal,
+ NULL);
+
+ cc->th_p2 = GNUNET_TRANSPORT_connect(cc->p2->cfg, NULL,
+ cc,
+ NULL,
+ ¬ify_connect_internal,
+ NULL);
+
+ GNUNET_assert (cc->th_p1 != NULL);
+ GNUNET_assert (cc->th_p2 != NULL);
+
+ GNUNET_TRANSPORT_get_hello (cc->th_p1, &exchange_hello, cc);
+ GNUNET_TRANSPORT_get_hello (cc->th_p2, &exchange_hello_last, cc);
cc->tct = GNUNET_SCHEDULER_add_now (&try_connect, cc);
}