2003-10-31 09:31:56 -05:00
|
|
|
|
/* Various utility functions.
|
2009-09-04 03:13:47 -04:00
|
|
|
|
Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2002, 2003, 2004,
|
|
|
|
|
2005, 2006, 2007, 2008, 2009 Free Software Foundation, Inc.
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2001-05-27 15:35:15 -04:00
|
|
|
|
This file is part of GNU Wget.
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2001-05-27 15:35:15 -04:00
|
|
|
|
GNU Wget is free software; you can redistribute it and/or modify
|
1999-12-02 02:42:23 -05:00
|
|
|
|
it under the terms of the GNU General Public License as published by
|
2007-07-10 01:53:22 -04:00
|
|
|
|
the Free Software Foundation; either version 3 of the License, or
|
1999-12-02 02:42:23 -05:00
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
2001-05-27 15:35:15 -04:00
|
|
|
|
GNU Wget is distributed in the hope that it will be useful,
|
1999-12-02 02:42:23 -05:00
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
2007-07-10 01:53:22 -04:00
|
|
|
|
along with Wget. If not, see <http://www.gnu.org/licenses/>.
|
2002-05-17 22:16:36 -04:00
|
|
|
|
|
2007-11-28 03:05:33 -05:00
|
|
|
|
Additional permission under GNU GPL version 3 section 7
|
|
|
|
|
|
|
|
|
|
If you modify this program, or any covered work, by linking or
|
|
|
|
|
combining it with the OpenSSL project's OpenSSL library (or a
|
|
|
|
|
modified version of that library), containing parts covered by the
|
|
|
|
|
terms of the OpenSSL or SSLeay licenses, the Free Software Foundation
|
|
|
|
|
grants you additional permission to convey the resulting work.
|
|
|
|
|
Corresponding Source for a non-source form of such a combination
|
|
|
|
|
shall include the source code for the parts of OpenSSL used as well
|
|
|
|
|
as that of the covered work. */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2007-10-18 23:50:40 -04:00
|
|
|
|
#include "wget.h"
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
#include <stdlib.h>
|
2005-06-19 18:34:58 -04:00
|
|
|
|
#include <string.h>
|
|
|
|
|
#include <time.h>
|
|
|
|
|
#ifdef HAVE_SYS_TIME_H
|
|
|
|
|
# include <sys/time.h>
|
|
|
|
|
#endif
|
1999-12-02 02:42:23 -05:00
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
|
|
|
# include <unistd.h>
|
|
|
|
|
#endif
|
2000-11-19 15:50:10 -05:00
|
|
|
|
#ifdef HAVE_MMAP
|
|
|
|
|
# include <sys/mman.h>
|
|
|
|
|
#endif
|
2007-10-06 15:44:17 -04:00
|
|
|
|
#ifdef HAVE_PROCESS_H
|
|
|
|
|
# include <process.h> /* getpid() */
|
|
|
|
|
#endif
|
1999-12-02 02:42:23 -05:00
|
|
|
|
#ifdef HAVE_UTIME_H
|
|
|
|
|
# include <utime.h>
|
|
|
|
|
#endif
|
|
|
|
|
#ifdef HAVE_SYS_UTIME_H
|
|
|
|
|
# include <sys/utime.h>
|
|
|
|
|
#endif
|
|
|
|
|
#include <errno.h>
|
2000-11-19 15:50:10 -05:00
|
|
|
|
#include <fcntl.h>
|
2000-11-10 13:01:35 -05:00
|
|
|
|
#include <assert.h>
|
2005-06-19 18:34:58 -04:00
|
|
|
|
#include <stdarg.h>
|
2005-06-29 17:45:44 -04:00
|
|
|
|
#include <locale.h>
|
2001-12-06 05:45:27 -05:00
|
|
|
|
|
|
|
|
|
/* For TIOCGWINSZ and friends: */
|
2001-11-22 23:59:52 -05:00
|
|
|
|
#ifdef HAVE_SYS_IOCTL_H
|
|
|
|
|
# include <sys/ioctl.h>
|
|
|
|
|
#endif
|
2001-12-06 05:45:27 -05:00
|
|
|
|
#ifdef HAVE_TERMIOS_H
|
|
|
|
|
# include <termios.h>
|
|
|
|
|
#endif
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2005-07-06 17:06:29 -04:00
|
|
|
|
/* Needed for Unix version of run_with_timeout. */
|
2005-06-19 19:03:27 -04:00
|
|
|
|
#include <signal.h>
|
2005-07-06 17:06:29 -04:00
|
|
|
|
#include <setjmp.h>
|
2003-10-23 08:15:39 -04:00
|
|
|
|
|
|
|
|
|
#ifndef HAVE_SIGSETJMP
|
2002-04-13 18:18:28 -04:00
|
|
|
|
/* If sigsetjmp is a macro, configure won't pick it up. */
|
2003-10-23 08:15:39 -04:00
|
|
|
|
# ifdef sigsetjmp
|
|
|
|
|
# define HAVE_SIGSETJMP
|
|
|
|
|
# endif
|
2002-04-13 18:18:28 -04:00
|
|
|
|
#endif
|
2003-10-23 08:15:39 -04:00
|
|
|
|
|
2005-07-08 03:33:05 -04:00
|
|
|
|
#if defined HAVE_SIGSETJMP || defined HAVE_SIGBLOCK
|
2005-06-21 21:01:12 -04:00
|
|
|
|
# define USE_SIGNAL_TIMEOUT
|
2002-04-13 18:18:28 -04:00
|
|
|
|
#endif
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
#include "utils.h"
|
2000-11-19 15:50:10 -05:00
|
|
|
|
#include "hash.h"
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2008-04-22 17:48:36 -04:00
|
|
|
|
#ifdef __VMS
|
|
|
|
|
#include "vms.h"
|
|
|
|
|
#endif /* def __VMS */
|
|
|
|
|
|
2006-03-15 09:55:29 -05:00
|
|
|
|
#ifdef TESTING
|
|
|
|
|
#include "test.h"
|
2009-09-21 23:39:44 -04:00
|
|
|
|
#endif
|
2006-03-15 09:55:29 -05:00
|
|
|
|
|
2008-04-19 15:13:38 -04:00
|
|
|
|
static void
|
|
|
|
|
memfatal (const char *context, long attempted_size)
|
|
|
|
|
{
|
|
|
|
|
/* Make sure we don't try to store part of the log line, and thus
|
|
|
|
|
call malloc. */
|
|
|
|
|
log_set_save_context (false);
|
|
|
|
|
|
|
|
|
|
/* We have different log outputs in different situations:
|
|
|
|
|
1) output without bytes information
|
|
|
|
|
2) output with bytes information */
|
|
|
|
|
if (attempted_size == UNKNOWN_ATTEMPTED_SIZE)
|
|
|
|
|
{
|
|
|
|
|
logprintf (LOG_ALWAYS,
|
|
|
|
|
_("%s: %s: Failed to allocate enough memory; memory exhausted.\n"),
|
|
|
|
|
exec_name, context);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
logprintf (LOG_ALWAYS,
|
|
|
|
|
_("%s: %s: Failed to allocate %ld bytes; memory exhausted.\n"),
|
|
|
|
|
exec_name, context, attempted_size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
exit (1);
|
|
|
|
|
}
|
|
|
|
|
|
2008-04-22 17:48:36 -04:00
|
|
|
|
/* Character property table for (re-)escaping VMS ODS5 extended file
|
|
|
|
|
names. Note that this table ignores Unicode.
|
|
|
|
|
|
|
|
|
|
ODS2 valid characters: 0-9 A-Z a-z $ - _ ~
|
|
|
|
|
|
|
|
|
|
ODS5 Invalid characters:
|
|
|
|
|
C0 control codes (0x00 to 0x1F inclusive)
|
|
|
|
|
Asterisk (*)
|
|
|
|
|
Question mark (?)
|
|
|
|
|
|
|
|
|
|
ODS5 Invalid characters only in VMS V7.2 (which no one runs, right?):
|
|
|
|
|
Double quotation marks (")
|
|
|
|
|
Backslash (\)
|
|
|
|
|
Colon (:)
|
|
|
|
|
Left angle bracket (<)
|
|
|
|
|
Right angle bracket (>)
|
|
|
|
|
Slash (/)
|
|
|
|
|
Vertical bar (|)
|
|
|
|
|
|
|
|
|
|
Characters escaped by "^":
|
2010-01-09 18:24:02 -05:00
|
|
|
|
SP ! " # % & ' ( ) + , . : ; =
|
|
|
|
|
@ [ \ ] ^ ` { | } ~
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
Either "^_" or "^ " is accepted as a space. Period (.) is a special
|
|
|
|
|
case. Note that un-escaped < and > can also confuse a directory
|
|
|
|
|
spec.
|
|
|
|
|
|
|
|
|
|
Characters put out as ^xx:
|
|
|
|
|
7F (DEL)
|
|
|
|
|
80-9F (C1 control characters)
|
|
|
|
|
A0 (nonbreaking space)
|
|
|
|
|
FF (Latin small letter y diaeresis)
|
|
|
|
|
|
|
|
|
|
Other cases:
|
|
|
|
|
Unicode: "^Uxxxx", where "xxxx" is four hex digits.
|
|
|
|
|
|
|
|
|
|
Property table values:
|
|
|
|
|
Normal escape: 1
|
|
|
|
|
Space: 2
|
|
|
|
|
Dot: 4
|
|
|
|
|
Hex-hex escape: 8
|
|
|
|
|
ODS2 normal: 16
|
|
|
|
|
ODS2 lower case: 32
|
|
|
|
|
Hex digit: 64
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
unsigned char char_prop[ 256] = {
|
|
|
|
|
|
|
|
|
|
/* NUL SOH STX ETX EOT ENQ ACK BEL BS HT LF VT FF CR SO SI */
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
|
|
|
|
|
/* DLE DC1 DC2 DC3 DC4 NAK SYN ETB CAN EM SUB ESC FS GS RS US */
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
|
|
|
|
|
/* SP ! " # $ % & ' ( ) * + , - . / */
|
2010-01-09 18:24:02 -05:00
|
|
|
|
2, 1, 1, 1, 16, 1, 1, 1, 1, 1, 0, 1, 1, 16, 4, 0,
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
/* 0 1 2 3 4 5 6 7 8 9 : ; < = > ? */
|
2010-01-09 18:24:02 -05:00
|
|
|
|
80, 80, 80, 80, 80, 80, 80, 80, 80, 80, 1, 1, 1, 1, 1, 1,
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
/* @ A B C D E F G H I J K L M N O */
|
|
|
|
|
1, 80, 80, 80, 80, 80, 80, 16, 16, 16, 16, 16, 16, 16, 16, 16,
|
|
|
|
|
|
|
|
|
|
/* P Q R S T U V W X Y Z [ \ ] ^ _ */
|
2010-01-09 18:24:02 -05:00
|
|
|
|
16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 1, 1, 1, 1, 16,
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
/* ` a b c d e f g h i j k l m n o */
|
|
|
|
|
1, 96, 96, 96, 96, 96, 96, 32, 32, 32, 32, 32, 32, 32, 32, 32,
|
|
|
|
|
|
|
|
|
|
/* p q r s t u v w x y z { | } ~ DEL */
|
2010-01-09 18:24:02 -05:00
|
|
|
|
32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 1, 1, 1, 17, 8,
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
|
|
|
|
|
8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
|
|
|
|
|
8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
|
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8
|
|
|
|
|
};
|
|
|
|
|
|
2001-11-24 22:10:34 -05:00
|
|
|
|
/* Utility function: like xstrdup(), but also lowercases S. */
|
|
|
|
|
|
|
|
|
|
char *
|
|
|
|
|
xstrdup_lower (const char *s)
|
|
|
|
|
{
|
|
|
|
|
char *copy = xstrdup (s);
|
|
|
|
|
char *p = copy;
|
|
|
|
|
for (; *p; p++)
|
2007-10-14 17:46:24 -04:00
|
|
|
|
*p = c_tolower (*p);
|
2001-11-24 22:10:34 -05:00
|
|
|
|
return copy;
|
|
|
|
|
}
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
/* Copy the string formed by two pointers (one on the beginning, other
|
|
|
|
|
on the char after the last char) to a new, malloc-ed location.
|
|
|
|
|
0-terminate it. */
|
|
|
|
|
char *
|
|
|
|
|
strdupdelim (const char *beg, const char *end)
|
|
|
|
|
{
|
2005-06-19 19:03:27 -04:00
|
|
|
|
char *res = xmalloc (end - beg + 1);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
memcpy (res, beg, end - beg);
|
|
|
|
|
res[end - beg] = '\0';
|
|
|
|
|
return res;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Parse a string containing comma-separated elements, and return a
|
|
|
|
|
vector of char pointers with the elements. Spaces following the
|
|
|
|
|
commas are ignored. */
|
|
|
|
|
char **
|
|
|
|
|
sepstring (const char *s)
|
|
|
|
|
{
|
|
|
|
|
char **res;
|
|
|
|
|
const char *p;
|
|
|
|
|
int i = 0;
|
|
|
|
|
|
|
|
|
|
if (!s || !*s)
|
|
|
|
|
return NULL;
|
|
|
|
|
res = NULL;
|
|
|
|
|
p = s;
|
|
|
|
|
while (*s)
|
|
|
|
|
{
|
|
|
|
|
if (*s == ',')
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
res = xrealloc (res, (i + 2) * sizeof (char *));
|
|
|
|
|
res[i] = strdupdelim (p, s);
|
|
|
|
|
res[++i] = NULL;
|
|
|
|
|
++s;
|
|
|
|
|
/* Skip the blanks following the ','. */
|
2007-10-14 17:46:24 -04:00
|
|
|
|
while (c_isspace (*s))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
++s;
|
|
|
|
|
p = s;
|
|
|
|
|
}
|
1999-12-02 02:42:23 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
++s;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2005-06-19 19:03:27 -04:00
|
|
|
|
res = xrealloc (res, (i + 2) * sizeof (char *));
|
1999-12-02 02:42:23 -05:00
|
|
|
|
res[i] = strdupdelim (p, s);
|
|
|
|
|
res[i + 1] = NULL;
|
|
|
|
|
return res;
|
|
|
|
|
}
|
|
|
|
|
|
2005-07-08 03:33:05 -04:00
|
|
|
|
/* Like sprintf, but prints into a string of sufficient size freshly
|
|
|
|
|
allocated with malloc, which is returned. If unable to print due
|
|
|
|
|
to invalid format, returns NULL. Inability to allocate needed
|
|
|
|
|
memory results in abort, as with xmalloc. This is in spirit
|
|
|
|
|
similar to the GNU/BSD extension asprintf, but somewhat easier to
|
|
|
|
|
use.
|
|
|
|
|
|
|
|
|
|
Internally the function either calls vasprintf or loops around
|
|
|
|
|
vsnprintf until the correct size is found. Since Wget also ships a
|
|
|
|
|
fallback implementation of vsnprintf, this should be portable. */
|
2003-11-29 13:40:01 -05:00
|
|
|
|
|
2008-04-11 15:52:57 -04:00
|
|
|
|
/* Constant is using for limits memory allocation for text buffer.
|
2009-09-21 23:39:44 -04:00
|
|
|
|
Applicable in situation when: vasprintf is not available in the system
|
2008-04-11 15:52:57 -04:00
|
|
|
|
and vsnprintf return -1 when long line is truncated (in old versions of
|
|
|
|
|
glibc and in other system where C99 doesn`t support) */
|
|
|
|
|
|
|
|
|
|
#define FMT_MAX_LENGTH 1048576
|
|
|
|
|
|
2003-11-29 13:40:01 -05:00
|
|
|
|
char *
|
|
|
|
|
aprintf (const char *fmt, ...)
|
|
|
|
|
{
|
2007-02-02 04:35:08 -05:00
|
|
|
|
#if defined HAVE_VASPRINTF && !defined DEBUG_MALLOC
|
2005-07-08 03:33:05 -04:00
|
|
|
|
/* Use vasprintf. */
|
|
|
|
|
int ret;
|
|
|
|
|
va_list args;
|
|
|
|
|
char *str;
|
|
|
|
|
va_start (args, fmt);
|
|
|
|
|
ret = vasprintf (&str, fmt, args);
|
|
|
|
|
va_end (args);
|
|
|
|
|
if (ret < 0 && errno == ENOMEM)
|
2008-04-12 16:09:43 -04:00
|
|
|
|
memfatal ("aprintf", UNKNOWN_ATTEMPTED_SIZE); /* for consistency
|
|
|
|
|
with xmalloc/xrealloc */
|
2005-07-08 03:33:05 -04:00
|
|
|
|
else if (ret < 0)
|
|
|
|
|
return NULL;
|
|
|
|
|
return str;
|
|
|
|
|
#else /* not HAVE_VASPRINTF */
|
2003-11-29 13:40:01 -05:00
|
|
|
|
|
2005-07-08 03:33:05 -04:00
|
|
|
|
/* vasprintf is unavailable. snprintf into a small buffer and
|
|
|
|
|
resize it as necessary. */
|
2003-11-29 13:40:01 -05:00
|
|
|
|
int size = 32;
|
|
|
|
|
char *str = xmalloc (size);
|
|
|
|
|
|
2005-07-08 03:36:59 -04:00
|
|
|
|
/* #### This code will infloop and eventually abort in xrealloc if
|
|
|
|
|
passed a FMT that causes snprintf to consistently return -1. */
|
|
|
|
|
|
2003-11-29 13:40:01 -05:00
|
|
|
|
while (1)
|
|
|
|
|
{
|
|
|
|
|
int n;
|
|
|
|
|
va_list args;
|
|
|
|
|
|
2005-06-19 18:34:58 -04:00
|
|
|
|
va_start (args, fmt);
|
2003-11-29 13:40:01 -05:00
|
|
|
|
n = vsnprintf (str, size, fmt, args);
|
|
|
|
|
va_end (args);
|
|
|
|
|
|
|
|
|
|
/* If the printing worked, return the string. */
|
|
|
|
|
if (n > -1 && n < size)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return str;
|
2003-11-29 13:40:01 -05:00
|
|
|
|
|
|
|
|
|
/* Else try again with a larger buffer. */
|
2007-08-02 23:38:21 -04:00
|
|
|
|
if (n > -1) /* C99 */
|
|
|
|
|
size = n + 1; /* precisely what is needed */
|
2008-04-12 16:09:43 -04:00
|
|
|
|
else if (size >= FMT_MAX_LENGTH) /* We have a huge buffer, */
|
|
|
|
|
{ /* maybe we have some wrong
|
|
|
|
|
format string? */
|
2009-09-21 23:39:44 -04:00
|
|
|
|
logprintf (LOG_ALWAYS,
|
2008-04-12 16:09:43 -04:00
|
|
|
|
_("%s: aprintf: text buffer is too big (%ld bytes), "
|
|
|
|
|
"aborting.\n"),
|
|
|
|
|
exec_name, size); /* printout a log message */
|
|
|
|
|
abort (); /* and abort... */
|
|
|
|
|
}
|
2003-11-29 13:40:01 -05:00
|
|
|
|
else
|
2008-04-11 15:52:57 -04:00
|
|
|
|
{
|
2008-04-12 16:09:43 -04:00
|
|
|
|
/* else, we continue to grow our
|
|
|
|
|
* buffer: Twice the old size. */
|
|
|
|
|
size <<= 1;
|
2008-04-11 15:52:57 -04:00
|
|
|
|
}
|
2003-11-29 13:40:01 -05:00
|
|
|
|
str = xrealloc (str, size);
|
|
|
|
|
}
|
2005-07-08 03:33:05 -04:00
|
|
|
|
#endif /* not HAVE_VASPRINTF */
|
2003-11-29 13:40:01 -05:00
|
|
|
|
}
|
2005-03-19 12:23:32 -05:00
|
|
|
|
|
|
|
|
|
/* Concatenate the NULL-terminated list of string arguments into
|
|
|
|
|
freshly allocated space. */
|
|
|
|
|
|
|
|
|
|
char *
|
|
|
|
|
concat_strings (const char *str0, ...)
|
|
|
|
|
{
|
|
|
|
|
va_list args;
|
2007-08-02 23:38:21 -04:00
|
|
|
|
int saved_lengths[5]; /* inspired by Apache's apr_pstrcat */
|
2005-03-19 12:23:32 -05:00
|
|
|
|
char *ret, *p;
|
|
|
|
|
|
|
|
|
|
const char *next_str;
|
|
|
|
|
int total_length = 0;
|
2008-05-31 01:42:36 -04:00
|
|
|
|
size_t argcount;
|
2005-03-19 12:23:32 -05:00
|
|
|
|
|
|
|
|
|
/* Calculate the length of and allocate the resulting string. */
|
|
|
|
|
|
|
|
|
|
argcount = 0;
|
2005-06-19 18:34:58 -04:00
|
|
|
|
va_start (args, str0);
|
2005-03-19 12:23:32 -05:00
|
|
|
|
for (next_str = str0; next_str != NULL; next_str = va_arg (args, char *))
|
|
|
|
|
{
|
|
|
|
|
int len = strlen (next_str);
|
|
|
|
|
if (argcount < countof (saved_lengths))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
saved_lengths[argcount++] = len;
|
2005-03-19 12:23:32 -05:00
|
|
|
|
total_length += len;
|
|
|
|
|
}
|
|
|
|
|
va_end (args);
|
|
|
|
|
p = ret = xmalloc (total_length + 1);
|
|
|
|
|
|
|
|
|
|
/* Copy the strings into the allocated space. */
|
|
|
|
|
|
|
|
|
|
argcount = 0;
|
2005-06-19 18:34:58 -04:00
|
|
|
|
va_start (args, str0);
|
2005-03-19 12:23:32 -05:00
|
|
|
|
for (next_str = str0; next_str != NULL; next_str = va_arg (args, char *))
|
|
|
|
|
{
|
|
|
|
|
int len;
|
|
|
|
|
if (argcount < countof (saved_lengths))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
len = saved_lengths[argcount++];
|
2005-03-19 12:23:32 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
len = strlen (next_str);
|
2005-03-19 12:23:32 -05:00
|
|
|
|
memcpy (p, next_str, len);
|
|
|
|
|
p += len;
|
|
|
|
|
}
|
|
|
|
|
va_end (args);
|
|
|
|
|
*p = '\0';
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
2003-11-29 13:40:01 -05:00
|
|
|
|
|
2006-08-08 10:32:53 -04:00
|
|
|
|
/* Format the provided time according to the specified format. The
|
|
|
|
|
format is a string with format elements supported by strftime. */
|
|
|
|
|
|
|
|
|
|
static char *
|
|
|
|
|
fmttime (time_t t, const char *fmt)
|
|
|
|
|
{
|
|
|
|
|
static char output[32];
|
|
|
|
|
struct tm *tm = localtime(&t);
|
|
|
|
|
if (!tm)
|
|
|
|
|
abort ();
|
|
|
|
|
if (!strftime(output, sizeof(output), fmt, tm))
|
|
|
|
|
abort ();
|
|
|
|
|
return output;
|
|
|
|
|
}
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
/* Return pointer to a static char[] buffer in which zero-terminated
|
2001-04-08 18:25:24 -04:00
|
|
|
|
string-representation of TM (in form hh:mm:ss) is printed.
|
|
|
|
|
|
2003-11-04 19:11:33 -05:00
|
|
|
|
If TM is NULL, the current time will be used. */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
|
|
|
|
char *
|
2006-08-08 10:32:53 -04:00
|
|
|
|
time_str (time_t t)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2006-08-08 10:32:53 -04:00
|
|
|
|
return fmttime(t, "%H:%M:%S");
|
2001-04-08 18:25:24 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Like the above, but include the date: YYYY-MM-DD hh:mm:ss. */
|
|
|
|
|
|
|
|
|
|
char *
|
2006-08-08 10:32:53 -04:00
|
|
|
|
datetime_str (time_t t)
|
2001-04-08 18:25:24 -04:00
|
|
|
|
{
|
2006-08-08 10:32:53 -04:00
|
|
|
|
return fmttime(t, "%Y-%m-%d %H:%M:%S");
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* The Windows versions of the following two functions are defined in
|
2007-09-28 17:47:46 -04:00
|
|
|
|
mswindows.c. On MSDOS this function should never be called. */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2008-04-22 17:48:36 -04:00
|
|
|
|
#ifdef __VMS
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
fork_to_background (void)
|
|
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#else /* def __VMS */
|
|
|
|
|
|
2007-09-28 17:47:46 -04:00
|
|
|
|
#if !defined(WINDOWS) && !defined(MSDOS)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
void
|
|
|
|
|
fork_to_background (void)
|
|
|
|
|
{
|
|
|
|
|
pid_t pid;
|
|
|
|
|
/* Whether we arrange our own version of opt.lfilename here. */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool logfile_changed = false;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2008-04-13 02:10:47 -04:00
|
|
|
|
if (!opt.lfilename && (!opt.quiet || opt.server_response))
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2005-03-04 14:26:18 -05:00
|
|
|
|
/* We must create the file immediately to avoid either a race
|
2007-08-02 23:38:21 -04:00
|
|
|
|
condition (which arises from using unique_name and failing to
|
|
|
|
|
use fopen_excl) or lying to the user about the log file name
|
|
|
|
|
(which arises from using unique_name, printing the name, and
|
|
|
|
|
using fopen_excl later on.) */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
FILE *new_log_fp = unique_create (DEFAULT_LOGFILE, false, &opt.lfilename);
|
2005-03-04 14:26:18 -05:00
|
|
|
|
if (new_log_fp)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
logfile_changed = true;
|
|
|
|
|
fclose (new_log_fp);
|
|
|
|
|
}
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
pid = fork ();
|
|
|
|
|
if (pid < 0)
|
|
|
|
|
{
|
|
|
|
|
/* parent, error */
|
|
|
|
|
perror ("fork");
|
|
|
|
|
exit (1);
|
|
|
|
|
}
|
|
|
|
|
else if (pid != 0)
|
|
|
|
|
{
|
|
|
|
|
/* parent, no error */
|
2008-04-13 02:20:28 -04:00
|
|
|
|
printf (_("Continuing in background, pid %d.\n"), (int) pid);
|
2005-03-04 14:26:18 -05:00
|
|
|
|
if (logfile_changed)
|
2008-04-16 06:35:03 -04:00
|
|
|
|
printf (_("Output will be written to %s.\n"), quote (opt.lfilename));
|
2007-08-02 23:38:21 -04:00
|
|
|
|
exit (0); /* #### should we use _exit()? */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2001-12-10 00:31:45 -05:00
|
|
|
|
|
|
|
|
|
/* child: give up the privileges and keep running. */
|
|
|
|
|
setsid ();
|
|
|
|
|
freopen ("/dev/null", "r", stdin);
|
|
|
|
|
freopen ("/dev/null", "w", stdout);
|
|
|
|
|
freopen ("/dev/null", "w", stderr);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2007-09-28 17:47:46 -04:00
|
|
|
|
#endif /* !WINDOWS && !MSDOS */
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
#endif /* def __VMS [else] */
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2005-05-05 10:47:48 -04:00
|
|
|
|
/* "Touch" FILE, i.e. make its mtime ("modified time") equal the time
|
|
|
|
|
specified with TM. The atime ("access time") is set to the current
|
|
|
|
|
time. */
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
void
|
|
|
|
|
touch (const char *file, time_t tm)
|
|
|
|
|
{
|
|
|
|
|
#ifdef HAVE_STRUCT_UTIMBUF
|
|
|
|
|
struct utimbuf times;
|
|
|
|
|
#else
|
2005-05-05 10:47:48 -04:00
|
|
|
|
struct {
|
|
|
|
|
time_t actime;
|
|
|
|
|
time_t modtime;
|
|
|
|
|
} times;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
#endif
|
2005-05-05 10:47:48 -04:00
|
|
|
|
times.modtime = tm;
|
|
|
|
|
times.actime = time (NULL);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
if (utime (file, ×) == -1)
|
2000-11-06 16:24:57 -05:00
|
|
|
|
logprintf (LOG_NOTQUIET, "utime(%s): %s\n", file, strerror (errno));
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Checks if FILE is a symbolic link, and removes it if it is. Does
|
|
|
|
|
nothing under MS-Windows. */
|
|
|
|
|
int
|
|
|
|
|
remove_link (const char *file)
|
|
|
|
|
{
|
|
|
|
|
int err = 0;
|
2005-02-23 17:21:04 -05:00
|
|
|
|
struct_stat st;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
|
|
|
|
if (lstat (file, &st) == 0 && S_ISLNK (st.st_mode))
|
|
|
|
|
{
|
|
|
|
|
DEBUGP (("Unlinking %s (symlink).\n", file));
|
|
|
|
|
err = unlink (file);
|
|
|
|
|
if (err != 0)
|
2008-04-16 06:35:03 -04:00
|
|
|
|
logprintf (LOG_VERBOSE, _("Failed to unlink symlink %s: %s\n"),
|
|
|
|
|
quote (file), strerror (errno));
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
return err;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Does FILENAME exist? This is quite a lousy implementation, since
|
|
|
|
|
it supplies no error codes -- only a yes-or-no answer. Thus it
|
|
|
|
|
will return that a file does not exist if, e.g., the directory is
|
|
|
|
|
unreadable. I don't mind it too much currently, though. The
|
|
|
|
|
proper way should, of course, be to have a third, error state,
|
|
|
|
|
other than true/false, but that would introduce uncalled-for
|
|
|
|
|
additional complexity to the callers. */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
1999-12-02 02:42:23 -05:00
|
|
|
|
file_exists_p (const char *filename)
|
|
|
|
|
{
|
|
|
|
|
#ifdef HAVE_ACCESS
|
|
|
|
|
return access (filename, F_OK) >= 0;
|
|
|
|
|
#else
|
2005-02-23 17:21:04 -05:00
|
|
|
|
struct_stat buf;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
return stat (filename, &buf) >= 0;
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Returns 0 if PATH is a directory, 1 otherwise (any kind of file).
|
|
|
|
|
Returns 0 on error. */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
1999-12-02 02:42:23 -05:00
|
|
|
|
file_non_directory_p (const char *path)
|
|
|
|
|
{
|
2005-02-23 17:21:04 -05:00
|
|
|
|
struct_stat buf;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
/* Use lstat() rather than stat() so that symbolic links pointing to
|
|
|
|
|
directories can be identified correctly. */
|
|
|
|
|
if (lstat (path, &buf) != 0)
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
|
|
|
|
return S_ISDIR (buf.st_mode) ? false : true;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2002-04-14 00:22:47 -04:00
|
|
|
|
/* Return the size of file named by FILENAME, or -1 if it cannot be
|
|
|
|
|
opened or seeked into. */
|
2005-02-23 17:21:04 -05:00
|
|
|
|
wgint
|
2002-04-14 00:22:47 -04:00
|
|
|
|
file_size (const char *filename)
|
|
|
|
|
{
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#if defined(HAVE_FSEEKO) && defined(HAVE_FTELLO)
|
|
|
|
|
wgint size;
|
2002-04-14 00:22:47 -04:00
|
|
|
|
/* We use fseek rather than stat to determine the file size because
|
2005-02-23 17:21:04 -05:00
|
|
|
|
that way we can also verify that the file is readable without
|
|
|
|
|
explicitly checking for permissions. Inspired by the POST patch
|
|
|
|
|
by Arnaud Wylie. */
|
2002-04-14 00:22:47 -04:00
|
|
|
|
FILE *fp = fopen (filename, "rb");
|
2003-10-04 18:26:58 -04:00
|
|
|
|
if (!fp)
|
|
|
|
|
return -1;
|
2005-02-23 17:21:04 -05:00
|
|
|
|
fseeko (fp, 0, SEEK_END);
|
|
|
|
|
size = ftello (fp);
|
2002-04-14 00:22:47 -04:00
|
|
|
|
fclose (fp);
|
|
|
|
|
return size;
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#else
|
|
|
|
|
struct_stat st;
|
|
|
|
|
if (stat (filename, &st) < 0)
|
|
|
|
|
return -1;
|
|
|
|
|
return st.st_size;
|
|
|
|
|
#endif
|
2002-04-14 00:22:47 -04:00
|
|
|
|
}
|
|
|
|
|
|
2008-04-22 17:48:36 -04:00
|
|
|
|
/* 2005-02-19 SMS.
|
|
|
|
|
If no UNIQ_SEP is defined (as on VMS), have unique_name() return the
|
|
|
|
|
original name. With the VMS file systems' versioning, everything
|
|
|
|
|
should be fine, and appending ".NN" just causes trouble.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#ifdef UNIQ_SEP
|
|
|
|
|
|
2003-09-16 17:47:49 -04:00
|
|
|
|
/* stat file names named PREFIX.1, PREFIX.2, etc., until one that
|
|
|
|
|
doesn't exist is found. Return a freshly allocated copy of the
|
|
|
|
|
unused file name. */
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
static char *
|
2003-09-16 17:47:49 -04:00
|
|
|
|
unique_name_1 (const char *prefix)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2003-09-16 17:47:49 -04:00
|
|
|
|
int count = 1;
|
|
|
|
|
int plen = strlen (prefix);
|
|
|
|
|
char *template = (char *)alloca (plen + 1 + 24);
|
|
|
|
|
char *template_tail = template + plen;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2003-09-16 17:47:49 -04:00
|
|
|
|
memcpy (template, prefix, plen);
|
2008-04-22 17:48:36 -04:00
|
|
|
|
*template_tail++ = UNIQ_SEP;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2003-09-16 17:47:49 -04:00
|
|
|
|
do
|
|
|
|
|
number_to_string (template_tail, count++);
|
|
|
|
|
while (file_exists_p (template));
|
|
|
|
|
|
|
|
|
|
return xstrdup (template);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2003-09-16 17:47:49 -04:00
|
|
|
|
/* Return a unique file name, based on FILE.
|
|
|
|
|
|
|
|
|
|
More precisely, if FILE doesn't exist, it is returned unmodified.
|
|
|
|
|
If not, FILE.1 is tried, then FILE.2, etc. The first FILE.<number>
|
|
|
|
|
file name that doesn't exist is returned.
|
|
|
|
|
|
2008-04-22 17:48:36 -04:00
|
|
|
|
2005-02-19 SMS. "." is now UNIQ_SEP, and may be different.
|
|
|
|
|
|
2003-09-16 17:47:49 -04:00
|
|
|
|
The resulting file is not created, only verified that it didn't
|
|
|
|
|
exist at the point in time when the function was called.
|
|
|
|
|
Therefore, where security matters, don't rely that the file created
|
|
|
|
|
by this function exists until you open it with O_EXCL or
|
2005-03-04 14:26:18 -05:00
|
|
|
|
equivalent.
|
2003-09-16 17:47:49 -04:00
|
|
|
|
|
|
|
|
|
If ALLOW_PASSTHROUGH is 0, it always returns a freshly allocated
|
|
|
|
|
string. Otherwise, it may return FILE if the file doesn't exist
|
|
|
|
|
(and therefore doesn't need changing). */
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
char *
|
2005-06-22 15:38:10 -04:00
|
|
|
|
unique_name (const char *file, bool allow_passthrough)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2003-09-16 17:47:49 -04:00
|
|
|
|
/* If the FILE itself doesn't exist, return it without
|
|
|
|
|
modification. */
|
|
|
|
|
if (!file_exists_p (file))
|
|
|
|
|
return allow_passthrough ? (char *)file : xstrdup (file);
|
|
|
|
|
|
|
|
|
|
/* Otherwise, find a numeric suffix that results in unused file name
|
|
|
|
|
and return it. */
|
|
|
|
|
return unique_name_1 (file);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2005-03-04 14:26:18 -05:00
|
|
|
|
|
2008-04-22 17:48:36 -04:00
|
|
|
|
#else /* def UNIQ_SEP */
|
|
|
|
|
|
|
|
|
|
/* Dummy unique_name() for VMS. Return the original name as easily as
|
|
|
|
|
possible.
|
|
|
|
|
*/
|
|
|
|
|
char *
|
|
|
|
|
unique_name (const char *file, bool allow_passthrough)
|
|
|
|
|
{
|
|
|
|
|
/* Return the FILE itself, without modification, irregardful. */
|
|
|
|
|
return allow_passthrough ? (char *)file : xstrdup (file);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif /* def UNIQ_SEP [else] */
|
|
|
|
|
|
2005-03-04 14:26:18 -05:00
|
|
|
|
/* Create a file based on NAME, except without overwriting an existing
|
|
|
|
|
file with that name. Providing O_EXCL is correctly implemented,
|
|
|
|
|
this function does not have the race condition associated with
|
|
|
|
|
opening the file returned by unique_name. */
|
|
|
|
|
|
|
|
|
|
FILE *
|
2005-06-22 15:38:10 -04:00
|
|
|
|
unique_create (const char *name, bool binary, char **opened_name)
|
2005-03-04 14:26:18 -05:00
|
|
|
|
{
|
|
|
|
|
/* unique file name, based on NAME */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
char *uname = unique_name (name, false);
|
2005-03-04 14:26:18 -05:00
|
|
|
|
FILE *fp;
|
|
|
|
|
while ((fp = fopen_excl (uname, binary)) == NULL && errno == EEXIST)
|
|
|
|
|
{
|
|
|
|
|
xfree (uname);
|
2005-06-22 15:38:10 -04:00
|
|
|
|
uname = unique_name (name, false);
|
2005-03-04 14:26:18 -05:00
|
|
|
|
}
|
|
|
|
|
if (opened_name && fp != NULL)
|
|
|
|
|
{
|
|
|
|
|
if (fp)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
*opened_name = uname;
|
2005-03-04 14:26:18 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
*opened_name = NULL;
|
|
|
|
|
xfree (uname);
|
|
|
|
|
}
|
2005-03-04 14:26:18 -05:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
xfree (uname);
|
|
|
|
|
return fp;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Open the file for writing, with the addition that the file is
|
|
|
|
|
opened "exclusively". This means that, if the file already exists,
|
|
|
|
|
this function will *fail* and errno will be set to EEXIST. If
|
|
|
|
|
BINARY is set, the file will be opened in binary mode, equivalent
|
|
|
|
|
to fopen's "wb".
|
|
|
|
|
|
|
|
|
|
If opening the file fails for any reason, including the file having
|
|
|
|
|
previously existed, this function returns NULL and sets errno
|
|
|
|
|
appropriately. */
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
2005-03-04 14:26:18 -05:00
|
|
|
|
FILE *
|
2009-09-11 22:41:57 -04:00
|
|
|
|
fopen_excl (const char *fname, int binary)
|
2005-03-04 14:26:18 -05:00
|
|
|
|
{
|
2005-03-05 18:35:15 -05:00
|
|
|
|
int fd;
|
2005-03-04 14:26:18 -05:00
|
|
|
|
#ifdef O_EXCL
|
2008-04-22 17:48:36 -04:00
|
|
|
|
|
|
|
|
|
/* 2005-04-14 SMS.
|
|
|
|
|
VMS lacks O_BINARY, but makes up for it in weird and wonderful ways.
|
|
|
|
|
It also has file versions which obviate all the O_EXCL effort.
|
|
|
|
|
O_TRUNC (something of a misnomer) requests a new version.
|
|
|
|
|
*/
|
|
|
|
|
# ifdef __VMS
|
|
|
|
|
/* Common open() optional arguments:
|
|
|
|
|
sequential access only, access callback function.
|
|
|
|
|
*/
|
|
|
|
|
# define OPEN_OPT_ARGS "fop=sqo", "acc", acc_cb, &open_id
|
|
|
|
|
|
|
|
|
|
int open_id;
|
|
|
|
|
int flags = O_WRONLY | O_CREAT | O_TRUNC;
|
|
|
|
|
|
|
|
|
|
if (binary > 1)
|
|
|
|
|
{
|
|
|
|
|
open_id = 11;
|
|
|
|
|
fd = open( fname, /* File name. */
|
|
|
|
|
flags, /* Flags. */
|
|
|
|
|
0777, /* Mode for default protection. */
|
|
|
|
|
"ctx=bin,stm", /* Binary, stream access. */
|
|
|
|
|
"rfm=stmlf", /* Stream_LF. */
|
|
|
|
|
OPEN_OPT_ARGS); /* Access callback. */
|
|
|
|
|
}
|
|
|
|
|
else if (binary)
|
|
|
|
|
{
|
|
|
|
|
open_id = 12;
|
|
|
|
|
fd = open( fname, /* File name. */
|
|
|
|
|
flags, /* Flags. */
|
|
|
|
|
0777, /* Mode for default protection. */
|
|
|
|
|
"ctx=bin,stm", /* Binary, stream access. */
|
|
|
|
|
"rfm=fix", /* Fixed-length, */
|
|
|
|
|
"mrs=512", /* 512-byte records. */
|
|
|
|
|
OPEN_OPT_ARGS); /* Access callback. */
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
open_id = 13;
|
|
|
|
|
fd = open( fname, /* File name. */
|
|
|
|
|
flags, /* Flags. */
|
|
|
|
|
0777, /* Mode for default protection.
|
|
|
|
|
*/
|
|
|
|
|
"rfm=stmlf", /* Stream_LF. */
|
|
|
|
|
OPEN_OPT_ARGS); /* Access callback. */
|
|
|
|
|
}
|
|
|
|
|
# else /* def __VMS */
|
2005-03-04 14:26:18 -05:00
|
|
|
|
int flags = O_WRONLY | O_CREAT | O_EXCL;
|
|
|
|
|
# ifdef O_BINARY
|
|
|
|
|
if (binary)
|
2005-03-05 18:35:15 -05:00
|
|
|
|
flags |= O_BINARY;
|
2005-03-04 14:26:18 -05:00
|
|
|
|
# endif
|
2005-03-05 18:35:15 -05:00
|
|
|
|
fd = open (fname, flags, 0666);
|
2008-04-22 17:48:36 -04:00
|
|
|
|
# endif /* def __VMS [else] */
|
|
|
|
|
|
2005-03-04 14:26:18 -05:00
|
|
|
|
if (fd < 0)
|
|
|
|
|
return NULL;
|
|
|
|
|
return fdopen (fd, binary ? "wb" : "w");
|
|
|
|
|
#else /* not O_EXCL */
|
2005-03-20 10:07:40 -05:00
|
|
|
|
/* Manually check whether the file exists. This is prone to race
|
|
|
|
|
conditions, but systems without O_EXCL haven't deserved
|
|
|
|
|
better. */
|
|
|
|
|
if (file_exists_p (fname))
|
|
|
|
|
{
|
|
|
|
|
errno = EEXIST;
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2005-03-04 14:26:18 -05:00
|
|
|
|
return fopen (fname, binary ? "wb" : "w");
|
|
|
|
|
#endif /* not O_EXCL */
|
|
|
|
|
}
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
|
|
|
|
/* Create DIRECTORY. If some of the pathname components of DIRECTORY
|
|
|
|
|
are missing, create them first. In case any mkdir() call fails,
|
|
|
|
|
return its error status. Returns 0 on successful completion.
|
|
|
|
|
|
|
|
|
|
The behaviour of this function should be identical to the behaviour
|
|
|
|
|
of `mkdir -p' on systems where mkdir supports the `-p' option. */
|
|
|
|
|
int
|
|
|
|
|
make_directory (const char *directory)
|
|
|
|
|
{
|
2005-03-05 21:11:10 -05:00
|
|
|
|
int i, ret, quit = 0;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
char *dir;
|
|
|
|
|
|
|
|
|
|
/* Make a copy of dir, to be able to write to it. Otherwise, the
|
|
|
|
|
function is unsafe if called with a read-only char *argument. */
|
|
|
|
|
STRDUP_ALLOCA (dir, directory);
|
|
|
|
|
|
|
|
|
|
/* If the first character of dir is '/', skip it (and thus enable
|
|
|
|
|
creation of absolute-pathname directories. */
|
|
|
|
|
for (i = (*dir == '/'); 1; ++i)
|
|
|
|
|
{
|
|
|
|
|
for (; dir[i] && dir[i] != '/'; i++)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
if (!dir[i])
|
2007-08-02 23:38:21 -04:00
|
|
|
|
quit = 1;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
dir[i] = '\0';
|
2002-01-16 20:03:33 -05:00
|
|
|
|
/* Check whether the directory already exists. Allow creation of
|
2007-08-02 23:38:21 -04:00
|
|
|
|
of intermediate directories to fail, as the initial path components
|
|
|
|
|
are not necessarily directories! */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
if (!file_exists_p (dir))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
ret = mkdir (dir, 0777);
|
2002-01-16 20:03:33 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
ret = 0;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
if (quit)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
break;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
dir[i] = '/';
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2002-01-16 20:03:33 -05:00
|
|
|
|
return ret;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2001-11-21 19:24:28 -05:00
|
|
|
|
|
|
|
|
|
/* Merge BASE with FILE. BASE can be a directory or a file name, FILE
|
2001-12-02 16:44:16 -05:00
|
|
|
|
should be a file name.
|
|
|
|
|
|
|
|
|
|
file_merge("/foo/bar", "baz") => "/foo/baz"
|
|
|
|
|
file_merge("/foo/bar/", "baz") => "/foo/bar/baz"
|
|
|
|
|
file_merge("foo", "bar") => "bar"
|
2001-11-21 19:24:28 -05:00
|
|
|
|
|
2005-07-06 15:56:20 -04:00
|
|
|
|
In other words, it's a simpler and gentler version of uri_merge. */
|
2001-11-21 19:24:28 -05:00
|
|
|
|
|
|
|
|
|
char *
|
|
|
|
|
file_merge (const char *base, const char *file)
|
|
|
|
|
{
|
|
|
|
|
char *result;
|
|
|
|
|
const char *cut = (const char *)strrchr (base, '/');
|
|
|
|
|
|
|
|
|
|
if (!cut)
|
2001-12-02 16:44:16 -05:00
|
|
|
|
return xstrdup (file);
|
2001-11-21 19:24:28 -05:00
|
|
|
|
|
2005-06-19 19:03:27 -04:00
|
|
|
|
result = xmalloc (cut - base + 1 + strlen (file) + 1);
|
2001-11-21 19:24:28 -05:00
|
|
|
|
memcpy (result, base, cut - base);
|
|
|
|
|
result[cut - base] = '/';
|
|
|
|
|
strcpy (result + (cut - base) + 1, file);
|
|
|
|
|
|
|
|
|
|
return result;
|
|
|
|
|
}
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2005-07-06 16:03:50 -04:00
|
|
|
|
/* Like fnmatch, but performs a case-insensitive match. */
|
2005-07-06 15:44:00 -04:00
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
fnmatch_nocase (const char *pattern, const char *string, int flags)
|
|
|
|
|
{
|
|
|
|
|
#ifdef FNM_CASEFOLD
|
2005-07-06 15:50:26 -04:00
|
|
|
|
/* The FNM_CASEFOLD flag started as a GNU extension, but it is now
|
|
|
|
|
also present on *BSD platforms, and possibly elsewhere. */
|
2005-07-06 15:44:00 -04:00
|
|
|
|
return fnmatch (pattern, string, flags | FNM_CASEFOLD);
|
|
|
|
|
#else
|
|
|
|
|
/* Turn PATTERN and STRING to lower case and call fnmatch on them. */
|
|
|
|
|
char *patcopy = (char *) alloca (strlen (pattern) + 1);
|
|
|
|
|
char *strcopy = (char *) alloca (strlen (string) + 1);
|
|
|
|
|
char *p;
|
|
|
|
|
for (p = patcopy; *pattern; pattern++, p++)
|
2007-10-14 17:46:24 -04:00
|
|
|
|
*p = c_tolower (*pattern);
|
2005-07-06 15:44:00 -04:00
|
|
|
|
*p = '\0';
|
|
|
|
|
for (p = strcopy; *string; string++, p++)
|
2007-10-14 17:46:24 -04:00
|
|
|
|
*p = c_tolower (*string);
|
2005-07-06 15:44:00 -04:00
|
|
|
|
*p = '\0';
|
|
|
|
|
return fnmatch (patcopy, strcopy, flags);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
static bool in_acclist (const char *const *, const char *, bool);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
|
|
|
|
/* Determine whether a file is acceptable to be followed, according to
|
|
|
|
|
lists of patterns to accept/reject. */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
1999-12-02 02:42:23 -05:00
|
|
|
|
acceptable (const char *s)
|
|
|
|
|
{
|
|
|
|
|
int l = strlen (s);
|
|
|
|
|
|
|
|
|
|
while (l && s[l] != '/')
|
|
|
|
|
--l;
|
|
|
|
|
if (s[l] == '/')
|
|
|
|
|
s += (l + 1);
|
|
|
|
|
if (opt.accepts)
|
|
|
|
|
{
|
|
|
|
|
if (opt.rejects)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return (in_acclist ((const char *const *)opt.accepts, s, true)
|
|
|
|
|
&& !in_acclist ((const char *const *)opt.rejects, s, true));
|
1999-12-02 02:42:23 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return in_acclist ((const char *const *)opt.accepts, s, true);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
else if (opt.rejects)
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return !in_acclist ((const char *const *)opt.rejects, s, true);
|
|
|
|
|
return true;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2006-03-15 09:55:29 -05:00
|
|
|
|
/* Check if D2 is a subdirectory of D1. E.g. if D1 is `/something', subdir_p()
|
2009-09-21 23:39:44 -04:00
|
|
|
|
will return true if and only if D2 begins with `/something/' or is exactly
|
2006-03-15 09:55:29 -05:00
|
|
|
|
'/something'. */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
2006-03-15 09:55:29 -05:00
|
|
|
|
subdir_p (const char *d1, const char *d2)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2007-12-05 23:56:23 -05:00
|
|
|
|
if (*d1 == '\0')
|
|
|
|
|
return true;
|
2005-07-06 15:44:00 -04:00
|
|
|
|
if (!opt.ignore_case)
|
2006-03-15 09:55:29 -05:00
|
|
|
|
for (; *d1 && *d2 && (*d1 == *d2); ++d1, ++d2)
|
2005-07-06 21:08:52 -04:00
|
|
|
|
;
|
2005-07-06 15:44:00 -04:00
|
|
|
|
else
|
2007-10-14 17:46:24 -04:00
|
|
|
|
for (; *d1 && *d2 && (c_tolower (*d1) == c_tolower (*d2)); ++d1, ++d2)
|
2005-07-06 21:08:52 -04:00
|
|
|
|
;
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
2006-03-15 09:55:29 -05:00
|
|
|
|
return *d1 == '\0' && (*d2 == '\0' || *d2 == '/');
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2006-03-15 09:55:29 -05:00
|
|
|
|
/* Iterate through DIRLIST (which must be NULL-terminated), and return the
|
|
|
|
|
first element that matches DIR, through wildcards or front comparison (as
|
|
|
|
|
appropriate). */
|
|
|
|
|
static bool
|
|
|
|
|
dir_matches_p (char **dirlist, const char *dir)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
|
|
|
|
char **x;
|
2005-07-06 15:44:00 -04:00
|
|
|
|
int (*matcher) (const char *, const char *, int)
|
|
|
|
|
= opt.ignore_case ? fnmatch_nocase : fnmatch;
|
|
|
|
|
|
2006-03-15 09:55:29 -05:00
|
|
|
|
for (x = dirlist; *x; x++)
|
2005-05-05 11:22:11 -04:00
|
|
|
|
{
|
2005-07-06 15:44:00 -04:00
|
|
|
|
/* Remove leading '/' */
|
|
|
|
|
char *p = *x + (**x == '/');
|
2005-05-05 11:22:11 -04:00
|
|
|
|
if (has_wildcards_p (p))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
if (matcher (p, dir, FNM_PATHNAME) == 0)
|
|
|
|
|
break;
|
|
|
|
|
}
|
2005-05-05 11:22:11 -04:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
if (subdir_p (p, dir))
|
|
|
|
|
break;
|
|
|
|
|
}
|
2005-05-05 11:22:11 -04:00
|
|
|
|
}
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
2006-03-15 09:55:29 -05:00
|
|
|
|
return *x ? true : false;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Returns whether DIRECTORY is acceptable for download, wrt the
|
|
|
|
|
include/exclude lists.
|
|
|
|
|
|
2005-07-06 15:44:00 -04:00
|
|
|
|
The leading `/' is ignored in paths; relative and absolute paths
|
|
|
|
|
may be freely intermixed. */
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
2005-07-06 15:44:00 -04:00
|
|
|
|
accdir (const char *directory)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
|
|
|
|
/* Remove starting '/'. */
|
2005-07-06 15:44:00 -04:00
|
|
|
|
if (*directory == '/')
|
1999-12-02 02:42:23 -05:00
|
|
|
|
++directory;
|
|
|
|
|
if (opt.includes)
|
|
|
|
|
{
|
2006-03-15 09:55:29 -05:00
|
|
|
|
if (!dir_matches_p (opt.includes, directory))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return false;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
if (opt.excludes)
|
|
|
|
|
{
|
2006-03-15 09:55:29 -05:00
|
|
|
|
if (dir_matches_p (opt.excludes, directory))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return false;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
/* Return true if STRING ends with TAIL. For instance:
|
2002-05-08 14:24:40 -04:00
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
match_tail ("abc", "bc", false) -> 1
|
|
|
|
|
match_tail ("abc", "ab", false) -> 0
|
|
|
|
|
match_tail ("abc", "abc", false) -> 1
|
2002-05-08 14:24:40 -04:00
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
If FOLD_CASE is true, the comparison will be case-insensitive. */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
|
|
|
|
match_tail (const char *string, const char *tail, bool fold_case)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
|
|
|
|
int i, j;
|
|
|
|
|
|
2002-05-08 14:24:40 -04:00
|
|
|
|
/* We want this to be fast, so we code two loops, one with
|
|
|
|
|
case-folding, one without. */
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
if (!fold_case)
|
2002-05-08 14:24:40 -04:00
|
|
|
|
{
|
|
|
|
|
for (i = strlen (string), j = strlen (tail); i >= 0 && j >= 0; i--, j--)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
if (string[i] != tail[j])
|
|
|
|
|
break;
|
2002-05-08 14:24:40 -04:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
for (i = strlen (string), j = strlen (tail); i >= 0 && j >= 0; i--, j--)
|
2007-10-14 17:46:24 -04:00
|
|
|
|
if (c_tolower (string[i]) != c_tolower (tail[j]))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
break;
|
2002-05-08 14:24:40 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* If the tail was exhausted, the match was succesful. */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
if (j == -1)
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
else
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Checks whether string S matches each element of ACCEPTS. A list
|
2001-12-01 00:08:03 -05:00
|
|
|
|
element are matched either with fnmatch() or match_tail(),
|
1999-12-02 02:42:23 -05:00
|
|
|
|
according to whether the element contains wildcards or not.
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
If the BACKWARD is false, don't do backward comparison -- just compare
|
1999-12-02 02:42:23 -05:00
|
|
|
|
them normally. */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
static bool
|
|
|
|
|
in_acclist (const char *const *accepts, const char *s, bool backward)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
|
|
|
|
for (; *accepts; accepts++)
|
|
|
|
|
{
|
|
|
|
|
if (has_wildcards_p (*accepts))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
int res = opt.ignore_case
|
|
|
|
|
? fnmatch_nocase (*accepts, s, 0) : fnmatch (*accepts, s, 0);
|
|
|
|
|
/* fnmatch returns 0 if the pattern *does* match the string. */
|
|
|
|
|
if (res == 0)
|
|
|
|
|
return true;
|
|
|
|
|
}
|
1999-12-02 02:42:23 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
if (backward)
|
|
|
|
|
{
|
|
|
|
|
if (match_tail (s, *accepts, opt.ignore_case))
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
int cmp = opt.ignore_case
|
|
|
|
|
? strcasecmp (s, *accepts) : strcmp (s, *accepts);
|
|
|
|
|
if (cmp == 0)
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
}
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2001-11-25 16:03:30 -05:00
|
|
|
|
/* Return the location of STR's suffix (file extension). Examples:
|
1999-12-02 02:42:23 -05:00
|
|
|
|
suffix ("foo.bar") -> "bar"
|
|
|
|
|
suffix ("foo.bar.baz") -> "baz"
|
|
|
|
|
suffix ("/foo/bar") -> NULL
|
|
|
|
|
suffix ("/foo.bar/baz") -> NULL */
|
|
|
|
|
char *
|
|
|
|
|
suffix (const char *str)
|
|
|
|
|
{
|
|
|
|
|
int i;
|
|
|
|
|
|
2001-11-25 16:03:30 -05:00
|
|
|
|
for (i = strlen (str); i && str[i] != '/' && str[i] != '.'; i--)
|
|
|
|
|
;
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
if (str[i++] == '.')
|
2001-11-25 16:03:30 -05:00
|
|
|
|
return (char *)str + i;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
else
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
/* Return true if S contains globbing wildcards (`*', `?', `[' or
|
2003-10-07 19:53:31 -04:00
|
|
|
|
`]'). */
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
2003-10-07 19:53:31 -04:00
|
|
|
|
has_wildcards_p (const char *s)
|
|
|
|
|
{
|
|
|
|
|
for (; *s; s++)
|
|
|
|
|
if (*s == '*' || *s == '?' || *s == '[' || *s == ']')
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
|
|
|
|
return false;
|
2003-10-07 19:53:31 -04:00
|
|
|
|
}
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
/* Return true if FNAME ends with a typical HTML suffix. The
|
|
|
|
|
following (case-insensitive) suffixes are presumed to be HTML
|
|
|
|
|
files:
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
2002-04-12 14:53:39 -04:00
|
|
|
|
html
|
|
|
|
|
htm
|
|
|
|
|
?html (`?' matches one character)
|
|
|
|
|
|
|
|
|
|
#### CAVEAT. This is not necessarily a good indication that FNAME
|
|
|
|
|
refers to a file that contains HTML! */
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
2002-04-12 14:53:39 -04:00
|
|
|
|
has_html_suffix_p (const char *fname)
|
|
|
|
|
{
|
|
|
|
|
char *suf;
|
|
|
|
|
|
|
|
|
|
if ((suf = suffix (fname)) == NULL)
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
2002-04-12 14:53:39 -04:00
|
|
|
|
if (!strcasecmp (suf, "html"))
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
2002-04-12 14:53:39 -04:00
|
|
|
|
if (!strcasecmp (suf, "htm"))
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
2002-04-12 14:53:39 -04:00
|
|
|
|
if (suf[0] && !strcasecmp (suf + 1, "html"))
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
|
|
|
|
return false;
|
2002-04-12 14:53:39 -04:00
|
|
|
|
}
|
|
|
|
|
|
2001-12-09 22:34:48 -05:00
|
|
|
|
/* Read a line from FP and return the pointer to freshly allocated
|
2003-10-04 18:58:34 -04:00
|
|
|
|
storage. The storage space is obtained through malloc() and should
|
|
|
|
|
be freed with free() when it is no longer needed.
|
2001-12-09 22:34:48 -05:00
|
|
|
|
|
|
|
|
|
The length of the line is not limited, except by available memory.
|
|
|
|
|
The newline character at the end of line is retained. The line is
|
|
|
|
|
terminated with a zero character.
|
2000-11-10 13:01:35 -05:00
|
|
|
|
|
|
|
|
|
After end-of-file is encountered without anything being read, NULL
|
|
|
|
|
is returned. NULL is also returned on error. To distinguish
|
2001-12-09 22:34:48 -05:00
|
|
|
|
between these two cases, use the stdio function ferror(). */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
|
|
|
|
char *
|
|
|
|
|
read_whole_line (FILE *fp)
|
|
|
|
|
{
|
2000-11-10 13:01:35 -05:00
|
|
|
|
int length = 0;
|
2001-12-09 22:34:48 -05:00
|
|
|
|
int bufsize = 82;
|
2005-06-19 19:03:27 -04:00
|
|
|
|
char *line = xmalloc (bufsize);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2000-11-10 13:01:35 -05:00
|
|
|
|
while (fgets (line + length, bufsize - length, fp))
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2000-11-10 13:01:35 -05:00
|
|
|
|
length += strlen (line + length);
|
2001-12-05 18:58:12 -05:00
|
|
|
|
if (length == 0)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
/* Possible for example when reading from a binary file where
|
|
|
|
|
a line begins with \0. */
|
|
|
|
|
continue;
|
2001-12-05 18:58:12 -05:00
|
|
|
|
|
2000-11-10 13:01:35 -05:00
|
|
|
|
if (line[length - 1] == '\n')
|
2007-08-02 23:38:21 -04:00
|
|
|
|
break;
|
2001-12-05 18:58:12 -05:00
|
|
|
|
|
2000-11-10 13:01:35 -05:00
|
|
|
|
/* fgets() guarantees to read the whole line, or to use up the
|
|
|
|
|
space we've given it. We can double the buffer
|
|
|
|
|
unconditionally. */
|
|
|
|
|
bufsize <<= 1;
|
|
|
|
|
line = xrealloc (line, bufsize);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2000-11-10 13:01:35 -05:00
|
|
|
|
if (length == 0 || ferror (fp))
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (line);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2000-11-10 13:01:35 -05:00
|
|
|
|
if (length + 1 < bufsize)
|
|
|
|
|
/* Relieve the memory from our exponential greediness. We say
|
|
|
|
|
`length + 1' because the terminating \0 is not included in
|
|
|
|
|
LENGTH. We don't need to zero-terminate the string ourselves,
|
|
|
|
|
though, because fgets() does that. */
|
|
|
|
|
line = xrealloc (line, length + 1);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
return line;
|
|
|
|
|
}
|
2000-11-19 15:50:10 -05:00
|
|
|
|
|
|
|
|
|
/* Read FILE into memory. A pointer to `struct file_memory' are
|
|
|
|
|
returned; use struct element `content' to access file contents, and
|
|
|
|
|
the element `length' to know the file length. `content' is *not*
|
|
|
|
|
zero-terminated, and you should *not* read or write beyond the [0,
|
|
|
|
|
length) range of characters.
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2000-11-19 15:50:10 -05:00
|
|
|
|
After you are done with the file contents, call read_file_free to
|
|
|
|
|
release the memory.
|
|
|
|
|
|
|
|
|
|
Depending on the operating system and the type of file that is
|
|
|
|
|
being read, read_file() either mmap's the file into memory, or
|
|
|
|
|
reads the file into the core using read().
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2000-11-19 15:50:10 -05:00
|
|
|
|
If file is named "-", fileno(stdin) is used for reading instead.
|
|
|
|
|
If you want to read from a real file named "-", use "./-" instead. */
|
|
|
|
|
|
|
|
|
|
struct file_memory *
|
|
|
|
|
read_file (const char *file)
|
|
|
|
|
{
|
|
|
|
|
int fd;
|
|
|
|
|
struct file_memory *fm;
|
2005-03-05 21:11:10 -05:00
|
|
|
|
long size;
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool inhibit_close = false;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
|
|
|
|
|
/* Some magic in the finest tradition of Perl and its kin: if FILE
|
|
|
|
|
is "-", just use stdin. */
|
|
|
|
|
if (HYPHENP (file))
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2000-11-19 15:50:10 -05:00
|
|
|
|
fd = fileno (stdin);
|
2005-06-22 15:38:10 -04:00
|
|
|
|
inhibit_close = true;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
/* Note that we don't inhibit mmap() in this case. If stdin is
|
|
|
|
|
redirected from a regular file, mmap() will still work. */
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
fd = open (file, O_RDONLY);
|
|
|
|
|
if (fd < 0)
|
|
|
|
|
return NULL;
|
2003-10-31 09:55:50 -05:00
|
|
|
|
fm = xnew (struct file_memory);
|
2000-11-19 15:50:10 -05:00
|
|
|
|
|
|
|
|
|
#ifdef HAVE_MMAP
|
|
|
|
|
{
|
2005-06-23 10:29:42 -04:00
|
|
|
|
struct_fstat buf;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
if (fstat (fd, &buf) < 0)
|
|
|
|
|
goto mmap_lose;
|
|
|
|
|
fm->length = buf.st_size;
|
|
|
|
|
/* NOTE: As far as I know, the callers of this function never
|
|
|
|
|
modify the file text. Relying on this would enable us to
|
|
|
|
|
specify PROT_READ and MAP_SHARED for a marginal gain in
|
|
|
|
|
efficiency, but at some cost to generality. */
|
|
|
|
|
fm->content = mmap (NULL, fm->length, PROT_READ | PROT_WRITE,
|
2007-08-02 23:38:21 -04:00
|
|
|
|
MAP_PRIVATE, fd, 0);
|
2001-04-06 00:43:13 -04:00
|
|
|
|
if (fm->content == (char *)MAP_FAILED)
|
2000-11-19 15:50:10 -05:00
|
|
|
|
goto mmap_lose;
|
|
|
|
|
if (!inhibit_close)
|
|
|
|
|
close (fd);
|
|
|
|
|
|
|
|
|
|
fm->mmap_p = 1;
|
|
|
|
|
return fm;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
mmap_lose:
|
|
|
|
|
/* The most common reason why mmap() fails is that FD does not point
|
|
|
|
|
to a plain file. However, it's also possible that mmap() doesn't
|
|
|
|
|
work for a particular type of file. Therefore, whenever mmap()
|
|
|
|
|
fails, we just fall back to the regular method. */
|
|
|
|
|
#endif /* HAVE_MMAP */
|
|
|
|
|
|
|
|
|
|
fm->length = 0;
|
2007-08-02 23:38:21 -04:00
|
|
|
|
size = 512; /* number of bytes fm->contents can
|
2000-11-19 15:50:10 -05:00
|
|
|
|
hold at any given time. */
|
|
|
|
|
fm->content = xmalloc (size);
|
|
|
|
|
while (1)
|
|
|
|
|
{
|
2005-02-23 17:21:04 -05:00
|
|
|
|
wgint nread;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
if (fm->length > size / 2)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
/* #### I'm not sure whether the whole exponential-growth
|
2000-11-19 15:50:10 -05:00
|
|
|
|
thing makes sense with kernel read. On Linux at least,
|
|
|
|
|
read() refuses to read more than 4K from a file at a
|
|
|
|
|
single chunk anyway. But other Unixes might optimize it
|
|
|
|
|
better, and it doesn't *hurt* anything, so I'm leaving
|
|
|
|
|
it. */
|
|
|
|
|
|
2007-08-02 23:38:21 -04:00
|
|
|
|
/* Normally, we grow SIZE exponentially to make the number
|
2000-11-19 15:50:10 -05:00
|
|
|
|
of calls to read() and realloc() logarithmic in relation
|
|
|
|
|
to file size. However, read() can read an amount of data
|
2003-10-04 18:58:34 -04:00
|
|
|
|
smaller than requested, and it would be unreasonable to
|
2000-11-19 15:50:10 -05:00
|
|
|
|
double SIZE every time *something* was read. Therefore,
|
|
|
|
|
we double SIZE only when the length exceeds half of the
|
|
|
|
|
entire allocated size. */
|
2007-08-02 23:38:21 -04:00
|
|
|
|
size <<= 1;
|
|
|
|
|
fm->content = xrealloc (fm->content, size);
|
|
|
|
|
}
|
2000-11-19 15:50:10 -05:00
|
|
|
|
nread = read (fd, fm->content + fm->length, size - fm->length);
|
|
|
|
|
if (nread > 0)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
/* Successful read. */
|
|
|
|
|
fm->length += nread;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
else if (nread < 0)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
/* Error. */
|
|
|
|
|
goto lose;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
else
|
2007-08-02 23:38:21 -04:00
|
|
|
|
/* EOF */
|
|
|
|
|
break;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2000-11-19 15:50:10 -05:00
|
|
|
|
if (!inhibit_close)
|
|
|
|
|
close (fd);
|
|
|
|
|
if (size > fm->length && fm->length != 0)
|
|
|
|
|
/* Due to exponential growth of fm->content, the allocated region
|
|
|
|
|
might be much larger than what is actually needed. */
|
|
|
|
|
fm->content = xrealloc (fm->content, fm->length);
|
|
|
|
|
fm->mmap_p = 0;
|
|
|
|
|
return fm;
|
|
|
|
|
|
|
|
|
|
lose:
|
|
|
|
|
if (!inhibit_close)
|
|
|
|
|
close (fd);
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (fm->content);
|
|
|
|
|
xfree (fm);
|
2000-11-19 15:50:10 -05:00
|
|
|
|
return NULL;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2000-11-19 15:50:10 -05:00
|
|
|
|
/* Release the resources held by FM. Specifically, this calls
|
2000-11-22 11:58:28 -05:00
|
|
|
|
munmap() or xfree() on fm->content, depending whether mmap or
|
2000-11-19 15:50:10 -05:00
|
|
|
|
malloc/read were used to read in the file. It also frees the
|
|
|
|
|
memory needed to hold the FM structure itself. */
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
read_file_free (struct file_memory *fm)
|
|
|
|
|
{
|
|
|
|
|
#ifdef HAVE_MMAP
|
|
|
|
|
if (fm->mmap_p)
|
|
|
|
|
{
|
|
|
|
|
munmap (fm->content, fm->length);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
#endif
|
|
|
|
|
{
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (fm->content);
|
2000-11-19 15:50:10 -05:00
|
|
|
|
}
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (fm);
|
2000-11-19 15:50:10 -05:00
|
|
|
|
}
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
/* Free the pointers in a NULL-terminated vector of pointers, then
|
|
|
|
|
free the pointer itself. */
|
|
|
|
|
void
|
|
|
|
|
free_vec (char **vec)
|
|
|
|
|
{
|
|
|
|
|
if (vec)
|
|
|
|
|
{
|
|
|
|
|
char **p = vec;
|
|
|
|
|
while (*p)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
xfree (*p++);
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (vec);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Append vector V2 to vector V1. The function frees V2 and
|
|
|
|
|
reallocates V1 (thus you may not use the contents of neither
|
|
|
|
|
pointer after the call). If V1 is NULL, V2 is returned. */
|
|
|
|
|
char **
|
|
|
|
|
merge_vecs (char **v1, char **v2)
|
|
|
|
|
{
|
|
|
|
|
int i, j;
|
|
|
|
|
|
|
|
|
|
if (!v1)
|
|
|
|
|
return v2;
|
|
|
|
|
if (!v2)
|
|
|
|
|
return v1;
|
|
|
|
|
if (!*v2)
|
|
|
|
|
{
|
|
|
|
|
/* To avoid j == 0 */
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (v2);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
return v1;
|
|
|
|
|
}
|
|
|
|
|
/* Count v1. */
|
2005-07-06 21:08:52 -04:00
|
|
|
|
for (i = 0; v1[i]; i++)
|
|
|
|
|
;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
/* Count v2. */
|
2005-07-06 21:08:52 -04:00
|
|
|
|
for (j = 0; v2[j]; j++)
|
|
|
|
|
;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
/* Reallocate v1. */
|
2005-06-19 19:03:27 -04:00
|
|
|
|
v1 = xrealloc (v1, (i + j + 1) * sizeof (char **));
|
1999-12-02 02:42:23 -05:00
|
|
|
|
memcpy (v1 + i, v2, (j + 1) * sizeof (char *));
|
2000-11-22 11:58:28 -05:00
|
|
|
|
xfree (v2);
|
1999-12-02 02:42:23 -05:00
|
|
|
|
return v1;
|
|
|
|
|
}
|
2005-05-30 09:31:24 -04:00
|
|
|
|
|
|
|
|
|
/* Append a freshly allocated copy of STR to VEC. If VEC is NULL, it
|
|
|
|
|
is allocated as needed. Return the new value of the vector. */
|
|
|
|
|
|
|
|
|
|
char **
|
|
|
|
|
vec_append (char **vec, const char *str)
|
|
|
|
|
{
|
2007-08-02 23:38:21 -04:00
|
|
|
|
int cnt; /* count of vector elements, including
|
|
|
|
|
the one we're about to append */
|
2005-05-30 09:31:24 -04:00
|
|
|
|
if (vec != NULL)
|
|
|
|
|
{
|
|
|
|
|
for (cnt = 0; vec[cnt]; cnt++)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
;
|
2005-05-30 09:31:24 -04:00
|
|
|
|
++cnt;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
cnt = 1;
|
|
|
|
|
/* Reallocate the array to fit the new element and the NULL. */
|
|
|
|
|
vec = xrealloc (vec, (cnt + 1) * sizeof (char *));
|
|
|
|
|
/* Append a copy of STR to the vector. */
|
|
|
|
|
vec[cnt - 1] = xstrdup (str);
|
|
|
|
|
vec[cnt] = NULL;
|
|
|
|
|
return vec;
|
|
|
|
|
}
|
2000-11-19 15:50:10 -05:00
|
|
|
|
|
|
|
|
|
/* Sometimes it's useful to create "sets" of strings, i.e. special
|
|
|
|
|
hash tables where you want to store strings as keys and merely
|
|
|
|
|
query for their existence. Here is a set of utility routines that
|
|
|
|
|
makes that transparent. */
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
string_set_add (struct hash_table *ht, const char *s)
|
|
|
|
|
{
|
2000-11-20 21:06:36 -05:00
|
|
|
|
/* First check whether the set element already exists. If it does,
|
|
|
|
|
do nothing so that we don't have to free() the old element and
|
|
|
|
|
then strdup() a new one. */
|
2001-05-12 16:06:41 -04:00
|
|
|
|
if (hash_table_contains (ht, s))
|
2000-11-20 21:06:36 -05:00
|
|
|
|
return;
|
|
|
|
|
|
2000-11-19 15:50:10 -05:00
|
|
|
|
/* We use "1" as value. It provides us a useful and clear arbitrary
|
|
|
|
|
value, and it consumes no memory -- the pointers to the same
|
2000-11-20 21:06:36 -05:00
|
|
|
|
string "1" will be shared by all the key-value pairs in all `set'
|
|
|
|
|
hash tables. */
|
2000-11-19 15:50:10 -05:00
|
|
|
|
hash_table_put (ht, xstrdup (s), "1");
|
|
|
|
|
}
|
|
|
|
|
|
2001-05-12 16:06:41 -04:00
|
|
|
|
/* Synonym for hash_table_contains... */
|
2000-11-20 21:06:36 -05:00
|
|
|
|
|
2000-11-19 15:50:10 -05:00
|
|
|
|
int
|
2001-05-12 16:06:41 -04:00
|
|
|
|
string_set_contains (struct hash_table *ht, const char *s)
|
2000-11-19 15:50:10 -05:00
|
|
|
|
{
|
2001-05-12 16:06:41 -04:00
|
|
|
|
return hash_table_contains (ht, s);
|
2000-11-19 15:50:10 -05:00
|
|
|
|
}
|
|
|
|
|
|
2005-03-30 14:27:34 -05:00
|
|
|
|
/* Convert the specified string set to array. ARRAY should be large
|
|
|
|
|
enough to hold hash_table_count(ht) char pointers. */
|
|
|
|
|
|
|
|
|
|
void string_set_to_array (struct hash_table *ht, char **array)
|
|
|
|
|
{
|
2005-08-27 09:05:39 -04:00
|
|
|
|
hash_table_iterator iter;
|
|
|
|
|
for (hash_table_iterate (ht, &iter); hash_table_iter_next (&iter); )
|
|
|
|
|
*array++ = iter.key;
|
2000-11-19 15:50:10 -05:00
|
|
|
|
}
|
|
|
|
|
|
2005-08-27 15:09:42 -04:00
|
|
|
|
/* Free the string set. This frees both the storage allocated for
|
|
|
|
|
keys and the actual hash table. (hash_table_destroy would only
|
|
|
|
|
destroy the hash table.) */
|
|
|
|
|
|
2000-11-19 15:50:10 -05:00
|
|
|
|
void
|
|
|
|
|
string_set_free (struct hash_table *ht)
|
|
|
|
|
{
|
2005-08-27 09:05:39 -04:00
|
|
|
|
hash_table_iterator iter;
|
|
|
|
|
for (hash_table_iterate (ht, &iter); hash_table_iter_next (&iter); )
|
|
|
|
|
xfree (iter.key);
|
2000-11-19 15:50:10 -05:00
|
|
|
|
hash_table_destroy (ht);
|
|
|
|
|
}
|
|
|
|
|
|
2005-08-27 15:09:42 -04:00
|
|
|
|
/* Utility function: simply call xfree() on all keys and values of HT. */
|
2000-11-19 15:50:10 -05:00
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
free_keys_and_values (struct hash_table *ht)
|
|
|
|
|
{
|
2005-08-27 09:05:39 -04:00
|
|
|
|
hash_table_iterator iter;
|
|
|
|
|
for (hash_table_iterate (ht, &iter); hash_table_iter_next (&iter); )
|
|
|
|
|
{
|
|
|
|
|
xfree (iter.key);
|
|
|
|
|
xfree (iter.value);
|
|
|
|
|
}
|
2000-11-19 15:50:10 -05:00
|
|
|
|
}
|
2000-11-01 16:51:25 -05:00
|
|
|
|
|
2005-08-10 08:40:31 -04:00
|
|
|
|
/* Get digit grouping data for thousand separors by calling
|
|
|
|
|
localeconv(). The data includes separator string and grouping info
|
|
|
|
|
and is cached after the first call to the function.
|
2005-06-26 22:07:31 -04:00
|
|
|
|
|
|
|
|
|
In locales that don't set a thousand separator (such as the "C"
|
2005-06-26 22:18:52 -04:00
|
|
|
|
locale), this forces it to be ",". We are now only showing
|
2005-06-26 22:07:31 -04:00
|
|
|
|
thousand separators in one place, so this shouldn't be a problem in
|
|
|
|
|
practice. */
|
|
|
|
|
|
2005-06-26 14:06:11 -04:00
|
|
|
|
static void
|
|
|
|
|
get_grouping_data (const char **sep, const char **grouping)
|
|
|
|
|
{
|
|
|
|
|
static const char *cached_sep;
|
|
|
|
|
static const char *cached_grouping;
|
|
|
|
|
static bool initialized;
|
|
|
|
|
if (!initialized)
|
|
|
|
|
{
|
|
|
|
|
/* Get the grouping info from the locale. */
|
2005-06-26 22:07:31 -04:00
|
|
|
|
struct lconv *lconv = localeconv ();
|
|
|
|
|
cached_sep = lconv->thousands_sep;
|
|
|
|
|
cached_grouping = lconv->grouping;
|
2008-02-06 21:10:24 -05:00
|
|
|
|
#if ! USE_NLS_PROGRESS_BAR
|
|
|
|
|
/* We can't count column widths, so ensure that the separator
|
|
|
|
|
* is single-byte only (let check below determine what byte). */
|
|
|
|
|
if (strlen(cached_sep) > 1)
|
|
|
|
|
cached_sep = "";
|
|
|
|
|
#endif
|
2005-06-26 22:07:31 -04:00
|
|
|
|
if (!*cached_sep)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
/* Many locales (such as "C" or "hr_HR") don't specify
|
|
|
|
|
grouping, which we still want to use it for legibility.
|
|
|
|
|
In those locales set the sep char to ',', unless that
|
|
|
|
|
character is used for decimal point, in which case set it
|
|
|
|
|
to ".". */
|
|
|
|
|
if (*lconv->decimal_point != ',')
|
|
|
|
|
cached_sep = ",";
|
|
|
|
|
else
|
|
|
|
|
cached_sep = ".";
|
|
|
|
|
cached_grouping = "\x03";
|
|
|
|
|
}
|
2005-06-26 14:06:11 -04:00
|
|
|
|
initialized = true;
|
|
|
|
|
}
|
|
|
|
|
*sep = cached_sep;
|
|
|
|
|
*grouping = cached_grouping;
|
|
|
|
|
}
|
|
|
|
|
|
2005-06-25 19:09:29 -04:00
|
|
|
|
/* Return a printed representation of N with thousand separators.
|
|
|
|
|
This should respect locale settings, with the exception of the "C"
|
|
|
|
|
locale which mandates no separator, but we use one anyway.
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2005-06-25 19:09:29 -04:00
|
|
|
|
Unfortunately, we cannot use %'d (in fact it would be %'j) to get
|
|
|
|
|
the separators because it's too non-portable, and it's hard to test
|
2005-08-10 08:40:31 -04:00
|
|
|
|
for this feature at configure time. Besides, it wouldn't display
|
|
|
|
|
separators in the "C" locale, still used by many Unix users. */
|
2005-06-25 19:09:29 -04:00
|
|
|
|
|
|
|
|
|
const char *
|
|
|
|
|
with_thousand_seps (wgint n)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2003-10-11 09:57:11 -04:00
|
|
|
|
static char outbuf[48];
|
2005-06-26 14:06:11 -04:00
|
|
|
|
char *p = outbuf + sizeof outbuf;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2005-06-26 14:06:11 -04:00
|
|
|
|
/* Info received from locale */
|
|
|
|
|
const char *grouping, *sep;
|
|
|
|
|
int seplen;
|
2003-10-11 09:57:11 -04:00
|
|
|
|
|
2005-06-26 14:06:11 -04:00
|
|
|
|
/* State information */
|
2005-06-25 19:09:29 -04:00
|
|
|
|
int i = 0, groupsize;
|
|
|
|
|
const char *atgroup;
|
|
|
|
|
|
2005-06-26 18:18:01 -04:00
|
|
|
|
bool negative = n < 0;
|
|
|
|
|
|
2005-06-26 14:06:11 -04:00
|
|
|
|
/* Initialize grouping data. */
|
|
|
|
|
get_grouping_data (&sep, &grouping);
|
|
|
|
|
seplen = strlen (sep);
|
|
|
|
|
atgroup = grouping;
|
2005-06-25 19:09:29 -04:00
|
|
|
|
groupsize = *atgroup++;
|
2003-10-11 09:57:11 -04:00
|
|
|
|
|
2005-08-10 08:40:31 -04:00
|
|
|
|
/* This would overflow on WGINT_MIN, but printing negative numbers
|
|
|
|
|
is not an important goal of this fuinction. */
|
2005-06-26 18:18:01 -04:00
|
|
|
|
if (negative)
|
|
|
|
|
n = -n;
|
|
|
|
|
|
2005-06-26 14:06:11 -04:00
|
|
|
|
/* Write the number into the buffer, backwards, inserting the
|
|
|
|
|
separators as necessary. */
|
|
|
|
|
*--p = '\0';
|
2005-06-25 19:09:29 -04:00
|
|
|
|
while (1)
|
|
|
|
|
{
|
|
|
|
|
*--p = n % 10 + '0';
|
|
|
|
|
n /= 10;
|
|
|
|
|
if (n == 0)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
break;
|
2005-06-26 14:06:11 -04:00
|
|
|
|
/* Prepend SEP to every groupsize'd digit and get new groupsize. */
|
2005-06-25 19:09:29 -04:00
|
|
|
|
if (++i == groupsize)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
if (seplen == 1)
|
|
|
|
|
*--p = *sep;
|
|
|
|
|
else
|
|
|
|
|
memcpy (p -= seplen, sep, seplen);
|
|
|
|
|
i = 0;
|
|
|
|
|
if (*atgroup)
|
|
|
|
|
groupsize = *atgroup++;
|
|
|
|
|
}
|
2005-06-25 19:09:29 -04:00
|
|
|
|
}
|
2005-06-26 18:18:01 -04:00
|
|
|
|
if (negative)
|
|
|
|
|
*--p = '-';
|
|
|
|
|
|
2005-06-25 19:09:29 -04:00
|
|
|
|
return p;
|
2005-03-22 08:20:02 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* N, a byte quantity, is converted to a human-readable abberviated
|
|
|
|
|
form a la sizes printed by `ls -lh'. The result is written to a
|
|
|
|
|
static buffer, a pointer to which is returned.
|
|
|
|
|
|
|
|
|
|
Unlike `with_thousand_seps', this approximates to the nearest unit.
|
|
|
|
|
Quoting GNU libit: "Most people visually process strings of 3-4
|
|
|
|
|
digits effectively, but longer strings of digits are more prone to
|
|
|
|
|
misinterpretation. Hence, converting to an abbreviated form
|
|
|
|
|
usually improves readability."
|
|
|
|
|
|
|
|
|
|
This intentionally uses kilobyte (KB), megabyte (MB), etc. in their
|
2005-08-10 08:28:01 -04:00
|
|
|
|
original computer-related meaning of "powers of 1024". We don't
|
|
|
|
|
use the "*bibyte" names invented in 1998, and seldom used in
|
|
|
|
|
practice. Wikipedia's entry on "binary prefix" discusses this in
|
|
|
|
|
some detail. */
|
2005-03-22 08:20:02 -05:00
|
|
|
|
|
|
|
|
|
char *
|
2005-06-25 19:09:29 -04:00
|
|
|
|
human_readable (HR_NUMTYPE n)
|
2005-03-22 08:20:02 -05:00
|
|
|
|
{
|
|
|
|
|
/* These suffixes are compatible with those of GNU `ls -lh'. */
|
|
|
|
|
static char powers[] =
|
|
|
|
|
{
|
2007-08-02 23:38:21 -04:00
|
|
|
|
'K', /* kilobyte, 2^10 bytes */
|
|
|
|
|
'M', /* megabyte, 2^20 bytes */
|
|
|
|
|
'G', /* gigabyte, 2^30 bytes */
|
|
|
|
|
'T', /* terabyte, 2^40 bytes */
|
|
|
|
|
'P', /* petabyte, 2^50 bytes */
|
|
|
|
|
'E', /* exabyte, 2^60 bytes */
|
2005-03-22 08:20:02 -05:00
|
|
|
|
};
|
|
|
|
|
static char buf[8];
|
2008-05-31 01:42:36 -04:00
|
|
|
|
size_t i;
|
2005-03-22 08:20:02 -05:00
|
|
|
|
|
|
|
|
|
/* If the quantity is smaller than 1K, just print it. */
|
|
|
|
|
if (n < 1024)
|
|
|
|
|
{
|
|
|
|
|
snprintf (buf, sizeof (buf), "%d", (int) n);
|
|
|
|
|
return buf;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Loop over powers, dividing N with 1024 in each iteration. This
|
|
|
|
|
works unchanged for all sizes of wgint, while still avoiding
|
|
|
|
|
non-portable `long double' arithmetic. */
|
|
|
|
|
for (i = 0; i < countof (powers); i++)
|
|
|
|
|
{
|
|
|
|
|
/* At each iteration N is greater than the *subsequent* power.
|
2007-08-02 23:38:21 -04:00
|
|
|
|
That way N/1024.0 produces a decimal number in the units of
|
|
|
|
|
*this* power. */
|
2005-06-25 17:16:51 -04:00
|
|
|
|
if ((n / 1024) < 1024 || i == countof (powers) - 1)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
double val = n / 1024.0;
|
|
|
|
|
/* Print values smaller than 10 with one decimal digits, and
|
|
|
|
|
others without any decimals. */
|
|
|
|
|
snprintf (buf, sizeof (buf), "%.*f%c",
|
|
|
|
|
val < 10 ? 1 : 0, val, powers[i]);
|
|
|
|
|
return buf;
|
|
|
|
|
}
|
2005-06-25 17:16:51 -04:00
|
|
|
|
n /= 1024;
|
2005-03-22 08:20:02 -05:00
|
|
|
|
}
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return NULL; /* unreached */
|
2000-11-01 16:51:25 -05:00
|
|
|
|
}
|
|
|
|
|
|
2005-03-29 15:03:42 -05:00
|
|
|
|
/* Count the digits in the provided number. Used to allocate space
|
|
|
|
|
when printing numbers. */
|
|
|
|
|
|
1999-12-02 02:42:23 -05:00
|
|
|
|
int
|
2005-02-23 17:21:04 -05:00
|
|
|
|
numdigit (wgint number)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2001-12-09 21:29:12 -05:00
|
|
|
|
int cnt = 1;
|
|
|
|
|
if (number < 0)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
++cnt; /* accomodate '-' */
|
2005-03-29 15:03:42 -05:00
|
|
|
|
while ((number /= 10) != 0)
|
2001-12-09 21:29:12 -05:00
|
|
|
|
++cnt;
|
|
|
|
|
return cnt;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
|
|
|
|
|
2005-04-01 13:22:38 -05:00
|
|
|
|
#define PR(mask) *p++ = n / (mask) + '0'
|
|
|
|
|
|
|
|
|
|
/* DIGITS_<D> is used to print a D-digit number and should be called
|
|
|
|
|
with mask==10^(D-1). It prints n/mask (the first digit), reducing
|
|
|
|
|
n to n%mask (the remaining digits), and calling DIGITS_<D-1>.
|
|
|
|
|
Recursively this continues until DIGITS_1 is invoked. */
|
|
|
|
|
|
|
|
|
|
#define DIGITS_1(mask) PR (mask)
|
|
|
|
|
#define DIGITS_2(mask) PR (mask), n %= (mask), DIGITS_1 ((mask) / 10)
|
|
|
|
|
#define DIGITS_3(mask) PR (mask), n %= (mask), DIGITS_2 ((mask) / 10)
|
|
|
|
|
#define DIGITS_4(mask) PR (mask), n %= (mask), DIGITS_3 ((mask) / 10)
|
|
|
|
|
#define DIGITS_5(mask) PR (mask), n %= (mask), DIGITS_4 ((mask) / 10)
|
|
|
|
|
#define DIGITS_6(mask) PR (mask), n %= (mask), DIGITS_5 ((mask) / 10)
|
|
|
|
|
#define DIGITS_7(mask) PR (mask), n %= (mask), DIGITS_6 ((mask) / 10)
|
|
|
|
|
#define DIGITS_8(mask) PR (mask), n %= (mask), DIGITS_7 ((mask) / 10)
|
|
|
|
|
#define DIGITS_9(mask) PR (mask), n %= (mask), DIGITS_8 ((mask) / 10)
|
|
|
|
|
#define DIGITS_10(mask) PR (mask), n %= (mask), DIGITS_9 ((mask) / 10)
|
|
|
|
|
|
|
|
|
|
/* DIGITS_<11-20> are only used on machines with 64-bit wgints. */
|
|
|
|
|
|
|
|
|
|
#define DIGITS_11(mask) PR (mask), n %= (mask), DIGITS_10 ((mask) / 10)
|
|
|
|
|
#define DIGITS_12(mask) PR (mask), n %= (mask), DIGITS_11 ((mask) / 10)
|
|
|
|
|
#define DIGITS_13(mask) PR (mask), n %= (mask), DIGITS_12 ((mask) / 10)
|
|
|
|
|
#define DIGITS_14(mask) PR (mask), n %= (mask), DIGITS_13 ((mask) / 10)
|
|
|
|
|
#define DIGITS_15(mask) PR (mask), n %= (mask), DIGITS_14 ((mask) / 10)
|
|
|
|
|
#define DIGITS_16(mask) PR (mask), n %= (mask), DIGITS_15 ((mask) / 10)
|
|
|
|
|
#define DIGITS_17(mask) PR (mask), n %= (mask), DIGITS_16 ((mask) / 10)
|
|
|
|
|
#define DIGITS_18(mask) PR (mask), n %= (mask), DIGITS_17 ((mask) / 10)
|
|
|
|
|
#define DIGITS_19(mask) PR (mask), n %= (mask), DIGITS_18 ((mask) / 10)
|
2005-02-23 17:21:04 -05:00
|
|
|
|
|
2005-04-01 13:22:38 -05:00
|
|
|
|
/* Shorthand for casting to wgint. */
|
|
|
|
|
#define W wgint
|
|
|
|
|
|
2005-02-23 17:21:04 -05:00
|
|
|
|
/* Print NUMBER to BUFFER in base 10. This is equivalent to
|
2005-03-29 15:03:42 -05:00
|
|
|
|
`sprintf(buffer, "%lld", (long long) number)', only typically much
|
|
|
|
|
faster and portable to machines without long long.
|
2001-04-23 22:06:34 -04:00
|
|
|
|
|
|
|
|
|
The speedup may make a difference in programs that frequently
|
|
|
|
|
convert numbers to strings. Some implementations of sprintf,
|
2006-02-09 05:24:07 -05:00
|
|
|
|
particularly the one in some versions of GNU libc, have been known
|
|
|
|
|
to be quite slow when converting integers to strings.
|
2001-04-23 22:06:34 -04:00
|
|
|
|
|
2001-12-09 21:29:12 -05:00
|
|
|
|
Return the pointer to the location where the terminating zero was
|
|
|
|
|
printed. (Equivalent to calling buffer+strlen(buffer) after the
|
|
|
|
|
function is done.)
|
2000-11-01 16:51:25 -05:00
|
|
|
|
|
2006-02-09 05:24:07 -05:00
|
|
|
|
BUFFER should be large enough to accept as many bytes as you expect
|
|
|
|
|
the number to take up. On machines with 64-bit wgints the maximum
|
2001-12-09 21:29:12 -05:00
|
|
|
|
needed size is 24 bytes. That includes the digits needed for the
|
|
|
|
|
largest 64-bit number, the `-' sign in case it's negative, and the
|
|
|
|
|
terminating '\0'. */
|
|
|
|
|
|
|
|
|
|
char *
|
2005-02-23 17:21:04 -05:00
|
|
|
|
number_to_string (char *buffer, wgint number)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2000-11-01 16:51:25 -05:00
|
|
|
|
char *p = buffer;
|
2005-02-23 17:21:04 -05:00
|
|
|
|
wgint n = number;
|
2001-04-23 22:06:34 -04:00
|
|
|
|
|
2006-02-03 05:27:59 -05:00
|
|
|
|
int last_digit_char = 0;
|
|
|
|
|
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#if (SIZEOF_WGINT != 4) && (SIZEOF_WGINT != 8)
|
2006-02-03 05:31:00 -05:00
|
|
|
|
/* We are running in a very strange environment. Leave the correct
|
|
|
|
|
printing to sprintf. */
|
|
|
|
|
p += sprintf (buf, "%j", (intmax_t) (n));
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#else /* (SIZEOF_WGINT == 4) || (SIZEOF_WGINT == 8) */
|
1999-12-02 02:42:23 -05:00
|
|
|
|
|
2001-04-23 22:06:34 -04:00
|
|
|
|
if (n < 0)
|
1999-12-02 02:42:23 -05:00
|
|
|
|
{
|
2005-02-23 17:21:04 -05:00
|
|
|
|
if (n < -WGINT_MAX)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
/* n = -n would overflow because -n would evaluate to a
|
2006-02-03 05:27:59 -05:00
|
|
|
|
wgint value larger than WGINT_MAX. Need to make n
|
|
|
|
|
smaller and handle the last digit separately. */
|
|
|
|
|
int last_digit = n % 10;
|
|
|
|
|
/* The sign of n%10 is implementation-defined. */
|
|
|
|
|
if (last_digit < 0)
|
|
|
|
|
last_digit_char = '0' - last_digit;
|
|
|
|
|
else
|
|
|
|
|
last_digit_char = '0' + last_digit;
|
|
|
|
|
/* After n is made smaller, -n will not overflow. */
|
|
|
|
|
n /= 10;
|
2007-08-02 23:38:21 -04:00
|
|
|
|
}
|
2002-04-14 15:18:33 -04:00
|
|
|
|
|
2000-11-01 16:51:25 -05:00
|
|
|
|
*p++ = '-';
|
2001-04-23 22:06:34 -04:00
|
|
|
|
n = -n;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2000-11-01 16:51:25 -05:00
|
|
|
|
|
2005-04-01 13:22:38 -05:00
|
|
|
|
/* Use the DIGITS_ macro appropriate for N's number of digits. That
|
|
|
|
|
way printing any N is fully open-coded without a loop or jump.
|
|
|
|
|
(Also see description of DIGITS_*.) */
|
|
|
|
|
|
|
|
|
|
if (n < 10) DIGITS_1 (1);
|
|
|
|
|
else if (n < 100) DIGITS_2 (10);
|
|
|
|
|
else if (n < 1000) DIGITS_3 (100);
|
|
|
|
|
else if (n < 10000) DIGITS_4 (1000);
|
|
|
|
|
else if (n < 100000) DIGITS_5 (10000);
|
|
|
|
|
else if (n < 1000000) DIGITS_6 (100000);
|
|
|
|
|
else if (n < 10000000) DIGITS_7 (1000000);
|
|
|
|
|
else if (n < 100000000) DIGITS_8 (10000000);
|
|
|
|
|
else if (n < 1000000000) DIGITS_9 (100000000);
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#if SIZEOF_WGINT == 4
|
2005-04-01 13:22:38 -05:00
|
|
|
|
/* wgint is 32 bits wide: no number has more than 10 digits. */
|
|
|
|
|
else DIGITS_10 (1000000000);
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#else
|
2005-06-19 18:34:58 -04:00
|
|
|
|
/* wgint is 64 bits wide: handle numbers with 9-19 decimal digits.
|
|
|
|
|
Constants are constructed by compile-time multiplication to avoid
|
|
|
|
|
dealing with different notations for 64-bit constants
|
|
|
|
|
(nL/nLL/nI64, depending on the compiler and architecture). */
|
2005-04-01 13:22:38 -05:00
|
|
|
|
else if (n < 10*(W)1000000000) DIGITS_10 (1000000000);
|
|
|
|
|
else if (n < 100*(W)1000000000) DIGITS_11 (10*(W)1000000000);
|
|
|
|
|
else if (n < 1000*(W)1000000000) DIGITS_12 (100*(W)1000000000);
|
|
|
|
|
else if (n < 10000*(W)1000000000) DIGITS_13 (1000*(W)1000000000);
|
|
|
|
|
else if (n < 100000*(W)1000000000) DIGITS_14 (10000*(W)1000000000);
|
|
|
|
|
else if (n < 1000000*(W)1000000000) DIGITS_15 (100000*(W)1000000000);
|
|
|
|
|
else if (n < 10000000*(W)1000000000) DIGITS_16 (1000000*(W)1000000000);
|
|
|
|
|
else if (n < 100000000*(W)1000000000) DIGITS_17 (10000000*(W)1000000000);
|
|
|
|
|
else if (n < 1000000000*(W)1000000000) DIGITS_18 (100000000*(W)1000000000);
|
|
|
|
|
else DIGITS_19 (1000000000*(W)1000000000);
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#endif
|
2001-04-23 22:06:34 -04:00
|
|
|
|
|
2006-02-03 05:27:59 -05:00
|
|
|
|
if (last_digit_char)
|
|
|
|
|
*p++ = last_digit_char;
|
|
|
|
|
|
2000-11-01 16:51:25 -05:00
|
|
|
|
*p = '\0';
|
2005-02-23 17:21:04 -05:00
|
|
|
|
#endif /* (SIZEOF_WGINT == 4) || (SIZEOF_WGINT == 8) */
|
2001-12-09 21:29:12 -05:00
|
|
|
|
|
|
|
|
|
return p;
|
1999-12-02 02:42:23 -05:00
|
|
|
|
}
|
2001-04-23 22:06:34 -04:00
|
|
|
|
|
2005-04-01 13:22:38 -05:00
|
|
|
|
#undef PR
|
|
|
|
|
#undef W
|
2005-06-27 14:19:22 -04:00
|
|
|
|
#undef SPRINTF_WGINT
|
2001-04-23 22:06:34 -04:00
|
|
|
|
#undef DIGITS_1
|
|
|
|
|
#undef DIGITS_2
|
|
|
|
|
#undef DIGITS_3
|
|
|
|
|
#undef DIGITS_4
|
|
|
|
|
#undef DIGITS_5
|
|
|
|
|
#undef DIGITS_6
|
|
|
|
|
#undef DIGITS_7
|
|
|
|
|
#undef DIGITS_8
|
|
|
|
|
#undef DIGITS_9
|
|
|
|
|
#undef DIGITS_10
|
|
|
|
|
#undef DIGITS_11
|
|
|
|
|
#undef DIGITS_12
|
|
|
|
|
#undef DIGITS_13
|
|
|
|
|
#undef DIGITS_14
|
|
|
|
|
#undef DIGITS_15
|
|
|
|
|
#undef DIGITS_16
|
|
|
|
|
#undef DIGITS_17
|
|
|
|
|
#undef DIGITS_18
|
|
|
|
|
#undef DIGITS_19
|
2005-02-23 17:21:04 -05:00
|
|
|
|
|
|
|
|
|
#define RING_SIZE 3
|
|
|
|
|
|
|
|
|
|
/* Print NUMBER to a statically allocated string and return a pointer
|
|
|
|
|
to the printed representation.
|
|
|
|
|
|
|
|
|
|
This function is intended to be used in conjunction with printf.
|
|
|
|
|
It is hard to portably print wgint values:
|
|
|
|
|
a) you cannot use printf("%ld", number) because wgint can be long
|
|
|
|
|
long on 32-bit machines with LFS.
|
|
|
|
|
b) you cannot use printf("%lld", number) because NUMBER could be
|
|
|
|
|
long on 32-bit machines without LFS, or on 64-bit machines,
|
|
|
|
|
which do not require LFS. Also, Windows doesn't support %lld.
|
|
|
|
|
c) you cannot use printf("%j", (int_max_t) number) because not all
|
|
|
|
|
versions of printf support "%j", the most notable being the one
|
|
|
|
|
on Windows.
|
|
|
|
|
d) you cannot #define WGINT_FMT to the appropriate format and use
|
|
|
|
|
printf(WGINT_FMT, number) because that would break translations
|
|
|
|
|
for user-visible messages, such as printf("Downloaded: %d
|
|
|
|
|
bytes\n", number).
|
|
|
|
|
|
|
|
|
|
What you should use instead is printf("%s", number_to_static_string
|
|
|
|
|
(number)).
|
|
|
|
|
|
|
|
|
|
CAVEAT: since the function returns pointers to static data, you
|
|
|
|
|
must be careful to copy its result before calling it again.
|
|
|
|
|
However, to make it more useful with printf, the function maintains
|
|
|
|
|
an internal ring of static buffers to return. That way things like
|
|
|
|
|
printf("%s %s", number_to_static_string (num1),
|
|
|
|
|
number_to_static_string (num2)) work as expected. Three buffers
|
|
|
|
|
are currently used, which means that "%s %s %s" will work, but "%s
|
|
|
|
|
%s %s %s" won't. If you need to print more than three wgints,
|
|
|
|
|
bump the RING_SIZE (or rethink your message.) */
|
|
|
|
|
|
|
|
|
|
char *
|
|
|
|
|
number_to_static_string (wgint number)
|
|
|
|
|
{
|
|
|
|
|
static char ring[RING_SIZE][24];
|
|
|
|
|
static int ringpos;
|
|
|
|
|
char *buf = ring[ringpos];
|
|
|
|
|
number_to_string (buf, number);
|
|
|
|
|
ringpos = (ringpos + 1) % RING_SIZE;
|
|
|
|
|
return buf;
|
|
|
|
|
}
|
2000-11-22 11:58:28 -05:00
|
|
|
|
|
2001-11-22 23:59:52 -05:00
|
|
|
|
/* Determine the width of the terminal we're running on. If that's
|
|
|
|
|
not possible, return 0. */
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
determine_screen_width (void)
|
|
|
|
|
{
|
|
|
|
|
/* If there's a way to get the terminal size using POSIX
|
|
|
|
|
tcgetattr(), somebody please tell me. */
|
2004-01-28 08:42:52 -05:00
|
|
|
|
#ifdef TIOCGWINSZ
|
2001-11-22 23:59:52 -05:00
|
|
|
|
int fd;
|
|
|
|
|
struct winsize wsz;
|
|
|
|
|
|
|
|
|
|
if (opt.lfilename != NULL)
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
|
|
fd = fileno (stderr);
|
|
|
|
|
if (ioctl (fd, TIOCGWINSZ, &wsz) < 0)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return 0; /* most likely ENOTTY */
|
2001-11-22 23:59:52 -05:00
|
|
|
|
|
|
|
|
|
return wsz.ws_col;
|
2005-06-19 21:37:23 -04:00
|
|
|
|
#elif defined(WINDOWS)
|
2004-01-28 08:42:52 -05:00
|
|
|
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
2004-01-29 07:38:52 -05:00
|
|
|
|
if (!GetConsoleScreenBufferInfo (GetStdHandle (STD_ERROR_HANDLE), &csbi))
|
2004-01-28 08:42:52 -05:00
|
|
|
|
return 0;
|
|
|
|
|
return csbi.dwSize.X;
|
2005-06-19 21:37:23 -04:00
|
|
|
|
#else /* neither TIOCGWINSZ nor WINDOWS */
|
2004-01-29 07:38:52 -05:00
|
|
|
|
return 0;
|
2005-06-19 21:37:23 -04:00
|
|
|
|
#endif /* neither TIOCGWINSZ nor WINDOWS */
|
2001-11-22 23:59:52 -05:00
|
|
|
|
}
|
2005-06-28 18:22:10 -04:00
|
|
|
|
|
|
|
|
|
/* Whether the rnd system (either rand or [dl]rand48) has been
|
|
|
|
|
seeded. */
|
|
|
|
|
static int rnd_seeded;
|
2001-11-29 09:57:30 -05:00
|
|
|
|
|
2001-12-17 09:05:08 -05:00
|
|
|
|
/* Return a random number between 0 and MAX-1, inclusive.
|
|
|
|
|
|
2005-06-28 18:22:10 -04:00
|
|
|
|
If the system does not support lrand48 and MAX is greater than the
|
|
|
|
|
value of RAND_MAX+1 on the system, the returned value will be in
|
|
|
|
|
the range [0, RAND_MAX]. This may be fixed in a future release.
|
2001-12-17 09:05:08 -05:00
|
|
|
|
The random number generator is seeded automatically the first time
|
|
|
|
|
it is called.
|
|
|
|
|
|
2005-06-28 18:22:10 -04:00
|
|
|
|
This uses lrand48 where available, rand elsewhere. DO NOT use it
|
|
|
|
|
for cryptography. It is only meant to be used in situations where
|
|
|
|
|
quality of the random numbers returned doesn't really matter. */
|
2001-12-17 09:05:08 -05:00
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
random_number (int max)
|
|
|
|
|
{
|
2005-06-28 18:22:10 -04:00
|
|
|
|
#ifdef HAVE_DRAND48
|
|
|
|
|
if (!rnd_seeded)
|
|
|
|
|
{
|
|
|
|
|
srand48 ((long) time (NULL) ^ (long) getpid ());
|
|
|
|
|
rnd_seeded = 1;
|
|
|
|
|
}
|
|
|
|
|
return lrand48 () % max;
|
|
|
|
|
#else /* not HAVE_DRAND48 */
|
|
|
|
|
|
2001-12-17 09:05:08 -05:00
|
|
|
|
double bounded;
|
|
|
|
|
int rnd;
|
2005-06-28 18:22:10 -04:00
|
|
|
|
if (!rnd_seeded)
|
2001-12-17 09:05:08 -05:00
|
|
|
|
{
|
2005-06-28 18:22:10 -04:00
|
|
|
|
srand ((unsigned) time (NULL) ^ (unsigned) getpid ());
|
|
|
|
|
rnd_seeded = 1;
|
2001-12-17 09:05:08 -05:00
|
|
|
|
}
|
|
|
|
|
rnd = rand ();
|
|
|
|
|
|
2005-06-28 18:22:10 -04:00
|
|
|
|
/* Like rand() % max, but uses the high-order bits for better
|
|
|
|
|
randomness on architectures where rand() is implemented using a
|
|
|
|
|
simple congruential generator. */
|
2001-12-17 09:05:08 -05:00
|
|
|
|
|
2005-06-28 18:22:10 -04:00
|
|
|
|
bounded = (double) max * rnd / (RAND_MAX + 1.0);
|
|
|
|
|
return (int) bounded;
|
2001-12-17 09:05:08 -05:00
|
|
|
|
|
2005-06-28 18:22:10 -04:00
|
|
|
|
#endif /* not HAVE_DRAND48 */
|
2001-12-17 09:05:08 -05:00
|
|
|
|
}
|
|
|
|
|
|
2003-09-20 19:12:18 -04:00
|
|
|
|
/* Return a random uniformly distributed floating point number in the
|
2005-06-28 18:22:10 -04:00
|
|
|
|
[0, 1) range. Uses drand48 where available, and a really lame
|
|
|
|
|
kludge elsewhere. */
|
2003-09-20 19:12:18 -04:00
|
|
|
|
|
|
|
|
|
double
|
|
|
|
|
random_float (void)
|
|
|
|
|
{
|
2005-06-28 18:22:10 -04:00
|
|
|
|
#ifdef HAVE_DRAND48
|
|
|
|
|
if (!rnd_seeded)
|
|
|
|
|
{
|
|
|
|
|
srand48 ((long) time (NULL) ^ (long) getpid ());
|
|
|
|
|
rnd_seeded = 1;
|
|
|
|
|
}
|
|
|
|
|
return drand48 ();
|
|
|
|
|
#else /* not HAVE_DRAND48 */
|
|
|
|
|
return ( random_number (10000) / 10000.0
|
2007-08-02 23:38:21 -04:00
|
|
|
|
+ random_number (10000) / (10000.0 * 10000.0)
|
|
|
|
|
+ random_number (10000) / (10000.0 * 10000.0 * 10000.0)
|
|
|
|
|
+ random_number (10000) / (10000.0 * 10000.0 * 10000.0 * 10000.0));
|
2005-06-28 18:22:10 -04:00
|
|
|
|
#endif /* not HAVE_DRAND48 */
|
2003-09-20 19:12:18 -04:00
|
|
|
|
}
|
2002-04-13 18:18:28 -04:00
|
|
|
|
|
2003-10-02 11:15:36 -04:00
|
|
|
|
/* Implementation of run_with_timeout, a generic timeout-forcing
|
|
|
|
|
routine for systems with Unix-like signal handling. */
|
|
|
|
|
|
2002-04-15 15:01:56 -04:00
|
|
|
|
#ifdef USE_SIGNAL_TIMEOUT
|
|
|
|
|
# ifdef HAVE_SIGSETJMP
|
|
|
|
|
# define SETJMP(env) sigsetjmp (env, 1)
|
2002-04-13 18:18:28 -04:00
|
|
|
|
|
|
|
|
|
static sigjmp_buf run_with_timeout_env;
|
|
|
|
|
|
2005-06-19 19:03:27 -04:00
|
|
|
|
static void
|
2002-04-13 18:18:28 -04:00
|
|
|
|
abort_run_with_timeout (int sig)
|
|
|
|
|
{
|
|
|
|
|
assert (sig == SIGALRM);
|
|
|
|
|
siglongjmp (run_with_timeout_env, -1);
|
|
|
|
|
}
|
2002-04-15 15:01:56 -04:00
|
|
|
|
# else /* not HAVE_SIGSETJMP */
|
|
|
|
|
# define SETJMP(env) setjmp (env)
|
2002-04-13 18:18:28 -04:00
|
|
|
|
|
|
|
|
|
static jmp_buf run_with_timeout_env;
|
|
|
|
|
|
2005-06-19 19:03:27 -04:00
|
|
|
|
static void
|
2002-04-13 18:18:28 -04:00
|
|
|
|
abort_run_with_timeout (int sig)
|
|
|
|
|
{
|
|
|
|
|
assert (sig == SIGALRM);
|
|
|
|
|
/* We don't have siglongjmp to preserve the set of blocked signals;
|
|
|
|
|
if we longjumped out of the handler at this point, SIGALRM would
|
|
|
|
|
remain blocked. We must unblock it manually. */
|
|
|
|
|
int mask = siggetmask ();
|
2003-10-04 18:58:34 -04:00
|
|
|
|
mask &= ~sigmask (SIGALRM);
|
2002-04-13 18:18:28 -04:00
|
|
|
|
sigsetmask (mask);
|
|
|
|
|
|
|
|
|
|
/* Now it's safe to longjump. */
|
|
|
|
|
longjmp (run_with_timeout_env, -1);
|
|
|
|
|
}
|
2002-04-15 15:01:56 -04:00
|
|
|
|
# endif /* not HAVE_SIGSETJMP */
|
2003-10-02 09:37:45 -04:00
|
|
|
|
|
|
|
|
|
/* Arrange for SIGALRM to be delivered in TIMEOUT seconds. This uses
|
|
|
|
|
setitimer where available, alarm otherwise.
|
|
|
|
|
|
|
|
|
|
TIMEOUT should be non-zero. If the timeout value is so small that
|
|
|
|
|
it would be rounded to zero, it is rounded to the least legal value
|
|
|
|
|
instead (1us for setitimer, 1s for alarm). That ensures that
|
|
|
|
|
SIGALRM will be delivered in all cases. */
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
|
alarm_set (double timeout)
|
|
|
|
|
{
|
|
|
|
|
#ifdef ITIMER_REAL
|
|
|
|
|
/* Use the modern itimer interface. */
|
|
|
|
|
struct itimerval itv;
|
2003-10-31 09:55:50 -05:00
|
|
|
|
xzero (itv);
|
2003-10-02 09:37:45 -04:00
|
|
|
|
itv.it_value.tv_sec = (long) timeout;
|
2005-04-01 13:22:38 -05:00
|
|
|
|
itv.it_value.tv_usec = 1000000 * (timeout - (long)timeout);
|
2003-10-02 09:37:45 -04:00
|
|
|
|
if (itv.it_value.tv_sec == 0 && itv.it_value.tv_usec == 0)
|
|
|
|
|
/* Ensure that we wait for at least the minimum interval.
|
|
|
|
|
Specifying zero would mean "wait forever". */
|
|
|
|
|
itv.it_value.tv_usec = 1;
|
|
|
|
|
setitimer (ITIMER_REAL, &itv, NULL);
|
|
|
|
|
#else /* not ITIMER_REAL */
|
|
|
|
|
/* Use the old alarm() interface. */
|
|
|
|
|
int secs = (int) timeout;
|
|
|
|
|
if (secs == 0)
|
|
|
|
|
/* Round TIMEOUTs smaller than 1 to 1, not to zero. This is
|
|
|
|
|
because alarm(0) means "never deliver the alarm", i.e. "wait
|
|
|
|
|
forever", which is not what someone who specifies a 0.5s
|
|
|
|
|
timeout would expect. */
|
|
|
|
|
secs = 1;
|
|
|
|
|
alarm (secs);
|
|
|
|
|
#endif /* not ITIMER_REAL */
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Cancel the alarm set with alarm_set. */
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
|
alarm_cancel (void)
|
|
|
|
|
{
|
|
|
|
|
#ifdef ITIMER_REAL
|
|
|
|
|
struct itimerval disable;
|
2003-10-31 09:55:50 -05:00
|
|
|
|
xzero (disable);
|
2003-10-02 09:37:45 -04:00
|
|
|
|
setitimer (ITIMER_REAL, &disable, NULL);
|
|
|
|
|
#else /* not ITIMER_REAL */
|
|
|
|
|
alarm (0);
|
|
|
|
|
#endif /* not ITIMER_REAL */
|
|
|
|
|
}
|
|
|
|
|
|
2003-10-04 18:58:34 -04:00
|
|
|
|
/* Call FUN(ARG), but don't allow it to run for more than TIMEOUT
|
2005-06-22 15:38:10 -04:00
|
|
|
|
seconds. Returns true if the function was interrupted with a
|
|
|
|
|
timeout, false otherwise.
|
2003-10-02 11:15:36 -04:00
|
|
|
|
|
|
|
|
|
This works by setting up SIGALRM to be delivered in TIMEOUT seconds
|
|
|
|
|
using setitimer() or alarm(). The timeout is enforced by
|
|
|
|
|
longjumping out of the SIGALRM handler. This has several
|
|
|
|
|
advantages compared to the traditional approach of relying on
|
|
|
|
|
signals causing system calls to exit with EINTR:
|
|
|
|
|
|
|
|
|
|
* The callback function is *forcibly* interrupted after the
|
|
|
|
|
timeout expires, (almost) regardless of what it was doing and
|
|
|
|
|
whether it was in a syscall. For example, a calculation that
|
|
|
|
|
takes a long time is interrupted as reliably as an IO
|
|
|
|
|
operation.
|
|
|
|
|
|
|
|
|
|
* It works with both SYSV and BSD signals because it doesn't
|
|
|
|
|
depend on the default setting of SA_RESTART.
|
|
|
|
|
|
2005-04-01 13:22:38 -05:00
|
|
|
|
* It doesn't require special handler setup beyond a simple call
|
|
|
|
|
to signal(). (It does use sigsetjmp/siglongjmp, but they're
|
2003-10-02 11:15:36 -04:00
|
|
|
|
optional.)
|
|
|
|
|
|
|
|
|
|
The only downside is that, if FUN allocates internal resources that
|
|
|
|
|
are normally freed prior to exit from the functions, they will be
|
|
|
|
|
lost in case of timeout. */
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
bool
|
2003-09-20 19:12:18 -04:00
|
|
|
|
run_with_timeout (double timeout, void (*fun) (void *), void *arg)
|
2002-04-13 18:18:28 -04:00
|
|
|
|
{
|
|
|
|
|
int saved_errno;
|
|
|
|
|
|
|
|
|
|
if (timeout == 0)
|
|
|
|
|
{
|
|
|
|
|
fun (arg);
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
2002-04-13 18:18:28 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
signal (SIGALRM, abort_run_with_timeout);
|
|
|
|
|
if (SETJMP (run_with_timeout_env) != 0)
|
|
|
|
|
{
|
|
|
|
|
/* Longjumped out of FUN with a timeout. */
|
|
|
|
|
signal (SIGALRM, SIG_DFL);
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return true;
|
2002-04-13 18:18:28 -04:00
|
|
|
|
}
|
2003-10-02 09:37:45 -04:00
|
|
|
|
alarm_set (timeout);
|
2002-04-13 18:18:28 -04:00
|
|
|
|
fun (arg);
|
|
|
|
|
|
|
|
|
|
/* Preserve errno in case alarm() or signal() modifies it. */
|
|
|
|
|
saved_errno = errno;
|
2003-10-02 09:37:45 -04:00
|
|
|
|
alarm_cancel ();
|
2002-04-13 18:18:28 -04:00
|
|
|
|
signal (SIGALRM, SIG_DFL);
|
|
|
|
|
errno = saved_errno;
|
|
|
|
|
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
2002-04-13 18:18:28 -04:00
|
|
|
|
}
|
2003-10-02 19:28:49 -04:00
|
|
|
|
|
|
|
|
|
#else /* not USE_SIGNAL_TIMEOUT */
|
|
|
|
|
|
|
|
|
|
#ifndef WINDOWS
|
|
|
|
|
/* A stub version of run_with_timeout that just calls FUN(ARG). Don't
|
|
|
|
|
define it under Windows, because Windows has its own version of
|
|
|
|
|
run_with_timeout that uses threads. */
|
|
|
|
|
|
2007-08-27 16:27:30 -04:00
|
|
|
|
bool
|
2003-10-02 19:28:49 -04:00
|
|
|
|
run_with_timeout (double timeout, void (*fun) (void *), void *arg)
|
|
|
|
|
{
|
|
|
|
|
fun (arg);
|
2005-06-22 15:38:10 -04:00
|
|
|
|
return false;
|
2003-10-02 19:28:49 -04:00
|
|
|
|
}
|
|
|
|
|
#endif /* not WINDOWS */
|
|
|
|
|
#endif /* not USE_SIGNAL_TIMEOUT */
|
2003-11-03 16:57:04 -05:00
|
|
|
|
|
|
|
|
|
#ifndef WINDOWS
|
|
|
|
|
|
|
|
|
|
/* Sleep the specified amount of seconds. On machines without
|
|
|
|
|
nanosleep(), this may sleep shorter if interrupted by signals. */
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
xsleep (double seconds)
|
|
|
|
|
{
|
|
|
|
|
#ifdef HAVE_NANOSLEEP
|
|
|
|
|
/* nanosleep is the preferred interface because it offers high
|
|
|
|
|
accuracy and, more importantly, because it allows us to reliably
|
2005-03-20 05:41:46 -05:00
|
|
|
|
restart receiving a signal such as SIGWINCH. (There was an
|
|
|
|
|
actual Debian bug report about --limit-rate malfunctioning while
|
|
|
|
|
the terminal was being resized.) */
|
2003-11-03 16:57:04 -05:00
|
|
|
|
struct timespec sleep, remaining;
|
|
|
|
|
sleep.tv_sec = (long) seconds;
|
2005-04-01 13:22:38 -05:00
|
|
|
|
sleep.tv_nsec = 1000000000 * (seconds - (long) seconds);
|
2003-11-03 16:57:04 -05:00
|
|
|
|
while (nanosleep (&sleep, &remaining) < 0 && errno == EINTR)
|
|
|
|
|
/* If nanosleep has been interrupted by a signal, adjust the
|
|
|
|
|
sleeping period and return to sleep. */
|
|
|
|
|
sleep = remaining;
|
2005-06-19 21:37:23 -04:00
|
|
|
|
#elif defined(HAVE_USLEEP)
|
2003-11-03 16:57:04 -05:00
|
|
|
|
/* If usleep is available, use it in preference to select. */
|
2004-01-24 22:16:33 -05:00
|
|
|
|
if (seconds >= 1)
|
2003-11-03 16:57:04 -05:00
|
|
|
|
{
|
2004-01-24 22:16:33 -05:00
|
|
|
|
/* On some systems, usleep cannot handle values larger than
|
2007-08-02 23:38:21 -04:00
|
|
|
|
1,000,000. If the period is larger than that, use sleep
|
|
|
|
|
first, then add usleep for subsecond accuracy. */
|
2003-11-03 16:57:04 -05:00
|
|
|
|
sleep (seconds);
|
|
|
|
|
seconds -= (long) seconds;
|
|
|
|
|
}
|
2005-04-01 13:22:38 -05:00
|
|
|
|
usleep (seconds * 1000000);
|
2005-06-21 21:01:12 -04:00
|
|
|
|
#else /* fall back select */
|
|
|
|
|
/* Note that, although Windows supports select, it can't be used to
|
|
|
|
|
implement sleeping because Winsock's select doesn't implement
|
|
|
|
|
timeout when it is passed NULL pointers for all fd sets. (But it
|
|
|
|
|
does under Cygwin, which implements Unix-compatible select.) */
|
2003-11-03 16:57:04 -05:00
|
|
|
|
struct timeval sleep;
|
|
|
|
|
sleep.tv_sec = (long) seconds;
|
2005-04-01 13:22:38 -05:00
|
|
|
|
sleep.tv_usec = 1000000 * (seconds - (long) seconds);
|
2003-11-03 16:57:04 -05:00
|
|
|
|
select (0, NULL, NULL, NULL, &sleep);
|
|
|
|
|
/* If select returns -1 and errno is EINTR, it means we were
|
|
|
|
|
interrupted by a signal. But without knowing how long we've
|
|
|
|
|
actually slept, we can't return to sleep. Using gettimeofday to
|
|
|
|
|
track sleeps is slow and unreliable due to clock skew. */
|
2005-06-19 21:37:23 -04:00
|
|
|
|
#endif
|
2003-11-03 16:57:04 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif /* not WINDOWS */
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2006-06-20 04:16:36 -04:00
|
|
|
|
/* Encode the octets in DATA of length LENGTH to base64 format,
|
|
|
|
|
storing the result to DEST. The output will be zero-terminated,
|
|
|
|
|
and must point to a writable buffer of at least
|
|
|
|
|
1+BASE64_LENGTH(length) bytes. The function returns the length of
|
|
|
|
|
the resulting base64 data, not counting the terminating zero.
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2006-06-20 04:16:36 -04:00
|
|
|
|
This implementation does not emit newlines after 76 characters of
|
2005-04-18 10:23:23 -04:00
|
|
|
|
base64 data. */
|
|
|
|
|
|
|
|
|
|
int
|
2006-06-20 04:16:36 -04:00
|
|
|
|
base64_encode (const void *data, int length, char *dest)
|
2005-04-06 16:42:22 -04:00
|
|
|
|
{
|
|
|
|
|
/* Conversion table. */
|
2006-06-20 04:16:36 -04:00
|
|
|
|
static const char tbl[64] = {
|
2006-06-21 04:32:55 -04:00
|
|
|
|
'A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P',
|
|
|
|
|
'Q','R','S','T','U','V','W','X','Y','Z','a','b','c','d','e','f',
|
|
|
|
|
'g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v',
|
|
|
|
|
'w','x','y','z','0','1','2','3','4','5','6','7','8','9','+','/'
|
2005-04-06 16:42:22 -04:00
|
|
|
|
};
|
2006-06-21 04:32:55 -04:00
|
|
|
|
/* Access bytes in DATA as unsigned char, otherwise the shifts below
|
|
|
|
|
don't work for data with MSB set. */
|
2006-06-20 04:16:36 -04:00
|
|
|
|
const unsigned char *s = data;
|
|
|
|
|
/* Theoretical ANSI violation when length < 3. */
|
2006-06-21 04:12:12 -04:00
|
|
|
|
const unsigned char *end = (const unsigned char *) data + length - 2;
|
2006-06-20 04:16:36 -04:00
|
|
|
|
char *p = dest;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
|
|
|
|
/* Transform the 3x8 bits to 4x6 bits, as required by base64. */
|
2006-06-19 16:35:53 -04:00
|
|
|
|
for (; s < end; s += 3)
|
2005-04-06 16:42:22 -04:00
|
|
|
|
{
|
|
|
|
|
*p++ = tbl[s[0] >> 2];
|
|
|
|
|
*p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
|
|
|
|
|
*p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
|
|
|
|
|
*p++ = tbl[s[2] & 0x3f];
|
|
|
|
|
}
|
2005-04-18 10:23:23 -04:00
|
|
|
|
|
2005-04-06 16:42:22 -04:00
|
|
|
|
/* Pad the result if necessary... */
|
2006-06-19 16:35:53 -04:00
|
|
|
|
switch (length % 3)
|
|
|
|
|
{
|
|
|
|
|
case 1:
|
|
|
|
|
*p++ = tbl[s[0] >> 2];
|
|
|
|
|
*p++ = tbl[(s[0] & 3) << 4];
|
|
|
|
|
*p++ = '=';
|
|
|
|
|
*p++ = '=';
|
|
|
|
|
break;
|
|
|
|
|
case 2:
|
|
|
|
|
*p++ = tbl[s[0] >> 2];
|
|
|
|
|
*p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
|
|
|
|
|
*p++ = tbl[((s[1] & 0xf) << 2)];
|
|
|
|
|
*p++ = '=';
|
|
|
|
|
break;
|
|
|
|
|
}
|
2005-04-06 16:42:22 -04:00
|
|
|
|
/* ...and zero-terminate it. */
|
|
|
|
|
*p = '\0';
|
2005-04-18 10:23:23 -04:00
|
|
|
|
|
2006-06-20 04:16:36 -04:00
|
|
|
|
return p - dest;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
}
|
|
|
|
|
|
2005-07-04 10:46:09 -04:00
|
|
|
|
/* Store in C the next non-whitespace character from the string, or \0
|
|
|
|
|
when end of string is reached. */
|
2007-08-02 23:38:21 -04:00
|
|
|
|
#define NEXT_CHAR(c, p) do { \
|
|
|
|
|
c = (unsigned char) *p++; \
|
2007-10-14 17:46:24 -04:00
|
|
|
|
} while (c_isspace (c))
|
2005-07-04 10:46:09 -04:00
|
|
|
|
|
|
|
|
|
#define IS_ASCII(c) (((c) & 0x80) == 0)
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2006-06-25 12:22:26 -04:00
|
|
|
|
/* Decode data from BASE64 (a null-terminated string) into memory
|
|
|
|
|
pointed to by DEST. DEST is assumed to be large enough to
|
|
|
|
|
accomodate the decoded data, which is guaranteed to be no more than
|
|
|
|
|
3/4*strlen(base64).
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2006-06-20 04:16:36 -04:00
|
|
|
|
Since DEST is assumed to contain binary data, it is not
|
2005-04-06 16:42:22 -04:00
|
|
|
|
NUL-terminated. The function returns the length of the data
|
|
|
|
|
written to TO. -1 is returned in case of error caused by malformed
|
2006-06-25 12:22:26 -04:00
|
|
|
|
base64 input.
|
|
|
|
|
|
|
|
|
|
This function originates from Free Recode. */
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
|
|
|
|
int
|
2006-06-20 04:16:36 -04:00
|
|
|
|
base64_decode (const char *base64, void *dest)
|
2005-04-06 16:42:22 -04:00
|
|
|
|
{
|
2005-04-18 10:23:23 -04:00
|
|
|
|
/* Table of base64 values for first 128 characters. Note that this
|
|
|
|
|
assumes ASCII (but so does Wget in other places). */
|
2006-06-20 04:18:59 -04:00
|
|
|
|
static const signed char base64_char_to_value[128] =
|
2005-04-06 16:42:22 -04:00
|
|
|
|
{
|
2007-08-02 23:38:21 -04:00
|
|
|
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 0- 9 */
|
|
|
|
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 10- 19 */
|
|
|
|
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 20- 29 */
|
|
|
|
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 30- 39 */
|
|
|
|
|
-1, -1, -1, 62, -1, -1, -1, 63, 52, 53, /* 40- 49 */
|
|
|
|
|
54, 55, 56, 57, 58, 59, 60, 61, -1, -1, /* 50- 59 */
|
|
|
|
|
-1, -1, -1, -1, -1, 0, 1, 2, 3, 4, /* 60- 69 */
|
|
|
|
|
5, 6, 7, 8, 9, 10, 11, 12, 13, 14, /* 70- 79 */
|
|
|
|
|
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, /* 80- 89 */
|
|
|
|
|
25, -1, -1, -1, -1, -1, -1, 26, 27, 28, /* 90- 99 */
|
|
|
|
|
29, 30, 31, 32, 33, 34, 35, 36, 37, 38, /* 100-109 */
|
|
|
|
|
39, 40, 41, 42, 43, 44, 45, 46, 47, 48, /* 110-119 */
|
|
|
|
|
49, 50, 51, -1, -1, -1, -1, -1 /* 120-127 */
|
2005-04-06 16:42:22 -04:00
|
|
|
|
};
|
2005-07-04 09:28:19 -04:00
|
|
|
|
#define BASE64_CHAR_TO_VALUE(c) ((int) base64_char_to_value[c])
|
2005-07-04 10:46:09 -04:00
|
|
|
|
#define IS_BASE64(c) ((IS_ASCII (c) && BASE64_CHAR_TO_VALUE (c) >= 0) || c == '=')
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
|
|
|
|
const char *p = base64;
|
2006-06-20 04:16:36 -04:00
|
|
|
|
char *q = dest;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
|
|
|
|
while (1)
|
|
|
|
|
{
|
2005-07-04 10:46:09 -04:00
|
|
|
|
unsigned char c;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
unsigned long value;
|
|
|
|
|
|
|
|
|
|
/* Process first byte of a quadruplet. */
|
2005-07-04 10:46:09 -04:00
|
|
|
|
NEXT_CHAR (c, p);
|
2005-04-06 16:42:22 -04:00
|
|
|
|
if (!c)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
break;
|
2005-07-04 10:46:09 -04:00
|
|
|
|
if (c == '=' || !IS_BASE64 (c))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* illegal char while decoding base64 */
|
2005-07-04 09:28:19 -04:00
|
|
|
|
value = BASE64_CHAR_TO_VALUE (c) << 18;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2005-07-04 10:46:09 -04:00
|
|
|
|
/* Process second byte of a quadruplet. */
|
|
|
|
|
NEXT_CHAR (c, p);
|
2005-04-06 16:42:22 -04:00
|
|
|
|
if (!c)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* premature EOF while decoding base64 */
|
2005-07-04 10:46:09 -04:00
|
|
|
|
if (c == '=' || !IS_BASE64 (c))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* illegal char while decoding base64 */
|
2005-07-04 09:28:19 -04:00
|
|
|
|
value |= BASE64_CHAR_TO_VALUE (c) << 12;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
*q++ = value >> 16;
|
|
|
|
|
|
|
|
|
|
/* Process third byte of a quadruplet. */
|
2005-07-04 10:46:09 -04:00
|
|
|
|
NEXT_CHAR (c, p);
|
2005-04-06 16:42:22 -04:00
|
|
|
|
if (!c)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* premature EOF while decoding base64 */
|
2005-07-04 10:46:09 -04:00
|
|
|
|
if (!IS_BASE64 (c))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* illegal char while decoding base64 */
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
|
|
|
|
if (c == '=')
|
2007-08-02 23:38:21 -04:00
|
|
|
|
{
|
|
|
|
|
NEXT_CHAR (c, p);
|
|
|
|
|
if (!c)
|
|
|
|
|
return -1; /* premature EOF while decoding base64 */
|
|
|
|
|
if (c != '=')
|
|
|
|
|
return -1; /* padding `=' expected but not found */
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2005-07-04 09:28:19 -04:00
|
|
|
|
value |= BASE64_CHAR_TO_VALUE (c) << 6;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
*q++ = 0xff & value >> 8;
|
|
|
|
|
|
|
|
|
|
/* Process fourth byte of a quadruplet. */
|
2005-07-04 10:46:09 -04:00
|
|
|
|
NEXT_CHAR (c, p);
|
2005-04-06 16:42:22 -04:00
|
|
|
|
if (!c)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* premature EOF while decoding base64 */
|
2005-04-06 16:42:22 -04:00
|
|
|
|
if (c == '=')
|
2007-08-02 23:38:21 -04:00
|
|
|
|
continue;
|
2005-07-04 10:46:09 -04:00
|
|
|
|
if (!IS_BASE64 (c))
|
2007-08-02 23:38:21 -04:00
|
|
|
|
return -1; /* illegal char while decoding base64 */
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2005-07-04 09:28:19 -04:00
|
|
|
|
value |= BASE64_CHAR_TO_VALUE (c);
|
2005-04-06 16:42:22 -04:00
|
|
|
|
*q++ = 0xff & value;
|
|
|
|
|
}
|
2005-07-04 10:46:09 -04:00
|
|
|
|
#undef IS_BASE64
|
2005-07-04 09:28:19 -04:00
|
|
|
|
#undef BASE64_CHAR_TO_VALUE
|
2005-04-06 16:42:22 -04:00
|
|
|
|
|
2006-06-20 04:16:36 -04:00
|
|
|
|
return q - (char *) dest;
|
2005-04-06 16:42:22 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#undef IS_ASCII
|
2005-07-04 10:48:19 -04:00
|
|
|
|
#undef NEXT_CHAR
|
2005-04-24 16:00:19 -04:00
|
|
|
|
|
|
|
|
|
/* Simple merge sort for use by stable_sort. Implementation courtesy
|
2005-05-06 07:03:21 -04:00
|
|
|
|
Zeljko Vrba with additional debugging by Nenad Barbutov. */
|
2005-04-24 16:00:19 -04:00
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
|
mergesort_internal (void *base, void *temp, size_t size, size_t from, size_t to,
|
2007-08-02 23:38:21 -04:00
|
|
|
|
int (*cmpfun) (const void *, const void *))
|
2005-04-24 16:00:19 -04:00
|
|
|
|
{
|
|
|
|
|
#define ELT(array, pos) ((char *)(array) + (pos) * size)
|
|
|
|
|
if (from < to)
|
|
|
|
|
{
|
|
|
|
|
size_t i, j, k;
|
|
|
|
|
size_t mid = (to + from) / 2;
|
|
|
|
|
mergesort_internal (base, temp, size, from, mid, cmpfun);
|
|
|
|
|
mergesort_internal (base, temp, size, mid + 1, to, cmpfun);
|
|
|
|
|
i = from;
|
|
|
|
|
j = mid + 1;
|
|
|
|
|
for (k = from; (i <= mid) && (j <= to); k++)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
if (cmpfun (ELT (base, i), ELT (base, j)) <= 0)
|
|
|
|
|
memcpy (ELT (temp, k), ELT (base, i++), size);
|
|
|
|
|
else
|
|
|
|
|
memcpy (ELT (temp, k), ELT (base, j++), size);
|
2005-04-24 16:00:19 -04:00
|
|
|
|
while (i <= mid)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
memcpy (ELT (temp, k++), ELT (base, i++), size);
|
2005-04-24 16:00:19 -04:00
|
|
|
|
while (j <= to)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
memcpy (ELT (temp, k++), ELT (base, j++), size);
|
2005-04-24 16:00:19 -04:00
|
|
|
|
for (k = from; k <= to; k++)
|
2007-08-02 23:38:21 -04:00
|
|
|
|
memcpy (ELT (base, k), ELT (temp, k), size);
|
2005-04-24 16:00:19 -04:00
|
|
|
|
}
|
|
|
|
|
#undef ELT
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Stable sort with interface exactly like standard library's qsort.
|
|
|
|
|
Uses mergesort internally, allocating temporary storage with
|
|
|
|
|
alloca. */
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
|
stable_sort (void *base, size_t nmemb, size_t size,
|
2007-08-02 23:38:21 -04:00
|
|
|
|
int (*cmpfun) (const void *, const void *))
|
2005-04-24 16:00:19 -04:00
|
|
|
|
{
|
|
|
|
|
if (size > 1)
|
|
|
|
|
{
|
|
|
|
|
void *temp = alloca (nmemb * size * sizeof (void *));
|
|
|
|
|
mergesort_internal (base, temp, size, 0, nmemb - 1, cmpfun);
|
|
|
|
|
}
|
|
|
|
|
}
|
2005-06-28 18:03:12 -04:00
|
|
|
|
|
|
|
|
|
/* Print a decimal number. If it is equal to or larger than ten, the
|
|
|
|
|
number is rounded. Otherwise it is printed with one significant
|
|
|
|
|
digit without trailing zeros and with no more than three fractional
|
|
|
|
|
digits total. For example, 0.1 is printed as "0.1", 0.035 is
|
|
|
|
|
printed as "0.04", 0.0091 as "0.009", and 0.0003 as simply "0".
|
|
|
|
|
|
|
|
|
|
This is useful for displaying durations because it provides
|
|
|
|
|
order-of-magnitude information without unnecessary clutter --
|
|
|
|
|
long-running downloads are shown without the fractional part, and
|
|
|
|
|
short ones still retain one significant digit. */
|
|
|
|
|
|
|
|
|
|
const char *
|
|
|
|
|
print_decimal (double number)
|
|
|
|
|
{
|
|
|
|
|
static char buf[32];
|
|
|
|
|
double n = number >= 0 ? number : -number;
|
|
|
|
|
|
|
|
|
|
if (n >= 9.95)
|
|
|
|
|
/* Cut off at 9.95 because the below %.1f would round 9.96 to
|
|
|
|
|
"10.0" instead of "10". OTOH 9.94 will print as "9.9". */
|
|
|
|
|
snprintf (buf, sizeof buf, "%.0f", number);
|
|
|
|
|
else if (n >= 0.95)
|
|
|
|
|
snprintf (buf, sizeof buf, "%.1f", number);
|
|
|
|
|
else if (n >= 0.001)
|
|
|
|
|
snprintf (buf, sizeof buf, "%.1g", number);
|
|
|
|
|
else if (n >= 0.0005)
|
|
|
|
|
/* round [0.0005, 0.001) to 0.001 */
|
|
|
|
|
snprintf (buf, sizeof buf, "%.3f", number);
|
|
|
|
|
else
|
|
|
|
|
/* print numbers close to 0 as 0, not 0.000 */
|
|
|
|
|
strcpy (buf, "0");
|
|
|
|
|
|
|
|
|
|
return buf;
|
|
|
|
|
}
|
2006-03-15 09:55:29 -05:00
|
|
|
|
|
|
|
|
|
#ifdef TESTING
|
|
|
|
|
|
|
|
|
|
const char *
|
|
|
|
|
test_subdir_p()
|
|
|
|
|
{
|
|
|
|
|
int i;
|
|
|
|
|
struct {
|
|
|
|
|
char *d1;
|
|
|
|
|
char *d2;
|
|
|
|
|
bool result;
|
|
|
|
|
} test_array[] = {
|
|
|
|
|
{ "/somedir", "/somedir", true },
|
|
|
|
|
{ "/somedir", "/somedir/d2", true },
|
|
|
|
|
{ "/somedir/d1", "/somedir", false },
|
|
|
|
|
};
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
|
|
|
|
for (i = 0; i < countof(test_array); ++i)
|
2006-03-15 09:55:29 -05:00
|
|
|
|
{
|
|
|
|
|
bool res = subdir_p (test_array[i].d1, test_array[i].d2);
|
|
|
|
|
|
2009-09-21 23:39:44 -04:00
|
|
|
|
mu_assert ("test_subdir_p: wrong result",
|
2006-03-15 09:55:29 -05:00
|
|
|
|
res == test_array[i].result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const char *
|
|
|
|
|
test_dir_matches_p()
|
|
|
|
|
{
|
|
|
|
|
int i;
|
|
|
|
|
struct {
|
|
|
|
|
char *dirlist[3];
|
|
|
|
|
char *dir;
|
|
|
|
|
bool result;
|
|
|
|
|
} test_array[] = {
|
|
|
|
|
{ { "/somedir", "/someotherdir", NULL }, "somedir", true },
|
|
|
|
|
{ { "/somedir", "/someotherdir", NULL }, "anotherdir", false },
|
|
|
|
|
{ { "/somedir", "/*otherdir", NULL }, "anotherdir", true },
|
|
|
|
|
{ { "/somedir/d1", "/someotherdir", NULL }, "somedir/d1", true },
|
2008-03-19 18:48:54 -04:00
|
|
|
|
{ { "*/*d1", "/someotherdir", NULL }, "somedir/d1", true },
|
2006-03-15 09:55:29 -05:00
|
|
|
|
{ { "/somedir/d1", "/someotherdir", NULL }, "d1", false },
|
2008-03-19 18:48:54 -04:00
|
|
|
|
{ { "!COMPLETE", NULL, NULL }, "!COMPLETE", true },
|
|
|
|
|
{ { "*COMPLETE", NULL, NULL }, "!COMPLETE", true },
|
|
|
|
|
{ { "*/!COMPLETE", NULL, NULL }, "foo/!COMPLETE", true },
|
|
|
|
|
{ { "*COMPLETE", NULL, NULL }, "foo/!COMPLETE", false },
|
|
|
|
|
{ { "*/*COMPLETE", NULL, NULL }, "foo/!COMPLETE", true },
|
|
|
|
|
{ { "/dir with spaces", NULL, NULL }, "dir with spaces", true },
|
|
|
|
|
{ { "/dir*with*spaces", NULL, NULL }, "dir with spaces", true },
|
2008-04-23 03:05:27 -04:00
|
|
|
|
{ { "/Tmp/has", NULL, NULL }, "/Tmp/has space", false },
|
|
|
|
|
{ { "/Tmp/has", NULL, NULL }, "/Tmp/has,comma", false },
|
2006-03-15 09:55:29 -05:00
|
|
|
|
};
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
|
|
|
|
for (i = 0; i < countof(test_array); ++i)
|
2006-03-15 09:55:29 -05:00
|
|
|
|
{
|
|
|
|
|
bool res = dir_matches_p (test_array[i].dirlist, test_array[i].dir);
|
2009-09-21 23:39:44 -04:00
|
|
|
|
|
|
|
|
|
mu_assert ("test_dir_matches_p: wrong result",
|
2006-03-15 09:55:29 -05:00
|
|
|
|
res == test_array[i].result);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif /* TESTING */
|
|
|
|
|
|