home *** CD-ROM | disk | FTP | other *** search
- #include "efence.h"
- #include <stdlib.h>
- #include <unistd.h>
- #include <stdarg.h>
- #include <string.h>
- #include <signal.h>
-
- /*
- * These routines do their printing without using stdio. Stdio can't
- * be used because it calls malloc(). Internal routines of a malloc()
- * debugger should not re-enter malloc(), so stdio is out.
- */
-
- /*
- * NUMBER_BUFFER_SIZE is the longest character string that could be needed
- * to represent an unsigned integer. Assuming unsigned integers might be as
- * large as 64 bits, and we might print in base 2, let's set it to 64 bytes.
- */
- #define NUMBER_BUFFER_SIZE 64
-
- static void
- printNumber(unsigned int number, unsigned int base)
- {
- char buffer[NUMBER_BUFFER_SIZE];
- char * s = &buffer[NUMBER_BUFFER_SIZE];
- int size;
-
- do {
- unsigned int digit;
-
- if ( --s == buffer )
- EF_Abort("Internal error printing number.");
-
- digit = number % base;
-
- if ( digit < 10 )
- *s = '0' + digit;
- else
- *s = 'a' + digit - 10;
-
- } while ( (number /= base) > 0 );
-
- size = &buffer[NUMBER_BUFFER_SIZE] - s;
-
- if ( size > 0 )
- write(2, s, size);
- }
-
- static void
- vprint(const char * pattern, va_list args)
- {
- static const char bad_pattern[] =
- "\nBad pattern specifier %%%c in EF_Print().\n";
- const char * s = pattern;
- char c;
-
- while ( (c = *s++) != '\0' ) {
- if ( c == '%' ) {
- c = *s++;
- switch ( c ) {
- case '%':
- (void) write(2, &c, 1);
- break;
- case 's':
- {
- const char * string;
- size_t length;
-
- string = va_arg(args, char *);
- length = strlen(string);
-
- (void) write(2, string, length);
- }
- break;
- case 'd':
- {
- int n = va_arg(args, int);
-
- if ( n < 0 ) {
- char c = '-';
- write(2, &c, 1);
- n = -n;
- }
- printNumber(n, 10);
- }
- break;
- case 'x':
- printNumber(va_arg(args, u_int), 0x10);
- break;
- case 'c':
- {
- char c = va_arg(args, char);
-
- (void) write(2, &c, 1);
- }
- break;
- default:
- {
- EF_Print(bad_pattern, c);
- }
-
- }
- }
- else
- (void) write(2, &c, 1);
- }
- }
-
- void
- EF_Abort(const char * pattern, ...)
- {
- va_list args;
-
- va_start(args, pattern);
-
- EF_Print("\nElectricFence Aborting: ");
- vprint(pattern, args);
- EF_Print("\n");
-
- va_end(args);
-
- /*
- * I use kill(getpid(), SIGILL) instead of abort() because some
- * mis-guided implementations of abort() flush stdio, which can
- * cause malloc() or free() to be called.
- */
- kill(getpid(), SIGILL);
- /* Just in case something handles SIGILL and returns, exit here. */
- _exit(-1);
- }
-
- void
- EF_Exit(const char * pattern, ...)
- {
- va_list args;
-
- va_start(args, pattern);
-
- EF_Print("\nElectricFence Exiting: ");
- vprint(pattern, args);
- EF_Print("\n");
-
- va_end(args);
-
- /*
- * I use _exit() because the regular exit() flushes stdio,
- * which may cause malloc() or free() to be called.
- */
- _exit(-1);
- }
-
- void
- EF_Print(const char * pattern, ...)
- {
- va_list args;
-
- va_start(args, pattern);
- vprint(pattern, args);
- va_end(args);
- }
-