mirror of
https://gitlab.com/procps-ng/procps.git
synced 2024-11-24 02:24:51 +08:00
30986cb22e
malloc and realloc could return NULL when no memory is available. The code doesn't handle errors, so use xmalloc/xrealloc instead. While at it, sync alloclen's type with len's type, so both are ssize_t.
150 lines
3.5 KiB
C
150 lines
3.5 KiB
C
/*
|
|
* pwdx.c - print process working directory
|
|
* Copyright 2004 Nicholas Miell
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <getopt.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "c.h"
|
|
#include "nls.h"
|
|
#include "xalloc.h"
|
|
#include "fileutils.h"
|
|
|
|
static void __attribute__ ((__noreturn__)) usage(FILE * out)
|
|
{
|
|
fputs(USAGE_HEADER, out);
|
|
fprintf(out, _(" %s [options] pid...\n"), program_invocation_short_name);
|
|
fputs(USAGE_OPTIONS, out);
|
|
fputs(USAGE_HELP, out);
|
|
fputs(USAGE_VERSION, out);
|
|
fprintf(out, USAGE_MAN_TAIL("pwdx(1)"));
|
|
|
|
exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
|
|
}
|
|
|
|
int check_pid_argument(char *input)
|
|
{
|
|
int skip = 0;
|
|
long pid;
|
|
char *end = NULL;
|
|
|
|
if (!strncmp("/proc/", input, 6))
|
|
skip = 6;
|
|
errno = 0;
|
|
pid = strtol(input + skip, &end, 10);
|
|
|
|
if (errno || input + skip == end || (end && *end))
|
|
return 1;
|
|
if (pid < 1)
|
|
return 1;
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
int ch;
|
|
int retval = 0, i;
|
|
ssize_t alloclen = 128;
|
|
char *pathbuf;
|
|
|
|
static const struct option longopts[] = {
|
|
{"version", no_argument, 0, 'V'},
|
|
{"help", no_argument, 0, 'h'},
|
|
{NULL, 0, 0, 0}
|
|
};
|
|
|
|
#ifdef HAVE_PROGRAM_INVOCATION_NAME
|
|
program_invocation_name = program_invocation_short_name;
|
|
#endif
|
|
setlocale (LC_ALL, "");
|
|
bindtextdomain(PACKAGE, LOCALEDIR);
|
|
textdomain(PACKAGE);
|
|
atexit(close_stdout);
|
|
|
|
while ((ch = getopt_long(argc, argv, "Vh", longopts, NULL)) != -1)
|
|
switch (ch) {
|
|
case 'V':
|
|
printf(PROCPS_NG_VERSION);
|
|
return EXIT_SUCCESS;
|
|
case 'h':
|
|
usage(stdout);
|
|
default:
|
|
usage(stderr);
|
|
}
|
|
|
|
argc -= optind;
|
|
argv += optind;
|
|
|
|
if (argc == 0)
|
|
usage(stderr);
|
|
|
|
pathbuf = xmalloc(alloclen);
|
|
|
|
for (i = 0; i < argc; i++) {
|
|
char *s;
|
|
ssize_t len, buflen;
|
|
/* Constant 10 is the length of strings "/proc/" + "/cwd" + 1 */
|
|
char *buf;
|
|
buflen = 10 + strlen(argv[i]) + 1;
|
|
buf = xmalloc(buflen);
|
|
|
|
if (check_pid_argument(argv[i]))
|
|
xerrx(EXIT_FAILURE, _("invalid process id: %s"),
|
|
argv[i]);
|
|
/*
|
|
* At this point, all arguments are in the form
|
|
* /proc/NNNN or NNNN, so a simple check based on
|
|
* the first char is possible
|
|
*/
|
|
if (argv[i][0] != '/')
|
|
snprintf(buf, buflen, "/proc/%s/cwd", argv[i]);
|
|
else
|
|
snprintf(buf, buflen, "%s/cwd", argv[i]);
|
|
|
|
/*
|
|
* buf contains /proc/NNNN/cwd symlink name
|
|
* on entry, the target of that symlink on return
|
|
*/
|
|
while ((len = readlink(buf, pathbuf, alloclen)) == alloclen) {
|
|
alloclen *= 2;
|
|
pathbuf = xrealloc(pathbuf, alloclen);
|
|
}
|
|
free(buf);
|
|
|
|
if (len < 0) {
|
|
s = strerror(errno == ENOENT ? ESRCH : errno);
|
|
retval = EXIT_FAILURE;
|
|
fprintf(stderr, "%s: %s\n", argv[i], s);
|
|
continue;
|
|
} else {
|
|
pathbuf[len] = 0;
|
|
s = pathbuf;
|
|
}
|
|
|
|
printf("%s: %s\n", argv[i], s);
|
|
}
|
|
free(pathbuf);
|
|
return retval;
|
|
}
|