xref: /third_party/ltp/lib/tst_module.c (revision f08c3bdf)
1/*
2 * Copyright (c) 2013 Oracle and/or its affiliates. All Rights Reserved.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License as
6 * published by the Free Software Foundation; either version 2 of
7 * the License, or (at your option) any later version.
8 *
9 * This program is distributed in the hope that it would be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write the Free Software Foundation,
16 * Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17 *
18 * Author: Alexey Kodanev <alexey.kodanev@oracle.com>
19 *
20 */
21
22#define _GNU_SOURCE
23#include <stdio.h>
24#include <stdlib.h>
25#include <unistd.h>
26
27#include "test.h"
28#include "ltp_priv.h"
29#include "old_module.h"
30
31void tst_module_exists_(void (cleanup_fn)(void),
32	const char *mod_name, char **mod_path)
33{
34	/* check current working directory */
35	if (access(mod_name, F_OK) == 0) {
36		if (mod_path != NULL)
37			*mod_path = strdup(mod_name);
38		return;
39	}
40	char *buf = NULL;
41	int err = -1;
42	/* check LTP installation path */
43	const char *ltproot = getenv("LTPROOT");
44	if (ltproot != NULL) {
45		if (asprintf(&buf, "%s/testcases/bin/%s",
46			ltproot, mod_name) == -1) {
47			tst_brkm(TBROK | TERRNO, cleanup_fn,
48				"asprintf failed at %s:%d",
49				__FILE__, __LINE__);
50			return;
51		}
52		err = access(buf, F_OK);
53	}
54	/* check start working directory */
55	if (err == -1 && tst_tmpdir_created()) {
56		free(buf);
57		if (asprintf(&buf, "%s/%s", tst_get_startwd(),
58			mod_name) == -1) {
59			tst_brkm(TBROK | TERRNO, cleanup_fn,
60				"asprintf failed at %s:%d",
61				__FILE__, __LINE__);
62			return;
63		}
64		err = access(buf, F_OK);
65	}
66
67	if (err != 0) {
68		free(buf);
69		tst_brkm(TCONF, cleanup_fn, "Failed to find module '%s'",
70			mod_name);
71		return;
72	}
73
74	if (mod_path != NULL)
75		*mod_path = buf;
76	else
77		free(buf);
78}
79
80void tst_module_load_(void (cleanup_fn)(void),
81	const char *mod_name, char *const argv[])
82{
83	char *mod_path = NULL;
84	tst_module_exists_(cleanup_fn, mod_name, &mod_path);
85
86	const int offset = 2; /* command name & module path */
87	int size = 0;
88	while (argv && argv[size])
89		++size;
90	size += offset;
91	const char *mod_argv[size + 1]; /* + NULL in the end */
92	mod_argv[size] = NULL;
93	mod_argv[0] = "insmod";
94	mod_argv[1] = mod_path;
95
96	int i;
97	for (i = offset; i < size; ++i)
98		mod_argv[i] = argv[i - offset];
99
100	tst_cmd(cleanup_fn, mod_argv, NULL, NULL, 0);
101	free(mod_path);
102}
103
104void tst_module_unload_(void (cleanup_fn)(void), const char *mod_name)
105{
106	int i, rc;
107
108	const char *const argv[] = { "rmmod", mod_name, NULL };
109
110	rc = 1;
111	for (i = 0; i < 50; i++) {
112		rc = tst_cmd(NULL, argv, "/dev/null", "/dev/null",
113				 TST_CMD_PASS_RETVAL);
114		if (!rc)
115			break;
116
117		usleep(20000);
118	}
119
120	if (rc) {
121		tst_brkm(TBROK, cleanup_fn,
122			 "could not unload %s module", mod_name);
123	}
124}
125