← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to testsuite/smbd/sec_ctx_torture.c

  • Committer: Chuck Short
  • Date: 2010-09-28 20:38:39 UTC
  • Revision ID: zulcss@ubuntu.com-20100928203839-pgjulytsi9ue63x1
Initial version

Show diffs side-by-side

added added

removed removed

Lines of Context:
testsuite/smbd/sec_ctx_torture.c
 
1
/* 
 
2
   Unix SMB/Netbios implementation.
 
3
   Version 1.9.
 
4
   Security context tests
 
5
   Copyright (C) Tim Potter 2000
 
6
   
 
7
   This program is free software; you can redistribute it and/or modify
 
8
   it under the terms of the GNU General Public License as published by
 
9
   the Free Software Foundation; either version 3 of the License, or
 
10
   (at your option) any later version.
 
11
   
 
12
   This program is distributed in the hope that it will be useful,
 
13
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
15
   GNU General Public License for more details.
 
16
   
 
17
   You should have received a copy of the GNU General Public License
 
18
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
19
*/
 
20
 
 
21
#include "includes.h"
 
22
#include "sec_ctx_utils.h"
 
23
 
 
24
#define NUM_TESTS 10000
 
25
 
 
26
int main (int argc, char **argv)
 
27
{
 
28
        int seed, level = 0, num_tests = 0;
 
29
 
 
30
        init_sec_ctx();
 
31
 
 
32
        if (argc == 1) {
 
33
                seed = time(NULL);
 
34
        } else {
 
35
                seed = atoi(argv[1]);
 
36
        }
 
37
 
 
38
        printf("seed = %d\n", seed);
 
39
 
 
40
        while(num_tests < NUM_TESTS) {
 
41
                switch (random() % 2) {
 
42
 
 
43
                        /* Push a random context */
 
44
 
 
45
                case 0:
 
46
                        if (level < MAX_SEC_CTX_DEPTH) {
 
47
                                int ngroups;
 
48
                                gid_t *groups;
 
49
 
 
50
                                if (!push_sec_ctx()) {
 
51
                                        printf("FAIL: push random ctx\n");
 
52
                                        return 1;
 
53
                                }
 
54
 
 
55
                                get_random_grouplist(&ngroups, &groups);
 
56
 
 
57
                                set_sec_ctx(random() % 32767, 
 
58
                                            random() % 32767, 
 
59
                                            ngroups, groups);
 
60
 
 
61
                                if (!verify_current_groups(ngroups, 
 
62
                                                           groups)) {
 
63
                                        printf("FAIL: groups did not stick\n");
 
64
                                        return 1;
 
65
                                }
 
66
 
 
67
                                printf("pushed (%d, %d) eff=(%d, %d)\n", 
 
68
                                       getuid(), getgid(), geteuid(), 
 
69
                                       getegid());
 
70
 
 
71
                                level++;
 
72
                                num_tests++;
 
73
 
 
74
                                free(groups);
 
75
                        }
 
76
                        break;
 
77
 
 
78
                        /* Pop a random context */
 
79
 
 
80
                case 1:
 
81
                        if (level > 0) {
 
82
                                if (!pop_sec_ctx()) {
 
83
                                        printf("FAIL: pop random ctx\n");
 
84
                                        return 1;
 
85
                                }
 
86
 
 
87
                                printf("popped (%d, %d) eff=(%d, %d)\n", 
 
88
                                       getuid(), getgid(), geteuid(), 
 
89
                                       getegid());
 
90
 
 
91
                                level--;
 
92
                                num_tests++;                    
 
93
                        }
 
94
                        break;
 
95
                }
 
96
        }
 
97
 
 
98
        /* Everything's cool */
 
99
 
 
100
        printf("PASS\n");
 
101
        return 0;
 
102
}