blob: 0be6ced0f547b6e4bb2bb9bcd5b00369f61e386b [file] [log] [blame]
/*
* Copyright (c) 2013 Oracle and/or its affiliates. All Rights Reserved.
*
* 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 would 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 the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* Author: Alexey Kodanev <alexey.kodanev@oracle.com>
*
*/
#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "test.h"
#include "ltp_priv.h"
#include "tst_module.h"
void tst_module_exists(void (cleanup_fn)(void),
const char *mod_name, char **mod_path)
{
/* check current working directory */
if (access(mod_name, F_OK) == 0) {
if (mod_path != NULL)
*mod_path = strdup(mod_name);
return;
}
char *buf = NULL;
int err = -1;
/* check LTP installation path */
const char *ltproot = getenv("LTPROOT");
if (ltproot != NULL) {
if (asprintf(&buf, "%s/testcases/bin/%s",
ltproot, mod_name) == -1) {
tst_brkm(TBROK | TERRNO, cleanup_fn,
"asprintf failed at %s:%d",
__FILE__, __LINE__);
}
err = access(buf, F_OK);
}
/* check start working directory */
if (err == -1 && tst_tmpdir_created()) {
free(buf);
if (asprintf(&buf, "%s/%s", tst_get_startwd(),
mod_name) == -1) {
tst_brkm(TBROK | TERRNO, cleanup_fn,
"asprintf failed at %s:%d",
__FILE__, __LINE__);
}
err = access(buf, F_OK);
}
if (err != 0) {
free(buf);
tst_brkm(TCONF, cleanup_fn, "Failed to find module '%s'",
mod_name);
}
if (mod_path != NULL)
*mod_path = buf;
else
free(buf);
}
void tst_module_load(void (cleanup_fn)(void),
const char *mod_name, char *const argv[])
{
char *mod_path = NULL;
tst_module_exists(cleanup_fn, mod_name, &mod_path);
const int offset = 2; /* command name & module path */
int size = 0;
while (argv && argv[size])
++size;
size += offset;
const char *mod_argv[size + 1]; /* + NULL in the end */
mod_argv[size] = NULL;
mod_argv[0] = "insmod";
mod_argv[1] = mod_path;
int i;
for (i = offset; i < size; ++i)
mod_argv[i] = argv[i - offset];
tst_run_cmd(cleanup_fn, mod_argv, NULL, NULL, 0);
free(mod_path);
}
void tst_module_unload(void (cleanup_fn)(void), const char *mod_name)
{
int i, rc;
const char *const argv[] = { "rmmod", mod_name, NULL };
rc = 1;
for (i = 0; i < 50; i++) {
rc = tst_run_cmd(NULL, argv, "/dev/null", "/dev/null", 1);
if (!rc)
break;
usleep(20000);
}
if (rc) {
tst_brkm(TBROK, cleanup_fn,
"could not unload %s module", mod_name);
}
}