Merge from vendor branch AWK:
[dragonfly.git] / tools / regression / p1003_1b / memlock.c
1 /*
2  * Copyright (c) 1996 - 1999
3  *      HD Associates, Inc.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by HD Associates, Inc
16  * 4. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY HD ASSOCIATES AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL HD ASSOCIATES OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  *
32  * $FreeBSD: src/tools/regression/p1003_1b/memlock.c,v 1.1 2000/02/16 14:28:40 dufault Exp $
33  * $DragonFly: src/tools/regression/p1003_1b/memlock.c,v 1.2 2003/06/17 04:29:11 dillon Exp $
34  */
35 #include <unistd.h>
36 #include <stdio.h>
37 #include <errno.h>
38 #include <sys/mman.h>
39
40 #include "prutil.h"
41
42 int memlock(int argc, char *argv[])
43 {
44         int e = 0;
45
46         /* Is memory locking configured?
47          */
48         errno = 0;
49         if (sysconf(_SC_MEMLOCK) == -1) {
50                 if (errno != 0) {
51                         /* This isn't valid - may be a standard violation
52                          */
53                         quit("(should not happen) sysconf(_SC_MEMLOCK)");
54                 }
55                 else {
56                         fprintf(stderr,
57                         "Memory locking is not supported in this environment.\n");
58                         e = -1;
59                 }
60         }
61
62         /* Lock yourself in memory:
63          */
64         if (mlockall(MCL_CURRENT | MCL_FUTURE) == -1) {
65                 perror("mlockall(MCL_CURRENT | MCL_FUTURE)");
66                 e = errno;
67         }
68         else if (munlockall() == -1) {
69                 perror("munlockall");
70                 e = errno;
71         }
72
73         return e;
74 }
75
76 #ifdef NO_MEMLOCK
77 int mlockall(int flags)
78 {
79         return EOPNOTSUPP;
80 }
81
82 int munlockall(void)
83 {
84         return EOPNOTSUPP;
85 }
86         
87
88 #endif
89
90 #ifdef STANDALONE_TESTS
91 int main(int argc, char *argv[]) { return memlock(argc, argv); }
92 #endif