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/cellconfig.h>
  34 #include <afs/afsutil.h>
  35
  36 #include <hcrypto/des.h>
  37
  38 #include "common.h"
  39
  40 static FILE *
  41 openConfigFile(char *dirname, char *filename) {
  42     char *path = NULL;
  43     FILE *file;
  44
  45     if (asprintf(&path, "%s/%s", dirname, filename) == -1)
  46         return NULL;
  47
  48     file = fopen(path, "w");
  49     free(path);
  50     return file;
  51 }
  52
  53 static void
  54 unlinkConfigFile(char *dirname, char *filename) {
  55     char *path;
  56
  57     if (asprintf(&path, "%s/%s", dirname, filename) != -1) {
  58         unlink(path);
  59         free(path);
  60     }
  61 }
  62
  63 /*!
  64  * Build a test configuration directory, containing a CellServDB and ThisCell
  65  * file for the "example.org" cell
  66  *
  67  * @return
  68  * The path to the configuration directory. This should be freed by the caller
  69  * using free()
  70  *
  71  */
  72
  73 char *
  74 afstest_BuildTestConfig(void) {
  75     char *dir = NULL;
  76     FILE *file;
  77
  78     if (asprintf(&dir, "%s/afs_XXXXXX", gettmpdir()) == -1)
  79         goto fail;
  80
  81     if (mkdtemp(dir) == NULL)
  82         goto fail;
  83
  84     /* Create a CellServDB */
  85     file = openConfigFile(dir, "CellServDB");
  86     fprintf(file, ">example.org # An example cell\n");
  87     fprintf(file, "127.0.0.1 #test.example.org\n");
  88     fclose(file);
  89
  90     /* Create a ThisCell file */
  91     file = openConfigFile(dir, "ThisCell");
  92     fprintf(file, "example.org");
  93     fclose(file);
  94
  95     return dir;
  96
  97 fail:
  98     if (dir)
  99         free(dir);
 100     return NULL;
 101 }
 102
 103 /*!
 104  * Delete at test configuration directory
 105  */
 106
 107 void
 108 afstest_UnlinkTestConfig(char *dir)
 109 {
 110     unlinkConfigFile(dir, "KeyFile");
 111     unlinkConfigFile(dir, "CellServDB");
 112     unlinkConfigFile(dir, "ThisCell");
 113     unlinkConfigFile(dir, "UserList");
 114     rmdir(dir);
 115 }
 116
 117 int
 118 afstest_AddDESKeyFile(struct afsconf_dir *dir)
 119 {
 120     char keymaterial[]="\x19\x17\xff\xe6\xbb\x77\x2e\xfc";
 121
 122     /* Make sure that it is actually a valid key */
 123     DES_set_odd_parity((DES_cblock *)keymaterial);
 124
 125     return afsconf_AddKey(dir, 1, keymaterial, 1);
 126 }