r24872: Use torture API a bit more
[kai/samba.git] / source4 / torture / util.c
1 /* 
2    Unix SMB/CIFS implementation.
3    SMB torture tester utility functions
4    Copyright (C) Jelmer Vernooij 2006
5    
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 3 of the License, or
9    (at your option) any later version.
10    
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "includes.h"
21 #include "system/filesys.h"
22 #include "system/wait.h"
23 #include "torture/torture.h"
24 #include "libcli/raw/interfaces.h"
25 #include "libcli/raw/libcliraw.h"
26
27 /**
28  create a temporary directory.
29 */
30 _PUBLIC_ NTSTATUS torture_temp_dir(struct torture_context *tctx, 
31                                                                    const char *prefix, 
32                                                                    char **tempdir)
33 {
34         *tempdir = talloc_asprintf(tctx, "%s/%s.XXXXXX", tctx->outputdir, prefix);
35
36         if (mkdtemp(*tempdir) == NULL)
37                 return NT_STATUS_UNSUCCESSFUL;
38
39         return NT_STATUS_OK;
40 }
41
42 /**
43   check if 2 NTTIMEs are equal.
44 */
45 BOOL nt_time_equal(NTTIME *t1, NTTIME *t2)
46 {
47         return *t1 == *t2;
48 }
49
50 NTSTATUS torture_second_tcon(TALLOC_CTX *mem_ctx,
51                              struct smbcli_session *session,
52                              const char *sharename,
53                              struct smbcli_tree **res)
54 {
55         union smb_tcon tcon;
56         struct smbcli_tree *result;
57         TALLOC_CTX *tmp_ctx;
58         NTSTATUS status;
59
60         if ((tmp_ctx = talloc_new(mem_ctx)) == NULL) {
61                 return NT_STATUS_NO_MEMORY;
62         }
63
64         result = smbcli_tree_init(session, tmp_ctx, False);
65         if (result == NULL) {
66                 talloc_free(tmp_ctx);
67                 return NT_STATUS_NO_MEMORY;
68         }
69
70         tcon.generic.level = RAW_TCON_TCONX;
71         tcon.tconx.in.flags = 0;
72
73         /* Ignore share mode security here */
74         tcon.tconx.in.password = data_blob(NULL, 0);
75         tcon.tconx.in.path = sharename;
76         tcon.tconx.in.device = "?????";
77
78         status = smb_raw_tcon(result, tmp_ctx, &tcon);
79         if (!NT_STATUS_IS_OK(status)) {
80                 talloc_free(tmp_ctx);
81                 return status;
82         }
83
84         result->tid = tcon.tconx.out.tid;
85         *res = talloc_steal(mem_ctx, result);
86         talloc_free(tmp_ctx);
87         return NT_STATUS_OK;
88 }