1 /* -*- mode: C; c-file-style: "gnu" -*- */
2 /* dbus-message.c DBusMessage object
4 * Copyright (C) 2002, 2003 Red Hat Inc.
5 * Copyright (C) 2002, 2003 CodeFactory AB
7 * Licensed under the Academic Free License version 1.2
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25 #include "dbus-internals.h"
26 #include "dbus-marshal.h"
27 #include "dbus-message.h"
28 #include "dbus-message-internal.h"
29 #include "dbus-memory.h"
30 #include "dbus-list.h"
31 #include "dbus-message-builder.h"
35 * @defgroup DBusMessageInternals DBusMessage implementation details
36 * @ingroup DBusInternals
37 * @brief DBusMessage private implementation details.
39 * The guts of DBusMessage and its methods.
57 static dbus_bool_t field_is_named[FIELD_LAST] =
59 FALSE, /* FIELD_HEADER_LENGTH */
60 FALSE, /* FIELD_BODY_LENGTH */
61 FALSE, /* FIELD_CLIENT_SERIAL */
62 TRUE, /* FIELD_NAME */
63 TRUE, /* FIELD_SERVICE */
64 TRUE, /* FIELD_SENDER */
65 TRUE /* FIELD_REPLY_SERIAL */
70 int offset; /**< Offset to start of field (location of name of field
76 * @brief Internals of DBusMessage
78 * Object representing a message received from or to be sent to
79 * another application. This is an opaque object, all members
84 dbus_atomic_t refcount; /**< Reference count */
86 DBusString header; /**< Header network data, stored
87 * separately from body so we can
88 * independently realloc it.
91 HeaderField header_fields[FIELD_LAST]; /**< Track the location
92 * of each field in "header"
94 int header_padding; /**< bytes of alignment in header */
96 DBusString body; /**< Body network data. */
98 char byte_order; /**< Message byte order. */
100 DBusCounter *size_counter; /**< Counter for the size of the message, or #NULL */
101 long size_counter_delta; /**< Size we incremented the size counter by. */
103 unsigned int locked : 1; /**< Message being sent, no modifications allowed. */
107 * @brief Internals of DBusMessageIter
109 * Object representing a position in a message. All fields are internal.
111 struct DBusMessageIter
113 int refcount; /**< Reference count */
115 int pos; /**< Current position in the string */
117 DBusMessage *message; /**< Message used */
121 * Gets the data to be sent over the network for this message.
122 * The header and then the body should be written out.
123 * This function is guaranteed to always return the same
124 * data once a message is locked (with _dbus_message_lock()).
126 * @param message the message.
127 * @param header return location for message header data.
128 * @param body return location for message body data.
131 _dbus_message_get_network_data (DBusMessage *message,
132 const DBusString **header,
133 const DBusString **body)
135 _dbus_assert (message->locked);
137 *header = &message->header;
138 *body = &message->body;
142 clear_header_padding (DBusMessage *message)
144 _dbus_string_shorten (&message->header,
145 message->header_padding);
146 message->header_padding = 0;
150 append_header_padding (DBusMessage *message)
153 old_len = _dbus_string_get_length (&message->header);
154 if (!_dbus_string_align_length (&message->header, 8))
157 message->header_padding = _dbus_string_get_length (&message->header) - old_len;
163 adjust_field_offsets (DBusMessage *message,
173 while (i < FIELD_LAST)
175 if (message->header_fields[i].offset > offsets_after)
176 message->header_fields[i].offset += delta;
183 get_string_field (DBusMessage *message,
187 int offset = message->header_fields[field].offset;
193 /* offset points to string length, string data follows it */
194 /* FIXME _dbus_demarshal_const_string() that returned
195 * a reference to the string plus its len might be nice.
199 *len = _dbus_demarshal_uint32 (&message->header,
204 _dbus_string_get_const_data (&message->header,
207 return data + (offset + 4);
211 get_int_field (DBusMessage *message,
214 int offset = message->header_fields[field].offset;
217 return -1; /* useless if -1 is a valid value of course */
219 return _dbus_demarshal_int32 (&message->header,
226 append_int_field (DBusMessage *message,
233 _dbus_assert (!message->locked);
235 clear_header_padding (message);
237 orig_len = _dbus_string_get_length (&message->header);
239 if (!_dbus_string_align_length (&message->header, 4))
242 if (!_dbus_string_append_len (&message->header, name, 4))
245 if (!_dbus_string_append_byte (&message->header, DBUS_TYPE_INT32))
248 if (!_dbus_string_align_length (&message->header, 4))
251 message->header_fields[FIELD_REPLY_SERIAL].offset =
252 _dbus_string_get_length (&message->header);
254 if (!_dbus_marshal_int32 (&message->header, message->byte_order,
258 if (!append_header_padding (message))
264 message->header_fields[field].offset = -1;
265 _dbus_string_set_length (&message->header, orig_len);
267 /* this must succeed because it was allocated on function entry and
268 * DBusString doesn't ever realloc smaller
270 if (!append_header_padding (message))
271 _dbus_assert_not_reached ("failed to reappend header padding");
276 append_string_field (DBusMessage *message,
283 _dbus_assert (!message->locked);
285 clear_header_padding (message);
287 orig_len = _dbus_string_get_length (&message->header);
289 if (!_dbus_string_align_length (&message->header, 4))
292 if (!_dbus_string_append_len (&message->header, name, 4))
295 if (!_dbus_string_append_byte (&message->header, DBUS_TYPE_STRING))
298 if (!_dbus_string_align_length (&message->header, 4))
301 message->header_fields[field].offset =
302 _dbus_string_get_length (&message->header);
304 if (!_dbus_marshal_string (&message->header, message->byte_order,
308 if (!append_header_padding (message))
314 message->header_fields[field].offset = -1;
315 _dbus_string_set_length (&message->header, orig_len);
317 /* this must succeed because it was allocated on function entry and
318 * DBusString doesn't ever realloc smaller
320 if (!append_header_padding (message))
321 _dbus_assert_not_reached ("failed to reappend header padding");
327 delete_int_field (DBusMessage *message,
330 int offset = message->header_fields[field].offset;
332 _dbus_assert (!message->locked);
333 _dbus_assert (field_is_named[field]);
338 clear_header_padding (message);
340 /* The field typecode and name take up 8 bytes */
341 _dbus_string_delete (&message->header,
345 message->header_fields[field].offset = -1;
347 adjust_field_offsets (message,
351 append_header_padding (message);
355 delete_string_field (DBusMessage *message,
358 int offset = message->header_fields[field].offset;
362 _dbus_assert (!message->locked);
363 _dbus_assert (field_is_named[field]);
368 clear_header_padding (message);
370 get_string_field (message, field, &len);
372 /* The field typecode and name take up 8 bytes, and the nul
373 * termination is 1 bytes, string length integer is 4 bytes
375 delete_len = 8 + 4 + 1 + len;
377 _dbus_string_delete (&message->header,
381 message->header_fields[field].offset = -1;
383 adjust_field_offsets (message,
387 append_header_padding (message);
391 set_int_field (DBusMessage *message,
395 int offset = message->header_fields[field].offset;
397 _dbus_assert (!message->locked);
401 /* need to append the field */
405 case FIELD_REPLY_SERIAL:
406 return append_int_field (message, field,
407 DBUS_HEADER_FIELD_REPLY,
410 _dbus_assert_not_reached ("appending an int field we don't support appending");
416 _dbus_marshal_set_int32 (&message->header,
425 set_uint_field (DBusMessage *message,
429 int offset = message->header_fields[field].offset;
431 _dbus_assert (!message->locked);
435 /* need to append the field */
440 _dbus_assert_not_reached ("appending a uint field we don't support appending");
446 _dbus_marshal_set_uint32 (&message->header,
455 set_string_field (DBusMessage *message,
459 int offset = message->header_fields[field].offset;
461 _dbus_assert (!message->locked);
462 _dbus_assert (value != NULL);
466 /* need to append the field */
471 return append_string_field (message, field,
472 DBUS_HEADER_FIELD_SENDER,
475 _dbus_assert_not_reached ("appending a string field we don't support appending");
485 old_len = _dbus_string_get_length (&message->header);
487 _dbus_string_init_const_len (&v, value,
488 strlen (value) + 1); /* include nul */
489 if (!_dbus_marshal_set_string (&message->header,
494 new_len = _dbus_string_get_length (&message->header);
496 adjust_field_offsets (message,
505 * Sets the client serial of a message.
506 * This can only be done once on a message.
508 * @todo client_serial should be called simply
509 * "serial"; it's in outgoing messages for both
510 * the client and the server, it's only client-specific
511 * in the message bus case. It's more like origin_serial
514 * @param message the message
515 * @param client_serial the client serial
518 _dbus_message_set_client_serial (DBusMessage *message,
519 dbus_int32_t client_serial)
521 _dbus_assert (!message->locked);
522 _dbus_assert (_dbus_message_get_client_serial (message) < 0);
524 set_int_field (message, FIELD_CLIENT_SERIAL,
529 * Sets the reply serial of a message (the client serial
530 * of the message this is a reply to).
532 * @param message the message
533 * @param reply_serial the client serial
534 * @returns #FALSE if not enough memory
537 _dbus_message_set_reply_serial (DBusMessage *message,
538 dbus_int32_t reply_serial)
540 _dbus_assert (!message->locked);
542 return set_int_field (message, FIELD_REPLY_SERIAL,
547 * Returns the client serial of a message or
548 * -1 if none has been specified.
550 * @todo see note in _dbus_message_set_client_serial()
551 * about how client_serial is a misnomer
553 * @todo this function should be public, after renaming it.
555 * @param message the message
556 * @returns the client serial
559 _dbus_message_get_client_serial (DBusMessage *message)
561 return get_int_field (message, FIELD_CLIENT_SERIAL);
565 * Returns the serial that the message is
566 * a reply to or -1 if none.
568 * @param message the message
569 * @returns the reply serial
572 _dbus_message_get_reply_serial (DBusMessage *message)
574 return get_int_field (message, FIELD_REPLY_SERIAL);
578 * Adds a counter to be incremented immediately with the
579 * size of this message, and decremented by the size
580 * of this message when this message if finalized.
582 * @param message the message
583 * @param counter the counter
586 _dbus_message_add_size_counter (DBusMessage *message,
587 DBusCounter *counter)
589 _dbus_assert (message->size_counter == NULL); /* If this fails we may need to keep a list of
590 * counters instead of just one
593 message->size_counter = counter;
594 _dbus_counter_ref (message->size_counter);
596 /* When we can change message size, we may want to
597 * update this each time we do so, or we may want to
598 * just KISS like this.
600 message->size_counter_delta =
601 _dbus_string_get_length (&message->header) +
602 _dbus_string_get_length (&message->body);
604 _dbus_verbose ("message has size %ld\n",
605 message->size_counter_delta);
607 _dbus_counter_adjust (message->size_counter, message->size_counter_delta);
611 dbus_message_create_header (DBusMessage *message,
617 if (!_dbus_string_append_byte (&message->header, message->byte_order))
621 if (!_dbus_string_append_byte (&message->header, flags))
624 if (!_dbus_string_append_byte (&message->header, DBUS_MAJOR_PROTOCOL_VERSION))
627 if (!_dbus_string_append_byte (&message->header, 0))
630 message->header_fields[FIELD_HEADER_LENGTH].offset = 4;
631 if (!_dbus_marshal_uint32 (&message->header, message->byte_order, 0))
634 message->header_fields[FIELD_BODY_LENGTH].offset = 8;
635 if (!_dbus_marshal_uint32 (&message->header, message->byte_order, 0))
638 message->header_fields[FIELD_CLIENT_SERIAL].offset = 12;
639 if (!_dbus_marshal_int32 (&message->header, message->byte_order, -1))
642 /* Marshal message service */
645 if (!append_string_field (message,
647 DBUS_HEADER_FIELD_SERVICE,
652 _dbus_assert (name != NULL);
653 if (!append_string_field (message,
655 DBUS_HEADER_FIELD_NAME,
663 * Locks a message. Allows checking that applications don't keep a
664 * reference to a message in the outgoing queue and change it
665 * underneath us. Messages are locked when they enter the outgoing
666 * queue (dbus_connection_send_message()), and the library complains
667 * if the message is modified while locked.
669 * @param message the message to lock.
672 _dbus_message_lock (DBusMessage *message)
674 if (!message->locked)
676 /* Fill in our lengths */
677 set_uint_field (message,
679 _dbus_string_get_length (&message->header));
681 set_uint_field (message,
683 _dbus_string_get_length (&message->body));
685 message->locked = TRUE;
692 * @defgroup DBusMessage DBusMessage
694 * @brief Message to be sent or received over a DBusConnection.
696 * A DBusMessage is the most basic unit of communication over a
697 * DBusConnection. A DBusConnection represents a stream of messages
698 * received from a remote application, and a stream of messages
699 * sent to a remote application.
705 * @typedef DBusMessage
707 * Opaque data type representing a message received from or to be
708 * sent to another application.
712 dbus_message_new_empty_header (void)
714 DBusMessage *message;
717 message = dbus_new0 (DBusMessage, 1);
721 message->refcount = 1;
722 message->byte_order = DBUS_COMPILER_BYTE_ORDER;
725 while (i < FIELD_LAST)
727 message->header_fields[i].offset = -1;
731 if (!_dbus_string_init (&message->header, _DBUS_INT_MAX))
737 if (!_dbus_string_init (&message->body, _DBUS_INT_MAX))
739 _dbus_string_free (&message->header);
749 * Constructs a new message. Returns #NULL if memory can't be
750 * allocated for the message. The service may be #NULL in which case
751 * no service is set; this is appropriate when using D-BUS in a
752 * peer-to-peer context (no message bus).
754 * @todo reverse the arguments, first 'name' then 'service'
755 * as 'name' is more fundamental
757 * @param service service that the message should be sent to or #NULL
758 * @param name name of the message
759 * @returns a new DBusMessage, free with dbus_message_unref()
760 * @see dbus_message_unref()
763 dbus_message_new (const char *service,
766 DBusMessage *message;
768 message = dbus_message_new_empty_header ();
772 if (!dbus_message_create_header (message, service, name))
774 dbus_message_unref (message);
782 * Constructs a message that is a reply to some other
783 * message. Returns #NULL if memory can't be allocated
786 * @param original_message the message which the created
787 * message is a reply to.
788 * @returns a new DBusMessage, free with dbus_message_unref()
789 * @see dbus_message_new(), dbus_message_unref()
792 dbus_message_new_reply (DBusMessage *original_message)
794 DBusMessage *message;
795 const char *sender, *name;
797 sender = get_string_field (original_message,
799 name = get_string_field (original_message,
802 _dbus_assert (sender != NULL);
804 message = dbus_message_new (sender, name);
809 if (!_dbus_message_set_reply_serial (message,
810 _dbus_message_get_client_serial (original_message)))
812 dbus_message_unref (message);
820 * Creates a new message that is an error reply to a certain message.
822 * @param original_message the original message
823 * @param error_name the error name
824 * @param error_message the error message string
825 * @returns a new error message
828 dbus_message_new_error_reply (DBusMessage *original_message,
829 const char *error_name,
830 const char *error_message)
832 DBusMessage *message;
835 sender = get_string_field (original_message,
838 _dbus_assert (sender != NULL);
840 message = dbus_message_new (sender, error_name);
845 if (!_dbus_message_set_reply_serial (message,
846 _dbus_message_get_client_serial (original_message)))
848 dbus_message_unref (message);
852 if (!dbus_message_append_string (message, error_message))
854 dbus_message_unref (message);
858 dbus_message_set_is_error (message, TRUE);
864 * Creates a new message that is an exact replica of the message
865 * specified, except that its refcount is set to 1.
867 * @param message the message.
868 * @returns the new message.
871 dbus_message_new_from_message (const DBusMessage *message)
876 retval = dbus_new0 (DBusMessage, 1);
880 retval->refcount = 1;
881 retval->byte_order = message->byte_order;
883 if (!_dbus_string_init (&retval->header, _DBUS_INT_MAX))
889 if (!_dbus_string_init (&retval->body, _DBUS_INT_MAX))
891 _dbus_string_free (&retval->header);
896 if (!_dbus_string_copy (&message->header, 0,
899 _dbus_string_free (&retval->header);
900 _dbus_string_free (&retval->body);
906 if (!_dbus_string_copy (&message->body, 0,
909 _dbus_string_free (&retval->header);
910 _dbus_string_free (&retval->body);
916 for (i = 0; i < FIELD_LAST; i++)
918 retval->header_fields[i].offset = message->header_fields[i].offset;
926 * Increments the reference count of a DBusMessage.
928 * @param message The message
929 * @see dbus_message_unref
932 dbus_message_ref (DBusMessage *message)
934 dbus_atomic_t refcount;
936 refcount = _dbus_atomic_inc (&message->refcount);
937 _dbus_assert (refcount > 1);
941 * Decrements the reference count of a DBusMessage.
943 * @param message The message
944 * @see dbus_message_ref
947 dbus_message_unref (DBusMessage *message)
949 dbus_atomic_t refcount;
951 refcount = _dbus_atomic_dec (&message->refcount);
953 _dbus_assert (refcount >= 0);
957 if (message->size_counter != NULL)
959 _dbus_counter_adjust (message->size_counter,
960 - message->size_counter_delta);
961 _dbus_counter_unref (message->size_counter);
964 _dbus_string_free (&message->header);
965 _dbus_string_free (&message->body);
972 * Gets the name of a message.
974 * @param message the message
975 * @returns the message name (should not be freed)
978 dbus_message_get_name (DBusMessage *message)
980 return get_string_field (message, FIELD_NAME, NULL);
984 * Gets the destination service of a message.
986 * @param message the message
987 * @returns the message destination service (should not be freed)
990 dbus_message_get_service (DBusMessage *message)
992 return get_string_field (message, FIELD_SERVICE, NULL);
996 * Appends fields to a message given a variable argument
997 * list. The variable argument list should contain the type
998 * of the argument followed by the value to add. Array values
999 * are specified by a pointer to the array followed by an int
1000 * giving the length of the array. The list is terminated
1003 * @param message the message
1004 * @param first_argument_type type of the first argument
1005 * @param ... value of first argument, list of additional type-value pairs
1006 * @returns #TRUE on success
1009 dbus_message_append_args (DBusMessage *message,
1016 va_start (var_args, first_arg_type);
1017 retval = dbus_message_append_args_valist (message,
1026 * This function takes a va_list for use by language bindings
1028 * @see dbus_message_append_args.
1029 * @param message the message
1030 * @param first_arg_type type of first argument
1031 * @param var_args value of first argument, then list of type/value pairs
1032 * @returns #TRUE on success
1035 dbus_message_append_args_valist (DBusMessage *message,
1041 old_len = _dbus_string_get_length (&message->body);
1043 type = first_arg_type;
1049 case DBUS_TYPE_INT32:
1050 if (!dbus_message_append_int32 (message, va_arg (var_args, dbus_int32_t)))
1053 case DBUS_TYPE_UINT32:
1054 if (!dbus_message_append_uint32 (message, va_arg (var_args, dbus_uint32_t)))
1057 case DBUS_TYPE_DOUBLE:
1058 if (!dbus_message_append_double (message, va_arg (var_args, double)))
1061 case DBUS_TYPE_STRING:
1062 if (!dbus_message_append_string (message, va_arg (var_args, const char *)))
1065 case DBUS_TYPE_BYTE_ARRAY:
1068 unsigned char *data;
1070 data = va_arg (var_args, unsigned char *);
1071 len = va_arg (var_args, int);
1073 if (!dbus_message_append_byte_array (message, data, len))
1077 case DBUS_TYPE_STRING_ARRAY:
1082 data = va_arg (var_args, const char **);
1083 len = va_arg (var_args, int);
1085 if (!dbus_message_append_string_array (message, data, len))
1091 _dbus_warn ("Unknown field type %d\n", type);
1094 type = va_arg (var_args, int);
1100 _dbus_string_set_length (&message->body, old_len);
1105 * Appends a 32 bit signed integer to the message.
1107 * @param message the message
1108 * @param value the integer value
1109 * @returns #TRUE on success
1112 dbus_message_append_int32 (DBusMessage *message,
1115 _dbus_assert (!message->locked);
1117 if (!_dbus_string_append_byte (&message->body, DBUS_TYPE_INT32))
1119 _dbus_string_shorten (&message->body, 1);
1123 return _dbus_marshal_int32 (&message->body,
1124 message->byte_order, value);
1128 * Appends a 32 bit unsigned integer to the message.
1130 * @param message the message
1131 * @param value the integer value
1132 * @returns #TRUE on success
1135 dbus_message_append_uint32 (DBusMessage *message,
1136 dbus_uint32_t value)
1138 _dbus_assert (!message->locked);
1140 if (!_dbus_string_append_byte (&message->body, DBUS_TYPE_UINT32))
1142 _dbus_string_shorten (&message->body, 1);
1146 return _dbus_marshal_uint32 (&message->body,
1147 message->byte_order, value);
1151 * Appends a double value to the message.
1153 * @param message the message
1154 * @param value the double value
1155 * @returns #TRUE on success
1158 dbus_message_append_double (DBusMessage *message,
1161 _dbus_assert (!message->locked);
1163 if (!_dbus_string_append_byte (&message->body, DBUS_TYPE_DOUBLE))
1165 _dbus_string_shorten (&message->body, 1);
1169 return _dbus_marshal_double (&message->body,
1170 message->byte_order, value);
1174 * Appends a UTF-8 string to the message.
1176 * @param message the message
1177 * @param value the string
1178 * @returns #TRUE on success
1181 dbus_message_append_string (DBusMessage *message,
1184 _dbus_assert (!message->locked);
1186 if (!_dbus_string_append_byte (&message->body, DBUS_TYPE_STRING))
1188 _dbus_string_shorten (&message->body, 1);
1192 return _dbus_marshal_string (&message->body,
1193 message->byte_order, value);
1197 * Appends a byte array to the message.
1199 * @param message the message
1200 * @param value the array
1201 * @param len the length of the array
1202 * @returns #TRUE on success
1205 dbus_message_append_byte_array (DBusMessage *message,
1206 unsigned const char *value,
1209 _dbus_assert (!message->locked);
1211 if (!_dbus_string_append_byte (&message->body, DBUS_TYPE_BYTE_ARRAY))
1213 _dbus_string_shorten (&message->body, 1);
1217 return _dbus_marshal_byte_array (&message->body,
1218 message->byte_order, value, len);
1222 * Appends a string array to the message.
1224 * @param message the message
1225 * @param value the array
1226 * @param len the length of the array
1227 * @returns #TRUE on success
1230 dbus_message_append_string_array (DBusMessage *message,
1234 _dbus_assert (!message->locked);
1236 if (!_dbus_string_append_byte (&message->body, DBUS_TYPE_STRING_ARRAY))
1238 _dbus_string_shorten (&message->body, 1);
1242 return _dbus_marshal_string_array (&message->body,
1243 message->byte_order, value, len);
1247 * Gets arguments from a message given a variable argument list.
1248 * The variable argument list should contain the type of the
1249 * argumen followed by a pointer to where the value should be
1250 * stored. The list is terminated with 0.
1252 * @param message the message
1253 * @param first_arg_type the first argument type
1254 * @param ... location for first argument value, then list of type-location pairs
1255 * @returns result code
1258 dbus_message_get_args (DBusMessage *message,
1262 DBusResultCode retval;
1265 va_start (var_args, first_arg_type);
1266 retval = dbus_message_get_args_valist (message, first_arg_type, var_args);
1273 * This function takes a va_list for use by language bindings
1275 * @todo this function (or some lower-level non-convenience function)
1276 * needs better error handling; should allow the application to
1277 * distinguish between out of memory, and bad data from the remote
1278 * app. It also needs to not leak a bunch of args when it gets
1279 * to the arg that's bad, as that would be a security hole
1280 * (allow one app to force another to leak memory)
1282 * @todo We need to free the argument data when an error occurs.
1284 * @see dbus_message_get_args
1285 * @param message the message
1286 * @param first_arg_type type of the first argument
1287 * @param var_args return location for first argument, followed by list of type/location pairs
1288 * @returns result code
1291 dbus_message_get_args_valist (DBusMessage *message,
1295 int spec_type, msg_type, i;
1296 DBusMessageIter *iter;
1298 iter = dbus_message_get_args_iter (message);
1301 return DBUS_RESULT_NO_MEMORY;
1303 spec_type = first_arg_type;
1306 while (spec_type != 0)
1308 msg_type = dbus_message_iter_get_arg_type (iter);
1310 if (msg_type != spec_type)
1312 _dbus_verbose ("Argument %d is specified to be of type \"%s\", but "
1313 "is actually of type \"%s\"\n", i,
1314 _dbus_type_to_string (spec_type),
1315 _dbus_type_to_string (msg_type));
1316 dbus_message_iter_unref (iter);
1318 return DBUS_RESULT_INVALID_ARGS;
1323 case DBUS_TYPE_INT32:
1327 ptr = va_arg (var_args, dbus_int32_t *);
1329 *ptr = dbus_message_iter_get_int32 (iter);
1332 case DBUS_TYPE_UINT32:
1336 ptr = va_arg (var_args, dbus_uint32_t *);
1338 *ptr = dbus_message_iter_get_uint32 (iter);
1342 case DBUS_TYPE_DOUBLE:
1346 ptr = va_arg (var_args, double *);
1348 *ptr = dbus_message_iter_get_double (iter);
1352 case DBUS_TYPE_STRING:
1356 ptr = va_arg (var_args, char **);
1358 *ptr = dbus_message_iter_get_string (iter);
1361 return DBUS_RESULT_NO_MEMORY;
1366 case DBUS_TYPE_BYTE_ARRAY:
1368 unsigned char **ptr;
1371 ptr = va_arg (var_args, unsigned char **);
1372 len = va_arg (var_args, int *);
1374 *ptr = dbus_message_iter_get_byte_array (iter, len);
1377 return DBUS_RESULT_NO_MEMORY;
1381 case DBUS_TYPE_STRING_ARRAY:
1386 ptr = va_arg (var_args, char ***);
1387 len = va_arg (var_args, int *);
1389 *ptr = dbus_message_iter_get_string_array (iter, len);
1392 return DBUS_RESULT_NO_MEMORY;
1397 _dbus_warn ("Unknown field type %d\n", spec_type);
1400 spec_type = va_arg (var_args, int);
1401 if (spec_type != 0 && !dbus_message_iter_next (iter))
1403 _dbus_verbose ("More fields than exist in the message were specified or field is corrupt\n");
1405 dbus_message_iter_unref (iter);
1406 return DBUS_RESULT_INVALID_ARGS;
1411 dbus_message_iter_unref (iter);
1412 return DBUS_RESULT_SUCCESS;
1416 * Returns a DBusMessageIter representing the arguments of the
1417 * message passed in.
1419 * @todo IMO the message iter should follow the GtkTextIter pattern,
1420 * a static object with a "stamp" value used to detect invalid
1421 * iter uses (uninitialized or after changing the message).
1422 * ref/unref is kind of annoying to deal with, and slower too.
1423 * This implies not ref'ing the message from the iter.
1425 * @param message the message
1426 * @returns a new iter.
1429 dbus_message_get_args_iter (DBusMessage *message)
1431 DBusMessageIter *iter;
1433 iter = dbus_new (DBusMessageIter, 1);
1435 dbus_message_ref (message);
1438 iter->message = message;
1445 * Increments the reference count of a DBusMessageIter.
1447 * @param iter the message iter
1448 * @see dbus_message_iter_unref
1451 dbus_message_iter_ref (DBusMessageIter *iter)
1453 _dbus_assert (iter->refcount > 0);
1455 iter->refcount += 1;
1459 * Decrements the reference count of a DBusMessageIter.
1461 * @param iter The message iter
1462 * @see dbus_message_iter_ref
1465 dbus_message_iter_unref (DBusMessageIter *iter)
1467 _dbus_assert (iter->refcount > 0);
1469 iter->refcount -= 1;
1471 if (iter->refcount == 0)
1473 dbus_message_unref (iter->message);
1480 * Checks if an iterator has any more fields.
1482 * @param iter the message iter
1483 * @returns #TRUE if there are more fields
1487 dbus_message_iter_has_next (DBusMessageIter *iter)
1491 if (!_dbus_marshal_get_arg_end_pos (&iter->message->body,
1492 iter->message->byte_order,
1493 iter->pos, &end_pos))
1496 if (end_pos >= _dbus_string_get_length (&iter->message->body))
1503 * Moves the iterator to the next field.
1505 * @param iter The message iter
1506 * @returns #TRUE if the iterator was moved to the next field
1509 dbus_message_iter_next (DBusMessageIter *iter)
1513 if (!_dbus_marshal_get_arg_end_pos (&iter->message->body,
1514 iter->message->byte_order,
1515 iter->pos, &end_pos))
1518 if (end_pos >= _dbus_string_get_length (&iter->message->body))
1521 iter->pos = end_pos;
1527 * Returns the argument type of the argument that the
1528 * message iterator points at.
1530 * @param iter the message iter
1531 * @returns the field type
1534 dbus_message_iter_get_arg_type (DBusMessageIter *iter)
1538 if (iter->pos >= _dbus_string_get_length (&iter->message->body))
1539 return DBUS_TYPE_INVALID;
1541 _dbus_string_get_const_data_len (&iter->message->body, &data, iter->pos, 1);
1543 if (*data > DBUS_TYPE_INVALID && *data <= DBUS_TYPE_STRING_ARRAY)
1546 return DBUS_TYPE_INVALID;
1550 * Returns the string value that an iterator may point to.
1551 * Note that you need to check that the iterator points to
1552 * a string value before using this function.
1554 * @see dbus_message_iter_get_field_type
1555 * @param iter the message iter
1556 * @returns the string
1559 dbus_message_iter_get_string (DBusMessageIter *iter)
1561 _dbus_assert (dbus_message_iter_get_arg_type (iter) == DBUS_TYPE_STRING);
1563 return _dbus_demarshal_string (&iter->message->body, iter->message->byte_order,
1564 iter->pos + 1, NULL);
1568 * Returns the 32 bit signed integer value that an iterator may point to.
1569 * Note that you need to check that the iterator points to
1570 * an integer value before using this function.
1572 * @see dbus_message_iter_get_field_type
1573 * @param iter the message iter
1574 * @returns the integer
1577 dbus_message_iter_get_int32 (DBusMessageIter *iter)
1579 return _dbus_demarshal_int32 (&iter->message->body, iter->message->byte_order,
1580 iter->pos + 1, NULL);
1584 * Returns the 32 bit unsigned integer value that an iterator may point to.
1585 * Note that you need to check that the iterator points to
1586 * an unsigned integer value before using this function.
1588 * @see dbus_message_iter_get_field_type
1589 * @param iter the message iter
1590 * @returns the integer
1593 dbus_message_iter_get_uint32 (DBusMessageIter *iter)
1595 return _dbus_demarshal_uint32 (&iter->message->body, iter->message->byte_order,
1596 iter->pos + 1, NULL);
1600 * Returns the double value that an iterator may point to.
1601 * Note that you need to check that the iterator points to
1602 * a string value before using this function.
1604 * @see dbus_message_iter_get_field_type
1605 * @param iter the message iter
1606 * @returns the double
1609 dbus_message_iter_get_double (DBusMessageIter *iter)
1611 return _dbus_demarshal_double (&iter->message->body, iter->message->byte_order,
1612 iter->pos + 1, NULL);
1616 * Returns the byte array that the iterator may point to.
1617 * Note that you need to check that the iterator points
1618 * to a byte array prior to using this function.
1620 * @todo this function should probably take "unsigned char **" as
1621 * an out param argument, and return boolean or result code.
1623 * @param iter the iterator
1624 * @param len return location for length of byte array
1625 * @returns the byte array
1628 dbus_message_iter_get_byte_array (DBusMessageIter *iter,
1631 _dbus_assert (dbus_message_iter_get_arg_type (iter) == DBUS_TYPE_BYTE_ARRAY);
1633 return _dbus_demarshal_byte_array (&iter->message->body, iter->message->byte_order,
1634 iter->pos + 1, NULL, len);
1638 * Returns the string array that the iterator may point to.
1639 * Note that you need to check that the iterator points
1640 * to a byte array prior to using this function.
1642 * @todo this function should probably take "char **" as
1643 * an out param argument, and return boolean or result code.
1645 * @param iter the iterator
1646 * @param len return location for length of byte array
1647 * @returns the byte array
1650 dbus_message_iter_get_string_array (DBusMessageIter *iter,
1653 _dbus_assert (dbus_message_iter_get_arg_type (iter) == DBUS_TYPE_STRING_ARRAY);
1655 return _dbus_demarshal_string_array (&iter->message->body, iter->message->byte_order,
1656 iter->pos + 1, NULL, len);
1660 * Sets the message sender.
1662 * @param message the message
1663 * @param sender the sender
1664 * @returns #FALSE if not enough memory
1667 dbus_message_set_sender (DBusMessage *message,
1670 _dbus_assert (!message->locked);
1674 delete_string_field (message, FIELD_SENDER);
1679 return set_string_field (message,
1686 * Sets a flag indicating that the message is an error reply
1687 * message, i.e. an "exception" rather than a normal response.
1689 * @param message the message
1690 * @param is_error_reply #TRUE if this is an error message.
1693 dbus_message_set_is_error (DBusMessage *message,
1694 dbus_bool_t is_error_reply)
1698 _dbus_assert (!message->locked);
1700 _dbus_string_get_data_len (&message->header, &header, 1, 1);
1703 *header |= DBUS_HEADER_FLAG_ERROR;
1705 *header &= ~DBUS_HEADER_FLAG_ERROR;
1709 * Returns #TRUE if the message is an error
1710 * reply to some previous message we sent.
1712 * @param message the message
1713 * @returns #TRUE if the message is an error
1716 dbus_message_get_is_error (DBusMessage *message)
1720 _dbus_string_get_const_data_len (&message->header, &header, 1, 1);
1722 return (*header & DBUS_HEADER_FLAG_ERROR) != 0;
1726 * Gets the service which originated this message,
1727 * or #NULL if unknown or inapplicable.
1729 * @param message the message
1730 * @returns the service name or #NULL
1733 dbus_message_get_sender (DBusMessage *message)
1735 return get_string_field (message, FIELD_SENDER, NULL);
1739 dbus_message_name_is (DBusMessage *message,
1742 if (dbus_message_get_name (message) &&
1743 strcmp (dbus_message_get_name (message), name) == 0)
1752 * @addtogroup DBusMessageInternals
1757 * @typedef DBusMessageLoader
1759 * The DBusMessageLoader object encapsulates the process of converting
1760 * a byte stream into a series of DBusMessage. It buffers the incoming
1761 * bytes as efficiently as possible, and generates a queue of
1762 * messages. DBusMessageLoader is typically used as part of a
1763 * DBusTransport implementation. The DBusTransport then hands off
1764 * the loaded messages to a DBusConnection, making the messages
1765 * visible to the application.
1769 /* we definitely use signed ints for sizes, so don't exceed
1770 * _DBUS_INT_MAX; and add 16 for paranoia, since a message
1771 * over 128M is pretty nuts anyhow.
1775 * The maximum sane message size.
1777 #define MAX_SANE_MESSAGE_SIZE (_DBUS_INT_MAX/16)
1780 * Implementation details of DBusMessageLoader.
1781 * All members are private.
1783 struct DBusMessageLoader
1785 int refcount; /**< Reference count. */
1787 DBusString data; /**< Buffered data */
1789 DBusList *messages; /**< Complete messages. */
1791 long max_message_size; /**< Maximum size of a message */
1793 unsigned int buffer_outstanding : 1; /**< Someone is using the buffer to read */
1795 unsigned int corrupted : 1; /**< We got broken data, and are no longer working */
1799 * The initial buffer size of the message loader.
1801 * @todo this should be based on min header size plus some average
1802 * body size, or something. Or rather, the min header size only, if we
1803 * want to try to read only the header, store that in a DBusMessage,
1804 * then read only the body and store that, etc., depends on
1805 * how we optimize _dbus_message_loader_get_buffer() and what
1806 * the exact message format is.
1808 #define INITIAL_LOADER_DATA_LEN 32
1811 * Creates a new message loader. Returns #NULL if memory can't
1814 * @returns new loader, or #NULL.
1817 _dbus_message_loader_new (void)
1819 DBusMessageLoader *loader;
1821 loader = dbus_new0 (DBusMessageLoader, 1);
1825 loader->refcount = 1;
1827 /* Try to cap message size at something that won't *totally* hose
1828 * the system if we have a couple of them.
1830 loader->max_message_size = _DBUS_ONE_MEGABYTE * 32;
1832 if (!_dbus_string_init (&loader->data, _DBUS_INT_MAX))
1838 /* preallocate the buffer for speed, ignore failure */
1839 _dbus_string_set_length (&loader->data, INITIAL_LOADER_DATA_LEN);
1840 _dbus_string_set_length (&loader->data, 0);
1846 * Increments the reference count of the loader.
1848 * @param loader the loader.
1851 _dbus_message_loader_ref (DBusMessageLoader *loader)
1853 loader->refcount += 1;
1857 * Decrements the reference count of the loader and finalizes the
1858 * loader when the count reaches zero.
1860 * @param loader the loader.
1863 _dbus_message_loader_unref (DBusMessageLoader *loader)
1865 loader->refcount -= 1;
1866 if (loader->refcount == 0)
1868 _dbus_list_foreach (&loader->messages,
1869 (DBusForeachFunction) dbus_message_unref,
1871 _dbus_list_clear (&loader->messages);
1872 _dbus_string_free (&loader->data);
1878 * Gets the buffer to use for reading data from the network. Network
1879 * data is read directly into an allocated buffer, which is then used
1880 * in the DBusMessage, to avoid as many extra memcpy's as possible.
1881 * The buffer must always be returned immediately using
1882 * _dbus_message_loader_return_buffer(), even if no bytes are
1883 * successfully read.
1885 * @todo this function can be a lot more clever. For example
1886 * it can probably always return a buffer size to read exactly
1887 * the body of the next message, thus avoiding any memory wastage
1890 * @param loader the message loader.
1891 * @param buffer the buffer
1894 _dbus_message_loader_get_buffer (DBusMessageLoader *loader,
1895 DBusString **buffer)
1897 _dbus_assert (!loader->buffer_outstanding);
1899 *buffer = &loader->data;
1901 loader->buffer_outstanding = TRUE;
1905 * The smallest header size that can occur.
1906 * (It won't be valid)
1908 #define DBUS_MINIMUM_HEADER_SIZE 16
1910 /** Pack four characters as in "abcd" into a uint32 */
1911 #define FOUR_CHARS_TO_UINT32(a, b, c, d) \
1912 ((((dbus_uint32_t)a) << 24) | \
1913 (((dbus_uint32_t)b) << 16) | \
1914 (((dbus_uint32_t)c) << 8) | \
1917 /** DBUS_HEADER_FIELD_NAME packed into a dbus_uint32_t */
1918 #define DBUS_HEADER_FIELD_NAME_AS_UINT32 \
1919 FOUR_CHARS_TO_UINT32 ('n', 'a', 'm', 'e')
1921 /** DBUS_HEADER_FIELD_SERVICE packed into a dbus_uint32_t */
1922 #define DBUS_HEADER_FIELD_SERVICE_AS_UINT32 \
1923 FOUR_CHARS_TO_UINT32 ('s', 'r', 'v', 'c')
1925 /** DBUS_HEADER_FIELD_REPLY packed into a dbus_uint32_t */
1926 #define DBUS_HEADER_FIELD_REPLY_AS_UINT32 \
1927 FOUR_CHARS_TO_UINT32 ('r', 'p', 'l', 'y')
1929 /** DBUS_HEADER_FIELD_SENDER Packed into a dbus_uint32_t */
1930 #define DBUS_HEADER_FIELD_SENDER_AS_UINT32 \
1931 FOUR_CHARS_TO_UINT32 ('s', 'n', 'd', 'r')
1934 decode_header_data (const DBusString *data,
1937 HeaderField fields[FIELD_LAST],
1938 int *message_padding)
1944 if (header_len < 16)
1948 while (i < FIELD_LAST)
1950 fields[i].offset = -1;
1954 fields[FIELD_HEADER_LENGTH].offset = 4;
1955 fields[FIELD_BODY_LENGTH].offset = 8;
1956 fields[FIELD_CLIENT_SERIAL].offset = 12;
1958 /* Now handle the named fields. A real named field is at least 4
1959 * bytes for the name, plus a type code (1 byte) plus padding. So
1960 * if we have less than 8 bytes left, it must be alignment padding,
1961 * not a field. While >= 8 bytes can't be entirely alignment
1965 while ((pos + 7) < header_len)
1967 pos = _DBUS_ALIGN_VALUE (pos, 4);
1969 if ((pos + 4) > header_len)
1972 _dbus_string_get_const_data_len (data, &field, pos, 4);
1975 _dbus_assert (_DBUS_ALIGN_ADDRESS (field, 4) == field);
1977 /* I believe FROM_BE is right, but if not we'll find out
1980 switch (DBUS_UINT32_FROM_BE (*(int*)field))
1982 case DBUS_HEADER_FIELD_SERVICE_AS_UINT32:
1983 if (fields[FIELD_SERVICE].offset >= 0)
1985 _dbus_verbose ("%s field provided twice\n",
1986 DBUS_HEADER_FIELD_SERVICE);
1990 fields[FIELD_SERVICE].offset = _DBUS_ALIGN_VALUE (pos + 1, 4);
1991 _dbus_verbose ("Found service name at offset %d\n",
1992 fields[FIELD_SERVICE].offset);
1995 case DBUS_HEADER_FIELD_NAME_AS_UINT32:
1996 if (fields[FIELD_NAME].offset >= 0)
1998 _dbus_verbose ("%s field provided twice\n",
1999 DBUS_HEADER_FIELD_NAME);
2003 fields[FIELD_NAME].offset = _DBUS_ALIGN_VALUE (pos + 1, 4);
2005 _dbus_verbose ("Found message name at offset %d\n",
2006 fields[FIELD_NAME].offset);
2008 case DBUS_HEADER_FIELD_SENDER_AS_UINT32:
2009 if (fields[FIELD_SENDER].offset >= 0)
2011 _dbus_verbose ("%s field provided twice\n",
2012 DBUS_HEADER_FIELD_SENDER);
2016 fields[FIELD_SENDER].offset = _DBUS_ALIGN_VALUE (pos + 1, 4);
2018 _dbus_verbose ("Found sender name at offset %d\n",
2019 fields[FIELD_NAME].offset);
2022 case DBUS_HEADER_FIELD_REPLY_AS_UINT32:
2023 if (fields[FIELD_REPLY_SERIAL].offset >= 0)
2025 _dbus_verbose ("%s field provided twice\n",
2026 DBUS_HEADER_FIELD_REPLY);
2030 fields[FIELD_REPLY_SERIAL].offset = _DBUS_ALIGN_VALUE (pos + 1, 4);
2032 _dbus_verbose ("Found reply serial at offset %d\n",
2033 fields[FIELD_REPLY_SERIAL].offset);
2037 _dbus_verbose ("Ignoring an unknown header field: %c%c%c%c at offset %d\n",
2038 field[0], field[1], field[2], field[3], pos);
2041 if (!_dbus_marshal_validate_arg (data, byte_order, pos, &new_pos))
2043 _dbus_verbose ("Failed to validate argument to named header field\n");
2047 if (new_pos > header_len)
2049 _dbus_verbose ("Named header field tries to extend beyond header length\n");
2056 if (pos < header_len)
2058 /* Alignment padding, verify that it's nul */
2059 _dbus_assert ((header_len - pos) < 8);
2061 if (!_dbus_string_validate_nul (data,
2062 pos, (header_len - pos)))
2064 _dbus_verbose ("header alignment padding is not nul\n");
2068 if (message_padding)
2069 *message_padding = header_len - pos;
2076 * Returns a buffer obtained from _dbus_message_loader_get_buffer(),
2077 * indicating to the loader how many bytes of the buffer were filled
2078 * in. This function must always be called, even if no bytes were
2079 * successfully read.
2081 * @todo if we run out of memory in here, we offer no way for calling
2082 * code to handle it, i.e. they can't re-run the message parsing
2083 * attempt. Perhaps much of this code could be moved to pop_message()?
2084 * But then that may need to distinguish NULL return for no messages
2085 * from NULL return for errors.
2087 * @param loader the loader.
2088 * @param buffer the buffer.
2089 * @param bytes_read number of bytes that were read into the buffer.
2092 _dbus_message_loader_return_buffer (DBusMessageLoader *loader,
2096 _dbus_assert (loader->buffer_outstanding);
2097 _dbus_assert (buffer == &loader->data);
2099 loader->buffer_outstanding = FALSE;
2101 if (loader->corrupted)
2104 while (_dbus_string_get_length (&loader->data) >= 16)
2106 DBusMessage *message;
2107 const char *header_data;
2108 int byte_order, header_len, body_len, header_padding;
2109 dbus_uint32_t header_len_unsigned, body_len_unsigned;
2111 _dbus_string_get_const_data_len (&loader->data, &header_data, 0, 16);
2113 _dbus_assert (_DBUS_ALIGN_ADDRESS (header_data, 4) == header_data);
2115 if (header_data[2] != DBUS_MAJOR_PROTOCOL_VERSION)
2117 _dbus_verbose ("Message has protocol version %d ours is %d\n",
2118 (int) header_data[2], DBUS_MAJOR_PROTOCOL_VERSION);
2119 loader->corrupted = TRUE;
2123 byte_order = header_data[0];
2125 if (byte_order != DBUS_LITTLE_ENDIAN &&
2126 byte_order != DBUS_BIG_ENDIAN)
2128 _dbus_verbose ("Message with bad byte order '%c' received\n",
2130 loader->corrupted = TRUE;
2134 header_len_unsigned = _dbus_unpack_uint32 (byte_order, header_data + 4);
2135 body_len_unsigned = _dbus_unpack_uint32 (byte_order, header_data + 8);
2137 if (header_len_unsigned < 16)
2139 _dbus_verbose ("Message had broken too-small header length %u\n",
2140 header_len_unsigned);
2141 loader->corrupted = TRUE;
2145 if (header_len_unsigned > (unsigned) MAX_SANE_MESSAGE_SIZE ||
2146 body_len_unsigned > (unsigned) MAX_SANE_MESSAGE_SIZE)
2148 _dbus_verbose ("Header or body length too large (%u %u)\n",
2149 header_len_unsigned,
2151 loader->corrupted = TRUE;
2155 /* Now that we know the values are in signed range, get
2156 * rid of stupid unsigned, just causes bugs
2158 header_len = header_len_unsigned;
2159 body_len = body_len_unsigned;
2161 if (_DBUS_ALIGN_VALUE (header_len, 8) != header_len_unsigned)
2163 _dbus_verbose ("header length %d is not aligned to 8 bytes\n",
2165 loader->corrupted = TRUE;
2169 if (header_len + body_len > loader->max_message_size)
2171 _dbus_verbose ("Message claimed length header = %d body = %d exceeds max message length %d\n",
2172 header_len, body_len, loader->max_message_size);
2173 loader->corrupted = TRUE;
2177 if (_dbus_string_get_length (&loader->data) >= (header_len + body_len))
2179 HeaderField fields[FIELD_LAST];
2183 _dbus_verbose_bytes_of_string (&loader->data, 0, header_len);
2184 if (!decode_header_data (&loader->data, header_len, byte_order,
2185 fields, &header_padding))
2187 _dbus_verbose ("Header was invalid\n");
2188 loader->corrupted = TRUE;
2192 next_arg = header_len;
2193 while (next_arg < (header_len + body_len))
2195 int prev = next_arg;
2197 if (!_dbus_marshal_validate_arg (&loader->data,
2202 loader->corrupted = TRUE;
2206 _dbus_assert (next_arg > prev);
2209 if (next_arg > (header_len + body_len))
2211 _dbus_verbose ("end of last arg at %d but message has len %d+%d=%d\n",
2212 next_arg, header_len, body_len,
2213 header_len + body_len);
2214 loader->corrupted = TRUE;
2218 message = dbus_message_new_empty_header ();
2219 if (message == NULL)
2220 break; /* ugh, postpone this I guess. */
2222 message->byte_order = byte_order;
2223 message->header_padding = header_padding;
2225 /* Copy in the offsets we found */
2227 while (i < FIELD_LAST)
2229 message->header_fields[i] = fields[i];
2233 if (!_dbus_list_append (&loader->messages, message))
2235 dbus_message_unref (message);
2239 _dbus_assert (_dbus_string_get_length (&message->header) == 0);
2240 _dbus_assert (_dbus_string_get_length (&message->body) == 0);
2242 _dbus_assert (_dbus_string_get_length (&loader->data) >=
2243 (header_len + body_len));
2245 if (!_dbus_string_move_len (&loader->data, 0, header_len, &message->header, 0))
2247 _dbus_list_remove_last (&loader->messages, message);
2248 dbus_message_unref (message);
2252 if (!_dbus_string_move_len (&loader->data, 0, body_len, &message->body, 0))
2256 /* put the header back, we'll try again later */
2257 result = _dbus_string_copy_len (&message->header, 0, header_len,
2259 _dbus_assert (result); /* because DBusString never reallocs smaller */
2261 _dbus_list_remove_last (&loader->messages, message);
2262 dbus_message_unref (message);
2266 _dbus_assert (_dbus_string_get_length (&message->header) == header_len);
2267 _dbus_assert (_dbus_string_get_length (&message->body) == body_len);
2269 _dbus_verbose ("Loaded message %p\n", message);
2277 * Pops a loaded message (passing ownership of the message
2278 * to the caller). Returns #NULL if no messages have been
2281 * @param loader the loader.
2282 * @returns the next message, or #NULL if none.
2285 _dbus_message_loader_pop_message (DBusMessageLoader *loader)
2287 return _dbus_list_pop_first (&loader->messages);
2292 * Checks whether the loader is confused due to bad data.
2293 * If messages are received that are invalid, the
2294 * loader gets confused and gives up permanently.
2295 * This state is called "corrupted."
2297 * @param loader the loader
2298 * @returns #TRUE if the loader is hosed.
2301 _dbus_message_loader_get_is_corrupted (DBusMessageLoader *loader)
2303 return loader->corrupted;
2307 * Sets the maximum size message we allow.
2309 * @param loader the loader
2310 * @param size the max message size in bytes
2313 _dbus_message_loader_set_max_message_size (DBusMessageLoader *loader,
2316 if (size > MAX_SANE_MESSAGE_SIZE)
2318 _dbus_verbose ("clamping requested max message size %ld to %d\n",
2319 size, MAX_SANE_MESSAGE_SIZE);
2320 size = MAX_SANE_MESSAGE_SIZE;
2322 loader->max_message_size = size;
2326 * Gets the maximum allowed message size in bytes.
2328 * @param loader the loader
2329 * @returns max size in bytes
2332 _dbus_message_loader_get_max_message_size (DBusMessageLoader *loader)
2334 return loader->max_message_size;
2338 #ifdef DBUS_BUILD_TESTS
2339 #include "dbus-test.h"
2343 message_iter_test (DBusMessage *message)
2345 DBusMessageIter *iter;
2348 iter = dbus_message_get_args_iter (message);
2351 if (dbus_message_iter_get_arg_type (iter) != DBUS_TYPE_STRING)
2352 _dbus_assert_not_reached ("Argument type isn't string");
2354 str = dbus_message_iter_get_string (iter);
2355 if (strcmp (str, "Test string") != 0)
2356 _dbus_assert_not_reached ("Strings differ");
2359 if (!dbus_message_iter_next (iter))
2360 _dbus_assert_not_reached ("Reached end of arguments");
2362 /* Signed integer tests */
2363 if (dbus_message_iter_get_arg_type (iter) != DBUS_TYPE_INT32)
2364 _dbus_assert_not_reached ("Argument type isn't int32");
2366 if (dbus_message_iter_get_int32 (iter) != -0x12345678)
2367 _dbus_assert_not_reached ("Signed integers differ");
2369 if (!dbus_message_iter_next (iter))
2370 _dbus_assert_not_reached ("Reached end of fields");
2372 /* Unsigned integer tests */
2373 if (dbus_message_iter_get_arg_type (iter) != DBUS_TYPE_UINT32)
2374 _dbus_assert_not_reached ("Argument type isn't int32");
2376 if (dbus_message_iter_get_int32 (iter) != 0xedd1e)
2377 _dbus_assert_not_reached ("Unsigned integers differ");
2379 if (!dbus_message_iter_next (iter))
2380 _dbus_assert_not_reached ("Reached end of arguments");
2383 if (dbus_message_iter_get_arg_type (iter) != DBUS_TYPE_DOUBLE)
2384 _dbus_assert_not_reached ("Argument type isn't double");
2386 if (dbus_message_iter_get_double (iter) != 3.14159)
2387 _dbus_assert_not_reached ("Doubles differ");
2389 if (dbus_message_iter_next (iter))
2390 _dbus_assert_not_reached ("Didn't reach end of arguments");
2392 dbus_message_iter_unref (iter);
2396 check_message_handling (DBusMessage *message)
2398 DBusMessageIter *iter;
2401 dbus_int32_t client_serial;
2406 client_serial = _dbus_message_get_client_serial (message);
2408 /* can't use set_client_serial due to the assertions at the start of it */
2409 set_int_field (message, FIELD_CLIENT_SERIAL,
2412 if (client_serial != _dbus_message_get_client_serial (message))
2414 _dbus_warn ("get/set cycle for client_serial did not succeed\n");
2418 /* If we implement message_set_arg (message, n, value)
2419 * then we would want to test it here
2422 iter = dbus_message_get_args_iter (message);
2423 while ((type = dbus_message_iter_get_arg_type (iter)) != DBUS_TYPE_INVALID)
2427 case DBUS_TYPE_STRING:
2430 str = dbus_message_iter_get_string (iter);
2436 if (!dbus_message_iter_next (iter))
2444 dbus_message_iter_unref (iter);
2450 check_have_valid_message (DBusMessageLoader *loader)
2452 DBusMessage *message;
2458 if (_dbus_message_loader_get_is_corrupted (loader))
2460 _dbus_warn ("loader corrupted on message that was expected to be valid\n");
2464 message = _dbus_message_loader_pop_message (loader);
2465 if (message == NULL)
2467 _dbus_warn ("didn't load message that was expected to be valid (message not popped)\n");
2471 if (_dbus_string_get_length (&loader->data) > 0)
2473 _dbus_warn ("had leftover bytes from expected-to-be-valid single message\n");
2477 /* Verify that we're able to properly deal with the message.
2478 * For example, this would detect improper handling of messages
2479 * in nonstandard byte order.
2481 if (!check_message_handling (message))
2488 dbus_message_unref (message);
2493 check_invalid_message (DBusMessageLoader *loader)
2499 if (!_dbus_message_loader_get_is_corrupted (loader))
2501 _dbus_warn ("loader not corrupted on message that was expected to be invalid\n");
2512 check_incomplete_message (DBusMessageLoader *loader)
2514 DBusMessage *message;
2520 if (_dbus_message_loader_get_is_corrupted (loader))
2522 _dbus_warn ("loader corrupted on message that was expected to be valid (but incomplete)\n");
2526 message = _dbus_message_loader_pop_message (loader);
2527 if (message != NULL)
2529 _dbus_warn ("loaded message that was expected to be incomplete\n");
2537 dbus_message_unref (message);
2542 check_loader_results (DBusMessageLoader *loader,
2543 DBusMessageValidity validity)
2547 case _DBUS_MESSAGE_VALID:
2548 return check_have_valid_message (loader);
2549 case _DBUS_MESSAGE_INVALID:
2550 return check_invalid_message (loader);
2551 case _DBUS_MESSAGE_INCOMPLETE:
2552 return check_incomplete_message (loader);
2553 case _DBUS_MESSAGE_UNKNOWN:
2557 _dbus_assert_not_reached ("bad DBusMessageValidity");
2563 * Loads the message in the given message file.
2565 * @param filename filename to load
2566 * @param is_raw if #TRUE load as binary data, if #FALSE as message builder language
2567 * @param data string to load message into
2568 * @returns #TRUE if the message was loaded
2571 dbus_internal_do_not_use_load_message_file (const DBusString *filename,
2581 DBusResultCode result;
2583 result = _dbus_file_get_contents (data, filename);
2584 if (result != DBUS_RESULT_SUCCESS)
2587 _dbus_string_get_const_data (filename, &s);
2588 _dbus_warn ("Could not load message file %s\n", s);
2594 if (!_dbus_message_data_load (data, filename))
2597 _dbus_string_get_const_data (filename, &s);
2598 _dbus_warn ("Could not load message file %s\n", s);
2611 * Tries loading the message in the given message file
2612 * and verifies that DBusMessageLoader can handle it.
2614 * @param filename filename to load
2615 * @param is_raw if #TRUE load as binary data, if #FALSE as message builder language
2616 * @param expected_validity what the message has to be like to return #TRUE
2617 * @returns #TRUE if the message has the expected validity
2620 dbus_internal_do_not_use_try_message_file (const DBusString *filename,
2622 DBusMessageValidity expected_validity)
2629 if (!_dbus_string_init (&data, _DBUS_INT_MAX))
2630 _dbus_assert_not_reached ("could not allocate string\n");
2632 if (!dbus_internal_do_not_use_load_message_file (filename, is_raw,
2636 retval = dbus_internal_do_not_use_try_message_data (&data, expected_validity);
2644 if (_dbus_string_get_length (&data) > 0)
2645 _dbus_verbose_bytes_of_string (&data, 0,
2646 _dbus_string_get_length (&data));
2648 _dbus_string_get_const_data (filename, &s);
2649 _dbus_warn ("Failed message loader test on %s\n",
2653 _dbus_string_free (&data);
2659 * Tries loading the given message data.
2662 * @param data the message data
2663 * @param expected_validity what the message has to be like to return #TRUE
2664 * @returns #TRUE if the message has the expected validity
2667 dbus_internal_do_not_use_try_message_data (const DBusString *data,
2668 DBusMessageValidity expected_validity)
2670 DBusMessageLoader *loader;
2678 /* Write the data one byte at a time */
2680 loader = _dbus_message_loader_new ();
2682 len = _dbus_string_get_length (data);
2683 for (i = 0; i < len; i++)
2687 _dbus_message_loader_get_buffer (loader, &buffer);
2688 _dbus_string_append_byte (buffer,
2689 _dbus_string_get_byte (data, i));
2690 _dbus_message_loader_return_buffer (loader, buffer, 1);
2693 if (!check_loader_results (loader, expected_validity))
2696 _dbus_message_loader_unref (loader);
2699 /* Write the data all at once */
2701 loader = _dbus_message_loader_new ();
2706 _dbus_message_loader_get_buffer (loader, &buffer);
2707 _dbus_string_copy (data, 0, buffer,
2708 _dbus_string_get_length (buffer));
2709 _dbus_message_loader_return_buffer (loader, buffer, 1);
2712 if (!check_loader_results (loader, expected_validity))
2715 _dbus_message_loader_unref (loader);
2718 /* Write the data 2 bytes at a time */
2720 loader = _dbus_message_loader_new ();
2722 len = _dbus_string_get_length (data);
2723 for (i = 0; i < len; i += 2)
2727 _dbus_message_loader_get_buffer (loader, &buffer);
2728 _dbus_string_append_byte (buffer,
2729 _dbus_string_get_byte (data, i));
2731 _dbus_string_append_byte (buffer,
2732 _dbus_string_get_byte (data, i+1));
2733 _dbus_message_loader_return_buffer (loader, buffer, 1);
2736 if (!check_loader_results (loader, expected_validity))
2739 _dbus_message_loader_unref (loader);
2747 _dbus_message_loader_unref (loader);
2753 process_test_subdir (const DBusString *test_base_dir,
2755 DBusMessageValidity validity,
2756 DBusForeachMessageFileFunc function,
2759 DBusString test_directory;
2760 DBusString filename;
2763 DBusResultCode result;
2768 if (!_dbus_string_init (&test_directory, _DBUS_INT_MAX))
2769 _dbus_assert_not_reached ("didn't allocate test_directory\n");
2771 _dbus_string_init_const (&filename, subdir);
2773 if (!_dbus_string_copy (test_base_dir, 0,
2774 &test_directory, 0))
2775 _dbus_assert_not_reached ("couldn't copy test_base_dir to test_directory");
2777 if (!_dbus_concat_dir_and_file (&test_directory, &filename))
2778 _dbus_assert_not_reached ("couldn't allocate full path");
2780 _dbus_string_free (&filename);
2781 if (!_dbus_string_init (&filename, _DBUS_INT_MAX))
2782 _dbus_assert_not_reached ("didn't allocate filename string\n");
2784 dir = _dbus_directory_open (&test_directory, &result);
2788 _dbus_string_get_const_data (&test_directory, &s);
2789 _dbus_warn ("Could not open %s: %s\n", s,
2790 dbus_result_to_string (result));
2794 printf ("Testing:\n");
2796 result = DBUS_RESULT_SUCCESS;
2798 while (_dbus_directory_get_next_file (dir, &filename, &result))
2800 DBusString full_path;
2803 if (!_dbus_string_init (&full_path, _DBUS_INT_MAX))
2804 _dbus_assert_not_reached ("couldn't init string");
2806 if (!_dbus_string_copy (&test_directory, 0, &full_path, 0))
2807 _dbus_assert_not_reached ("couldn't copy dir to full_path");
2809 if (!_dbus_concat_dir_and_file (&full_path, &filename))
2810 _dbus_assert_not_reached ("couldn't concat file to dir");
2812 if (_dbus_string_ends_with_c_str (&filename, ".message"))
2814 else if (_dbus_string_ends_with_c_str (&filename, ".message-raw"))
2818 const char *filename_c;
2819 _dbus_string_get_const_data (&filename, &filename_c);
2820 _dbus_verbose ("Skipping non-.message file %s\n",
2822 _dbus_string_free (&full_path);
2828 _dbus_string_get_const_data (&filename, &s);
2829 printf (" %s\n", s);
2832 _dbus_verbose (" expecting %s\n",
2833 validity == _DBUS_MESSAGE_VALID ? "valid" :
2834 (validity == _DBUS_MESSAGE_INVALID ? "invalid" :
2835 (validity == _DBUS_MESSAGE_INCOMPLETE ? "incomplete" : "unknown")));
2837 if (! (*function) (&full_path, is_raw, validity, user_data))
2839 _dbus_string_free (&full_path);
2843 _dbus_string_free (&full_path);
2846 if (result != DBUS_RESULT_SUCCESS)
2849 _dbus_string_get_const_data (&test_directory, &s);
2850 _dbus_warn ("Could not get next file in %s: %s\n",
2851 s, dbus_result_to_string (result));
2860 _dbus_directory_close (dir);
2861 _dbus_string_free (&test_directory);
2862 _dbus_string_free (&filename);
2868 * Runs the given function on every message file in the test suite.
2869 * The function should return #FALSE on test failure or fatal error.
2871 * @param test_data_dir root dir of the test suite data files (top_srcdir/test/data)
2872 * @param func the function to run
2873 * @param user_data data for function
2874 * @returns #FALSE if there's a failure
2877 dbus_internal_do_not_use_foreach_message_file (const char *test_data_dir,
2878 DBusForeachMessageFileFunc func,
2881 DBusString test_directory;
2886 _dbus_string_init_const (&test_directory, test_data_dir);
2888 if (!process_test_subdir (&test_directory, "valid-messages",
2889 _DBUS_MESSAGE_VALID, func, user_data))
2892 if (!process_test_subdir (&test_directory, "invalid-messages",
2893 _DBUS_MESSAGE_INVALID, func, user_data))
2896 if (!process_test_subdir (&test_directory, "incomplete-messages",
2897 _DBUS_MESSAGE_INCOMPLETE, func, user_data))
2904 _dbus_string_free (&test_directory);
2910 * @ingroup DBusMessageInternals
2911 * Unit test for DBusMessage.
2913 * @returns #TRUE on success.
2916 _dbus_message_test (const char *test_data_dir)
2918 DBusMessage *message;
2919 DBusMessageLoader *loader;
2922 dbus_int32_t our_int;
2926 /* Test the vararg functions */
2927 message = dbus_message_new ("org.freedesktop.DBus.Test", "testMessage");
2928 _dbus_message_set_client_serial (message, 1);
2929 dbus_message_append_args (message,
2930 DBUS_TYPE_INT32, -0x12345678,
2931 DBUS_TYPE_STRING, "Test string",
2932 DBUS_TYPE_DOUBLE, 3.14159,
2934 _dbus_verbose_bytes_of_string (&message->header, 0,
2935 _dbus_string_get_length (&message->header));
2936 _dbus_verbose_bytes_of_string (&message->body, 0,
2937 _dbus_string_get_length (&message->body));
2939 if (dbus_message_get_args (message,
2940 DBUS_TYPE_INT32, &our_int,
2941 DBUS_TYPE_STRING, &our_str,
2942 DBUS_TYPE_DOUBLE, &our_double,
2943 0) != DBUS_RESULT_SUCCESS)
2944 _dbus_assert_not_reached ("Could not get arguments");
2946 if (our_int != -0x12345678)
2947 _dbus_assert_not_reached ("integers differ!");
2949 if (our_double != 3.14159)
2950 _dbus_assert_not_reached ("doubles differ!");
2952 if (strcmp (our_str, "Test string") != 0)
2953 _dbus_assert_not_reached ("strings differ!");
2955 dbus_free (our_str);
2956 dbus_message_unref (message);
2958 message = dbus_message_new ("org.freedesktop.DBus.Test", "testMessage");
2959 _dbus_message_set_client_serial (message, 1);
2960 _dbus_message_set_reply_serial (message, 0x12345678);
2962 dbus_message_append_string (message, "Test string");
2963 dbus_message_append_int32 (message, -0x12345678);
2964 dbus_message_append_uint32 (message, 0xedd1e);
2965 dbus_message_append_double (message, 3.14159);
2967 message_iter_test (message);
2969 /* Message loader test */
2970 _dbus_message_lock (message);
2971 loader = _dbus_message_loader_new ();
2973 /* Write the header data one byte at a time */
2974 _dbus_string_get_const_data (&message->header, &data);
2975 for (i = 0; i < _dbus_string_get_length (&message->header); i++)
2979 _dbus_message_loader_get_buffer (loader, &buffer);
2980 _dbus_string_append_byte (buffer, data[i]);
2981 _dbus_message_loader_return_buffer (loader, buffer, 1);
2984 /* Write the body data one byte at a time */
2985 _dbus_string_get_const_data (&message->body, &data);
2986 for (i = 0; i < _dbus_string_get_length (&message->body); i++)
2990 _dbus_message_loader_get_buffer (loader, &buffer);
2991 _dbus_string_append_byte (buffer, data[i]);
2992 _dbus_message_loader_return_buffer (loader, buffer, 1);
2995 dbus_message_unref (message);
2997 /* Now pop back the message */
2998 if (_dbus_message_loader_get_is_corrupted (loader))
2999 _dbus_assert_not_reached ("message loader corrupted");
3001 message = _dbus_message_loader_pop_message (loader);
3003 _dbus_assert_not_reached ("received a NULL message");
3005 if (_dbus_message_get_reply_serial (message) != 0x12345678)
3006 _dbus_assert_not_reached ("reply serial fields differ");
3008 message_iter_test (message);
3010 dbus_message_unref (message);
3011 _dbus_message_loader_unref (loader);
3013 /* Now load every message in test_data_dir if we have one */
3014 if (test_data_dir == NULL)
3017 return dbus_internal_do_not_use_foreach_message_file (test_data_dir,
3018 (DBusForeachMessageFileFunc)
3019 dbus_internal_do_not_use_try_message_file,
3023 #endif /* DBUS_BUILD_TESTS */