2003-10-14 Havoc Pennington <hp@redhat.com>
[platform/upstream/dbus.git] / dbus / dbus-internals.c
1 /* -*- mode: C; c-file-style: "gnu" -*- */
2 /* dbus-internals.c  random utility stuff (internal to D-BUS implementation)
3  *
4  * Copyright (C) 2002, 2003  Red Hat, Inc.
5  *
6  * Licensed under the Academic Free License version 1.2
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 #include "dbus-internals.h"
24 #include "dbus-protocol.h"
25 #include "dbus-test.h"
26 #include <stdio.h>
27 #include <stdarg.h>
28 #include <string.h>
29 #include <sys/types.h>
30 #include <errno.h>
31 #include <unistd.h>
32 #include <fcntl.h>
33 #include <stdlib.h>
34
35 /**
36  * @defgroup DBusInternals D-BUS internal implementation details
37  * @brief Documentation useful when developing or debugging D-BUS itself.
38  * 
39  */
40
41 /**
42  * @defgroup DBusInternalsUtils Utilities and portability
43  * @ingroup DBusInternals
44  * @brief Utility functions (_dbus_assert(), _dbus_warn(), etc.)
45  * @{
46  */
47
48 /**
49  * @def _dbus_assert
50  *
51  * Aborts with an error message if the condition is false.
52  * 
53  * @param condition condition which must be true.
54  */
55
56 /**
57  * @def _dbus_assert_not_reached
58  *
59  * Aborts with an error message if called.
60  * The given explanation will be printed.
61  * 
62  * @param explanation explanation of what happened if the code was reached.
63  */
64
65 /**
66  * @def _DBUS_N_ELEMENTS
67  *
68  * Computes the number of elements in a fixed-size array using
69  * sizeof().
70  *
71  * @param array the array to count elements in.
72  */
73
74 /**
75  * @def _DBUS_POINTER_TO_INT
76  *
77  * Safely casts a void* to an integer; should only be used on void*
78  * that actually contain integers, for example one created with
79  * _DBUS_INT_TO_POINTER.  Only guaranteed to preserve 32 bits.
80  * (i.e. it's used to store 32-bit ints in pointers, but
81  * can't be used to store 64-bit pointers in ints.)
82  *
83  * @param pointer pointer to extract an integer from.
84  */
85 /**
86  * @def _DBUS_INT_TO_POINTER
87  *
88  * Safely stuffs an integer into a pointer, to be extracted later with
89  * _DBUS_POINTER_TO_INT. Only guaranteed to preserve 32 bits.
90  *
91  * @param integer the integer to stuff into a pointer.
92  */
93 /**
94  * @def _DBUS_ZERO
95  *
96  * Sets all bits in an object to zero.
97  *
98  * @param object the object to be zeroed.
99  */
100 /**
101  * @def _DBUS_INT_MIN
102  *
103  * Minimum value of type "int"
104  */
105 /**
106  * @def _DBUS_INT_MAX
107  *
108  * Maximum value of type "int"
109  */
110
111 /**
112  * @typedef DBusForeachFunction
113  * 
114  * Used to iterate over each item in a collection, such as
115  * a DBusList.
116  */
117
118 /**
119  * @def _DBUS_LOCK_NAME
120  *
121  * Expands to name of a global lock variable.
122  */
123
124 /**
125  * @def _DBUS_DEFINE_GLOBAL_LOCK
126  *
127  * Defines a global lock variable with the given name.
128  * The lock must be added to the list to initialize
129  * in dbus_threads_init().
130  */
131
132 /**
133  * @def _DBUS_DECLARE_GLOBAL_LOCK
134  *
135  * Expands to declaration of a global lock defined
136  * with _DBUS_DEFINE_GLOBAL_LOCK.
137  * The lock must be added to the list to initialize
138  * in dbus_threads_init().
139  */
140
141 /**
142  * @def _DBUS_LOCK
143  *
144  * Locks a global lock
145  */
146
147 /**
148  * @def _DBUS_UNLOCK
149  *
150  * Unlocks a global lock
151  */
152
153 /**
154  * Fixed "out of memory" error message, just to avoid
155  * making up a different string every time and wasting
156  * space.
157  */
158 const char _dbus_no_memory_message[] = "Not enough memory";
159
160 /**
161  * Prints a warning message to stderr.
162  *
163  * @param format printf-style format string.
164  */
165 void
166 _dbus_warn (const char *format,
167             ...)
168 {
169   /* FIXME not portable enough? */
170   va_list args;
171
172   va_start (args, format);
173   vfprintf (stderr, format, args);
174   va_end (args);
175 }
176
177 static dbus_bool_t verbose_initted = FALSE;
178
179 /**
180  * Prints a warning message to stderr
181  * if the user has enabled verbose mode.
182  * This is the real function implementation,
183  * use _dbus_verbose() macro in code.
184  *
185  * @param format printf-style format string.
186  */
187 void
188 _dbus_verbose_real (const char *format,
189                     ...)
190 {
191   va_list args;
192   static dbus_bool_t verbose = TRUE;
193   static dbus_bool_t need_pid = TRUE;
194   
195   /* things are written a bit oddly here so that
196    * in the non-verbose case we just have the one
197    * conditional and return immediately.
198    */
199   if (!verbose)
200     return;
201   
202   if (!verbose_initted)
203     {
204       verbose = _dbus_getenv ("DBUS_VERBOSE") != NULL;
205       verbose_initted = TRUE;
206       if (!verbose)
207         return;
208     }
209
210   if (need_pid)
211     {
212       int len;
213       
214       fprintf (stderr, "%lu: ", _dbus_getpid ());
215
216       len = strlen (format);
217       if (format[len-1] == '\n')
218         need_pid = TRUE;
219       else
220         need_pid = FALSE;
221     }
222   
223   va_start (args, format);
224   vfprintf (stderr, format, args);
225   va_end (args);
226 }
227
228 /**
229  * Reinitializes the verbose logging code, used
230  * as a hack in dbus-spawn.c so that a child
231  * process re-reads its pid
232  *
233  */
234 void
235 _dbus_verbose_reset_real (void)
236 {
237   verbose_initted = FALSE;
238 }
239
240 /**
241  * Duplicates a string. Result must be freed with
242  * dbus_free(). Returns #NULL if memory allocation fails.
243  * If the string to be duplicated is #NULL, returns #NULL.
244  * 
245  * @param str string to duplicate.
246  * @returns newly-allocated copy.
247  */
248 char*
249 _dbus_strdup (const char *str)
250 {
251   size_t len;
252   char *copy;
253   
254   if (str == NULL)
255     return NULL;
256   
257   len = strlen (str);
258
259   copy = dbus_malloc (len + 1);
260   if (copy == NULL)
261     return NULL;
262
263   memcpy (copy, str, len + 1);
264   
265   return copy;
266 }
267
268 /**
269  * Duplicates a block of memory. Returns
270  * #NULL on failure.
271  *
272  * @param mem memory to copy
273  * @param n_bytes number of bytes to copy
274  * @returns the copy
275  */
276 void*
277 _dbus_memdup (const void  *mem,
278               size_t       n_bytes)
279 {
280   void *copy;
281
282   copy = dbus_malloc (n_bytes);
283   if (copy == NULL)
284     return NULL;
285
286   memcpy (copy, mem, n_bytes);
287   
288   return copy;
289 }
290
291 /**
292  * Duplicates a string array. Result may be freed with
293  * dbus_free_string_array(). Returns #NULL if memory allocation fails.
294  * If the array to be duplicated is #NULL, returns #NULL.
295  * 
296  * @param array array to duplicate.
297  * @returns newly-allocated copy.
298  */
299 char**
300 _dbus_dup_string_array (const char **array)
301 {
302   int len;
303   int i;
304   char **copy;
305   
306   if (array == NULL)
307     return NULL;
308
309   for (len = 0; array[len] != NULL; ++len)
310     ;
311
312   copy = dbus_new0 (char*, len + 1);
313   if (copy == NULL)
314     return NULL;
315
316   i = 0;
317   while (i < len)
318     {
319       copy[i] = _dbus_strdup (array[i]);
320       if (copy[i] == NULL)
321         {
322           dbus_free_string_array (copy);
323           return NULL;
324         }
325
326       ++i;
327     }
328
329   return copy;
330 }
331
332 /**
333  * Checks whether a string array contains the given string.
334  * 
335  * @param array array to search.
336  * @param str string to look for
337  * @returns #TRUE if array contains string
338  */
339 dbus_bool_t
340 _dbus_string_array_contains (const char **array,
341                              const char  *str)
342 {
343   int i;
344
345   i = 0;
346   while (array[i] != NULL)
347     {
348       if (strcmp (array[i], str) == 0)
349         return TRUE;
350       ++i;
351     }
352
353   return FALSE;
354 }
355
356 /**
357  * Returns a string describing the given type.
358  *
359  * @param type the type to describe
360  * @returns a constant string describing the type
361  */
362 const char *
363 _dbus_type_to_string (int type)
364 {
365   switch (type)
366     {
367     case DBUS_TYPE_INVALID:
368       return "invalid";
369     case DBUS_TYPE_NIL:
370       return "nil";
371     case DBUS_TYPE_BOOLEAN:
372       return "boolean";
373     case DBUS_TYPE_INT32:
374       return "int32";
375     case DBUS_TYPE_UINT32:
376       return "uint32";
377     case DBUS_TYPE_DOUBLE:
378       return "double";
379     case DBUS_TYPE_STRING:
380       return "string";
381     case DBUS_TYPE_NAMED:
382       return "named";
383     case DBUS_TYPE_ARRAY:
384       return "array";
385     case DBUS_TYPE_DICT:
386       return "dict";
387     default:
388       return "unknown";
389     }
390 }
391
392 /**
393  * Returns a string describing the given name.
394  *
395  * @param header_field the field to describe
396  * @returns a constant string describing the field
397  */
398 const char *
399 _dbus_header_field_to_string (int header_field)
400 {
401   switch (header_field)
402     {
403     case DBUS_HEADER_FIELD_INVALID:
404       return "invalid";
405     case DBUS_HEADER_FIELD_PATH:
406       return "path";
407     case DBUS_HEADER_FIELD_INTERFACE:
408       return "interface";
409     case DBUS_HEADER_FIELD_MEMBER:
410       return "member";
411     case DBUS_HEADER_FIELD_ERROR_NAME:
412       return "error-name";
413     case DBUS_HEADER_FIELD_REPLY_SERIAL:
414       return "reply-serial";
415     case DBUS_HEADER_FIELD_SERVICE:
416       return "service";
417     case DBUS_HEADER_FIELD_SENDER_SERVICE:
418       return "sender-service";
419     default:
420       return "unknown";
421     }
422 }
423
424 #ifndef DBUS_DISABLE_CHECKS
425 /** String used in _dbus_return_if_fail macro */
426 const char _dbus_return_if_fail_warning_format[] =
427 "Arguments to %s() were incorrect, assertion \"%s\" failed in file %s line %d.\n"
428 "This is normally a bug in some application using the D-BUS library.\n";
429 #endif
430
431 #ifndef DBUS_DISABLE_ASSERT
432 /**
433  * Internals of _dbus_assert(); it's a function
434  * rather than a macro with the inline code so
435  * that the assertion failure blocks don't show up
436  * in test suite coverage, and to shrink code size.
437  *
438  * @param condition TRUE if assertion succeeded
439  * @param condition_text condition as a string
440  * @param file file the assertion is in
441  * @param line line the assertion is in
442  */
443 void
444 _dbus_real_assert (dbus_bool_t  condition,
445                    const char  *condition_text,
446                    const char  *file,
447                    int          line)
448 {
449   if (!condition)
450     {
451       _dbus_warn ("Assertion failed \"%s\" file \"%s\" line %d process %lu\n",
452                   condition_text, file, line, _dbus_getpid ());
453       _dbus_abort ();
454     }
455 }
456
457 /**
458  * Internals of _dbus_assert_not_reached(); it's a function
459  * rather than a macro with the inline code so
460  * that the assertion failure blocks don't show up
461  * in test suite coverage, and to shrink code size.
462  *
463  * @param explanation what was reached that shouldn't have been
464  * @param file file the assertion is in
465  * @param line line the assertion is in
466  */
467 void
468 _dbus_real_assert_not_reached (const char *explanation,
469                                const char *file,
470                                int         line)
471 {
472   _dbus_warn ("File \"%s\" line %d process %lu should not have been reached: %s\n",
473               file, line, _dbus_getpid (), explanation);
474   _dbus_abort ();
475 }
476 #endif /* DBUS_DISABLE_ASSERT */
477   
478 #ifdef DBUS_BUILD_TESTS
479 static dbus_bool_t
480 run_failing_each_malloc (int                    n_mallocs,
481                          const char            *description,
482                          DBusTestMemoryFunction func,
483                          void                  *data)
484 {
485   n_mallocs += 10; /* fudge factor to ensure reallocs etc. are covered */
486   
487   while (n_mallocs >= 0)
488     {      
489       _dbus_set_fail_alloc_counter (n_mallocs);
490
491       _dbus_verbose ("\n===\n%s: (will fail malloc %d with %d failures)\n===\n",
492                      description, n_mallocs,
493                      _dbus_get_fail_alloc_failures ());
494
495       if (!(* func) (data))
496         return FALSE;
497       
498       n_mallocs -= 1;
499     }
500
501   _dbus_set_fail_alloc_counter (_DBUS_INT_MAX);
502
503   return TRUE;
504 }                        
505
506 /**
507  * Tests how well the given function responds to out-of-memory
508  * situations. Calls the function repeatedly, failing a different
509  * call to malloc() each time. If the function ever returns #FALSE,
510  * the test fails. The function should return #TRUE whenever something
511  * valid (such as returning an error, or succeeding) occurs, and #FALSE
512  * if it gets confused in some way.
513  *
514  * @param description description of the test used in verbose output
515  * @param func function to call
516  * @param data data to pass to function
517  * @returns #TRUE if the function never returns FALSE
518  */
519 dbus_bool_t
520 _dbus_test_oom_handling (const char             *description,
521                          DBusTestMemoryFunction  func,
522                          void                   *data)
523 {
524   int approx_mallocs;
525
526   /* Run once to see about how many mallocs are involved */
527   
528   _dbus_set_fail_alloc_counter (_DBUS_INT_MAX);
529
530   _dbus_verbose ("Running once to count mallocs\n");
531   
532   if (!(* func) (data))
533     return FALSE;
534   
535   approx_mallocs = _DBUS_INT_MAX - _dbus_get_fail_alloc_counter ();
536
537   _dbus_verbose ("\n=================\n%s: about %d mallocs total\n=================\n",
538                  description, approx_mallocs);
539
540   _dbus_set_fail_alloc_failures (1);
541   if (!run_failing_each_malloc (approx_mallocs, description, func, data))
542     return FALSE;
543
544   _dbus_set_fail_alloc_failures (2);
545   if (!run_failing_each_malloc (approx_mallocs, description, func, data))
546     return FALSE;
547   
548   _dbus_set_fail_alloc_failures (3);
549   if (!run_failing_each_malloc (approx_mallocs, description, func, data))
550     return FALSE;
551
552   _dbus_set_fail_alloc_failures (4);
553   if (!run_failing_each_malloc (approx_mallocs, description, func, data))
554     return FALSE;
555   
556   _dbus_verbose ("\n=================\n%s: all iterations passed\n=================\n",
557                  description);
558
559   return TRUE;
560 }
561 #endif /* DBUS_BUILD_TESTS */
562
563 /** @} */