1 /*        $NetBSD: ioperm.c,v 1.2 2008/01/04 16:18:52 yamt Exp $      */
2 
3 /*-
4  * Copyright (c)2008 YAMAMOTO Takashi,
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28 
29 #include <sys/cdefs.h>
30 #ifndef lint
31 __RCSID("$NetBSD: ioperm.c,v 1.2 2008/01/04 16:18:52 yamt Exp $");
32 #endif /* not lint */
33 
34 #include <err.h>
35 #include <signal.h>
36 #include <setjmp.h>
37 #include <stdbool.h>
38 #include <stdint.h>
39 #include <stdio.h>
40 #include <stdlib.h>
41 #include <string.h>
42 #include <unistd.h>
43 
44 int i386_get_ioperm(unsigned long *);
45 int i386_set_ioperm(unsigned long *);
46 
47 /* arbitrary port to test */
48 #define   IO_TIMER1 0x40
49 #define   TIMER_CNTR0         0
50 #define   PORT      (IO_TIMER1+TIMER_CNTR0)
51 
52 static uint8_t
inb(uint16_t port)53 inb(uint16_t port)
54 {
55           uint8_t data;
56 
57           __asm __volatile("inb %1, %0" : "=a"(data) : "id"(port));
58           return data;
59 }
60 
61 sigjmp_buf env;
62 int gotsig;
63 
64 static void
sighandler(int sig)65 sighandler(int sig)
66 {
67 
68           siglongjmp(env, sig);
69 }
70 
71 static void
try(const char * msg,bool success)72 try(const char *msg, bool success)
73 {
74           int sig;
75 
76           sig = sigsetjmp(env, 1);
77           if (sig == 0) {
78                     inb(PORT);
79                     if (!success) {
80                               errx(EXIT_FAILURE, "%s: unexpected success of inb",
81                                   msg);
82                     }
83                     return;
84           }
85           if (success) {
86                     errx(EXIT_FAILURE, "%s: got signal %d\n", msg, sig);
87           }
88 }
89 
90 int
main(int argc,char * argv[])91 main(int argc, char *argv[])
92 {
93           unsigned long buf[32];
94           unsigned long buf2[32];
95           int ret;
96           int i;
97 
98           ret = i386_get_ioperm(buf);
99           if (ret == -1) {
100                     err(EXIT_FAILURE, "get_ioperm 1");
101           }
102           for (i = 0; i < __arraycount(buf); i++) {
103                     if (buf[i] != 0xffffffff) {
104                               errx(EXIT_FAILURE, "buf[%d] == 0x%lx", i, buf[i]);
105                     }
106           }
107 
108           signal(SIGSEGV, sighandler);
109 
110           try("1", false);
111           sleep(1);
112 
113           memset(buf2, 0x00, sizeof(buf2));
114           strcpy((char *)buf2, "foobarbaz");
115           buf2[PORT / 8 / sizeof(unsigned long)] &=
116               ~(1 << (PORT % (8 * sizeof(unsigned long))));
117           ret = i386_set_ioperm(buf2);
118           if (ret == -1) {
119                     err(EXIT_FAILURE, "set_ioperm 1");
120           }
121           ret = i386_get_ioperm(buf);
122           if (ret == -1) {
123                     err(EXIT_FAILURE, "get_ioperm 2");
124           }
125           if (memcmp(buf, buf2, sizeof(buf))) {
126                     errx(EXIT_FAILURE, "iomap mismatch");
127           }
128 
129           sleep(1);
130           try("2", true);
131           sleep(1);
132 
133           buf2[PORT / 8 / sizeof(unsigned long)] |=
134               (1 << (PORT % (8 * sizeof(unsigned long))));
135           ret = i386_set_ioperm(buf2);
136           if (ret == -1) {
137                     err(EXIT_FAILURE, "set_ioperm 3");
138           }
139 
140           sleep(1);
141           try("3", false);
142 }
143