tests/helpers/testsuite_tools.h - master - githubmirror/libreport

tests/helpers/testsuite_tools.h
87 строк3 KB
Michal Fabik
Replace libreport_xstrndup with g_strndup
17 июн 2020, 13:45
17 июн 2020, 13:45df386b0
/*
    Copyright (C) 2015  ABRT team <crash-catcher@lists.fedorahosted.org>
    Copyright (C) 2015  RedHat 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 of the License, 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, write to the Free Software Foundation, Inc.,
    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

    ----

    Helper functions
*/

#include "testsuite.h"

/* Creates a new dump directory in a new temporary directory
 *
 * @param uid Owner's uid
 * @param mode Dump dir mode or -1 for the sane default value 0640.
 * @param ts_falgs Unused
 */
static struct dump_dir *testsuite_dump_dir_create(uid_t uid, mode_t mode, int ts_flags)
{
    char dump_dir_name[] = "/tmp/XXXXXX/dump_dir";

    char *last_slash = strrchr(dump_dir_name, '/');
    *last_slash = '\0';

    if (mkdtemp(dump_dir_name) == NULL) {
        perror("mkdtemp()");
        abort();
    }

    fprintf(stdout, "Test temp directory: %s\n", dump_dir_name);
    fflush(stdout);

    *last_slash = '/';

    gid_t dd_default_gid_bck = dd_g_fs_group_gid;

    if (getuid() != 0 && dd_g_fs_group_gid == (gid_t)-1) {
        /* dd_create changes fsgid of the resulting directory and its files to
         * abrt's gid. However, this does not work for unprivileged users.
         * It is quiet common that test suites are not run under root user.
         * We can leave this hack to users of the testsuite API but we believe
         * that this situation is so common, that it is better to that
         * automatically.
         *
         * If dd_g_fs_group_id == -1, then someone has changed intentionally
         * and we should not touch it.
         */
        dd_g_fs_group_gid = getgid();
    }

    struct dump_dir *dd = dd_create(dump_dir_name, uid, mode == (mode_t)-1 ? 0640 : mode);
    assert(dd != NULL);

    dd_g_fs_group_gid = dd_default_gid_bck;

    return dd;
}

/* Removes the dump directory in and the temporary directory
 *
 * See testsuite_dump_dir_create()
 */
static void testsuite_dump_dir_delete(struct dump_dir *dd)
{
    g_autofree char *tmp_dir = g_strndup(dd->dd_dirname, strrchr(dd->dd_dirname, '/') - dd->dd_dirname);
    assert(dd_delete(dd) == 0);

    if(rmdir(tmp_dir) != 0)
    {
        perror("rmdir()");
        abort();
    }
}