1716fd348SMartin Matuska /*
2716fd348SMartin Matuska * CDDL HEADER START
3716fd348SMartin Matuska *
4716fd348SMartin Matuska * The contents of this file are subject to the terms of the
5716fd348SMartin Matuska * Common Development and Distribution License (the "License").
6716fd348SMartin Matuska * You may not use this file except in compliance with the License.
7716fd348SMartin Matuska *
8716fd348SMartin Matuska * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9*271171e0SMartin Matuska * or https://opensource.org/licenses/CDDL-1.0.
10716fd348SMartin Matuska * See the License for the specific language governing permissions
11716fd348SMartin Matuska * and limitations under the License.
12716fd348SMartin Matuska *
13716fd348SMartin Matuska * When distributing Covered Code, include this CDDL HEADER in each
14716fd348SMartin Matuska * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15716fd348SMartin Matuska * If applicable, add the following below this CDDL HEADER, with the
16716fd348SMartin Matuska * fields enclosed by brackets "[]" replaced with your own identifying
17716fd348SMartin Matuska * information: Portions Copyright [yyyy] [name of copyright owner]
18716fd348SMartin Matuska *
19716fd348SMartin Matuska * CDDL HEADER END
20716fd348SMartin Matuska */
21716fd348SMartin Matuska /*
22716fd348SMartin Matuska * Copyright (c) 2019 by Tomohiro Kusumi. All rights reserved.
23716fd348SMartin Matuska */
24716fd348SMartin Matuska
25716fd348SMartin Matuska #include <stdio.h>
26716fd348SMartin Matuska #include <stdlib.h>
27716fd348SMartin Matuska #include <string.h>
28716fd348SMartin Matuska #include <sys/types.h>
29716fd348SMartin Matuska #include <sys/stat.h>
30716fd348SMartin Matuska #include <unistd.h>
31716fd348SMartin Matuska #include <fcntl.h>
32716fd348SMartin Matuska #include <stdbool.h>
33716fd348SMartin Matuska
34716fd348SMartin Matuska int
main(int argc,char * argv[])35716fd348SMartin Matuska main(int argc, char *argv[])
36716fd348SMartin Matuska {
37716fd348SMartin Matuska const char *name, *phase;
38716fd348SMartin Matuska mode_t extra;
39716fd348SMartin Matuska struct stat st;
40716fd348SMartin Matuska
41716fd348SMartin Matuska if (argc < 3) {
42716fd348SMartin Matuska fprintf(stderr, "Invalid argc\n");
43716fd348SMartin Matuska exit(1);
44716fd348SMartin Matuska }
45716fd348SMartin Matuska
46716fd348SMartin Matuska name = argv[1];
47716fd348SMartin Matuska if (strcmp(name, "SUID") == 0) {
48716fd348SMartin Matuska extra = S_ISUID;
49716fd348SMartin Matuska } else if (strcmp(name, "SGID") == 0) {
50716fd348SMartin Matuska extra = S_ISGID;
51716fd348SMartin Matuska } else if (strcmp(name, "SUID_SGID") == 0) {
52716fd348SMartin Matuska extra = S_ISUID | S_ISGID;
53716fd348SMartin Matuska } else if (strcmp(name, "NONE") == 0) {
54716fd348SMartin Matuska extra = 0;
55716fd348SMartin Matuska } else {
56716fd348SMartin Matuska fprintf(stderr, "Invalid name %s\n", name);
57716fd348SMartin Matuska exit(1);
58716fd348SMartin Matuska }
59716fd348SMartin Matuska
60716fd348SMartin Matuska const char *testdir = getenv("TESTDIR");
61716fd348SMartin Matuska if (!testdir) {
62716fd348SMartin Matuska fprintf(stderr, "getenv(TESTDIR)\n");
63716fd348SMartin Matuska exit(1);
64716fd348SMartin Matuska }
65716fd348SMartin Matuska
66716fd348SMartin Matuska umask(0);
67716fd348SMartin Matuska if (stat(testdir, &st) == -1 && mkdir(testdir, 0777) == -1) {
68716fd348SMartin Matuska perror("mkdir");
69716fd348SMartin Matuska exit(2);
70716fd348SMartin Matuska }
71716fd348SMartin Matuska
72716fd348SMartin Matuska char fpath[1024];
73716fd348SMartin Matuska snprintf(fpath, sizeof (fpath), "%s/%s", testdir, name);
74716fd348SMartin Matuska
75716fd348SMartin Matuska
76716fd348SMartin Matuska phase = argv[2];
77716fd348SMartin Matuska if (strcmp(phase, "PRECRASH") == 0) {
78716fd348SMartin Matuska
79716fd348SMartin Matuska /* clean up last run */
80716fd348SMartin Matuska unlink(fpath);
81716fd348SMartin Matuska if (stat(fpath, &st) == 0) {
82716fd348SMartin Matuska fprintf(stderr, "%s exists\n", fpath);
83716fd348SMartin Matuska exit(3);
84716fd348SMartin Matuska }
85716fd348SMartin Matuska
86716fd348SMartin Matuska int fd;
87716fd348SMartin Matuska
88716fd348SMartin Matuska fd = creat(fpath, 0777 | extra);
89716fd348SMartin Matuska if (fd == -1) {
90716fd348SMartin Matuska perror("creat");
91716fd348SMartin Matuska exit(4);
92716fd348SMartin Matuska }
93716fd348SMartin Matuska close(fd);
94716fd348SMartin Matuska
95716fd348SMartin Matuska if (setuid(65534) == -1) {
96716fd348SMartin Matuska perror("setuid");
97716fd348SMartin Matuska exit(5);
98716fd348SMartin Matuska }
99716fd348SMartin Matuska
100716fd348SMartin Matuska fd = open(fpath, O_RDWR);
101716fd348SMartin Matuska if (fd == -1) {
102716fd348SMartin Matuska perror("open");
103716fd348SMartin Matuska exit(6);
104716fd348SMartin Matuska }
105716fd348SMartin Matuska
106716fd348SMartin Matuska const char buf[] = "test";
107716fd348SMartin Matuska if (write(fd, buf, sizeof (buf)) == -1) {
108716fd348SMartin Matuska perror("write");
109716fd348SMartin Matuska exit(7);
110716fd348SMartin Matuska }
111716fd348SMartin Matuska close(fd);
112716fd348SMartin Matuska
113716fd348SMartin Matuska } else if (strcmp(phase, "REPLAY") == 0) {
114716fd348SMartin Matuska /* created in PRECRASH run */
115716fd348SMartin Matuska } else {
116716fd348SMartin Matuska fprintf(stderr, "Invalid phase %s\n", phase);
117716fd348SMartin Matuska exit(1);
118716fd348SMartin Matuska }
119716fd348SMartin Matuska
120716fd348SMartin Matuska if (stat(fpath, &st) == -1) {
121716fd348SMartin Matuska perror("stat");
122716fd348SMartin Matuska exit(8);
123716fd348SMartin Matuska }
124716fd348SMartin Matuska
125716fd348SMartin Matuska /* Verify SUID/SGID are dropped */
126716fd348SMartin Matuska mode_t res = st.st_mode & (0777 | S_ISUID | S_ISGID);
127716fd348SMartin Matuska if (res != 0777) {
128716fd348SMartin Matuska fprintf(stderr, "stat(2) %o\n", res);
129716fd348SMartin Matuska exit(9);
130716fd348SMartin Matuska }
131716fd348SMartin Matuska
132716fd348SMartin Matuska return (0);
133716fd348SMartin Matuska }
134