tests/LTbasic2.c - master - githubmirror/lsof

tests/LTbasic2.c
160 строк5 KB
Jiajie Chen
[netbsd] skip file name validation in LTbasic2
10 июн 2026, 13:32
10 июн 2026, 13:321db0076
/*
 * LTbasic2.c -- Lsof Test basic tests 2
 *
 * The basic tests measure the finding by liblsof of its own open CWD, open
 * executable (when possible) and opened regular file.
 *
 * V. Abell
 * Purdue University
 */

/*
 * Copyright 2002 Purdue Research Foundation, West Lafayette, Indiana
 * 47907.  All rights reserved.
 *
 * Written by V. Abell.
 *
 * This software is not subject to any license of the American Telephone
 * and Telegraph Company or the Regents of the University of California.
 *
 * Permission is granted to anyone to use this software for any purpose on
 * any computer system, and to alter it and redistribute it freely, subject
 * to the following restrictions:
 *
 * 1. Neither the authors nor Purdue University are responsible for any
 *    consequences of the use of this software.
 *
 * 2. The origin of this software must not be misrepresented, either by
 *    explicit claim or by omission.  Credit to the authors and Purdue
 *    University must appear in documentation and sources.
 *
 * 3. Altered versions must be plainly marked as such, and must not be
 *    misrepresented as being the original software.
 *
 * 4. This notice may not be removed or altered.
 */

#include "lsof.h"
#include <stdio.h>
#include <fcntl.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>

int main(int argc, char **argv) {
    struct lsof_result *result;
    struct lsof_context *ctx;
    struct lsof_process *p;
    struct lsof_file *f;
    int pi, fi;
    char buffer[128];
    struct stat exec_stat;
    struct stat cwd_stat;
    struct stat tmp_stat;
    int fd;
    int round;
    int exec_found; /* executable found in result */
    int cwd_found;  /* cwd found in result */
    int fd_found;   /* opened fd found in result */

    for (round = 0; round < 3; round++) {
        printf("Testing round %d\n", round);

        exec_found = 0;
        cwd_found = 0;
        fd_found = 0;
        fd = -1;
        if (stat(argv[0], &exec_stat)) {
            fprintf(stderr, "Cannot stat %s, skipping executable check\n",
                    argv[0]);
            exec_found = 1;
        }
        if (stat(".", &cwd_stat)) {
            fprintf(stderr, "Cannot stat '.', skipping cwd check\n");
            cwd_found = 1;
        }
        if ((fd = open("LTbasic2-tmp", O_CREAT, 0644)) < 0) {
            fprintf(stderr,
                    "Cannot create 'LTbasic2-tmp' in current directory, "
                    "skipping fd check\n");
            fd_found = 1;
        } else {
            /* on FreeBSD 14.3, the path for newly created file is not returned
             * immediately, so let's trigger it */
            stat("LTbasic2-tmp", &tmp_stat);
        }

        ctx = lsof_new();
        if (round == 0) {
            lsof_select_process(ctx, "LTbasic2", 0);
        } else if (round == 1) {
            lsof_select_process_regex(ctx, "/^LTbasic2$/");
        } else if (round == 2) {
            lsof_select_pid(ctx, getpid(), 0);
        }
        lsof_freeze(ctx);
        lsof_gather(ctx, &result);

        for (pi = 0; pi < result->num_processes; pi++) {
            p = &result->processes[pi];
            for (fi = 0; fi < p->num_files; fi++) {
                f = &p->files[fi];
                if (f->fd_type == LSOF_FD_PROGRAM_TEXT) {
                    /* check if device and inode matches */
                    if ((f->flags & (LSOF_FILE_FLAG_DEV_VALID |
                                     LSOF_FILE_FLAG_INODE_VALID)) &&
                        f->dev == exec_stat.st_dev &&
                        f->inode == exec_stat.st_ino) {
                        exec_found = 1;
                    }
                } else if (f->fd_type == LSOF_FD_CWD) {
                    /* check if device and inode matches */
                    if ((f->flags & (LSOF_FILE_FLAG_DEV_VALID |
                                     LSOF_FILE_FLAG_INODE_VALID)) &&
                        f->dev == cwd_stat.st_dev &&
                        f->inode == cwd_stat.st_ino) {
                        cwd_found = 1;
                    }
                } else if (f->fd_type == LSOF_FD_NUMERIC) {
                    /* check if fd matches */
                    if (f->fd_num == fd && f->inode == tmp_stat.st_ino &&
                        f->dev == tmp_stat.st_dev) {
#ifdef NETBSDV
                        /* on NetBSD, the file path for new created file may be
                         * unavailable, skip validation */
                        fd_found = 1;
#else
                        if (f->name && strstr(f->name, "LTbasic2-tmp")) {
                            fd_found = 1;
                        }
#endif
                    }
                }
            }
        }

        lsof_free_result(result);
        lsof_destroy(ctx);

        if (!exec_found) {
            fprintf(stderr,
                    "ERROR!!!  open LTbasic2 executable wasn't found.\n");
        }
        if (!cwd_found) {
            fprintf(stderr,
                    "ERROR!!!  current working directory wasn't found.\n");
        }
        if (!fd_found) {
            fprintf(stderr, "ERROR!!!  opened regular file wasn't found.\n");
        }
        /* cleanup created temporary file */
        if (fd >= 0) {
            close(fd);
            unlink("LTbasic2-tmp");
        }
        if (!(exec_found && cwd_found && fd_found)) {
            return 1;
        }
    }
    return 0;
}