1 /* PSPP - computes sample statistics.
2 Copyright (C) 1997-9, 2000 Free Software Foundation, Inc.
3 Written by Ben Pfaff <blp@gnu.org>.
5 This program is free software; you can redistribute it and/or
6 modify it under the terms of the GNU General Public License as
7 published by the Free Software Foundation; either version 2 of the
8 License, or (at your option) any later version.
10 This program 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.
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
30 /* sprintf() wrapper functions for convenience. */
34 spprintf (char *buf, const char *format,...)
41 va_start (args, format);
45 vsprintf (buf, format, args);
51 return strchr (buf, 0);
54 #endif /* !__GNUC__ */
56 #if !__GNUC__ && !HAVE_GOOD_SPRINTF
58 nsprintf (char *buf, const char *format,...)
62 va_start (args, format);
63 vsprintf (buf, format, args);
70 nvsprintf (char *buf, const char *format, va_list args)
72 vsprintf (buf, format, args);
75 #endif /* Not GNU C and not good sprintf(). */
77 /* Reverses the order of NBYTES bytes at address P, thus converting
78 between little- and big-endian byte orders. */
80 mm_reverse (void *p, size_t nbytes)
82 unsigned char *h = p, *t = &h[nbytes - 1];
94 /* Finds the last NEEDLE of length NEEDLE_LEN in a HAYSTACK of length
95 HAYSTACK_LEN. Returns a pointer to the needle found. */
97 mm_find_reverse (const char *haystack, size_t haystack_len,
98 const char *needle, size_t needle_len)
101 for (i = haystack_len - needle_len; i >= 0; i--)
102 if (!memcmp (needle, &haystack[i], needle_len))
103 return (char *) &haystack[i];
107 /* Compares the SIZE bytes in A to those in B, disregarding case,
108 and returns a strcmp()-type result. */
110 mm_case_compare (const void *a_, const void *b_, size_t size)
112 const unsigned char *a = a_;
113 const unsigned char *b = b_;
117 unsigned char ac = toupper (*a++);
118 unsigned char bc = toupper (*b++);
121 return ac > bc ? 1 : -1;
127 /* Compares A of length A_LEN to B of length B_LEN. The shorter
128 string is considered to be padded with spaces to the length of
131 st_compare_pad (const char *a, size_t a_len, const char *b, size_t b_len)
136 min_len = a_len < b_len ? a_len : b_len;
137 result = memcmp (a, b, min_len);
146 for (idx = min_len; idx < b_len; idx++)
148 return ' ' > b[idx] ? 1 : -1;
152 for (idx = min_len; idx < a_len; idx++)
154 return a[idx] > ' ' ? 1 : -1;
160 /* Copies SRC to DEST, truncating to N characters or right-padding
161 with spaces to N characters as necessary. Does not append a null
162 character. SRC must be null-terminated. */
164 st_bare_pad_copy (char *dest, const char *src, size_t n)
170 memcpy (dest, src, n);
173 memcpy (dest, src, len);
174 memset (&dest[len], ' ', n - len);
178 /* Copies SRC to DEST, truncating SRC to N characters or right-padding
179 with spaces to N characters if necessary. Does not append a null
180 character. SRC must be LEN characters long but does not need to be
183 st_bare_pad_len_copy (char *dest, const char *src, size_t n, size_t len)
186 memmove (dest, src, n);
189 memmove (dest, src, len);
190 memset (&dest[len], ' ', n - len);
194 /* Copies SRC to DEST, truncating SRC to N-1 characters or
195 right-padding with spaces to N-1 characters if necessary. Always
196 appends a null character. */
198 st_pad_copy (char *dest, const char *src, size_t n)
205 else if (len < n - 1)
207 memcpy (dest, src, len);
208 memset (&dest[len], ' ', n - 1 - len);
213 memcpy (dest, src, n - 1);
218 /* Copies SRC to DST, truncating DST to N-1 characters if
219 necessary. Always appends a null character. */
221 st_trim_copy (char *dst, const char *src, size_t n)
223 size_t len = strlen (src);
226 memcpy (dst, src, len + 1);
229 memcpy (dst, src, n - 1);
235 /* Converts each character in S to uppercase. */
237 st_uppercase (char *s)
239 for (; *s != '\0'; s++)
240 *s = toupper ((unsigned char) *s);
243 /* Initializes ST with initial contents S. */
245 ds_create (struct string *st, const char *s)
247 st->length = strlen (s);
248 st->capacity = 8 + st->length * 2;
249 st->string = xmalloc (st->capacity + 1);
250 strcpy (st->string, s);
253 /* Initializes ST, making room for at least CAPACITY characters. */
255 ds_init (struct string *st, size_t capacity)
259 st->capacity = capacity;
262 st->string = xmalloc (st->capacity + 1);
265 /* Replaces the contents of ST with STRING. STRING may overlap with
268 ds_replace (struct string *st, const char *string)
270 size_t new_length = strlen (string);
271 if (new_length > st->capacity)
273 /* The new length is longer than the allocated length, so
274 there can be no overlap. */
276 ds_concat (st, string, new_length);
280 /* Overlap is possible, but the new string will fit in the
281 allocated space, so we can just copy data. */
282 st->length = new_length;
283 memmove (st->string, string, st->length);
289 ds_destroy (struct string *st)
294 /* Truncates ST to zero length. */
296 ds_clear (struct string *st)
301 /* Pad ST on the right with copies of PAD until ST is at least
302 LENGTH characters in size. If ST is initially LENGTH
303 characters or longer, this is a no-op. */
305 ds_rpad (struct string *st, size_t length, char pad)
308 if (st->length < length)
310 if (st->capacity < length)
311 ds_extend (st, length);
312 memset (&st->string[st->length], pad, length - st->length);
317 /* Ensures that ST can hold at least MIN_CAPACITY characters plus a null
320 ds_extend (struct string *st, size_t min_capacity)
322 if (min_capacity > st->capacity)
325 if (st->capacity < min_capacity)
326 st->capacity = min_capacity * 2;
328 st->string = xrealloc (st->string, st->capacity + 1);
332 /* Shrink ST to the minimum capacity need to contain its content. */
334 ds_shrink (struct string *st)
336 if (st->capacity != st->length)
338 st->capacity = st->length;
339 st->string = xrealloc (st->string, st->capacity + 1);
343 /* Truncates ST to at most LENGTH characters long. */
345 ds_truncate (struct string *st, size_t length)
347 if (length >= st->length)
352 /* Returns the length of ST. */
354 ds_length (const struct string *st)
359 /* Returns the allocation size of ST. */
361 ds_capacity (const struct string *st)
366 /* Returns the value of ST as a null-terminated string. */
368 ds_c_str (const struct string *st)
370 ((char *) st->string)[st->length] = '\0';
374 /* Returns the string data inside ST. */
376 ds_data (const struct string *st)
381 /* Returns a pointer to the null terminator ST.
382 This might not be an actual null character unless ds_c_str() has
383 been called since the last modification to ST. */
385 ds_end (const struct string *st)
387 return st->string + st->length;
390 /* Concatenates S onto ST. */
392 ds_puts (struct string *st, const char *s)
399 ds_extend (st, st->length + s_len);
400 strcpy (st->string + st->length, s);
404 /* Concatenates LEN characters from BUF onto ST. */
406 ds_concat (struct string *st, const char *buf, size_t len)
408 ds_extend (st, st->length + len);
409 memcpy (st->string + st->length, buf, len);
413 void ds_vprintf (struct string *st, const char *format, va_list args);
416 /* Formats FORMAT as a printf string and appends the result to ST. */
418 ds_printf (struct string *st, const char *format, ...)
422 va_start (args, format);
423 ds_vprintf(st,format,args);
428 /* Formats FORMAT as a printf string and appends the result to ST. */
430 ds_vprintf (struct string *st, const char *format, va_list args)
432 /* Fscking glibc silently changed behavior between 2.0 and 2.1.
433 Fsck fsck fsck. Before, it returned -1 on buffer overflow. Now,
434 it returns the number of characters (not bytes) that would have
439 avail = st->capacity - st->length + 1;
440 needed = vsnprintf (st->string + st->length, avail, format, args);
445 ds_extend (st, st->length + needed);
447 vsprintf (st->string + st->length, format, args);
452 ds_extend (st, (st->capacity + 1) * 2);
453 avail = st->capacity - st->length + 1;
455 needed = vsnprintf (st->string + st->length, avail, format, args);
459 st->length += needed;
462 /* Appends character CH to ST. */
464 ds_putc (struct string *st, int ch)
466 if (st->length == st->capacity)
467 ds_extend (st, st->length + 1);
468 st->string[st->length++] = ch;
471 /* Appends to ST a newline-terminated line read from STREAM.
472 Newline is the last character of ST on return, unless an I/O error
473 or end of file is encountered after reading some characters.
474 Returns 1 if a line is successfully read, or 0 if no characters at
475 all were read before an I/O error or end of file was
478 ds_gets (struct string *st, FILE *stream)
498 /* Reads a line from STREAM into ST, then preprocesses as follows:
500 - Splices lines terminated with `\'.
502 - Deletes comments introduced by `#' outside of single or double
505 - Trailing whitespace will be deleted.
507 Increments cust_ln as appropriate.
509 Returns nonzero only if a line was successfully read. */
511 ds_get_config_line (FILE *stream, struct string *st, struct file_locator *where)
513 /* Read the first line. */
515 where->line_number++;
516 if (!ds_gets (st, stream))
519 /* Read additional lines, if any. */
522 /* Remove trailing whitespace. */
524 char *s = ds_c_str (st);
525 size_t len = ds_length (st);
527 while (len > 0 && isspace ((unsigned char) s[len - 1]))
529 ds_truncate (st, len);
532 /* Check for trailing \. Remove if found, bail otherwise. */
533 if (ds_length (st) == 0 || ds_c_str (st)[ds_length (st) - 1] != '\\')
535 ds_truncate (st, ds_length (st) - 1);
537 /* Append another line and go around again. */
539 int success = ds_gets (st, stream);
540 where->line_number++;
546 /* Find a comment and remove. */
551 for (cp = ds_c_str (st); *cp; cp++)
556 else if (*cp == '\\')
559 else if (*cp == '\'' || *cp == '"')
563 ds_truncate (st, cp - ds_c_str (st));
571 /* Lengthed strings. */
573 /* Creates a new lengthed string LS with contents as a copy of
576 ls_create (struct fixed_string *ls, const char *s)
578 ls->length = strlen (s);
579 ls->string = xmalloc (ls->length + 1);
580 memcpy (ls->string, s, ls->length + 1);
583 /* Creates a new lengthed string LS with contents as a copy of
584 BUFFER with length LEN. */
586 ls_create_buffer (struct fixed_string *ls,
587 const char *buffer, size_t len)
590 ls->string = xmalloc (len + 1);
591 memcpy (ls->string, buffer, len);
592 ls->string[len] = '\0';
595 /* Sets the fields of LS to the specified values. */
597 ls_init (struct fixed_string *ls, const char *string, size_t length)
599 ls->string = (char *) string;
603 /* Copies the fields of SRC to DST. */
605 ls_shallow_copy (struct fixed_string *dst, const struct fixed_string *src)
610 /* Frees the memory backing LS. */
612 ls_destroy (struct fixed_string *ls)
617 /* Sets LS to a null pointer value. */
619 ls_null (struct fixed_string *ls)
624 /* Returns nonzero only if LS has a null pointer value. */
626 ls_null_p (const struct fixed_string *ls)
628 return ls->string == NULL;
631 /* Returns nonzero only if LS is a null pointer or has length 0. */
633 ls_empty_p (const struct fixed_string *ls)
635 return ls->string == NULL || ls->length == 0;
638 /* Returns the length of LS, which must not be null. */
640 ls_length (const struct fixed_string *ls)
645 /* Returns a pointer to the character string in LS. */
647 ls_c_str (const struct fixed_string *ls)
649 return (char *) ls->string;
652 /* Returns a pointer to the null terminator of the character string in
655 ls_end (const struct fixed_string *ls)
657 return (char *) (ls->string + ls->length);