freebsd-skq/usr.bin/fstat/fuser.c
Pedro F. Giffuni 1de7b4b805 various: general adoption of SPDX licensing ID tags.
Mainly focus on files that use BSD 2-Clause license, however the tool I
was using misidentified many licenses so this was mostly a manual - error
prone - task.

The Software Package Data Exchange (SPDX) group provides a specification
to make it easier for automated tools to detect and summarize well known
opensource licenses. We are gradually adopting the specification, noting
that the tags are considered only advisory and do not, in any way,
superceed or replace the license texts.

No functional change intended.
2017-11-27 15:37:16 +00:00

371 lines
8.5 KiB
C

/*-
* SPDX-License-Identifier: BSD-2-Clause-FreeBSD
*
* Copyright (c) 2005-2009 Stanislav Sedov <stas@FreeBSD.org>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
*/
#include <sys/cdefs.h>
__FBSDID("$FreeBSD$");
#include <sys/queue.h>
#include <sys/stat.h>
#include <sys/sysctl.h>
#include <sys/user.h>
#include <assert.h>
#include <ctype.h>
#include <err.h>
#include <fcntl.h>
#include <libprocstat.h>
#include <limits.h>
#include <paths.h>
#include <pwd.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <unistd.h>
#include "functions.h"
/*
* File access mode flags table.
*/
static const struct {
int flag;
char ch;
} fflags[] = {
{PS_FST_FFLAG_WRITE, 'w'},
{PS_FST_FFLAG_APPEND, 'a'},
{PS_FST_FFLAG_DIRECT, 'd'},
{PS_FST_FFLAG_SHLOCK, 's'},
{PS_FST_FFLAG_EXLOCK, 'e'}
};
#define NFFLAGS (sizeof(fflags) / sizeof(*fflags))
/*
* Usage flags translation table.
*/
static const struct {
int flag;
char ch;
} uflags[] = {
{PS_FST_UFLAG_RDIR, 'r'},
{PS_FST_UFLAG_CDIR, 'c'},
{PS_FST_UFLAG_JAIL, 'j'},
{PS_FST_UFLAG_TRACE, 't'},
{PS_FST_UFLAG_TEXT, 'x'},
{PS_FST_UFLAG_MMAP, 'm'},
{PS_FST_UFLAG_CTTY, 'y'}
};
#define NUFLAGS (sizeof(uflags) / sizeof(*uflags))
struct consumer {
pid_t pid;
uid_t uid;
int fd;
int flags;
int uflags;
STAILQ_ENTRY(consumer) next;
};
struct reqfile {
uint32_t fsid;
uint64_t fileid;
const char *name;
STAILQ_HEAD(, consumer) consumers;
};
/*
* Option flags.
*/
#define UFLAG 0x01 /* -u flag: show users */
#define FFLAG 0x02 /* -f flag: specified files only */
#define CFLAG 0x04 /* -c flag: treat as mpoints */
#define MFLAG 0x10 /* -m flag: mmapped files too */
#define KFLAG 0x20 /* -k flag: send signal (SIGKILL by default) */
static int flags = 0; /* Option flags. */
static void printflags(struct consumer *consumer);
static int str2sig(const char *str);
static void usage(void) __dead2;
static int addfile(const char *path, struct reqfile *reqfile);
static void dofiles(struct procstat *procstat, struct kinfo_proc *kp,
struct reqfile *reqfiles, size_t nfiles);
static void
usage(void)
{
fprintf(stderr,
"usage: fuser [-cfhkmu] [-M core] [-N system] [-s signal] file ...\n");
exit(EX_USAGE);
}
static void
printflags(struct consumer *cons)
{
unsigned int i;
assert(cons);
for (i = 0; i < NUFLAGS; i++)
if ((cons->uflags & uflags[i].flag) != 0)
fputc(uflags[i].ch, stderr);
for (i = 0; i < NFFLAGS; i++)
if ((cons->flags & fflags[i].flag) != 0)
fputc(fflags[i].ch, stderr);
}
/*
* Add file to the list.
*/
static int
addfile(const char *path, struct reqfile *reqfile)
{
struct stat sb;
assert(path);
if (stat(path, &sb) != 0) {
warn("%s", path);
return (1);
}
reqfile->fileid = sb.st_ino;
reqfile->fsid = sb.st_dev;
reqfile->name = path;
STAILQ_INIT(&reqfile->consumers);
return (0);
}
int
do_fuser(int argc, char *argv[])
{
struct consumer *consumer;
struct kinfo_proc *p, *procs;
struct procstat *procstat;
struct reqfile *reqfiles;
char *ep, *nlistf, *memf;
int ch, cnt, sig;
unsigned int i, nfiles;
sig = SIGKILL; /* Default to kill. */
nlistf = NULL;
memf = NULL;
while ((ch = getopt(argc, argv, "M:N:cfhkms:u")) != -1)
switch(ch) {
case 'f':
if ((flags & CFLAG) != 0)
usage();
flags |= FFLAG;
break;
case 'c':
if ((flags & FFLAG) != 0)
usage();
flags |= CFLAG;
break;
case 'N':
nlistf = optarg;
break;
case 'M':
memf = optarg;
break;
case 'u':
flags |= UFLAG;
break;
case 'm':
flags |= MFLAG;
break;
case 'k':
flags |= KFLAG;
break;
case 's':
if (isdigit(*optarg)) {
sig = strtol(optarg, &ep, 10);
if (*ep != '\0' || sig < 0 || sig >= sys_nsig)
errx(EX_USAGE, "illegal signal number" ": %s",
optarg);
} else {
sig = str2sig(optarg);
if (sig < 0)
errx(EX_USAGE, "illegal signal name: "
"%s", optarg);
}
break;
case 'h':
/* PASSTHROUGH */
default:
usage();
/* NORETURN */
}
argv += optind;
argc -= optind;
assert(argc >= 0);
if (argc == 0)
usage();
/* NORETURN */
/*
* Process named files.
*/
reqfiles = malloc(argc * sizeof(struct reqfile));
if (reqfiles == NULL)
err(EX_OSERR, "malloc()");
nfiles = 0;
while (argc--)
if (!addfile(*(argv++), &reqfiles[nfiles]))
nfiles++;
if (nfiles == 0)
errx(EX_IOERR, "files not accessible");
if (memf != NULL)
procstat = procstat_open_kvm(nlistf, memf);
else
procstat = procstat_open_sysctl();
if (procstat == NULL)
errx(1, "procstat_open()");
procs = procstat_getprocs(procstat, KERN_PROC_PROC, 0, &cnt);
if (procs == NULL)
errx(1, "procstat_getprocs()");
/*
* Walk through process table and look for matching files.
*/
p = procs;
while(cnt--)
if (p->ki_stat != SZOMB)
dofiles(procstat, p++, reqfiles, nfiles);
for (i = 0; i < nfiles; i++) {
fprintf(stderr, "%s:", reqfiles[i].name);
fflush(stderr);
STAILQ_FOREACH(consumer, &reqfiles[i].consumers, next) {
if (consumer->flags != 0) {
fprintf(stdout, "%6d", consumer->pid);
fflush(stdout);
printflags(consumer);
if ((flags & UFLAG) != 0)
fprintf(stderr, "(%s)",
user_from_uid(consumer->uid, 0));
if ((flags & KFLAG) != 0)
kill(consumer->pid, sig);
fflush(stderr);
}
}
(void)fprintf(stderr, "\n");
}
procstat_freeprocs(procstat, procs);
procstat_close(procstat);
free(reqfiles);
return (0);
}
static void
dofiles(struct procstat *procstat, struct kinfo_proc *kp,
struct reqfile *reqfiles, size_t nfiles)
{
struct vnstat vn;
struct consumer *cons;
struct filestat *fst;
struct filestat_list *head;
int error, match;
unsigned int i;
char errbuf[_POSIX2_LINE_MAX];
head = procstat_getfiles(procstat, kp, flags & MFLAG);
if (head == NULL)
return;
STAILQ_FOREACH(fst, head, next) {
if (fst->fs_type != PS_FST_TYPE_VNODE)
continue;
error = procstat_get_vnode_info(procstat, fst, &vn, errbuf);
if (error != 0)
continue;
for (i = 0; i < nfiles; i++) {
if (flags & CFLAG && reqfiles[i].fsid == vn.vn_fsid) {
break;
}
else if (reqfiles[i].fsid == vn.vn_fsid &&
reqfiles[i].fileid == vn.vn_fileid) {
break;
}
else if (!(flags & FFLAG) &&
(vn.vn_type == PS_FST_VTYPE_VCHR ||
vn.vn_type == PS_FST_VTYPE_VBLK) &&
vn.vn_fsid == reqfiles[i].fileid) {
break;
}
}
if (i == nfiles)
continue; /* No match. */
/*
* Look for existing entries.
*/
match = 0;
STAILQ_FOREACH(cons, &reqfiles[i].consumers, next)
if (cons->pid == kp->ki_pid) {
match = 1;
break;
}
if (match == 1) { /* Use old entry. */
cons->flags |= fst->fs_fflags;
cons->uflags |= fst->fs_uflags;
} else {
/*
* Create new entry in the consumer chain.
*/
cons = calloc(1, sizeof(struct consumer));
if (cons == NULL) {
warn("malloc()");
continue;
}
cons->uid = kp->ki_uid;
cons->pid = kp->ki_pid;
cons->uflags = fst->fs_uflags;
cons->flags = fst->fs_fflags;
STAILQ_INSERT_TAIL(&reqfiles[i].consumers, cons, next);
}
}
procstat_freefiles(procstat, head);
}
/*
* Returns signal number for it's string representation.
*/
static int
str2sig(const char *str)
{
int i;
if (!strncasecmp(str, "SIG", 3))
str += 3;
for (i = 1; i < sys_nsig; i++) {
if (!strcasecmp(sys_signame[i], str))
return (i);
}
return (-1);
}