mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
* ruby.c (get_arglen): skip the last terminator of argv before
checking environ. * ruby.c (get_arglen): duplicate environ area if setenv and unsetenv are provided. * ruby.c (set_arg0): keep empty strings. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_8@27468 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
fcecd66b98
commit
720160f364
2 changed files with 27 additions and 29 deletions
10
ChangeLog
10
ChangeLog
|
@ -1,3 +1,13 @@
|
|||
Sat Apr 24 13:06:57 2010 Nobuyoshi Nakada <nobu@ruby-lang.org>
|
||||
|
||||
* ruby.c (get_arglen): skip the last terminator of argv before
|
||||
checking environ.
|
||||
|
||||
* ruby.c (get_arglen): duplicate environ area if setenv and unsetenv
|
||||
are provided.
|
||||
|
||||
* ruby.c (set_arg0): keep empty strings.
|
||||
|
||||
Sat Apr 24 09:44:40 2010 Nobuyoshi Nakada <nobu@ruby-lang.org>
|
||||
|
||||
* ext/syck/yaml2byte.c (bytestring_append, bytestring_extend):
|
||||
|
|
46
ruby.c
46
ruby.c
|
@ -1039,36 +1039,12 @@ VALUE rb_argv0;
|
|||
|
||||
#if defined(PSTAT_SETCMD) || defined(HAVE_SETPROCTITLE)
|
||||
#elif defined(_WIN32)
|
||||
#elif defined(HAVE_SETENV) && defined(HAVE_UNSETENV)
|
||||
#else
|
||||
#define USE_ENVSPACE_FOR_ARG0
|
||||
#endif
|
||||
|
||||
#ifdef USE_ENVSPACE_FOR_ARG0
|
||||
static struct {
|
||||
char *begin, *end;
|
||||
} envspace;
|
||||
extern char **environ;
|
||||
|
||||
static void
|
||||
set_arg0space()
|
||||
{
|
||||
char *s;
|
||||
int i;
|
||||
|
||||
if (!environ || (s = environ[0]) == NULL) return;
|
||||
envspace.begin = s;
|
||||
s += strlen(s);
|
||||
for (i = 1; environ[i]; i++) {
|
||||
if (environ[i] == s + 1) {
|
||||
s++;
|
||||
s += strlen(s); /* this one is ok too */
|
||||
}
|
||||
}
|
||||
envspace.end = s;
|
||||
}
|
||||
#else
|
||||
#define set_arg0space() ((void)0)
|
||||
#endif
|
||||
|
||||
static int
|
||||
|
@ -1090,7 +1066,8 @@ get_arglen(int argc, char **argv)
|
|||
}
|
||||
}
|
||||
#if defined(USE_ENVSPACE_FOR_ARG0)
|
||||
if (environ && (s == environ[0])) {
|
||||
if (environ && (s+1 == environ[0])) {
|
||||
s++;
|
||||
s += strlen(s);
|
||||
for (i = 1; environ[i]; i++) {
|
||||
if (environ[i] == s + 1) {
|
||||
|
@ -1098,7 +1075,19 @@ get_arglen(int argc, char **argv)
|
|||
s += strlen(s); /* this one is ok too */
|
||||
}
|
||||
}
|
||||
# if defined(HAVE_SETENV) && defined(HAVE_UNSETENV)
|
||||
{
|
||||
char *t = malloc(s - environ[0] + 1);
|
||||
for (i = 0; environ[i]; i++) {
|
||||
size_t len = strlen(environ[i]) + 1;
|
||||
memcpy(t, environ[i], len);
|
||||
environ[i] = t;
|
||||
t += len;
|
||||
}
|
||||
}
|
||||
# else
|
||||
ruby_setenv("", NULL); /* duplicate environ vars */
|
||||
# endif
|
||||
}
|
||||
#endif
|
||||
return s - argv[0];
|
||||
|
@ -1144,13 +1133,13 @@ set_arg0(val, id)
|
|||
len = get_arglen(origargc, origargv);
|
||||
}
|
||||
|
||||
if (i >= len) {
|
||||
i = len;
|
||||
if (i > len - origargc) {
|
||||
i = len - origargc;
|
||||
}
|
||||
memcpy(origargv[0], s, i);
|
||||
s = origargv[0] + i;
|
||||
*s = '\0';
|
||||
if (++i < len) memset(s + 1, ' ', len - i);
|
||||
if (++i < len) memset(s + 1, '\0', len - i);
|
||||
for (i = len-1, j = origargc-1; j > 0 && i >= 0; --i, --j) {
|
||||
origargv[j] = origargv[0] + i;
|
||||
*origargv[j] = '\0';
|
||||
|
@ -1274,7 +1263,6 @@ ruby_process_options(argc, argv)
|
|||
#if defined(USE_DLN_A_OUT)
|
||||
dln_argv0 = argv[0];
|
||||
#endif
|
||||
set_arg0space();
|
||||
proc_options(argc, argv);
|
||||
|
||||
if (do_check && ruby_nerrs == 0) {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue