Home | History | Annotate | Download | only in tests 
      1 /*
      2  * Check decoding of setreuid/setregid/setreuid32/setregid32 syscalls.
      3  *
      4  * Copyright (c) 2016 Dmitry V. Levin <ldv (at) altlinux.org>
      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  * 3. The name of the author may not be used to endorse or promote products
     16  *    derived from this software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     19  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     20  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     21  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     23  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     24  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     28  */
     29 
     30 #include <errno.h>
     31 #include <stdio.h>
     32 #include <unistd.h>
     33 
     34 static int
     35 ugid2int(const unsigned UGID_TYPE ugid)
     36 {
     37 	if ((unsigned UGID_TYPE) -1U == ugid)
     38 		return -1;
     39 	else
     40 		return ugid;
     41 }
     42 
     43 static void
     44 print_int(const unsigned int num)
     45 {
     46 	if (num == -1U)
     47 		printf("-1");
     48 	else
     49 		printf("%u", num);
     50 }
     51 
     52 static int
     53 num_matches_id(const unsigned int num, const unsigned int ugid)
     54 {
     55 	return num == ugid || num == -1U;
     56 }
     57 
     58 #define PAIR(val)	{ val, ugid }, { ugid, val }
     59 
     60 int
     61 main(void)
     62 {
     63 	unsigned int ugid = GETUGID;
     64 	CHECK_OVERFLOWUGID(ugid);
     65 
     66 	const struct {
     67 		const long r, e;
     68 	} tests[] = {
     69 		{ ugid, ugid },
     70 		PAIR((unsigned long) 0xffffffff00000000ULL | ugid),
     71 		PAIR(-1U),
     72 		PAIR(-1L),
     73 		PAIR(0xffff0000U | ugid),
     74 		PAIR(0xffff),
     75 		PAIR(0xc0deffffU)
     76 	};
     77 
     78 	unsigned int i;
     79 
     80 	for (i = 0; i < ARRAY_SIZE(tests); ++i) {
     81 		const unsigned int rn = ugid2int(tests[i].r);
     82 		const unsigned int en = ugid2int(tests[i].e);
     83 
     84 		if (!num_matches_id(rn, ugid) || !num_matches_id(en, ugid))
     85 			continue;
     86 
     87 		if (syscall(SYSCALL_NR, tests[i].r, tests[i].e)) {
     88 			if (!i && ENOSYS == errno) {
     89 				printf("%s(%u, %u) = -1 ENOSYS (%m)\n",
     90 				       SYSCALL_NAME, ugid, ugid);
     91 				break;
     92 			}
     93 			perror_msg_and_fail("%s(%#lx, %#lx)", SYSCALL_NAME,
     94 					    tests[i].r, tests[i].e);
     95 		}
     96 
     97 		printf("%s(", SYSCALL_NAME);
     98 		print_int(rn);
     99 		printf(", ");
    100 		print_int(en);
    101 		printf(") = 0\n");
    102 	}
    103 
    104 	puts("+++ exited with 0 +++");
    105 	return 0;
    106 }
    107