mirror of
https://github.com/git/git.git
synced 2024-11-27 12:03:55 +08:00
568a05c5ec
Calculating the sum of two array indexes to find the midpoint between
them can overflow, i.e. code like this is unsafe for big arrays:
mid = (first + last) >> 1;
Make sure the intermediate value stays within the boundaries instead,
like this:
mid = first + ((last - first) >> 1);
The loop condition of the binary search makes sure that 'last' is
always greater than 'first', so this is safe as long as 'first' is
not negative. And that can be verified easily using the pre-context
of each change, except for name-hash.c, so add an assertion to that
effect there.
The unsafe calculations were found with:
git grep '(.*+.*) *>> *1'
This is a continuation of 19716b21a4 (cleanup: fix possible overflow
errors in binary search, 2017-10-08).
Signed-off-by: Rene Scharfe <l.s.r@web.de>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
428 lines
10 KiB
C
428 lines
10 KiB
C
/*
|
|
* sh-i18n--envsubst.c - a stripped-down version of gettext's envsubst(1)
|
|
*
|
|
* Copyright (C) 2010 Ævar Arnfjörð Bjarmason
|
|
*
|
|
* This is a modified version of
|
|
* 67d0871a8c:gettext-runtime/src/envsubst.c from the gettext.git
|
|
* repository. It has been stripped down to only implement the
|
|
* envsubst(1) features that we need in the git-sh-i18n fallbacks.
|
|
*
|
|
* The "Close standard error" part in main() is from
|
|
* 8dac033df0:gnulib-local/lib/closeout.c. The copyright notices for
|
|
* both files are reproduced immediately below.
|
|
*/
|
|
|
|
#include "git-compat-util.h"
|
|
#include "trace2.h"
|
|
|
|
/* Substitution of environment variables in shell format strings.
|
|
Copyright (C) 2003-2007 Free Software Foundation, Inc.
|
|
Written by Bruno Haible <bruno@clisp.org>, 2003.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2, or (at your option)
|
|
any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
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
|
|
along with this program; if not, see <http://www.gnu.org/licenses/>. */
|
|
|
|
/* closeout.c - close standard output and standard error
|
|
Copyright (C) 1998-2007 Free Software Foundation, Inc.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2, or (at your option)
|
|
any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
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
|
|
along with this program; if not, see <http://www.gnu.org/licenses/>. */
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
/* If true, substitution shall be performed on all variables. */
|
|
static unsigned short int all_variables;
|
|
|
|
/* Forward declaration of local functions. */
|
|
static void print_variables (const char *string);
|
|
static void note_variables (const char *string);
|
|
static void subst_from_stdin (void);
|
|
|
|
int
|
|
cmd_main (int argc, const char *argv[])
|
|
{
|
|
/* Default values for command line options. */
|
|
/* unsigned short int show_variables = 0; */
|
|
|
|
trace2_cmd_name("sh-i18n--envsubst");
|
|
|
|
switch (argc)
|
|
{
|
|
case 1:
|
|
error ("we won't substitute all variables on stdin for you");
|
|
break;
|
|
/*
|
|
all_variables = 1;
|
|
subst_from_stdin ();
|
|
*/
|
|
case 2:
|
|
/* echo '$foo and $bar' | git sh-i18n--envsubst --variables '$foo and $bar' */
|
|
all_variables = 0;
|
|
note_variables (argv[1]);
|
|
subst_from_stdin ();
|
|
break;
|
|
case 3:
|
|
/* git sh-i18n--envsubst --variables '$foo and $bar' */
|
|
if (strcmp(argv[1], "--variables"))
|
|
error ("first argument must be --variables when two are given");
|
|
/* show_variables = 1; */
|
|
print_variables (argv[2]);
|
|
break;
|
|
default:
|
|
error ("too many arguments");
|
|
break;
|
|
}
|
|
|
|
/* Close standard error. This is simpler than fwriteerror_no_ebadf, because
|
|
upon failure we don't need an errno - all we can do at this point is to
|
|
set an exit status. */
|
|
errno = 0;
|
|
if (ferror (stderr) || fflush (stderr))
|
|
{
|
|
fclose (stderr);
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
if (fclose (stderr) && errno != EBADF)
|
|
exit (EXIT_FAILURE);
|
|
|
|
exit (EXIT_SUCCESS);
|
|
}
|
|
|
|
/* Parse the string and invoke the callback each time a $VARIABLE or
|
|
${VARIABLE} construct is seen, where VARIABLE is a nonempty sequence
|
|
of ASCII alphanumeric/underscore characters, starting with an ASCII
|
|
alphabetic/underscore character.
|
|
We allow only ASCII characters, to avoid dependencies w.r.t. the current
|
|
encoding: While "${\xe0}" looks like a variable access in ISO-8859-1
|
|
encoding, it doesn't look like one in the BIG5, BIG5-HKSCS, GBK, GB18030,
|
|
SHIFT_JIS, JOHAB encodings, because \xe0\x7d is a single character in these
|
|
encodings. */
|
|
static void
|
|
find_variables (const char *string,
|
|
void (*callback) (const char *var_ptr, size_t var_len))
|
|
{
|
|
for (; *string != '\0';)
|
|
if (*string++ == '$')
|
|
{
|
|
const char *variable_start;
|
|
const char *variable_end;
|
|
unsigned short int valid;
|
|
char c;
|
|
|
|
if (*string == '{')
|
|
string++;
|
|
|
|
variable_start = string;
|
|
c = *string;
|
|
if ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || c == '_')
|
|
{
|
|
do
|
|
c = *++string;
|
|
while ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z')
|
|
|| (c >= '0' && c <= '9') || c == '_');
|
|
variable_end = string;
|
|
|
|
if (variable_start[-1] == '{')
|
|
{
|
|
if (*string == '}')
|
|
{
|
|
string++;
|
|
valid = 1;
|
|
}
|
|
else
|
|
valid = 0;
|
|
}
|
|
else
|
|
valid = 1;
|
|
|
|
if (valid)
|
|
callback (variable_start, variable_end - variable_start);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/* Print a variable to stdout, followed by a newline. */
|
|
static void
|
|
print_variable (const char *var_ptr, size_t var_len)
|
|
{
|
|
fwrite (var_ptr, var_len, 1, stdout);
|
|
putchar ('\n');
|
|
}
|
|
|
|
/* Print the variables contained in STRING to stdout, each one followed by a
|
|
newline. */
|
|
static void
|
|
print_variables (const char *string)
|
|
{
|
|
find_variables (string, &print_variable);
|
|
}
|
|
|
|
|
|
/* Type describing list of immutable strings,
|
|
implemented using a dynamic array. */
|
|
typedef struct string_list_ty string_list_ty;
|
|
struct string_list_ty
|
|
{
|
|
const char **item;
|
|
size_t nitems;
|
|
size_t nitems_max;
|
|
};
|
|
|
|
/* Initialize an empty list of strings. */
|
|
static inline void
|
|
string_list_init (string_list_ty *slp)
|
|
{
|
|
slp->item = NULL;
|
|
slp->nitems = 0;
|
|
slp->nitems_max = 0;
|
|
}
|
|
|
|
/* Append a single string to the end of a list of strings. */
|
|
static inline void
|
|
string_list_append (string_list_ty *slp, const char *s)
|
|
{
|
|
/* Grow the list. */
|
|
if (slp->nitems >= slp->nitems_max)
|
|
{
|
|
slp->nitems_max = slp->nitems_max * 2 + 4;
|
|
REALLOC_ARRAY(slp->item, slp->nitems_max);
|
|
}
|
|
|
|
/* Add the string to the end of the list. */
|
|
slp->item[slp->nitems++] = s;
|
|
}
|
|
|
|
/* Compare two strings given by reference. */
|
|
static int
|
|
cmp_string (const void *pstr1, const void *pstr2)
|
|
{
|
|
const char *str1 = *(const char **)pstr1;
|
|
const char *str2 = *(const char **)pstr2;
|
|
|
|
return strcmp (str1, str2);
|
|
}
|
|
|
|
/* Sort a list of strings. */
|
|
static inline void
|
|
string_list_sort (string_list_ty *slp)
|
|
{
|
|
QSORT(slp->item, slp->nitems, cmp_string);
|
|
}
|
|
|
|
/* Test whether a sorted string list contains a given string. */
|
|
static int
|
|
sorted_string_list_member (const string_list_ty *slp, const char *s)
|
|
{
|
|
size_t j1, j2;
|
|
|
|
j1 = 0;
|
|
j2 = slp->nitems;
|
|
if (j2 > 0)
|
|
{
|
|
/* Binary search. */
|
|
while (j2 - j1 > 1)
|
|
{
|
|
/* Here we know that if s is in the list, it is at an index j
|
|
with j1 <= j < j2. */
|
|
size_t j = j1 + ((j2 - j1) >> 1);
|
|
int result = strcmp (slp->item[j], s);
|
|
|
|
if (result > 0)
|
|
j2 = j;
|
|
else if (result == 0)
|
|
return 1;
|
|
else
|
|
j1 = j + 1;
|
|
}
|
|
if (j2 > j1)
|
|
if (strcmp (slp->item[j1], s) == 0)
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
/* Set of variables on which to perform substitution.
|
|
Used only if !all_variables. */
|
|
static string_list_ty variables_set;
|
|
|
|
/* Adds a variable to variables_set. */
|
|
static void
|
|
note_variable (const char *var_ptr, size_t var_len)
|
|
{
|
|
char *string = xmemdupz (var_ptr, var_len);
|
|
|
|
string_list_append (&variables_set, string);
|
|
}
|
|
|
|
/* Stores the variables occurring in the string in variables_set. */
|
|
static void
|
|
note_variables (const char *string)
|
|
{
|
|
string_list_init (&variables_set);
|
|
find_variables (string, ¬e_variable);
|
|
string_list_sort (&variables_set);
|
|
}
|
|
|
|
|
|
static int
|
|
do_getc (void)
|
|
{
|
|
int c = getc (stdin);
|
|
|
|
if (c == EOF)
|
|
{
|
|
if (ferror (stdin))
|
|
error ("error while reading standard input");
|
|
}
|
|
|
|
return c;
|
|
}
|
|
|
|
static inline void
|
|
do_ungetc (int c)
|
|
{
|
|
if (c != EOF)
|
|
ungetc (c, stdin);
|
|
}
|
|
|
|
/* Copies stdin to stdout, performing substitutions. */
|
|
static void
|
|
subst_from_stdin (void)
|
|
{
|
|
static char *buffer;
|
|
static size_t bufmax;
|
|
static size_t buflen;
|
|
int c;
|
|
|
|
for (;;)
|
|
{
|
|
c = do_getc ();
|
|
if (c == EOF)
|
|
break;
|
|
/* Look for $VARIABLE or ${VARIABLE}. */
|
|
if (c == '$')
|
|
{
|
|
unsigned short int opening_brace = 0;
|
|
unsigned short int closing_brace = 0;
|
|
|
|
c = do_getc ();
|
|
if (c == '{')
|
|
{
|
|
opening_brace = 1;
|
|
c = do_getc ();
|
|
}
|
|
if ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || c == '_')
|
|
{
|
|
unsigned short int valid;
|
|
|
|
/* Accumulate the VARIABLE in buffer. */
|
|
buflen = 0;
|
|
do
|
|
{
|
|
if (buflen >= bufmax)
|
|
{
|
|
bufmax = 2 * bufmax + 10;
|
|
buffer = xrealloc (buffer, bufmax);
|
|
}
|
|
buffer[buflen++] = c;
|
|
|
|
c = do_getc ();
|
|
}
|
|
while ((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z')
|
|
|| (c >= '0' && c <= '9') || c == '_');
|
|
|
|
if (opening_brace)
|
|
{
|
|
if (c == '}')
|
|
{
|
|
closing_brace = 1;
|
|
valid = 1;
|
|
}
|
|
else
|
|
{
|
|
valid = 0;
|
|
do_ungetc (c);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
valid = 1;
|
|
do_ungetc (c);
|
|
}
|
|
|
|
if (valid)
|
|
{
|
|
/* Terminate the variable in the buffer. */
|
|
if (buflen >= bufmax)
|
|
{
|
|
bufmax = 2 * bufmax + 10;
|
|
buffer = xrealloc (buffer, bufmax);
|
|
}
|
|
buffer[buflen] = '\0';
|
|
|
|
/* Test whether the variable shall be substituted. */
|
|
if (!all_variables
|
|
&& !sorted_string_list_member (&variables_set, buffer))
|
|
valid = 0;
|
|
}
|
|
|
|
if (valid)
|
|
{
|
|
/* Substitute the variable's value from the environment. */
|
|
const char *env_value = getenv (buffer);
|
|
|
|
if (env_value != NULL)
|
|
fputs (env_value, stdout);
|
|
}
|
|
else
|
|
{
|
|
/* Perform no substitution at all. Since the buffered input
|
|
contains no other '$' than at the start, we can just
|
|
output all the buffered contents. */
|
|
putchar ('$');
|
|
if (opening_brace)
|
|
putchar ('{');
|
|
fwrite (buffer, buflen, 1, stdout);
|
|
if (closing_brace)
|
|
putchar ('}');
|
|
}
|
|
}
|
|
else
|
|
{
|
|
do_ungetc (c);
|
|
putchar ('$');
|
|
if (opening_brace)
|
|
putchar ('{');
|
|
}
|
|
}
|
|
else
|
|
putchar (c);
|
|
}
|
|
}
|