1170240dddaa440924a66d8fdbfc700154dfa466
[openafs.git] / tests / common / config.c
1 /*
2  * Copyright (c) 2010 Your File System Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR `AS IS'' AND ANY EXPRESS OR
14  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23  */
24
25 /*!
26  * Common functions for building a configuration directory
27  */
28
29 #include <afsconfig.h>
30 #include <afs/param.h>
31 #include <roken.h>
32
33 #include <afs/afsutil.h>
34
35 #include "common.h"
36
37 static FILE *
38 openConfigFile(char *dirname, char *filename) {
39     char *path = NULL;
40     FILE *file;
41
42     if (asprintf(&path, "%s/%s", dirname, filename) == -1)
43         return NULL;
44
45     file = fopen(path, "w");
46     free(path);
47     return file;
48 }
49
50 static void
51 unlinkConfigFile(char *dirname, char *filename) {
52     char *path;
53
54     if (asprintf(&path, "%s/%s", dirname, filename) != -1) {
55         unlink(path);
56         free(path);
57     }
58 }
59
60 /*!
61  * Build a test configuration directory, containing a CellServDB and ThisCell
62  * file for the "example.org" cell
63  *
64  * @return
65  * The path to the configuration directory. This should be freed by the caller
66  * using free()
67  *
68  */
69
70 char *
71 afstest_BuildTestConfig(void) {
72     char *dir = NULL;
73     FILE *file;
74
75     if (asprintf(&dir, "%s/afs_XXXXXX", gettmpdir()) == -1)
76         goto fail;
77
78     if (mkdtemp(dir) == NULL)
79         goto fail;
80
81     /* Create a CellServDB */
82     file = openConfigFile(dir, "CellServDB");
83     fprintf(file, ">example.org # An example cell\n");
84     fprintf(file, "127.0.0.1 #test.example.org\n");
85     fclose(file);
86
87     /* Create a ThisCell file */
88     file = openConfigFile(dir, "ThisCell");
89     fprintf(file, "example.org");
90     fclose(file);
91
92     return dir;
93
94 fail:
95     if (dir)
96         free(dir);
97     return NULL;
98 }
99
100 /*!
101  * Delete at test configuration directory
102  */
103
104 void
105 afstest_UnlinkTestConfig(char *dir)
106 {
107     unlinkConfigFile(dir, "KeyFile");
108     unlinkConfigFile(dir, "CellServDB");
109     unlinkConfigFile(dir, "ThisCell");
110     unlinkConfigFile(dir, "UserList");
111     rmdir(dir);
112 }