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
tst_module_exists(void (cleanup_fn)(void),const char * mod_name,char ** mod_path)31 void 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 }
51 err = access(buf, F_OK);
52 }
53 /* check start working directory */
54 if (err == -1 && tst_tmpdir_created()) {
55 free(buf);
56 if (asprintf(&buf, "%s/%s", tst_get_startwd(),
57 mod_name) == -1) {
58 tst_brkm(TBROK | TERRNO, cleanup_fn,
59 "asprintf failed at %s:%d",
60 __FILE__, __LINE__);
61 }
62 err = access(buf, F_OK);
63 }
64
65 if (err != 0) {
66 free(buf);
67 tst_brkm(TCONF, cleanup_fn, "Failed to find module '%s'",
68 mod_name);
69 }
70
71 if (mod_path != NULL)
72 *mod_path = buf;
73 else
74 free(buf);
75 }
76
tst_module_load(void (cleanup_fn)(void),const char * mod_name,char * const argv[])77 void tst_module_load(void (cleanup_fn)(void),
78 const char *mod_name, char *const argv[])
79 {
80 char *mod_path = NULL;
81 tst_module_exists(cleanup_fn, mod_name, &mod_path);
82
83 const int offset = 2; /* command name & module path */
84 int size = 0;
85 while (argv && argv[size])
86 ++size;
87 size += offset;
88 const char *mod_argv[size + 1]; /* + NULL in the end */
89 mod_argv[size] = NULL;
90 mod_argv[0] = "insmod";
91 mod_argv[1] = mod_path;
92
93 int i;
94 for (i = offset; i < size; ++i)
95 mod_argv[i] = argv[i - offset];
96
97 tst_run_cmd(cleanup_fn, mod_argv, NULL, NULL, 0);
98 free(mod_path);
99 }
100
tst_module_unload(void (cleanup_fn)(void),const char * mod_name)101 void tst_module_unload(void (cleanup_fn)(void), const char *mod_name)
102 {
103 int i, rc;
104
105 const char *const argv[] = { "rmmod", mod_name, NULL };
106
107 rc = 1;
108 for (i = 0; i < 50; i++) {
109 rc = tst_run_cmd(NULL, argv, "/dev/null", "/dev/null", 1);
110 if (!rc)
111 break;
112
113 usleep(20000);
114 }
115
116 if (rc) {
117 tst_brkm(TBROK, cleanup_fn,
118 "could not unload %s module", mod_name);
119 }
120 }
121