use enum GNUNET_ATS_Network_Type instead of uint32_t where appropriate
[oweals/gnunet.git] / src / transport / test_transport_testing_restart.c
1 /*
2      This file is part of GNUnet.
3      (C) 2009, 2010 Christian Grothoff (and other contributing authors)
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20 /**
21  * @file transport/test_transport_testing_restart.c
22  * @brief test case for transport testing library:
23  * start the peer, get the HELLO message, restart and stop the peer
24  *
25  */
26 #include "platform.h"
27 #include "gnunet_transport_service.h"
28 #include "transport-testing.h"
29
30 /**
31  * How long until we give up on transmitting the message?
32  */
33 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
34
35 GNUNET_SCHEDULER_TaskIdentifier timeout_task;
36
37 static struct PeerContext *p;
38
39 struct GNUNET_TRANSPORT_TESTING_handle *tth;
40
41 static int ret = 0;
42
43 static void
44 end ()
45 {
46   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopping peers\n");
47
48   if (timeout_task != GNUNET_SCHEDULER_NO_TASK)
49     GNUNET_SCHEDULER_cancel (timeout_task);
50
51   GNUNET_TRANSPORT_TESTING_stop_peer (tth, p);
52   GNUNET_TRANSPORT_TESTING_done (tth);
53 }
54
55 static void
56 end_badly ()
57 {
58   timeout_task = GNUNET_SCHEDULER_NO_TASK;
59   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Fail! Stopping peers\n");
60
61   if (NULL != p)
62     GNUNET_TRANSPORT_TESTING_stop_peer (tth, p);
63
64   if (NULL != tth)
65     GNUNET_TRANSPORT_TESTING_done (tth);
66
67   ret = GNUNET_SYSERR;
68 }
69
70 static void
71 restart_cb (struct PeerContext *p, void *cls)
72 {
73   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer %u (`%s') successfully restarted\n",
74               p->no,
75               GNUNET_i2s (&p->id));
76
77   ret = 0;
78   GNUNET_SCHEDULER_add_now (&end, NULL);
79 }
80
81
82 static void
83 restart_task ()
84 {
85   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer %u (`%s') restarting, \n",
86               p->no,
87               GNUNET_i2s (&p->id));
88   GNUNET_TRANSPORT_TESTING_restart_peer (tth, p, NULL, restart_cb, p);
89 }
90
91 static void
92 start_cb (struct PeerContext *p, void *cls)
93 {
94   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer %u (`%s') successfully started\n",
95               p->no,
96               GNUNET_i2s (&p->id));
97
98   GNUNET_SCHEDULER_add_now (&restart_task, NULL);
99 }
100
101
102 static void
103 run (void *cls, char *const *args, const char *cfgfile,
104      const struct GNUNET_CONFIGURATION_Handle *cfg)
105 {
106   ret = 1;
107   tth = GNUNET_TRANSPORT_TESTING_init ();
108   GNUNET_assert (NULL != tth);
109
110   timeout_task =
111       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES, &end_badly, NULL);
112
113   p = GNUNET_TRANSPORT_TESTING_start_peer(tth, cfgfile, 1,
114                                           NULL, /* receive cb */
115                                           NULL, /* connect cb */
116                                           NULL, /* disconnect cb */
117                                           start_cb, /* startup cb */
118                                           NULL); /* closure */
119   if (NULL == p)
120   {
121     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Failed to start peer\n");
122     if (timeout_task != GNUNET_SCHEDULER_NO_TASK)
123       GNUNET_SCHEDULER_cancel (timeout_task);
124     timeout_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL);
125   }
126 }
127
128 int
129 main (int argc, char *argv[])
130 {
131   GNUNET_log_setup ("test_transport_testing_restart",
132                     "WARNING",
133                     NULL);
134
135   char *const argv_1[] = { "test_transport_testing_restart",
136     "-c",
137     "test_transport_api_data.conf",
138     NULL
139   };
140
141   struct GNUNET_GETOPT_CommandLineOption options[] = {
142     GNUNET_GETOPT_OPTION_END
143   };
144
145   GNUNET_PROGRAM_run ((sizeof (argv_1) / sizeof (char *)) - 1, argv_1,
146                       "test_transport_testing_restart", "nohelp", options, &run, &ret);
147
148   return ret;
149 }
150
151 /* end of test_transport_testing_restart.c */