dbus/dbus-message-factory.c

   1 /* -*- mode: C; c-file-style: "gnu" -*- */
   2 /* dbus-message-factory.c Generator of valid and invalid message data for test suite
   3  *
   4  * Copyright (C) 2005 Red Hat Inc.
   5  *
   6  * Licensed under the Academic Free License version 2.1
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  21  *
  22  */
  23
  24 #ifdef DBUS_BUILD_TESTS
  25
  26 #include "dbus-message-factory.h"
  27 #include "dbus-message-private.h"
  28
  29 typedef dbus_bool_t (* DBusMessageGeneratorFunc) (int           sequence,
  30                                                   DBusMessage **message_p);
  31
  32 static void
  33 set_reply_serial (DBusMessage *message)
  34 {
  35   if (message == NULL)
  36     _dbus_assert_not_reached ("oom");
  37   if (!dbus_message_set_reply_serial (message, 100))
  38     _dbus_assert_not_reached ("oom");
  39 }
  40
  41 static dbus_bool_t
  42 generate_trivial (int           sequence,
  43                   DBusMessage **message_p)
  44 {
  45   DBusMessage *message;
  46
  47   switch (sequence)
  48     {
  49     case 0:
  50       message = dbus_message_new_method_call ("org.freedesktop.TextEditor",
  51                                               "/foo/bar",
  52                                               "org.freedesktop.DocumentFactory",
  53                                               "Create");
  54       break;
  55     case 1:
  56       message = dbus_message_new (DBUS_MESSAGE_TYPE_METHOD_RETURN);
  57       set_reply_serial (message);
  58       break;
  59     case 2:
  60       message = dbus_message_new_signal ("/foo/bar",
  61                                          "org.freedesktop.DocumentFactory",
  62                                          "Created");
  63       break;
  64     case 3:
  65       message = dbus_message_new (DBUS_MESSAGE_TYPE_ERROR);
  66       set_reply_serial (message);
  67       break;
  68     default:
  69       return FALSE;
  70     }
  71
  72   if (message == NULL)
  73     _dbus_assert_not_reached ("oom");
  74
  75   *message_p = message;
  76
  77   return TRUE;
  78 }
  79
  80 static const DBusMessageGeneratorFunc generators[] = {
  81   generate_trivial
  82 };
  83
  84 void
  85 _dbus_message_data_free (DBusMessageData *data)
  86 {
  87   _dbus_string_free (&data->data);
  88 }
  89
  90 void
  91 _dbus_message_data_iter_init (DBusMessageDataIter *iter)
  92 {
  93   iter->generator = 0;
  94   iter->sequence = 0;
  95 }
  96
  97 dbus_bool_t
  98 _dbus_message_data_iter_get_and_next (DBusMessageDataIter *iter,
  99                                       DBusMessageData     *data)
 100 {
 101   DBusMessageGeneratorFunc func;
 102   DBusMessage *message;
 103
 104   if (iter->generator == _DBUS_N_ELEMENTS (generators))
 105     return FALSE;
 106
 107   func = generators[iter->generator];
 108
 109   if ((*func)(iter->sequence, &message))
 110     iter->sequence += 1;
 111   else
 112     {
 113       iter->generator += 1;
 114       iter->sequence = 0;
 115     }
 116
 117   _dbus_assert (message != NULL);
 118
 119   if (!_dbus_string_init (&data->data))
 120     _dbus_assert_not_reached ("oom");
 121
 122   /* move for efficiency, since we'll nuke the message anyway */
 123   if (!_dbus_string_move (&message->header.data, 0,
 124                           &data->data, 0))
 125     _dbus_assert_not_reached ("oom");
 126
 127   if (!_dbus_string_copy (&message->body, 0,
 128                           &data->data, _dbus_string_get_length (&data->data)))
 129     _dbus_assert_not_reached ("oom");
 130
 131   dbus_message_unref (message);
 132
 133   return TRUE;
 134 }
 135
 136 #endif /* DBUS_BUILD_TESTS */