Merge branch 'master' of ssh://gnunet.org/gnunet
[oweals/gnunet.git] / src / util / test_client.c
1 /*
2      This file is part of GNUnet.
3      Copyright (C) 2009, 2016 GNUnet e.V.
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., 51 Franklin Street, Fifth Floor,
18      Boston, MA 02110-1301, USA.
19 */
20 /**
21  * @file util/test_client.c
22  * @brief tests for client.c
23  * @author Christian Grothoff
24  */
25 #include "platform.h"
26 #include "gnunet_util_lib.h"
27
28 static int global_ret;
29
30 static struct GNUNET_MQ_Handle *client_mq;
31
32 #define MY_TYPE 130
33
34
35 /**
36  * Callback that just bounces the message back to the sender.
37  */
38 static void
39 handle_echo (void *cls,
40              const struct GNUNET_MessageHeader *message)
41 {
42   struct GNUNET_SERVICE_Client *c = cls;
43   struct GNUNET_MQ_Handle *mq = GNUNET_SERVICE_client_get_mq (c);
44   struct GNUNET_MQ_Envelope *env;
45
46   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
47               "Receiving message from client, bouncing back\n");
48   env = GNUNET_MQ_msg_copy (message);
49   GNUNET_MQ_send (mq,
50                   env);
51   GNUNET_SERVICE_client_continue (c);
52 }
53
54
55 static void
56 handle_bounce (void *cls,
57                const struct GNUNET_MessageHeader *got)
58 {
59   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
60               "Receiving bounce, checking content\n");
61   GNUNET_assert (NULL != got);
62   global_ret = 2;
63   GNUNET_MQ_destroy (client_mq);
64   client_mq = NULL;
65 }
66
67
68 /**
69  * Generic error handler, called with the appropriate error code and
70  * the same closure specified at the creation of the message queue.
71  * Not every message queue implementation supports an error handler.
72  *
73  * @param cls closure with the `struct GNUNET_STATISTICS_Handle *`
74  * @param error error code
75  */
76 static void
77 mq_error_handler (void *cls,
78                   enum GNUNET_MQ_Error error)
79 {
80   GNUNET_assert (0); /* should never happen */
81 }
82
83
84 static void
85 task (void *cls,
86       const struct GNUNET_CONFIGURATION_Handle *cfg,
87       struct GNUNET_SERVICE_Handle *sh)
88 {
89   struct GNUNET_MQ_MessageHandler chandlers[] = {
90     GNUNET_MQ_hd_fixed_size (bounce,
91                              MY_TYPE,
92                              struct GNUNET_MessageHeader,
93                              cls),
94     GNUNET_MQ_handler_end ()
95   };
96   struct GNUNET_MQ_Envelope *env;
97   struct GNUNET_MessageHeader *msg;
98
99   /* test that ill-configured client fails instantly */
100   GNUNET_assert (NULL ==
101                  GNUNET_CLIENT_connect (cfg,
102                                         "invalid-service",
103                                         NULL,
104                                         &mq_error_handler,
105                                         NULL));
106   client_mq = GNUNET_CLIENT_connect (cfg,
107                                      "test_client",
108                                      chandlers,
109                                      &mq_error_handler,
110                                      NULL);
111   GNUNET_assert (NULL != client_mq);
112   env = GNUNET_MQ_msg (msg,
113                        MY_TYPE);
114   GNUNET_MQ_send (client_mq,
115                   env);
116 }
117
118
119 /**
120  * Function called when the client connects to the service.
121  *
122  * @param cls the name of the service
123  * @param c connecting client
124  * @param mq message queue to talk to the client
125  * @return @a c
126  */
127 static void *
128 connect_cb (void *cls,
129             struct GNUNET_SERVICE_Client *c,
130             struct GNUNET_MQ_Handle *mq)
131 {
132   return c;
133 }
134
135
136 /**
137  * Function called when the client disconnects.
138  *
139  * @param cls our service name
140  * @param c disconnecting client
141  * @param internal_cls must match @a c
142  */ 
143 static void
144 disconnect_cb (void *cls,
145                struct GNUNET_SERVICE_Client *c,
146                void *internal_cls)
147 {
148   if (2 == global_ret)
149   {
150     GNUNET_SCHEDULER_shutdown ();
151     global_ret = 0;
152   }
153 }
154
155
156 int
157 main (int argc,
158       char *argv[])
159 {
160   struct GNUNET_MQ_MessageHandler shandlers[] = {
161     GNUNET_MQ_hd_fixed_size (echo,
162                              MY_TYPE,
163                              struct GNUNET_MessageHeader,
164                              NULL),
165     GNUNET_MQ_handler_end ()
166   };
167   char * test_argv[] = {
168     (char *) "test_client",
169     "-c",
170     "test_client_data.conf",
171     NULL
172   };
173
174   GNUNET_log_setup ("test_client",
175                     "WARNING",
176                     NULL);
177   if (0 != strstr (argv[0],
178                    "unix"))
179     test_argv[2] = "test_client_unix.conf";
180   global_ret = 1;
181   if (0 !=
182       GNUNET_SERVICE_run_ (3,
183                            test_argv,
184                            "test_client",
185                            GNUNET_SERVICE_OPTION_NONE,
186                            &task,
187                            &connect_cb,
188                            &disconnect_cb,
189                            NULL,
190                            shandlers))
191     global_ret = 3;
192   return global_ret;
193 }
194
195 /* end of test_client.c */